1 | /* |
2 | * CellDecomposer.hpp |
3 | * |
4 | * Created on: Apr 1, 2015 |
5 | * Author: Pietro Incardona |
6 | */ |
7 | |
8 | #ifndef CELLDECOMPOSER_HPP_ |
9 | #define CELLDECOMPOSER_HPP_ |
10 | |
11 | #include "Space/SpaceBox.hpp" |
12 | #include "Space/Matrix.hpp" |
13 | #include "util/copy_compare/meta_compare.hpp" |
14 | #include "Grid/grid_sm.hpp" |
15 | |
16 | #define CELL_DECOMPOSER 8001lu |
17 | |
18 | |
19 | |
20 | |
21 | /*! This Class apply a shift transformation before converting to Cell-ID |
22 | * |
23 | * |
24 | */ |
25 | template<unsigned int dim, typename T> |
26 | class shift |
27 | { |
28 | //! Shift point |
29 | Point<dim,T> sh; |
30 | |
31 | //! Matrix transformation |
32 | Matrix<dim,T> mat; |
33 | |
34 | public: |
35 | |
36 | /*! \brief Constructor |
37 | * |
38 | * \param t Matrix transformation |
39 | * \param s shift |
40 | * |
41 | */ |
42 | __device__ __host__ shift(const Matrix<dim,T> & t, const Point<dim,T> & s) |
43 | :sh(s) |
44 | { |
45 | mat.identity(); |
46 | } |
47 | |
48 | /*! \brief Shift the point transformation |
49 | * |
50 | * \param s source point |
51 | * \param i coordinate |
52 | * |
53 | * \return the transformed coordinate |
54 | * |
55 | */ |
56 | __device__ __host__ inline T transform(const T(&s)[dim], const size_t i) const |
57 | { |
58 | return s[i] - sh.get(i); |
59 | } |
60 | |
61 | /*! \brief Shift the point transformation |
62 | * |
63 | * \param s source point |
64 | * \param i coordinate |
65 | * |
66 | * \return the transformed coordinate |
67 | * |
68 | */ |
69 | __device__ __host__ inline T transform(const Point<dim,T> & s, const size_t i) const |
70 | { |
71 | return s.get(i) - sh.get(i); |
72 | } |
73 | |
74 | /*! \brief Shift the point transformation |
75 | * |
76 | * \param s source point |
77 | * \param i coordinate |
78 | * |
79 | * \return the transformed coordinate |
80 | * |
81 | */ |
82 | template<typename Mem> inline __device__ __host__ T transform(const encapc<1,Point<dim,T>,Mem> & s, const size_t i) const |
83 | { |
84 | return s.template get<0>()[i] - sh.get(i); |
85 | } |
86 | |
87 | /*! \brief Set the transformation Matrix and shift |
88 | * |
89 | * \param mat Matrix transformation |
90 | * \param orig origin point |
91 | * |
92 | */ |
93 | __device__ __host__ inline void setTransform(const Matrix<dim,T> & mat, const Point<dim,T> & orig) |
94 | { |
95 | for (size_t i = 0 ; i < dim ; i++) |
96 | sh.get(i) = orig.get(i); |
97 | } |
98 | |
99 | /*! \brief Get the shift vector |
100 | * |
101 | * \return the shift vector |
102 | * |
103 | */ |
104 | inline const Point<dim,T> & getOrig() const |
105 | { |
106 | return sh; |
107 | } |
108 | |
109 | /*! \brief Get the transformation Matrix |
110 | * |
111 | * \return the transformation matrix |
112 | * |
113 | */ |
114 | inline const Matrix<dim,T> & getMat() const |
115 | { |
116 | return mat; |
117 | } |
118 | |
119 | /*! \brief It return true if the shift match |
120 | * |
121 | * \param s shift to compare with |
122 | * |
123 | * \return true if it match |
124 | * |
125 | */ |
126 | inline bool operator==(const shift<dim,T> & s) |
127 | { |
128 | return sh == s.sh; |
129 | } |
130 | |
131 | /*! \brief It return true if the shift is different |
132 | * |
133 | * \param s shift to compare with |
134 | * |
135 | * \return true if the shift is different |
136 | * |
137 | */ |
138 | inline bool operator!=(const shift<dim,T> & s) |
139 | { |
140 | return !this->operator==(s); |
141 | } |
142 | }; |
143 | |
144 | /*! This Class apply a shift transformation before converting to Cell-ID |
145 | * |
146 | * |
147 | */ |
148 | template<unsigned int dim, typename T> |
149 | class shift_only |
150 | { |
151 | //! Shift point |
152 | Point<dim,T> sh; |
153 | |
154 | public: |
155 | |
156 | /*! \brief Default constructor |
157 | * |
158 | */ |
159 | shift_only() |
160 | { |
161 | sh.zero(); |
162 | } |
163 | |
164 | /*! \brief Constructor |
165 | * |
166 | * \param t Matrix transformation |
167 | * \param s shift |
168 | * |
169 | */ |
170 | shift_only(const Matrix<dim,T> & t, const Point<dim,T> & s) |
171 | :sh(s) |
172 | { |
173 | } |
174 | |
175 | /*! \brief Shift the point transformation |
176 | * |
177 | * \param s source point |
178 | * \param i coordinate |
179 | * |
180 | * \return the transformed coordinate |
181 | * |
182 | */ |
183 | __device__ __host__ inline T transform(const T * s, const int i) const |
184 | { |
185 | return s[i] - sh.get(i); |
186 | } |
187 | |
188 | /*! \brief Shift the point transformation |
189 | * |
190 | * \param s source point |
191 | * \param i coordinate |
192 | * |
193 | * \return the transformed coordinate |
194 | * |
195 | */ |
196 | __device__ __host__ inline T transform(const T(&s)[dim], const int i) const |
197 | { |
198 | return s[i] - sh.get(i); |
199 | } |
200 | |
201 | /*! \brief Shift the point transformation |
202 | * |
203 | * \param s source point |
204 | * \param i coordinate |
205 | * |
206 | * \return the transformed coordinate |
207 | * |
208 | */ |
209 | __device__ __host__ inline T transform(const Point<dim,T> & s, const int i) const |
210 | { |
211 | return s.get(i) - sh.get(i); |
212 | } |
213 | |
214 | /*! \brief Shift the point transformation |
215 | * |
216 | * \param s source point |
217 | * \param i coordinate |
218 | * |
219 | * \return the transformed coordinate |
220 | * |
221 | */ |
222 | template<typename Mem> __device__ __host__ inline T transform(const encapc<1,Point<dim,T>,Mem> & s, const int i) const |
223 | { |
224 | return s.template get<0>()[i] - sh.get(i); |
225 | } |
226 | |
227 | /*! \brief Set the transformation Matrix and shift |
228 | * |
229 | * \param mat Matrix transformation |
230 | * \param orig origin point |
231 | * |
232 | */ |
233 | inline void setTransform(const Matrix<dim,T> & mat, const Point<dim,T> & orig) |
234 | { |
235 | for (size_t i = 0 ; i < dim ; i++) |
236 | sh.get(i) = orig.get(i); |
237 | } |
238 | |
239 | /*! \brief Get the shift vector |
240 | * |
241 | * \return the shift vector |
242 | * |
243 | */ |
244 | __device__ __host__ inline const Point<dim,T> & getOrig() const |
245 | { |
246 | return sh; |
247 | } |
248 | |
249 | |
250 | /*! \brief It return true if the shift match |
251 | * |
252 | * \param s shift to compare with |
253 | * |
254 | * \return true if it match |
255 | * |
256 | */ |
257 | inline bool operator==(const shift<dim,T> & s) |
258 | { |
259 | return sh == s.sh; |
260 | } |
261 | |
262 | /*! \brief It return true if the shift is different |
263 | * |
264 | * \param s shift to compare with |
265 | * |
266 | * \return true if the shift is different |
267 | * |
268 | */ |
269 | inline bool operator!=(const shift<dim,T> & s) |
270 | { |
271 | return !this->operator==(s); |
272 | } |
273 | }; |
274 | |
275 | //! No transformation |
276 | template<unsigned int dim, typename T> |
277 | class no_transform |
278 | { |
279 | //! shift transform |
280 | Point<dim,T> orig; |
281 | |
282 | //! Matrix transform |
283 | Matrix<dim,T> mat; |
284 | |
285 | public: |
286 | |
287 | /*! \brief Constructor |
288 | * |
289 | * \param t Matrix transformation |
290 | * \param s shift |
291 | * |
292 | */ |
293 | __device__ __host__ no_transform(const Matrix<dim,T> & t, const Point<dim,T> & s) |
294 | { |
295 | orig.zero(); |
296 | mat.identity(); |
297 | } |
298 | |
299 | /*! \brief Shift the point transformation |
300 | * |
301 | * \param s source point |
302 | * \param i coordinate |
303 | * |
304 | * \return the transformed coordinate |
305 | * |
306 | */ |
307 | __device__ __host__ inline T transform(const T(&s)[dim], const size_t i) const |
308 | { |
309 | return s[i]; |
310 | } |
311 | |
312 | /*! \brief No transformation |
313 | * |
314 | * \param s source point |
315 | * \param i coordinate |
316 | * |
317 | * \return the source point coordinate |
318 | * |
319 | */ |
320 | __device__ __host__ inline T transform(const Point<dim,T> & s, const size_t i) const |
321 | { |
322 | return s.get(i); |
323 | } |
324 | |
325 | /*! \brief No transformation |
326 | * |
327 | * \param s source point |
328 | * \param i coordinate |
329 | * |
330 | * \return the point coordinate |
331 | * |
332 | */ |
333 | template<typename Mem> __device__ __host__ inline T transform(const encapc<1,Point<dim,T>,Mem> & s, const size_t i) const |
334 | { |
335 | return s.template get<Point<dim,T>::x>()[i]; |
336 | } |
337 | |
338 | /*! \brief Set the transformation Matrix and shift |
339 | * |
340 | * \param mat Matrix transformation |
341 | * \param orig origin point |
342 | * |
343 | */ |
344 | inline void setTransform(const Matrix<dim,T> & mat, const Point<dim,T> & orig) |
345 | { |
346 | |
347 | } |
348 | |
349 | /*! \brief It return always true true |
350 | * |
351 | * There is nothing to compare |
352 | * |
353 | * \param nt unused |
354 | * |
355 | * \return true |
356 | * |
357 | */ |
358 | inline bool operator==(const no_transform<dim,T> & nt) |
359 | { |
360 | return true; |
361 | } |
362 | |
363 | /*! \brief It return always false |
364 | * |
365 | * There is nothing to compare they cannot be differents |
366 | * |
367 | * \param nt unused |
368 | * |
369 | * \return false |
370 | * |
371 | */ |
372 | inline bool operator!=(const no_transform<dim,T> & nt) |
373 | { |
374 | return false; |
375 | } |
376 | |
377 | /*! \brief Get the origin |
378 | * |
379 | * \return the shift vector |
380 | * |
381 | */ |
382 | inline const Point<dim,T> & getOrig() const |
383 | { |
384 | return orig; |
385 | } |
386 | |
387 | /*! \brief Get the transformation Matrix |
388 | * |
389 | * \return get the return the matrix |
390 | * |
391 | */ |
392 | inline const Matrix<dim,T> & getMat() const |
393 | { |
394 | return mat; |
395 | } |
396 | }; |
397 | |
398 | //! No transformation |
399 | template<unsigned int dim, typename T> |
400 | class no_transform_only |
401 | { |
402 | |
403 | public: |
404 | |
405 | /*! \brief Constructor |
406 | * |
407 | * \param t Matrix transformation |
408 | * \param s shift |
409 | * |
410 | */ |
411 | no_transform_only(const Matrix<dim,T> & t, const Point<dim,T> & s) |
412 | { |
413 | } |
414 | |
415 | /*! \brief Shift the point transformation |
416 | * |
417 | * \param s source point |
418 | * \param i coordinate |
419 | * |
420 | * \return the transformed coordinate |
421 | * |
422 | */ |
423 | __device__ __host__ inline T transform(const T * s, const int i) const |
424 | { |
425 | return s[i]; |
426 | } |
427 | |
428 | /*! \brief Shift the point transformation |
429 | * |
430 | * \param s source point |
431 | * \param i coordinate |
432 | * |
433 | * \return the transformed coordinate |
434 | * |
435 | */ |
436 | __device__ __host__ inline T transform(const T(&s)[dim], const int i) const |
437 | { |
438 | return s[i]; |
439 | } |
440 | |
441 | /*! \brief No transformation |
442 | * |
443 | * \param s source point |
444 | * \param i coordinate |
445 | * |
446 | * \return the source point coordinate |
447 | * |
448 | */ |
449 | __device__ __host__ inline T transform(const Point<dim,T> & s, const int i) const |
450 | { |
451 | return s.get(i); |
452 | } |
453 | |
454 | /*! \brief No transformation |
455 | * |
456 | * \param s source point |
457 | * \param i coordinate |
458 | * |
459 | * \return the point coordinate |
460 | * |
461 | */ |
462 | template<typename Mem> __device__ __host__ inline T transform(const encapc<1,Point<dim,T>,Mem> & s, const int i) const |
463 | { |
464 | return s.template get<Point<dim,T>::x>()[i]; |
465 | } |
466 | |
467 | /*! \brief Set the transformation Matrix and shift |
468 | * |
469 | * \param mat Matrix transformation |
470 | * \param orig origin point |
471 | * |
472 | */ |
473 | inline void setTransform(const Matrix<dim,T> & mat, const Point<dim,T> & orig) |
474 | { |
475 | |
476 | } |
477 | |
478 | /*! \brief It return always true true |
479 | * |
480 | * There is nothing to compare |
481 | * |
482 | * \param nt unused |
483 | * |
484 | * \return true |
485 | * |
486 | */ |
487 | inline bool operator==(const no_transform<dim,T> & nt) |
488 | { |
489 | return true; |
490 | } |
491 | |
492 | /*! \brief It return always false |
493 | * |
494 | * There is nothing to compare they cannot be differents |
495 | * |
496 | * \param nt unused |
497 | * |
498 | * \return false |
499 | * |
500 | */ |
501 | inline bool operator!=(const no_transform<dim,T> & nt) |
502 | { |
503 | return false; |
504 | } |
505 | }; |
506 | |
507 | /*! \brief Decompose a space into cells |
508 | * |
509 | * It is a convenient class for cell decomposition of an N dimensional space into cells |
510 | * and index linearization with getCell. Transform parameter is used to shift the space |
511 | * from the origin where the cell list is defined. It basically apply a transformation to the |
512 | * point every time we call getCell of getCellGrid. The shift is given by the starting point (p1) |
513 | * of the box that define where the Cell decomposition live |
514 | * |
515 | * \tparam dim dimension of the space divided by the cell |
516 | * \tparam T information the cell contain |
517 | * \tparam transform type of transformation (no_transformation shift ... ), carefully if p1 it different from {0, ... 0} |
518 | * shift class must be used |
519 | * |
520 | * Example of a 2D space decompised into Cells, 6x6 structure with padding 1 without shift, cell indicated with p are padding cell |
521 | * the origin of the cell or point (0,0) is marked with cell number 9 |
522 | * |
523 | * \verbatim |
524 | * +-----------------------+ |
525 | * |p |p |p |p |p |p |p |p | |
526 | * +-----------------------+ |
527 | * |p | | | | | | |p | |
528 | * +-----------------------+ |
529 | * |p | | | | | | |p | |
530 | * +-----------------------+ |
531 | * |p | | | | | | |p | |
532 | * +-----------------------+ |
533 | * |p |9 | | | | | |p | |
534 | * +-----------------------+ |
535 | * |p |p |p |p |p |p |p |p | |
536 | * +-----------------------+ |
537 | * \endverbatim |
538 | * |
539 | * ### Cell decomposer without shift |
540 | * \snippet CellDecomposer_unit_tests.hpp Cell decomposer use without shift |
541 | * ### Cell decomposer with padding |
542 | * \snippet CellDecomposer_unit_tests.hpp Test Cell decomposer with padding |
543 | * ### Cell decomposer with shift |
544 | * \snippet CellDecompose_unit_tests.hpp Test Cell decomposer with shift |
545 | * |
546 | */ |
547 | template<unsigned int dim,typename T, typename transform = no_transform<dim,T>> |
548 | class CellDecomposer_sm |
549 | { |
550 | // Point in the middle of a cell |
551 | // |
552 | // \verbatim |
553 | // |
554 | // C (0.1,0.1) |
555 | // +-----+ |
556 | // | | |
557 | // | .P | |
558 | // | | |
559 | // +-----+ |
560 | // |
561 | // \endverbatim |
562 | // |
563 | // C is the cell and P is the point inside the middle of the cell |
564 | // for example if the cell is (0.1,0.1) P is (0.05,0.05) |
565 | // |
566 | // |
567 | Point<dim,T> p_middle; |
568 | |
569 | // Point transformation before get the Cell object (useful for example to shift the cell list) |
570 | transform t; |
571 | |
572 | /*! \brief Convert the coordinates into id |
573 | * |
574 | * \param x coordinate |
575 | * \param s dimension |
576 | * |
577 | */ |
578 | inline size_t ConvertToID(const T (&x)[dim] ,size_t s) const |
579 | { |
580 | size_t id = openfpm::math::size_t_floor(t.transform(x,s) / box_unit.getHigh(s)) + off[s]; |
581 | id = (id >= gr_cell.size(s))?(gr_cell.size(s)-1-cell_shift.get(s)):id-cell_shift.get(s); |
582 | return id; |
583 | } |
584 | |
585 | /*! \brief Convert the coordinates into id |
586 | * |
587 | * \param x point |
588 | * \param s dimension |
589 | * |
590 | */ |
591 | inline size_t ConvertToID(const Point<dim,T> & x ,size_t s, size_t sc = 0) const |
592 | { |
593 | size_t id = openfpm::math::size_t_floor(t.transform(x,s) / box_unit.getHigh(s)) + off[s]; |
594 | id = (id >= gr_cell.size(s))?(gr_cell.size(s)-1-cell_shift.get(s)):id-cell_shift.get(s); |
595 | return id; |
596 | } |
597 | |
598 | /*! \brief Convert the coordinates into id with negative machine precision expansion |
599 | * |
600 | * \param x point |
601 | * \param s dimension |
602 | * |
603 | */ |
604 | inline size_t ConvertToID_me(const Point<dim,T> & x ,size_t s, size_t sc = 0) const |
605 | { |
606 | T cc = t.transform(x,s) / box_unit.getHigh(s) - 0.015625; |
607 | size_t id = openfpm::math::size_t_floor(cc) + off[s]; |
608 | id = (id >= gr_cell.size(s))?(gr_cell.size(s)-1-cell_shift.get(s)):id-cell_shift.get(s); |
609 | return id; |
610 | } |
611 | |
612 | /*! \brief Convert the coordinates into id with positive machine precision expansion |
613 | * |
614 | * \param x point |
615 | * \param s dimension |
616 | * |
617 | */ |
618 | inline size_t ConvertToID_pe(const Point<dim,T> & x ,size_t s, size_t sc = 0) const |
619 | { |
620 | T cc = t.transform(x,s) / box_unit.getHigh(s) + 0.015625; |
621 | size_t id = openfpm::math::size_t_floor(cc) + off[s]; |
622 | id = (id >= gr_cell.size(s))?(gr_cell.size(s)-1-cell_shift.get(s)):id-cell_shift.get(s); |
623 | return id; |
624 | } |
625 | |
626 | /*! \brief Convert the coordinates into id without apply shift |
627 | * |
628 | * \param x coordinate |
629 | * \param s dimension |
630 | * |
631 | */ |
632 | inline size_t ConvertToID_ns(const T (&x)[dim] ,size_t s) const |
633 | { |
634 | size_t id = openfpm::math::size_t_floor(t.transform(x,s) / box_unit.getHigh(s)) + off[s]; |
635 | id = (id >= gr_cell.size(s))?(gr_cell.size(s)-1):id; |
636 | return id; |
637 | } |
638 | |
639 | /*! \brief Convert the coordinates into id without apply shift |
640 | * |
641 | * \param x point |
642 | * \param s dimension |
643 | * |
644 | */ |
645 | inline size_t ConvertToID_ns(const Point<dim,T> & x ,size_t s, size_t sc = 0) const |
646 | { |
647 | size_t id = openfpm::math::size_t_floor(t.transform(x,s) / box_unit.getHigh(s)) + off[s]; |
648 | id = (id >= gr_cell.size(s))?(gr_cell.size(s)-1):id; |
649 | return id; |
650 | } |
651 | |
652 | /*! \brief Convert the coordinates into id |
653 | * |
654 | * \param x point |
655 | * \param s dimension |
656 | * |
657 | */ |
658 | template <typename Mem> inline size_t ConvertToID_(const encapc<1,Point<dim,T>,Mem> & x ,size_t s, size_t sc = 0) const |
659 | { |
660 | size_t id = (size_t)(t.transform(x,s) / box_unit.getHigh(s)) + off[s]; |
661 | id = (id >= gr_cell.size(s))?(gr_cell.size(s)-1-cell_shift.get(s)):id-cell_shift.get(s); |
662 | return id; |
663 | } |
664 | |
665 | /*! \Check if a particle is outside the domain of the cell-list |
666 | * |
667 | * \param pos position of the particle |
668 | * \param s coordinate to check |
669 | * |
670 | */ |
671 | template<typename Ele> inline void check_and_print_error(const Ele & pos ,size_t s) const |
672 | { |
673 | #ifdef SE_CLASS1 |
674 | if (tot_n_cell == 0) |
675 | { |
676 | std::cerr << "Error: " << __FILE__ << ":" << __LINE__ << " using an uninitialized CellDecomposer" ; |
677 | ACTION_ON_ERROR(CELL_DECOMPOSER); |
678 | } |
679 | |
680 | if (pos[s] < box.getLow(s) - off[s]*box_unit.getP2()[s] || pos[s] > box.getHigh(s) + off[s]*box_unit.getP2()[s]) |
681 | { |
682 | std::cerr << "Error: " << __FILE__ << ":" << __LINE__ << " point " << Point<dim,T>(pos).toString() << " is not inside the cell space" ; |
683 | ACTION_ON_ERROR(CELL_DECOMPOSER); |
684 | } |
685 | #endif |
686 | } |
687 | |
688 | |
689 | template<typename Ele> inline size_t getCellDom_impl(const Ele & pos) const |
690 | { |
691 | check_and_print_error(pos,0); |
692 | |
693 | size_t cell_id = ConvertToID_ns(pos,0); |
694 | |
695 | cell_id = (cell_id == gr_cell.size(0) - off[0])?gr_cell.size(0) - off[0] - 1:cell_id; |
696 | cell_id = (cell_id == off[0]-1)?off[0]:cell_id; |
697 | |
698 | cell_id -= cell_shift.get(0); |
699 | |
700 | for (size_t s = 1 ; s < dim ; s++) |
701 | { |
702 | /* coverity[dead_error_begin] */ |
703 | check_and_print_error(pos,s); |
704 | |
705 | size_t cell_idt = ConvertToID_ns(pos,s); |
706 | |
707 | cell_idt = (cell_idt == gr_cell.size(s) - off[s])?gr_cell.size(s) - off[s] - 1:cell_idt; |
708 | cell_idt = (cell_idt == off[s]-1)?off[s]:cell_idt; |
709 | |
710 | cell_idt -= cell_shift.get(s); |
711 | |
712 | cell_id += gr_cell2.size_s(s-1) * cell_idt; |
713 | } |
714 | |
715 | return cell_id; |
716 | } |
717 | |
718 | template<typename Ele> inline size_t getCellPad_impl(const Ele & pos) const |
719 | { |
720 | check_and_print_error(pos,0); |
721 | |
722 | size_t cell_id = ConvertToID_ns(pos,0); |
723 | cell_id = (cell_id == off[0])?off[0]-1:cell_id; |
724 | cell_id = (cell_id == gr_cell.size(0) - off[0] - 1)?gr_cell.size(0) - off[0]:cell_id; |
725 | |
726 | cell_id -= cell_shift.get(0); |
727 | |
728 | for (size_t s = 1 ; s < dim ; s++) |
729 | { |
730 | check_and_print_error(pos,s); |
731 | |
732 | size_t cell_idt = ConvertToID_ns(pos,s); |
733 | cell_idt = (cell_idt == off[s])?off[s]-1:cell_idt; |
734 | cell_idt = (cell_idt == gr_cell.size(s) - off[s] - 1)?gr_cell.size(s) - off[s]:cell_idt; |
735 | |
736 | cell_idt -= cell_shift.get(s); |
737 | |
738 | cell_id += gr_cell2.size_s(s-1) * cell_idt; |
739 | } |
740 | |
741 | return cell_id; |
742 | } |
743 | |
744 | |
745 | protected: |
746 | |
747 | //! Total number of cell |
748 | size_t tot_n_cell; |
749 | |
750 | //! Domain of the cell list |
751 | SpaceBox<dim,T> box; |
752 | |
753 | //! Unit box of the Cell list |
754 | SpaceBox<dim,T> box_unit; |
755 | |
756 | //! Grid structure of the Cell list |
757 | grid_sm<dim,void> gr_cell; |
758 | |
759 | //! Grid structure of the internal Cell list |
760 | grid_sm<dim,void> gr_cell2; |
761 | |
762 | //! Box in continuum for the gr_cell2 |
763 | Box<dim,T> box_gr_cell2; |
764 | |
765 | //! cell padding on each dimension |
766 | size_t off[dim]; |
767 | |
768 | //! cell_shift |
769 | Point<dim,long int> cell_shift; |
770 | |
771 | // temporal buffer |
772 | mutable size_t div_wp[dim]; |
773 | |
774 | /*! \brief Initialize all the structures |
775 | * |
776 | */ |
777 | void Initialize(const size_t pad, const size_t (& div)[dim]) |
778 | { |
779 | #ifdef SE_CLASS1 |
780 | |
781 | for (size_t i = 0 ; i < dim ; i++) |
782 | { |
783 | if (div[i] == 0) |
784 | { |
785 | std::cerr << "Error " << __FILE__ << ":" << __LINE__ << " the number of cells on each dimension must be different from zero\n" ; |
786 | ACTION_ON_ERROR(CELL_DECOMPOSER) |
787 | } |
788 | } |
789 | |
790 | #endif |
791 | |
792 | // set div_wp to zero |
793 | for (size_t i = 0 ; i < dim ; i++) |
794 | {div_wp[i] = 0;} |
795 | |
796 | // created a padded div |
797 | size_t div_p[dim]; |
798 | |
799 | for (size_t i = 0 ; i < dim ; i++) |
800 | div_p[i] = div[i] + 2*pad; |
801 | |
802 | gr_cell.setDimensions(div_p); |
803 | gr_cell2.setDimensions(div_p); |
804 | |
805 | tot_n_cell = 1; |
806 | |
807 | // Total number of cells and calculate the unit cell size |
808 | |
809 | for (size_t i = 0 ; i < dim ; i++) |
810 | { |
811 | tot_n_cell *= gr_cell.size(i); |
812 | |
813 | // Cell are padded by |
814 | box_unit.setHigh(i,(box.getHigh(i) - box.getLow(i)) / (gr_cell.size(i)- 2*pad) ); |
815 | } |
816 | |
817 | for (size_t i = 0; i < dim ; i++) |
818 | off[i] = pad; |
819 | |
820 | // Initialize p_middle |
821 | |
822 | p_middle = box_unit.getP2(); |
823 | p_middle = p_middle / 2; |
824 | } |
825 | |
826 | public: |
827 | |
828 | /*! \brief Return the transformation |
829 | * |
830 | * \return the transform |
831 | * |
832 | */ |
833 | const transform & getTransform() |
834 | { |
835 | return t; |
836 | } |
837 | |
838 | /*! \brief Return the underlying grid information of the cell list |
839 | * |
840 | * \return the grid infos |
841 | * |
842 | */ |
843 | const grid_sm<dim,void> & getGrid() const |
844 | { |
845 | #ifdef DEBUG |
846 | if (tot_n_cell == 0) |
847 | std::cerr << "Error: " << __FILE__ << ":" << __LINE__ << " using an uninitialized CellDecomposer" ; |
848 | #endif |
849 | |
850 | return gr_cell; |
851 | } |
852 | |
853 | /*! \brief Get the cell-ids |
854 | * |
855 | * Convert the point coordinates into the cell ids |
856 | * |
857 | * \param pos Point position |
858 | * |
859 | * \return the cell-ids ad a grid_key_dx<dim> |
860 | * |
861 | */ |
862 | inline grid_key_dx<dim> getCellGrid(const T (& pos)[dim]) const |
863 | { |
864 | #ifdef SE_CLASS1 |
865 | if (tot_n_cell == 0) |
866 | { |
867 | std::cerr << "Error: " << __FILE__ << ":" << __LINE__ << " using an uninitialized CellDecomposer" ; |
868 | ACTION_ON_ERROR(CELL_DECOMPOSER); |
869 | } |
870 | #endif |
871 | |
872 | grid_key_dx<dim> key; |
873 | key.set_d(0,ConvertToID(pos,0)); |
874 | |
875 | for (size_t s = 1 ; s < dim ; s++) |
876 | { |
877 | #ifdef SE_CLASS1 |
878 | if ((size_t)(t.transform(pos,s) / box_unit.getHigh(s)) + off[s] < 0) |
879 | { |
880 | std::cerr << "Error: " << __FILE__ << ":" << __LINE__ << " point is not inside the cell space\n" ; |
881 | ACTION_ON_ERROR(CELL_DECOMPOSER); |
882 | } |
883 | #endif |
884 | key.set_d(s,ConvertToID(pos,s)); |
885 | |
886 | } |
887 | |
888 | return key; |
889 | } |
890 | |
891 | /*! \brief Get the cell-id |
892 | * |
893 | * Convert the point coordinates into the cell id |
894 | * |
895 | * \param cell-id in grid units |
896 | * |
897 | * \return the cell-id |
898 | * |
899 | */ |
900 | inline size_t getCellLin(grid_key_dx<dim> && k) const |
901 | { |
902 | #ifdef SE_CLASS1 |
903 | if (tot_n_cell == 0) |
904 | { |
905 | std::cerr << "Error: " << __FILE__ << ":" << __LINE__ << " using an uninitialized CellDecomposer" ; |
906 | ACTION_ON_ERROR(CELL_DECOMPOSER); |
907 | } |
908 | |
909 | if (gr_cell2.size(0) < k.get(0) + off[0] - cell_shift.get(0)) |
910 | { |
911 | std::cerr << "Error: " << __FILE__ << ":" << __LINE__ << " cell coordinate 0 = " << k.get(0) + off[0] - cell_shift.get(0) << " bigger than cell space " << gr_cell2.size(0) << std::endl; |
912 | ACTION_ON_ERROR(CELL_DECOMPOSER); |
913 | } |
914 | #endif |
915 | |
916 | size_t cell_id = k.get(0) + off[0] - cell_shift.get(0); |
917 | |
918 | for (size_t s = 1 ; s < dim ; s++) |
919 | { |
920 | #ifdef SE_CLASS1 |
921 | if (gr_cell2.size(s) < k.get(s) + off[s] - cell_shift.get(s)) |
922 | { |
923 | std::cerr << "Error: " << __FILE__ << ":" << __LINE__ << " cell coordinate 0 = " << k.get(0) + off[0] - cell_shift.get(0) << " bigger than cell space " << gr_cell2.size(s) << std::endl; |
924 | ACTION_ON_ERROR(CELL_DECOMPOSER); |
925 | } |
926 | #endif |
927 | cell_id += gr_cell2.size_s(s-1) * (k.get(s) + off[s] -cell_shift.get(s)); |
928 | } |
929 | |
930 | return cell_id; |
931 | } |
932 | |
933 | /*! \brief Get the cell-id |
934 | * |
935 | * Convert the point coordinates into the cell id |
936 | * |
937 | * \param cell-id in grid units |
938 | * |
939 | * \return the cell-id |
940 | * |
941 | */ |
942 | inline size_t getCellLin(const grid_key_dx<dim> & k) const |
943 | { |
944 | #ifdef SE_CLASS1 |
945 | if (tot_n_cell == 0) |
946 | { |
947 | std::cerr << "Error: " << __FILE__ << ":" << __LINE__ << " using an uninitialized CellDecomposer" ; |
948 | ACTION_ON_ERROR(CELL_DECOMPOSER); |
949 | } |
950 | |
951 | if (gr_cell2.size(0) < k.get(0) + off[0] - cell_shift.get(0)) |
952 | { |
953 | std::cerr << "Error: " << __FILE__ << ":" << __LINE__ << " cell coordinate 0 = " << k.get(0) + off[0] - cell_shift.get(0) << " bigger than cell space " << gr_cell2.size(0) << std::endl; |
954 | ACTION_ON_ERROR(CELL_DECOMPOSER); |
955 | } |
956 | #endif |
957 | |
958 | size_t cell_id = k.get(0) + off[0] -cell_shift.get(0); |
959 | |
960 | for (size_t s = 1 ; s < dim ; s++) |
961 | { |
962 | #ifdef SE_CLASS1 |
963 | if (gr_cell2.size(s) < k.get(s) + off[s] - cell_shift.get(s)) |
964 | { |
965 | std::cerr << "Error: " << __FILE__ << ":" << __LINE__ << " cell coordinate 0 = " << k.get(0) + off[0] - cell_shift.get(0) << " bigger than cell space " << gr_cell2.size(s) << std::endl; |
966 | ACTION_ON_ERROR(CELL_DECOMPOSER); |
967 | } |
968 | #endif |
969 | cell_id += gr_cell2.size_s(s-1) * (k.get(s) + off[s] -cell_shift.get(s)); |
970 | } |
971 | |
972 | return cell_id; |
973 | } |
974 | |
975 | /*! \brief Get the cell-ids with negative machine precision expansion |
976 | * |
977 | * Convert the point coordinates into the cell ids (Careful it include padding) |
978 | * |
979 | * \param pos Point position |
980 | * |
981 | * \return the cell-ids ad a grid_key_dx<dim> |
982 | * |
983 | */ |
984 | inline grid_key_dx<dim> getCellGrid_me(const Point<dim,T> & pos) const |
985 | { |
986 | #ifdef SE_CLASS1 |
987 | if (tot_n_cell == 0) |
988 | { |
989 | std::cerr << "Error: " << __FILE__ << ":" << __LINE__ << " using an uninitialized CellDecomposer" << std::endl; |
990 | ACTION_ON_ERROR(CELL_DECOMPOSER); |
991 | } |
992 | #endif |
993 | |
994 | grid_key_dx<dim> key; |
995 | key.set_d(0,ConvertToID_me(pos,0)); |
996 | |
997 | for (size_t s = 1 ; s < dim ; s++) |
998 | { |
999 | #ifdef SE_CLASS1 |
1000 | if ((size_t)(t.transform(pos,s) / box_unit.getHigh(s)) + off[s] < 0) |
1001 | { |
1002 | std::cerr << "Error: " << __FILE__ << ":" << __LINE__ << " point is not inside the cell space" << std::endl; |
1003 | ACTION_ON_ERROR(CELL_DECOMPOSER); |
1004 | } |
1005 | #endif |
1006 | /* coverity[dead_error_line] */ |
1007 | key.set_d(s,ConvertToID_me(pos,s)); |
1008 | } |
1009 | |
1010 | return key; |
1011 | } |
1012 | |
1013 | /*! \brief Get the cell-ids with positive machine precision expansion |
1014 | * |
1015 | * Convert the point coordinates into the cell ids (Careful it include padding) |
1016 | * |
1017 | * \param pos Point position |
1018 | * |
1019 | * \return the cell-ids ad a grid_key_dx<dim> |
1020 | * |
1021 | */ |
1022 | inline grid_key_dx<dim> getCellGrid_pe(const Point<dim,T> & pos) const |
1023 | { |
1024 | #ifdef SE_CLASS1 |
1025 | if (tot_n_cell == 0) |
1026 | { |
1027 | std::cerr << "Error: " << __FILE__ << ":" << __LINE__ << " using an uninitialized CellDecomposer" << std::endl; |
1028 | ACTION_ON_ERROR(CELL_DECOMPOSER); |
1029 | } |
1030 | #endif |
1031 | |
1032 | grid_key_dx<dim> key; |
1033 | key.set_d(0,ConvertToID_pe(pos,0)); |
1034 | |
1035 | for (size_t s = 1 ; s < dim ; s++) |
1036 | { |
1037 | #ifdef SE_CLASS1 |
1038 | if ((size_t)(t.transform(pos,s) / box_unit.getHigh(s)) + off[s] < 0) |
1039 | { |
1040 | std::cerr << "Error: " << __FILE__ << ":" << __LINE__ << " point is not inside the cell space" << std::endl; |
1041 | ACTION_ON_ERROR(CELL_DECOMPOSER); |
1042 | } |
1043 | #endif |
1044 | /* coverity[dead_error_line] */ |
1045 | key.set_d(s,ConvertToID_pe(pos,s)); |
1046 | } |
1047 | |
1048 | return key; |
1049 | } |
1050 | |
1051 | /*! \brief Get the cell-ids |
1052 | * |
1053 | * Convert the point coordinates into the cell ids (Careful it include padding) |
1054 | * |
1055 | * \param pos Point position |
1056 | * |
1057 | * \return the cell-ids ad a grid_key_dx<dim> |
1058 | * |
1059 | */ |
1060 | inline grid_key_dx<dim> getCellGrid(const Point<dim,T> & pos) const |
1061 | { |
1062 | #ifdef SE_CLASS1 |
1063 | if (tot_n_cell == 0) |
1064 | { |
1065 | std::cerr << "Error: " << __FILE__ << ":" << __LINE__ << " using an uninitialized CellDecomposer" << std::endl; |
1066 | ACTION_ON_ERROR(CELL_DECOMPOSER); |
1067 | } |
1068 | #endif |
1069 | |
1070 | grid_key_dx<dim> key; |
1071 | key.set_d(0,ConvertToID(pos,0)); |
1072 | |
1073 | for (size_t s = 1 ; s < dim ; s++) |
1074 | { |
1075 | #ifdef SE_CLASS1 |
1076 | if ((size_t)(t.transform(pos,s) / box_unit.getHigh(s)) + off[s] < 0) |
1077 | { |
1078 | std::cerr << "Error: " << __FILE__ << ":" << __LINE__ << " point is not inside the cell space" << std::endl; |
1079 | ACTION_ON_ERROR(CELL_DECOMPOSER); |
1080 | } |
1081 | #endif |
1082 | /* coverity[dead_error_line] */ |
1083 | key.set_d(s,ConvertToID(pos,s)); |
1084 | } |
1085 | |
1086 | return key; |
1087 | } |
1088 | |
1089 | /*! \brief Get the cell-id enforcing that is NOT a cell from the padding |
1090 | * |
1091 | * Convert the point coordinates into the cell id |
1092 | * |
1093 | * \note this function is in general used to bypass round-off error |
1094 | * |
1095 | * \param pos Point position |
1096 | * |
1097 | * \return the cell-id |
1098 | * |
1099 | */ |
1100 | inline size_t getCellDom(const Point<dim,T> & pos) const |
1101 | { |
1102 | return getCellDom_impl<Point<dim,T>>(pos); |
1103 | } |
1104 | |
1105 | |
1106 | /*! \brief Get the cell-id enforcing that is NOT a cell from the padding |
1107 | * |
1108 | * Convert the point coordinates into the cell id |
1109 | * |
1110 | * \note this function is in general used to bypass round-off error |
1111 | * |
1112 | * \param pos Point position |
1113 | * |
1114 | * \return the cell-id |
1115 | * |
1116 | */ |
1117 | inline size_t getCellDom(const T (& pos)[dim]) const |
1118 | { |
1119 | return getCellDom_impl<T[dim]>(pos); |
1120 | } |
1121 | |
1122 | /*! \brief Get the cell-id enforcing that is from a padding cell |
1123 | * |
1124 | * Convert the point coordinates into the cell id |
1125 | * |
1126 | * \note this function is in general used to bypass round-off error |
1127 | * |
1128 | * \param pos Point position |
1129 | * |
1130 | * \return the cell-id |
1131 | * |
1132 | */ |
1133 | inline size_t getCellPad(const Point<dim,T> & pos) const |
1134 | { |
1135 | return getCellPad_impl<Point<dim,T>>(pos); |
1136 | } |
1137 | |
1138 | /*! \brief Get the cell-id enforcing that is from a padding cell |
1139 | * |
1140 | * Convert the point coordinates into the cell id |
1141 | * |
1142 | * \note this function is in general used to bypass round-off error |
1143 | * |
1144 | * \param pos Point position |
1145 | * |
1146 | * \return the cell-id |
1147 | * |
1148 | */ |
1149 | inline size_t getCellPad(const T (& pos)[dim]) const |
1150 | { |
1151 | return getCellPad_impl<T[dim]>(pos); |
1152 | } |
1153 | |
1154 | /*! \brief Get the cell-id |
1155 | * |
1156 | * Convert the point coordinates into the cell id |
1157 | * |
1158 | * \param pos Point position |
1159 | * |
1160 | * \return the cell-id |
1161 | * |
1162 | */ |
1163 | inline size_t getCell(const T (& pos)[dim]) const |
1164 | { |
1165 | #ifdef SE_CLASS1 |
1166 | if (tot_n_cell == 0) |
1167 | { |
1168 | std::cerr << "Error: " << __FILE__ << ":" << __LINE__ << " using an uninitialized CellDecomposer" ; |
1169 | ACTION_ON_ERROR(CELL_DECOMPOSER); |
1170 | } |
1171 | |
1172 | if (pos[0] < box.getLow(0) - off[0]*box_unit.getP2()[0] || pos[0] > box.getHigh(0) + off[0]*box_unit.getP2()[0]) |
1173 | { |
1174 | std::cerr << "Error: " << __FILE__ << ":" << __LINE__ << " point " << toPointString(pos) << " is not inside the cell space" ; |
1175 | ACTION_ON_ERROR(CELL_DECOMPOSER); |
1176 | } |
1177 | #endif |
1178 | |
1179 | size_t cell_id = ConvertToID(pos,0); |
1180 | |
1181 | for (size_t s = 1 ; s < dim ; s++) |
1182 | { |
1183 | #ifdef SE_CLASS1 |
1184 | if (pos[s] < box.getLow(s) - off[s]*box_unit.getP2()[s] || pos[s] > box.getHigh(s) + off[s]*box_unit.getP2()[s]) |
1185 | { |
1186 | std::cerr << "Error: " << __FILE__ << ":" << __LINE__ << " point " << toPointString(pos) << " is not inside the cell space" ; |
1187 | ACTION_ON_ERROR(CELL_DECOMPOSER); |
1188 | } |
1189 | #endif |
1190 | cell_id += gr_cell2.size_s(s-1) * ConvertToID(pos,s); |
1191 | } |
1192 | |
1193 | return cell_id; |
1194 | } |
1195 | |
1196 | /*! \brief Get the cell-id |
1197 | * |
1198 | * Convert the point coordinates into the cell id |
1199 | * |
1200 | * \param pos Point position |
1201 | * |
1202 | * \return the cell-id |
1203 | * |
1204 | */ |
1205 | inline size_t getCell(const Point<dim,T> & pos) const |
1206 | { |
1207 | #ifdef SE_CLASS1 |
1208 | if (tot_n_cell == 0) |
1209 | { |
1210 | std::cerr << "Error: " << __FILE__ << ":" << __LINE__ << " using an uninitialized CellDecomposer" ; |
1211 | ACTION_ON_ERROR(CELL_DECOMPOSER); |
1212 | } |
1213 | |
1214 | if (pos.get(0) < box.getLow(0) - off[0]*box_unit.getP2()[0] || pos.get(0) > box.getHigh(0) + off[0]*box_unit.getP2()[0]) |
1215 | { |
1216 | std::cerr << "Error: " << __FILE__ << ":" << __LINE__ << " point " << pos.toPointString() << " is not inside the cell space" ; |
1217 | ACTION_ON_ERROR(CELL_DECOMPOSER); |
1218 | } |
1219 | #endif |
1220 | |
1221 | size_t cell_id = ConvertToID(pos,0); |
1222 | |
1223 | for (size_t s = 1 ; s < dim ; s++) |
1224 | { |
1225 | #ifdef SE_CLASS1 |
1226 | if (pos.get(s) < box.getLow(s) - off[s]*box_unit.getP2()[s] || pos.get(s) > box.getHigh(s) + off[s]*box_unit.getP2()[s]) |
1227 | { |
1228 | std::cerr << "Error: " << __FILE__ << ":" << __LINE__ << " point " << pos.toPointString() << " is not inside the cell space" ; |
1229 | ACTION_ON_ERROR(CELL_DECOMPOSER); |
1230 | } |
1231 | #endif |
1232 | /* coverity[dead_error_line] */ |
1233 | cell_id += gr_cell2.size_s(s-1) * ConvertToID(pos,s); |
1234 | } |
1235 | |
1236 | return cell_id; |
1237 | } |
1238 | |
1239 | /*! \brief Get the cell-id |
1240 | * |
1241 | * Convert the point coordinates into the cell id |
1242 | * |
1243 | * \param pos Point position |
1244 | * |
1245 | * \return the cell-id |
1246 | * |
1247 | */ |
1248 | template<typename Mem> inline size_t getCell(const encapc<1,Point<dim,T>,Mem> & pos) const |
1249 | { |
1250 | |
1251 | #ifdef SE_CLASS1 |
1252 | if (tot_n_cell == 0) |
1253 | { |
1254 | std::cerr << "Error: " << __FILE__ << ":" << __LINE__ << " using an uninitialized CellDecomposer" ; |
1255 | ACTION_ON_ERROR(CELL_DECOMPOSER); |
1256 | } |
1257 | |
1258 | if (pos.template get<0>()[0] < box.getLow(0) - off[0]*box_unit.getP2()[0] || pos.template get<0>()[0] > box.getHigh(0) + off[0]*box_unit.getP2()[0]) |
1259 | { |
1260 | std::cerr << "Error: " << __FILE__ << ":" << __LINE__ << " point " << toPointString(pos) << " is not inside the cell space " << box.toString() << std::endl; |
1261 | ACTION_ON_ERROR(CELL_DECOMPOSER); |
1262 | } |
1263 | #endif |
1264 | |
1265 | size_t cell_id = ConvertToID_(pos,0); |
1266 | |
1267 | for (size_t s = 1 ; s < dim ; s++) |
1268 | { |
1269 | #ifdef SE_CLASS1 |
1270 | |
1271 | if (pos.template get<0>()[s] < box.getLow(s) - off[s]*box_unit.getP2()[s] || pos.template get<0>()[s] > box.getHigh(s) + off[s]*box_unit.getP2()[s]) |
1272 | { |
1273 | std::cerr << "Error: " << __FILE__ << ":" << __LINE__ << " point " << toPointString(pos) << " is not inside the cell space" ; |
1274 | ACTION_ON_ERROR(CELL_DECOMPOSER); |
1275 | } |
1276 | #endif |
1277 | cell_id += gr_cell2.size_s(s-1) * ConvertToID_(pos,s); |
1278 | } |
1279 | |
1280 | return cell_id; |
1281 | } |
1282 | |
1283 | /*! \brief Return the smallest box containing the grid points |
1284 | * |
1285 | * Suppose a grid 5x5 defined on a Box<2,float> box({0.0,0.0},{1.0,1.0}) |
1286 | * and feeding to the function a Box<2,float>({0.4,0.4},{0.8,0.8}), it will return |
1287 | * a Box<2,size_t> (2,2) and (3,3). A visualization it is shown in the |
1288 | * picture below. (the grid points are centered on each cell) |
1289 | * |
1290 | * \verbatim |
1291 | * |
1292 | +----------+ |
1293 | |. . . . . | |
1294 | | | |
1295 | |. . . . . | |
1296 | | +---+ | |
1297 | |. .|. .|. | |
1298 | | | | | |
1299 | |. .|. .|. | |
1300 | | +---+ | |
1301 | |. . . . . | |
1302 | +----------+ |
1303 | |
1304 | \endverbatim |
1305 | * |
1306 | * |
1307 | * \div grid size on each dimension |
1308 | * \box Small box in the picture |
1309 | * |
1310 | * The big box is defined by "this" box |
1311 | * |
1312 | * \return the box containing the grid points |
1313 | * |
1314 | */ |
1315 | inline Box<dim,size_t> getGridPoints(const Box<dim,T> & s_box) const |
1316 | { |
1317 | // Box with inside grid |
1318 | Box<dim,size_t> bx; |
1319 | |
1320 | // Point p2 |
1321 | Point<dim,T> p2 = s_box.getP2(); |
1322 | p2 = p2 - p_middle; |
1323 | |
1324 | // Point p1 |
1325 | Point<dim,T> p1 = s_box.getP1(); |
1326 | p1 = p1 + p_middle; |
1327 | |
1328 | bx.setP2(getCellGrid(p2)); |
1329 | bx.setP1(getCellGrid(p1)); |
1330 | |
1331 | return bx; |
1332 | } |
1333 | |
1334 | /*! \brief Set the domain to decompose |
1335 | * |
1336 | * \param box Domain to decompose |
1337 | * \param div array with the number of cells on each dimensions |
1338 | * \param pad padding cell |
1339 | * |
1340 | */ |
1341 | inline void setDimensions(const Box<dim,T> & box, const size_t (&div)[dim], const size_t (&div2)[dim], const size_t pad, Point<dim,long int> cell_shift) |
1342 | { |
1343 | Matrix<dim,T> mat; |
1344 | mat.identity(); |
1345 | t.setTransform(mat,box.getP1()); |
1346 | this->box = box; |
1347 | |
1348 | Initialize(pad,div); |
1349 | |
1350 | size_t cells[dim]; |
1351 | |
1352 | for (size_t i = 0 ; i < dim ; i++) |
1353 | cells[i] = div2[i] + 2*pad; |
1354 | |
1355 | gr_cell2.setDimensions(cells); |
1356 | |
1357 | for (size_t i = 0 ; i < dim ; i++) |
1358 | this->cell_shift.get(i) = cell_shift.get(i) - off[i]; |
1359 | } |
1360 | |
1361 | /*! \brief Set the domain to decompose |
1362 | * |
1363 | * \param box Domain to decompose |
1364 | * \param div array with the number of cells on each dimensions |
1365 | * \param pad padding cell |
1366 | * |
1367 | */ |
1368 | inline void setDimensions(const Box<dim,T> & box, const size_t (&div)[dim], const size_t pad) |
1369 | { |
1370 | Matrix<dim,T> mat; |
1371 | mat.identity(); |
1372 | t.setTransform(mat,box.getP1()); |
1373 | this->box = box; |
1374 | Initialize(pad,div); |
1375 | this->cell_shift = 0; |
1376 | } |
1377 | |
1378 | /*! \brief Set the cell decomposition parameters + the nested |
1379 | * |
1380 | * \param box Domain to decompose |
1381 | * \param div array with the number of cells on each dimensions |
1382 | * \param div2 array with the number of cells on each dimension for the nested decomposer |
1383 | * \param mat transformation matrix the cell space is transformed by p' = A * p |
1384 | * \param orig origin of the cell decomposition |
1385 | * \param pad padding cell |
1386 | * |
1387 | */ |
1388 | inline void setDimensions(const Box<dim,T> & box, const size_t (&div)[dim], const size_t (&div2)[dim], const Matrix<dim,T> & mat, const size_t pad, Point<dim,long int> cell_shift) |
1389 | { |
1390 | t.setTransform(mat,box.getP1()); |
1391 | this->box = box; |
1392 | |
1393 | Initialize(pad,div); |
1394 | |
1395 | // The nested cell is big div2 + 2*off |
1396 | |
1397 | size_t div_with_off[dim]; |
1398 | |
1399 | for(size_t i = 0 ; i < dim ; i++) |
1400 | div_with_off[i] = div2[i] + 2*off[i]; |
1401 | |
1402 | gr_cell2.setDimensions(div_with_off); |
1403 | |
1404 | for (size_t i = 0 ; i < dim ; i++) |
1405 | this->cell_shift.get(i) = cell_shift.get(i) - off[i]; |
1406 | } |
1407 | |
1408 | /*! \brief Set the cell decomposition parameters |
1409 | * |
1410 | * \param box Domain to decompose |
1411 | * \param div array with the number of cells on each dimensions |
1412 | * \param mat transformation matrix the cell space is transformed by p' = A * p |
1413 | * \param orig origin of the cell decomposition |
1414 | * \param pad padding cell |
1415 | * |
1416 | */ |
1417 | inline void setDimensions(const Box<dim,T> & box, const size_t (&div)[dim], const Matrix<dim,T> & mat, const size_t pad) |
1418 | { |
1419 | t.setTransform(mat,box.getP1()); |
1420 | this->box = box; |
1421 | Initialize(pad,div); |
1422 | this->cell_shift = 0; |
1423 | } |
1424 | |
1425 | |
1426 | /*! \brief Set a consistent cell decomposer |
1427 | * |
1428 | * When we want to create a new extended CellDecomposer consistent with the old one |
1429 | * this function can be used to guarantee such costrain. |
1430 | * With consistent we mean that for each cell of the old CellDecomposer exist |
1431 | * a Cell in the new CellDecomposer that perfectly overlap |
1432 | * |
1433 | * \param cd OLD CellDecomposer |
1434 | * \param cell_extension extension of the CellDecomposer in term of cells to add in each directions (like Ghost) |
1435 | * |
1436 | */ |
1437 | inline void setDimensions(const CellDecomposer_sm<dim,T,transform> & cd, const Box<dim,size_t> & cell_extension) |
1438 | { |
1439 | this->cell_shift = 0; |
1440 | |
1441 | // Get the space transformation |
1442 | |
1443 | t.setTransform(cd.getMat(),cd.getOrig()); |
1444 | |
1445 | // The domain is equivalent to the old one |
1446 | this->box = cd.box; |
1447 | |
1448 | // The padding must be calculated |
1449 | |
1450 | size_t pad = 0; |
1451 | |
1452 | for (size_t i = 0 ; i < dim ; i++) |
1453 | { |
1454 | if (pad < cell_extension.getLow(i)) |
1455 | pad = cell_extension.getLow(i); |
1456 | else if (pad > cell_extension.getHigh(i)) |
1457 | pad = cell_extension.getHigh(i); |
1458 | } |
1459 | |
1460 | // We have to give the old division |
1461 | |
1462 | size_t sz_div[dim]; |
1463 | |
1464 | for (size_t i = 0 ; i < dim ; i++) |
1465 | sz_div[i] = cd.gr_cell.size(i) - 2*cd.off[i]; |
1466 | |
1467 | Initialize(pad,sz_div); |
1468 | } |
1469 | |
1470 | /*! \brief Constructor |
1471 | * |
1472 | * \param box Space where is defined the cell list |
1473 | * \param div Reference array to the number of divisions on each dimensions |
1474 | * \param mat Transformation matrix, the point is transformed as p' = mat * p |
1475 | * \param pad cell padding |
1476 | * |
1477 | * Example for div = {6,6} and pad = 1 |
1478 | * |
1479 | * \verbatim |
1480 | * +-----------------------+ |
1481 | * |p |p |p |p |p |p |p |p | |
1482 | * +-----------------------+ |
1483 | * |p | | | | | | |p | |
1484 | * +-----------------------+ |
1485 | * |p | | | | | | |p | |
1486 | * +-----------------------+ |
1487 | * |p | | | | | | |p | |
1488 | * +-----------------------+ |
1489 | * |p |9 | | | | | |p | |
1490 | * +-----------------------+ |
1491 | * |p |p |p |p |p |p |p |p | |
1492 | * +-----------------------+ |
1493 | * \endverbatim |
1494 | * |
1495 | * Cell with p are padding cell cell that are around but external the box, the cell number 9 that |
1496 | * is at the origin of the box is identified with 9 |
1497 | * |
1498 | */ |
1499 | CellDecomposer_sm(const SpaceBox<dim,T> & box, const size_t (&div)[dim], Matrix<dim,T> & mat, const size_t pad) |
1500 | :t(Matrix<dim,T>::identity(),box.getP1()),box(box),gr_cell() |
1501 | { |
1502 | Initialize(pad); |
1503 | } |
1504 | |
1505 | /*! \brief Constructor |
1506 | * |
1507 | * \param box Space where is defined the cell list (it is assumed p1 = {0, .... 0}) |
1508 | * \param div Reference array to the number of divisions on each dimensions |
1509 | * \param pad cell padding |
1510 | * |
1511 | * Example for div = {7,7} and pad = 1 |
1512 | * |
1513 | * \verbatim |
1514 | * +-----------------------+ |
1515 | * |p |p |p |p |p |p |p |p | |
1516 | * +-----------------------+ |
1517 | * |p | | | | | | |p | |
1518 | * +-----------------------+ |
1519 | * |p | | | | | | |p | |
1520 | * +-----------------------+ |
1521 | * |p | | | | | | |p | |
1522 | * +-----------------------+ |
1523 | * |p |9 | | | | | |p | |
1524 | * +-----------------------+ |
1525 | * |p |p |p |p |p |p |p |p | |
1526 | * +-----------------------+ |
1527 | * |
1528 | * \endverbatim |
1529 | * |
1530 | * Cell with p are padding cell cell that are around but external the box, the cell number 9 that |
1531 | * is at the origin of the box is identified with 9 |
1532 | * |
1533 | */ |
1534 | CellDecomposer_sm(const SpaceBox<dim,T> & box, const size_t (&div)[dim], const size_t pad) |
1535 | :t(Matrix<dim,T>::identity(),box.getP1()),box(box),gr_cell(div) |
1536 | { |
1537 | Initialize(pad,div); |
1538 | } |
1539 | |
1540 | /*! \brief Constructor for a consistent CellDecomposer construction |
1541 | * |
1542 | * \param cd Old CellDecomposer |
1543 | * \param ext Extension box |
1544 | * |
1545 | * |
1546 | */ |
1547 | CellDecomposer_sm(const CellDecomposer_sm<dim,T,transform> & cd, Box<dim,size_t> & ext) |
1548 | :t(Matrix<dim,T>::identity(),cd.getOrig()) |
1549 | { |
1550 | setDimensions(cd,ext); |
1551 | } |
1552 | |
1553 | |
1554 | //! default constructor |
1555 | CellDecomposer_sm() |
1556 | :t(Matrix<dim,T>::identity(),Point<dim,T>::zero_p()),tot_n_cell(0) |
1557 | { |
1558 | |
1559 | } |
1560 | |
1561 | /*! \brief Return the box that represent the cell |
1562 | * |
1563 | * Can be considered the spacing between vertices of the cells |
1564 | * |
1565 | * \return the box |
1566 | * |
1567 | */ |
1568 | inline const Box<dim,T> & getCellBox() const |
1569 | { |
1570 | return box_unit; |
1571 | } |
1572 | |
1573 | /*! \brief Get the transformation Matrix of the cell decomposer |
1574 | * |
1575 | * \return the transformation Matrix |
1576 | * |
1577 | */ |
1578 | inline const Matrix<dim,T> & getMat() const |
1579 | { |
1580 | return t.getMat(); |
1581 | } |
1582 | |
1583 | /*! \brief Get the origin of the cell decomposer |
1584 | * |
1585 | * \return the origin |
1586 | * |
1587 | */ |
1588 | inline const Point<dim,T> & getOrig() const |
1589 | { |
1590 | return t.getOrig(); |
1591 | } |
1592 | |
1593 | /*! \brief Convert a Box in the domain space into cell units (Negative contour included, positive contour excluded) |
1594 | * |
1595 | * Given the following |
1596 | * |
1597 | * \verbatim |
1598 | * |
1599 | +-----+-----+-----+-----+-----+-----+ (1.0. 1.0) Domain box |
1600 | | | | | | | | |
1601 | | | | | | | | |
1602 | | +-----------------+ | | | |
1603 | +-----+-----+-----+-----+-----+-----+ |
1604 | | | | | | | | | | |
1605 | Box "b" <-----------------+ | | | | | | Grid (7, 6) |
1606 | (0.1 , 0.42) | | | | | | | | | Cells (6,5) |
1607 | (0.64, 0.85) +-----+-----+-----+-----+-----+-----+ |
1608 | | | | | | | | | | |
1609 | | | | | | | | | | |
1610 | | +-----------------+ | | | |
1611 | +-----+-----+-----+-----+-----+-----+ |
1612 | | | | | | | | |
1613 | | | | | | | | |
1614 | +-----+-----+-----+-----+-----+-----+ |
1615 | | | | | | | | |
1616 | | | | | | | | |
1617 | | | | | | | | |
1618 | +-----+-----+-----+-----+-----+-----+ |
1619 | (0.0, 0.0) |
1620 | |
1621 | |
1622 | + = grid points |
1623 | |
1624 | \verbatim |
1625 | |
1626 | It return a Box with P1 = (0,2), P2 = (3,4) |
1627 | |
1628 | * |
1629 | * \param b Box in domain space |
1630 | * \param bc boundary conditions |
1631 | * |
1632 | * \return Box in grid units, if P2 < P1 the box does not include any grid points |
1633 | * |
1634 | */ |
1635 | inline Box<dim,long int> convertDomainSpaceIntoCellUnits(const Box<dim,T> & b_d, const size_t (& bc)[dim]) const |
1636 | { |
1637 | Box<dim,long int> g_box; |
1638 | Box<dim,T> b = b_d; |
1639 | b -= getOrig(); |
1640 | |
1641 | // Convert b into grid units |
1642 | b /= getCellBox().getP2(); |
1643 | |
1644 | // Considering that we are interested in a box decomposition of the space |
1645 | // where each box does not intersect any other boxes in the decomposition we include the negative |
1646 | // countour and exclude the positive one. So ceilP1 do the job for P1 while ceilP2 - 1 |
1647 | // do the job for P2 |
1648 | |
1649 | b.floorP1(); |
1650 | b.ceilP2(); |
1651 | |
1652 | g_box = b; |
1653 | |
1654 | // Translate the box by the offset |
1655 | |
1656 | for (size_t i = 0 ; i < dim ; i++) |
1657 | { |
1658 | g_box.setLow(i,g_box.getLow(i) + off[i]); |
1659 | g_box.setHigh(i,g_box.getHigh(i) + off[i]); |
1660 | } |
1661 | |
1662 | // on the other hand with non periodic boundary condition, the positive border of the |
1663 | // sub-domain at the edge of the domain must be included |
1664 | |
1665 | Point<dim,size_t> p_move; |
1666 | |
1667 | for (size_t i = 0 ; i < dim ; i++) |
1668 | { |
1669 | // we are at the positive border (We are assuming that there are not rounding error in the decomposition) |
1670 | if (b_d.getHigh(i) == box.getHigh(i)) |
1671 | { |
1672 | if (bc[i] == NON_PERIODIC) |
1673 | { |
1674 | // Here the positive boundary is included |
1675 | g_box.setHigh(i,gr_cell.size(i) - off[i]); |
1676 | } |
1677 | else |
1678 | { |
1679 | // Carefull in periodic gr_cell is one bigger than the non-periodic |
1680 | // and the positive boundary is excluded |
1681 | g_box.setHigh(i,gr_cell.size(i)-1 - off[i]); |
1682 | } |
1683 | } |
1684 | |
1685 | |
1686 | if (b_d.getLow(i) == box.getHigh(i)) |
1687 | { |
1688 | if (bc[i] == NON_PERIODIC) |
1689 | { |
1690 | // The instruction is the same but the meaning is different |
1691 | // for this reason there is anyway a branch |
1692 | // Here the border is not included |
1693 | g_box.setLow(i,gr_cell.size(i) - off[i]); |
1694 | } |
1695 | else |
1696 | { |
1697 | // Carefull in periodic gr_cell is one bigger than the non-periodic |
1698 | // Here the border is included |
1699 | g_box.setLow(i,gr_cell.size(i) - off[i]); |
1700 | } |
1701 | } |
1702 | |
1703 | /////////// Low boundary |
1704 | |
1705 | // we are at the positive border (We are assuming that there are not rounding error in the decomposition) |
1706 | /* coverity[copy_paste_error] */ |
1707 | if (b_d.getHigh(i) == box.getLow(i)) |
1708 | g_box.setHigh(i,off[i]); |
1709 | |
1710 | |
1711 | if (b_d.getLow(i) == box.getLow(i)) |
1712 | g_box.setLow(i,off[i]); |
1713 | } |
1714 | |
1715 | return g_box; |
1716 | } |
1717 | |
1718 | |
1719 | /*! \brief Convert a Box in the domain space into grid units (Negative contour included, positive contour excluded) |
1720 | * |
1721 | * Given the following |
1722 | * |
1723 | * \verbatim |
1724 | * |
1725 | +-----+-----+-----+-----+-----+-----+ (1.0. 1.0) Domain box |
1726 | | | | | | | | |
1727 | | | | | | | | |
1728 | | +-----------------+ | | | |
1729 | +-----+-----+-----+-----+-----+-----+ |
1730 | | | | | | | | | | |
1731 | Box "b" <-----------------+ | | | | | | Grid (7, 6) |
1732 | (0.1 , 0.42) | | | | | | | | | |
1733 | (0.64, 0.85) +-----+-----+-----+-----+-----+-----+ |
1734 | | | | | | | | | | |
1735 | | | | | | | | | | |
1736 | | +-----------------+ | | | |
1737 | +-----+-----+-----+-----+-----+-----+ |
1738 | | | | | | | | |
1739 | | | | | | | | |
1740 | +-----+-----+-----+-----+-----+-----+ |
1741 | | | | | | | | |
1742 | | | | | | | | |
1743 | | | | | | | | |
1744 | +-----+-----+-----+-----+-----+-----+ |
1745 | (0.0, 0.0) |
1746 | |
1747 | |
1748 | + = grid points |
1749 | |
1750 | \verbatim |
1751 | |
1752 | It return a Box with P1 = (1,3), P2 = (3,4) |
1753 | |
1754 | * |
1755 | * \param b Box in domain space |
1756 | * \param bc boundary conditions |
1757 | * |
1758 | * \return Box in grid units, if P2 < P1 the box does not include any grid points |
1759 | * |
1760 | */ |
1761 | inline Box<dim,long int> convertDomainSpaceIntoGridUnits(const Box<dim,T> & b_d, const size_t (& bc)[dim]) const |
1762 | { |
1763 | Box<dim,long int> g_box; |
1764 | Box<dim,T> b = b_d; |
1765 | b -= getOrig(); |
1766 | |
1767 | // Convert b into grid units |
1768 | b /= getCellBox().getP2(); |
1769 | |
1770 | // Considering that we are interested in a box decomposition of the space |
1771 | // where each box does not intersect any other boxes in the decomposition we include the negative |
1772 | // countour and exclude the positive one. So ceilP1 do the job for P1 while ceilP2 - 1 |
1773 | // do the job for P2 |
1774 | |
1775 | b.ceilP1(); |
1776 | |
1777 | // (we do -1 later) |
1778 | b.ceilP2(); |
1779 | for (size_t i = 0 ; i < dim ; i++) {b.setHigh(i,b.getHigh(i)-1);} |
1780 | |
1781 | g_box = b; |
1782 | |
1783 | // on the other hand with non periodic boundary condition, the positive border of the |
1784 | // sub-domain at the edge of the domain must be included |
1785 | |
1786 | Point<dim,size_t> p_move; |
1787 | |
1788 | for (size_t i = 0 ; i < dim ; i++) |
1789 | { |
1790 | // we are at the positive border (We are assuming that there are not rounding error in the decomposition) |
1791 | if (b_d.getHigh(i) == box.getHigh(i)) |
1792 | { |
1793 | if (bc[i] == NON_PERIODIC) |
1794 | { |
1795 | // Here the positive boundary is included |
1796 | g_box.setHigh(i,gr_cell.size(i) - off[i]); |
1797 | } |
1798 | else |
1799 | { |
1800 | // Carefull in periodic gr_cell is one bigger than the non-periodic |
1801 | // and the positive boundary is excluded |
1802 | g_box.setHigh(i,gr_cell.size(i)-1 - off[i]); |
1803 | } |
1804 | } |
1805 | |
1806 | |
1807 | if (b_d.getLow(i) == box.getHigh(i)) |
1808 | { |
1809 | if (bc[i] == NON_PERIODIC) |
1810 | { |
1811 | // The instruction is the same but the meaning is different |
1812 | // for this reason there is anyway a branch |
1813 | // Here the border is not included |
1814 | g_box.setLow(i,gr_cell.size(i) - off[i]); |
1815 | } |
1816 | else |
1817 | { |
1818 | // Carefull in periodic gr_cell is one bigger than the non-periodic |
1819 | // Here the border is included |
1820 | g_box.setLow(i,gr_cell.size(i) - off[i]); |
1821 | } |
1822 | } |
1823 | |
1824 | /////////// Low boundary |
1825 | |
1826 | // we are at the positive border (We are assuming that there are not rounding error in the decomposition) |
1827 | /* coverity[copy_paste_error] */ |
1828 | if (b_d.getHigh(i) == box.getLow(i)) |
1829 | g_box.setHigh(i,off[i]); |
1830 | |
1831 | |
1832 | if (b_d.getLow(i) == box.getLow(i)) |
1833 | g_box.setLow(i,off[i]); |
1834 | } |
1835 | |
1836 | return g_box; |
1837 | } |
1838 | |
1839 | |
1840 | /*! \brief Convert a Box from grid units into domain space (Negative contour included, positive contour excluded) |
1841 | * |
1842 | * Given the following |
1843 | * |
1844 | * \verbatim |
1845 | |
1846 | +-----+-----+-----+-----+-----+-----+ (1.0. 1.0) Domain box |
1847 | | | | | | | | |
1848 | | | | | | | | |
1849 | | | | | | | | |
1850 | +-----+-----+-----+-----+----{2}----+ |
1851 | | | | | | | | |
1852 | | | | | | | | | |
1853 | | | | | | | | |
1854 | +-----+-----+-----+-----+-----+-----+ |
1855 | | | | | | | | |
1856 | | | | | | | | |
1857 | | | | | | | |
1858 | +-----+----{1}----+-----+-----+-----+ |
1859 | | | | | | | | |
1860 | | | | | | | | |
1861 | +-----+-----+-----+-----+-----+-----+ |
1862 | | | | | | | | |
1863 | | | | | | | | |
1864 | | | | | | | | |
1865 | +-----+-----+-----+-----+-----+-----+ |
1866 | (0.0, 0.0) |
1867 | |
1868 | |
1869 | + = grid points |
1870 | |
1871 | \verbatim |
1872 | |
1873 | Giving a box P1 = (2,2), P2 = (5,4) |
1874 | it return a box (0.333333,0.33333333) (0.8333333,0.8) |
1875 | |
1876 | * |
1877 | * \param b Box in grid units |
1878 | * |
1879 | * \return the box in domain space |
1880 | * |
1881 | */ |
1882 | inline Box<dim,T> convertCellUnitsIntoDomainSpace(const Box<dim,long int> & b_d) const |
1883 | { |
1884 | Box<dim,T> be; |
1885 | |
1886 | for (size_t i = 0 ; i < dim ; i++) |
1887 | { |
1888 | if ((long int)gr_cell.size(i) - (long int)off[i] == b_d.getLow(i)) |
1889 | be.setLow(i,box.getHigh(i)); |
1890 | else if ((long int)off[i] == b_d.getLow(i)) |
1891 | be.setLow(i,box.getLow(i)); |
1892 | else |
1893 | be.setLow(i,(b_d.getLow(i) - off[i]) * box_unit.getP2()[i] + box.getLow(i)); |
1894 | |
1895 | if ((long int)gr_cell.size(i) - (long int)off[i] == b_d.getHigh(i)) |
1896 | be.setHigh(i,box.getHigh(i)); |
1897 | else if ((long int)off[i] == b_d.getHigh(i)) |
1898 | be.setHigh(i,box.getLow(i)); |
1899 | else |
1900 | be.setHigh(i,(b_d.getHigh(i) - off[i]) * box_unit.getP2()[i] + box.getLow(i)); |
1901 | } |
1902 | |
1903 | return be; |
1904 | } |
1905 | |
1906 | |
1907 | /*! \brief Convert a Box from grid units into domain space (Negative contour included, positive contour excluded) |
1908 | * |
1909 | * Given the following |
1910 | * |
1911 | * \verbatim |
1912 | |
1913 | +-----+-----+-----+-----+-----+-----+ (1.0. 1.0) Domain box |
1914 | | | | | | | | |
1915 | | | *-----------------------* | |
1916 | | | | | | | | | | |
1917 | +-----+--|--+-----+-----+----{2}-|--+ |
1918 | | | | | | | | | | |
1919 | | | | | | | | | | |
1920 | | | | | | | | | | |
1921 | +-----+--|--+-----+-----+-----+--|--+ |
1922 | | | | | | | | | | |
1923 | | | | | | | | | | |
1924 | | | | | | | | | | |
1925 | +-----+--|-{1}----+-----+-----+--|--+ |
1926 | | | | | | | | | | |
1927 | | | *-----------------------* | |
1928 | | | | | | | | |
1929 | +-----+-----+-----+-----+-----+-----+ |
1930 | | | | | | | | |
1931 | | | | | | | | |
1932 | | | | | | | | |
1933 | +-----+-----+-----+-----+-----+-----+ |
1934 | (0.0, 0.0) |
1935 | |
1936 | |
1937 | + = grid points |
1938 | |
1939 | \verbatim |
1940 | |
1941 | Giving a box P1 = (2,2), P2 = (5,4) |
1942 | it return a box (0.25,0.25) (0.916666,0.9) |
1943 | |
1944 | * |
1945 | * \param b Box in grid units |
1946 | * |
1947 | * The box is never allowed to be bigger than the domain and is always cropped |
1948 | * by the domain size |
1949 | * |
1950 | * \return the box in domain space |
1951 | * |
1952 | */ |
1953 | inline Box<dim,T> convertCellUnitsIntoDomainSpaceMiddle(const Box<dim,long int> & b_d) const |
1954 | { |
1955 | Box<dim,T> be; |
1956 | |
1957 | for (size_t i = 0 ; i < dim ; i++) |
1958 | { |
1959 | if ((long int)gr_cell.size(i) - (long int)off[i] == b_d.getLow(i)) |
1960 | {be.setLow(i,box.getHigh(i));} |
1961 | else if ((long int)off[i] == b_d.getLow(i)) |
1962 | {be.setLow(i,box.getLow(i));} |
1963 | else |
1964 | {be.setLow(i,(b_d.getLow(i) - off[i]) * box_unit.getP2()[i] + box.getLow(i) - box_unit.getP2()[i] / 2.0);} |
1965 | |
1966 | if ((long int)gr_cell.size(i) - (long int)off[i] == b_d.getHigh(i)) |
1967 | {be.setHigh(i,box.getHigh(i));} |
1968 | else if ((long int)off[i] == b_d.getHigh(i)) |
1969 | {be.setHigh(i,box.getLow(i));} |
1970 | else |
1971 | {be.setHigh(i,(b_d.getHigh(i) - off[i]) * box_unit.getP2()[i] + box.getLow(i) + box_unit.getP2()[i] / 2.0);} |
1972 | } |
1973 | |
1974 | return be; |
1975 | } |
1976 | |
1977 | /*! \brief Return the number of divisions of the Cell Decomposer (including padding) |
1978 | * |
1979 | * \return the number of divisions |
1980 | * |
1981 | */ |
1982 | const size_t (& getDiv() const)[dim] |
1983 | { |
1984 | return gr_cell.getSize(); |
1985 | } |
1986 | |
1987 | |
1988 | /*! \brief Return the number of divisions of the Cell Decomposer (without padding) |
1989 | * |
1990 | * \return the number of divisions |
1991 | * |
1992 | */ |
1993 | const size_t (& getDivWP() const)[dim] |
1994 | { |
1995 | for (size_t i = 0 ; i < dim ; i++) |
1996 | {div_wp[i] = gr_cell.getSize()[i] - 2*getPadding(i);} |
1997 | |
1998 | return div_wp; |
1999 | } |
2000 | |
2001 | /*! \brief Return the domain where the CellDecomposer is defined |
2002 | * |
2003 | * \return The domain of the remote machine |
2004 | * |
2005 | */ |
2006 | const Box<dim,T> & getDomain() const |
2007 | { |
2008 | return box; |
2009 | } |
2010 | |
2011 | /*! \brief it swap the content of two Cell Decomposer |
2012 | * |
2013 | * \param cd CellDecomposer to swap with |
2014 | * |
2015 | */ |
2016 | inline void swap(CellDecomposer_sm<dim,T,transform> & cd) |
2017 | { |
2018 | // swap all the members |
2019 | p_middle.swap(cd.p_middle); |
2020 | |
2021 | // Point transformation before get the Cell object (useful for example to shift the cell list) |
2022 | transform t_t = t; |
2023 | t = cd.t; |
2024 | cd.t = t_t; |
2025 | |
2026 | // Total number of cell |
2027 | size_t tot_n_cell_t = tot_n_cell; |
2028 | tot_n_cell = cd.tot_n_cell; |
2029 | cd.tot_n_cell = tot_n_cell_t; |
2030 | |
2031 | box.swap(cd.box); |
2032 | box_unit.swap(cd.box_unit); |
2033 | gr_cell.swap(cd.gr_cell); |
2034 | gr_cell2.swap(cd.gr_cell2); |
2035 | |
2036 | for (size_t i = 0 ; i < dim ; i++) |
2037 | { |
2038 | size_t off_t = off[i]; |
2039 | off[i] = cd.off[i]; |
2040 | cd.off[i] = off_t; |
2041 | |
2042 | size_t cs_t = cell_shift.get(i); |
2043 | cell_shift.get(i) = cd.cell_shift.get(i); |
2044 | cd.cell_shift.get(i) = cs_t; |
2045 | } |
2046 | } |
2047 | |
2048 | /*! \brief Check that the CellDecomposer is the same |
2049 | * |
2050 | * \param cd Cell decomposer |
2051 | * |
2052 | * \return true if the two CellDecomposer are the same |
2053 | * |
2054 | */ |
2055 | inline bool operator==(const CellDecomposer_sm<dim,T,transform> & cd) |
2056 | { |
2057 | if (meta_compare<Point<dim,T>>::meta_compare_f(p_middle,cd.p_middle) == false) |
2058 | return false; |
2059 | |
2060 | if (t != cd.t) |
2061 | return false; |
2062 | |
2063 | if (tot_n_cell != cd.tot_n_cell) |
2064 | return false; |
2065 | |
2066 | if (box != cd.box) |
2067 | return false; |
2068 | |
2069 | if (box_unit != cd.box_unit) |
2070 | return false; |
2071 | |
2072 | if (gr_cell != cd.gr_cell) |
2073 | return false; |
2074 | |
2075 | if (gr_cell2 != cd.gr_cell2) |
2076 | return false; |
2077 | |
2078 | for (size_t i = 0 ; i < dim ; i++) |
2079 | { |
2080 | if (off[i] != cd.off[i]) |
2081 | return false; |
2082 | |
2083 | if (cell_shift.get(i) != cd.cell_shift.get(i)) |
2084 | return false; |
2085 | } |
2086 | |
2087 | return true; |
2088 | } |
2089 | |
2090 | /*! \brief Check that the CellDecomposer is the same |
2091 | * |
2092 | * \param cd Cell decomposer |
2093 | * |
2094 | * \return true if the two CellDecomposer is different |
2095 | * |
2096 | */ |
2097 | inline bool operator!=(const CellDecomposer_sm<dim,T,transform> & cd) |
2098 | { |
2099 | return ! this->operator==(cd); |
2100 | } |
2101 | |
2102 | /*! \brief Return the number of padding cells of the Cell decomposer |
2103 | * |
2104 | * \param i dimension |
2105 | * |
2106 | * \return the number of padding cells |
2107 | * |
2108 | */ |
2109 | size_t getPadding(size_t i) const |
2110 | { |
2111 | return off[i]; |
2112 | } |
2113 | |
2114 | /*! \brief Return the number of padding cells of the Cell decomposer as an array |
2115 | * |
2116 | * |
2117 | * \return the number of padding cells |
2118 | * |
2119 | */ |
2120 | size_t (& getPadding())[dim] |
2121 | { |
2122 | return off; |
2123 | } |
2124 | |
2125 | /*! \brief Return the index of the first cell in the domain |
2126 | * |
2127 | * This function make sense if the CellDecomposer has been |
2128 | * created with setDimensions with div and div2 |
2129 | * |
2130 | * \return the first domain cell |
2131 | * |
2132 | */ |
2133 | grid_key_dx<dim> getStartDomainCell() const |
2134 | { |
2135 | grid_key_dx<dim> key; |
2136 | |
2137 | for (size_t i = 0 ; i < dim ; i++) |
2138 | { |
2139 | key.set_d(i, cell_shift.get(i)); |
2140 | } |
2141 | |
2142 | return key; |
2143 | } |
2144 | |
2145 | /*! \brief Return the index of the last cell in the domain |
2146 | * |
2147 | * This function make sense if the CellDecomposer has been |
2148 | * created with setDimensions with div and div2 |
2149 | * |
2150 | * \return the last domain cell |
2151 | * |
2152 | */ |
2153 | grid_key_dx<dim> getStopDomainCell() const |
2154 | { |
2155 | grid_key_dx<dim> key; |
2156 | |
2157 | for (size_t i = 0 ; i < dim ; i++) |
2158 | { |
2159 | key.set_d(i,cell_shift.get(i) + gr_cell2.size(i) - 2*getPadding(i) - 1); |
2160 | } |
2161 | |
2162 | return key; |
2163 | } |
2164 | |
2165 | /*! \brief Return the cell shift |
2166 | * |
2167 | * \return the cell shifting |
2168 | * |
2169 | */ |
2170 | grid_key_dx<dim> getShift() const |
2171 | { |
2172 | grid_key_dx<dim> k; |
2173 | |
2174 | for (size_t i = 0 ; i < dim ; i++) |
2175 | k.set_d(i,cell_shift.get(i)); |
2176 | |
2177 | return k; |
2178 | } |
2179 | |
2180 | /*! \brief Return the grid structure of the internal cell list |
2181 | * |
2182 | * \return the grid information |
2183 | * |
2184 | */ |
2185 | const grid_sm<dim,void> & getInternalGrid() const |
2186 | { |
2187 | return gr_cell2; |
2188 | } |
2189 | }; |
2190 | |
2191 | |
2192 | #endif /* CELLDECOMPOSER_HPP_ */ |
2193 | |