ViewVC Help
View File | Revision Log | Show Annotations | View Changeset | Root Listing
root/OpenMD/trunk/src/applications/hydrodynamics/RoughShell.hpp
(Generate patch)

Comparing trunk/src/applications/hydrodynamics/RoughShell.hpp (file contents):
Revision 906 by tim, Fri Mar 17 23:20:35 2006 UTC vs.
Revision 972 by gezelter, Wed May 24 18:31:12 2006 UTC

# Line 44 | Line 44 | namespace oopse {
44   #include "applications/hydrodynamics/ApproximationModel.hpp"
45   #include "applications/hydrodynamics/CompositeShape.hpp"
46   namespace oopse {
47 < /**
48 < * @class Grid3d
49 < * A generic 3d grid class
50 < */
51 < template<class Elem>
52 < class Grid3D {
53 <    public:
54 <        Grid3D(unsigned int dim1, unsigned int dim2, unsigned int dim3) : dim1_(dim1), dim2_(dim2), dim3_(dim3) {
55 <            data_.resize(dim1_*dim2_*dim3_);
56 <        }
57 <        Elem& operator ()(unsigned int i, unsigned int j, unsigned int k) {
58 <            int index = isValidGrid(i, j , k);
59 <            assert(index != -1);
60 <            return data_[index];
61 <        }
62 <
63 <        const Elem& operator () (unsigned int i, unsigned int j, unsigned int k) const {
64 <            int index = isValidGrid(i, j , k);
65 <            assert(index != -1);
66 <            return data_[index];
67 <        }
68 <        
69 <        std::vector<Elem> getAllNeighbors(unsigned int i, unsigned int j, unsigned int k) {
70 <            std::vector<Elem> result;
71 <            int index;
72 <            index = isValidGrid(i-1, j, k);
73 <            if (index != -1)
74 <                result.push_back(data_[index]);
75 <
76 <            index = isValidGrid(i+1, j, k);
77 <            if (index != -1)
78 <                result.push_back(data_[index]);
79 <
80 <            index = isValidGrid(i, j-1, k);
81 <            if (index != -1)
82 <                result.push_back(data_[index]);
83 <
84 <            index = isValidGrid(i, j+1, k);
85 <            if (index != -1)
86 <                result.push_back(data_[index]);
87 <
88 <            index = isValidGrid(i, j, k-1);
89 <            if (index != -1)
90 <                result.push_back(data_[index]);
91 <
92 <            index = isValidGrid(i, j, k+1);
93 <            if (index != -1)
94 <                result.push_back(data_[index]);
95 <
96 <            return result;
97 <        }
98 <    private:
99 <
100 <        int isValidGrid(unsigned int i, unsigned int j, unsigned int k) const {
101 <            int index = i * dim2_*dim3_ + j * dim3_ + k;
102 <            return index < data_.size() ? index : -1;
103 <        };
104 <        
105 <        unsigned int dim1_;
106 <        unsigned int dim2_;
107 <        unsigned int dim3_;
108 <        std::vector<Elem> data_;
109 <
110 < };
47 >  /**
48 >   * @class Grid3d
49 >   * A generic 3d grid class
50 >   */
51 >  template<class Elem>
52 >  class Grid3D {
53 >  public:
54 >    Grid3D(unsigned int dim1, unsigned int dim2, unsigned int dim3) : dim1_(dim1), dim2_(dim2), dim3_(dim3) {
55 >      data_.resize(dim1_*dim2_*dim3_);
56 >    }
57 >    Elem& operator ()(unsigned int i, unsigned int j, unsigned int k) {
58 >      int index = isValidGrid(i, j , k);
59 >      assert(index != -1);
60 >      return data_[index];
61 >    }
62      
63 +    const Elem& operator () (unsigned int i, unsigned int j, unsigned int k) const {
64 +      int index = isValidGrid(i, j , k);
65 +      assert(index != -1);
66 +      return data_[index];
67 +    }
68      
69 < class RoughShell : public ApproximationModel {
70 <    public:
71 <        RoughShell(StuntDouble* sd, SimInfo* info);
72 <        virtual ~RoughShell() { delete shape_;}
73 <        void setSigma(double sigma) {sigma_ = sigma;}
74 <        double getSigma() {return sigma_;}
75 <    private:
76 <        virtual bool createBeads(std::vector<BeadParam>& beads);
77 <        //StuntDoubleShape sdShape_;
78 <        double sigma_;
79 <        Shape* shape_;
80 < };
81 <
69 >    std::vector<Elem> getAllNeighbors(unsigned int i, unsigned int j, unsigned int k) {
70 >      std::vector<Elem> result;
71 >      int index;
72 >      index = isValidGrid(i-1, j, k);
73 >      if (index != -1)
74 >        result.push_back(data_[index]);
75 >      
76 >      index = isValidGrid(i+1, j, k);
77 >      if (index != -1)
78 >        result.push_back(data_[index]);
79 >      
80 >      index = isValidGrid(i, j-1, k);
81 >      if (index != -1)
82 >        result.push_back(data_[index]);
83 >      
84 >      index = isValidGrid(i, j+1, k);
85 >      if (index != -1)
86 >        result.push_back(data_[index]);
87 >      
88 >      index = isValidGrid(i, j, k-1);
89 >      if (index != -1)
90 >        result.push_back(data_[index]);
91 >      
92 >      index = isValidGrid(i, j, k+1);
93 >      if (index != -1)
94 >        result.push_back(data_[index]);
95 >      
96 >      return result;
97 >    }
98 >  private:
99 >    
100 >    int isValidGrid(unsigned int i, unsigned int j, unsigned int k) const {
101 >      int index = i * dim2_*dim3_ + j * dim3_ + k;
102 >      return index < data_.size() ? index : -1;
103 >    };
104 >    
105 >    unsigned int dim1_;
106 >    unsigned int dim2_;
107 >    unsigned int dim3_;
108 >    std::vector<Elem> data_;
109 >    
110 >  };
111 >  
112 >  
113 >  class RoughShell : public ApproximationModel {
114 >  public:
115 >    RoughShell(StuntDouble* sd, SimInfo* info);
116 >    virtual ~RoughShell() { delete shape_;}
117 >    void setSigma(RealType sigma) {sigma_ = sigma;}
118 >    RealType getSigma() {return sigma_;}
119 >  private:
120 >    virtual bool createBeads(std::vector<BeadParam>& beads);
121 >    //StuntDoubleShape sdShape_;
122 >    RealType sigma_;
123 >    Shape* shape_;
124 >  };
125 >  
126   }
127   #endif

Diff Legend

– Removed lines
+ Added lines
< Changed lines
> Changed lines