File: ClosePackIterator.h

package info (click to toggle)
esys-particle 2.1-4
  • links: PTS, VCS
  • area: main
  • in suites: wheezy
  • size: 7,284 kB
  • sloc: cpp: 77,304; python: 5,647; makefile: 1,176; sh: 10
file content (129 lines) | stat: -rw-r--r-- 3,496 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
/////////////////////////////////////////////////////////////
//                                                         //
// Copyright (c) 2003-2011 by The University of Queensland //
// Earth Systems Science Computational Centre (ESSCC)      //
// http://www.uq.edu.au/esscc                              //
//                                                         //
// Primary Business: Brisbane, Queensland, Australia       //
// Licensed under the Open Software License version 3.0    //
// http://www.opensource.org/licenses/osl-3.0.php          //
//                                                         //
/////////////////////////////////////////////////////////////


#ifndef ESYS_LSMCLOSEPACKITERATOR_H
#define ESYS_LSMCLOSEPACKITERATOR_H

#include "Foundation/BoundingBox.h"
#include "Foundation/vec3.h"
#include "Geometry/Vec3L.h"
#include "Geometry/ClosePackOrientation.h"

namespace esys
{
  namespace lsm
  {
    template <int NI, int NJ, int NK>
    class TmplMatrix
    {
    public:
      TmplMatrix();

      TmplMatrix(const TmplMatrix &m);

      TmplMatrix &operator=(const TmplMatrix &m);

      const double &operator()(int i, int j, int k) const;

      double &operator()(int i, int j, int k);

      int getNumI() const;

      int getNumJ() const;

      int getNumK() const;

    private:
      double m_matrix[NI][NJ][NK];
    };

    /**
     * Base class for iterators used to generate centre-points of spheres
     * arranged in a close-packing.
     */
    class ClosePackIterator
    {
    public:
      static const double SQRT_1_OVER_3;
      static const double SQRT_8_OVER_3;
      static const double SQRT_3;

      /**
       * Creates default empty iterator.
       */
      inline ClosePackIterator();

      /**
       * Creates an iterator which will iterate over numI*numJ*numK
       * centre points of spheres with radius sphereRadius.
       * @param numI number of spheres in the i direction.
       * @param numJ number of spheres in the j direction.
       * @param numK number of spheres in the k direction.
       * @param sphereRadius radius of spheres in the packing.
       * @param orientation specifies the axis alignment of layers.
       */
      inline ClosePackIterator(
        int numI,
        int numJ,
        int numK,
        double sphereRadius,
        ClosePackOrientation orientation = DEFAULT_ORIENT
      );

      /**
       * Returns whether there is another centre point in the
       * iteration sequence.
       */
      inline bool hasNext() const;

      /**
       * Returns the next centre-point in the iteration sequence.
       */
      inline Vec3 next();

      /**
       * Returns the radius of spheres used in the iteration.
       */
      inline double getRadius() const;

    protected:
      typedef TmplMatrix<3,6,6> OffsetMatrix;

      inline void incrementDimIndex();

      inline double getOffset(int i) const;

      inline const Vec3 &getMinPt() const;

      inline void setMinPt(const Vec3 &pt) const;

      inline void setDimRepeat(const Vec3L &dimRepeat);

      inline void setOffsetMatrix(const OffsetMatrix &offsetMatrix);

    private:
      static  Vec3L s_orientationDimMap[NUM_ORIENTATIONS];
      double        m_radius;
      Vec3          m_minPt;
      OffsetMatrix  m_offsetMatrix;
      Vec3L         m_dimRepeat;
      Vec3L         m_dimCount;
      Vec3L         m_dimIdx;
      Vec3L         m_dim;
    };
  }
}

#include "Geometry/ClosePackIterator.hpp"

#endif