View Javadoc
1   // ******************************************************************************
2   //
3   // Title:       Force Field X.
4   // Description: Force Field X - Software for Molecular Biophysics.
5   // Copyright:   Copyright (c) Michael J. Schnieders 2001-2024.
6   //
7   // This file is part of Force Field X.
8   //
9   // Force Field X is free software; you can redistribute it and/or modify it
10  // under the terms of the GNU General Public License version 3 as published by
11  // the Free Software Foundation.
12  //
13  // Force Field X is distributed in the hope that it will be useful, but WITHOUT
14  // ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
15  // FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
16  // details.
17  //
18  // You should have received a copy of the GNU General Public License along with
19  // Force Field X; if not, write to the Free Software Foundation, Inc., 59 Temple
20  // Place, Suite 330, Boston, MA 02111-1307 USA
21  //
22  // Linking this library statically or dynamically with other modules is making a
23  // combined work based on this library. Thus, the terms and conditions of the
24  // GNU General Public License cover the whole combination.
25  //
26  // As a special exception, the copyright holders of this library give you
27  // permission to link this library with independent modules to produce an
28  // executable, regardless of the license terms of these independent modules, and
29  // to copy and distribute the resulting executable under terms of your choice,
30  // provided that you also meet, for each linked independent module, the terms
31  // and conditions of the license of that module. An independent module is a
32  // module which is not derived from or based on this library. If you modify this
33  // library, you may extend this exception to your version of the library, but
34  // you are not obligated to do so. If you do not wish to do so, delete this
35  // exception statement from your version.
36  //
37  // ******************************************************************************
38  package ffx.algorithms.dynamics.integrators;
39  
40  import ffx.potential.SystemState;
41  import ffx.numerics.Constraint;
42  import ffx.numerics.Potential;
43  import ffx.potential.ForceFieldEnergy;
44  import java.util.ArrayList;
45  import java.util.Collection;
46  import java.util.List;
47  import java.util.logging.Logger;
48  import java.util.stream.Collectors;
49  
50  /**
51   * The Integrator class is responsible for propagation of degrees of freedom through time.
52   * Implementations must define their behavior at pre-force and post-force evaluation time points.
53   *
54   * @author Michael J. Schnieders
55   * @since 1.0
56   */
57  public abstract class Integrator {
58  
59    private static final Logger logger = Logger.getLogger(Integrator.class.getName());
60    /**
61     * Numerical tolerance (as a fraction of bond length) permitted for numerical solutions to
62     * constraints.
63     */
64    protected final double constraintTolerance = ForceFieldEnergy.DEFAULT_CONSTRAINT_TOLERANCE;
65  
66    /**
67     * The MDState class contains the current state of the Molecular Dynamics simulation.
68     */
69    protected final SystemState state;
70    /** Time step (psec). */
71    protected double dt;
72    /** Any geometric constraints to apply during integration. */
73    protected List<Constraint> constraints = new ArrayList<>();
74    /** If there are constraints present. */
75    protected boolean useConstraints = false;
76    /** Half the time step (psec). */
77    double dt_2;
78  
79    /**
80     * Constructor for Integrator.
81     *
82     * @param state The MD state to operate on.
83     */
84    public Integrator(SystemState state) {
85      this.state = state;
86      dt = 1.0e-3;
87      dt_2 = dt / 2.0;
88    }
89  
90  
91    /**
92     * Parse an integrator String into an instance of the IntegratorEnum enum.
93     *
94     * @param str Integrator string.
95     * @return Integrator enum.
96     */
97    public static IntegratorEnum parseIntegrator(String str) {
98      try {
99        String integrator = str.toUpperCase().replaceAll("\\s+", "");
100       integrator = integrator.replaceAll("-", "_");
101       return IntegratorEnum.valueOf(integrator);
102     } catch (Exception e) {
103       logger.info(String.format(" Could not parse %s as an integrator; defaulting to Verlet.", str));
104       return IntegratorEnum.VERLET;
105     }
106   }
107 
108   /**
109    * Adds a set of Constraints that this Integrator must respect.
110    *
111    * @param addedConstraints Constraints to add.
112    */
113   public void addConstraints(List<Constraint> addedConstraints) {
114     constraints.addAll(addedConstraints);
115     useConstraints = true;
116   }
117 
118   /** Copy acceleration to previous acceleration. */
119   public void copyAccelerationToPrevious() {
120     state.copyAccelerationsToPrevious();
121   }
122 
123   /**
124    * Returns a copy of the list of Constraints.
125    *
126    * @return All Constraints this Integrator respects.
127    */
128   public List<Constraint> getConstraints() {
129     return new ArrayList<>(constraints);
130   }
131 
132   /**
133    * Get the time step.
134    *
135    * @return the time step (psec).
136    */
137   public double getTimeStep() {
138     return dt;
139   }
140 
141   /**
142    * Set the time step.
143    *
144    * @param dt the time step (psec).
145    */
146   public abstract void setTimeStep(double dt);
147 
148   /**
149    * Integrator post-force evaluation operation.
150    *
151    * @param gradient the gradient for the post-force operation.
152    */
153   public abstract void postForce(double[] gradient);
154 
155   /**
156    * Integrator pre-force evaluation operation.
157    *
158    * @param potential the Potential this integrator operates on.
159    */
160   public abstract void preForce(Potential potential);
161 
162 }