Changeset 67f102 for molecuilder/src/molecules.cpp

Timestamp:

Aug 6, 2008, 12:00:18 PM (17 years ago)

Author:

Frederik Heber <heber@…>

Children:

dc89d1

Parents:

8f8621

Message:

config::Save() and config::Load() parse MD steps into and out of molecule::Trajectories. Tested.

File:

• molecuilder/src/molecules.cpp (modified) (6 diffs)

molecuilder/src/molecules.cpp

@@ -47 +47 @@
   link(first,last);
   // other stuff 
+  MDSteps = 0;
   last_atom = 0; 
   elemente = teil;
@@ -983 +984 @@
   int i, Ion[2];
   double a, IonMass;
+  unsigned int size;
 
   CountElements();  // make sure ElementsInMolecule is up to date
@@ -1044 +1046 @@
             
             
+            // check size of vectors
+            size = Trajectories[walker].R.size(); 
+            if (size <= (unsigned int)(MDSteps)) {
+              cout << "Increasing size for trajectory array of " << *walker << " to " << (size+10) << "." << endl;
+              Trajectories[walker].R.resize(size+10);
+              Trajectories[walker].U.resize(size+10);
+              Trajectories[walker].F.resize(size+10);
+            }
             // add trajectory point
 
@@ -1294 +1304 @@
 bool molecule::Output(ofstream *out)
 {
-  element *runner = elemente->start;
+  element *runner;
   atom *walker = NULL;
   int ElementNo, AtomNo;
@@ -1304 +1314 @@
     *out << "#Ion_TypeNr._Nr.R[0]    R[1]    R[2]    MoveType (0 MoveIon, 1 FixedIon)" << endl;
     ElementNo = 0;
+    runner = elemente->start;
     while (runner->next != elemente->end) { // go through every element
                 runner = runner->next;
@@ -1314 +1325 @@
           if (walker->type == runner) { // if this atom fits to element
             AtomNo++;
-            walker->Output(ElementNo, AtomNo, out);
+            walker->Output(ElementNo, AtomNo, out); // removed due to trajectories
+          }
+        }
+      }
+    }
+    return true;
+  }
+};
+
+/** Prints molecule with all atomic trajectory positions to *out.
+ * \param *out output stream
+ */
+bool molecule::OutputTrajectories(ofstream *out)
+{
+  element *runner = NULL;
+  atom *walker = NULL;
+  int ElementNo, AtomNo;
+  CountElements();
+  
+  if (out == NULL) {
+    return false;
+  } else {
+    for (int step = 0; step < MDSteps; step++) {
+      if (step == 0) {
+        *out << "#Ion_TypeNr._Nr.R[0]    R[1]    R[2]    MoveType (0 MoveIon, 1 FixedIon)" << endl;
+      } else {
+        *out << "# ====== MD step " << step << " =========" << endl;
+      }
+      ElementNo = 0;
+      runner = elemente->start;
+      while (runner->next != elemente->end) { // go through every element
+        runner = runner->next;
+        if (ElementsInMolecule[runner->Z]) { // if this element got atoms
+          ElementNo++;
+          AtomNo = 0;
+          walker = start;
+          while (walker->next != end) { // go through every atom of this element
+            walker = walker->next;
+            if (walker->type == runner) { // if this atom fits to element
+              AtomNo++;
+              *out << "Ion_Type" << ElementNo << "_" << AtomNo << "\t"  << fixed << setprecision(9) << showpoint;
+              *out << Trajectories[walker].R.at(step).x[0] << "\t" << Trajectories[walker].R.at(step).x[1] << "\t" << Trajectories[walker].R.at(step).x[2];
+              *out << "\t" << walker->FixedIon;
+              if (Trajectories[walker].U.at(step).Norm() > MYEPSILON)
+                *out << "\t" << scientific << setprecision(6) << Trajectories[walker].U.at(step).x[0] << "\t" << Trajectories[walker].U.at(step).x[1] << "\t" << Trajectories[walker].U.at(step).x[2] << "\t";
+              if (Trajectories[walker].F.at(step).Norm() > MYEPSILON)
+                *out << "\t" << scientific << setprecision(6) << Trajectories[walker].F.at(step).x[0] << "\t" << Trajectories[walker].F.at(step).x[1] << "\t" << Trajectories[walker].F.at(step).x[2] << "\t";
+              *out << " # Number in molecule " << walker->nr << endl;
+            }
           }
         }