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