Download as
Browse Code »

Commit 46111ecbdc7714b1671d775ae3405953b431322d

Authored by Thiago Franco de Moraes
1 parent 6102130b
Exists in watershed

micro-optimization

Showing 1 changed file with 2 additions and 7 deletions   Show diff stats
invesalius/data/styles.py
  Diff comments   View file @46111ec
@@ -42,7 +42,7 @@ ORIENTATIONS = { @@ -42,7 +42,7 @@ ORIENTATIONS = {
42 42
43 BRUSH_FOREGROUND=1 43 BRUSH_FOREGROUND=1
44 BRUSH_BACKGROUND=2 44 BRUSH_BACKGROUND=2
45 -BRUSH_ERASE=3 45 +BRUSH_ERASE=0
46 46
47 def get_LUT_value(data, window, level): 47 def get_LUT_value(data, window, level):
48 return np.piecewise(data, 48 return np.piecewise(data,
@@ -1009,12 +1009,7 @@ class WaterShedInteractorStyle(DefaultInteractorStyle): @@ -1009,12 +1009,7 @@ class WaterShedInteractorStyle(DefaultInteractorStyle):
1009 1009
1010 # Checking if roi_i has at least one element. 1010 # Checking if roi_i has at least one element.
1011 if roi_m.size: 1011 if roi_m.size:
1012 - if operation == BRUSH_FOREGROUND:  
1013 - roi_m[index] = 1  
1014 - elif operation == BRUSH_BACKGROUND:  
1015 - roi_m[index] = 2  
1016 - elif operation == BRUSH_ERASE:  
1017 - roi_m[index] = 0 1012 + roi_m[index] = operation
1018 1013
1019 def expand_watershed(self, pubsub_evt): 1014 def expand_watershed(self, pubsub_evt):
1020 markers = self.matrix 1015 markers = self.matrix