| [57dd40] | 1 | /*
 | 
|---|
 | 2 |  * UndoRedoHelpers.hpp
 | 
|---|
 | 3 |  *
 | 
|---|
 | 4 |  *  Created on: Apr 5, 2012
 | 
|---|
 | 5 |  *      Author: heber
 | 
|---|
 | 6 |  */
 | 
|---|
 | 7 | 
 | 
|---|
 | 8 | #ifndef UNDOREDOHELPERS_HPP_
 | 
|---|
 | 9 | #define UNDOREDOHELPERS_HPP_
 | 
|---|
 | 10 | 
 | 
|---|
 | 11 | 
 | 
|---|
 | 12 | // include config.h
 | 
|---|
 | 13 | #ifdef HAVE_CONFIG_H
 | 
|---|
 | 14 | #include <config.h>
 | 
|---|
 | 15 | #endif
 | 
|---|
 | 16 | 
 | 
|---|
 | 17 | #include <vector>
 | 
|---|
 | 18 | 
 | 
|---|
 | 19 | #include "Atom/AtomicInfo.hpp"
 | 
|---|
| [af9be32] | 20 | #include "Bond/BondInfo.hpp"
 | 
|---|
| [57dd40] | 21 | 
 | 
|---|
 | 22 | namespace MoleCuilder {
 | 
|---|
 | 23 | 
 | 
|---|
 | 24 |   /** Adds removed atoms back to the world whose state is stored as AtomicInfo.
 | 
|---|
 | 25 |    *
 | 
|---|
 | 26 |    * @param atoms vector of atomicInfo
 | 
|---|
 | 27 |    * @return restoral was successful, at least atom could not be restored.
 | 
|---|
 | 28 |    */
 | 
|---|
| [596cfa] | 29 |   bool AddAtomsFromAtomicInfo(const std::vector<AtomicInfo> &atoms);
 | 
|---|
 | 30 | 
 | 
|---|
 | 31 |   /** Adds removed molecules with their atoms back to the world.
 | 
|---|
 | 32 |    *
 | 
|---|
 | 33 |    * @param mol_atoms map of molecules with ids and their atoms as AtomicInfo
 | 
|---|
 | 34 |    * \return true - restoral was successful, at least one atom or molecule could not be restored
 | 
|---|
 | 35 |    */
 | 
|---|
 | 36 |   bool AddMoleculesFromAtomicInfo(std::map< moleculeId_t, std::vector<AtomicInfo> > &mol_atoms);
 | 
|---|
| [57dd40] | 37 | 
 | 
|---|
 | 38 |   /** Removes atoms whose state information is stored as AtomicInfo.
 | 
|---|
 | 39 |    *
 | 
|---|
 | 40 |    * @param atoms vector of atomicInfo
 | 
|---|
 | 41 |    */
 | 
|---|
 | 42 |   void RemoveAtomsFromAtomicInfo(std::vector<AtomicInfo> &atoms);
 | 
|---|
 | 43 | 
 | 
|---|
| [af9be32] | 44 |   /** Stores the required bond information in for all \a atoms in \a bonds.
 | 
|---|
 | 45 |    *
 | 
|---|
 | 46 |    * @param atoms atoms whose bonds to store
 | 
|---|
 | 47 |    * @param bonds vector with bond information on return
 | 
|---|
 | 48 |    */
 | 
|---|
 | 49 |   void StoreBondInformationFromAtoms(
 | 
|---|
 | 50 |       const std::vector<const atom*> &atoms,
 | 
|---|
 | 51 |       std::vector<BondInfo> &bonds);
 | 
|---|
 | 52 | 
 | 
|---|
 | 53 |   /** Recreates bonds from information stored in \a bonds.
 | 
|---|
 | 54 |    *
 | 
|---|
 | 55 |    * @param bonds bond state information
 | 
|---|
 | 56 |    * @return true - all bonds restored, false - at least one bond could not be restored
 | 
|---|
 | 57 |    */
 | 
|---|
 | 58 |   bool AddBondsFromBondInfo(const std::vector< BondInfo > &bonds);
 | 
|---|
 | 59 | 
 | 
|---|
| [57dd40] | 60 |   /** Sets atoms to state information stored as AtomicInfo.
 | 
|---|
 | 61 |    *
 | 
|---|
 | 62 |    * @param movedatoms vector of atomicInfo
 | 
|---|
 | 63 |    */
 | 
|---|
| [7e51e1] | 64 |   void SetAtomsFromAtomicInfo(const std::vector<AtomicInfo> &_movedatoms);
 | 
|---|
| [57dd40] | 65 | 
 | 
|---|
 | 66 |   /** Selects all atoms inside the given vector
 | 
|---|
 | 67 |    *
 | 
|---|
 | 68 |    * @param movedatoms vector of atomicInfo
 | 
|---|
 | 69 |    */
 | 
|---|
| [7e51e1] | 70 |   void SelectAtomsFromAtomicInfo(const std::vector<AtomicInfo> &_movedatoms);
 | 
|---|
 | 71 | 
 | 
|---|
 | 72 |   /** Helper function to allow setting arbitrary atom vectorial information via some
 | 
|---|
 | 73 |    * \a setter function.
 | 
|---|
 | 74 |    *
 | 
|---|
 | 75 |    * @param movedatoms atoms whose info to change
 | 
|---|
 | 76 |    * @param MovedToVector vector with old vectorial information
 | 
|---|
 | 77 |    */
 | 
|---|
 | 78 |   void ResetByFunction(
 | 
|---|
 | 79 |       const std::vector<AtomicInfo> &movedatoms,
 | 
|---|
 | 80 |       const std::vector<Vector> &MovedToVector,
 | 
|---|
 | 81 |       boost::function<void(atom *, const Vector&)> &setter);
 | 
|---|
| [57dd40] | 82 | 
 | 
|---|
 | 83 |   /** Sets the atoms whose id is stored in given AtomicInfo in \a movedatoms
 | 
|---|
 | 84 |    *  to position in \a MovedToVector.
 | 
|---|
 | 85 |    *
 | 
|---|
 | 86 |    * @param movedatoms atoms whose position to change
 | 
|---|
 | 87 |    * @param MovedToVector vector with old positions
 | 
|---|
 | 88 |    */
 | 
|---|
 | 89 |   void ResetAtomPosition(const std::vector<AtomicInfo> &movedatoms, const std::vector<Vector> &MovedToVector);
 | 
|---|
| [8ea3e7] | 90 | 
 | 
|---|
| [7e51e1] | 91 |   /** Sets the atoms whose id is stored in given AtomicInfo in \a movedatoms
 | 
|---|
 | 92 |    *  to position in \a MovedToVector.
 | 
|---|
 | 93 |    *
 | 
|---|
 | 94 |    * @param movedatoms atoms whose position to change
 | 
|---|
 | 95 |    * @param VelocityVector vector with old velocities
 | 
|---|
 | 96 |    */
 | 
|---|
 | 97 |   void ResetAtomVelocity(const std::vector<AtomicInfo> &movedatoms, const std::vector<Vector> &VelocityVector);
 | 
|---|
 | 98 | 
 | 
|---|
 | 99 |   /** Sets the atoms whose id is stored in given AtomicInfo in \a movedatoms
 | 
|---|
 | 100 |    *  to position in \a MovedToVector.
 | 
|---|
 | 101 |    *
 | 
|---|
 | 102 |    * @param movedatoms atoms whose position to change
 | 
|---|
 | 103 |    * @param ForceVector vector with old forces
 | 
|---|
 | 104 |    */
 | 
|---|
 | 105 |   void ResetAtomForce(const std::vector<AtomicInfo> &movedatoms, const std::vector<Vector> &ForceVector);
 | 
|---|
 | 106 | 
 | 
|---|
| [8ea3e7] | 107 |   /** Remove all molecules identified by their ids given in \a ids.
 | 
|---|
 | 108 |    *
 | 
|---|
 | 109 |    * @param ids vector of molecular ids to remove
 | 
|---|
 | 110 |    */
 | 
|---|
 | 111 |   void RemoveMoleculesWithAtomsByIds(const std::vector<moleculeId_t> &ids);
 | 
|---|
| [7e51e1] | 112 | 
 | 
|---|
 | 113 |   /** Removes the last time step for all \a movedatoms
 | 
|---|
 | 114 |    *
 | 
|---|
 | 115 |    * @param movedatoms atoms whose last step in time to remove
 | 
|---|
| [8cc22f] | 116 |    * @param _step which trajectory to remove
 | 
|---|
| [7e51e1] | 117 |    */
 | 
|---|
| [8cc22f] | 118 |   void removeLastStep(const std::vector<atomId_t> &movedatoms, const unsigned int _step);
 | 
|---|
| [7e51e1] | 119 | 
 | 
|---|
 | 120 |   /** Adds another time step to all \a movedatoms.
 | 
|---|
 | 121 |    *
 | 
|---|
 | 122 |    * Note that the time step is initialized to zero.
 | 
|---|
 | 123 |    *
 | 
|---|
 | 124 |    * @param _ids atoms whose last step in time to remove
 | 
|---|
| [8cc22f] | 125 |    * @param _step which trajectory to insert/assign
 | 
|---|
| [7e51e1] | 126 |    */
 | 
|---|
| [8cc22f] | 127 |   void addNewStep(const std::vector<atomId_t> &_ids, const unsigned int _step);
 | 
|---|
| [7e51e1] | 128 | 
 | 
|---|
 | 129 |   /** Adds another time step to all \a movedatoms.
 | 
|---|
 | 130 |    *
 | 
|---|
 | 131 |    * Note that the time step is initialized to zero. This gives you a chance
 | 
|---|
 | 132 |    * to set the time step and call setAtomsFromAtomicInfo() yourself.
 | 
|---|
 | 133 |    *
 | 
|---|
 | 134 |    * @param _movedatoms atoms whose last step in time to remove
 | 
|---|
| [8cc22f] | 135 |    * @param _step which trajectory to insert/assign
 | 
|---|
| [7e51e1] | 136 |    */
 | 
|---|
| [8cc22f] | 137 |   void addNewStep(const std::vector<AtomicInfo> &_movedatoms, const unsigned int _step);
 | 
|---|
| [7e51e1] | 138 | 
 | 
|---|
 | 139 |   /** Helper function to extract id information from vector of AtomicInfo.
 | 
|---|
 | 140 |    *
 | 
|---|
 | 141 |    * @param movedatoms atoms whose ids to extract
 | 
|---|
 | 142 |    */
 | 
|---|
 | 143 |   std::vector<atomId_t> getIdsFromAtomicInfo(const std::vector<AtomicInfo> &movedatoms);
 | 
|---|
| [57dd40] | 144 | }
 | 
|---|
 | 145 | 
 | 
|---|
 | 146 | 
 | 
|---|
 | 147 | 
 | 
|---|
 | 148 | #endif /* UNDOREDOHELPERS_HPP_ */
 | 
|---|