| 787 |
|
int beginRigidBodyIndex; |
| 788 |
|
int beginCutoffGroupIndex; |
| 789 |
|
int nGlobalAtoms = info->getNGlobalAtoms(); |
| 790 |
+ |
int nGlobalRigidBodies = info->getNGlobalRigidBodies(); |
| 791 |
|
|
| 792 |
|
beginAtomIndex = 0; |
| 793 |
|
//rigidbody's index begins right after atom's |
| 863 |
|
#endif |
| 864 |
|
|
| 865 |
|
//fill molMembership |
| 866 |
< |
std::vector<int> globalMolMembership(info->getNGlobalAtoms(), 0); |
| 866 |
> |
std::vector<int> globalMolMembership(info->getNGlobalAtoms() + |
| 867 |
> |
info->getNGlobalRigidBodies(), 0); |
| 868 |
|
|
| 869 |
< |
for(mol = info->beginMolecule(mi); mol != NULL; mol = info->nextMolecule(mi)) { |
| 869 |
> |
for(mol = info->beginMolecule(mi); mol != NULL; |
| 870 |
> |
mol = info->nextMolecule(mi)) { |
| 871 |
|
for(atom = mol->beginAtom(ai); atom != NULL; atom = mol->nextAtom(ai)) { |
| 872 |
|
globalMolMembership[atom->getGlobalIndex()] = mol->getGlobalIndex(); |
| 873 |
|
} |
| 874 |
+ |
for (rb = mol->beginRigidBody(ri); rb != NULL; |
| 875 |
+ |
rb = mol->nextRigidBody(ri)) { |
| 876 |
+ |
globalMolMembership[rb->getGlobalIndex()] = mol->getGlobalIndex(); |
| 877 |
+ |
} |
| 878 |
|
} |
| 879 |
|
|
| 880 |
|
#ifdef IS_MPI |
| 881 |
< |
std::vector<int> tmpMolMembership(info->getNGlobalAtoms(), 0); |
| 881 |
> |
std::vector<int> tmpMolMembership(info->getNGlobalAtoms() + |
| 882 |
> |
info->getNGlobalRigidBodies(), 0); |
| 883 |
|
MPI::COMM_WORLD.Allreduce(&globalMolMembership[0], &tmpMolMembership[0], |
| 884 |
< |
nGlobalAtoms, |
| 884 |
> |
nGlobalAtoms + nGlobalRigidBodies, |
| 885 |
|
MPI::INT, MPI::SUM); |
| 886 |
|
|
| 887 |
|
info->setGlobalMolMembership(tmpMolMembership); |
| 893 |
|
// here the molecules are listed by their global indices. |
| 894 |
|
|
| 895 |
|
std::vector<int> nIOPerMol(info->getNGlobalMolecules(), 0); |
| 896 |
< |
for (mol = info->beginMolecule(mi); mol != NULL; mol = info->nextMolecule(mi)) { |
| 896 |
> |
for (mol = info->beginMolecule(mi); mol != NULL; |
| 897 |
> |
mol = info->nextMolecule(mi)) { |
| 898 |
|
nIOPerMol[mol->getGlobalIndex()] = mol->getNIntegrableObjects(); |
| 899 |
|
} |
| 900 |
|
|
| 915 |
|
} |
| 916 |
|
|
| 917 |
|
std::vector<StuntDouble*> IOIndexToIntegrableObject(info->getNGlobalIntegrableObjects(), (StuntDouble*)NULL); |
| 918 |
< |
for (mol = info->beginMolecule(mi); mol != NULL; mol = info->nextMolecule(mi)) { |
| 918 |
> |
for (mol = info->beginMolecule(mi); mol != NULL; |
| 919 |
> |
mol = info->nextMolecule(mi)) { |
| 920 |
|
int myGlobalIndex = mol->getGlobalIndex(); |
| 921 |
|
int globalIO = startingIOIndexForMol[myGlobalIndex]; |
| 922 |
|
for (StuntDouble* sd = mol->beginIntegrableObject(ioi); sd != NULL; |