1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38 package ffx.potential.openmm;
39
40 import ffx.openmm.Integrator;
41 import ffx.openmm.LangevinIntegrator;
42 import ffx.openmm.VerletIntegrator;
43 import ffx.potential.parameters.ForceField;
44 import org.apache.commons.configuration2.CompositeConfiguration;
45
46 import java.util.logging.Logger;
47
48 import static ffx.potential.ForceFieldEnergy.DEFAULT_CONSTRAINT_TOLERANCE;
49 import static java.lang.String.format;
50
51
52
53
54
55
56
57
58
59
60 public class OpenMMIntegrator {
61
62 private static final Logger logger = Logger.getLogger(OpenMMIntegrator.class.getName());
63
64
65
66
67 private final static double constraintTolerance = DEFAULT_CONSTRAINT_TOLERANCE;
68
69
70
71
72 private OpenMMIntegrator() {
73
74 }
75
76
77
78
79
80
81
82
83
84
85 public static Integrator createIntegrator(String name, double timeStep, double temperature, OpenMMSystem openMMSystem) {
86 switch (name) {
87 default -> {
88 return createVerletIntegrator(timeStep);
89 }
90 case "LANGEVIN" -> {
91 return createLangevinIntegrator(timeStep, temperature, openMMSystem.getForceField());
92 }
93 case "MTS" -> {
94 return createCustomMTSIntegrator(timeStep, openMMSystem);
95 }
96 case "LANGEVIN-MTS" -> {
97 return createCustomMTSLangevinIntegrator(timeStep, temperature, openMMSystem);
98 }
99 }
100 }
101
102
103
104
105
106
107
108
109 public static LangevinIntegrator createLangevinIntegrator(double dt, double temperature, ForceField forceField) {
110 CompositeConfiguration properties = forceField.getProperties();
111 int seed = 0;
112 if (properties.containsKey("integrator-seed")) {
113 seed = properties.getInt("integrator-seed", 0);
114 }
115 double frictionCoeff = forceField.getDouble("FRICTION_COEFF", 91.0);
116 LangevinIntegrator langevinIntegrator = new LangevinIntegrator(dt, temperature, frictionCoeff);
117 langevinIntegrator.setRandomNumberSeed(seed);
118 langevinIntegrator.setConstraintTolerance(constraintTolerance);
119 logger.info(" Langevin Integrator");
120 logger.info(format(" Target Temperature: %6.2f (K)", temperature));
121 logger.info(format(" Friction Coefficient: %6.2f (1/psec)", frictionCoeff));
122 logger.info(format(" Time step: %6.2f (fsec)", dt * 1000));
123 return langevinIntegrator;
124 }
125
126
127
128
129
130
131 public static CustomMTSIntegrator createCustomMTSIntegrator(double dt, OpenMMSystem openMMSystem) {
132 return new CustomMTSIntegrator(dt, constraintTolerance, openMMSystem.hasAmoebaCavitationForce());
133 }
134
135
136
137
138
139
140
141
142
143 public static CustomMTSLangevinIntegrator createCustomMTSLangevinIntegrator(double dt, double temperature, OpenMMSystem openMMSystem) {
144
145 ForceField forceField = openMMSystem.getForceField();
146 double frictionCoeff = forceField.getDouble("FRICTION_COEFF", 91.0);
147 CustomMTSLangevinIntegrator customMTSLangevinIntegrator =
148 new CustomMTSLangevinIntegrator(dt, temperature, frictionCoeff, openMMSystem.hasAmoebaCavitationForce());
149 return customMTSLangevinIntegrator;
150 }
151
152
153
154
155
156
157 public static VerletIntegrator createVerletIntegrator(double dt) {
158 VerletIntegrator verletIntegrator = new VerletIntegrator(dt);
159 verletIntegrator.setConstraintTolerance(constraintTolerance);
160 logger.info(" Verlet Integrator");
161 logger.info(format(" Time step: %6.2f (fsec)", dt * 1000));
162 return verletIntegrator;
163 }
164
165 }