| 40 |
|
*/ |
| 41 |
|
|
| 42 |
|
#include <algorithm> |
| 43 |
+ |
#include <functional> |
| 44 |
|
#include "applications/staticProps/DensityPlot.hpp" |
| 45 |
|
#include "utils/simError.h" |
| 46 |
|
#include "io/DumpReader.hpp" |
| 49 |
|
namespace oopse { |
| 50 |
|
|
| 51 |
|
|
| 52 |
< |
DensityPlot::DensityPlot(SimInfo* info, const std::string& filename, const std::string& sele, const std::string& cmSele, double len, int nrbins) |
| 52 |
> |
DensityPlot::DensityPlot(SimInfo* info, const std::string& filename, const std::string& sele, const std::string& cmSele, RealType len, int nrbins) |
| 53 |
|
: StaticAnalyser(info, filename), selectionScript_(sele), evaluator_(info), seleMan_(info), |
| 54 |
|
cmSelectionScript_(cmSele), cmEvaluator_(info), cmSeleMan_(info), |
| 55 |
|
len_(len), nRBins_(nrbins), halfLen_(len/2) { |
| 84 |
|
|
| 85 |
|
DumpReader reader(info_, dumpFilename_); |
| 86 |
|
int nFrames = reader.getNFrames(); |
| 87 |
< |
int i; |
| 87 |
< |
for (i = 0; i < nFrames; i += step_) { |
| 87 |
> |
for (int i = 0; i < nFrames; i += step_) { |
| 88 |
|
reader.readFrame(i); |
| 89 |
|
currentSnapshot_ = info_->getSnapshotManager()->getCurrentSnapshot(); |
| 90 |
|
|
| 107 |
|
Vector3d origin = calcNewOrigin(); |
| 108 |
|
|
| 109 |
|
Mat3x3d hmat = currentSnapshot_->getHmat(); |
| 110 |
< |
double slabVolume = deltaR_ * hmat(0, 0) * hmat(1, 1); |
| 111 |
< |
|
| 112 |
< |
for (StuntDouble* sd = seleMan_.beginSelected(i); sd != NULL; sd = seleMan_.nextSelected(i)) { |
| 110 |
> |
RealType slabVolume = deltaR_ * hmat(0, 0) * hmat(1, 1); |
| 111 |
> |
int k; |
| 112 |
> |
for (StuntDouble* sd = seleMan_.beginSelected(k); sd != NULL; sd = seleMan_.nextSelected(k)) { |
| 113 |
|
|
| 114 |
|
|
| 115 |
|
if (!sd->isAtom()) { |
| 137 |
|
simError(); |
| 138 |
|
} |
| 139 |
|
|
| 140 |
< |
double nelectron = doubleData->getData(); |
| 140 |
> |
RealType nelectron = doubleData->getData(); |
| 141 |
|
|
| 142 |
|
data = atom->getAtomType()->getPropertyByName("LennardJones"); |
| 143 |
|
if (data == NULL) { |
| 157 |
|
} |
| 158 |
|
|
| 159 |
|
LJParam ljParam = ljData->getData(); |
| 160 |
< |
double sigma = ljParam.sigma * 0.5; |
| 161 |
< |
double sigma2 = sigma * sigma; |
| 160 |
> |
RealType sigma = ljParam.sigma * 0.5; |
| 161 |
> |
RealType sigma2 = sigma * sigma; |
| 162 |
|
|
| 163 |
|
Vector3d pos = sd->getPos() - origin; |
| 164 |
– |
/* |
| 165 |
– |
currentSnapshot_->wrapVector(pos); |
| 166 |
– |
double wrappedZdist = pos.z() + halfLen_; |
| 167 |
– |
if (wrappedZdist < 0.0 || wrappedZdist > len_) { |
| 168 |
– |
continue; |
| 169 |
– |
} |
| 170 |
– |
|
| 171 |
– |
int which =wrappedZdist / deltaR_; |
| 172 |
– |
density_[which] += nelectron; |
| 173 |
– |
*/ |
| 164 |
|
for (int j =0; j < nRBins_; ++j) { |
| 165 |
|
Vector3d tmp(pos); |
| 166 |
< |
double zdist =j * deltaR_ - halfLen_; |
| 166 |
> |
RealType zdist =j * deltaR_ - halfLen_; |
| 167 |
|
tmp[2] += zdist; |
| 168 |
< |
currentSnapshot_->wrapVector(tmp); |
| 168 |
> |
if (usePeriodicBoundaryConditions_) |
| 169 |
> |
currentSnapshot_->wrapVector(tmp); |
| 170 |
|
|
| 171 |
< |
double wrappedZdist = tmp.z() + halfLen_; |
| 171 |
> |
RealType wrappedZdist = tmp.z() + halfLen_; |
| 172 |
|
if (wrappedZdist < 0.0 || wrappedZdist > len_) { |
| 173 |
|
continue; |
| 174 |
|
} |
| 184 |
|
} |
| 185 |
|
|
| 186 |
|
int nProcessed = nFrames /step_; |
| 187 |
< |
std::transform(density_.begin(), density_.end(), density_.begin(), std::bind2nd(std::divides<double>(), nProcessed)); |
| 187 |
> |
std::transform(density_.begin(), density_.end(), density_.begin(), std::bind2nd(std::divides<RealType>(), nProcessed)); |
| 188 |
|
writeDensity(); |
| 189 |
|
|
| 190 |
|
|
| 195 |
|
|
| 196 |
|
int i; |
| 197 |
|
Vector3d newOrigin(0.0); |
| 198 |
< |
double totalMass = 0.0; |
| 198 |
> |
RealType totalMass = 0.0; |
| 199 |
|
for (StuntDouble* sd = seleMan_.beginSelected(i); sd != NULL; sd = seleMan_.nextSelected(i)) { |
| 200 |
< |
double mass = sd->getMass(); |
| 200 |
> |
RealType mass = sd->getMass(); |
| 201 |
|
totalMass += mass; |
| 202 |
|
newOrigin += sd->getPos() * mass; |
| 203 |
|
} |