diff --git a/nbmolviz/viewers/viewer3d.py b/nbmolviz/viewers/viewer3d.py index 2ab587e..e1b8bc3 100644 --- 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): """