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 1627 by gezelter, Tue Sep 13 22:05:04 2011 UTC vs.
Revision 1715 by gezelter, Tue May 22 21:55:31 2012 UTC

# Line 36 | Line 36
36   * [1]  Meineke, et al., J. Comp. Chem. 26, 252-271 (2005).            
37   * [2]  Fennell & Gezelter, J. Chem. Phys. 124, 234104 (2006).          
38   * [3]  Sun, Lin & Gezelter, J. Chem. Phys. 128, 24107 (2008).          
39 < * [4]  Vardeman & Gezelter, in progress (2009).                        
39 > * [4]  Kuang & Gezelter,  J. Chem. Phys. 133, 164101 (2010).
40 > * [5]  Vardeman, Stocker & Gezelter, J. Chem. Theory Comput. 7, 834 (2011).
41   */
42  
43   /**
# Line 74 | Line 75
75   #include "antlr/MismatchedTokenException.hpp"
76   #include "antlr/NoViableAltForCharException.hpp"
77   #include "antlr/NoViableAltException.hpp"
78 +
79 + #include "types/DirectionalAdapter.hpp"
80 + #include "types/MultipoleAdapter.hpp"
81 + #include "types/EAMAdapter.hpp"
82 + #include "types/SuttonChenAdapter.hpp"
83 + #include "types/PolarizableAdapter.hpp"
84 + #include "types/FixedChargeAdapter.hpp"
85 + #include "types/FluctuatingChargeAdapter.hpp"
86  
87   #ifdef IS_MPI
88   #include "mpi.h"
# Line 413 | Line 422 | namespace OpenMD {
422      //create the molecules
423      createMolecules(info);
424      
425 +    //find the storage layout
426 +
427 +    int storageLayout = computeStorageLayout(info);
428 +
429      //allocate memory for DataStorage(circular reference, need to
430      //break it)
431 <    info->setSnapshotManager(new SimSnapshotManager(info));
431 >    info->setSnapshotManager(new SimSnapshotManager(info, storageLayout));
432      
433      //set the global index of atoms, rigidbodies and cutoffgroups
434      //(only need to be set once, the global index will never change
# Line 652 | Line 665 | namespace OpenMD {
665      } //end for(int i=0)  
666    }
667      
668 +  int SimCreator::computeStorageLayout(SimInfo* info) {
669 +
670 +    Globals* simParams = info->getSimParams();
671 +    int nRigidBodies = info->getNGlobalRigidBodies();
672 +    set<AtomType*> atomTypes = info->getSimulatedAtomTypes();
673 +    set<AtomType*>::iterator i;
674 +    bool hasDirectionalAtoms = false;
675 +    bool hasFixedCharge = false;
676 +    bool hasMultipoles = false;    
677 +    bool hasPolarizable = false;    
678 +    bool hasFluctuatingCharge = false;    
679 +    bool hasMetallic = false;
680 +    int storageLayout = 0;
681 +    storageLayout |= DataStorage::dslPosition;
682 +    storageLayout |= DataStorage::dslVelocity;
683 +    storageLayout |= DataStorage::dslForce;
684 +
685 +    for (i = atomTypes.begin(); i != atomTypes.end(); ++i) {
686 +
687 +      DirectionalAdapter da = DirectionalAdapter( (*i) );
688 +      MultipoleAdapter ma = MultipoleAdapter( (*i) );
689 +      EAMAdapter ea = EAMAdapter( (*i) );
690 +      SuttonChenAdapter sca = SuttonChenAdapter( (*i) );
691 +      PolarizableAdapter pa = PolarizableAdapter( (*i) );
692 +      FixedChargeAdapter fca = FixedChargeAdapter( (*i) );
693 +      FluctuatingChargeAdapter fqa = FluctuatingChargeAdapter( (*i) );
694 +
695 +      if (da.isDirectional()){
696 +        hasDirectionalAtoms = true;
697 +      }
698 +      if (ma.isMultipole()){
699 +        hasMultipoles = true;
700 +      }
701 +      if (ea.isEAM() || sca.isSuttonChen()){
702 +        hasMetallic = true;
703 +      }
704 +      if ( fca.isFixedCharge() ){
705 +        hasFixedCharge = true;
706 +      }
707 +      if ( fqa.isFluctuatingCharge() ){
708 +        hasFluctuatingCharge = true;
709 +      }
710 +      if ( pa.isPolarizable() ){
711 +        hasPolarizable = true;
712 +      }
713 +    }
714 +    
715 +    if (nRigidBodies > 0 || hasDirectionalAtoms) {
716 +      storageLayout |= DataStorage::dslAmat;
717 +      if(storageLayout & DataStorage::dslVelocity) {
718 +        storageLayout |= DataStorage::dslAngularMomentum;
719 +      }
720 +      if (storageLayout & DataStorage::dslForce) {
721 +        storageLayout |= DataStorage::dslTorque;
722 +      }
723 +    }
724 +    if (hasMultipoles) {
725 +      storageLayout |= DataStorage::dslElectroFrame;
726 +    }
727 +    if (hasFixedCharge || hasFluctuatingCharge) {
728 +      storageLayout |= DataStorage::dslSkippedCharge;
729 +    }
730 +    if (hasMetallic) {
731 +      storageLayout |= DataStorage::dslDensity;
732 +      storageLayout |= DataStorage::dslFunctional;
733 +      storageLayout |= DataStorage::dslFunctionalDerivative;
734 +    }
735 +    if (hasPolarizable) {
736 +      storageLayout |= DataStorage::dslElectricField;
737 +    }
738 +    if (hasFluctuatingCharge){
739 +      storageLayout |= DataStorage::dslFlucQPosition;
740 +      if(storageLayout & DataStorage::dslVelocity) {
741 +        storageLayout |= DataStorage::dslFlucQVelocity;
742 +      }
743 +      if (storageLayout & DataStorage::dslForce) {
744 +        storageLayout |= DataStorage::dslFlucQForce;
745 +      }
746 +    }
747 +    
748 +    // if the user has asked for them, make sure we've got the memory for the
749 +    // objects defined.
750 +
751 +    if (simParams->getOutputParticlePotential()) {
752 +      storageLayout |= DataStorage::dslParticlePot;
753 +    }
754 +    if (simParams->getOutputElectricField()) {
755 +      storageLayout |= DataStorage::dslElectricField;
756 +    }
757 +    if (simParams->getOutputFluctuatingCharges()) {
758 +      storageLayout |= DataStorage::dslFlucQPosition;
759 +      storageLayout |= DataStorage::dslFlucQVelocity;
760 +      storageLayout |= DataStorage::dslFlucQForce;
761 +    }
762 +
763 +    return storageLayout;
764 +  }
765 +
766    void SimCreator::setGlobalIndex(SimInfo *info) {
767      SimInfo::MoleculeIterator mi;
768      Molecule::AtomIterator ai;

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines