ViewVC Help
View File | Revision Log | Show Annotations | View Changeset | Root Listing
root/group/trunk/OOPSE-3.0/src/types/MoleculeStamp.cpp
Revision: 2544
Committed: Wed Jan 11 19:01:20 2006 UTC (18 years, 5 months ago) by tim
File size: 24077 byte(s)
Log Message:
instead of printing to std::cout, throwing an exception when error is found.

File Contents

# User Rev Content
1 gezelter 2204 /*
2 gezelter 1930 * Copyright (c) 2005 The University of Notre Dame. All Rights Reserved.
3     *
4     * The University of Notre Dame grants you ("Licensee") a
5     * non-exclusive, royalty free, license to use, modify and
6     * redistribute this software in source and binary code form, provided
7     * that the following conditions are met:
8     *
9     * 1. Acknowledgement of the program authors must be made in any
10     * publication of scientific results based in part on use of the
11     * program. An acceptable form of acknowledgement is citation of
12     * the article in which the program was described (Matthew
13     * A. Meineke, Charles F. Vardeman II, Teng Lin, Christopher
14     * J. Fennell and J. Daniel Gezelter, "OOPSE: An Object-Oriented
15     * Parallel Simulation Engine for Molecular Dynamics,"
16     * J. Comput. Chem. 26, pp. 252-271 (2005))
17     *
18     * 2. Redistributions of source code must retain the above copyright
19     * notice, this list of conditions and the following disclaimer.
20     *
21     * 3. Redistributions in binary form must reproduce the above copyright
22     * notice, this list of conditions and the following disclaimer in the
23     * documentation and/or other materials provided with the
24     * distribution.
25     *
26     * This software is provided "AS IS," without a warranty of any
27     * kind. All express or implied conditions, representations and
28     * warranties, including any implied warranty of merchantability,
29     * fitness for a particular purpose or non-infringement, are hereby
30     * excluded. The University of Notre Dame and its licensors shall not
31     * be liable for any damages suffered by licensee as a result of
32     * using, modifying or distributing the software or its
33     * derivatives. In no event will the University of Notre Dame or its
34     * licensors be liable for any lost revenue, profit or data, or for
35     * direct, indirect, special, consequential, incidental or punitive
36     * damages, however caused and regardless of the theory of liability,
37     * arising out of the use of or inability to use software, even if the
38     * University of Notre Dame has been advised of the possibility of
39     * such damages.
40     */
41 tim 2544 #include <algorithm>
42 tim 2485 #include <functional>
43 gezelter 1490 #include <iostream>
44 tim 2485 #include <sstream>
45 tim 1492 #include "types/MoleculeStamp.hpp"
46 tim 2483 #include "utils/Tuple.hpp"
47 tim 2515 #include "utils/MemoryUtils.hpp"
48 tim 2469 namespace oopse {
49 tim 2485
50     template<class ContainerType>
51 tim 2544 bool hasDuplicateElement(const ContainerType& cont) {
52     ContainerType tmp = cont;
53     std::sort(tmp.begin(), tmp.end());
54     tmp.erase(std::unique(tmp.begin(), tmp.end()), tmp.end());
55     return tmp.size() != cont.size();
56 tim 2485 }
57    
58 tim 2469 MoleculeStamp::MoleculeStamp() {
59     DefineParameter(Name, "name");
60    
61     deprecatedKeywords_.insert("nAtoms");
62     deprecatedKeywords_.insert("nBonds");
63     deprecatedKeywords_.insert("nBends");
64     deprecatedKeywords_.insert("nTorsions");
65     deprecatedKeywords_.insert("nRigidBodies");
66     deprecatedKeywords_.insert("nCutoffGroups");
67    
68     }
69 gezelter 1490
70 tim 2469 MoleculeStamp::~MoleculeStamp() {
71 tim 2515 MemoryUtils::deletePointers(atomStamps_);
72     MemoryUtils::deletePointers(bondStamps_);
73     MemoryUtils::deletePointers(bendStamps_);
74     MemoryUtils::deletePointers(torsionStamps_);
75     MemoryUtils::deletePointers(rigidBodyStamps_);
76     MemoryUtils::deletePointers(cutoffGroupStamps_);
77     MemoryUtils::deletePointers(fragmentStamps_);
78 tim 2469 }
79 gezelter 1490
80 tim 2469 bool MoleculeStamp::addAtomStamp( AtomStamp* atom) {
81     bool ret = addIndexSensitiveStamp(atomStamps_, atom);
82     if (!ret) {
83 tim 2544 std::ostringstream oss;
84     oss<< "Error in Molecule " << getName() << ": multiple atoms have the same indices"<< atom->getIndex() <<"\n";
85     throw OOPSEException(oss.str());
86 tim 2469 }
87     return ret;
88    
89     }
90 gezelter 1490
91 tim 2469 bool MoleculeStamp::addBondStamp( BondStamp* bond) {
92     bondStamps_.push_back(bond);
93     return true;
94 gezelter 1490 }
95    
96 tim 2469 bool MoleculeStamp::addBendStamp( BendStamp* bend) {
97     bendStamps_.push_back(bend);
98     return true;
99     }
100 gezelter 1490
101 tim 2469 bool MoleculeStamp::addTorsionStamp( TorsionStamp* torsion) {
102     torsionStamps_.push_back(torsion);
103     return true;
104     }
105 gezelter 1490
106 tim 2469 bool MoleculeStamp::addRigidBodyStamp( RigidBodyStamp* rigidbody) {
107     bool ret = addIndexSensitiveStamp(rigidBodyStamps_, rigidbody);
108     if (!ret) {
109 tim 2544 std::ostringstream oss;
110     oss<< "Error in Molecule " << getName() << ": multiple rigidbodies have the same indices: " << rigidbody->getIndex() <<"\n";
111     throw OOPSEException(oss.str());
112 tim 2469 }
113     return ret;
114     }
115 gezelter 1490
116 tim 2469 bool MoleculeStamp::addCutoffGroupStamp( CutoffGroupStamp* cutoffgroup) {
117     cutoffGroupStamps_.push_back(cutoffgroup);
118     return true;
119 gezelter 1490 }
120    
121 tim 2469 bool MoleculeStamp::addFragmentStamp( FragmentStamp* fragment) {
122     return addIndexSensitiveStamp(fragmentStamps_, fragment);
123 gezelter 1490 }
124    
125 tim 2469 void MoleculeStamp::validate() {
126     DataHolder::validate();
127 gezelter 1490
128 tim 2483 atom2Rigidbody.resize(getNAtoms());
129     // negative number means atom is a free atom, does not belong to rigidbody
130     //every element in atom2Rigidbody has unique negative number at the very beginning
131     for(int i = 0; i < atom2Rigidbody.size(); ++i) {
132     atom2Rigidbody[i] = -1 - i;
133     }
134     for (int i = 0; i < getNRigidBodies(); ++i) {
135     RigidBodyStamp* rbStamp = getRigidBodyStamp(i);
136     std::vector<int> members = rbStamp->getMembers();
137     for(std::vector<int>::iterator j = members.begin(); j != members.end(); ++j) {
138     atom2Rigidbody[*j] = i;
139     }
140     }
141    
142     checkAtoms();
143     checkBonds();
144     fillBondInfo();
145     checkBends();
146     checkTorsions();
147     checkRigidBodies();
148     checkCutoffGroups();
149     checkFragments();
150    
151     int nrigidAtoms = 0;
152     for (int i = 0; i < getNRigidBodies(); ++i) {
153     RigidBodyStamp* rbStamp = getRigidBodyStamp(i);
154     nrigidAtoms += rbStamp->getNMembers();
155     }
156     nintegrable_ = getNAtoms()+ getNRigidBodies() - nrigidAtoms;
157    
158     }
159    
160     void MoleculeStamp::checkAtoms() {
161 tim 2469 std::vector<AtomStamp*>::iterator ai = std::find(atomStamps_.begin(), atomStamps_.end(), static_cast<AtomStamp*>(NULL));
162     if (ai != atomStamps_.end()) {
163 tim 2544 std::ostringstream oss;
164     oss << "Error in Molecule " << getName() << ": atom[" << ai - atomStamps_.begin()<< "] is missing\n";
165     throw OOPSEException(oss.str());
166 gezelter 1490 }
167    
168 tim 2485 }
169    
170     void MoleculeStamp::checkBonds() {
171 tim 2544 std::ostringstream oss;
172 tim 2469 //make sure index is not out of range
173     int natoms = getNAtoms();
174     for(int i = 0; i < getNBonds(); ++i) {
175     BondStamp* bondStamp = getBondStamp(i);
176 tim 2544 if (bondStamp->getA() > natoms-1 || bondStamp->getA() < 0 || bondStamp->getB() > natoms-1 || bondStamp->getB() < 0 || bondStamp->getA() == bondStamp->getB()) {
177    
178     oss << "Error in Molecule " << getName() << ": bond(" << bondStamp->getA() << ", " << bondStamp->getB() << ") is invalid\n";
179     throw OOPSEException(oss.str());
180 tim 2469 }
181 gezelter 1490 }
182 tim 2485
183 tim 2483 //make sure bonds are unique
184     std::set<std::pair<int, int> > allBonds;
185     for(int i = 0; i < getNBonds(); ++i) {
186     BondStamp* bondStamp= getBondStamp(i);
187     std::pair<int, int> bondPair(bondStamp->getA(), bondStamp->getB());
188     //make sure bondTuple.first is always less than or equal to bondTuple.third
189     if (bondPair.first > bondPair.second) {
190     std::swap(bondPair.first, bondPair.second);
191     }
192    
193     std::set<std::pair<int, int> >::iterator iter = allBonds.find(bondPair);
194     if ( iter != allBonds.end()) {
195 tim 2544
196     oss << "Error in Molecule " << getName() << ": " << "bond(" <<iter->first << ", "<< iter->second << ") appears multiple times\n";
197     throw OOPSEException(oss.str());
198 tim 2483 } else {
199     allBonds.insert(bondPair);
200     }
201     }
202    
203     //make sure atoms belong to same rigidbody do not bond to each other
204     for(int i = 0; i < getNBonds(); ++i) {
205     BondStamp* bondStamp = getBondStamp(i);
206     if (atom2Rigidbody[bondStamp->getA()] == atom2Rigidbody[bondStamp->getB()]) {
207 tim 2544
208     oss << "Error in Molecule " << getName() << ": "<<"bond(" << bondStamp->getA() << ", " << bondStamp->getB() << ") belong to same rigidbody " << atom2Rigidbody[bondStamp->getA()] << "\n";
209     throw OOPSEException(oss.str());
210 tim 2483 }
211     }
212    
213     }
214    
215     void MoleculeStamp::checkBends() {
216 tim 2544 std::ostringstream oss;
217 tim 2469 for(int i = 0; i < getNBends(); ++i) {
218     BendStamp* bendStamp = getBendStamp(i);
219     std::vector<int> bendAtoms = bendStamp->getMembers();
220 tim 2483 std::vector<int>::iterator j =std::find_if(bendAtoms.begin(), bendAtoms.end(), std::bind2nd(std::greater<int>(), getNAtoms()-1));
221 tim 2544 std::vector<int>::iterator k =std::find_if(bendAtoms.begin(), bendAtoms.end(), std::bind2nd(std::less<int>(), 0));
222    
223     if (j != bendAtoms.end() || k != bendAtoms.end()) {
224    
225     oss << "Error in Molecule " << getName() << " : atoms of bend" << containerToString(bendAtoms) << " have invalid indices\n";
226     throw OOPSEException(oss.str());
227 tim 2469 }
228 tim 2544
229     if (hasDuplicateElement(bendAtoms)) {
230     oss << "Error in Molecule " << getName() << " : atoms of bend" << containerToString(bendAtoms) << " have duplicated indices\n";
231     throw OOPSEException(oss.str());
232     }
233    
234 tim 2483 if (bendAtoms.size() == 2 ) {
235     if (!bendStamp->haveGhostVectorSource()) {
236 tim 2544
237     oss << "Error in Molecule " << getName() << ": ghostVectorSouce is missing\n";
238     throw OOPSEException(oss.str());
239 tim 2483 }else{
240     int ghostIndex = bendStamp->getGhostVectorSource();
241     if (ghostIndex < getNAtoms()) {
242     if (std::find(bendAtoms.begin(), bendAtoms.end(), ghostIndex) == bendAtoms.end()) {
243 tim 2544
244     oss << "Error in Molecule " << getName() << ": ghostVectorSouce "<< ghostIndex<<"is invalid\n";
245     throw OOPSEException(oss.str());
246 tim 2483 }
247     if (!getAtomStamp(ghostIndex)->haveOrientation()) {
248 tim 2544
249     oss << "Error in Molecule " << getName() << ": ghost atom must be a directioanl atom\n";
250     throw OOPSEException(oss.str());
251 tim 2483 }
252     }else {
253 tim 2544 oss << "Error in Molecule " << getName() << ": ghostVectorsource " << ghostIndex<< " is invalid\n";
254     throw OOPSEException(oss.str());
255 tim 2483 }
256     }
257     } else if (bendAtoms.size() == 3 && bendStamp->haveGhostVectorSource()) {
258 tim 2544 oss << "Error in Molecule " << getName() << ": normal bend should not have ghostVectorSouce\n";
259     throw OOPSEException(oss.str());
260 tim 2469 }
261 gezelter 1490 }
262    
263 tim 2469 for(int i = 0; i < getNBends(); ++i) {
264     BendStamp* bendStamp = getBendStamp(i);
265     std::vector<int> bendAtoms = bendStamp->getMembers();
266     std::vector<int> rigidSet(getNRigidBodies(), 0);
267     std::vector<int>::iterator j;
268     for( j = bendAtoms.begin(); j != bendAtoms.end(); ++j) {
269     int rigidbodyIndex = atom2Rigidbody[*j];
270     if (rigidbodyIndex >= 0) {
271     ++rigidSet[rigidbodyIndex];
272     if (rigidSet[rigidbodyIndex] > 1) {
273 tim 2544 oss << "Error in Molecule " << getName() << ": bend" << containerToString(bendAtoms) << " belong to same rigidbody " << rigidbodyIndex << "\n";
274     throw OOPSEException(oss.str());
275 tim 2469 }
276     }
277     }
278 tim 2483 }
279    
280    
281 tim 2507 std::set<IntTuple3> allBends;
282     std::set<IntTuple3>::iterator iter;
283 tim 2483 for(int i = 0; i < getNBends(); ++i) {
284     BendStamp* bendStamp= getBendStamp(i);
285     std::vector<int> bend = bendStamp->getMembers();
286     if (bend.size() == 2) {
287     // in case we have two ghost bend. For example,
288     // bend {
289     // members (0, 1);
290     // ghostVectorSource = 0;
291     // }
292     // and
293     // bend {
294     // members (0, 1);
295     // ghostVectorSource = 0;
296     // }
297     // In order to distinguish them. we expand them to Tuple3.
298     // the first one is expanded to (0, 0, 1) while the second one is expaned to (0, 1, 1)
299     int ghostIndex = bendStamp->getGhostVectorSource();
300     std::vector<int>::iterator j = std::find(bend.begin(), bend.end(), ghostIndex);
301     if (j != bend.end()) {
302     bend.insert(j, ghostIndex);
303     }
304     }
305    
306     IntTuple3 bendTuple(bend[0], bend[1], bend[2]);
307     //make sure bendTuple.first is always less than or equal to bendTuple.third
308     if (bendTuple.first > bendTuple.third) {
309     std::swap(bendTuple.first, bendTuple.third);
310     }
311    
312     iter = allBends.find(bendTuple);
313     if ( iter != allBends.end()) {
314 tim 2544 oss << "Error in Molecule " << getName() << ": " << "Bend" << containerToString(bend)<< " appears multiple times\n";
315     throw OOPSEException(oss.str());
316 tim 2483 } else {
317     allBends.insert(bendTuple);
318     }
319     }
320    
321     for (int i = 0; i < getNBonds(); ++i) {
322     BondStamp* bondStamp = getBondStamp(i);
323     int a = bondStamp->getA();
324     int b = bondStamp->getB();
325    
326     AtomStamp* atomA = getAtomStamp(a);
327     AtomStamp* atomB = getAtomStamp(b);
328    
329     //find bend c--a--b
330     AtomStamp::AtomIter ai;
331 gezelter 2543 for(int c= atomA->getFirstBondedAtom(ai);c != -1;c = atomA->getNextBondedAtom(ai))
332 tim 2483 {
333     if(b == c)
334     continue;
335    
336     IntTuple3 newBend(c, a, b);
337     if (newBend.first > newBend.third) {
338     std::swap(newBend.first, newBend.third);
339     }
340    
341     if (allBends.find(newBend) == allBends.end() ) {
342     allBends.insert(newBend);
343     BendStamp * newBendStamp = new BendStamp();
344     newBendStamp->setMembers(newBend);
345     addBendStamp(newBendStamp);
346     }
347     }
348    
349     //find bend a--b--c
350 gezelter 2543 for(int c= atomB->getFirstBondedAtom(ai);c != -1;c = atomB->getNextBondedAtom(ai))
351 tim 2483 {
352     if(a == c)
353     continue;
354    
355     IntTuple3 newBend( a, b, c);
356     if (newBend.first > newBend.third) {
357     std::swap(newBend.first, newBend.third);
358     }
359     if (allBends.find(newBend) == allBends.end() ) {
360     allBends.insert(newBend);
361     BendStamp * newBendStamp = new BendStamp();
362     newBendStamp->setMembers(newBend);
363     addBendStamp(newBendStamp);
364     }
365     }
366     }
367    
368     }
369    
370     void MoleculeStamp::checkTorsions() {
371 tim 2544 std::ostringstream oss;
372 tim 2493 for(int i = 0; i < getNTorsions(); ++i) {
373 tim 2485 TorsionStamp* torsionStamp = getTorsionStamp(i);
374     std::vector<int> torsionAtoms = torsionStamp ->getMembers();
375     std::vector<int>::iterator j =std::find_if(torsionAtoms.begin(), torsionAtoms.end(), std::bind2nd(std::greater<int>(), getNAtoms()-1));
376 tim 2544 std::vector<int>::iterator k =std::find_if(torsionAtoms.begin(), torsionAtoms.end(), std::bind2nd(std::less<int>(), 0));
377    
378     if (j != torsionAtoms.end() || k != torsionAtoms.end()) {
379     oss << "Error in Molecule " << getName() << ": atoms of torsion" << containerToString(torsionAtoms) << " have invalid indices\n";
380     throw OOPSEException(oss.str());
381 tim 2485 }
382 tim 2544 if (hasDuplicateElement(torsionAtoms)) {
383     oss << "Error in Molecule " << getName() << " : atoms of torsion" << containerToString(torsionAtoms) << " have duplicated indices\n";
384     throw OOPSEException(oss.str());
385     }
386 tim 2485 }
387    
388 tim 2469 for(int i = 0; i < getNTorsions(); ++i) {
389     TorsionStamp* torsionStamp = getTorsionStamp(i);
390     std::vector<int> torsionAtoms = torsionStamp->getMembers();
391     std::vector<int> rigidSet(getNRigidBodies(), 0);
392     std::vector<int>::iterator j;
393     for( j = torsionAtoms.begin(); j != torsionAtoms.end(); ++j) {
394     int rigidbodyIndex = atom2Rigidbody[*j];
395     if (rigidbodyIndex >= 0) {
396     ++rigidSet[rigidbodyIndex];
397     if (rigidSet[rigidbodyIndex] > 1) {
398 tim 2544 oss << "Error in Molecule " << getName() << ": torsion" << containerToString(torsionAtoms) << "is invalid\n";
399     throw OOPSEException(oss.str());
400 tim 2469 }
401     }
402     }
403 tim 2483 }
404 gezelter 1490
405 tim 2507 std::set<IntTuple4> allTorsions;
406     std::set<IntTuple4>::iterator iter;
407 tim 2483 for(int i = 0; i < getNTorsions(); ++i) {
408     TorsionStamp* torsionStamp= getTorsionStamp(i);
409     std::vector<int> torsion = torsionStamp->getMembers();
410     if (torsion.size() == 3) {
411     int ghostIndex = torsionStamp->getGhostVectorSource();
412     std::vector<int>::iterator j = std::find(torsion.begin(), torsion.end(), ghostIndex);
413     if (j != torsion.end()) {
414     torsion.insert(j, ghostIndex);
415     }
416     }
417 gezelter 1490
418 tim 2483 IntTuple4 torsionTuple(torsion[0], torsion[1], torsion[2], torsion[3]);
419     if (torsionTuple.first > torsionTuple.fourth) {
420     std::swap(torsionTuple.first, torsionTuple.fourth);
421     std::swap(torsionTuple.second, torsionTuple.third);
422     }
423 gezelter 1490
424 tim 2483 iter = allTorsions.find(torsionTuple);
425     if ( iter == allTorsions.end()) {
426     allTorsions.insert(torsionTuple);
427     } else {
428 tim 2544 oss << "Error in Molecule " << getName() << ": " << "Torsion" << containerToString(torsion)<< " appears multiple times\n";
429     throw OOPSEException(oss.str());
430 tim 2483 }
431     }
432    
433     for (int i = 0; i < getNBonds(); ++i) {
434     BondStamp* bondStamp = getBondStamp(i);
435     int b = bondStamp->getA();
436     int c = bondStamp->getB();
437    
438     AtomStamp* atomB = getAtomStamp(b);
439     AtomStamp* atomC = getAtomStamp(c);
440    
441     AtomStamp::AtomIter ai2;
442     AtomStamp::AtomIter ai3;
443    
444 gezelter 2543 for(int a = atomB->getFirstBondedAtom(ai2);a != -1;a = atomB->getNextBondedAtom(ai2))
445 tim 2483 {
446     if(a == c)
447     continue;
448    
449 gezelter 2543 for(int d = atomC->getFirstBondedAtom(ai3);d != -1;d = atomC->getNextBondedAtom(ai3))
450 tim 2483 {
451     if(d == b)
452     continue;
453    
454     IntTuple4 newTorsion(a, b, c, d);
455     //make sure the first element is always less than or equal to the fourth element in IntTuple4
456     if (newTorsion.first > newTorsion.fourth) {
457     std::swap(newTorsion.first, newTorsion.fourth);
458     std::swap(newTorsion.second, newTorsion.third);
459     }
460     if (allTorsions.find(newTorsion) == allTorsions.end() ) {
461     allTorsions.insert(newTorsion);
462     TorsionStamp * newTorsionStamp = new TorsionStamp();
463     newTorsionStamp->setMembers(newTorsion);
464     addTorsionStamp(newTorsionStamp);
465     }
466     }
467     }
468     }
469    
470     }
471    
472     void MoleculeStamp::checkRigidBodies() {
473 tim 2544 std::ostringstream oss;
474 tim 2483 std::vector<RigidBodyStamp*>::iterator ri = std::find(rigidBodyStamps_.begin(), rigidBodyStamps_.end(), static_cast<RigidBodyStamp*>(NULL));
475     if (ri != rigidBodyStamps_.end()) {
476 tim 2544 oss << "Error in Molecule " << getName() << ":rigidBody[" << ri - rigidBodyStamps_.begin()<< "] is missing\n";
477     throw OOPSEException(oss.str());
478 tim 2483 }
479    
480 tim 2469 for (int i = 0; i < getNRigidBodies(); ++i) {
481     RigidBodyStamp* rbStamp = getRigidBodyStamp(i);
482 tim 2483 std::vector<int> rigidAtoms = rbStamp ->getMembers();
483     std::vector<int>::iterator j =std::find_if(rigidAtoms.begin(), rigidAtoms.end(), std::bind2nd(std::greater<int>(), getNAtoms()-1));
484     if (j != rigidAtoms.end()) {
485 tim 2544 oss << "Error in Molecule " << getName();
486     throw OOPSEException(oss.str());
487 tim 2483 }
488    
489     }
490     }
491 gezelter 1490
492 tim 2483 void MoleculeStamp::checkCutoffGroups() {
493     for(int i = 0; i < getNCutoffGroups(); ++i) {
494     CutoffGroupStamp* cutoffGroupStamp = getCutoffGroupStamp(i);
495     std::vector<int> cutoffGroupAtoms = cutoffGroupStamp ->getMembers();
496     std::vector<int>::iterator j =std::find_if(cutoffGroupAtoms.begin(), cutoffGroupAtoms.end(), std::bind2nd(std::greater<int>(), getNAtoms()-1));
497     if (j != cutoffGroupAtoms.end()) {
498 tim 2544 std::ostringstream oss;
499     oss << "Error in Molecule " << getName() << ": cutoffGroup" << " is out of range\n";
500     throw OOPSEException(oss.str());
501 tim 2483 }
502     }
503 gezelter 1490 }
504    
505 tim 2483 void MoleculeStamp::checkFragments() {
506 gezelter 1490
507 tim 2483 std::vector<FragmentStamp*>::iterator fi = std::find(fragmentStamps_.begin(), fragmentStamps_.end(), static_cast<FragmentStamp*>(NULL));
508     if (fi != fragmentStamps_.end()) {
509 tim 2544 std::ostringstream oss;
510     oss << "Error in Molecule " << getName() << ":fragment[" << fi - fragmentStamps_.begin()<< "] is missing\n";
511     throw OOPSEException(oss.str());
512 tim 2483 }
513    
514 gezelter 1490 }
515    
516 tim 2483 void MoleculeStamp::fillBondInfo() {
517 gezelter 1490
518 tim 2483 for (int i = 0; i < getNBonds(); ++i) {
519     BondStamp* bondStamp = getBondStamp(i);
520     int a = bondStamp->getA();
521     int b = bondStamp->getB();
522     AtomStamp* atomA = getAtomStamp(a);
523     AtomStamp* atomB = getAtomStamp(b);
524     atomA->addBond(i);
525     atomA->addBondedAtom(b);
526     atomB->addBond(i);
527     atomB->addBondedAtom(a);
528    
529     }
530 gezelter 1490 }
531    
532 tim 2483
533    
534 gezelter 1490 //Function Name: isBondInSameRigidBody
535     //Return true is both atoms of the bond belong to the same rigid body, otherwise return false
536     bool MoleculeStamp::isBondInSameRigidBody(BondStamp* bond){
537     int rbA;
538     int rbB;
539     int consAtomA;
540     int consAtomB;
541    
542     if (!isAtomInRigidBody(bond->getA(),rbA, consAtomA))
543     return false;
544    
545     if(!isAtomInRigidBody(bond->getB(),rbB, consAtomB) )
546     return false;
547    
548     if(rbB == rbA)
549     return true;
550     else
551     return false;
552     }
553    
554     // Function Name: isAtomInRigidBody
555     //return false if atom does not belong to a rigid body, otherwise return true
556     bool MoleculeStamp::isAtomInRigidBody(int atomIndex){
557 tim 2485 return atom2Rigidbody[atomIndex] >=0 ;
558 gezelter 1490
559     }
560    
561     // Function Name: isAtomInRigidBody
562     //return false if atom does not belong to a rigid body otherwise return true and set whichRigidBody
563     //and consAtomIndex
564     //atomIndex : the index of atom in component
565     //whichRigidBody: the index of rigidbody in component
566     //consAtomIndex: the position of joint atom apears in rigidbody's definition
567     bool MoleculeStamp::isAtomInRigidBody(int atomIndex, int& whichRigidBody, int& consAtomIndex){
568    
569 tim 2485
570    
571 gezelter 1490 whichRigidBody = -1;
572     consAtomIndex = -1;
573    
574 tim 2485 if (atom2Rigidbody[atomIndex] >=0) {
575     whichRigidBody = atom2Rigidbody[atomIndex];
576     RigidBodyStamp* rbStamp = getRigidBodyStamp(whichRigidBody);
577     int numAtom = rbStamp->getNMembers();
578     for(int j = 0; j < numAtom; j++) {
579 tim 2469 if (rbStamp->getMemberAt(j) == atomIndex){
580 gezelter 1490 consAtomIndex = j;
581     return true;
582     }
583 tim 2485 }
584 gezelter 1490 }
585    
586     return false;
587    
588     }
589    
590     //return the position of joint atom apears in rigidbody's definition
591     //for the time being, we will use the most inefficient algorithm, the complexity is O(N2)
592     //actually we could improve the complexity to O(NlgN) by sorting the atom index in rigid body first
593 gezelter 2204 std::vector<std::pair<int, int> > MoleculeStamp::getJointAtoms(int rb1, int rb2){
594 gezelter 1490 RigidBodyStamp* rbStamp1;
595     RigidBodyStamp* rbStamp2;
596     int natomInRb1;
597     int natomInRb2;
598     int atomIndex1;
599     int atomIndex2;
600 gezelter 2204 std::vector<std::pair<int, int> > jointAtomIndexPair;
601 gezelter 1490
602 tim 2469 rbStamp1 = this->getRigidBodyStamp(rb1);
603 gezelter 1490 natomInRb1 =rbStamp1->getNMembers();
604    
605 tim 2469 rbStamp2 = this->getRigidBodyStamp(rb2);
606 gezelter 1490 natomInRb2 =rbStamp2->getNMembers();
607    
608     for(int i = 0; i < natomInRb1; i++){
609 tim 2469 atomIndex1 = rbStamp1->getMemberAt(i);
610 gezelter 1490
611     for(int j= 0; j < natomInRb1; j++){
612 tim 2469 atomIndex2 = rbStamp2->getMemberAt(j);
613 gezelter 1490
614     if(atomIndex1 == atomIndex2){
615 gezelter 1930 jointAtomIndexPair.push_back(std::make_pair(i, j));
616 gezelter 1490 break;
617     }
618    
619 tim 2469 }
620 gezelter 1490
621 tim 2469 }
622 gezelter 1490
623     return jointAtomIndexPair;
624     }
625 tim 2469
626     }