~esys-p-dev/esys-particle/2.1

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
/////////////////////////////////////////////////////////////
//                                                         //
// 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_LSMCIRCULARNEIGHBOURTABLE_H
#define ESYS_LSMCIRCULARNEIGHBOURTABLE_H

#include "Geometry/NeighbourTable.h"
#include <boost/pool/object_pool.hpp>
#include <boost/shared_ptr.hpp>

#include <sstream>
#include <stdexcept>
#include <set>

namespace esys
{
  namespace lsm
  {
    /**
     *
     */
    template <class TmplParticle>
    class CircularNeighbourTable : public NeighbourTable<TmplParticle>
    {
    public:
      typedef NeighbourTable<TmplParticle>                  Inherited;
      typedef typename Inherited::Particle                  Particle;
      typedef typename Inherited::ParticleVector            ParticleVector;
      typedef std::set<typename ParticleVector::value_type> ParticleSet;
      typedef boost::object_pool<Particle>                  ParticlePool;
      typedef boost::shared_ptr<ParticlePool>               ParticlePoolPtr;
      typedef std::vector<bool>                             BoolVector;

    public:
      CircularNeighbourTable(
        const BoundingBox  &bBox,
        double             gridSpacing,
        const BoolVector   &periodicDimensions = BoolVector(3, false),
        double             circBorderWidth = 0.0
      );

      CircularNeighbourTable(
        const BoundingBox  &bBox,
        double             gridSpacing,
        ParticlePoolPtr    particlePoolPtr,
        const BoolVector   &periodicDimensions = BoolVector(3, false),
        double             circBorderWidth = 0.0
      );

    public:
      void checkPeriodicDimensions();

      virtual ~CircularNeighbourTable();

      void setCircularBorderWidth(
        double circBorderWidth,
        double gridSpacing
      );
      
      void setCircularBorderWidth(double circBorderWidth);

      void resize(
        const BoundingBox &bBox,
        double gridSpacing,
        double circBorderWidth
      );

      void resize(
        const BoundingBox &bBox,
        double gridSpacing
      );
      
      void insertClone(Particle *pParticle, const Vec3 &newPosition);

      bool havePeriodicDimensions() const;

      Vec3 getModdedPosn(const Vec3 &posn) const;
      
      void insert(Particle *pParticle);

      void insert(Particle &particle);

      size_t getNumClonedParticles() const;

      size_t getNumParticles() const;

      const BoolVector &getPeriodicDimensions() const;

    protected:
      bool isClone(Particle *p) const;
      
      ParticleVector getNonClonedParticles();
      
      void clearClonedParticles();

    private:
      BoolVector      m_periodicDimensions;
      ParticlePoolPtr m_particlePoolPtr;
      ParticleSet     m_clonedParticleSet;
      int             m_circGridWidth;
      int             m_periodicDimIndex;
    };
  };
};

#include "Geometry/CircularNeighbourTable.hpp"

#endif