Commit c944ccf56d8a8462219089dd6251c62c8e807eab

Authored by Thiago Franco de Moraes
1 parent 7c2bd0b8
Exists in master

Using get_LUT_value from imagedata_utils in styles.py

Showing 1 changed file with 1 additions and 20 deletions   Show diff stats
invesalius/data/styles.py
@@ -45,6 +45,7 @@ import invesalius.utils as utils @@ -45,6 +45,7 @@ import invesalius.utils as utils
45 from invesalius.data.measures import (CircleDensityMeasure, MeasureData, 45 from invesalius.data.measures import (CircleDensityMeasure, MeasureData,
46 PolygonDensityMeasure) 46 PolygonDensityMeasure)
47 47
  48 +from invesalius.data.imagedata_utils import get_LUT_value, get_LUT_value_255
48 from invesalius_cy import floodfill 49 from invesalius_cy import floodfill
49 50
50 # For tracts 51 # For tracts
@@ -69,26 +70,6 @@ WATERSHED_OPERATIONS = {_("Erase"): BRUSH_ERASE, @@ -69,26 +70,6 @@ WATERSHED_OPERATIONS = {_("Erase"): BRUSH_ERASE,
69 _("Foreground"): BRUSH_FOREGROUND, 70 _("Foreground"): BRUSH_FOREGROUND,
70 _("Background"): BRUSH_BACKGROUND,} 71 _("Background"): BRUSH_BACKGROUND,}
71 72
72 -def get_LUT_value(data, window, level):  
73 - shape = data.shape  
74 - data_ = data.ravel()  
75 - data = np.piecewise(data_,  
76 - [data_ <= (level - 0.5 - (window-1)/2),  
77 - data_ > (level - 0.5 + (window-1)/2)],  
78 - [0, window, lambda data_: ((data_ - (level - 0.5))/(window-1) + 0.5)*(window)])  
79 - data.shape = shape  
80 - return data  
81 -  
82 -def get_LUT_value_255(data, window, level):  
83 - shape = data.shape  
84 - data_ = data.ravel()  
85 - data = np.piecewise(data_,  
86 - [data_ <= (level - 0.5 - (window-1)/2),  
87 - data_ > (level - 0.5 + (window-1)/2)],  
88 - [0, 255, lambda data_: ((data_ - (level - 0.5))/(window-1) + 0.5)*(255)])  
89 - data.shape = shape  
90 - return data  
91 -  
92 73
93 class BaseImageInteractorStyle(vtk.vtkInteractorStyleImage): 74 class BaseImageInteractorStyle(vtk.vtkInteractorStyleImage):
94 def __init__(self, viewer): 75 def __init__(self, viewer):