46 |
|
#include "math/SquareMatrix3.hpp" |
47 |
|
#include "math/DynamicRectMatrix.hpp" |
48 |
|
#include "primitives/Molecule.hpp" |
49 |
< |
#include "applications/hydrodynamics/StuntDoubleShape.hpp" |
50 |
< |
#include "utils/any.hpp" |
51 |
< |
namespace oopse { |
52 |
< |
struct HydrodynamicProps { |
53 |
< |
Vector3d diffCenter; |
54 |
< |
Mat3x3d transDiff; |
55 |
< |
Mat3x3d rotDiff; |
56 |
< |
Mat3x3d transRotDiff; |
57 |
< |
}; |
49 |
> |
#include "hydrodynamics/HydroProp.hpp" |
50 |
> |
#include "utils/OOPSEConstant.hpp" |
51 |
|
|
52 |
< |
struct BeadParam { |
52 |
> |
namespace oopse { |
53 |
> |
|
54 |
> |
struct BeadParam { |
55 |
|
std::string atomName; |
56 |
|
Vector3d pos; |
57 |
< |
double radius; |
58 |
< |
}; |
57 |
> |
RealType radius; |
58 |
> |
}; |
59 |
> |
|
60 |
> |
class Shape; |
61 |
> |
class Sphere; |
62 |
> |
class Ellipsoid; |
63 |
> |
class CompositeShape; |
64 |
> |
|
65 |
> |
class HydrodynamicsModel { |
66 |
> |
public: |
67 |
> |
HydrodynamicsModel(StuntDouble* sd, SimInfo* info) : sd_(sd), info_(info) {} |
68 |
> |
virtual ~HydrodynamicsModel() {} |
69 |
|
|
70 |
< |
typedef std::map<std::string, boost::any> DynamicProperty; |
71 |
< |
|
72 |
< |
class HydrodynamicsModel { |
73 |
< |
public: |
74 |
< |
HydrodynamicsModel(StuntDouble* sd, const DynamicProperty& extraParams); |
75 |
< |
bool calcHydrodyanmicsProps(); |
76 |
< |
|
77 |
< |
Vector3d getDiffCenter(); |
78 |
< |
Mat3x3d getTransDiff(); |
79 |
< |
Mat3x3d getRotDiff(); |
80 |
< |
Mat3x3d getTransRotDiff(); |
81 |
< |
void writeBeads(std::ostream& os); |
82 |
< |
void writeDiffCenterAndDiffTensor(std::ostream& os); |
83 |
< |
protected: |
84 |
< |
StuntDouble* sd_; |
85 |
< |
private: |
86 |
< |
virtual bool createBeads(std::vector<BeadParam>& beads) = 0; |
87 |
< |
|
88 |
< |
|
89 |
< |
HydrodynamicProps props_; |
85 |
< |
std::vector<BeadParam> beads_; |
86 |
< |
double viscosity_; |
87 |
< |
double temperature_; |
88 |
< |
|
89 |
< |
}; |
90 |
< |
|
91 |
< |
|
70 |
> |
virtual bool calcHydroProps(Shape* shape, RealType viscosity, RealType temperature); |
71 |
> |
|
72 |
> |
virtual void init() {}; |
73 |
> |
virtual void writeBeads(std::ostream& os) = 0; |
74 |
> |
void writeHydroProps(std::ostream& os); |
75 |
> |
HydroProp* getHydroPropsAtCR() {return cr_;} |
76 |
> |
HydroProp* getHydroPropsAtCD() {return cd_;} |
77 |
> |
|
78 |
> |
void setCR(HydroProp* cr) {cr_ = cr;} |
79 |
> |
void setCD(HydroProp* cd) {cd_ = cd;} |
80 |
> |
std::string getStuntDoubleName() { return sd_->getType();} |
81 |
> |
protected: |
82 |
> |
StuntDouble* sd_; |
83 |
> |
SimInfo* info_; |
84 |
> |
private: |
85 |
> |
HydroProp* cr_; |
86 |
> |
HydroProp* cd_; |
87 |
> |
std::vector<BeadParam> beads_; |
88 |
> |
}; |
89 |
> |
|
90 |
|
} |
91 |
|
|
92 |
|
#endif |