ViewVC Help
View File | Revision Log | Show Annotations | View Changeset | Root Listing
root/OpenMD/branches/development/src/brains/SimCreator.cpp
(Generate patch)

Comparing branches/development/src/brains/SimCreator.cpp (file contents):
Revision 1597 by gezelter, Tue Jul 26 15:49:24 2011 UTC vs.
Revision 1613 by gezelter, Thu Aug 18 20:18:19 2011 UTC

# Line 81 | Line 81 | namespace OpenMD {
81  
82   namespace OpenMD {
83    
84 <  Globals* SimCreator::parseFile(std::istream& rawMetaDataStream, const std::string& filename, int startOfMetaDataBlock ){
84 >  Globals* SimCreator::parseFile(std::istream& rawMetaDataStream, const std::string& filename, int mdFileVersion, int startOfMetaDataBlock ){
85      Globals* simParams = NULL;
86      try {
87  
# Line 92 | Line 92 | namespace OpenMD {
92        const int masterNode = 0;
93        int commStatus;
94        if (worldRank == masterNode) {
95 < #endif
96 <                
95 >        commStatus = MPI_Bcast(&mdFileVersion, 1, MPI_INT, masterNode, MPI_COMM_WORLD);
96 > #endif                
97          SimplePreprocessor preprocessor;
98          preprocessor.preprocess(rawMetaDataStream, filename, startOfMetaDataBlock, ppStream);
99                  
# Line 106 | Line 106 | namespace OpenMD {
106              
107                  
108        } else {
109 +
110 +        commStatus = MPI_Bcast(&mdFileVersion, 1, MPI_INT, masterNode, MPI_COMM_WORLD);
111 +
112          //get stream size
113          commStatus = MPI_Bcast(&streamSize, 1, MPI_LONG, masterNode, MPI_COMM_WORLD);  
114  
# Line 229 | Line 232 | namespace OpenMD {
232        simError();
233      }
234  
235 +    simParams->setMDfileVersion(mdFileVersion);
236      return simParams;
237    }
238    
# Line 243 | Line 247 | namespace OpenMD {
247      int metaDataBlockEnd = -1;
248      int i;
249      int mdOffset;
250 +    int mdFileVersion;
251  
252   #ifdef IS_MPI            
253      const int masterNode = 0;
# Line 276 | Line 281 | namespace OpenMD {
281          painCave.isFatal = 1;
282          simError();
283        }
284 +      
285 +      // found the correct opening string, now try to get the file
286 +      // format version number.
287  
288 +      StringTokenizer tokenizer(line, "=<> \t\n\r");
289 +      std::string fileType = tokenizer.nextToken();
290 +      toUpper(fileType);
291 +
292 +      mdFileVersion = 0;
293 +
294 +      if (fileType == "OPENMD") {
295 +        while (tokenizer.hasMoreTokens()) {
296 +          std::string token(tokenizer.nextToken());
297 +          toUpper(token);
298 +          if (token == "VERSION") {
299 +            mdFileVersion = tokenizer.nextTokenAsInt();
300 +            break;
301 +          }
302 +        }
303 +      }
304 +            
305        //scan through the input stream and find MetaData tag        
306        while(mdFile_.getline(buffer, bufferSize)) {
307          ++lineNo;
# Line 332 | Line 357 | namespace OpenMD {
357      std::stringstream rawMetaDataStream(mdRawData);
358  
359      //parse meta-data file
360 <    Globals* simParams = parseFile(rawMetaDataStream, mdFileName, metaDataBlockStart+1);
360 >    Globals* simParams = parseFile(rawMetaDataStream, mdFileName, mdFileVersion,
361 >                                   metaDataBlockStart + 1);
362      
363      //create the force field
364      ForceField * ff = ForceFieldFactory::getInstance()->createForceField(simParams->getForceField());
# Line 589 | Line 615 | namespace OpenMD {
615        MPI_Bcast(&molToProcMap[0], nGlobalMols, MPI_INT, 0, MPI_COMM_WORLD);
616      }
617      
592 –    cerr << "molToProcMap:\n";
593 –    for (int i = 0; i < molToProcMap.size(); i++) {
594 –      cerr << "m = " << i << " mtpr[m] = " << molToProcMap[i] <<"\n";
595 –    }
596 –
618      info->setMolToProcMap(molToProcMap);
619      sprintf(checkPointMsg,
620              "Successfully divided the molecules among the processors.\n");
# Line 614 | Line 635 | namespace OpenMD {
635   #endif
636          
637          stampId = info->getMoleculeStampId(i);
638 <        Molecule * mol = molCreator.createMolecule(info->getForceField(), info->getMoleculeStamp(stampId),
639 <                                                   stampId, i, info->getLocalIndexManager());
638 >        Molecule * mol = molCreator.createMolecule(info->getForceField(),
639 >                                                   info->getMoleculeStamp(stampId),
640 >                                                   stampId, i,
641 >                                                   info->getLocalIndexManager());
642          
643          info->addMolecule(mol);
644          
# Line 642 | Line 665 | namespace OpenMD {
665      int beginRigidBodyIndex;
666      int beginCutoffGroupIndex;
667      int nGlobalAtoms = info->getNGlobalAtoms();
645 –
646 –    /**@todo fixme */
647 – #ifndef IS_MPI
668      
669      beginAtomIndex = 0;
670      beginRigidBodyIndex = 0;
671      beginCutoffGroupIndex = 0;
672 <    
673 < #else
654 <    
655 <    int nproc;
656 <    int myNode;
657 <    
658 <    myNode = worldRank;
659 <    MPI_Comm_size(MPI_COMM_WORLD, &nproc);
660 <    
661 <    std::vector < int > tmpAtomsInProc(nproc, 0);
662 <    std::vector < int > tmpRigidBodiesInProc(nproc, 0);
663 <    std::vector < int > tmpCutoffGroupsInProc(nproc, 0);
664 <    std::vector < int > NumAtomsInProc(nproc, 0);
665 <    std::vector < int > NumRigidBodiesInProc(nproc, 0);
666 <    std::vector < int > NumCutoffGroupsInProc(nproc, 0);
667 <    
668 <    tmpAtomsInProc[myNode] = info->getNAtoms();
669 <    tmpRigidBodiesInProc[myNode] = info->getNRigidBodies();
670 <    tmpCutoffGroupsInProc[myNode] = info->getNCutoffGroups();
671 <    
672 <    //do MPI_ALLREDUCE to exchange the total number of atoms, rigidbodies and cutoff groups
673 <    MPI_Allreduce(&tmpAtomsInProc[0], &NumAtomsInProc[0], nproc, MPI_INT,
674 <                  MPI_SUM, MPI_COMM_WORLD);
675 <    MPI_Allreduce(&tmpRigidBodiesInProc[0], &NumRigidBodiesInProc[0], nproc,
676 <                  MPI_INT, MPI_SUM, MPI_COMM_WORLD);
677 <    MPI_Allreduce(&tmpCutoffGroupsInProc[0], &NumCutoffGroupsInProc[0], nproc,
678 <                  MPI_INT, MPI_SUM, MPI_COMM_WORLD);
679 <    
680 <    beginAtomIndex = 0;
681 <    beginRigidBodyIndex = 0;
682 <    beginCutoffGroupIndex = 0;
683 <    
684 <    for(int i = 0; i < myNode; i++) {
685 <      beginAtomIndex += NumAtomsInProc[i];
686 <      beginRigidBodyIndex += NumRigidBodiesInProc[i];
687 <      beginCutoffGroupIndex += NumCutoffGroupsInProc[i];
688 <    }
689 <    
690 < #endif
691 <    
692 <    //rigidbody's index begins right after atom's
693 <    beginRigidBodyIndex += info->getNGlobalAtoms();
694 <    
695 <    for(mol = info->beginMolecule(mi); mol != NULL;
696 <        mol = info->nextMolecule(mi)) {
672 >
673 >    for(int i = 0; i < info->getNGlobalMolecules(); i++) {
674        
675 <      //local index(index in DataStorge) of atom is important
676 <      for(atom = mol->beginAtom(ai); atom != NULL; atom = mol->nextAtom(ai)) {
677 <        atom->setGlobalIndex(beginAtomIndex++);
675 > #ifdef IS_MPI      
676 >      if (info->getMolToProc(i) == worldRank) {
677 > #endif        
678 >        // stuff to do if I own this molecule
679 >        mol = info->getMoleculeByGlobalIndex(i);
680 >
681 >        //local index(index in DataStorge) of atom is important
682 >        for(atom = mol->beginAtom(ai); atom != NULL; atom = mol->nextAtom(ai)) {
683 >          atom->setGlobalIndex(beginAtomIndex++);
684 >        }
685 >        
686 >        for(rb = mol->beginRigidBody(ri); rb != NULL;
687 >            rb = mol->nextRigidBody(ri)) {
688 >          rb->setGlobalIndex(beginRigidBodyIndex++);
689 >        }
690 >        
691 >        //local index of cutoff group is trivial, it only depends on
692 >        //the order of travesing
693 >        for(cg = mol->beginCutoffGroup(ci); cg != NULL;
694 >            cg = mol->nextCutoffGroup(ci)) {
695 >          cg->setGlobalIndex(beginCutoffGroupIndex++);
696 >        }        
697 >        
698 > #ifdef IS_MPI        
699 >      }  else {
700 >
701 >        // stuff to do if I don't own this molecule
702 >        
703 >        int stampId = info->getMoleculeStampId(i);
704 >        MoleculeStamp* stamp = info->getMoleculeStamp(stampId);
705 >
706 >        beginAtomIndex += stamp->getNAtoms();
707 >        beginRigidBodyIndex += stamp->getNRigidBodies();
708 >        beginCutoffGroupIndex += stamp->getNCutoffGroups() + stamp->getNFreeAtoms();
709        }
710 <      
711 <      for(rb = mol->beginRigidBody(ri); rb != NULL;
712 <          rb = mol->nextRigidBody(ri)) {
713 <        rb->setGlobalIndex(beginRigidBodyIndex++);
706 <      }
707 <      
708 <      //local index of cutoff group is trivial, it only depends on the order of travesing
709 <      for(cg = mol->beginCutoffGroup(ci); cg != NULL;
710 <          cg = mol->nextCutoffGroup(ci)) {
711 <        cg->setGlobalIndex(beginCutoffGroupIndex++);
712 <      }
713 <    }
714 <    
710 > #endif          
711 >
712 >    } //end for(int i=0)  
713 >
714      //fill globalGroupMembership
715 <    std::vector<int> globalGroupMembership(info->getNGlobalAtoms(), -1);
715 >    std::vector<int> globalGroupMembership(info->getNGlobalAtoms(), 0);
716      for(mol = info->beginMolecule(mi); mol != NULL; mol = info->nextMolecule(mi)) {        
717        for (cg = mol->beginCutoffGroup(ci); cg != NULL; cg = mol->nextCutoffGroup(ci)) {
718          
# Line 732 | Line 731 | namespace OpenMD {
731      // docs said we could.
732      std::vector<int> tmpGroupMembership(info->getNGlobalAtoms(), 0);
733      MPI_Allreduce(&globalGroupMembership[0], &tmpGroupMembership[0], nGlobalAtoms,
734 <                  MPI_INT, MPI_MAX, MPI_COMM_WORLD);
734 >                  MPI_INT, MPI_SUM, MPI_COMM_WORLD);
735      info->setGlobalGroupMembership(tmpGroupMembership);
737 –
738 –    cerr << "ggm:\n";
739 –    for (int i = 0; i < tmpGroupMembership.size(); i++)
740 –      cerr << "i = " << i << "\t ggm(i) = " << tmpGroupMembership[i] << "\n";
741 –
736   #else
737      info->setGlobalGroupMembership(globalGroupMembership);
738   #endif
# Line 798 | Line 792 | namespace OpenMD {
792          globalIO++;
793        }
794      }
801 –    cerr << "ioi2io:\n";
802 –    for (int i = 0; i < IOIndexToIntegrableObject.size(); i++) {
803 –      if (IOIndexToIntegrableObject[i] != NULL) {
804 –        cerr << "i = " << i << "globalIOindex = " << IOIndexToIntegrableObject[i]->getGlobalIntegrableObjectIndex() << "\n";
805 –      }
806 –    }
795        
796      info->setIOIndexToIntegrableObject(IOIndexToIntegrableObject);
797      

Diff Legend

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