Skip to content

Commit

Permalink
Created toml file
Browse files Browse the repository at this point in the history 
formatted and type hinted style.py, and presets.py
  • Loading branch information
@omar-abdelgawad
omar-abdelgawad committed May 9, 2024
1 parent db5c4ba commit 6bead8e
Show file tree
Hide file tree
Showing 3 changed files with 221 additions and 141 deletions.
228 changes: 119 additions & 109 deletions invesalius/presets.py
View file
Original file line number Diff line number Diff line change
@@ -1,10 +1,10 @@
#--------------------------------------------------------------------------
# --------------------------------------------------------------------------
# Software: InVesalius - Software de Reconstrucao 3D de Imagens Medicas
# Copyright: (C) 2001 Centro de Pesquisas Renato Archer
# Homepage: http://www.softwarepublico.gov.br
# Contact: invesalius@cti.gov.br
# License: GNU - GPL 2 (LICENSE.txt/LICENCA.txt)
#--------------------------------------------------------------------------
# --------------------------------------------------------------------------
# Este programa e software livre; voce pode redistribui-lo e/ou
# modifica-lo sob os termos da Licenca Publica Geral GNU, conforme
# publicada pela Free Software Foundation; de acordo com a versao 2
Expand All @@ -15,108 +15,115 @@
# COMERCIALIZACAO ou de ADEQUACAO A QUALQUER PROPOSITO EM
# PARTICULAR. Consulte a Licenca Publica Geral GNU para obter mais
# detalhes.
#--------------------------------------------------------------------------
# --------------------------------------------------------------------------
import glob
import os
import plistlib

import invesalius.constants as const

from invesalius.pubsub import pub as Publisher
import typing

from invesalius import inv_paths
from invesalius.i18n import tr
from invesalius.pubsub import pub as Publisher
from invesalius.utils import TwoWaysDictionary
from invesalius.i18n import tr as _

class Presets():
if typing.TYPE_CHECKING:
from pathlib import Path


class Presets:
def __init__(self):
self.thresh_ct = TwoWaysDictionary({
_("Bone"):(226,3071),
_("Soft Tissue"):(-700,225),
_("Enamel (Adult)"):(1553,2850),
_("Enamel (Child)"):(2042,3071),
_("Compact Bone (Adult)"):(662,1988),
_("Compact Bone (Child)"):(586,2198),
_("Spongial Bone (Adult)"):(148,661),
_("Spongial Bone (Child)"):(156,585),
_("Muscle Tissue (Adult)"):(-5,135),
_("Muscle Tissue (Child)"):(-25,139),
_("Fat Tissue (Adult)"):(-205,-51),
_("Fat Tissue (Child)"):(-212,-72),
_("Skin Tissue (Adult)"):(-718,-177),
_("Skin Tissue (Child)"):(-766,-202),
_("Custom"):(0, 0)
})

self.thresh_mri = TwoWaysDictionary({
_("Bone"):(1250,4095),
_("Soft Tissue"):(324,1249),
_("Enamel (Adult)"):(2577,3874),
_("Enamel (Child)"):(3066,4095),
_("Compact Bone (Adult)"):(1686,3012),
_("Compact Bone (Child)"):(1610,3222),
_("Spongial Bone (Adult)"):(1172,1685),
_("Spongial Bone (Child)"):(1180,1609),
_("Muscle Tissue (Adult)"):(1019,1159),
_("Muscle Tissue (Child)"):(999,1163),
_("Fat Tissue (Adult)"):(819,973),
_("Fat Tissue (Child)"):(812,952),
_("Skin Tissue (Adult)"):(306,847),
_("Skin Tissue (Child)"):(258,822),
_("Custom"):(0, 0)
})
self.thresh_ct = TwoWaysDictionary(
{
tr("Bone"): (226, 3071),
tr("Soft Tissue"): (-700, 225),
tr("Enamel (Adult)"): (1553, 2850),
tr("Enamel (Child)"): (2042, 3071),
tr("Compact Bone (Adult)"): (662, 1988),
tr("Compact Bone (Child)"): (586, 2198),
tr("Spongial Bone (Adult)"): (148, 661),
tr("Spongial Bone (Child)"): (156, 585),
tr("Muscle Tissue (Adult)"): (-5, 135),
tr("Muscle Tissue (Child)"): (-25, 139),
tr("Fat Tissue (Adult)"): (-205, -51),
tr("Fat Tissue (Child)"): (-212, -72),
tr("Skin Tissue (Adult)"): (-718, -177),
tr("Skin Tissue (Child)"): (-766, -202),
tr("Custom"): (0, 0),
}
)

