From bcd7b0aab32b20e5469893a0c0fa17bb83339284 Mon Sep 17 00:00:00 2001 From: Marco Biasini <marco.biasini@unibas.ch> Date: Tue, 15 Nov 2011 11:29:40 +0100 Subject: [PATCH] added support for selection in TrajWidget --- modules/gui/pymod/traj.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/modules/gui/pymod/traj.py b/modules/gui/pymod/traj.py index 6df252b35..1fcb1c779 100644 --- a/modules/gui/pymod/traj.py +++ b/modules/gui/pymod/traj.py @@ -21,12 +21,13 @@ from PyQt4.QtGui import * from ost import * class TrajWidget(QWidget): - def __init__(self, traj=None, render_mode=gfx.SIMPLE, parent=None): + def __init__(self, traj=None, render_mode=gfx.SIMPLE, sel='', parent=None): QWidget.__init__(self, parent, Qt.Tool) self.render_mode=render_mode vb=QVBoxLayout() hb=QHBoxLayout() hb2=QHBoxLayout() + self.selection=sel self.callback=None self._slider=QSlider(self) self._slider.setOrientation(Qt.Horizontal) @@ -116,7 +117,8 @@ class TrajWidget(QWidget): self._traj=traj if self._traj: ev=traj.GetEntity() - self.gfx_entity=gfx.Entity("mol",self.render_mode, ev) + self.gfx_entity=gfx.Entity("mol",self.render_mode, + ev.Select(self.selection)) gfx.Scene().Add(self.gfx_entity) gfx.Scene().CenterOn(self.gfx_entity) # enable the blur effect -- GitLab