1 |
/* |
2 |
* Copyright (C) 2000-2004 Object Oriented Parallel Simulation Engine (OOPSE) project |
3 |
* |
4 |
* Contact: oopse@oopse.org |
5 |
* |
6 |
* This program is free software; you can redistribute it and/or |
7 |
* modify it under the terms of the GNU Lesser General Public License |
8 |
* as published by the Free Software Foundation; either version 2.1 |
9 |
* of the License, or (at your option) any later version. |
10 |
* All we ask is that proper credit is given for our work, which includes |
11 |
* - but is not limited to - adding the above copyright notice to the beginning |
12 |
* of your source code files, and to any copyright notice that you may distribute |
13 |
* with programs based on this work. |
14 |
* |
15 |
* This program is distributed in the hope that it will be useful, |
16 |
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
17 |
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
18 |
* GNU Lesser General Public License for more details. |
19 |
* |
20 |
* You should have received a copy of the GNU Lesser General Public License |
21 |
* along with this program; if not, write to the Free Software |
22 |
* Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. |
23 |
* |
24 |
*/ |
25 |
#ifndef CONSTRAINTS_ZCONSTRAINTFORCEMANAGER_HPP |
26 |
#define CONSTRAINTS_ZCONSTRAINTFORCEMANAGER_HPP |
27 |
#include <list> |
28 |
#include <string> |
29 |
#include <vector> |
30 |
#include "brains/ForceManager.hpp" |
31 |
#include "primitives/Molecule.hpp" |
32 |
namespace oopse { |
33 |
|
34 |
struct ZconstraintParam { |
35 |
double zTargetPos; /**< target zconstraint position */ |
36 |
double kz; /**< force constant */ |
37 |
double cantVel; /**< The velocity of cantilever */ |
38 |
}; |
39 |
|
40 |
struct ZconstraintMol { |
41 |
Molecule* mol; |
42 |
ZconstraintParam param; |
43 |
double fz; |
44 |
double zpos; |
45 |
double cantPos; /**< current position of cantilever */ |
46 |
double endFixingTime; |
47 |
}; |
48 |
|
49 |
class ZconstraintForceManager : public ForceManager { |
50 |
public: |
51 |
ZconstraintForceManager(SimInfo* info); |
52 |
|
53 |
virtual void calcForces(bool needPotential, bool needStress); |
54 |
|
55 |
double getZConsTime() { return zconsTime_; } |
56 |
std::string getZConsOutput() { return zconsOutput_; } |
57 |
|
58 |
void update(); |
59 |
|
60 |
private: |
61 |
|
62 |
bool isZMol(Molecule* mol); |
63 |
void thermalize(void); |
64 |
|
65 |
void zeroVelocity(); |
66 |
void doZconstraintForce(); |
67 |
void doHarmonic(); |
68 |
bool checkZConsState(); |
69 |
bool haveFixedZMols(); |
70 |
bool haveMovingMols(); |
71 |
void updateZPos(); |
72 |
void updateCantPos(); |
73 |
void calcTotalMassMovingZMols(); |
74 |
double getZTargetPos(int index); |
75 |
double getZFOfFixedZMols(Molecule* mol, StuntDouble* sd, double totalForce) ; |
76 |
double getZFOfMovingMols(StuntDouble* sd, double totalForce) ; |
77 |
double getHFOfFixedZMols(Molecule* mol, StuntDouble* sd, double totalForce); |
78 |
double getHFOfUnconsMols(StuntDouble* sd, double totalForce); |
79 |
|
80 |
std::list<ZconstraintMol> movingZMols_; /**< moving zconstraint molecules*/ |
81 |
std::list<ZconstraintMol> fixedZMols_; /**< fixed zconstraint molecules*/ |
82 |
std::vector<Molecule*> unzconsMols_; /**< free molecules*/ |
83 |
|
84 |
double zconsTime_; |
85 |
std::string zconsOutput_; |
86 |
double zconsTol_; |
87 |
bool usingSMD_; |
88 |
double zconsFixingTime_; |
89 |
double zconsGap_; |
90 |
bool usingZconsGap_; |
91 |
double dt_; |
92 |
|
93 |
const static int whichDirection = 2; |
94 |
|
95 |
std::map<int, ZconstraintParam> allZMolIndices_; |
96 |
|
97 |
Snapshot* currSnapshot_; |
98 |
double currZconsTime_; |
99 |
|
100 |
double totMassMovingZMols_; |
101 |
double totMassUnconsMols_; /**< mass of unconstraint molecules in the whole system (never change) */ |
102 |
|
103 |
ZConsWriter* fzOut; |
104 |
const static double infiniteTime = 1e31; |
105 |
}; |
106 |
|
107 |
} |
108 |
#endif |