Changeset 1826 for trunk/python/interactivemask.py

Timestamp:

08/03/10 11:41:13 (14 years ago)

Author:

Malte Marquarding

Message:

Tidy up of imports (now imported from asap.). Also fixed some whitespace/tab issues

File:

• trunk/python/interactivemask.py (modified) (16 diffs)

trunk/python/interactivemask.py

@@ -1 @@
-from asap import rcParams
-from asap import _n_bools, mask_and, mask_or
+from asap.parameters import rcParams
+from asap.utils import _n_bools, mask_and, mask_or
 from asap.scantable import scantable
 
@@ -15 +15 @@
            my_mask.finish_selection(callback=func)
            mask=my_mask.get_mask()
-           
-        Modify mask region by selecting a region on a plot with mouse. 
+
+        Modify mask region by selecting a region on a plot with mouse.
         """
 
@@ -22 +22 @@
                 """
                 Create a interactive masking object.
-                Either or both 'plotter' or/and 'scan' should be defined. 
+                Either or both 'plotter' or/and 'scan' should be defined.
 
                 Parameters:
@@ -63 +63 @@
                 self.ydataold=None
                 self._polygons=[]
-        
+
 
         def set_basemask(self,masklist=[],invert=False):
                 """
                 Set initial channel mask.
-                
+
                 Parameters:
                     masklist:  [[min, max], [min2, max2], ...]
-                               A list of pairs of start/end points (inclusive) 
+                               A list of pairs of start/end points (inclusive)
                                specifying the regions to be masked
                     invert:    optional argument. If specified as True,
@@ -77 +77 @@
                                specified are excluded
                 You can reset the mask selection by running this method with
-                the default parameters. 
+                the default parameters.
                 """
                 # Verify input parameters
@@ -97 +97 @@
                 """
                 Inherit an event from the parent function.
-                
+
                 Parameters:
                     event: 'button_press_event' object to be inherited to
-                           start interactive region selection . 
+                           start interactive region selection .
                 """
                 from matplotlib.backend_bases import MouseEvent
-                if isinstance(event,MouseEvent) and event.name=='button_press_event': 
+                if isinstance(event,MouseEvent) and event.name=='button_press_event':
                         self.event=event
                 else:
                         msg="Invalid event."
-                        raise TypeError(msg)    
+                        raise TypeError(msg)
 
         def set_callback(self,callback):
                 """
-                Set callback function to run when finish_selection() is executed. 
+                Set callback function to run when finish_selection() is executed.
                     callback: The post processing function to run after
                               the mask selections are completed.
@@ -123 +123 @@
                 Do interactive mask selection.
                 Modify masks interactively by adding/deleting regions with
-                mouse drawing.(left-button: mask; right-button: UNmask) 
-                Note that the interactive region selection is available only 
-                when GUI plotter is active. 
+                mouse drawing.(left-button: mask; right-button: UNmask)
+                Note that the interactive region selection is available only
+                when GUI plotter is active.
 
                 Parameters:
                     once:     If specified as True, you can modify masks only
-                              once. Else if False, you can modify them repeatedly. 
+                              once. Else if False, you can modify them repeatedly.
                     showmask: If specified as True, the masked regions are plotted
                               on the plotter.
                               Note this parameter is valid only when once=True.
-                              Otherwise, maskes are forced to be plotted for reference. 
+                              Otherwise, maskes are forced to be plotted for reference.
                 """
                 # Return if GUI is not active
@@ -181 +181 @@
                 if self.event != None:
                         self._region_start(self.event)
-                else: 
+                else:
                         self.p._plotter.register('button_press',None)
                         self.p._plotter.register('button_press',self._region_start)
@@ -205 +205 @@
         def _region_draw(self,event):
                 sameaxes=(event.inaxes == self.rect['axes'])
-                if sameaxes: 
+                if sameaxes:
                         xnow=event.x
                         ynow=event.y
@@ -224 +224 @@
                 # Delete the rubber band
                 self.p._plotter.figmgr.toolbar.release(event)
-                
-                if event.inaxes == self.rect['axes']: 
+
+                if event.inaxes == self.rect['axes']:
                         xend=event.x
                         yend=event.y
@@ -235 +235 @@
                         xdataend=self.xdataold
                         ydataend=self.ydataold
-                        
+
                 self.rect['world'][2:4] = [xdataend, ydataend]
                 self.rect['pixel'][2:4] = [xend, yend]
@@ -300 +300 @@
                                 if len(ifs) > 1:
                                         for k in xrange(len(ifs)-1):
-                                                self._polygons.append(self.p._plotter.subplots[j]['axes'].axvspan(projs[k][i][0],projs[k][i][1],facecolor='#aaaaaa'))                   
+                                                self._polygons.append(self.p._plotter.subplots[j]['axes'].axvspan(projs[k][i][0],projs[k][i][1],facecolor='#aaaaaa'))
                                 self._polygons.append(self.p._plotter.subplots[j]['axes'].axvspan(msks[i][0],msks[i][1],facecolor='yellow'))
                 self.p._plotter.canvas.draw()
@@ -314 +314 @@
                               Specifying the callback function here will overwrite
                               the one set by set_callback(func)
-                
+
                 Note this function is automatically called at the end of
-                select_mask() if once=True. 
+                select_mask() if once=True.
                 """
                 if callback: self.callback=callback
@@ -324 +324 @@
                 if not self.newplot:
                         self.clear_polygon()
-                else: 
+                else:
                         self.p._plotter.unmap()
                         self.p._plotter = None
@@ -334 +334 @@
         def clear_polygon(self):
                 """
-                Erase masks plots from the plotter. 
+                Erase masks plots from the plotter.
                 """
                 if len(self._polygons)>0:
@@ -347 +347 @@
                 Get the interactively selected channel mask.
                 Returns:
-                    A list of channel mask. 
+                    A list of channel mask.
                 """
                 return self.mask
-
-