| 1 | 
  | 
#include "Integrator.hpp" | 
| 2 | 
  | 
#include "simError.h" | 
| 3 | 
< | 
#include <cmath> | 
| 3 | 
> | 
#include <math.h> | 
| 4 | 
  | 
template<typename T> ZConstraint<T>::ZConstraint(SimInfo* theInfo, ForceFields* the_ff)  | 
| 5 | 
< | 
                                    : T(theInfo, the_ff), fz(NULL), curZPos(NULL), | 
| 6 | 
< | 
                                 indexOfZConsMols(NULL), forcePolicy(NULL), curZconsTime(0) | 
| 5 | 
> | 
                                    : T(theInfo, the_ff), indexOfZConsMols(NULL), fz(NULL), curZPos(NULL), | 
| 6 | 
> | 
                                      fzOut(NULL), curZconsTime(0), forcePolicy(NULL) | 
| 7 | 
  | 
{ | 
| 8 | 
  | 
 | 
| 9 | 
  | 
  //get properties from SimInfo | 
| 27 | 
  | 
  double halfOfLargestBox = max(info->boxL[0], max(info->boxL[1], info->boxL[2])) /2; | 
| 28 | 
  | 
  zForceConst = Kb * info->target_temp /(halfOfLargestBox * halfOfLargestBox); | 
| 29 | 
  | 
 | 
| 30 | 
< | 
  //creat force substraction policy | 
| 30 | 
> | 
  //creat force Subtraction policy | 
| 31 | 
  | 
  data = info->getProperty(ZCONSFORCEPOLICY_ID); | 
| 32 | 
  | 
  if(!data){ | 
| 33 | 
  | 
    sprintf( painCave.errMsg, | 
| 34 | 
< | 
               "ZConstraint Warning: User does not set force substraction policy, " | 
| 35 | 
< | 
               "average force substraction policy is used\n"); | 
| 34 | 
> | 
               "ZConstraint Warning: User does not set force Subtraction policy, " | 
| 35 | 
> | 
               "PolicyByMass is used\n"); | 
| 36 | 
  | 
    painCave.isFatal = 0; | 
| 37 | 
  | 
    simError();       | 
| 38 | 
  | 
 | 
| 39 | 
< | 
    forcePolicy = (ForceSubstractionPolicy*) new PolicyByNumber(this); | 
| 39 | 
> | 
    forcePolicy = (ForceSubtractionPolicy*) new PolicyByMass(this); | 
| 40 | 
  | 
  } | 
| 41 | 
  | 
  else{ | 
| 42 | 
  | 
    policy = dynamic_cast<StringData*>(data); | 
| 44 | 
  | 
    if(!policy){ | 
| 45 | 
  | 
      sprintf( painCave.errMsg, | 
| 46 | 
  | 
                 "ZConstraint Error: Convertion from GenericData to StringData failure, " | 
| 47 | 
< | 
                 "average force substraction policy is used\n"); | 
| 47 | 
> | 
                 "PolicyByMass is used\n"); | 
| 48 | 
  | 
      painCave.isFatal = 0; | 
| 49 | 
  | 
      simError();       | 
| 50 | 
  | 
 | 
| 51 | 
< | 
      forcePolicy = (ForceSubstractionPolicy*) new PolicyByNumber(this); | 
| 51 | 
> | 
      forcePolicy = (ForceSubtractionPolicy*) new PolicyByMass(this); | 
| 52 | 
  | 
    } | 
| 53 | 
  | 
    else{ | 
| 54 | 
  | 
      if(policy->getData() == "BYNUMBER") | 
| 55 | 
< | 
        forcePolicy = (ForceSubstractionPolicy*) new PolicyByNumber(this); | 
| 55 | 
> | 
        forcePolicy = (ForceSubtractionPolicy*) new PolicyByNumber(this); | 
| 56 | 
  | 
      else if(policy->getData() == "BYMASS") | 
| 57 | 
< | 
        forcePolicy = (ForceSubstractionPolicy*) new PolicyByMass(this); | 
| 57 | 
> | 
        forcePolicy = (ForceSubtractionPolicy*) new PolicyByMass(this); | 
| 58 | 
  | 
      else{ | 
| 59 | 
  | 
        sprintf( painCave.errMsg, | 
| 60 | 
< | 
                  "ZConstraint Warning: unknown force substraction policy, " | 
| 61 | 
< | 
                  "average force substraction policy is used\n"); | 
| 60 | 
> | 
                  "ZConstraint Warning: unknown force Subtraction policy, " | 
| 61 | 
> | 
                  "PolicyByMass is used\n"); | 
| 62 | 
  | 
        painCave.isFatal = 0; | 
| 63 | 
  | 
        simError();       | 
| 64 | 
+ | 
        forcePolicy = (ForceSubtractionPolicy*) new PolicyByMass(this); | 
| 65 | 
  | 
      }    | 
| 66 | 
  | 
    } | 
| 67 | 
  | 
  } | 
| 213 | 
  | 
 | 
| 214 | 
  | 
      //if user does not specify the zpos for the zconstraint molecule | 
| 215 | 
  | 
      //its initial z coordinate  will be used as default | 
