Changeset 67e885 for src/Parser/TremoloParser.cpp

Timestamp:

Jul 3, 2017, 3:07:28 PM (8 years ago)

Author:

Frederik Heber <frederik.heber@…>

Branches:

ForceAnnealing_oldresults, IndependentFragmentGrids_IntegrationTest

Children:

092ede

Parents:

ef8667

git-author:

Frederik Heber <frederik.heber@…> (06/17/17 23:09:44)

git-committer:

Frederik Heber <frederik.heber@…> (07/03/17 15:07:28)

Message:

TremoloParser::save() now supports multiple time steps.

• TESTS: Removed XFAIL from tests Tremolo with multiple time steps.

File:

• src/Parser/TremoloParser.cpp (modified) (4 diffs)

src/Parser/TremoloParser.cpp

@@ -146 +146 @@
   size_t timestep = 0;
   atom *addedatom = NULL;
-  std::vector<atom *> addedatoms;
-  std::vector<atom *>::iterator atomiter = addedatoms.begin();
+  std::list<atom *> addedatoms;
+  std::list<atom *>::iterator atomiter = addedatoms.begin();
   while (file->good()) {
     std::getline(*file, line, '\n');
@@ -226 +226 @@
   distributeContinuousIds(AtomList);
 
-  // store atomdata
-  save_AtomDataLine(file);
-
-  // store box
-  save_BoxLine(file);
-
-  // store particles
-  for (std::vector<const atom*>::const_iterator  atomIt = AtomList.begin();
-      atomIt != AtomList.end(); ++atomIt)
-    saveLine(file, *atomIt);
+  std::pair<size_t, size_t> minmax_trajectories =
+      getMinMaxTrajectories(AtomList);
+  LOG(2, "INFO: There are " << minmax_trajectories.second << " steps to save.");
+
+  for (size_t step = 0; (step < minmax_trajectories.second) || (step == 0); ++step) {
+    // store atomdata
+    save_AtomDataLine(file);
+
+    // store box only on first step
+    if (step == 0)
+      save_BoxLine(file);
+
+    // store particles
+    for (std::vector<const atom*>::const_iterator  atomIt = AtomList.begin();
+        atomIt != AtomList.end(); ++atomIt)
+      saveLine(file, *atomIt, step);
+  }
 }
 
@@ -495 +502 @@
 void FormatParser< tremolo >::saveLine(
     std::ostream* file,
-    const atom * const currentAtom)
+    const atom * const currentAtom,
+    const size_t _timestep)
 {
   TremoloKey::atomDataKey currentField;
 
-  LOG(4, "INFO: Saving atom " << *currentAtom << ", its father id is " << currentAtom->GetTrueFather()->getId());
+  LOG(4, "INFO: Saving atom " << *currentAtom << ", its father id is "
+      << currentAtom->GetTrueFather()->getId() << " at time step " << _timestep);
 
   for (usedFields_t::iterator it = usedFields_save.begin(); it != usedFields_save.end(); it++) {
@@ -506 +515 @@
       case TremoloKey::x :
         // for the moment, assume there are always three dimensions
-        LOG(3, "Writing for type " << knownKeyNames[currentField] << ": " << currentAtom->getPosition());
-        *file << currentAtom->at(0) << "\t";
-        *file << currentAtom->at(1) << "\t";
-        *file << currentAtom->at(2) << "\t";
+        LOG(3, "Writing for type " << knownKeyNames[currentField] << ": " << currentAtom->getPositionAtStep(_timestep));
+        *file << currentAtom->atStep(0, _timestep) << "\t";
+        *file << currentAtom->atStep(1, _timestep) << "\t";
+        *file << currentAtom->atStep(2, _timestep) << "\t";
         break;
       case TremoloKey::u :
         // for the moment, assume there are always three dimensions
-        LOG(3, "Writing for type " << knownKeyNames[currentField] << ": " << currentAtom->getAtomicVelocity());
-        *file << currentAtom->getAtomicVelocity()[0] << "\t";
-        *file << currentAtom->getAtomicVelocity()[1] << "\t";
-        *file << currentAtom->getAtomicVelocity()[2] << "\t";
+        LOG(3, "Writing for type " << knownKeyNames[currentField] << ": " << currentAtom->getAtomicVelocityAtStep(_timestep));
+        *file << currentAtom->getAtomicVelocityAtStep(_timestep)[0] << "\t";
+        *file << currentAtom->getAtomicVelocityAtStep(_timestep)[1] << "\t";
+        *file << currentAtom->getAtomicVelocityAtStep(_timestep)[2] << "\t";
         break;
       case TremoloKey::F :
         // for the moment, assume there are always three dimensions
-        LOG(3, "Writing for type " << knownKeyNames[currentField] << ": " << currentAtom->getAtomicForce());
-        *file << currentAtom->getAtomicForce()[0] << "\t";
-        *file << currentAtom->getAtomicForce()[1] << "\t";
-        *file << currentAtom->getAtomicForce()[2] << "\t";
+        LOG(3, "Writing for type " << knownKeyNames[currentField] << ": " << currentAtom->getAtomicForceAtStep(_timestep));
+        *file << currentAtom->getAtomicForceAtStep(_timestep)[0] << "\t";
+        *file << currentAtom->getAtomicForceAtStep(_timestep)[1] << "\t";
+        *file << currentAtom->getAtomicForceAtStep(_timestep)[2] << "\t";
         break;
       case TremoloKey::type :