self.thresh_mri = TwoWaysDictionary(
{
tr("Bone"): (1250, 4095),
tr("Soft Tissue"): (324, 1249),
tr("Enamel (Adult)"): (2577, 3874),
tr("Enamel (Child)"): (3066, 4095),
tr("Compact Bone (Adult)"): (1686, 3012),
tr("Compact Bone (Child)"): (1610, 3222),
tr("Spongial Bone (Adult)"): (1172, 1685),
tr("Spongial Bone (Child)"): (1180, 1609),
tr("Muscle Tissue (Adult)"): (1019, 1159),
tr("Muscle Tissue (Child)"): (999, 1163),
tr("Fat Tissue (Adult)"): (819, 973),
tr("Fat Tissue (Child)"): (812, 952),
tr("Skin Tissue (Adult)"): (306, 847),
tr("Skin Tissue (Child)"): (258, 822),
tr("Custom"): (0, 0),
}
)
self.__bind_events()

def __bind_events(self):
Publisher.subscribe(self.UpdateThresholdModes,
'Update threshold limits list')
def __bind_events(self) -> None:
Publisher.subscribe(self.UpdateThresholdModes, "Update threshold limits list")

def UpdateThresholdModes(self, threshold_range):
def UpdateThresholdModes(self, threshold_range: tuple[int, int]) -> None:
thresh_min, thresh_max = threshold_range
presets_list = (self.thresh_ct, self.thresh_mri)

for presets in presets_list:
for key in presets:
(t_min, t_max) = presets[key]

if (t_min is None) or (t_max is None): # setting custom preset
if (t_min is None) or (t_max is None): # setting custom preset
t_min = thresh_min
t_max = thresh_max
t_min = max(t_min, thresh_min)
t_max = min(t_max, thresh_max)

# This has happened in Analyze files
# TODO: find a good solution for presets in Analyze files
if (t_min > thresh_max):
if t_min > thresh_max:
t_min = thresh_min
if (t_max < thresh_min):
if t_max < thresh_min:
t_max = thresh_max

presets[key] = (t_min, t_max)

Publisher.sendMessage('Update threshold limits',
threshold_range=(thresh_min, thresh_max))
Publisher.sendMessage(
"Update threshold limits", threshold_range=(thresh_min, thresh_max)
)

def SavePlist(self, filename):
filename = "%s$%s" % (filename, 'presets.plist')
def SavePlist(self, filename: str | Path) -> str:
filename = "{}${}".format(filename, "presets.plist")
preset = {}

translate_to_en = {_("Bone"):"Bone",
_("Soft Tissue"):"Soft Tissue",
_("Enamel (Adult)"):"Enamel (Adult)",
_("Enamel (Child)"): "Enamel (Child)",
_("Compact Bone (Adult)"):"Compact Bone (Adult)",
_("Compact Bone (Child)"):"Compact Bone (Child)",
_("Spongial Bone (Adult)"):"Spongial Bone (Adult)",
_("Spongial Bone (Child)"):"Spongial Bone (Child)",
_("Muscle Tissue (Adult)"):"Muscle Tissue (Adult)",
_("Muscle Tissue (Child)"):"Muscle Tissue (Child)",
_("Fat Tissue (Adult)"):"Fat Tissue (Adult)",
_("Fat Tissue (Child)"):"Fat Tissue (Child)",
_("Skin Tissue (Adult)"):"Skin Tissue (Adult)",
_("Skin Tissue (Child)"):"Skin Tissue (Child)",
_("Custom"):"Custom"}

translate_to_en = {
tr("Bone"): "Bone",
tr("Soft Tissue"): "Soft Tissue",
tr("Enamel (Adult)"): "Enamel (Adult)",
tr("Enamel (Child)"): "Enamel (Child)",
tr("Compact Bone (Adult)"): "Compact Bone (Adult)",
tr("Compact Bone (Child)"): "Compact Bone (Child)",
tr("Spongial Bone (Adult)"): "Spongial Bone (Adult)",
tr("Spongial Bone (Child)"): "Spongial Bone (Child)",
tr("Muscle Tissue (Adult)"): "Muscle Tissue (Adult)",
tr("Muscle Tissue (Child)"): "Muscle Tissue (Child)",
tr("Fat Tissue (Adult)"): "Fat Tissue (Adult)",
tr("Fat Tissue (Child)"): "Fat Tissue (Child)",
tr("Skin Tissue (Adult)"): "Skin Tissue (Adult)",
tr("Skin Tissue (Child)"): "Skin Tissue (Child)",
tr("Custom"): "Custom",
}

thresh_mri_new = {}
for name in self.thresh_mri.keys():
Expand All @@ -126,67 +133,70 @@ def SavePlist(self, filename):
for name in self.thresh_ct.keys():
thresh_ct_new[translate_to_en[name]] = self.thresh_ct[name]

