From c944ccf56d8a8462219089dd6251c62c8e807eab Mon Sep 17 00:00:00 2001 From: Thiago Franco de Moraes Date: Tue, 12 Oct 2021 19:21:10 -0300 Subject: [PATCH] Using get_LUT_value from imagedata_utils in styles.py --- invesalius/data/styles.py | 21 +-------------------- 1 file changed, 1 insertion(+), 20 deletions(-) diff --git a/invesalius/data/styles.py b/invesalius/data/styles.py index e4b5dd2..7a7dc66 100644 --- a/invesalius/data/styles.py +++ b/invesalius/data/styles.py @@ -45,6 +45,7 @@ import invesalius.utils as utils from invesalius.data.measures import (CircleDensityMeasure, MeasureData, PolygonDensityMeasure) +from invesalius.data.imagedata_utils import get_LUT_value, get_LUT_value_255 from invesalius_cy import floodfill # For tracts @@ -69,26 +70,6 @@ WATERSHED_OPERATIONS = {_("Erase"): BRUSH_ERASE, _("Foreground"): BRUSH_FOREGROUND, _("Background"): BRUSH_BACKGROUND,} -def get_LUT_value(data, window, level): - shape = data.shape - data_ = data.ravel() - data = np.piecewise(data_, - [data_ <= (level - 0.5 - (window-1)/2), - data_ > (level - 0.5 + (window-1)/2)], - [0, window, lambda data_: ((data_ - (level - 0.5))/(window-1) + 0.5)*(window)]) - data.shape = shape - return data - -def get_LUT_value_255(data, window, level): - shape = data.shape - data_ = data.ravel() - data = np.piecewise(data_, - [data_ <= (level - 0.5 - (window-1)/2), - data_ > (level - 0.5 + (window-1)/2)], - [0, 255, lambda data_: ((data_ - (level - 0.5))/(window-1) + 0.5)*(255)]) - data.shape = shape - return data - class BaseImageInteractorStyle(vtk.vtkInteractorStyleImage): def __init__(self, viewer): -- libgit2 0.21.2