44 |
|
#include "applications/staticProps/RadialDistrFunc.hpp" |
45 |
|
namespace oopse { |
46 |
|
|
47 |
< |
class GofRAngle : public RadialDistrFunc { |
47 |
> |
class GofRAngle : public RadialDistrFunc { |
48 |
|
|
49 |
< |
public: |
50 |
< |
GofRAngle(SimInfo* info, const std::string& filename, const std::string& sele1, const std::string& sele2); |
49 |
> |
public: |
50 |
> |
GofRAngle(SimInfo* info, const std::string& filename, const std::string& sele1, const std::string& sele2, RealType len, int nrbins, int nangleBins); |
51 |
|
|
52 |
< |
void setNRBins(int nbins) { |
53 |
< |
assert(nbins > 0); |
54 |
< |
nRBins_ = nbins; |
55 |
< |
deltaR_ = len_ / nRBins_; |
56 |
< |
for (int i = 0 ; i < nRBins_; ++i) { |
57 |
< |
histogram_[i].resize(nAngleBins_); |
58 |
< |
avgGofr_[i].resize(nAngleBins_); |
59 |
< |
} |
60 |
< |
} |
52 |
> |
int getNRBins() { |
53 |
> |
return nRBins_; |
54 |
> |
} |
55 |
|
|
56 |
< |
int getNRBins() { |
57 |
< |
return nRBins_; |
58 |
< |
} |
56 |
> |
RealType getLength() { |
57 |
> |
return len_; |
58 |
> |
} |
59 |
|
|
60 |
< |
void setLength(double len) { |
67 |
< |
len_ = len; |
68 |
< |
deltaR_ = len_ /nRBins_; |
69 |
< |
} |
70 |
< |
|
71 |
< |
double getLength() { |
72 |
< |
return len_; |
73 |
< |
} |
60 |
> |
int getNAngleBins() {return nAngleBins_;} |
61 |
|
|
62 |
< |
void setNAngleBins(int nbins) { |
76 |
< |
assert(nbins >0); |
77 |
< |
nAngleBins_ = nbins; |
78 |
< |
deltaCosAngle_ = 2.0 / nAngleBins_; |
79 |
< |
for (int i = 0 ; i < nRBins_; ++i) { |
80 |
< |
histogram_[i].resize(nAngleBins_); |
81 |
< |
avgGofr_[i].resize(nAngleBins_); |
82 |
< |
} |
83 |
< |
} |
84 |
< |
|
85 |
< |
int getNAngleBins() {return nAngleBins_;} |
86 |
< |
|
87 |
< |
private: |
62 |
> |
private: |
63 |
|
|
64 |
< |
virtual void preProcess(); |
65 |
< |
virtual void initalizeHistogram(); |
66 |
< |
virtual void collectHistogram(StuntDouble* sd1, StuntDouble* sd2); |
67 |
< |
virtual void processHistogram(); |
64 |
> |
virtual void preProcess(); |
65 |
> |
virtual void initalizeHistogram(); |
66 |
> |
virtual void collectHistogram(StuntDouble* sd1, StuntDouble* sd2); |
67 |
> |
virtual void processHistogram(); |
68 |
|
|
69 |
< |
virtual double evaluateAngle(StuntDouble* sd1, StuntDouble* sd2) = 0; |
69 |
> |
virtual RealType evaluateAngle(StuntDouble* sd1, StuntDouble* sd2) = 0; |
70 |
|
|
71 |
< |
virtual void writeRdf(); |
71 |
> |
virtual void writeRdf(); |
72 |
|
|
73 |
< |
double deltaCosAngle_; |
74 |
< |
int nAngleBins_; |
75 |
< |
double len_; |
76 |
< |
int nRBins_; |
77 |
< |
double deltaR_; |
73 |
> |
RealType deltaCosAngle_; |
74 |
> |
int nAngleBins_; |
75 |
> |
RealType len_; |
76 |
> |
int nRBins_; |
77 |
> |
RealType deltaR_; |
78 |
|
|
79 |
< |
std::vector<std::vector<int> > histogram_; |
80 |
< |
std::vector<std::vector<double> > avgGofr_; |
81 |
< |
int npairs_; |
82 |
< |
}; |
79 |
> |
std::vector<std::vector<int> > histogram_; |
80 |
> |
std::vector<std::vector<RealType> > avgGofr_; |
81 |
> |
int npairs_; |
82 |
> |
}; |
83 |
|
|
84 |
|
|
85 |
< |
class GofRTheta : public GofRAngle { |
86 |
< |
public: |
87 |
< |
GofRTheta(SimInfo* info, const std::string& filename, const std::string& sele1, const std::string& sele2) |
88 |
< |
: GofRAngle (info, filename, sele1, sele2) { |
89 |
< |
setOutputName(getPrefix(filename) + ".gt"); |
90 |
< |
} |
85 |
> |
class GofRTheta : public GofRAngle { |
86 |
> |
public: |
87 |
> |
GofRTheta(SimInfo* info, const std::string& filename, const std::string& sele1, const std::string& sele2, RealType len, int nrbins, int nangleBins) |
88 |
> |
: GofRAngle (info, filename, sele1, sele2, len, nrbins, nangleBins) { |
89 |
> |
setOutputName(getPrefix(filename) + ".gofrt"); |
90 |
> |
} |
91 |
|
|
92 |
< |
private: |
92 |
> |
private: |
93 |
|
|
94 |
< |
virtual double evaluateAngle(StuntDouble* sd1, StuntDouble* sd2); |
95 |
< |
}; |
94 |
> |
virtual RealType evaluateAngle(StuntDouble* sd1, StuntDouble* sd2); |
95 |
> |
}; |
96 |
|
|
97 |
|
|
98 |
< |
class GofROmega : public GofRAngle { |
99 |
< |
public: |
100 |
< |
GofROmega(SimInfo* info, const std::string& filename, const std::string& sele1, const std::string& sele2) |
101 |
< |
: GofRAngle (info, filename, sele1, sele2) { |
102 |
< |
setOutputName(getPrefix(filename) + ".go"); |
103 |
< |
} |
98 |
> |
class GofROmega : public GofRAngle { |
99 |
> |
public: |
100 |
> |
GofROmega(SimInfo* info, const std::string& filename, const std::string& sele1, const std::string& sele2, RealType len, int nrbins, int nangleBins) |
101 |
> |
: GofRAngle (info, filename, sele1, sele2, len, nrbins, nangleBins) { |
102 |
> |
setOutputName(getPrefix(filename) + ".gofro"); |
103 |
> |
} |
104 |
|
|
105 |
< |
private: |
106 |
< |
virtual double evaluateAngle(StuntDouble* sd1, StuntDouble* sd2); |
107 |
< |
}; |
105 |
> |
private: |
106 |
> |
virtual RealType evaluateAngle(StuntDouble* sd1, StuntDouble* sd2); |
107 |
> |
}; |
108 |
|
|
109 |
|
} |
110 |
|
#endif |