Mercurial > hg > orthanc
annotate Resources/Samples/Python/AutoClassify.py @ 1183:6ef2c81581cd
more fault-tolerant sample
author | Sebastien Jodogne <s.jodogne@gmail.com> |
---|---|
date | Fri, 10 Oct 2014 08:52:56 +0200 |
parents | d74ac5d0bcaf |
children | 4e9d517503ae |
rev | line source |
---|---|
1182
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
1 #!/usr/bin/python |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
2 # -*- coding: utf-8 -*- |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
3 |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
4 # Orthanc - A Lightweight, RESTful DICOM Store |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
5 # Copyright (C) 2012-2014 Medical Physics Department, CHU of Liege, |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
6 # Belgium |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
7 # |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
8 # This program is free software: you can redistribute it and/or |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
9 # modify it under the terms of the GNU General Public License as |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
10 # published by the Free Software Foundation, either version 3 of the |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
11 # License, or (at your option) any later version. |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
12 # |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
13 # This program is distributed in the hope that it will be useful, but |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
14 # WITHOUT ANY WARRANTY; without even the implied warranty of |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
15 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
16 # General Public License for more details. |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
17 # |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
18 # You should have received a copy of the GNU General Public License |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
19 # along with this program. If not, see <http://www.gnu.org/licenses/>. |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
20 |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
21 |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
22 import argparse |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
23 import time |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
24 import os |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
25 import os.path |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
26 import RestToolbox |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
27 |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
28 parser = argparse.ArgumentParser( |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
29 description = 'Automated classification of DICOM files from Orthanc.', |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
30 formatter_class = argparse.ArgumentDefaultsHelpFormatter) |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
31 |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
32 parser.add_argument('--host', default = 'localhost', |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
33 help = 'The host address that runs Orthanc') |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
34 parser.add_argument('--port', type = int, default = '8042', |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
35 help = 'The port number to which Orthanc is listening for the REST API') |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
36 parser.add_argument('--target', default = 'OrthancFiles', |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
37 help = 'The target directory where to store the DICOM files') |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
38 parser.add_argument('--all', action = 'store_true', |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
39 help = 'Replay the entire history on startup (disabled by default)') |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
40 parser.set_defaults(all = False) |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
41 parser.add_argument('--remove', action = 'store_true', |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
42 help = 'Remove DICOM files from Orthanc once classified (disabled by default)') |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
43 parser.set_defaults(remove = False) |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
44 |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
45 |
1183
6ef2c81581cd
more fault-tolerant sample
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
1182
diff
changeset
|
46 def FixPath(p): |
6ef2c81581cd
more fault-tolerant sample
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
1182
diff
changeset
|
47 return p.encode('ascii', 'ignore').strip() |
6ef2c81581cd
more fault-tolerant sample
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
1182
diff
changeset
|
48 |
6ef2c81581cd
more fault-tolerant sample
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
1182
diff
changeset
|
49 def GetTag(resource, tag): |
6ef2c81581cd
more fault-tolerant sample
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
1182
diff
changeset
|
50 if ('MainDicomTags' in resource and |
6ef2c81581cd
more fault-tolerant sample
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
1182
diff
changeset
|
51 tag in resource['MainDicomTags']): |
6ef2c81581cd
more fault-tolerant sample
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
1182
diff
changeset
|
52 return resource['MainDicomTags'][tag] |
6ef2c81581cd
more fault-tolerant sample
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
1182
diff
changeset
|
53 else: |
6ef2c81581cd
more fault-tolerant sample
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
1182
diff
changeset
|
54 return 'No' + tag |
6ef2c81581cd
more fault-tolerant sample
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
1182
diff
changeset
|
55 |
6ef2c81581cd
more fault-tolerant sample
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
1182
diff
changeset
|
56 def ClassifyInstance(instanceId): |
6ef2c81581cd
more fault-tolerant sample
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
1182
diff
changeset
|
57 # Extract the patient, study, series and instance information |
6ef2c81581cd
more fault-tolerant sample
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
1182
diff
changeset
|
58 instance = RestToolbox.DoGet('%s/instances/%s' % (URL, instanceId)) |
6ef2c81581cd
more fault-tolerant sample
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
1182
diff
changeset
|
59 series = RestToolbox.DoGet('%s/series/%s' % (URL, instance['ParentSeries'])) |
6ef2c81581cd
more fault-tolerant sample
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
1182
diff
changeset
|
60 study = RestToolbox.DoGet('%s/studies/%s' % (URL, series['ParentStudy'])) |
6ef2c81581cd
more fault-tolerant sample
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
1182
diff
changeset
|
61 patient = RestToolbox.DoGet('%s/patients/%s' % (URL, study['ParentPatient'])) |
6ef2c81581cd
more fault-tolerant sample
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
1182
diff
changeset
|
62 |
6ef2c81581cd
more fault-tolerant sample
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
1182
diff
changeset
|
63 # Construct a target path |
6ef2c81581cd
more fault-tolerant sample
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
1182
diff
changeset
|
64 a = '%s - %s' % (GetTag(patient, 'PatientID'), |
6ef2c81581cd
more fault-tolerant sample
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
1182
diff
changeset
|
65 GetTag(patient, 'PatientName')) |
6ef2c81581cd
more fault-tolerant sample
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
1182
diff
changeset
|
66 b = GetTag(study, 'StudyDescription') |
6ef2c81581cd
more fault-tolerant sample
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
1182
diff
changeset
|
67 c = '%s - %s' % (GetTag(series, 'Modality'), |
6ef2c81581cd
more fault-tolerant sample
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
1182
diff
changeset
|
68 GetTag(series, 'SeriesDescription')) |
6ef2c81581cd
more fault-tolerant sample
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
1182
diff
changeset
|
69 d = '%s.dcm' % GetTag(instance, 'SOPInstanceUID') |
6ef2c81581cd
more fault-tolerant sample
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
1182
diff
changeset
|
70 |
6ef2c81581cd
more fault-tolerant sample
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
1182
diff
changeset
|
71 p = os.path.join(args.target, FixPath(a), FixPath(b), FixPath(c)) |
6ef2c81581cd
more fault-tolerant sample
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
1182
diff
changeset
|
72 f = os.path.join(p, FixPath(d)) |
6ef2c81581cd
more fault-tolerant sample
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
1182
diff
changeset
|
73 |
6ef2c81581cd
more fault-tolerant sample
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
1182
diff
changeset
|
74 # Copy the DICOM file to the target path |
6ef2c81581cd
more fault-tolerant sample
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
1182
diff
changeset
|
75 print('Writing new DICOM file: %s' % f) |
6ef2c81581cd
more fault-tolerant sample
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
1182
diff
changeset
|
76 |
6ef2c81581cd
more fault-tolerant sample
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
1182
diff
changeset
|
77 try: |
6ef2c81581cd
more fault-tolerant sample
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
1182
diff
changeset
|
78 os.makedirs(p) |
6ef2c81581cd
more fault-tolerant sample
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
1182
diff
changeset
|
79 except: |
6ef2c81581cd
more fault-tolerant sample
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
1182
diff
changeset
|
80 # Already existing directory, ignore the error |
6ef2c81581cd
more fault-tolerant sample
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
1182
diff
changeset
|
81 pass |
6ef2c81581cd
more fault-tolerant sample
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
1182
diff
changeset
|
82 |
6ef2c81581cd
more fault-tolerant sample
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
1182
diff
changeset
|
83 dcm = RestToolbox.DoGet('%s/instances/%s/file' % (URL, instanceId)) |
6ef2c81581cd
more fault-tolerant sample
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
1182
diff
changeset
|
84 with open(f, 'wb') as g: |
6ef2c81581cd
more fault-tolerant sample
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
1182
diff
changeset
|
85 g.write(dcm) |
6ef2c81581cd
more fault-tolerant sample
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
1182
diff
changeset
|
86 |
6ef2c81581cd
more fault-tolerant sample
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
1182
diff
changeset
|
87 |
1182
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
88 # Parse the arguments |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
89 args = parser.parse_args() |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
90 URL = 'http://%s:%d' % (args.host, args.port) |
1183
6ef2c81581cd
more fault-tolerant sample
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
1182
diff
changeset
|
91 print('Connecting to Orthanc on address: %s' % URL) |
1182
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
92 |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
93 # Compute the starting point for the changes loop |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
94 if args.all: |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
95 current = 0 |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
96 else: |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
97 current = RestToolbox.DoGet(URL + '/changes?last')['Last'] |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
98 |
1183
6ef2c81581cd
more fault-tolerant sample
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
1182
diff
changeset
|
99 # Polling loop using the 'changes' API of Orthanc, waiting for the |
1182
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
100 # incoming of new DICOM files |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
101 while True: |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
102 r = RestToolbox.DoGet(URL + '/changes', { |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
103 'since' : current, |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
104 'limit' : 4 # Retrieve at most 4 changes at once |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
105 }) |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
106 |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
107 for change in r['Changes']: |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
108 # We are only interested interested in the arrival of new instances |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
109 if change['ChangeType'] == 'NewInstance': |
1183
6ef2c81581cd
more fault-tolerant sample
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
1182
diff
changeset
|
110 try: |
6ef2c81581cd
more fault-tolerant sample
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
1182
diff
changeset
|
111 ClassifyInstance(change['ID']) |
6ef2c81581cd
more fault-tolerant sample
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
1182
diff
changeset
|
112 except: |
6ef2c81581cd
more fault-tolerant sample
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
1182
diff
changeset
|
113 print('Unable to write instance %s to the disk' % change['ID']) |
1182
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
114 |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
115 # If requested, remove the instance once it has been copied |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
116 if args.remove: |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
117 RestToolbox.DoDelete('%s/instances/%s' % (URL, change['ID'])) |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
118 |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
119 current = r['Last'] |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
120 |
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
121 if r['Done']: |
1183
6ef2c81581cd
more fault-tolerant sample
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
1182
diff
changeset
|
122 print('Everything has been processed: Waiting...') |
1182
d74ac5d0bcaf
New sample: Automated classification of DICOM files
Sebastien Jodogne <s.jodogne@gmail.com>
parents:
diff
changeset
|
123 time.sleep(1) |