58 |
|
#include "utils/simError.h" |
59 |
|
#include "utils/MemoryUtils.hpp" |
60 |
|
#include "utils/StringTokenizer.hpp" |
61 |
+ |
#include "brains/Thermo.hpp" |
62 |
|
|
63 |
|
#ifdef IS_MPI |
63 |
– |
|
64 |
|
#include <mpi.h> |
65 |
< |
#define TAKE_THIS_TAG_CHAR 0 |
66 |
< |
#define TAKE_THIS_TAG_INT 1 |
65 |
> |
#endif |
66 |
|
|
68 |
– |
#endif // is_mpi |
67 |
|
|
70 |
– |
|
68 |
|
namespace OpenMD { |
69 |
|
|
70 |
|
DumpReader::DumpReader(SimInfo* info, const std::string& filename) |
242 |
|
|
243 |
|
if (needCOMprops_) { |
244 |
|
Snapshot* s = info_->getSnapshotManager()->getCurrentSnapshot(); |
245 |
+ |
Thermo thermo(info_); |
246 |
|
Vector3d com; |
249 |
– |
Vector3d comvel; |
250 |
– |
Vector3d comw; |
251 |
– |
if (needPos_ && needVel_){ |
252 |
– |
info_->getComAll(com, comvel); |
253 |
– |
comw = info_->getAngularMomentum(); |
254 |
– |
}else{ |
255 |
– |
com = info_->getCom(); |
256 |
– |
comvel = 0.0; |
257 |
– |
comw = 0.0; |
258 |
– |
} |
259 |
– |
s->setCOMprops(com, comvel, comw); |
260 |
– |
} |
247 |
|
|
248 |
+ |
if (needPos_ && needVel_) { |
249 |
+ |
Vector3d comvel; |
250 |
+ |
Vector3d comw; |
251 |
+ |
thermo.getComAll(com, comvel); |
252 |
+ |
comw = thermo.getAngularMomentum(); |
253 |
+ |
} else { |
254 |
+ |
com = thermo.getCom(); |
255 |
+ |
} |
256 |
+ |
} |
257 |
|
} |
258 |
|
|
259 |
|
void DumpReader::readSet(int whichFrame) { |
709 |
|
hmat(2, 2) = tokenizer.nextTokenAsDouble(); |
710 |
|
s->setHmat(hmat); |
711 |
|
} else if (propertyName == "Thermostat") { |
712 |
< |
RealType chi = tokenizer.nextTokenAsDouble(); |
713 |
< |
RealType integralOfChiDt = tokenizer.nextTokenAsDouble(); |
714 |
< |
s->setChi(chi); |
715 |
< |
s->setIntegralOfChiDt(integralOfChiDt); |
712 |
> |
pair<RealType, RealType> thermostat; |
713 |
> |
thermostat.first = tokenizer.nextTokenAsDouble(); |
714 |
> |
thermostat.second = tokenizer.nextTokenAsDouble(); |
715 |
> |
s->setThermostat(thermostat); |
716 |
|
} else if (propertyName == "Barostat") { |
717 |
|
Mat3x3d eta; |
718 |
|
eta(0, 0) = tokenizer.nextTokenAsDouble(); |
724 |
|
eta(2, 0) = tokenizer.nextTokenAsDouble(); |
725 |
|
eta(2, 1) = tokenizer.nextTokenAsDouble(); |
726 |
|
eta(2, 2) = tokenizer.nextTokenAsDouble(); |
727 |
< |
s->setEta(eta); |
727 |
> |
s->setBarostat(eta); |
728 |
|
} else { |
729 |
|
sprintf(painCave.errMsg, |
730 |
|
"DumpReader Error: %s is an invalid property in <FrameData>\n", propertyName.c_str()); |