ViewVC Help
View File | Revision Log | Show Annotations | View Changeset | Root Listing
root/group/trunk/OOPSE-4/src/applications/dump2Xyz/Dump2XYZ.cpp
(Generate patch)

Comparing trunk/OOPSE-4/src/applications/dump2Xyz/Dump2XYZ.cpp (file contents):
Revision 2224 by chrisfen, Thu May 12 19:43:48 2005 UTC vs.
Revision 3115 by chuckv, Tue Jan 16 22:17:27 2007 UTC

# Line 57 | Line 57
57   #include "selection/SelectionEvaluator.hpp"
58   #include "selection/SelectionManager.hpp"
59   #include "visitors/LipidTransVisitor.hpp"
60 + #include "visitors/AtomNameVisitor.hpp"
61  
62   using namespace oopse;
63  
# Line 67 | Line 68 | int main(int argc, char* argv[]){
68    
69    gengetopt_args_info args_info;
70    std::string dumpFileName;
70 –  std::string mdFileName;
71    std::string xyzFileName;
72    
73    //parse the command line option
# Line 83 | Line 83 | int main(int argc, char* argv[]){
83      exit(1);
84    }
85    
86 –  mdFileName = dumpFileName;
87 –  mdFileName = mdFileName.substr(0, mdFileName.rfind(".")) + ".md";
88 –
86    if (args_info.output_given){
87      xyzFileName = args_info.output_arg;
88    } else {
# Line 95 | Line 92 | int main(int argc, char* argv[]){
92    
93    //parse md file and set up the system
94    SimCreator creator;
95 <  SimInfo* info = creator.createSim(mdFileName, false);
95 >  SimInfo* info = creator.createSim(dumpFileName, false);
96    
97 <  
101 <  
97 >
98    //create visitor list
99    CompositeVisitor* compositeVisitor = new CompositeVisitor();
100      
# Line 114 | Line 110 | int main(int argc, char* argv[]){
110    
111    LinearAtomVisitor* linearVisitor = new LinearAtomVisitor(info);
112    compositeVisitor->addVisitor(linearVisitor, 750);
113 +  if (args_info.gb_given) {
114 +    linearVisitor->addGayBerneAtomType(args_info.gb_arg);
115 +  }
116    
117 +  GBLipidAtomVisitor* gbLipidVisitor = new GBLipidAtomVisitor(info);
118 +  compositeVisitor->addVisitor(gbLipidVisitor, 740);
119 +
120    //create default atom visitor
121    DefaultAtomVisitor* defaultAtomVisitor = new DefaultAtomVisitor(info);
122    compositeVisitor->addVisitor(defaultAtomVisitor, 700);
# Line 124 | Line 126 | int main(int argc, char* argv[]){
126      WaterTypeVisitor* waterTypeVisitor = new WaterTypeVisitor;
127      compositeVisitor->addVisitor(waterTypeVisitor, 600);
128    }
129 +
130 +  if (args_info.basetype_flag) {
131 +      AtomNameVisitor* atomNameVisitor = new AtomNameVisitor(info);
132 +      compositeVisitor->addVisitor(atomNameVisitor, 550);
133 +
134 +  }
135    
136    //create ZconsVisitor
137    if(args_info.zconstraint_flag){
# Line 139 | Line 147 | int main(int argc, char* argv[]){
147    
148    //create wrapping visitor
149    
150 <  if(args_info.periodicBox_flag){
151 <    WrappingVisitor* wrappingVisitor = new WrappingVisitor(info);
152 <    compositeVisitor->addVisitor(wrappingVisitor, 400);
153 <  }
150 >  //if(args_info.periodicBox_flag){
151 >  //  WrappingVisitor* wrappingVisitor = new WrappingVisitor(info);
152 >  //  compositeVisitor->addVisitor(wrappingVisitor, 400);
153 >  //}
154  
155    //create replicate visitor
156    if(args_info.repeatX_given > 0 || args_info.repeatY_given > 0 ||args_info.repeatY_given > 0){
# Line 162 | Line 170 | int main(int argc, char* argv[]){
170      
171    //create xyzVisitor
172    XYZVisitor* xyzVisitor;
173 +
174    if (args_info.selection_given) {
175      xyzVisitor = new XYZVisitor(info, args_info.selection_arg);
176    } else {
177      xyzVisitor = new XYZVisitor(info);
178    }
179 +
180 +  if(args_info.printPosOnly_flag){
181 +    bool posOnly = true;
182 +    xyzVisitor->setPosOnly(posOnly);
183 +  }
184 +
185    compositeVisitor->addVisitor(xyzVisitor, 200);
186    
187    std::cout << compositeVisitor->toString();
# Line 189 | Line 204 | int main(int argc, char* argv[]){
204    Molecule* mol;
205    StuntDouble* integrableObject;
206    RigidBody* rb;
207 +  Vector3d molCom;
208 +  Vector3d newMolCom;
209 +  Vector3d displacement;
210 +  Mat3x3d hmat;
211 +  Snapshot* currentSnapshot;
212        
213    for (int i = 0; i < nframes; i += args_info.frame_arg){
214      dumpReader->readFrame(i);
215 <    
215 >
216 >    //wrapping the molecule
217 >    if(args_info.periodicBox_flag) {
218 >      currentSnapshot = info->getSnapshotManager()->getCurrentSnapshot();    
219 >      for (mol = info->beginMolecule(miter); mol != NULL; mol = info->nextMolecule(miter)) {
220 >          molCom = mol->getCom();
221 >          newMolCom = molCom;
222 >          currentSnapshot->wrapVector(newMolCom);
223 >          displacement = newMolCom - molCom;
224 >        for (integrableObject = mol->beginIntegrableObject(iiter); integrableObject != NULL;
225 >             integrableObject = mol->nextIntegrableObject(iiter)) {  
226 >          integrableObject->setPos(integrableObject->getPos() + displacement);
227 >        }
228 >      }    
229 >    }
230      //update atoms of rigidbody
231      for (mol = info->beginMolecule(miter); mol != NULL; mol = info->nextMolecule(miter)) {
232        

Diff Legend

– Removed lines
+ Added lines
< Changed lines
> Changed lines