00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024
00025
00026
00027
00028
00029
00030
00031
00032
00033
00034
00035
00036
00037
00038
00039
00040
00041
00042
00043
00044
00045
00046
00047
00048
00049
00050
00051
00052 #ifndef _BOXLAYOUT_H_
00053 #define _BOXLAYOUT_H_
00054
00055 #ifndef WRAPPER
00056 #include "Box.H"
00057 #include "Vector.H"
00058 class DataIterator;
00059 class LayoutIterator;
00060 #include "RefCountedPtr.H"
00061 #include "DataIndex.H"
00062 #include "SPMD.H"
00063 #endif
00064
00065 #ifndef DOXYGEN
00066
00067 struct Entry
00068 {
00069 Entry()
00070 :
00071 m_procID(procID())
00072 {}
00073
00074 Entry(const Box& a_box)
00075 :
00076 box(a_box),
00077 m_procID(procID())
00078 {}
00079
00080 Entry(const Box& a_box, const unsigned int a_index)
00081 :
00082 box(a_box),
00083 index(a_index),
00084 m_procID(procID())
00085 {}
00086
00087 bool operator < (const Entry& rhs) const
00088 { return box < rhs.box;}
00089
00090 Box box;
00091 unsigned int index;
00092 unsigned int m_procID;
00093 };
00094
00095 #endif
00096
00098
00153 class BoxLayout
00154 {
00155 public:
00156
00161
00163
00166 BoxLayout();
00167
00169
00172 BoxLayout(const Vector<Box>& a_boxes,
00173 const Vector<int>& a_procIDs);
00174
00176
00181 virtual
00182 ~BoxLayout();
00183
00185
00188 BoxLayout& operator=(const BoxLayout& a_rhs);
00189
00191
00197 virtual void
00198 define(const Vector<Box>& a_boxes,
00199 const Vector<int>& a_procIDs);
00200
00207
00209
00211 const Box&
00212 operator[](const LayoutIndex& it) const;
00213
00215
00217 Box&
00218 operator[](const LayoutIndex& it);
00219
00221
00228 Box get(const LayoutIndex& it) const;
00229
00231
00233 Box get(const LayoutIterator& it) const;
00234
00236
00238 Box get(const DataIterator& it) const;
00239
00241
00249 unsigned int
00250 procID(const LayoutIndex& a_index) const ;
00251
00253
00255 int numBoxes(const int procID) const;
00256
00258
00260 inline unsigned int
00261 size() const ;
00262
00265 unsigned int index(const LayoutIndex& index) const;
00266
00273
00275
00278 inline bool
00279 operator==(const BoxLayout& rhs) const ;
00280
00282
00285 bool eq(const BoxLayout& rhs) const { return *this == rhs;}
00286
00290 bool
00291 isClosed() const;
00292
00294
00296 bool
00297 isSorted() const {return isClosed();}
00298
00301 bool check(const LayoutIndex& index) const
00302 { return index.m_layoutIntPtr == m_layout;}
00303
00310
00312
00316 DataIndex
00317 addBox(const Box& box, int procID);
00318
00320
00322 virtual void
00323 close();
00324
00326
00351 virtual void
00352 deepCopy(const BoxLayout& a_source);
00353
00355
00361 bool coarsenable(int refRatio) const;
00362
00364
00379 friend void coarsen(BoxLayout& output,
00380 const BoxLayout& input,
00381 int refinement);
00382
00384
00399 friend void refine(BoxLayout& output,
00400 const BoxLayout& input,
00401 int refinement);
00402
00404
00407 void
00408 setProcID(const LayoutIndex& a_index, unsigned int a_procID);
00409
00410
00412
00416 void
00417 sort();
00418
00421 void aliasAddBox(const Box& box);
00422
00425 void aliasClose();
00426
00433
00435
00439 DataIterator
00440 dataIterator() const;
00441
00443
00447 LayoutIterator
00448 layoutIterator() const;
00449
00456
00457 #ifndef WRAPPER
00458
00459
00462 void
00463 print() const;
00464 #endif
00465
00466
00469 void p() const { print();}
00470
00473 protected:
00474
00475 void setIndexVector();
00476 friend class LayoutIterator;
00477 friend class DataIterator;
00478
00479 RefCountedPtr<Vector<Entry> > m_boxes;
00480 RefCountedPtr<Vector<unsigned int> > m_index;
00481 RefCountedPtr<int> m_layout;
00482 RefCountedPtr<bool> m_closed;
00483 RefCountedPtr<DataIterator> m_dataIterator;
00484
00485 private:
00486
00487 };
00488
00489 void coarsen_bl(BoxLayout& output, const BoxLayout& input, int refinement);
00490
00491 void refine_bl(BoxLayout& output, const BoxLayout& input, int refinement);
00492
00493 #ifndef WRAPPER
00494
00495 inline
00496 void coarsen_bl(BoxLayout& output, const BoxLayout& input, int refinement)
00497 { coarsen(output, input, refinement);}
00498
00499 inline
00500 void refine_bl(BoxLayout& output, const BoxLayout& input, int refinement)
00501 { refine(output, input, refinement);}
00502
00503
00504
00505
00506
00507
00508
00509
00510
00511
00512
00513
00514
00515
00516
00517
00518
00519
00520
00521
00522
00523
00524
00525
00526
00527
00528
00529
00530
00531
00532 inline const Box&
00533 BoxLayout::operator[](const LayoutIndex& index) const
00534 {
00535 assert(check(index));
00536 return m_boxes->operator[](m_index->operator[](index.m_index)).box;
00537 }
00538
00539 inline Box&
00540 BoxLayout::operator[](const LayoutIndex& index)
00541 {
00542 if(*m_closed)
00543 {
00544 MayDay::Error("attempt to modify closed BoxLayout");
00545 }
00546 assert(check(index));
00547 return m_boxes->operator[](m_index->operator[](index.m_index)).box;
00548 }
00549
00550 inline bool
00551 BoxLayout::operator==(const BoxLayout& rhs) const
00552 {
00553 return m_boxes == rhs.m_boxes;
00554 }
00555
00556
00557
00558
00559 inline Box
00560 BoxLayout::get(const LayoutIndex& index) const
00561 {
00562 assert(check(index));
00563 return m_boxes->operator[](m_index->operator[](index.m_index)).box;
00564 }
00565
00566 inline unsigned int
00567 BoxLayout::index(const LayoutIndex& a_index) const
00568 {
00569 return m_index->operator[](a_index.m_index);
00570 }
00571
00572 inline bool
00573 BoxLayout::isClosed() const
00574 {
00575 return *m_closed;
00576 }
00577
00578 inline unsigned int
00579 BoxLayout::procID(const LayoutIndex& a_index) const
00580 {
00581 assert(check(a_index));
00582 return m_boxes->operator[](m_index->operator[](a_index.m_index)).m_procID;
00583 }
00584
00585 inline void
00586 BoxLayout::setProcID(const LayoutIndex& a_index, unsigned int a_procID)
00587 {
00588 assert(check(a_index));
00589 m_boxes->operator[](m_index->operator[](a_index.m_index)).m_procID = a_procID;
00590 }
00591
00592 inline unsigned int
00593 BoxLayout::size() const
00594 {
00595 return m_boxes->size();
00596 }
00597
00598
00599
00600
00601 std::ostream& operator<<(std::ostream& os, const BoxLayout& a_layout);
00602
00603 #endif
00604
00605 #endif // DATAPLAN_H