44 |
|
* @file ForceManager.cpp |
45 |
|
* @author tlin |
46 |
|
* @date 11/09/2004 |
47 |
– |
* @time 10:39am |
47 |
|
* @version 1.0 |
48 |
|
*/ |
49 |
|
|
57 |
|
#include "primitives/Torsion.hpp" |
58 |
|
#include "primitives/Inversion.hpp" |
59 |
|
#include "nonbonded/NonBondedInteraction.hpp" |
60 |
+ |
#include "perturbations/ElectricField.hpp" |
61 |
|
#include "parallel/ForceMatrixDecomposition.hpp" |
62 |
|
|
63 |
|
#include <cstdio> |
369 |
|
} |
370 |
|
switcher_->setSwitchType(sft_); |
371 |
|
switcher_->setSwitch(rSwitch_, rCut_); |
372 |
– |
interactionMan_->setSwitchingRadius(rSwitch_); |
372 |
|
} |
373 |
|
|
374 |
|
|
392 |
|
doParticlePot_ = info_->getSimParams()->getOutputParticlePotential(); |
393 |
|
doHeatFlux_ = info_->getSimParams()->getPrintHeatFlux(); |
394 |
|
if (doHeatFlux_) doParticlePot_ = true; |
395 |
+ |
|
396 |
+ |
doElectricField_ = info_->getSimParams()->getOutputElectricField(); |
397 |
|
|
398 |
|
} |
399 |
|
|
424 |
|
electrostaticScale_[2] = fopts.getelectrostatic13scale(); |
425 |
|
electrostaticScale_[3] = fopts.getelectrostatic14scale(); |
426 |
|
|
427 |
+ |
if (info_->getSimParams()->haveElectricField()) { |
428 |
+ |
ElectricField* eField = new ElectricField(info_); |
429 |
+ |
perturbations_.push_back(eField); |
430 |
+ |
} |
431 |
+ |
|
432 |
|
fDecomp_->distributeInitialData(); |
433 |
|
|
434 |
|
initialized_ = true; |
455 |
|
Molecule::CutoffGroupIterator ci; |
456 |
|
CutoffGroup* cg; |
457 |
|
|
458 |
< |
// forces are zeroed here, before any are accumulated. |
458 |
> |
// forces and potentials are zeroed here, before any are |
459 |
> |
// accumulated. |
460 |
|
|
461 |
+ |
Snapshot* snap = info_->getSnapshotManager()->getCurrentSnapshot(); |
462 |
+ |
|
463 |
+ |
snap->setBondPotential(0.0); |
464 |
+ |
snap->setBendPotential(0.0); |
465 |
+ |
snap->setTorsionPotential(0.0); |
466 |
+ |
snap->setInversionPotential(0.0); |
467 |
+ |
|
468 |
+ |
potVec zeroPot(0.0); |
469 |
+ |
snap->setLongRangePotential(zeroPot); |
470 |
+ |
snap->setExcludedPotentials(zeroPot); |
471 |
+ |
|
472 |
+ |
snap->setRestraintPotential(0.0); |
473 |
+ |
snap->setRawPotential(0.0); |
474 |
+ |
|
475 |
|
for (mol = info_->beginMolecule(mi); mol != NULL; |
476 |
|
mol = info_->nextMolecule(mi)) { |
477 |
|
for(atom = mol->beginAtom(ai); atom != NULL; |
607 |
|
} |
608 |
|
} |
609 |
|
} |
610 |
< |
|
611 |
< |
RealType shortRangePotential = bondPotential + bendPotential + |
612 |
< |
torsionPotential + inversionPotential; |
610 |
> |
|
611 |
> |
#ifdef IS_MPI |
612 |
> |
// Collect from all nodes. This should eventually be moved into a |
613 |
> |
// SystemDecomposition, but this is a better place than in |
614 |
> |
// Thermo to do the collection. |
615 |
> |
MPI::COMM_WORLD.Allreduce(MPI::IN_PLACE, &bondPotential, 1, MPI::REALTYPE, |
616 |
> |
MPI::SUM); |
617 |
> |
MPI::COMM_WORLD.Allreduce(MPI::IN_PLACE, &bendPotential, 1, MPI::REALTYPE, |
618 |
> |
MPI::SUM); |
619 |
> |
MPI::COMM_WORLD.Allreduce(MPI::IN_PLACE, &torsionPotential, 1, |
620 |
> |
MPI::REALTYPE, MPI::SUM); |
621 |
> |
MPI::COMM_WORLD.Allreduce(MPI::IN_PLACE, &inversionPotential, 1, |
622 |
> |
MPI::REALTYPE, MPI::SUM); |
623 |
> |
#endif |
624 |
> |
|
625 |
|
Snapshot* curSnapshot = info_->getSnapshotManager()->getCurrentSnapshot(); |
626 |
< |
curSnapshot->statData[Stats::SHORT_RANGE_POTENTIAL] = shortRangePotential; |
627 |
< |
curSnapshot->statData[Stats::BOND_POTENTIAL] = bondPotential; |
628 |
< |
curSnapshot->statData[Stats::BEND_POTENTIAL] = bendPotential; |
629 |
< |
curSnapshot->statData[Stats::DIHEDRAL_POTENTIAL] = torsionPotential; |
630 |
< |
curSnapshot->statData[Stats::INVERSION_POTENTIAL] = inversionPotential; |
626 |
> |
|
627 |
> |
curSnapshot->setBondPotential(bondPotential); |
628 |
> |
curSnapshot->setBendPotential(bendPotential); |
629 |
> |
curSnapshot->setTorsionPotential(torsionPotential); |
630 |
> |
curSnapshot->setInversionPotential(inversionPotential); |
631 |
> |
|
632 |
> |
// RealType shortRangePotential = bondPotential + bendPotential + |
633 |
> |
// torsionPotential + inversionPotential; |
634 |
> |
|
635 |
> |
// curSnapshot->setShortRangePotential(shortRangePotential); |
636 |
|
} |
637 |
|
|
638 |
|
void ForceManager::longRangeInteractions() { |
681 |
|
InteractionData idat; |
682 |
|
SelfData sdat; |
683 |
|
RealType mf; |
646 |
– |
RealType lrPot; |
684 |
|
RealType vpair; |
685 |
|
RealType dVdFQ1(0.0); |
686 |
|
RealType dVdFQ2(0.0); |
687 |
|
potVec longRangePotential(0.0); |
688 |
|
potVec workPot(0.0); |
689 |
+ |
potVec exPot(0.0); |
690 |
+ |
Vector3d eField1(0.0); |
691 |
+ |
Vector3d eField2(0.0); |
692 |
|
vector<int>::iterator ia, jb; |
693 |
|
|
694 |
|
int loopStart, loopEnd; |
696 |
|
idat.vdwMult = &vdwMult; |
697 |
|
idat.electroMult = &electroMult; |
698 |
|
idat.pot = &workPot; |
699 |
+ |
idat.excludedPot = &exPot; |
700 |
|
sdat.pot = fDecomp_->getEmbeddingPotential(); |
701 |
+ |
sdat.excludedPot = fDecomp_->getExcludedSelfPotential(); |
702 |
|
idat.vpair = &vpair; |
703 |
|
idat.dVdFQ1 = &dVdFQ1; |
704 |
|
idat.dVdFQ2 = &dVdFQ2; |
705 |
+ |
idat.eField1 = &eField1; |
706 |
+ |
idat.eField2 = &eField2; |
707 |
|
idat.f1 = &f1; |
708 |
|
idat.sw = &sw; |
709 |
|
idat.shiftedPot = (cutoffMethod_ == SHIFTED_POTENTIAL) ? true : false; |
710 |
|
idat.shiftedForce = (cutoffMethod_ == SHIFTED_FORCE) ? true : false; |
711 |
|
idat.doParticlePot = doParticlePot_; |
712 |
+ |
idat.doElectricField = doElectricField_; |
713 |
|
sdat.doParticlePot = doParticlePot_; |
714 |
|
|
715 |
|
loopEnd = PAIR_LOOP; |
749 |
|
|
750 |
|
in_switching_region = switcher_->getSwitch(rgrpsq, sw, dswdr, |
751 |
|
rgrp); |
752 |
+ |
|
753 |
|
atomListRow = fDecomp_->getAtomsInGroupRow(cg1); |
754 |
|
atomListColumn = fDecomp_->getAtomsInGroupColumn(cg2); |
755 |
|
|
764 |
|
jb != atomListColumn.end(); ++jb) { |
765 |
|
atom2 = (*jb); |
766 |
|
|
767 |
< |
if (!fDecomp_->skipAtomPair(atom1, atom2)) { |
767 |
> |
if (!fDecomp_->skipAtomPair(atom1, atom2, cg1, cg2)) { |
768 |
> |
|
769 |
|
vpair = 0.0; |
770 |
|
workPot = 0.0; |
771 |
+ |
exPot = 0.0; |
772 |
|
f1 = V3Zero; |
773 |
|
dVdFQ1 = 0.0; |
774 |
|
dVdFQ2 = 0.0; |
879 |
|
|
880 |
|
fDecomp_->collectIntermediateData(); |
881 |
|
|
882 |
< |
for (int atom1 = 0; atom1 < info_->getNAtoms(); atom1++) { |
882 |
> |
for (unsigned int atom1 = 0; atom1 < info_->getNAtoms(); atom1++) { |
883 |
|
fDecomp_->fillSelfData(sdat, atom1); |
884 |
|
interactionMan_->doPreForce(sdat); |
885 |
|
} |
890 |
|
} |
891 |
|
} |
892 |
|
|
893 |
+ |
// collects pairwise information |
894 |
|
fDecomp_->collectData(); |
895 |
|
|
896 |
|
if (info_->requiresSelfCorrection()) { |
897 |
< |
|
849 |
< |
for (int atom1 = 0; atom1 < info_->getNAtoms(); atom1++) { |
897 |
> |
for (unsigned int atom1 = 0; atom1 < info_->getNAtoms(); atom1++) { |
898 |
|
fDecomp_->fillSelfData(sdat, atom1); |
899 |
|
interactionMan_->doSelfCorrection(sdat); |
900 |
|
} |
853 |
– |
|
901 |
|
} |
902 |
|
|
903 |
+ |
// collects single-atom information |
904 |
+ |
fDecomp_->collectSelfData(); |
905 |
+ |
|
906 |
|
longRangePotential = *(fDecomp_->getEmbeddingPotential()) + |
907 |
|
*(fDecomp_->getPairwisePotential()); |
908 |
|
|
909 |
< |
lrPot = longRangePotential.sum(); |
909 |
> |
curSnapshot->setLongRangePotential(longRangePotential); |
910 |
|
|
911 |
< |
//store the stressTensor and long range potential |
912 |
< |
curSnapshot->statData[Stats::LONG_RANGE_POTENTIAL] = lrPot; |
913 |
< |
curSnapshot->statData[Stats::VANDERWAALS_POTENTIAL] = longRangePotential[VANDERWAALS_FAMILY]; |
914 |
< |
curSnapshot->statData[Stats::ELECTROSTATIC_POTENTIAL] = longRangePotential[ELECTROSTATIC_FAMILY]; |
911 |
> |
// collects single-atom information |
912 |
> |
fDecomp_->collectSelfData(); |
913 |
> |
|
914 |
> |
longRangePotential = *(fDecomp_->getEmbeddingPotential()) + |
915 |
> |
*(fDecomp_->getPairwisePotential()); |
916 |
> |
|
917 |
> |
curSnapshot->setLongRangePotential(longRangePotential); |
918 |
> |
|
919 |
> |
curSnapshot->setExcludedPotentials(*(fDecomp_->getExcludedSelfPotential()) + |
920 |
> |
*(fDecomp_->getExcludedPotential())); |
921 |
> |
|
922 |
|
} |
923 |
|
|
924 |
|
|
925 |
|
void ForceManager::postCalculation() { |
926 |
+ |
|
927 |
+ |
vector<Perturbation*>::iterator pi; |
928 |
+ |
for (pi = perturbations_.begin(); pi != perturbations_.end(); ++pi) { |
929 |
+ |
(*pi)->applyPerturbation(); |
930 |
+ |
} |
931 |
+ |
|
932 |
|
SimInfo::MoleculeIterator mi; |
933 |
|
Molecule* mol; |
934 |
|
Molecule::RigidBodyIterator rbIter; |
935 |
|
RigidBody* rb; |
936 |
|
Snapshot* curSnapshot = info_->getSnapshotManager()->getCurrentSnapshot(); |
937 |
< |
|
937 |
> |
|
938 |
|
// collect the atomic forces onto rigid bodies |
939 |
|
|
940 |
|
for (mol = info_->beginMolecule(mi); mol != NULL; |
947 |
|
} |
948 |
|
|
949 |
|
#ifdef IS_MPI |
887 |
– |
|
950 |
|
MPI::COMM_WORLD.Allreduce(MPI::IN_PLACE, stressTensor.getArrayPointer(), 9, |
951 |
|
MPI::REALTYPE, MPI::SUM); |
952 |
|
#endif |
953 |
|
curSnapshot->setStressTensor(stressTensor); |
954 |
|
|
955 |
|
} |
894 |
– |
|
956 |
|
} //end namespace OpenMD |