(fit.py) Quick kludgy patch for crash of issue 0028

[hooke.git] / viewer.py

diff --git a/viewer.py b/viewer.py
index bae4355ae3b88b3da24ee81f94d7665922769e6c..42f4b6d38c3ae62da449931394fa297302e704e7 100644 (file)
--- a/viewer.py
+++ b/viewer.py
@@ -10,25 +10,30 @@ This program is released under the GNU General Public License version 2.
 
 
 import libviewer as lview
+import libinput as linput
 
 class viewerCommands:
        
     def _plug_init(self):
-       self.viewerlist=[]
+        self.viewerlist=[]
+        #we keep a list of different viewers so it's possible to retrieve different data
+        #or process the same data differently
                
 
     def do_vwnew(self,args):
-       self.viewerlist.append(lview.Ascii(self.outlet))
-       dt=raw_input('What type of data will this viewer handle? (force/distance/all)')
-       print dt
-       self.viewerlist[-1].setdtype(dt)
+        #creates a new viewer
+        self.viewerlist.append(lview.Ascii(self.outlet))
+        dt=linput.safeinput('What type of data will this viewer handle? (force/distance/all)',['force', 'distance', 'all']) 
+                    #TODO update types, make a list somewhere?
+        print dt
+        self.viewerlist[-1].setdtype(dt)
 
 
     def do_vwaction(self,args):
-       '''
-       triggers default action of viewer number n (default 0)
-       '''
+        '''
+        triggers default action of viewer number n (default 0)
+        '''
 
-       if len(args)==0:
-               args=0
-       self.viewerlist[int(args)].action()
+        if len(args)==0:
+            args=0
+        self.viewerlist[int(args)].action()
\ No newline at end of file