| 1 |
|
#include <iostream> |
| 2 |
|
#include <stdlib.h> |
| 3 |
|
#include <math.h> |
| 4 |
< |
|
| 4 |
> |
#include "Rattle.hpp" |
| 5 |
|
#ifdef IS_MPI |
| 6 |
|
#include "mpiSimulation.hpp" |
| 7 |
|
#include <unistd.h> |
| 33 |
|
nAtoms = info->n_atoms; |
| 34 |
|
integrableObjects = info->integrableObjects; |
| 35 |
|
|
| 36 |
+ |
rattle = new RattleFramework(info); |
| 37 |
+ |
|
| 38 |
+ |
if(rattle == NULL){ |
| 39 |
+ |
sprintf(painCave.errMsg, |
| 40 |
+ |
"Integrator::Intergrator() Error: Memory allocation error for RattleFramework" ); |
| 41 |
+ |
painCave.isFatal = 1; |
| 42 |
+ |
simError(); |
| 43 |
+ |
} |
| 44 |
+ |
|
| 45 |
+ |
/* |
| 46 |
|
// check for constraints |
| 47 |
|
|
| 48 |
|
constrainedA = NULL; |
| 55 |
|
nConstrained = 0; |
| 56 |
|
|
| 57 |
|
checkConstraints(); |
| 58 |
+ |
*/ |
| 59 |
|
} |
| 60 |
|
|
| 61 |
|
template<typename T> Integrator<T>::~Integrator(){ |
| 62 |
+ |
if (rattle != NULL) |
| 63 |
+ |
delete rattle; |
| 64 |
+ |
/* |
| 65 |
|
if (nConstrained){ |
| 66 |
|
delete[] constrainedA; |
| 67 |
|
delete[] constrainedB; |
| 70 |
|
delete[] moved; |
| 71 |
|
delete[] oldPos; |
| 72 |
|
} |
| 73 |
+ |
*/ |
| 74 |
|
} |
| 75 |
|
|
| 76 |
+ |
/* |
| 77 |
|
template<typename T> void Integrator<T>::checkConstraints(void){ |
| 78 |
|
isConstrained = 0; |
| 79 |
|
|
| 166 |
|
|
| 167 |
|
delete[] temp_con; |
| 168 |
|
} |
| 169 |
+ |
*/ |
| 170 |
|
|
| 154 |
– |
|
| 171 |
|
template<typename T> void Integrator<T>::integrate(void){ |
| 172 |
|
|
| 173 |
|
double runTime = info->run_time; |
| 176 |
|
double thermalTime = info->thermalTime; |
| 177 |
|
double resetTime = info->resetTime; |
| 178 |
|
|
| 179 |
< |
|
| 179 |
> |
double difference; |
| 180 |
|
double currSample; |
| 181 |
|
double currThermal; |
| 182 |
|
double currStatus; |
| 199 |
|
// that was drifting |
| 200 |
|
tStats->removeCOMdrift(); |
| 201 |
|
|
| 202 |
+ |
// initialize the retraints if necessary |
| 203 |
+ |
if (info->useSolidThermInt && !info->useLiquidThermInt) { |
| 204 |
+ |
myFF->initRestraints(); |
| 205 |
+ |
} |
| 206 |
+ |
|
| 207 |
|
// initialize the forces before the first step |
| 208 |
|
|
| 209 |
|
calcForce(1, 1); |
| 210 |
< |
|
| 211 |
< |
if (nConstrained){ |
| 212 |
< |
preMove(); |
| 213 |
< |
constrainA(); |
| 214 |
< |
calcForce(1, 1); |
| 215 |
< |
constrainB(); |
| 195 |
< |
} |
| 210 |
> |
|
| 211 |
> |
//execute constraint algorithm to make sure at the very beginning the system is constrained |
| 212 |
> |
rattle->doPreConstraint(); |
| 213 |
> |
rattle->doRattleA(); |
| 214 |
> |
calcForce(1, 1); |
| 215 |
> |
rattle->doRattleB(); |
| 216 |
|
|
| 217 |
|
if (info->setTemp){ |
| 218 |
|
thermalize(); |
| 235 |
|
#endif // is_mpi |
| 236 |
|
|
| 237 |
|
while (info->getTime() < runTime && !stopIntegrator()){ |
| 238 |
< |
if ((info->getTime() + dt) >= currStatus){ |
| 238 |
> |
difference = info->getTime() + dt - currStatus; |
| 239 |
> |
if (difference > 0 || fabs(difference) < 1e-4 ){ |
| 240 |
|
calcPot = 1; |
| 241 |
|
calcStress = 1; |
| 242 |
|
} |
| 291 |
|
#endif // is_mpi |
| 292 |
|
} |
| 293 |
|
|
| 294 |
+ |
// dump out a file containing the omega values for the final configuration |
| 295 |
+ |
if (info->useSolidThermInt && !info->useLiquidThermInt) |
| 296 |
+ |
myFF->dumpzAngle(); |
| 297 |
+ |
|
| 298 |
+ |
|
| 299 |
|
delete dumpOut; |
| 300 |
|
delete statOut; |
| 301 |
|
} |
| 308 |
|
startProfile(pro3); |
| 309 |
|
#endif //profile |
| 310 |
|
|
| 311 |
< |
preMove(); |
| 311 |
> |
//save old state (position, velocity etc) |
| 312 |
> |
rattle->doPreConstraint(); |
| 313 |
|
|
| 314 |
|
#ifdef PROFILE |
| 315 |
|
endProfile(pro3); |
| 331 |
|
MPIcheckPoint(); |
| 332 |
|
#endif // is_mpi |
| 333 |
|
|
| 307 |
– |
|
| 334 |
|
// calc forces |
| 309 |
– |
|
| 335 |
|
calcForce(calcPot, calcStress); |
| 336 |
|
|
| 337 |
|
#ifdef IS_MPI |
| 366 |
|
double Tb[3], ji[3]; |
| 367 |
|
double vel[3], pos[3], frc[3]; |
| 368 |
|
double mass; |
| 369 |
+ |
double omega; |
| 370 |
|
|
| 371 |
|
for (i = 0; i < integrableObjects.size() ; i++){ |
| 372 |
|
integrableObjects[i]->getVel(vel); |
| 373 |
|
integrableObjects[i]->getPos(pos); |
| 374 |
|
integrableObjects[i]->getFrc(frc); |
| 349 |
– |
|
| 350 |
– |
std::cerr << "i =\t" << i << "\t" << frc[0] << "\t" << frc[1]<< "\t" << frc[2] << "\n"; |
| 375 |
|
|
| 376 |
|
mass = integrableObjects[i]->getMass(); |
| 377 |
|
|
| 405 |
|
} |
| 406 |
|
} |
| 407 |
|
|
| 408 |
< |
if (nConstrained){ |
| 385 |
< |
constrainA(); |
| 386 |
< |
} |
| 408 |
> |
rattle->doRattleA(); |
| 409 |
|
} |
| 410 |
|
|
| 411 |
|
|
| 446 |
|
} |
| 447 |
|
} |
| 448 |
|
|
| 449 |
< |
if (nConstrained){ |
| 428 |
< |
constrainB(); |
| 429 |
< |
} |
| 449 |
> |
rattle->doRattleB(); |
| 450 |
|
} |
| 451 |
|
|
| 452 |
+ |
/* |
| 453 |
|
template<typename T> void Integrator<T>::preMove(void){ |
| 454 |
|
int i, j; |
| 455 |
|
double pos[3]; |
| 708 |
|
simError(); |
| 709 |
|
} |
| 710 |
|
} |
| 711 |
< |
|
| 711 |
> |
*/ |
| 712 |
|
template<typename T> void Integrator<T>::rotationPropagation |
| 713 |
|
( StuntDouble* sd, double ji[3] ){ |
| 714 |
|
|
| 747 |
|
|
| 748 |
|
// rotate about the z-axis |
| 749 |
|
angle = dt * ji[2] / I[2][2]; |
| 750 |
+ |
sd->addZangle(angle); |
| 751 |
|
this->rotate( 0, 1, angle, ji, A); |
| 752 |
|
|
| 753 |
|
// rotate about the y-axis |