A function was added to view normal mode displacement vectors from QM…

… packages.
Autodesk · Jul 27, 2017 · cc2f9a5 · cc2f9a5
1 parent a28be53
commit cc2f9a5
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/nbmolviz/viewers/viewer3d.py b/nbmolviz/viewers/viewer3d.py
@@ -282,6 +282,10 @@ def draw_forces(self, **kwargs):
     def draw_momenta(self, **kwargs):
         return self.draw_atom_vectors(self.mol.momenta, **kwargs)
 
+    def draw_normalmodes(self, normode_num, **kwargs):
+        normodes = self.mol.properties['normalmodes_displacements'][normode_num]
+        return self.draw_atom_vectors(normodes , **kwargs)
+
     def highlight_atoms(self, atoms=None):
         """