234 |
|
|
235 |
|
SimInfo* SimCreator::createSim(const std::string & mdFileName, |
236 |
|
bool loadInitCoords) { |
237 |
< |
|
237 |
> |
|
238 |
|
const int bufferSize = 65535; |
239 |
|
char buffer[bufferSize]; |
240 |
|
int lineNo = 0; |
369 |
|
} |
370 |
|
|
371 |
|
ff->parse(forcefieldFileName); |
372 |
– |
ff->setFortranForceOptions(); |
372 |
|
//create SimInfo |
373 |
|
SimInfo * info = new SimInfo(ff, simParams); |
374 |
|
|
386 |
|
//create the molecules |
387 |
|
createMolecules(info); |
388 |
|
|
390 |
– |
|
389 |
|
//allocate memory for DataStorage(circular reference, need to |
390 |
|
//break it) |
391 |
|
info->setSnapshotManager(new SimSnapshotManager(info)); |
411 |
|
|
412 |
|
if (loadInitCoords) |
413 |
|
loadCoordinates(info, mdFileName); |
416 |
– |
|
414 |
|
return info; |
415 |
|
} |
416 |
|
|
589 |
|
MPI_Bcast(&molToProcMap[0], nGlobalMols, MPI_INT, 0, MPI_COMM_WORLD); |
590 |
|
} |
591 |
|
|
592 |
+ |
cerr << "molToProcMap:\n"; |
593 |
+ |
for (int i = 0; i < molToProcMap.size(); i++) { |
594 |
+ |
cerr << "m = " << i << " mtpr[m] = " << molToProcMap[i] <<"\n"; |
595 |
+ |
} |
596 |
+ |
|
597 |
|
info->setMolToProcMap(molToProcMap); |
598 |
|
sprintf(checkPointMsg, |
599 |
|
"Successfully divided the molecules among the processors.\n"); |
614 |
|
#endif |
615 |
|
|
616 |
|
stampId = info->getMoleculeStampId(i); |
617 |
< |
Molecule * mol = molCreator.createMolecule(info->getForceField(), info->getMoleculeStamp(stampId), |
618 |
< |
stampId, i, info->getLocalIndexManager()); |
617 |
> |
Molecule * mol = molCreator.createMolecule(info->getForceField(), |
618 |
> |
info->getMoleculeStamp(stampId), |
619 |
> |
stampId, i, |
620 |
> |
info->getLocalIndexManager()); |
621 |
|
|
622 |
|
info->addMolecule(mol); |
623 |
|
|
644 |
|
int beginRigidBodyIndex; |
645 |
|
int beginCutoffGroupIndex; |
646 |
|
int nGlobalAtoms = info->getNGlobalAtoms(); |
643 |
– |
|
644 |
– |
/**@todo fixme */ |
645 |
– |
#ifndef IS_MPI |
646 |
– |
|
647 |
– |
beginAtomIndex = 0; |
648 |
– |
beginRigidBodyIndex = 0; |
649 |
– |
beginCutoffGroupIndex = 0; |
650 |
– |
|
651 |
– |
#else |
652 |
– |
|
653 |
– |
int nproc; |
654 |
– |
int myNode; |
655 |
– |
|
656 |
– |
myNode = worldRank; |
657 |
– |
MPI_Comm_size(MPI_COMM_WORLD, &nproc); |
658 |
– |
|
659 |
– |
std::vector < int > tmpAtomsInProc(nproc, 0); |
660 |
– |
std::vector < int > tmpRigidBodiesInProc(nproc, 0); |
661 |
– |
std::vector < int > tmpCutoffGroupsInProc(nproc, 0); |
662 |
– |
std::vector < int > NumAtomsInProc(nproc, 0); |
663 |
– |
std::vector < int > NumRigidBodiesInProc(nproc, 0); |
664 |
– |
std::vector < int > NumCutoffGroupsInProc(nproc, 0); |
647 |
|
|
666 |
– |
tmpAtomsInProc[myNode] = info->getNAtoms(); |
667 |
– |
tmpRigidBodiesInProc[myNode] = info->getNRigidBodies(); |
668 |
– |
tmpCutoffGroupsInProc[myNode] = info->getNCutoffGroups(); |
669 |
– |
|
670 |
– |
//do MPI_ALLREDUCE to exchange the total number of atoms, rigidbodies and cutoff groups |
671 |
– |
MPI_Allreduce(&tmpAtomsInProc[0], &NumAtomsInProc[0], nproc, MPI_INT, |
672 |
– |
MPI_SUM, MPI_COMM_WORLD); |
673 |
– |
MPI_Allreduce(&tmpRigidBodiesInProc[0], &NumRigidBodiesInProc[0], nproc, |
674 |
– |
MPI_INT, MPI_SUM, MPI_COMM_WORLD); |
675 |
– |
MPI_Allreduce(&tmpCutoffGroupsInProc[0], &NumCutoffGroupsInProc[0], nproc, |
676 |
– |
MPI_INT, MPI_SUM, MPI_COMM_WORLD); |
677 |
– |
|
648 |
|
beginAtomIndex = 0; |
649 |
|
beginRigidBodyIndex = 0; |
650 |
|
beginCutoffGroupIndex = 0; |
651 |
< |
|
652 |
< |
for(int i = 0; i < myNode; i++) { |
683 |
< |
beginAtomIndex += NumAtomsInProc[i]; |
684 |
< |
beginRigidBodyIndex += NumRigidBodiesInProc[i]; |
685 |
< |
beginCutoffGroupIndex += NumCutoffGroupsInProc[i]; |
686 |
< |
} |
687 |
< |
|
688 |
< |
#endif |
689 |
< |
|
690 |
< |
//rigidbody's index begins right after atom's |
691 |
< |
beginRigidBodyIndex += info->getNGlobalAtoms(); |
692 |
< |
|
693 |
< |
for(mol = info->beginMolecule(mi); mol != NULL; |
694 |
< |
mol = info->nextMolecule(mi)) { |
651 |
> |
|
652 |
> |
for(int i = 0; i < info->getNGlobalMolecules(); i++) { |
653 |
|
|
654 |
< |
//local index(index in DataStorge) of atom is important |
655 |
< |
for(atom = mol->beginAtom(ai); atom != NULL; atom = mol->nextAtom(ai)) { |
656 |
< |
atom->setGlobalIndex(beginAtomIndex++); |
654 |
> |
#ifdef IS_MPI |
655 |
> |
if (info->getMolToProc(i) == worldRank) { |
656 |
> |
#endif |
657 |
> |
// stuff to do if I own this molecule |
658 |
> |
mol = info->getMoleculeByGlobalIndex(i); |
659 |
> |
|
660 |
> |
//local index(index in DataStorge) of atom is important |
661 |
> |
for(atom = mol->beginAtom(ai); atom != NULL; atom = mol->nextAtom(ai)) { |
662 |
> |
atom->setGlobalIndex(beginAtomIndex++); |
663 |
> |
} |
664 |
> |
|
665 |
> |
for(rb = mol->beginRigidBody(ri); rb != NULL; |
666 |
> |
rb = mol->nextRigidBody(ri)) { |
667 |
> |
rb->setGlobalIndex(beginRigidBodyIndex++); |
668 |
> |
} |
669 |
> |
|
670 |
> |
//local index of cutoff group is trivial, it only depends on |
671 |
> |
//the order of travesing |
672 |
> |
for(cg = mol->beginCutoffGroup(ci); cg != NULL; |
673 |
> |
cg = mol->nextCutoffGroup(ci)) { |
674 |
> |
cg->setGlobalIndex(beginCutoffGroupIndex++); |
675 |
> |
} |
676 |
> |
|
677 |
> |
#ifdef IS_MPI |
678 |
> |
} else { |
679 |
> |
|
680 |
> |
// stuff to do if I don't own this molecule |
681 |
> |
|
682 |
> |
int stampId = info->getMoleculeStampId(i); |
683 |
> |
MoleculeStamp* stamp = info->getMoleculeStamp(stampId); |
684 |
> |
|
685 |
> |
beginAtomIndex += stamp->getNAtoms(); |
686 |
> |
beginRigidBodyIndex += stamp->getNRigidBodies(); |
687 |
> |
beginCutoffGroupIndex += stamp->getNCutoffGroups() + stamp->getNFreeAtoms(); |
688 |
|
} |
689 |
< |
|
690 |
< |
for(rb = mol->beginRigidBody(ri); rb != NULL; |
691 |
< |
rb = mol->nextRigidBody(ri)) { |
692 |
< |
rb->setGlobalIndex(beginRigidBodyIndex++); |
704 |
< |
} |
705 |
< |
|
706 |
< |
//local index of cutoff group is trivial, it only depends on the order of travesing |
707 |
< |
for(cg = mol->beginCutoffGroup(ci); cg != NULL; |
708 |
< |
cg = mol->nextCutoffGroup(ci)) { |
709 |
< |
cg->setGlobalIndex(beginCutoffGroupIndex++); |
710 |
< |
} |
711 |
< |
} |
712 |
< |
|
689 |
> |
#endif |
690 |
> |
|
691 |
> |
} //end for(int i=0) |
692 |
> |
|
693 |
|
//fill globalGroupMembership |
694 |
|
std::vector<int> globalGroupMembership(info->getNGlobalAtoms(), 0); |
695 |
|
for(mol = info->beginMolecule(mi); mol != NULL; mol = info->nextMolecule(mi)) { |
701 |
|
|
702 |
|
} |
703 |
|
} |
704 |
< |
|
704 |
> |
|
705 |
|
#ifdef IS_MPI |
706 |
|
// Since the globalGroupMembership has been zero filled and we've only |
707 |
|
// poked values into the atoms we know, we can do an Allreduce |
712 |
|
MPI_Allreduce(&globalGroupMembership[0], &tmpGroupMembership[0], nGlobalAtoms, |
713 |
|
MPI_INT, MPI_SUM, MPI_COMM_WORLD); |
714 |
|
info->setGlobalGroupMembership(tmpGroupMembership); |
715 |
+ |
|
716 |
+ |
cerr << "ggm:\n"; |
717 |
+ |
for (int i = 0; i < tmpGroupMembership.size(); i++) |
718 |
+ |
cerr << "i = " << i << "\t ggm(i) = " << tmpGroupMembership[i] << "\n"; |
719 |
+ |
|
720 |
|
#else |
721 |
|
info->setGlobalGroupMembership(globalGroupMembership); |
722 |
|
#endif |
776 |
|
globalIO++; |
777 |
|
} |
778 |
|
} |
779 |
< |
|
779 |
> |
cerr << "ioi2io:\n"; |
780 |
> |
for (int i = 0; i < IOIndexToIntegrableObject.size(); i++) { |
781 |
> |
if (IOIndexToIntegrableObject[i] != NULL) { |
782 |
> |
cerr << "i = " << i << "globalIOindex = " << IOIndexToIntegrableObject[i]->getGlobalIntegrableObjectIndex() << "\n"; |
783 |
> |
} |
784 |
> |
} |
785 |
> |
|
786 |
|
info->setIOIndexToIntegrableObject(IOIndexToIntegrableObject); |
787 |
|
|
788 |
|
} |
789 |
|
|
790 |
|
void SimCreator::loadCoordinates(SimInfo* info, const std::string& mdFileName) { |
791 |
|
Globals* simParams; |
792 |
+ |
|
793 |
|
simParams = info->getSimParams(); |
794 |
|
|
803 |
– |
|
795 |
|
DumpReader reader(info, mdFileName); |
796 |
|
int nframes = reader.getNFrames(); |
797 |
< |
|
797 |
> |
|
798 |
|
if (nframes > 0) { |
799 |
|
reader.readFrame(nframes - 1); |
800 |
|
} else { |
805 |
|
painCave.isFatal = 1; |
806 |
|
simError(); |
807 |
|
} |
817 |
– |
|
808 |
|
//copy the current snapshot to previous snapshot |
809 |
|
info->getSnapshotManager()->advance(); |
810 |
|
} |