| 216 | 
< | 
      for(int i = 0; i < parameters->size(); i++){ | 
| 216 | 
> | 
      for(int i = 0; i < (int)(parameters->size()); i++){ | 
| 217 | 
  | 
 | 
| 218 | 
  | 
        if(!(*parameters)[i].havingZPos){ | 
| 219 | 
  | 
#ifndef IS_MPI | 
| 227 | 
  | 
            //query which processor current zconstraint molecule belongs to | 
| 228 | 
  | 
          int *MolToProcMap; | 
| 229 | 
  | 
          int whichNode; | 
| 230 | 
< | 
    double initZPos; | 
| 230 | 
> | 
           | 
| 231 | 
  | 
          MolToProcMap = mpiSim->getMolToProcMap(); | 
| 232 | 
  | 
          whichNode = MolToProcMap[(*parameters)[i].zconsIndex]; | 
| 233 | 
  | 
         | 
| 244 | 
  | 
          | 
| 245 | 
  | 
          } | 
| 246 | 
  | 
 | 
| 247 | 
< | 
          MPI_Bcast(&COM[whichDirection], 1, MPI_DOUBLE_PRECISION, whichNode, MPI_COMM_WORLD);         | 
| 247 | 
> | 
          MPI_Bcast(&COM[whichDirection], 1, MPI_DOUBLE, whichNode, MPI_COMM_WORLD);         | 
| 248 | 
  | 
#endif | 
| 249 | 
  | 
             | 
| 250 | 
  | 
          (*parameters)[i].zPos = COM[whichDirection]; | 
| 251 | 
  | 
 | 
| 252 | 
  | 
          sprintf( painCave.errMsg, | 
| 253 | 
< | 
                     "ZConstraint warningr: Does not specify zpos for z-constraint molecule " | 
| 253 | 
> | 
                     "ZConstraint warning: Does not specify zpos for z-constraint molecule " | 
| 254 | 
  | 
                     "initial z coornidate will be used \n"); | 
| 255 | 
  | 
           painCave.isFatal = 0; | 
| 256 | 
  | 
           simError();   | 
| 277 | 
  | 
      massOfZConsMols.push_back(molecules[i].getTotalMass());   | 
| 278 | 
  | 
 | 
| 279 | 
  | 
      zPos.push_back((*parameters)[searchResult].zPos); | 
| 280 | 
< | 
      cout << "index: "<< (*parameters)[searchResult].zconsIndex  | 
| 281 | 
< | 
             <<"\tzPos = " << (*parameters)[searchResult].zPos << endl;  | 
| 281 | 
< | 
      kz.push_back((*parameters)[searchResult]. kRatio * zForceConst); | 
| 280 | 
> | 
//       cout << "index: "<< (*parameters)[searchResult].zconsIndex  | 
| 281 | 
> | 
//              <<"\tzPos = " << (*parameters)[searchResult].zPos << endl;  | 
| 282 | 
  | 
       | 
| 283 | 
+ | 
      kz.push_back((*parameters)[searchResult]. kRatio * zForceConst); | 
| 284 | 
  | 
      molecules[i].getCOM(COM); | 
| 285 | 
  | 
    } | 
| 286 | 
  | 
    else | 
| 304 | 
  | 
  } | 
| 305 | 
  | 
 | 
| 306 | 
  | 
  //determine the states of z-constraint molecules | 
| 307 | 
< | 
  for(int i = 0; i < zconsMols.size(); i++){ | 
| 307 | 
> | 
  for(int i = 0; i < (int)(zconsMols.size()); i++){ | 
| 308 | 
  | 
    indexOfZConsMols[i] = zconsMols[i]->getGlobalIndex(); | 
| 309 | 
  | 
 | 
| 310 | 
  | 
    zconsMols[i]->getCOM(COM); | 
| 320 | 
  | 
  double totalMassOfUncons_local; | 
| 321 | 
  | 
  totalMassOfUncons_local = 0; | 
| 322 | 
  | 
   | 
| 323 | 
< | 
  for(int i = 0; i < unconsMols.size(); i++) | 
| 323 | 
> | 
  for(int i = 0; i < (int)(unconsMols.size()); i++) | 
| 324 | 
  | 
    totalMassOfUncons_local += unconsMols[i]->getTotalMass(); | 
| 325 | 
  | 
     | 
| 326 | 
  | 
#ifndef IS_MPI | 
| 330 | 
  | 
                      MPI_DOUBLE, MPI_SUM, MPI_COMM_WORLD);   | 
| 331 | 
  | 
#endif | 
| 332 | 
  | 
 | 
| 332 | 
– | 
 | 
| 333 | 
  | 
  //get total number of unconstrained atoms | 
| 334 | 
  | 
  int nUnconsAtoms_local; | 
| 335 | 
  | 
  nUnconsAtoms_local = 0; | 
| 336 | 
< | 
  for(int i = 0; i < unconsMols.size(); i++) | 
| 336 | 
> | 
  for(int i = 0; i < (int)(unconsMols.size()); i++) | 
| 337 | 
  | 
    nUnconsAtoms_local += unconsMols[i]->getNAtoms(); | 
| 338 | 
  | 
     | 
| 339 | 
  | 
#ifndef IS_MPI | 
| 343 | 
  | 
                      MPI_INT,MPI_SUM, MPI_COMM_WORLD);   | 
| 344 | 
  | 
#endif   | 
| 345 | 
  | 
 | 
| 346 | 
– | 
  // creat zconsWriter   | 
| 347 | 
– | 
  fzOut = new ZConsWriter(zconsOutput.c_str(), parameters);    | 
| 348 | 
– | 
   | 
| 349 | 
– | 
  if(!fzOut){ | 
| 350 | 
– | 
    sprintf( painCave.errMsg, | 
| 351 | 
– | 
             "Memory allocation failure in class Zconstraint\n"); | 
| 352 | 
– | 
    painCave.isFatal = 1; | 
| 353 | 
– | 
    simError(); | 
| 354 | 
– | 
  } | 
| 355 | 
– | 
 | 
| 346 | 
  | 
  forcePolicy->update(); | 
| 347 | 
  | 
} | 
| 348 | 
  | 
 | 