preset['thresh_mri'] = thresh_mri_new
preset['thresh_ct'] = thresh_ct_new
with open(filename, 'w+b') as f:
preset["thresh_mri"] = thresh_mri_new
preset["thresh_ct"] = thresh_ct_new
with open(filename, "w+b") as f:
plistlib.dump(preset, f)
return os.path.split(filename)[1]

def OpenPlist(self, filename):

translate_to_x = {"Bone":_("Bone"),
"Soft Tissue":_("Soft Tissue"),
"Enamel (Adult)":_("Enamel (Adult)"),
"Enamel (Child)": _("Enamel (Child)"),
"Compact Bone (Adult)": _("Compact Bone (Adult)"),
"Compact Bone (Child)":_("Compact Bone (Child)"),
"Spongial Bone (Adult)":_("Spongial Bone (Adult)"),
"Spongial Bone (Child)":_("Spongial Bone (Child)"),
"Muscle Tissue (Adult)":_("Muscle Tissue (Adult)"),
"Muscle Tissue (Child)":_("Muscle Tissue (Child)"),
"Fat Tissue (Adult)":_("Fat Tissue (Adult)"),
"Fat Tissue (Child)":_("Fat Tissue (Child)"),
"Skin Tissue (Adult)":_("Skin Tissue (Adult)"),
"Skin Tissue (Child)":_("Skin Tissue (Child)"),
"Custom":_("Custom")}


with open(filename, 'rb') as f:
def OpenPlist(self, filename: str | Path) -> None:
translate_to_x = {
"Bone": tr("Bone"),
"Soft Tissue": tr("Soft Tissue"),
"Enamel (Adult)": tr("Enamel (Adult)"),
"Enamel (Child)": tr("Enamel (Child)"),
"Compact Bone (Adult)": tr("Compact Bone (Adult)"),
"Compact Bone (Child)": tr("Compact Bone (Child)"),
"Spongial Bone (Adult)": tr("Spongial Bone (Adult)"),
"Spongial Bone (Child)": tr("Spongial Bone (Child)"),
"Muscle Tissue (Adult)": tr("Muscle Tissue (Adult)"),
"Muscle Tissue (Child)": tr("Muscle Tissue (Child)"),
"Fat Tissue (Adult)": tr("Fat Tissue (Adult)"),
"Fat Tissue (Child)": tr("Fat Tissue (Child)"),
"Skin Tissue (Adult)": tr("Skin Tissue (Adult)"),
"Skin Tissue (Child)": tr("Skin Tissue (Child)"),
"Custom": tr("Custom"),
}

with open(filename, "rb") as f:
p = plistlib.load(f, fmt=plistlib.FMT_XML)
thresh_mri = p['thresh_mri'].copy()
thresh_ct = p['thresh_ct'].copy()
thresh_mri = p["thresh_mri"].copy()
thresh_ct = p["thresh_ct"].copy()

thresh_ct_new = {}
for name in thresh_ct.keys():
for name in thresh_ct.keys():
thresh_ct_new[translate_to_x[name]] = thresh_ct[name]

thresh_mri_new = {}
for name in thresh_mri.keys():
thresh_mri_new[translate_to_x[name]] = thresh_mri[name]
for name in thresh_mri.keys():
thresh_mri_new[translate_to_x[name]] = thresh_mri[name]

self.thresh_mri = TwoWaysDictionary(thresh_mri_new)
self.thresh_ct = TwoWaysDictionary(thresh_ct_new)



def get_wwwl_presets():
files = glob.glob(os.path.join(inv_paths.RAYCASTING_PRESETS_COLOR_DIRECTORY, '*.plist'))
presets = {}
def get_wwwl_presets() -> dict[str, str]:
files = glob.glob(
os.path.join(inv_paths.RAYCASTING_PRESETS_COLOR_DIRECTORY, "*.plist")
)
presets: dict[str, str] = {}
for f in files:
p = os.path.splitext(os.path.basename(f))[0]
presets[p] = f
return presets


def get_wwwl_preset_colours(pfile):
with open(pfile, 'rb') as f:
def get_wwwl_preset_colours(
pfile: str | Path,
) -> list[tuple[int, int, int]]:
with open(pfile, "rb") as f:
preset = plistlib.load(f, fmt=plistlib.FMT_XML)
ncolours = len(preset['Blue'])
ncolours = len(preset["Blue"])
colours = []
for i in range(ncolours):
r = preset['Red'][i]
g = preset['Green'][i]
b = preset['Blue'][i]
r = preset["Red"][i]
g = preset["Green"][i]
b = preset["Blue"][i]

colours.append((r, g, b))

Expand Down
Loading

0 comments on commit 6bead8e

Please sign in to comment.