Changeset 13100 for fact/tools/pyscripts/pyfact/plotters.py

Timestamp:

03/14/12 14:59:57 (13 years ago)

Author:

neise

Message:

debugged handling of boolean masks in CamPlotter

File:

• fact/tools/pyscripts/pyfact/plotters.py (modified) (3 diffs)

fact/tools/pyscripts/pyfact/plotters.py

@@ -131 +131 @@
                 return
             
-            if mask.dtype == bool and data.ndim ==1 and len(mask)==1440:
+            elif mask.dtype == bool and data.ndim ==1 and len(mask)==1440:
                 length = mask.sum()
-                mask = np.where(mask)
+                mask = np.where(mask)[0]
                 mxe = np.empty( length )
                 mye = np.empty( length )
                 mdata = np.empty( length )
                 for i,chid in enumerate(mask):
+                    #print i , chid
                     mxe[i] = xe[chid]
                     mye[i] = ye[chid]
                     mdata[i] = data[chid]
+                #print 'mxe', mxe, 'len', len(mxe)
+                #print 'mye', mye, 'len', len(mye)
+                #print 'mxe', mdata, 'len', len(mdata)
 
                 self.ax.axis([-22,22,-22,22])
@@ -156 +160 @@
 
 
-            if mask.dtype == int  and data.ndim ==1:
+            elif mask.dtype == int  and data.ndim ==1:
                 length = len(mask)
                 mxe = np.empty( length )
@@ -179 +183 @@
                 plt.draw()
                 
+            else:
+                print "there is a mask, but I don't know how to treat it!!!"
+                sys.exit(-1)
         else: # i.e. when mask is None
         # handle 1D and 2D case differently