def cropDcm(self):
oseg = self.oseg
if oseg.data3d is None:
self.statusBar().showMessage('No DICOM data!')
return
self.statusBar().showMessage('Cropping DICOM data...')
QApplication.processEvents()
pyed = QTSeedEditor(oseg.data3d, mode='crop',
voxelSize=oseg.voxelsize_mm)
# @TODO
mx = self.oseg.viewermax
mn = self.oseg.viewermin
width = mx - mn
# enter = (float(mx)-float(mn))
center = np.average([mx, mn])
logger.debug("window params max %f min %f width, %f center %f" %
(mx, mn, width, center))
pyed.changeC(center)
pyed.changeW(width)
pyed.exec_()
crinfo = pyed.getROI()
if crinfo is not None:
tmpcrinfo = []
for ii in crinfo:
tmpcrinfo.append([ii.start, ii.stop])
# seg.data3d = qmisc.crop(oseg.data3d, oseg.crinfo)
oseg.crop(tmpcrinfo)
self.setLabelText(self.text_dcm_data, self.getDcmInfo())
self.statusBar().showMessage('Ready')
def visualization(self):
"""
Run viewer with output data3d and segmentation
"""
try:
from pysegbase.seed_editor_qt import QTSeedEditor
except:
logger.warning("Deprecated of pyseg_base as submodule")
from seed_editor_qt import QTSeedEditor
from PyQt4.QtGui import QApplication
import numpy as np
#, QMainWindow
app = QApplication(sys.argv)
#pyed = QTSeedEditor(self.data3d, contours=(self.segmentation>0))
pyed = QTSeedEditor(self.segmentation)
pyed.exec_()
#import pdb; pdb.set_trace()
#pyed = QTSeedEditor(deletemask, mode='draw')
#pyed.exec_()
app.exit()
def maskRegion(self):
if self.oseg.data3d is None:
self.statusBar().showMessage('No DICOM data!')
return
self.statusBar().showMessage('Mask region...')
QApplication.processEvents()
pyed = QTSeedEditor(
self.oseg.data3d, mode='mask',
voxelSize=self.oseg.voxelsize_mm,
contours=((self.oseg.segmentation == 0).astype(np.int8)*2)
)
pyed.contours_old = pyed.contours.copy()
# initial mask set
# pyed.masked = np.ones(self.oseg.data3d.shape, np.int8)
# pyed.masked = (self.oseg.segmentation == 0).astype(np.int8)
mx = self.oseg.viewermax
mn = self.oseg.viewermin
width = mx - mn
# enter = (float(mx)-float(mn))
center = np.average([mx, mn])
logger.debug("window params max %f min %f width, %f center %f" %
(mx, mn, width, center))
pyed.changeC(center)
pyed.changeW(width)
pyed.exec_()
self.statusBar().showMessage('Ready')
def visualization(self):
"""
Run viewer with output data3d and segmentation
"""
try:
from pysegbase.seed_editor_qt import QTSeedEditor
except:
logger.warning("Deprecated of pyseg_base as submodule")
from seed_editor_qt import QTSeedEditor
from PyQt4.QtGui import QApplication
import numpy as np
#, QMainWindow
app = QApplication(sys.argv)
#pyed = QTSeedEditor(self.data3d, contours=(self.segmentation>0))
pyed = QTSeedEditor(self.segmentation)
pyed.exec_()
#import pdb; pdb.set_trace()
#pyed = QTSeedEditor(deletemask, mode='draw')
#pyed.exec_()
app.exit()
def cropDcm(self):
oseg = self.oseg
if oseg.data3d is None:
self.statusBar().showMessage('No DICOM data!')
return
self.statusBar().showMessage('Cropping DICOM data...')
QApplication.processEvents()
pyed = QTSeedEditor(oseg.data3d, mode='crop',
voxelSize=oseg.voxelsize_mm)
# @TODO
mx = self.oseg.viewermax
mn = self.oseg.viewermin
width = mx - mn
# enter = (float(mx)-float(mn))
center = np.average([mx, mn])
logger.debug("window params max %f min %f width, %f center %f" %
(mx, mn, width, center))
pyed.changeC(center)
pyed.changeW(width)
pyed.exec_()
crinfo = pyed.getROI()
if crinfo is not None:
tmpcrinfo = []
for ii in crinfo:
tmpcrinfo.append([ii.start, ii.stop])
# seg.data3d = qmisc.crop(oseg.data3d, oseg.crinfo)
oseg.crop(tmpcrinfo)
self.setLabelText(self.text_dcm_data, self.getDcmInfo())
self.statusBar().showMessage('Ready')
def test_data_editor(self):
"""
Funkce provádí změnu vstupních dat - data3d
"""
#pyed = sed3.sed3(self.data3d, contour = oseg.segmentation)
#pyed.show()
try:
from pysegbase.seed_editor_qt import QTSeedEditor
except:
logger.warning("Deprecated of pyseg_base as submodule")
from seed_editor_qt import QTSeedEditor
from PyQt4.QtGui import QApplication
import numpy as np
im3d = np.random.rand(15,15,15)
print ("Select pixels for deletion (it will be setted to 0)")
#, QMainWindow
app = QApplication(sys.argv)
pyed = QTSeedEditor(im3d, mode='draw')
pyed.exec_()
deletemask = pyed.getSeeds()
print ("If it is ok, press 'Return'. If it is wrong, click into image and press 'Return'")
# rewrite input data
im3d [deletemask != 0] = 0
pyed = QTSeedEditor(im3d)
app.exec_()
sds = pyed.getSeeds()
# if usere select pixel, there will be 1 or 2 in sds
self.assertLess(np.max(sds), 1)
def test_viewer_seeds(self):
try:
from pysegbase.seed_editor_qt import QTSeedEditor
except:
print("Deprecated of pyseg_base as submodule")
from seed_editor_qt import QTSeedEditor
from PyQt4.QtGui import QApplication
import numpy as np
img3d = (np.random.rand(30, 30, 30)*10).astype(np.int16)
seeds = (np.zeros(img3d.shape)).astype(np.int8)
seeds[3:6, 12:18, 9:16] = 1
seeds[3:6, 19:22, 21:27] = 2
# , QMainWindow
app = QApplication(sys.argv)
pyed = QTSeedEditor(img3d, seeds=seeds)
pyed.exec_()
# deletemask = pyed.getSeeds()
# import pdb; pdb.set_trace()
# pyed = QTSeedEditor(deletemask, mode='draw')
# pyed.exec_()
app.exit()
def maskRegion(self):
if self.oseg.data3d is None:
self.statusBar().showMessage('No DICOM data!')
return
self.statusBar().showMessage('Mask region...')
QApplication.processEvents()
pyed = QTSeedEditor(
self.oseg.data3d, mode='mask',
voxelSize=self.oseg.voxelsize_mm,
contours=((self.oseg.segmentation == 0).astype(np.int8)*2)
)
pyed.contours_old = pyed.contours.copy()
# initial mask set
# pyed.masked = np.ones(self.oseg.data3d.shape, np.int8)
# pyed.masked = (self.oseg.segmentation == 0).astype(np.int8)
mx = self.oseg.viewermax
mn = self.oseg.viewermin
width = mx - mn
# enter = (float(mx)-float(mn))
center = np.average([mx, mn])
logger.debug("window params max %f min %f width, %f center %f" %
(mx, mn, width, center))
pyed.changeC(center)
pyed.changeW(width)
pyed.exec_()
self.statusBar().showMessage('Ready')
def test_data_editor(self):
"""
Funkce provádí změnu vstupních dat - data3d
"""
#pyed = sed3.sed3(self.data3d, contour = oseg.segmentation)
#pyed.show()
try:
from pysegbase.seed_editor_qt import QTSeedEditor
except:
logger.warning("Deprecated of pyseg_base as submodule")
from seed_editor_qt import QTSeedEditor
from PyQt4.QtGui import QApplication
import numpy as np
im3d = np.random.rand(15, 15, 15)
print("Select pixels for deletion (it will be setted to 0)")
#, QMainWindow
app = QApplication(sys.argv)
pyed = QTSeedEditor(im3d, mode='draw')
pyed.exec_()
deletemask = pyed.getSeeds()
print(
"If it is ok, press 'Return'. If it is wrong, click into image and press 'Return'"
)
# rewrite input data
im3d[deletemask != 0] = 0
pyed = QTSeedEditor(im3d)
app.exec_()
sds = pyed.getSeeds()
# if usere select pixel, there will be 1 or 2 in sds
self.assertLess(np.max(sds), 1)
def manualSeg(self):
if self.dcm_3Ddata is None:
self.statusBar().showMessage('No DICOM data!')
return
pyed = QTSeedEditor(self.dcm_3Ddata,
seeds=self.segmentation_data,
mode='draw',
voxelSize=self.voxel_sizemm)
pyed.exec_()
self.segmentation_data = pyed.getSeeds()
self.checkSegData()
def maskRegion(self):
if self.dcm_3Ddata is None:
self.statusBar().showMessage('No DICOM data!')
return
self.statusBar().showMessage('Mask region...')
QApplication.processEvents()
pyed = QTSeedEditor(self.dcm_3Ddata, mode='mask',
voxelSize=self.voxel_sizemm)
pyed.exec_()
self.statusBar().showMessage('Ready')
def manualSeg(self):
oseg = self.oseg
# rint 'ms d3d ', oseg.data3d.shape
# rint 'ms seg ', oseg.segmentation.shape
# rint 'crinfo ', oseg.crinfo
if oseg.data3d is None:
self.statusBar().showMessage("No DICOM data!")
return
sgm = oseg.segmentation.astype(np.uint8)
pyed = QTSeedEditor(oseg.data3d, seeds=sgm, mode="draw", voxelSize=oseg.voxelsize_mm, volume_unit="ml")
pyed.exec_()
oseg.segmentation = pyed.getSeeds()
self.oseg.processing_time = datetime.datetime.now() - self.oseg.time_start
self.checkSegData("manual seg., ")
def autoSeg(self):
if self.dcm_3Ddata is None:
self.statusBar().showMessage('No DICOM data!')
return
igc = pycut.ImageGraphCut(self.dcm_3Ddata,
voxelsize=self.voxel_sizemm)
pyed = QTSeedEditor(self.dcm_3Ddata,
seeds=self.segmentation_seeds,
modeFun=igc.interactivity_loop,
voxelSize=self.voxel_sizemm)
pyed.exec_()
self.segmentation_data = pyed.getContours()
self.segmentation_seeds = pyed.getSeeds()
self.checkSegData()
def cropDcm(self):
if self.dcm_3Ddata is None:
self.statusBar().showMessage('No DICOM data!')
return
self.statusBar().showMessage('Cropping DICOM data...')
QApplication.processEvents()
pyed = QTSeedEditor(self.dcm_3Ddata, mode='crop',
voxelSize=self.voxel_sizemm)
pyed.exec_()
self.dcm_3Ddata = pyed.getImg()
self.dcm_offsetmm = pyed.getOffset()
self.setLabelText(self.text_dcm_data, self.getDcmInfo())
self.statusBar().showMessage('Ready')
self.statusBar().showMessage('Ready')
def interactivity(self, min_val=None, max_val=None):
try:
from pysegbase.seed_editor_qt import QTSeedEditor
except:
logger.warning("Deprecated of pyseg_base as submodule")
from seed_editor_qt import QTSeedEditor
import_gui()
logger.debug('interactivity')
# if self.edit_data:
# self.data3d = self.data_editor(self.data3d)
igc = self._interactivity_begin()
# from PyQt4.QtCore import pyqtRemoveInputHook
# pyqtRemoveInputHook()
# import ipdb; ipdb.set_trace() # noqa BREAKPOINT
pyed = QTSeedEditor(igc.img,
seeds=igc.seeds,
modeFun=igc.interactivity_loop,
voxelSize=igc.voxelsize,
volume_unit='ml')
# set window
if min_val is None:
min_val = np.min(self.data3d)
if max_val is None:
max_val = np.max(self.data3d)
window_c = ((max_val + min_val) / 2)
window_w = (max_val - min_val)
pyed.changeC(window_c)
pyed.changeW(window_w)
pyed.exec_()
# @TODO někde v igc.interactivity() dochází k přehození nul za jedničy,
# tady se to řeší hackem
if igc.segmentation is not None:
self.segmentation = (igc.segmentation == 0).astype(np.int8)
self._interactivity_end(igc)
def btnManualSeg(self):
oseg = self.oseg
# rint 'ms d3d ', oseg.data3d.shape
# rint 'ms seg ', oseg.segmentation.shape
# rint 'crinfo ', oseg.crinfo
if oseg.data3d is None:
self.statusBar().showMessage('No DICOM data!')
return
sgm = oseg.segmentation.astype(np.uint8)
pyed = QTSeedEditor(oseg.data3d,
seeds=sgm,
mode='draw',
voxelSize=oseg.voxelsize_mm, volume_unit='ml')
pyed.exec_()
oseg.segmentation = pyed.getSeeds()
self.oseg.processing_time = \
datetime.datetime.now() - self.oseg.time_start
self.checkSegData('manual seg., ')
def maskRegion(self):
if self.oseg.data3d is None:
self.statusBar().showMessage("No DICOM data!")
return
self.statusBar().showMessage("Mask region...")
QApplication.processEvents()
pyed = QTSeedEditor(self.oseg.data3d, mode="mask", voxelSize=self.oseg.voxelsize_mm)
mx = self.oseg.viewermax
mn = self.oseg.viewermin
width = mx - mn
# enter = (float(mx)-float(mn))
center = np.average([mx, mn])
logger.debug("window params max %f min %f width, %f center %f" % (mx, mn, width, center))
pyed.changeC(center)
pyed.changeW(width)
pyed.exec_()
self.statusBar().showMessage("Ready")