| 394 | 
  | 
      zconsMols.push_back(&molecules[i]);       | 
| 395 | 
  | 
      zPos.push_back((*parameters)[index].zPos); | 
| 396 | 
  | 
      kz.push_back((*parameters)[index].kRatio * zForceConst); | 
| 407 | 
– | 
       | 
| 397 | 
  | 
      massOfZConsMols.push_back(molecules[i].getTotalMass());   | 
| 398 | 
  | 
        | 
| 399 | 
  | 
      molecules[i].getCOM(COM); | 
| 408 | 
  | 
  } | 
| 409 | 
  | 
 | 
| 410 | 
  | 
  //determine the states of z-constraint molecules | 
| 411 | 
< | 
  for(int i = 0; i < zconsMols.size(); i++){ | 
| 411 | 
> | 
  for(int i = 0; i < (int)(zconsMols.size()); i++){ | 
| 412 | 
  | 
     zconsMols[i]->getCOM(COM); | 
| 413 | 
  | 
      if (fabs(zPos[i] - COM[whichDirection]) < zconsTol) | 
| 414 | 
  | 
      states.push_back(zcsFixed); | 
| 440 | 
  | 
      simError(); | 
| 441 | 
  | 
    } | 
| 442 | 
  | 
         | 
| 443 | 
< | 
    for(int i = 0; i < zconsMols.size(); i++){ | 
| 443 | 
> | 
    for(int i = 0; i < (int)(zconsMols.size()); i++){ | 
| 444 | 
  | 
      indexOfZConsMols[i] = zconsMols[i]->getGlobalIndex(); | 
| 445 | 
  | 
    } | 
| 446 | 
  | 
 | 
| 494 | 
  | 
} | 
| 495 | 
  | 
 | 
| 496 | 
  | 
template<typename T> void ZConstraint<T>::integrate(){ | 
| 497 | 
+ | 
 | 
| 498 | 
+ | 
  // creat zconsWriter   | 
| 499 | 
+ | 
  fzOut = new ZConsWriter(zconsOutput.c_str(), parameters);    | 
| 500 | 
  | 
   | 
| 501 | 
+ | 
  if(!fzOut){ | 
| 502 | 
+ | 
    sprintf( painCave.errMsg, | 
| 503 | 
+ | 
             "Memory allocation failure in class Zconstraint\n"); | 
| 504 | 
+ | 
    painCave.isFatal = 1; | 
| 505 | 
+ | 
    simError(); | 
| 506 | 
+ | 
  } | 
| 507 | 
+ | 
   | 
| 508 | 
  | 
  //zero out the velocities of center of mass of unconstrained molecules  | 
| 509 | 
  | 
  //and the velocities of center of mass of every single z-constrained molecueles | 
| 510 | 
  | 
  zeroOutVel(); | 
| 530 | 
  | 
 | 
| 531 | 
  | 
  T::calcForce(calcPot, calcStress); | 
| 532 | 
  | 
 | 
| 533 | 
< | 
  if (checkZConsState()){ | 
| 534 | 
< | 
     | 
| 536 | 
< | 
#ifdef IS_MPI | 
| 537 | 
< | 
    if(worldRank == 0){ | 
| 538 | 
< | 
#endif | 
| 539 | 
< | 
      std::cerr << "\n" | 
| 540 | 
< | 
                << "*******************************************\n" | 
| 541 | 
< | 
                << " about to call zeroOutVel()\n" | 
| 542 | 
< | 
                << "*******************************************\n" | 
| 543 | 
< | 
                << "\n"; | 
| 544 | 
< | 
#ifdef IS_MPI | 
| 545 | 
< | 
    } | 
| 546 | 
< | 
#endif | 
| 547 | 
< | 
    zeroOutVel(); | 
| 548 | 
< | 
 | 
| 549 | 
< | 
#ifdef IS_MPI | 
| 550 | 
< | 
    if(worldRank == 0){ | 
| 551 | 
< | 
#endif | 
| 552 | 
< | 
      std::cerr << "\n" | 
| 553 | 
< | 
                << "*******************************************\n" | 
| 554 | 
< | 
                << " finished zeroOutVel()\n" | 
| 555 | 
< | 
                << "*******************************************\n" | 
| 556 | 
< | 
                << "\n"; | 
| 557 | 
< | 
#ifdef IS_MPI | 
| 558 | 
< | 
    } | 
| 559 | 
< | 
#endif | 
| 560 | 
< | 
     | 
| 533 | 
> | 
  if (checkZConsState()){     | 
| 534 | 
> | 
    zeroOutVel();     | 
| 535 | 
  | 
    forcePolicy->update(); | 
| 536 | 
  | 
  }   | 
| 537 | 
  | 
   | 
| 540 | 
  | 
#ifdef IS_MPI | 
| 541 | 
  | 
  if(worldRank == 0){ | 
| 542 | 
  | 
#endif | 
| 543 | 
< | 
    cout << "---------------------------------------------------------------------" <<endl; | 
| 544 | 
< | 
    cout << "current time: " << info->getTime() << endl; | 
| 545 | 
< | 
    cout << "center of mass at z: " << zsys << endl;      | 
| 546 | 
< | 
    cout << "before calcForce, the COMVel of system is " << zSysCOMVel <<endl; | 
| 543 | 
> | 
     //cout << "---------------------------------------------------------------------" <<endl; | 
| 544 | 
> | 
     //cout << "current time: " << info->getTime() << endl; | 
| 545 | 
> | 
     //cout << "center of mass at z: " << zsys << endl;      | 
| 546 | 
> | 
     //cout << "before calcForce, the COMVel of system is " << zSysCOMVel <<endl; | 
| 547 | 
  | 
 | 
| 548 | 
  | 
#ifdef IS_MPI | 
| 549 | 
  | 
  } | 
| 552 | 
  | 
  //do zconstraint force;  | 
| 553 | 
  | 
  if (haveFixedZMols()) | 
| 554 | 
  | 
    this->doZconstraintForce(); | 
| 555 | 
< | 
      | 
| 555 | 
> | 
 | 
| 556 | 
  | 
  //use harmonical poteintial to move the molecules to the specified positions | 
| 557 | 
  | 
  if (haveMovingZMols()) | 
| 558 | 
  | 
    this->doHarmonic(); | 
| 559 | 
  | 
 | 
| 560 | 
  | 
  //write out forces and current positions of z-constraint molecules | 
| 561 | 
  | 
  if(info->getTime() >= curZconsTime){     | 
| 562 | 
< | 
   for(int i = 0; i < zconsMols.size(); i++){ | 
| 562 | 
> | 
   for(int i = 0; i < (int)(zconsMols.size()); i++){ | 
| 563 | 
  | 
      zconsMols[i]->getCOM(COM); | 
| 564 | 
  | 
    curZPos[i] = COM[whichDirection]; | 
| 565 | 
  | 
 | 
| 582 | 
  | 
#ifdef IS_MPI | 
| 583 | 
  | 
  if(worldRank == 0){ | 
| 584 | 
  | 
#endif | 
| 585 | 
< | 
    cout << "after calcForce, the COMVel of system is " << zSysCOMVel <<endl; | 
| 585 | 
> | 
     //cout << "after calcForce, the COMVel of system is " << zSysCOMVel <<endl; | 
| 586 | 
  | 
#ifdef IS_MPI | 
| 587 | 
  | 
  } | 
| 588 | 
  | 
#endif | 
| 615 | 
– | 
 | 
| 589 | 
  | 
} | 
| 590 | 
  | 
 | 
| 591 | 
  | 
 | 
| 654 | 
  | 
 | 
| 655 | 
  | 
  //zero out the velocities of center of mass of fixed z-constrained molecules | 
| 656 | 
  | 
   | 
| 657 | 
< | 
  for(int i = 0; i < zconsMols.size(); i++){ | 
| 657 | 
> | 
  for(int i = 0; i < (int)(zconsMols.size()); i++){ | 
| 658 | 
  | 
 | 
| 659 | 
  | 
    if (states[i] == zcsFixed){  | 
| 660 | 
  | 
 | 
| 661 | 
  | 
     zconsMols[i]->getCOMvel(COMvel);       | 
| 662 | 
< | 
    //cout << "before resetting " << indexOfZConsMols[i] <<"'s vz is " << COMvel[whichDirection] << endl; | 
| 662 | 
> | 
     //cout << "before resetting " << indexOfZConsMols[i] <<"'s vz is " << COMvel[whichDirection] << endl; | 
| 663 | 
  | 
 | 
| 664 | 
  | 
      fixedZAtoms = zconsMols[i]->getMyAtoms();  | 
| 665 | 
  | 
     | 
| 681 | 
  | 
#ifdef IS_MPI | 
| 682 | 
  | 
  if(worldRank == 0){ | 
| 683 | 
  | 
#endif | 
| 684 | 
< | 
    cout << "before resetting the COMVel of sytem is " << zSysCOMVel << endl;   | 
| 684 | 
> | 
     //cout << "before resetting the COMVel of sytem is " << zSysCOMVel << endl;   | 
| 685 | 
  | 
#ifdef IS_MPI | 
| 686 | 
  | 
  } | 
| 687 | 
  | 
#endif | 
| 695 | 
  | 
  MVzOfMovingMols_local = 0; | 
| 696 | 
  | 
  totalMassOfMovingZMols_local = 0; | 
| 697 | 
  | 
 | 
| 698 | 
< | 
  for(int i =0; i < unconsMols.size(); i++){ | 
| 698 | 
> | 
  for(int i =0; i < (int)(unconsMols.size()); i++){ | 
| 699 | 
  | 
    unconsMols[i]->getCOMvel(COMvel); | 
| 700 | 
  | 
    MVzOfMovingMols_local += massOfUnconsMols[i] * COMvel[whichDirection];       | 
| 701 | 
  | 
  }  | 
| 702 | 
  | 
 | 
| 703 | 
< | 
  for(int i = 0; i < zconsMols.size(); i++){ | 
| 703 | 
> | 
  for(int i = 0; i < (int)(zconsMols.size()); i++){ | 
| 704 | 
  | 
    if (states[i] == zcsMoving){ | 
| 705 | 
  | 
      zconsMols[i]->getCOMvel(COMvel); | 
| 706 | 
  | 
      MVzOfMovingMols_local += massOfZConsMols[i] * COMvel[whichDirection];    | 
| 722 | 
  | 
 | 
| 723 | 
  | 
  //modify the velocites of unconstrained molecules   | 
| 724 | 
  | 
  Atom** unconsAtoms; | 
| 725 | 
< | 
  for(int i = 0; i < unconsMols.size(); i++){ | 
| 725 | 
> | 
  for(int i = 0; i < (int)(unconsMols.size()); i++){ | 
| 726 | 
  | 
   | 
| 727 | 
  | 
    unconsAtoms = unconsMols[i]->getMyAtoms(); | 
| 728 | 
  | 
    for(int j = 0; j < unconsMols[i]->getNAtoms();j++){ | 
| 735 | 
  | 
 | 
| 736 | 
  | 
  //modify the velocities of moving z-constrained molecuels | 
| 737 | 
  | 
  Atom** movingZAtoms; | 
| 738 | 
< | 
  for(int i = 0; i < zconsMols.size(); i++){ | 
| 738 | 
> | 
  for(int i = 0; i < (int)(zconsMols.size()); i++){ | 
| 739 | 
  | 
 | 
| 740 | 
  | 
    if (states[i] ==zcsMoving){ | 
| 741 | 
  | 
    | 
| 755 | 
  | 
#ifdef IS_MPI | 
| 756 | 
  | 
  if(worldRank == 0){ | 
| 757 | 
  | 
#endif | 
| 758 | 
< | 
    cout << "after resetting the COMVel of moving molecules is " << zSysCOMVel << endl;   | 
| 758 | 
> | 
     //cout << "after resetting the COMVel of moving molecules is " << zSysCOMVel << endl;   | 
| 759 | 
  | 
#ifdef IS_MPI | 
| 760 | 
  | 
  } | 
| 761 | 
  | 
#endif | 
| 771 | 
  | 
  Atom** zconsAtoms; | 
| 772 | 
  | 
  double totalFZ;  | 
| 773 | 
  | 
  double totalFZ_local; | 
| 801 | 
– | 
  double COMvel[3];    | 
| 774 | 
  | 
  double COM[3]; | 
| 775 | 
  | 
  double force[3]; | 
| 776 | 
  | 
 | 
| 805 | 
– | 
 | 
| 806 | 
– | 
 | 
| 777 | 
  | 
  //constrain the molecules which do not reach the specified positions   | 
| 778 | 
  | 
     | 
| 779 | 
  | 
  //Zero Out the force of z-contrained molecules     | 
| 780 | 
  | 
  totalFZ_local = 0; | 
| 781 | 
  | 
 | 
| 782 | 
  | 
  //calculate the total z-contrained force of fixed z-contrained molecules | 
| 783 | 
+ | 
   | 
| 784 | 
+ | 
  //cout << "before zero out z-constraint force on fixed z-constraint molecuels " | 
| 785 | 
+ | 
  //       << "total force is " << calcTotalForce() << endl; | 
| 786 | 
  | 
 | 
| 787 | 
< | 
  for(int i = 0; i < zconsMols.size(); i++){ | 
| 787 | 
> | 
  for(int i = 0; i < (int)(zconsMols.size()); i++){ | 
| 788 | 
  | 
     | 
| 789 | 
  | 
    if (states[i] == zcsFixed){ | 
| 790 | 
  | 
       | 
| 798 | 
  | 
      }  | 
| 799 | 
  | 
      totalFZ_local += fz[i]; | 
| 800 | 
  | 
 | 
| 801 | 
< | 
//       cout << "Fixed Molecule\tindex: " << indexOfZConsMols[i]  | 
| 802 | 
< | 
//              <<"\tcurrent zpos: " << COM[whichDirection]  | 
| 803 | 
< | 
//              << "\tcurrent fz: " <<fz[i] << endl; | 
| 801 | 
> | 
      //cout << "Fixed Molecule\tindex: " << indexOfZConsMols[i]  | 
| 802 | 
> | 
      //      <<"\tcurrent zpos: " << COM[whichDirection]  | 
| 803 | 
> | 
      //      << "\tcurrent fz: " <<fz[i] << endl; | 
| 804 | 
  | 
 | 
| 805 | 
+ | 
 | 
| 806 | 
  | 
    } | 
| 807 | 
  | 
     | 
| 808 | 
  | 
  }  | 
| 820 | 
  | 
  force[1]= 0; | 
| 821 | 
  | 
  force[2]= 0; | 
| 822 | 
  | 
 | 
| 823 | 
< | 
  for(int i = 0; i < zconsMols.size(); i++){ | 
| 823 | 
> | 
  for(int i = 0; i < (int)(zconsMols.size()); i++){ | 
| 824 | 
  | 
 | 
| 825 | 
  | 
    if (states[i] == zcsFixed){   | 
| 826 | 
  | 
   | 
| 828 | 
  | 
      zconsAtoms = zconsMols[i]->getMyAtoms();   | 
| 829 | 
  | 
     | 
| 830 | 
  | 
      for(int j =0; j < nAtomOfCurZConsMol; j++) { | 
| 831 | 
< | 
      force[whichDirection] = -fz[i]/ nAtomOfCurZConsMol; | 
| 832 | 
< | 
        //force[whichDirection] = - forcePolicy->getZFOfFixedZMols(zconsMols[i], zconsAtoms[j], fz[i]); | 
| 831 | 
> | 
        //force[whichDirection] = -fz[i]/ nAtomOfCurZConsMol; | 
| 832 | 
> | 
        force[whichDirection] = - forcePolicy->getZFOfFixedZMols(zconsMols[i], zconsAtoms[j], fz[i]); | 
| 833 | 
  | 
        zconsAtoms[j]->addFrc(force); | 
| 834 | 
  | 
      } | 
| 835 | 
  | 
     | 
| 837 | 
  | 
   | 
| 838 | 
  | 
  }  | 
| 839 | 
  | 
 | 
| 840 | 
< | 
//   cout << "after zero out z-constraint force on fixed z-constraint molecuels " | 
| 841 | 
< | 
//        << "total force is " << calcTotalForce() << endl; | 
| 840 | 
> | 
  //cout << "after zero out z-constraint force on fixed z-constraint molecuels " | 
| 841 | 
> | 
  //      << "total force is " << calcTotalForce() << endl; | 
| 842 | 
> | 
    | 
| 843 | 
  | 
 | 
| 869 | 
– | 
  //calculate the number of atoms of moving z-constrained molecules | 
| 870 | 
– | 
  int nMovingZAtoms_local; | 
| 871 | 
– | 
  int nMovingZAtoms; | 
| 872 | 
– | 
   | 
| 873 | 
– | 
  nMovingZAtoms_local = 0; | 
| 874 | 
– | 
  for(int i = 0; i < zconsMols.size(); i++) | 
| 875 | 
– | 
    if(states[i] == zcsMoving) | 
| 876 | 
– | 
     nMovingZAtoms_local += zconsMols[i]->getNAtoms(); | 
| 877 | 
– | 
   | 
| 878 | 
– | 
#ifdef IS_MPI | 
| 879 | 
– | 
  MPI_Allreduce(&nMovingZAtoms_local, &nMovingZAtoms, 1,  | 
| 880 | 
– | 
                      MPI_DOUBLE,MPI_SUM, MPI_COMM_WORLD); | 
| 881 | 
– | 
#else | 
| 882 | 
– | 
  nMovingZAtoms = nMovingZAtoms_local; | 
| 883 | 
– | 
#endif | 
| 884 | 
– | 
 | 
| 844 | 
  | 
  force[0]= 0; | 
| 845 | 
  | 
  force[1]= 0; | 
| 846 | 
  | 
  force[2]= 0; | 
| 847 | 
  | 
 | 
| 848 | 
  | 
  //modify the forces of unconstrained molecules | 
| 849 | 
< | 
  for(int i = 0; i < unconsMols.size(); i++){ | 
| 849 | 
> | 
  for(int i = 0; i < (int)(unconsMols.size()); i++){ | 
| 850 | 
  | 
      | 
| 851 | 
  | 
     Atom** unconsAtoms = unconsMols[i]->getMyAtoms(); | 
| 852 | 
  | 
      | 
| 853 | 
  | 
     for(int j = 0; j < unconsMols[i]->getNAtoms(); j++){           | 
| 854 | 
< | 
       force[whichDirection] = totalFZ / (totNumOfUnconsAtoms + nMovingZAtoms); | 
| 855 | 
< | 
       //force[whichDirection] = forcePolicy->getZFOfMovingMols(unconsAtoms[j],totalFZ); | 
| 854 | 
> | 
       //force[whichDirection] = totalFZ / (totNumOfUnconsAtoms + nMovingZAtoms); | 
| 855 | 
> | 
       force[whichDirection] = forcePolicy->getZFOfMovingMols(unconsAtoms[j],totalFZ); | 
| 856 | 
  | 
       unconsAtoms[j]->addFrc(force); | 
| 857 | 
  | 
     } | 
| 858 | 
  | 
      | 
| 859 | 
  | 
  }       | 
| 860 | 
  | 
 | 
| 861 | 
  | 
 //modify the forces of moving z-constrained molecules | 
| 862 | 
< | 
  for(int i = 0; i < zconsMols.size(); i++) { | 
| 862 | 
> | 
  for(int i = 0; i < (int)(zconsMols.size()); i++) { | 
| 863 | 
  | 
    if (states[i] == zcsMoving){ | 
| 864 | 
  | 
     | 
| 865 | 
  | 
      Atom** movingZAtoms = zconsMols[i]->getMyAtoms();      | 
| 866 | 
  | 
 | 
| 867 | 
  | 
      for(int j = 0; j < zconsMols[i]->getNAtoms(); j++){ | 
| 868 | 
< | 
        force[whichDirection] = totalFZ / (totNumOfUnconsAtoms + nMovingZAtoms); | 
| 869 | 
< | 
        //force[whichDirection] = forcePolicy->getZFOfMovingMols(movingZAtoms[j],totalFZ); | 
| 868 | 
> | 
        //force[whichDirection] = totalFZ / (totNumOfUnconsAtoms + nMovingZAtoms); | 
| 869 | 
> | 
        force[whichDirection] = forcePolicy->getZFOfMovingMols(movingZAtoms[j],totalFZ); | 
| 870 | 
  | 
        movingZAtoms[j]->addFrc(force); | 
| 871 | 
  | 
      } | 
| 872 | 
  | 
    } | 
| 873 | 
  | 
  } | 
| 874 | 
+ | 
//  cout << "after substracting z-constraint force from moving molecuels " | 
| 875 | 
+ | 
//        << "total force is " << calcTotalForce()  << endl; | 
| 876 | 
  | 
 | 
| 916 | 
– | 
  //cout << "after substracting z-constraint force from moving molecuels " | 
| 917 | 
– | 
  //      << "total force is " << calcTotalForce()  << endl; | 
| 918 | 
– | 
 | 
| 877 | 
  | 
} | 
| 878 | 
  | 
 | 
| 879 | 
  | 
/** | 
| 896 | 
  | 
 | 
| 897 | 
  | 
  totalFZ_local = 0; | 
| 898 | 
  | 
 | 
| 899 | 
< | 
  for(int i = 0; i < zconsMols.size(); i++) { | 
| 899 | 
> | 
  for(int i = 0; i < (int)(zconsMols.size()); i++) { | 
| 900 | 
  | 
 | 
| 901 | 
  | 
    if (states[i] == zcsMoving){ | 
| 902 | 
  | 
      zconsMols[i]->getCOM(COM); | 
| 903 | 
< | 
      cout << "Moving Molecule\tindex: " << indexOfZConsMols[i]  | 
| 904 | 
< | 
           << "\tcurrent zpos: " << COM[whichDirection] << endl; | 
| 903 | 
> | 
//       cout << "Moving Molecule\tindex: " << indexOfZConsMols[i]  | 
| 904 | 
> | 
//         << "\tcurrent zpos: " << COM[whichDirection] << endl; | 
| 905 | 
> | 
 | 
| 906 | 
> | 
      diff = COM[whichDirection] -zPos[i]; | 
| 907 | 
  | 
     | 
| 948 | 
– | 
    diff = COM[whichDirection] -zPos[i]; | 
| 949 | 
– | 
     | 
| 908 | 
  | 
      harmonicU = 0.5 * kz[i] * diff * diff;   | 
| 909 | 
< | 
    info->lrPot += harmonicU; | 
| 909 | 
> | 
      info->lrPot += harmonicU; | 
| 910 | 
  | 
 | 
| 911 | 
  | 
      harmonicF =  - kz[i] * diff; | 
| 912 | 
  | 
      totalFZ_local += harmonicF; | 
| 916 | 
  | 
      Atom** movingZAtoms = zconsMols[i]->getMyAtoms();      | 
| 917 | 
  | 
 | 
| 918 | 
  | 
       for(int j = 0; j < zconsMols[i]->getNAtoms(); j++){           | 
| 919 | 
< | 
        force[whichDirection] = harmonicF / zconsMols[i]->getNAtoms(); | 
| 920 | 
< | 
         //force[whichDirection] = forcePolicy->getHFOfFixedZMols(zconsMols[i], movingZAtoms[j], harmonicF); | 
| 919 | 
> | 
        //force[whichDirection] = harmonicF / zconsMols[i]->getNAtoms(); | 
| 920 | 
> | 
        force[whichDirection] = forcePolicy->getHFOfFixedZMols(zconsMols[i], movingZAtoms[j], harmonicF); | 
| 921 | 
  | 
         movingZAtoms[j]->addFrc(force); | 
| 922 | 
  | 
       } | 
| 923 | 
  | 
    } | 
| 930 | 
  | 
  MPI_Allreduce(&totalFZ_local, &totalFZ, 1, MPI_DOUBLE,MPI_SUM, MPI_COMM_WORLD);   | 
| 931 | 
  | 
#endif | 
| 932 | 
  | 
 | 
| 933 | 
+ | 
  //cout << "before substracting harmonic force from moving molecuels " | 
| 934 | 
+ | 
  //      << "total force is " << calcTotalForce()  << endl; | 
| 935 | 
+ | 
 | 
| 936 | 
  | 
  force[0]= 0; | 
| 937 | 
  | 
  force[1]= 0; | 
| 938 | 
  | 
  force[2]= 0; | 
| 939 | 
  | 
 | 
| 940 | 
  | 
  //modify the forces of unconstrained molecules | 
| 941 | 
< | 
  for(int i = 0; i < unconsMols.size(); i++){ | 
| 941 | 
> | 
  for(int i = 0; i < (int)(unconsMols.size()); i++){ | 
| 942 | 
  | 
      | 
| 943 | 
  | 
     Atom** unconsAtoms = unconsMols[i]->getMyAtoms(); | 
| 944 | 
  | 
      | 
| 945 | 
  | 
     for(int j = 0; j < unconsMols[i]->getNAtoms(); j++){           | 
| 946 | 
< | 
       force[whichDirection] = - totalFZ /totNumOfUnconsAtoms; | 
| 947 | 
< | 
       //force[whichDirection] = - forcePolicy->getHFOfUnconsMols(unconsAtoms[j], totalFZ); | 
| 946 | 
> | 
       //force[whichDirection] = - totalFZ /totNumOfUnconsAtoms; | 
| 947 | 
> | 
       force[whichDirection] = - forcePolicy->getHFOfUnconsMols(unconsAtoms[j], totalFZ); | 
| 948 | 
  | 
       unconsAtoms[j]->addFrc(force);      | 
| 949 | 
  | 
     } | 
| 950 | 
  | 
  }    | 
| 951 | 
  | 
 | 
| 952 | 
+ | 
  //cout << "after substracting harmonic force from moving molecuels " | 
| 953 | 
+ | 
  //      << "total force is " << calcTotalForce()  << endl; | 
| 954 | 
+ | 
 | 
| 955 | 
  | 
} | 
| 956 | 
  | 
 | 
| 957 | 
  | 
/** | 
| 967 | 
  | 
   | 
| 968 | 
  | 
  changed_local = 0; | 
| 969 | 
  | 
   | 
| 970 | 
< | 
  for(int i =0; i < zconsMols.size(); i++){ | 
| 970 | 
> | 
  for(int i =0; i < (int)(zconsMols.size()); i++){ | 
| 971 | 
  | 
 | 
| 972 | 
  | 
    zconsMols[i]->getCOM(COM); | 
| 973 | 
  | 
    diff = fabs(COM[whichDirection] - zPos[i]);   | 
| 987 | 
  | 
#else | 
| 988 | 
  | 
  MPI_Allreduce(&changed_local, &changed, 1, MPI_INT,MPI_SUM, MPI_COMM_WORLD); | 
| 989 | 
  | 
#endif | 
| 990 | 
< | 
   | 
| 990 | 
> | 
 | 
| 991 | 
  | 
  return (changed > 0); | 
| 992 | 
+ | 
 | 
| 993 | 
  | 
} | 
| 994 | 
  | 
 | 
| 995 | 
  | 
template<typename T> bool ZConstraint<T>::haveFixedZMols(){ | 
| 999 | 
  | 
 | 
| 1000 | 
  | 
  havingFixed_local = 0; | 
| 1001 | 
  | 
 | 
| 1002 | 
< | 
  for(int i = 0; i < zconsMols.size(); i++) | 
| 1002 | 
> | 
  for(int i = 0; i < (int)(zconsMols.size()); i++) | 
| 1003 | 
  | 
    if (states[i] == zcsFixed){ | 
| 1004 | 
  | 
      havingFixed_local = 1; | 
| 1005 | 
  | 
    break; | 
| 1011 | 
  | 
  MPI_Allreduce(&havingFixed_local, &havingFixed, 1, MPI_INT,MPI_SUM, MPI_COMM_WORLD); | 
| 1012 | 
  | 
#endif | 
| 1013 | 
  | 
 | 
| 1014 | 
< | 
  return havingFixed > 0 ? true : false; | 
| 1014 | 
> | 
  return (havingFixed > 0); | 
| 1015 | 
  | 
} | 
| 1016 | 
  | 
 | 
| 1017 | 
  | 
 | 
| 1025 | 
  | 
 | 
| 1026 | 
  | 
  havingMoving_local = 0; | 
| 1027 | 
  | 
 | 
| 1028 | 
< | 
  for(int i = 0; i < zconsMols.size(); i++) | 
| 1028 | 
> | 
  for(int i = 0; i < (int)(zconsMols.size()); i++) | 
| 1029 | 
  | 
    if (states[i] == zcsMoving){ | 
| 1030 | 
  | 
      havingMoving_local = 1; | 
| 1031 | 
  | 
    break; | 
| 1037 | 
  | 
  MPI_Allreduce(&havingMoving_local, &havingMoving, 1, MPI_INT,MPI_SUM, MPI_COMM_WORLD); | 
| 1038 | 
  | 
#endif | 
| 1039 | 
  | 
 | 
| 1040 | 
< | 
  return havingMoving > 0 ? true : false; | 
| 1040 | 
> | 
  return (havingMoving > 0); | 
| 1041 | 
  | 
   | 
| 1042 | 
  | 
} | 
| 1043 | 
  | 
 | 
| 1107 | 
  | 
 | 
| 1108 | 
  | 
  massOfZCons_local = 0; | 
| 1109 | 
  | 
   | 
| 1110 | 
< | 
  for(int i = 0; i < massOfZConsMols.size(); i++){ | 
| 1110 | 
> | 
  for(int i = 0; i < (int)(massOfZConsMols.size()); i++){ | 
| 1111 | 
  | 
    massOfZCons_local += massOfZConsMols[i]; | 
| 1112 | 
  | 
  } | 
| 1113 | 
  | 
#ifndef IS_MPI | 
| 1158 | 
  | 
  int nMovingZAtoms; | 
| 1159 | 
  | 
   | 
| 1160 | 
  | 
  nMovingZAtoms_local = 0; | 
| 1161 | 
< | 
  for(int i = 0; i < (zconsIntegrator->zconsMols).size(); i++) | 
| 1161 | 
> | 
  for(int i = 0; i < (int)((zconsIntegrator->zconsMols).size()); i++) | 
| 1162 | 
  | 
    if((zconsIntegrator->states)[i] == (zconsIntegrator->zcsMoving)) | 
| 1163 | 
  | 
     nMovingZAtoms_local += (zconsIntegrator->zconsMols)[i]->getNAtoms(); | 
| 1164 | 
  | 
   | 
| 1168 | 
  | 
  nMovingZAtoms = nMovingZAtoms_local; | 
| 1169 | 
  | 
#endif | 
| 1170 | 
  | 
  totNumOfMovingAtoms = nMovingZAtoms + zconsIntegrator->totNumOfUnconsAtoms; | 
| 1206 | 
– | 
 | 
| 1207 | 
– | 
#ifdef IS_MPI | 
| 1208 | 
– | 
  if(worldRank == 0){ | 
| 1209 | 
– | 
#endif | 
| 1210 | 
– | 
    std::cerr << "\n" | 
| 1211 | 
– | 
              << "*******************************************\n" | 
| 1212 | 
– | 
              << " fiished Policy by numbr()\n" | 
| 1213 | 
– | 
              << "*******************************************\n" | 
| 1214 | 
– | 
              << "\n"; | 
| 1215 | 
– | 
#ifdef IS_MPI | 
| 1216 | 
– | 
  } | 
| 1217 | 
– | 
#endif | 
| 1171 | 
  | 
} | 
| 1172 | 
  | 
 | 
| 1173 | 
  | 
template<typename T>double ZConstraint<T>::PolicyByNumber::getZFOfFixedZMols(Molecule* mol, Atom* atom, double totalForce){ | 
| 1196 | 
  | 
  double massOfMovingZAtoms; | 
| 1197 | 
  | 
   | 
| 1198 | 
  | 
  massOfMovingZAtoms_local = 0; | 
| 1199 | 
< | 
  for(int i = 0; i < (zconsIntegrator->zconsMols).size(); i++) | 
| 1199 | 
> | 
  for(int i = 0; i < (int)((zconsIntegrator->zconsMols).size()); i++) | 
| 1200 | 
  | 
    if((zconsIntegrator->states)[i] == (zconsIntegrator->zcsMoving)) | 
| 1201 | 
  | 
     massOfMovingZAtoms_local += (zconsIntegrator->zconsMols)[i]->getTotalMass(); | 
| 1202 | 
  | 
   | 
| 1205 | 
  | 
#else | 
| 1206 | 
  | 
  massOfMovingZAtoms = massOfMovingZAtoms_local; | 
| 1207 | 
  | 
#endif | 
| 1208 | 
< | 
  totMassOfMovingAtoms = massOfMovingZAtoms_local + zconsIntegrator->totalMassOfUncons; | 
| 1208 | 
> | 
  totMassOfMovingAtoms = massOfMovingZAtoms + zconsIntegrator->totalMassOfUncons; | 
| 1209 | 
  | 
} | 
| 1210 | 
  | 
 | 
| 1211 | 
  | 
template<typename T> double ZConstraint<T>::PolicyByMass::getZFOfFixedZMols(Molecule* mol, Atom* atom, double totalForce){ |