| 16 |
|
gengetopt_args_info args_info; |
| 17 |
|
string dumpFileName; |
| 18 |
|
string bassFileName; |
| 19 |
+ |
char inFileName[2002]; |
| 20 |
|
string xyzFileName; |
| 21 |
|
SimInfo* info; |
| 22 |
|
SimSetup startMe; |
| 26 |
|
Molecule* mol; |
| 27 |
|
vector<StuntDouble*> integrableObjects; |
| 28 |
|
vector<StuntDouble*>::iterator iter; |
| 29 |
+ |
vector<RigidBody*> myRigidBodies; |
| 30 |
+ |
vector<RigidBody*>::iterator rbIter; |
| 31 |
|
|
| 32 |
|
CompositeVisitor* compositeVisitor; |
| 33 |
|
SSDAtomVisitor* ssdVisitor; |
| 34 |
|
DefaultAtomVisitor* defaultAtomVisitor; |
| 35 |
|
LipidHeadVisitor* lipidVisitor; |
| 36 |
+ |
RBCOMVisitor* rbCOMVisitor; |
| 37 |
|
ReplicateVisitor* replicateVisitor; |
| 38 |
|
WrappingVisitor* wrappingVisitor; |
| 39 |
|
IgnoreVisitor* ignoreVisitor; |
| 40 |
|
XYZVisitor* xyzVisitor; |
| 41 |
|
ZConsVisitor* zconsVisitor; |
| 42 |
|
PrepareVisitor* prepareVisitor; |
| 43 |
+ |
WaterTypeVisitor* waterTypeVisitor; |
| 44 |
|
|
| 45 |
|
//parse the command line option |
| 46 |
|
if (cmdline_parser (argc, argv, &args_info) != 0) |
| 70 |
|
info = new SimInfo(); |
| 71 |
|
startMe.setSimInfo(info ); |
| 72 |
|
|
| 73 |
< |
startMe.parseFile( bassFileName.c_str()); |
| 73 |
> |
strcpy(inFileName, bassFileName.c_str() ); |
| 74 |
> |
startMe.parseFile( inFileName ); |
| 75 |
|
|
| 76 |
|
startMe.createSim(); |
| 77 |
|
|
| 104 |
|
} |
| 105 |
|
|
| 106 |
|
//creat RigidBody Visitor |
| 107 |
< |
lipidVisitor = new LipidHeadVisitor(info); |
| 107 |
> |
if(args_info.rigidbody_flag){ |
| 108 |
> |
rbCOMVisitor = new RBCOMVisitor(info); |
| 109 |
> |
compositeVisitor->addVisitor(rbCOMVisitor, 900); |
| 110 |
> |
} |
| 111 |
> |
//creat RigidBody Visitor |
| 112 |
> |
//lipidVisitor = new LipidHeadVisitor(info); |
| 113 |
|
|
| 114 |
|
//adding names of lipid head group |
| 115 |
< |
lipidVisitor->addLipidHeadName("PC_RB_0"); |
| 116 |
< |
lipidVisitor->addLipidHeadName("PE_RB_0"); |
| 115 |
> |
//lipidVisitor->addLipidHeadName("PC_RB_0"); |
| 116 |
> |
//lipidVisitor->addLipidHeadName("PE_RB_0"); |
| 117 |
|
|
| 118 |
< |
compositeVisitor->addVisitor(lipidVisitor, 900); |
| 118 |
> |
//compositeVisitor->addVisitor(lipidVisitor, 900); |
| 119 |
|
|
| 120 |
|
//creat SSD atom visitor |
| 121 |
|
ssdVisitor = new SSDAtomVisitor(info); |
| 123 |
|
|
| 124 |
|
//creat default atom visitor |
| 125 |
|
defaultAtomVisitor = new DefaultAtomVisitor(info); |
| 126 |
< |
compositeVisitor->addVisitor(ssdVisitor, 700); |
| 126 |
> |
compositeVisitor->addVisitor(defaultAtomVisitor, 700); |
| 127 |
|
|
| 128 |
+ |
//creat waterType visitor |
| 129 |
+ |
if(args_info.watertype_flag){ |
| 130 |
+ |
waterTypeVisitor = new WaterTypeVisitor; |
| 131 |
+ |
compositeVisitor->addVisitor(waterTypeVisitor, 600); |
| 132 |
+ |
} |
| 133 |
+ |
|
| 134 |
|
//create ZconsVisitor |
| 135 |
< |
if(args_info.replace_given){ |
| 135 |
> |
if(args_info.zconstraint_flag){ |
| 136 |
|
|
| 137 |
|
zconsVisitor = new ZConsVisitor(info); |
| 138 |
|
|
| 139 |
|
if(zconsVisitor->haveZconsMol()) |
| 140 |
< |
compositeVisitor->addVisitor(zconsVisitor, 600); |
| 140 |
> |
compositeVisitor->addVisitor(zconsVisitor, 500); |
| 141 |
|
else |
| 142 |
|
delete zconsVisitor; |
| 143 |
|
} |
| 146 |
|
|
| 147 |
|
if(args_info.periodicBox_flag){ |
| 148 |
|
wrappingVisitor = new WrappingVisitor(info); |
| 149 |
< |
compositeVisitor->addVisitor(wrappingVisitor, 500); |
| 149 |
> |
compositeVisitor->addVisitor(wrappingVisitor, 400); |
| 150 |
|
} |
| 151 |
|
|
| 152 |
|
//creat replicate visitor |
| 153 |
|
if(args_info.repeatX_given > 0 || args_info.repeatY_given > 0 ||args_info.repeatY_given > 0){ |
| 154 |
|
IntVec3 replicateOpt(args_info.repeatX_arg, args_info.repeatY_arg, args_info.repeatZ_arg); |
| 155 |
|
replicateVisitor = new ReplicateVisitor(info, replicateOpt); |
| 156 |
< |
compositeVisitor->addVisitor(replicateVisitor, 400); |
| 156 |
> |
compositeVisitor->addVisitor(replicateVisitor, 300); |
| 157 |
|
} |
| 158 |
|
|
| 159 |
|
//creat xyzVisitor |
| 160 |
|
xyzVisitor = new XYZVisitor(info); |
| 161 |
< |
compositeVisitor->addVisitor(wrappingVisitor, 300); |
| 161 |
> |
compositeVisitor->addVisitor(xyzVisitor, 200); |
| 162 |
|
|
| 163 |
+ |
cout << compositeVisitor->toString(); |
| 164 |
|
|
| 165 |
|
//creat prepareVisitor |
| 166 |
|
prepareVisitor = new PrepareVisitor(); |
| 175 |
|
dumpReader->readFrame(info, i); |
| 176 |
|
|
| 177 |
|
mol = info->molecules; |
| 178 |
+ |
|
| 179 |
+ |
//update atoms of rigidbody |
| 180 |
+ |
for(int j = 0; j < info->n_mol; j++){ |
| 181 |
+ |
myRigidBodies = mol[j].getMyRigidBodies(); |
| 182 |
+ |
|
| 183 |
+ |
for(rbIter = myRigidBodies.begin(); rbIter != myRigidBodies.end(); ++rbIter) |
| 184 |
+ |
(*rbIter)->updateAtoms(); |
| 185 |
+ |
} |
| 186 |
+ |
|
| 187 |
|
|
| 188 |
|
//prepare visit |
| 189 |
|
for(int j = 0; j < info->n_mol; j++){ |
| 190 |
< |
integrableObjects = mol[i].getIntegrableObjects(); |
| 190 |
> |
integrableObjects = mol[j].getIntegrableObjects(); |
| 191 |
|
|
| 192 |
|
for(iter = integrableObjects.begin(); iter != integrableObjects.end(); ++iter) |
| 193 |
|
(*iter)->accept(prepareVisitor); |
| 198 |
|
|
| 199 |
|
//visit stuntdouble |
| 200 |
|
for(int j = 0; j < info->n_mol; j++){ |
| 201 |
< |
integrableObjects = mol[i].getIntegrableObjects(); |
| 201 |
> |
integrableObjects = mol[j].getIntegrableObjects(); |
| 202 |
|
|
| 203 |
|
for(iter = integrableObjects.begin(); iter != integrableObjects.end(); ++iter) |
| 204 |
|
(*iter)->accept(compositeVisitor); |