|
@@ -3,9 +3,13 @@ import os
|
|
|
import radiomics
|
|
|
import SimpleITK
|
|
|
import sys
|
|
|
+import json
|
|
|
|
|
|
def main(parFile='../templates/statistics.json'):
|
|
|
setup=statUtils.loadSetup(parFile)
|
|
|
+ rFile='radiomics.json'
|
|
|
+ with open(rFile,'w') as f:
|
|
|
+ f.write(json.dumps(setup['radiomics']))
|
|
|
setup['db'],setup['fb']=statUtils.connectDB('onko-nix')
|
|
|
users=statUtils.getUsers(setup['db'],setup['project'])
|
|
|
qFilter=[]
|
|
@@ -14,14 +18,14 @@ def main(parFile='../templates/statistics.json'):
|
|
|
qFilter.append({'variable':'ParticipantId','value':vList,'oper':'in'})
|
|
|
except KeyError:
|
|
|
pass
|
|
|
- ds=setup['db'].selectRows(setup['project'],'study','Imaging1',qFilter)
|
|
|
+ ds=setup['db'].selectRows(setup['project'],'study',setup['imagingDataset'],qFilter)
|
|
|
if not os.path.isdir(setup['localDir']):
|
|
|
os.mkdir(setup['localDir'])
|
|
|
#select just the first row; debugging
|
|
|
rows=ds['rows']
|
|
|
setup['values']=['COM','MTV','TLG','SUVmean','SUVmax']
|
|
|
- params=os.path.join('..','templates','radiomics.yaml')
|
|
|
- setup['featureExtractor']=radiomics.featureextractor.RadiomicsFeatureExtractor(params)
|
|
|
+ #params=os.path.join('..','templates','radiomics.yaml')
|
|
|
+ setup['featureExtractor']=radiomics.featureextractor.RadiomicsFeatureExtractor(rFile)
|
|
|
for r in rows:
|
|
|
print(r)
|
|
|
for q in ['petResampled']:
|
|
@@ -49,12 +53,12 @@ def main(parFile='../templates/statistics.json'):
|
|
|
output.update({x:r[x] for x in ['ParticipantId','SequenceNum','patientCode','visitCode']})
|
|
|
output['User']=x
|
|
|
output['segment']=ids[id]
|
|
|
- statUtils.updateDatasetRows(setup['db'],setup['project'],'SUVanalysis',[output])
|
|
|
+ statUtils.updateDatasetRows(setup['db'],setup['project'],setup['SUVdataset'],[output])
|
|
|
#print(output)
|
|
|
#cleanup
|
|
|
- for x in segPath:
|
|
|
- os.remove(segPath[x])
|
|
|
+ for x in segPaths:
|
|
|
+ os.remove(segPaths[x])
|
|
|
os.remove(localPath)
|
|
|
-
|
|
|
+ os.remove(rFile)
|
|
|
if __name__=='__main__':
|
|
|
main(sys.argv[1])
|