64 |
|
Shape* shape; |
65 |
|
}; |
66 |
|
void registerHydrodynamicsModels(); |
67 |
< |
void calcHydrodynamicsProp(HydrodynamicsModel* model, Shape* shape,double viscosity, double temperature, std::ostream& os, const std::string& prefix); |
67 |
> |
void writeHydroProps(std::ostream& os); |
68 |
|
|
69 |
|
int main(int argc, char* argv[]){ |
70 |
|
//register force fields |
89 |
|
exit(1); |
90 |
|
} |
91 |
|
|
92 |
– |
mdFileName = dumpFileName; |
93 |
– |
mdFileName = mdFileName.substr(0, mdFileName.rfind(".")) + ".md"; |
94 |
– |
|
92 |
|
if (args_info.output_given){ |
93 |
|
prefix = args_info.output_arg; |
94 |
|
} else { |
95 |
< |
prefix = getPrefix(mdFileName); |
95 |
> |
prefix = getPrefix(dumpFileName); |
96 |
|
} |
97 |
|
std::string outputFilename = prefix + ".diff"; |
98 |
|
|
99 |
|
//parse md file and set up the system |
100 |
|
SimCreator creator; |
101 |
< |
SimInfo* info = creator.createSim(mdFileName, true); |
101 |
> |
SimInfo* info = creator.createSim(dumpFileName, true); |
102 |
|
|
103 |
|
SimInfo::MoleculeIterator mi; |
104 |
|
Molecule* mol; |
110 |
|
identMat(2,2) = 1.0; |
111 |
|
|
112 |
|
Globals* simParams = info->getSimParams(); |
113 |
< |
double temperature; |
114 |
< |
double viscosity; |
113 |
> |
RealType temperature; |
114 |
> |
RealType viscosity; |
115 |
|
|
116 |
|
if (simParams->haveViscosity()) { |
117 |
|
viscosity = simParams->getViscosity(); |
124 |
|
if (simParams->haveTargetTemp()) { |
125 |
|
temperature = simParams->getTargetTemp(); |
126 |
|
} else { |
127 |
< |
sprintf(painCave.errMsg, "target temperaturemust be set\n"); |
127 |
> |
sprintf(painCave.errMsg, "target temperature must be set\n"); |
128 |
|
painCave.isFatal = 1; |
129 |
|
simError(); |
130 |
|
} |
136 |
|
integrableObject = mol->nextIntegrableObject(ii)) { |
137 |
|
if (uniqueStuntDoubles.find(integrableObject->getType()) == uniqueStuntDoubles.end()) { |
138 |
|
|
142 |
– |
SDShape tmp; |
143 |
– |
tmp.shape = ShapeBuilder::createShape(integrableObject); |
144 |
– |
tmp.sd = integrableObject; |
145 |
– |
uniqueStuntDoubles.insert(std::map<std::string, SDShape>::value_type(integrableObject->getType(), tmp)); |
139 |
|
integrableObject->setPos(V3Zero); |
140 |
|
integrableObject->setA(identMat); |
141 |
|
if (integrableObject->isRigidBody()) { |
142 |
|
RigidBody* rb = static_cast<RigidBody*>(integrableObject); |
143 |
|
rb->updateAtoms(); |
144 |
|
} |
145 |
+ |
|
146 |
+ |
SDShape tmp; |
147 |
+ |
tmp.shape = ShapeBuilder::createShape(integrableObject); |
148 |
+ |
tmp.sd = integrableObject; |
149 |
+ |
uniqueStuntDoubles.insert(std::map<std::string, SDShape>::value_type(integrableObject->getType(), tmp)); |
150 |
+ |
|
151 |
|
} |
152 |
|
} |
153 |
|
} |
167 |
|
} else { |
168 |
|
model = new BeadModel(sd, info); |
169 |
|
} |
170 |
< |
calcHydrodynamicsProp(model, shape, viscosity, temperature, outputDiff, prefix); |
170 |
> |
|
171 |
> |
model->init(); |
172 |
> |
|
173 |
> |
std::ofstream ofs; |
174 |
> |
std::stringstream outputBeads; |
175 |
> |
outputBeads << prefix << "_" << model->getStuntDoubleName() << ".xyz"; |
176 |
> |
ofs.open(outputBeads.str().c_str()); |
177 |
> |
model->writeBeads(ofs); |
178 |
> |
ofs.close(); |
179 |
> |
|
180 |
> |
//if beads option is turned on, skip the calculation |
181 |
> |
if (!args_info.beads_flag) { |
182 |
> |
model->calcHydroProps(shape, viscosity, temperature); |
183 |
> |
model->writeHydroProps(outputDiff); |
184 |
> |
} |
185 |
> |
|
186 |
|
delete model; |
187 |
|
} |
188 |
|
|
198 |
|
HydrodynamicsModelFactory::getInstance()->registerHydrodynamicsModel(new HydrodynamicsModelBuilder<AnalyticalModel>("AnalyticalModel")); |
199 |
|
|
200 |
|
} |
187 |
– |
void calcHydrodynamicsProp(HydrodynamicsModel* model, Shape* shape,double viscosity, double temperature, std::ostream& os, const std::string& prefix) { |
188 |
– |
|
189 |
– |
shape->calcHydroProps(model, viscosity, temperature); |
190 |
– |
model->writeHydroProps(os); |
191 |
– |
std::ofstream ofs; |
192 |
– |
std::stringstream outputBeads; |
193 |
– |
outputBeads << prefix << "_" << model->getStuntDoubleName() << ".xyz"; |
194 |
– |
ofs.open(outputBeads.str().c_str()); |
195 |
– |
model->writeBeads(ofs); |
196 |
– |
ofs.close(); |
197 |
– |
} |