| 68 |
|
|
| 69 |
|
gengetopt_args_info args_info; |
| 70 |
|
std::string dumpFileName; |
| 71 |
– |
std::string mdFileName; |
| 71 |
|
std::string xyzFileName; |
| 72 |
|
|
| 73 |
|
//parse the command line option |
| 83 |
|
exit(1); |
| 84 |
|
} |
| 85 |
|
|
| 87 |
– |
mdFileName = dumpFileName; |
| 88 |
– |
mdFileName = mdFileName.substr(0, mdFileName.rfind(".")) + ".md"; |
| 89 |
– |
|
| 86 |
|
if (args_info.output_given){ |
| 87 |
|
xyzFileName = args_info.output_arg; |
| 88 |
|
} else { |
| 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 |
< |
|
| 102 |
< |
|
| 97 |
> |
|
| 98 |
|
//create visitor list |
| 99 |
|
CompositeVisitor* compositeVisitor = new CompositeVisitor(); |
| 100 |
|
|
| 117 |
|
GBLipidAtomVisitor* gbLipidVisitor = new GBLipidAtomVisitor(info); |
| 118 |
|
compositeVisitor->addVisitor(gbLipidVisitor, 740); |
| 119 |
|
|
| 120 |
+ |
Ring5gbAtomVisitor* ring5Visitor = new Ring5gbAtomVisitor(info); |
| 121 |
+ |
compositeVisitor->addVisitor(ring5Visitor, 730); |
| 122 |
+ |
|
| 123 |
+ |
HeadAtomVisitor* headVisitor = new HeadAtomVisitor(info); |
| 124 |
+ |
compositeVisitor->addVisitor(headVisitor, 720); |
| 125 |
+ |
|
| 126 |
|
//create default atom visitor |
| 127 |
|
DefaultAtomVisitor* defaultAtomVisitor = new DefaultAtomVisitor(info); |
| 128 |
|
compositeVisitor->addVisitor(defaultAtomVisitor, 700); |
| 129 |
|
|
| 130 |
< |
//create waterType visitor |
| 131 |
< |
if(args_info.watertype_flag){ |
| 132 |
< |
WaterTypeVisitor* waterTypeVisitor = new WaterTypeVisitor; |
| 133 |
< |
compositeVisitor->addVisitor(waterTypeVisitor, 600); |
| 130 |
> |
// if we gave the -w option, we want to skip the waters: |
| 131 |
> |
std::cerr << "-w flag was set to:" << args_info.water_given << "\n"; |
| 132 |
> |
if (!args_info.water_given) { |
| 133 |
> |
//create waterType visitor |
| 134 |
> |
if(args_info.watertype_flag){ |
| 135 |
> |
WaterTypeVisitor* waterTypeVisitor = new WaterTypeVisitor; |
| 136 |
> |
compositeVisitor->addVisitor(waterTypeVisitor, 600); |
| 137 |
> |
} |
| 138 |
|
} |
| 139 |
|
|
| 140 |
|
if (args_info.basetype_flag) { |
| 180 |
|
|
| 181 |
|
//create xyzVisitor |
| 182 |
|
XYZVisitor* xyzVisitor; |
| 183 |
+ |
|
| 184 |
|
if (args_info.selection_given) { |
| 185 |
|
xyzVisitor = new XYZVisitor(info, args_info.selection_arg); |
| 186 |
|
} else { |
| 187 |
|
xyzVisitor = new XYZVisitor(info); |
| 188 |
|
} |
| 189 |
+ |
|
| 190 |
+ |
if(args_info.printPosOnly_flag){ |
| 191 |
+ |
bool posOnly = true; |
| 192 |
+ |
xyzVisitor->setPosOnly(posOnly); |
| 193 |
+ |
} |
| 194 |
+ |
|
| 195 |
|
compositeVisitor->addVisitor(xyzVisitor, 200); |
| 196 |
|
|
| 197 |
|
std::cout << compositeVisitor->toString(); |
| 204 |
|
int nframes = dumpReader->getNFrames(); |
| 205 |
|
|
| 206 |
|
|
| 207 |
< |
std::ofstream xyzStream; |
| 196 |
< |
xyzStream .open(xyzFileName.c_str()); |
| 207 |
> |
std::ofstream xyzStream(xyzFileName.c_str()); |
| 208 |
|
|
| 198 |
– |
|
| 209 |
|
SimInfo::MoleculeIterator miter; |
| 210 |
|
Molecule::IntegrableObjectIterator iiter; |
| 211 |
|
Molecule::RigidBodyIterator rbIter; |
| 268 |
|
xyzVisitor->clear(); |
| 269 |
|
|
| 270 |
|
}//end for (int i = 0; i < nframes; i += args_info.frame_arg) |
| 271 |
< |
|
| 271 |
> |
|
| 272 |
|
xyzStream.close(); |
| 263 |
– |
|
| 273 |
|
delete prepareVisitor; |
| 274 |
|
delete compositeVisitor; |
| 275 |
|
delete info; |
| 267 |
– |
|
| 276 |
|
} |