imageBrowser.py 26 KB

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