| 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 |
– |
} |