Incorporation of phase constant to dihedral plot

martimunicoy · Sep 23, 2020 · 9c76113 · 9c76113
1 parent 66a2154
commit 9c76113
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/offpele/topology/topology.py b/offpele/topology/topology.py
@@ -476,7 +476,8 @@ def plot(self):
 
         x = unit.Quantity(np.arange(0, np.pi, 0.1), unit=unit.radians)
         pyplot.plot(x, self.constant * (1 + self.prefactor
-                                        * np.cos(self.periodicity * x)),
+                                        * np.cos(self.periodicity * x
+                                                 + self.phase)),
                     'r--')
 
         pyplot.show()