imageBrowser.py 28 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831
  1. import os
  2. import unittest
  3. from __main__ import vtk, qt, ctk, slicer
  4. from slicer.ScriptedLoadableModule import *
  5. import json
  6. import datetime
  7. import sys
  8. import nixModule
  9. import pathlib
  10. import chardet
  11. import re
  12. #
  13. # labkeySlicerPythonExtension
  14. #
  15. class imageBrowser(ScriptedLoadableModule):
  16. """Uses ScriptedLoadableModule base class, available at:
  17. https://github.com/Slicer/Slicer/blob/master/Base/Python/slicer/ScriptedLoadableModule.py
  18. """
  19. def __init__(self, parent):
  20. ScriptedLoadableModule.__init__(self, parent)
  21. self.parent.title = "image Browser"
  22. # TODO make this more human readable by adding spaces
  23. self.parent.categories = ["LabKey"]
  24. self.parent.dependencies = []
  25. self.parent.contributors = ["Andrej Studen (UL/FMF)"]
  26. # replace with "Firstname Lastname (Organization)"
  27. self.parent.helpText = """
  28. Interface to irAEMM files in LabKey
  29. """
  30. self.parent.acknowledgementText = """
  31. Developed within the medical physics research programme
  32. of the Slovenian research agency.
  33. """ # replace with organization, grant and thanks.
  34. #
  35. # labkeySlicerPythonExtensionWidget
  36. #
  37. class imageBrowserWidget(ScriptedLoadableModuleWidget):
  38. """Uses ScriptedLoadableModuleWidget base class, available at:
  39. https://github.com/Slicer/Slicer/blob/master/Base/Python/slicer/ScriptedLoadableModule.py
  40. """
  41. def setup(self):
  42. print("Setting up imageBrowserWidget")
  43. ScriptedLoadableModuleWidget.setup(self)
  44. # Instantiate and connect widgets ...
  45. self.logic=imageBrowserLogic(self)
  46. self.addInfoSection()
  47. self.addSetupSection()
  48. self.addPatientsSelector()
  49. self.addSegmentEditor()
  50. self.addWindowManipulator()
  51. def addInfoSection(self):
  52. #a python overview of json settings
  53. infoCollapsibleButton = ctk.ctkCollapsibleButton()
  54. infoCollapsibleButton.text = "Info"
  55. self.layout.addWidget(infoCollapsibleButton)
  56. infoLayout = qt.QFormLayout(infoCollapsibleButton)
  57. self.participantField=qt.QLabel("PatientId")
  58. infoLayout.addRow("Participant field:",self.participantField)
  59. self.ctField=qt.QLabel("ctResampled")
  60. infoLayout.addRow("Data field (CT):",self.ctField)
  61. self.petField=qt.QLabel("petResampled")
  62. infoLayout.addRow("Data field (PET):",self.petField)
  63. self.userField=qt.QLabel("Loading")
  64. infoLayout.addRow("User",self.userField)
  65. self.idField=qt.QLabel("Loading")
  66. infoLayout.addRow("ID",self.idField)
  67. #Add logic at some point
  68. #self.logic=imageBrowserLogic(self)
  69. def addPatientsSelector(self):
  70. #
  71. # Patients Area
  72. #
  73. patientsCollapsibleButton = ctk.ctkCollapsibleButton()
  74. patientsCollapsibleButton.text = "Patients"
  75. #don't add it yet
  76. self.layout.addWidget(patientsCollapsibleButton)
  77. patientsFormLayout = qt.QFormLayout(patientsCollapsibleButton)
  78. self.patientList=qt.QComboBox()
  79. self.patientList.currentIndexChanged.connect(self.onPatientListChanged)
  80. patientsFormLayout.addRow("Patient:",self.patientList)
  81. self.visitList=qt.QComboBox()
  82. self.visitList.currentIndexChanged.connect(self.onVisitListChanged)
  83. patientsFormLayout.addRow("Visit:",self.visitList)
  84. self.ctCode=qt.QLabel("ctCode")
  85. patientsFormLayout.addRow("CT:",self.ctCode)
  86. self.petCode=qt.QLabel("petCode")
  87. patientsFormLayout.addRow("PET:",self.petCode)
  88. self.patientLoad=qt.QPushButton("Load")
  89. self.patientLoad.clicked.connect(self.onPatientLoadButtonClicked)
  90. patientsFormLayout.addRow("Load patient",self.patientLoad)
  91. self.patientSave=qt.QPushButton("Save")
  92. self.patientSave.clicked.connect(self.onPatientSaveButtonClicked)
  93. patientsFormLayout.addRow("Save segmentation",self.patientSave)
  94. self.patientClear=qt.QPushButton("Clear")
  95. self.patientClear.clicked.connect(self.onPatientClearButtonClicked)
  96. patientsFormLayout.addRow("Clear patient",self.patientClear)
  97. self.keepCached=qt.QCheckBox("keep Cached")
  98. self.keepCached.setChecked(1)
  99. patientsFormLayout.addRow("Keep cached",self.keepCached)
  100. def addSetupSection(self):
  101. setupCollapsibleButton = ctk.ctkCollapsibleButton()
  102. setupCollapsibleButton.text = "Setup"
  103. self.layout.addWidget(setupCollapsibleButton)
  104. #Form layout (maybe one can think of more intuitive layouts)
  105. setupFormLayout = qt.QFormLayout(setupCollapsibleButton)
  106. self.serverList=qt.QComboBox()
  107. self.serverList.addItem('<Select>')
  108. self.serverList.addItem("astuden")
  109. self.serverList.addItem("adoma")
  110. self.serverList.addItem("kzevnik")
  111. self.serverList.currentIndexChanged.connect(self.onServerListChanged)
  112. setupFormLayout.addRow("Select user:",self.serverList)
  113. self.setupList=qt.QComboBox()
  114. self.setupList.addItem('<Select>')
  115. self.setupList.addItem("limfomiPET_iBrowser.json")
  116. self.setupList.addItem("iraemm_iBrowser.json")
  117. self.setupList.currentIndexChanged.connect(self.onSetupListChanged)
  118. setupFormLayout.addRow("Setup:",self.setupList)
  119. def addReviewSection(self):
  120. #
  121. # Review Area
  122. #
  123. reviewCollapsibleButton = ctk.ctkCollapsibleButton()
  124. reviewCollapsibleButton.text = "Review"
  125. self.layout.addWidget(reviewCollapsibleButton)
  126. self.reviewBoxLayout = qt.QVBoxLayout(reviewCollapsibleButton)
  127. self.reviewFormLayout = qt.QFormLayout()
  128. self.reviewSegment=qt.QComboBox()
  129. self.reviewSegment.currentIndexChanged.connect(\
  130. self.onReviewSegmentChanged)
  131. self.reviewFormLayout.addRow("Selected region:",self.reviewSegment)
  132. self.reviewResult=qt.QComboBox()
  133. sLabel="What do you think about the segmentation:"
  134. self.reviewFormLayout.addRow(sLabel,self.reviewResult)
  135. reviewOptions=['Select','Excellent','Minor deficiencies',\
  136. 'Major deficiencies','Unusable']
  137. for opt in reviewOptions:
  138. self.reviewResult.addItem(opt)
  139. self.aeResult=qt.QComboBox()
  140. aeLabel="Is organ suffering from adverse effect?"
  141. self.reviewFormLayout.addRow(aeLabel,self.aeResult)
  142. aeOptions=['Select','Yes','No']
  143. for opt in aeOptions:
  144. self.aeResult.addItem(opt)
  145. #self.aeResult.setCurrentIndex(0)
  146. self.updateReview=qt.QPushButton("Save")
  147. saLabel="Save segmentation and AE decision for current segment"
  148. self.reviewFormLayout.addRow(saLabel,self.updateReview)
  149. self.updateReview.clicked.connect(self.onUpdateReviewButtonClicked)
  150. self.reviewBoxLayout.addLayout(self.reviewFormLayout)
  151. submitFrame=qt.QGroupBox("Submit data")
  152. self.submitFormLayout=qt.QFormLayout()
  153. self.reviewComment=qt.QTextEdit("this is a test")
  154. self.submitFormLayout.addRow("Comments (optional)",self.reviewComment)
  155. self.submitReviewButton=qt.QPushButton("Submit")
  156. self.submitFormLayout.addRow("Submit to database",\
  157. self.submitReviewButton)
  158. self.submitReviewButton.clicked.connect(\
  159. self.onSubmitReviewButtonClicked)
  160. submitFrame.setLayout(self.submitFormLayout)
  161. submitFrame.setFlat(1)
  162. #submitFrame.setFrameShape(qt.QFrame.StyledPanel)
  163. #submitFrame.setFrameShadow(qt.QFrame.Sunken)
  164. submitFrame.setStyleSheet("background-color:rgba(220,215,180,45)")
  165. self.reviewBoxLayout.addWidget(submitFrame)
  166. def addSegmentEditor(self):
  167. editorCollapsibleButton = ctk.ctkCollapsibleButton()
  168. editorCollapsibleButton.text = "Segment Editor"
  169. self.layout.addWidget(editorCollapsibleButton)
  170. hLayout=qt.QVBoxLayout(editorCollapsibleButton)
  171. self.segmentEditorWidget=slicer.qMRMLSegmentEditorWidget()
  172. hLayout.addWidget(self.segmentEditorWidget)
  173. self.segmentEditorWidget.setMRMLScene(slicer.mrmlScene)
  174. segEditorNode=slicer.vtkMRMLSegmentEditorNode()
  175. slicer.mrmlScene.AddNode(segEditorNode)
  176. self.segmentEditorWidget.setMRMLSegmentEditorNode(segEditorNode)
  177. def addWindowManipulator(self):
  178. windowManipulatorCollapsibleButton=ctk.ctkCollapsibleButton()
  179. windowManipulatorCollapsibleButton.text="CT Window Manipulator"
  180. self.layout.addWidget(windowManipulatorCollapsibleButton)
  181. hLayout=qt.QHBoxLayout(windowManipulatorCollapsibleButton)
  182. ctWins={'CT:bone':self.onCtBoneButtonClicked,
  183. 'CT:air':self.onCtAirButtonClicked,
  184. 'CT:abdomen':self.onCtAbdomenButtonClicked,
  185. 'CT:brain':self.onCtBrainButtonClicked,
  186. 'CT:lung':self.onCtLungButtonClicked}
  187. for ctWin in ctWins:
  188. ctButton=qt.QPushButton(ctWin)
  189. ctButton.clicked.connect(ctWins[ctWin])
  190. hLayout.addWidget(ctButton)
  191. def onSetupListChanged(self,i):
  192. status=self.logic.setConfig(self.setupList.currentText)
  193. try:
  194. if status['error']=='FILE NOT FOUND':
  195. print('File {} not found.'.format(self.setupList.currentText))
  196. return
  197. except KeyError:
  198. pass
  199. self.updatePatientList(status['ids'])
  200. self.onPatientListChanged(0)
  201. def onServerListChanged(self,i):
  202. status=self.logic.setServer(self.serverList.currentText)
  203. try:
  204. if status['error']=='KEY ERROR':
  205. self.serverList.setStyleSheet('background-color: violet')
  206. if status['error']=='ID ERROR':
  207. self.serverList.setStyleSheet('background-color: red')
  208. return
  209. except KeyError:
  210. pass
  211. self.idField.setText(status['id'])
  212. self.userField.setText(status['displayName'])
  213. self.serverList.setStyleSheet('background-color: green')
  214. def onPatientListChanged(self,i):
  215. ds=self.logic.getDataset(dbFilter={'participant':self.patientList.currentText})
  216. visitVar=self.logic.getVarName(var='visitField')
  217. dt=datetime.datetime
  218. #label is a combination of sequence number and date of imaging
  219. seq={row['SequenceNum']:
  220. {'label':row[visitVar],
  221. 'date': dt.strptime(row['studyDate'],'%Y/%m/%d %H:%M:%S')}
  222. for row in ds['rows']}
  223. #apply lookup to visitVar if available
  224. try:
  225. seq={x:
  226. {'label':ds['lookups'][visitVar][seq[x]['label']],
  227. 'date':seq[x]['date']}
  228. for x in seq}
  229. except KeyError:
  230. pass
  231. #format label
  232. seq={x:'{} ({})'.format(seq[x]['label'],dt.strftime(seq[x]['date'],'%d-%b-%Y'))
  233. for x in seq}
  234. self.visitList.clear()
  235. for s in seq:
  236. #onVisitListChanged is called for every addItem
  237. self.visitList.addItem(seq[s],s)
  238. #self.onVisitListChanged(0)
  239. def onVisitListChanged(self,i):
  240. #ignore calls on empty list
  241. if self.visitList.count==0:
  242. return
  243. #get sequence num
  244. s=self.visitList.itemData(i)
  245. print("Visit: SequenceNum:{}, label{}".format(s,self.visitList.currentText))
  246. dbFilter={'participant':self.patientList.currentText,
  247. 'seqNum':s}
  248. ds=self.logic.getDataset(dbFilter=dbFilter)
  249. if not len(ds['rows'])==1:
  250. print("Found incorrect number {} of matches for [{}]/[{}]".\
  251. format(len(ds['rows']),\
  252. self.patientList.currentText,s))
  253. row=ds['rows'][0]
  254. #copy row properties for data access
  255. self.currentRow=row
  256. self.petCode.setText(row[self.petField.text])
  257. self.ctCode.setText(row[self.ctField.text])
  258. #self.segmentationCode.setText(row[self.segmentationField.text])
  259. def updatePatientList(self,ids):
  260. self.patientList.clear()
  261. for id in ids:
  262. self.patientList.addItem(id)
  263. def onPatientLoadButtonClicked(self):
  264. print("Load")
  265. #delegate loading to logic
  266. self.logic.loadImages(self.currentRow,self.keepCached.isChecked())
  267. self.logic.loadSegmentation(self.currentRow)
  268. self.setSegmentEditor()
  269. #self.logic.loadReview(self.currentRow)
  270. #self.logic.loadAE(self.currentRow)
  271. #self.onReviewSegmentChanged()
  272. def setSegmentEditor(self):
  273. #use current row to set segment in segment editor
  274. self.segmentEditorWidget.setSegmentationNode(
  275. self.logic.volumeNode['Segmentation'])
  276. self.segmentEditorWidget.setMasterVolumeNode(
  277. self.logic.volumeNode['PET'])
  278. def onReviewSegmentChanged(self):
  279. pass
  280. def onPatientClearButtonClicked(self):
  281. self.logic.clearVolumesAndSegmentations()
  282. def onPatientSaveButtonClicked(self):
  283. self.logic.saveSegmentation()
  284. def onCtBoneButtonClicked(self):
  285. self.logic.setWindow('CT:bone')
  286. def onCtAirButtonClicked(self):
  287. self.logic.setWindow('CT:air')
  288. def onCtAbdomenButtonClicked(self):
  289. self.logic.setWindow('CT:abdomen')
  290. def onCtBrainButtonClicked(self):
  291. self.logic.setWindow('CT:brain')
  292. def onCtLungButtonClicked(self):
  293. self.logic.setWindow('CT:lung')
  294. def cleanup(self):
  295. pass
  296. def loadLibrary(name):
  297. #utility function to load nix library from git
  298. fwrapper=nixModule.getWrapper('nixWrapper.py')
  299. p=pathlib.Path(fwrapper)
  300. sys.path.append(str(p.parent))
  301. import nixWrapper
  302. return nixWrapper.loadLibrary(name)
  303. #
  304. # imageBrowserLogic
  305. #
  306. class imageBrowserLogic(ScriptedLoadableModuleLogic):
  307. """This class should implement all the actual
  308. computation done by your module. The interface
  309. should be such that other python code can import
  310. this class and make use of the functionality without
  311. requiring an instance of the Widget.
  312. Uses ScriptedLoadableModuleLogic base class, available at:
  313. https://github.com/Slicer/Slicer/blob/master/Base/Python/slicer/ScriptedLoadableModule.py
  314. """
  315. def __init__(self,parent=None):
  316. ScriptedLoadableModuleLogic.__init__(self, parent)
  317. print('imageBrowserLogic loading')
  318. if not parent==None:
  319. #use layout and data from parent widget
  320. self.parent=parent
  321. fhome=os.path.expanduser('~')
  322. fsetup=os.path.join(fhome,'.labkey','setup.json')
  323. try:
  324. with open(fsetup) as f:
  325. self.setup=json.load(f)
  326. except FileNotFoundError:
  327. self.setup={}
  328. try:
  329. pt=self.setup['paths']
  330. except KeyError:
  331. self.setup['paths']={}
  332. lName='labkeyInterface'
  333. loadLibrary(lName)
  334. import labkeyInterface
  335. import labkeyDatabaseBrowser
  336. import labkeyFileBrowser
  337. self.network=labkeyInterface.labkeyInterface()
  338. self.dbBrowser=labkeyDatabaseBrowser
  339. self.fBrowser=labkeyFileBrowser
  340. self.tempDir=os.path.join(os.path.expanduser('~'),'temp')
  341. if not os.path.isdir(self.tempDir):
  342. os.mkdir(self.tempDir)
  343. self.lookups={}
  344. self.segmentList=["Liver","Blood","Marrow","Disease","Deauville"]
  345. print('imageBrowserLogic setup complete')
  346. def setServer(self,serverName):
  347. #additional way of setting the labkey network interface
  348. #if no parent was provided in logic initialization (stand-alone mode)
  349. status={}
  350. fileName="NONE"
  351. if serverName=="astuden":
  352. fileName="astuden.json"
  353. if serverName=="adoma":
  354. fileName="adoma.json"
  355. if serverName=="kzevnik":
  356. fileName="kzevnik.json"
  357. if fileName=="NONE":
  358. print("No path was associated with server {}".format(serverName))
  359. status['error']='KEY ERROR'
  360. return status
  361. fconfig=os.path.join(os.path.expanduser('~'),'.labkey',fileName)
  362. self.network.init(fconfig)
  363. self.remoteId=self.network.getUserId()
  364. if self.remoteId==None:
  365. status['error']='ID ERROR'
  366. return status
  367. status['displayName']=self.remoteId['displayName']
  368. status['id']=self.remoteId['id']
  369. #reset db and fb (they are thin classes anyhow)
  370. self.db=self.dbBrowser.labkeyDB(self.network)
  371. self.fb=self.fBrowser.labkeyFileBrowser(self.network)
  372. return status
  373. def setConfig(self,configName):
  374. status={}
  375. fileName=os.path.join(os.path.expanduser('~'),'.labkey',configName)
  376. if not os.path.isfile(fileName):
  377. status['error']='FILE NOT FOUND'
  378. return status
  379. with open(fileName,'r') as f:
  380. self.isetup=json.load(f)
  381. #self.project=self.isetup['project']
  382. #"iPNUMMretro/Study"
  383. #self.schema='study'
  384. #self.dataset=self.isetup['query']
  385. ds=self.getDataset()
  386. ids=[row[self.isetup['participantField']] for row in ds['rows']]
  387. status['ids']=list(set(ids))
  388. return status
  389. def getVarName(self,name="Imaging",var="visitField"):
  390. dset=self.isetup['datasets'][name]
  391. defaults={"visitField":"imagingVisitId"}
  392. try:
  393. return dset[var]
  394. except KeyError:
  395. return defaults[var]
  396. def getDataset(self,name="Imaging",dbFilter=[]):
  397. dset=self.isetup['datasets'][name]
  398. project=dset['project']
  399. schema=dset['schema']
  400. query=dset['query']
  401. qFilter=[]
  402. for f in dbFilter:
  403. if f=='participant':
  404. qFilter.append({'variable':self.isetup['participantField'],
  405. 'value':dbFilter[f],'oper':'eq'})
  406. continue
  407. if f=='seqNum':
  408. qFilter.append({'variable':'SequenceNum',
  409. 'value':'{}'.format(dbFilter[f]),
  410. 'oper':'eq'})
  411. continue
  412. qFilter.append({'variable':f,'value':dbFilter[f],'oper':'eq'})
  413. try:
  414. ds=self.db.selectRows(project,schema,query, \
  415. qFilter,dset['view'])
  416. except KeyError:
  417. ds=self.db.selectRows(project,schema,query,qFilter)
  418. #get lookups as well
  419. lookups={}
  420. for f in ds['metaData']['fields']:
  421. try:
  422. lookup=f['lookup']
  423. except KeyError:
  424. continue
  425. var=f['name']
  426. lookupCode='{}:{}'.format(lookup['schema'],lookup['queryName'])
  427. try:
  428. lookups[var]=self.lookups[lookupCode]
  429. except KeyError:
  430. self.lookups[lookupCode]=self.loadLookup(project,lookup)
  431. lookups[var]=self.lookups[lookupCode]
  432. return {'rows':ds['rows'],'lookups':lookups}
  433. def loadLookup(self,project,lookup):
  434. qData={}
  435. key=lookup['keyColumn']
  436. value=lookup['displayColumn']
  437. fSet=self.db.selectRows(project,lookup['schema'],lookup['queryName'],[])
  438. for q in fSet['rows']:
  439. qData[q[key]]=q[value]
  440. return qData
  441. def loadImage(self,iData):
  442. #unpack iData
  443. idx=iData['idx']
  444. path=iData['path']
  445. keepCached=iData['keepCached']
  446. dset=self.isetup['datasets'][iData['dataset']]
  447. localPath=os.path.join(self.tempDir,path[-1])
  448. if not os.path.isfile(localPath):
  449. #download from server
  450. remotePath=self.fb.formatPathURL(dset['project'],'/'.join(path))
  451. if not self.fb.entryExists(remotePath):
  452. print("Failed to get {}".format(remotePath))
  453. return
  454. self.fb.readFileToFile(remotePath,localPath)
  455. properties={}
  456. filetype='VolumeFile'
  457. #make sure segmentation gets loaded as a labelmap
  458. if idx=="Segmentation":
  459. filetype='SegmentationFile'
  460. #properties["labelmap"]=1
  461. self.volumeNode[idx]=slicer.util.loadNodeFromFile(localPath,
  462. filetype=filetype,properties=properties)
  463. if not keepCached:
  464. os.remove(localPath)
  465. def loadImages(self,row,keepCached):
  466. #fields={'ctResampled':True,'petResampled':False}
  467. fields={"CT":self.parent.ctField.text,\
  468. "PET":self.parent.petField.text}
  469. path=[self.isetup['imageDir'],row['patientCode'],row['visitCode']]
  470. relativePaths={x:path+[row[y]] for (x,y) in fields.items()}
  471. self.volumeNode={}
  472. for f in relativePaths:
  473. iData={'idx':f,'path':relativePaths[f],
  474. 'keepCached':keepCached,'dataset':'Imaging'}
  475. self.loadImage(iData)
  476. #mimic abdominalCT standardized window setting
  477. self.volumeNode['CT'].GetScalarVolumeDisplayNode().\
  478. SetAutoWindowLevel(False)
  479. self.volumeNode['CT'].GetScalarVolumeDisplayNode().\
  480. SetWindowLevel(1400, -500)
  481. #set colormap for PET to PET-Heat (this is a verbatim setting from
  482. #the Volumes->Display->Lookup Table colormap identifier)
  483. self.volumeNode['PET'].GetScalarVolumeDisplayNode().\
  484. SetAndObserveColorNodeID(\
  485. slicer.util.getNode('Inferno').GetID())
  486. slicer.util.setSliceViewerLayers(background=self.volumeNode['CT'],\
  487. foreground=self.volumeNode['PET'],foregroundOpacity=0.5,fit=True)
  488. def loadSegmentation(self,row):
  489. dbFilter={'User':'{}'.format(self.remoteId['id']),
  490. 'participant':row[self.isetup['participantField']],
  491. 'visitCode':row['visitCode']}
  492. ds=self.getDataset(name='SegmentationsMaster',
  493. dbFilter=dbFilter)
  494. if len(ds['rows'])>1:
  495. print('Multiple segmentations found!')
  496. return
  497. if len(ds['rows'])==1:
  498. #update self.segmentationEntry
  499. self.segmentationEntry=ds['rows'][0]
  500. self.segmentationEntry['origin']='database'
  501. self.loadSegmentationFromEntry()
  502. return
  503. #create new segmentation
  504. self.createSegmentation(row)
  505. def getSegmentationPath(self):
  506. path=[self.isetup['imageDir'],
  507. self.segmentationEntry['patientCode'],
  508. self.segmentationEntry['visitCode']]
  509. path.append('Segmentations')
  510. return path
  511. def loadSegmentationFromEntry(self):
  512. #compile path
  513. entry=self.segmentationEntry
  514. path=self.getSegmentationPath()
  515. path.append(entry['latestFile'])
  516. iData={'idx':'Segmentation','path':path,
  517. 'keepCached':0,'dataset':'SegmentationsMaster'}
  518. self.loadImage(iData)
  519. #look for missing segments
  520. segNode=self.volumeNode['Segmentation']
  521. seg=segNode.GetSegmentation()
  522. segNames=[seg.GetNthSegmentID(i) for i in range(seg.GetNumberOfSegments())]
  523. print('Segments')
  524. for s in self.segmentList:
  525. if s not in segNames:
  526. seg.AddEmptySegment(s,s)
  527. print(s)
  528. print('Done')
  529. def saveSegmentation(self):
  530. #get the latest key by adding an entry to SegmentationList
  531. copyFields=['ParticipantId','patientCode','visitCode','User']
  532. outRow={x:self.segmentationEntry[x] for x in copyFields}
  533. sList=self.isetup['datasets']['SegmentationsList']
  534. resp=self.db.modifyRows('insert',sList['project'],
  535. sList['schema'],sList['query'],[outRow])
  536. encoding=chardet.detect(resp)['encoding']
  537. respJSON=json.loads(resp.decode(encoding))
  538. outRow=respJSON['rows'][0]
  539. #print(outRow)
  540. #construct file name with known key
  541. uName=re.sub(' ','_',self.remoteId['displayName'])
  542. fName='Segmentation_{}-{}_{}_{}.nrrd'.format(
  543. self.segmentationEntry['patientCode'],
  544. self.segmentationEntry['visitCode'],
  545. uName,outRow['Key'])
  546. path=self.getSegmentationPath()
  547. path.append(fName)
  548. self.saveNode(self.volumeNode['Segmentation'],'SegmentationsMaster',path)
  549. #update SegmentationList with know file name
  550. outRow['Segmentation']=fName
  551. self.db.modifyRows('update',sList['project'],
  552. sList['schema'],sList['query'],[outRow])
  553. #update SegmentationsMaster
  554. self.segmentationEntry['latestFile']=fName
  555. self.segmentationEntry['version']=outRow['Key']
  556. des=self.isetup['datasets']['SegmentationsMaster']
  557. op='insert'
  558. if self.segmentationEntry['origin']=='database':
  559. op='update'
  560. self.db.modifyRows(op,des['project'],
  561. des['schema'],des['query'],[self.segmentationEntry])
  562. #since we loaded a version, origin should be set to dataset
  563. self.segmentationEntry['origin']='dataset'
  564. def saveNode(self,node,dataset,path):
  565. fName=path[-1]
  566. localPath=os.path.join(self.tempDir,fName)
  567. slicer.util.saveNode(node,localPath)
  568. dset=self.isetup['datasets'][dataset]
  569. #exclude file name when building directory structure
  570. remotePath=self.fb.buildPathURL(dset['project'],path[:-1])
  571. remotePath+='/'+fName
  572. self.fb.writeFileToFile(localPath,remotePath)
  573. #add entry to segmentation list
  574. def createSegmentation(self,entry):
  575. #create segmentation entry for database update
  576. #note that origin is not set to database
  577. copyFields=['ParticipantId','patientCode','visitCode','SequenceNum']
  578. self.segmentationEntry={x:entry[x] for x in copyFields}
  579. self.segmentationEntry['User']=self.remoteId['id']
  580. self.segmentationEntry['origin']='created'
  581. self.segmentationEntry['version']=-1111
  582. #create a segmentation node
  583. uName=re.sub(' ','_',self.remoteId['displayName'])
  584. segNode=slicer.vtkMRMLSegmentationNode()
  585. self.volumeNode['Segmentation']=segNode
  586. segNode.SetName('Segmentation_{}_{}_{}'.
  587. format(entry['patientCode'],entry['visitCode'],uName))
  588. slicer.mrmlScene.AddNode(segNode)
  589. segNode.CreateDefaultDisplayNodes()
  590. segNode.SetReferenceImageGeometryParameterFromVolumeNode(self.volumeNode['PET'])
  591. for s in self.segmentList:
  592. segNode.GetSegmentation().AddEmptySegment(s,s)
  593. def clearVolumesAndSegmentations(self):
  594. nodes=slicer.util.getNodesByClass("vtkMRMLVolumeNode")
  595. nodes.extend(slicer.util.getNodesByClass("vtkMRMLSegmentationNode"))
  596. res=[slicer.mrmlScene.RemoveNode(f) for f in nodes]
  597. #self.segmentationNode=None
  598. #self.reviewResult={}
  599. #self.aeList={}
  600. def setWindow(self,windowName):
  601. #default
  602. w=1400
  603. l=-500
  604. if windowName=='CT:bone':
  605. w=1000
  606. l=400
  607. if windowName=='CT:air':
  608. w=1000
  609. l=-426
  610. if windowName=='CT:abdomen':
  611. w=350
  612. l=40
  613. if windowName=='CT:brain':
  614. w=100
  615. l=50
  616. if windowName=='CT:lung':
  617. w=1400
  618. l=-500
  619. self.volumeNode['CT'].GetScalarVolumeDisplayNode().\
  620. SetWindowLevel(w,l)
  621. print('setWindow: {} {}/{}'.format(windowName,w,l))
  622. class imageBrowserTest(ScriptedLoadableModuleTest):
  623. """
  624. This is the test case for your scripted module.
  625. Uses ScriptedLoadableModuleTest base class, available at:
  626. https://github.com/Slicer/Slicer/blob/master/Base/Python/slicer/ScriptedLoadableModule.py
  627. """
  628. def setup(self):
  629. """ Do whatever is needed to reset the state - typically a scene clear will be enough.
  630. """
  631. slicer.mrmlScene.Clear(0)
  632. def runTest(self):
  633. """Run as few or as many tests as needed here.
  634. """
  635. self.setUp()
  636. self.test_irAEMMBrowser()
  637. def test_irAEMMBrowser(self):
  638. """ Ideally you should have several levels of tests. At the lowest level
  639. tests sould exercise the functionality of the logic with different inputs
  640. (both valid and invalid). At higher levels your tests should emulate the
  641. way the user would interact with your code and confirm that it still works
  642. the way you intended.
  643. One of the most important features of the tests is that it should alert other
  644. developers when their changes will have an impact on the behavior of your
  645. module. For example, if a developer removes a feature that you depend on,
  646. your test should break so they know that the feature is needed.
  647. """
  648. self.delayDisplay("Starting the test")
  649. #
  650. # first, get some data
  651. #