ViewVC Help
View File | Revision Log | Show Annotations | View Changeset | Root Listing
root/group/branches/new_design/OOPSE-2.0/src/primitives/Molecule.hpp
(Generate patch)

Comparing branches/new_design/OOPSE-2.0/src/primitives/Molecule.hpp (file contents):
Revision 1692 by tim, Mon Nov 1 20:15:58 2004 UTC vs.
Revision 1733 by tim, Fri Nov 12 06:19:04 2004 UTC

# Line 38 | Line 38
38   #include "math/Vector3.hpp"
39   #include "primitives/Atom.hpp"
40   #include "primitives/RigidBody.hpp"
41 + //#include "primitives/Bond.hpp"
42 + //#include "primitives/Bend.hpp"
43 + //#include "primitives/Torsion.hpp"
44 + #include "primitives/SRI.hpp"
45 + #include "primitives/CutoffGroup.hpp"
46  
47   namespace oopse{
48  
49 + class Constraint;
50 +
51   /**
52   * @class Molecule Molecule.hpp "primitives/Molecule.hpp"
53   * @brief
# Line 48 | Line 55 | class Molecule {
55   class Molecule {
56      public:
57  
58 <        Molecule();
58 >        typedef std::vector<Atom*>::iterator AtomIterator;
59 >        typedef std::vector<Bond*>::iterator BondIterator;
60 >        typedef std::vector<Bend*>::iterator BendIterator;
61 >        typedef std::vector<Torsion*>::iterator TorsionIterator;
62 >        typedef std::vector<RigidBody*>::iterator RigidBodyIterator;
63 >        typedef std::vector<CutoffGroup*>::iterator CutoffGroupIterator;
64 >        typedef std::vector<StuntDouble*>::iterator IntegrableObjectIterator;        
65 >
66 >        Molecule(int stampId, int globalIndex, const std::string& molName);
67          virtual ~Molecule();
68  
69          /**
# Line 60 | Line 75 | class Molecule {
75          }
76  
77          /**
78 <         * Sets the global index of this molecule.
79 <         * @param new global index to be set
78 >         * Returns the stamp id of this molecule
79 >         * @note Ideally, every molecule should keep a pointer of its molecule stamp instead of its
80 >         * stamp id. However, the pointer will become invalid, if the molecule migrate to other processor.
81           */
82 <        void setGlobalIndex(int index) {
83 <            return globalIndex_;
82 >        int getStampId() {
83 >            return stampId_;
84          }
85 +
86 +        /** Returns the name of the molecule */
87 +        std::string getType() {
88 +            return moleculeName_;
89 +        }
90          
91 <        /**
92 <         * Returns the local index of this molecule
93 <         * @return the local index of this molecule
91 >        /**
92 >         * Sets the global index of this molecule.
93 >         * @param new global index to be set
94           */
95 <        int getLocalIndex() {
96 <            return localIndex_;
95 >        int setGlobalIndex(int index) {
96 >            return globalIndex_;
97          }
98  
99 +        
100          /** add an atom into this molecule */
101          void addAtom(Atom* atom);
102  
# Line 131 | Line 153 | class Molecule {
153              return cutoffGroups_.size();
154          }
155  
156 +        /** Returns the total number of constraints in this molecule */
157 +        unsigned int getNConstraints() {
158 +            return constraints_.size();
159 +        }
160 +
161 +        Atom* getAtomAt(unsigned int i) {
162 +            assert(i < atoms_.size());
163 +            return atoms_[i];
164 +        }
165          /**
166           * Returns the first atom in this molecule and initialize the iterator.
167           * @return the first atom, return NULL if there is not cut off group in this molecule
# Line 198 | Line 229 | class Molecule {
229           */        
230          CutoffGroup* nextCutoffGroup(std::vector<CutoffGroup*>::iterator& i);
231  
232 +        Constraint* beginConstraint(std::vector<Constraint*>::iterator& i);
233 +
234 +        Constraint* nextConstraint(std::vector<Constraint*>::iterator& i);
235 +        
236          //void setStampID( int info ) {stampID = info;}
237  
238          void calcForces( void );
# Line 212 | Line 247 | class Molecule {
247          Vector3d getCom();
248  
249          /** Moves the center of this molecule */
250 <        void moveCom(const Vetor3d& delta);
250 >        void moveCom(const Vector3d& delta);
251  
252          /** Returns the velocity of center of mass of this molecule */
253          Vector3d getComVel();
# Line 220 | Line 255 | class Molecule {
255          /** Returns the total mass of this molecule */
256          double getTotalMass();
257  
258 <        friend std::ostream& operator <<(std::ostream& o, const Molecule& mol);
258 >        friend std::ostream& operator <<(std::ostream& o, Molecule& mol);
259          
260      private:
261 <        int localIndex_;
261 >        
262          int globalIndex_;
263  
264          std::vector<Atom*> atoms_;
# Line 233 | Line 268 | class Molecule {
268          std::vector<RigidBody*> rigidBodies_;
269          std::vector<StuntDouble*> integrableObjects_;
270          std::vector<CutoffGroup*> cutoffGroups_;
271 +        std::vector<Constraint*> constraints_;
272 +
273 +        int stampId_;
274 +        std::string moleculeName_;
275   };
276  
277   } //namespace oopse

Diff Legend

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