Commit dd1c83b9b56330fe51d672d5261df8937ec223da
1 parent
aa5e2841
Exists in
master
and in
67 other branches
ADD: Showing patient series in CFind
Showing
3 changed files
with
145 additions
and
20 deletions
Show diff stats
invesalius/gui/import_network_panel.py
... | ... | @@ -219,7 +219,7 @@ class TextPanel(wx.Panel): |
219 | 219 | |
220 | 220 | def __bind_pubsub_evt(self): |
221 | 221 | #ps.Publisher().subscribe(self.SelectSeries, 'Select series in import panel') |
222 | - pass | |
222 | + ps.Publisher().subscribe(self.Populate, 'Populate tree') | |
223 | 223 | |
224 | 224 | def __bind_events_wx(self): |
225 | 225 | self.Bind(wx.EVT_SIZE, self.OnSize) |
... | ... | @@ -268,10 +268,83 @@ class TextPanel(wx.Panel): |
268 | 268 | def SelectSeries(self, pubsub_evt): |
269 | 269 | group_index = pubsub_evt.data |
270 | 270 | |
271 | - def Populate(self, patient_list): | |
271 | + def Populate(self, pubsub_evt): | |
272 | 272 | tree = self.tree |
273 | + #print ">>>>>>>>>>>>>>>>>>>>>>>>>>", dir(tree.GetRootItem()) | |
274 | + #print ">>>>>>>>>>>>",dir(self.tree) | |
275 | + patients = pubsub_evt.data | |
273 | 276 | |
274 | 277 | first = 0 |
278 | + self.idserie_treeitem = {} | |
279 | + | |
280 | + for patient in patients.keys(): | |
281 | + | |
282 | + | |
283 | + #ngroups = patient.ngroups | |
284 | + #dicom = patient.GetDicomSample() | |
285 | + #title = dicom.patient.name + " (%d series)"%(ngroups) | |
286 | + #date_time = "%s %s"%(dicom.acquisition.date, | |
287 | + # dicom.acquisition.time) | |
288 | + | |
289 | + first_serie = patients[patient].keys()[0] | |
290 | + title = patients[patient][first_serie]['name'] | |
291 | + p = patients[patient][first_serie] | |
292 | + | |
293 | + p_id = patient | |
294 | + age = p['age'] | |
295 | + gender = p['gender'] | |
296 | + study_description = p['study_description'] | |
297 | + modality = p['modality'] | |
298 | + date = p['acquisition_date'] | |
299 | + time = p['acquisition_time'] | |
300 | + institution = p['institution'] | |
301 | + birthdate = p['date_of_birth'] | |
302 | + acession_number = p['acession_number'] | |
303 | + physician = p['ref_physician'] | |
304 | + | |
305 | + parent = tree.AppendItem(self.root, title) | |
306 | + | |
307 | + n_amount_images = 0 | |
308 | + for se in patients[patient]: | |
309 | + n_amount_images = n_amount_images + patients[patient][se]['n_images'] | |
310 | + | |
311 | + | |
312 | + tree.SetItemPyData(parent, patient) | |
313 | + tree.SetItemText(parent, "%s" % p_id, 1) | |
314 | + tree.SetItemText(parent, "%s" % age, 2) | |
315 | + tree.SetItemText(parent, "%s" % gender, 3) | |
316 | + tree.SetItemText(parent, "%s" % study_description, 4) | |
317 | + tree.SetItemText(parent, "%s" % modality, 5) | |
318 | + tree.SetItemText(parent, "%s" % date + " " + time, 6) | |
319 | + tree.SetItemText(parent, "%s" % str(n_amount_images), 7) | |
320 | + tree.SetItemText(parent, "%s" % institution, 8) | |
321 | + tree.SetItemText(parent, "%s" % birthdate, 9) | |
322 | + tree.SetItemText(parent, "%s" % acession_number, 10) | |
323 | + tree.SetItemText(parent, "%s" % physician, 11) | |
324 | + | |
325 | + | |
326 | + | |
327 | + for series in patients[patient].keys(): | |
328 | + | |
329 | + serie_description = patients[patient][series]['serie_description'] | |
330 | + n_images = patients[patient][series]['n_images'] | |
331 | + date = patients[patient][series]['acquisition_date'] | |
332 | + time = patients[patient][series]['acquisition_time'] | |
333 | + | |
334 | + child = tree.AppendItem(parent, serie_description) | |
335 | + tree.SetItemPyData(child, series) | |
336 | + | |
337 | + tree.SetItemText(child, "%s" % serie_description, 0) | |
338 | + #tree.SetItemText(child, "%s" % dicom.acquisition.protocol_name, 4) | |
339 | + #tree.SetItemText(child, "%s" % dicom.acquisition.modality, 5) | |
340 | + tree.SetItemText(child, "%s" % date + " " + time, 6) | |
341 | + tree.SetItemText(child, "%s" % n_images , 7) | |
342 | + | |
343 | + self.idserie_treeitem[child] = (patient, series) | |
344 | + | |
345 | + | |
346 | + """ | |
347 | + | |
275 | 348 | for patient in patient_list: |
276 | 349 | if not isinstance(patient, dcm.PatientGroup): |
277 | 350 | return None |
... | ... | @@ -319,7 +392,7 @@ class TextPanel(wx.Panel): |
319 | 392 | tree.Expand(self.root) |
320 | 393 | tree.SelectItem(parent_select) |
321 | 394 | tree.Bind(wx.EVT_TREE_ITEM_ACTIVATED, self.OnActivate) |
322 | - tree.Bind(wx.EVT_TREE_SEL_CHANGED, self.OnSelChanged) | |
395 | + tree.Bind(wx.EVT_TREE_SEL_CHANGED, self.OnSelChanged)""" | |
323 | 396 | |
324 | 397 | def OnSelChanged(self, evt): |
325 | 398 | item = self.tree.GetSelection() |
... | ... | @@ -436,7 +509,8 @@ class FindPanel(wx.Panel): |
436 | 509 | dn.SetAETitleCall(self.hosts[key][3]) |
437 | 510 | dn.SetAETitle(self.hosts[0][3]) |
438 | 511 | dn.SetSearchWord(self.find_txt.GetValue()) |
439 | - print dn.RunCFind() | |
512 | + | |
513 | + ps.Publisher().sendMessage('Populate tree', dn.RunCFind()) | |
440 | 514 | |
441 | 515 | |
442 | 516 | def SetHostsList(self, evt_pub): | ... | ... |
invesalius/net/dicom.py
1 | 1 | import gdcm |
2 | +import utils | |
3 | + | |
2 | 4 | |
3 | 5 | class DicomNet: |
4 | 6 | |
... | ... | @@ -34,6 +36,8 @@ class DicomNet: |
34 | 36 | def GetValueFromDICOM(self, ret, tag): |
35 | 37 | value = str(ret.GetDataElement(gdcm.Tag(tag[0],\ |
36 | 38 | tag[1])).GetValue()) |
39 | + if value == 'None' and tag != (0x0008,0x103E): | |
40 | + value = '' | |
37 | 41 | return value |
38 | 42 | |
39 | 43 | |
... | ... | @@ -46,20 +50,22 @@ class DicomNet: |
46 | 50 | |
47 | 51 | tags = [(0x0010, 0x0010), (0x0010, 0x1010), (0x0010,0x0040), (0x0008,0x1030),\ |
48 | 52 | (0x0008,0x0060), (0x0008,0x0022), (0x0008,0x0080), (0x0010,0x0030),\ |
49 | - (0x0008,0x0050), (0x0008,0x0090)] | |
53 | + (0x0008,0x0050), (0x0008,0x0090), (0x0008,0x103E), (0x0008,0x0033),\ | |
54 | + (0x0008,0x0032)] | |
50 | 55 | |
51 | 56 | |
52 | 57 | ds = gdcm.DataSet() |
53 | 58 | |
54 | 59 | for tag in tags: |
55 | 60 | |
56 | - bit_size = len(self.search_word) + 1 | |
57 | 61 | |
58 | 62 | tg = gdcm.Tag(tag[0], tag[1]) |
59 | 63 | |
60 | 64 | de = gdcm.DataElement(tg) |
61 | 65 | |
62 | - if self.search_type == 'patient': | |
66 | + if self.search_type == 'patient' and tag == (0x0010, 0x0010): | |
67 | + | |
68 | + bit_size = len(self.search_word) + 1 | |
63 | 69 | de.SetByteValue(str(self.search_word + '*'), gdcm.VL(bit_size)) |
64 | 70 | else: |
65 | 71 | de.SetByteValue('*', gdcm.VL(1)) |
... | ... | @@ -90,26 +96,29 @@ class DicomNet: |
90 | 96 | |
91 | 97 | rt = ret[i] |
92 | 98 | |
93 | - | |
94 | - name = self.GetValueFromDICOM(rt, ((0x0010, 0x0010))) | |
95 | - age = self.GetValueFromDICOM(rt, ((0x0010, 0x1010))) | |
96 | - gender = self.GetValueFromDICOM(rt, ((0x0010),(0x0040))) | |
97 | - study_description = self.GetValueFromDICOM(rt, ((0x0008),(0x1030))) | |
98 | - modality = self.GetValueFromDICOM(rt, ((0x0008),(0x0060))) | |
99 | - date_acquisition = self.GetValueFromDICOM(rt, ((0x0008),(0x0022))) | |
100 | - institution = self.GetValueFromDICOM(rt, ((0x0008),(0x0080))) | |
101 | - date_of_birth = self.GetValueFromDICOM(rt, ((0x0010),(0x0030))) | |
102 | - acession_number = self.GetValueFromDICOM(rt, ((0x0008),(0x0050))) | |
103 | - ref_physician = self.GetValueFromDICOM(rt, ((0x0008),(0x0090))) | |
99 | + name = self.GetValueFromDICOM(rt, (0x0010, 0x0010)) | |
100 | + age = self.GetValueFromDICOM(rt, (0x0010, 0x1010)) | |
101 | + gender = self.GetValueFromDICOM(rt, (0x0010,0x0040)) | |
102 | + study_description = self.GetValueFromDICOM(rt, (0x0008,0x1030)) | |
103 | + modality = self.GetValueFromDICOM(rt, (0x0008,0x0060)) | |
104 | + institution = self.GetValueFromDICOM(rt, (0x0008,0x0080)) | |
105 | + date_of_birth = utils.format_date(self.GetValueFromDICOM(rt, (0x0010,0x0030))) | |
106 | + acession_number = self.GetValueFromDICOM(rt, (0x0008,0x0050)) | |
107 | + ref_physician = self.GetValueFromDICOM(rt, (0x0008,0x0090)) | |
108 | + serie_description = self.GetValueFromDICOM(rt, (0x0008,0x103E)) | |
109 | + acquisition_time = utils.format_time(self.GetValueFromDICOM(rt, (0x0008,0x0032))) | |
110 | + acquisition_date = utils.format_date(self.GetValueFromDICOM(rt, (0x0008,0x0022))) | |
104 | 111 | |
105 | 112 | patients[patient_id][serie_id] = {'name':name, 'age':age, 'gender':gender,\ |
106 | 113 | 'study_description':study_description,\ |
107 | 114 | 'modality':modality, \ |
108 | - 'date_acquision':date_acquisition,\ | |
115 | + 'acquisition_time':acquisition_time,\ | |
116 | + 'acquisition_date':acquisition_date,\ | |
109 | 117 | 'institution':institution,\ |
110 | 118 | 'date_of_birth':date_of_birth,\ |
111 | 119 | 'acession_number':acession_number,\ |
112 | - 'ref_physician':ref_physician} | |
120 | + 'ref_physician':ref_physician,\ | |
121 | + 'serie_description':serie_description} | |
113 | 122 | |
114 | 123 | patients[patient_id][serie_id]['n_images'] = 1 |
115 | 124 | else: | ... | ... |
invesalius/utils.py
... | ... | @@ -17,11 +17,53 @@ |
17 | 17 | # detalhes. |
18 | 18 | #-------------------------------------------------------------------------- |
19 | 19 | import platform |
20 | +import time | |
20 | 21 | import sigar |
21 | 22 | import sys |
22 | 23 | import re |
23 | 24 | import locale |
24 | 25 | |
26 | +def format_time(value): | |
27 | + sp1 = value.split(".") | |
28 | + sp2 = value.split(":") | |
29 | + | |
30 | + if (len(sp1) == 2) and (len(sp2) == 3): | |
31 | + new_value = str(sp2[0]+sp2[1]+ | |
32 | + str(int(float(sp2[2])))) | |
33 | + data = time.strptime(new_value, "%H%M%S") | |
34 | + elif (len(sp1) == 2): | |
35 | + data = time.gmtime(float(value)) | |
36 | + elif (len(sp1) > 2): | |
37 | + data = time.strptime(value, "%H.%M.%S") | |
38 | + elif(len(sp2) > 1): | |
39 | + data = time.strptime(value, "%H:%M:%S") | |
40 | + else: | |
41 | + try: | |
42 | + data = time.strptime(value, "%H%M%S") | |
43 | + # If the time is not in a bad format only return it. | |
44 | + except ValueError: | |
45 | + return value | |
46 | + return time.strftime("%H:%M:%S",data) | |
47 | + | |
48 | +def format_date(value): | |
49 | + | |
50 | + sp1 = value.split(".") | |
51 | + try: | |
52 | + | |
53 | + if (len(sp1) > 1): | |
54 | + if (len(sp1[0]) <= 2): | |
55 | + data = time.strptime(value, "%D.%M.%Y") | |
56 | + else: | |
57 | + data = time.strptime(value, "%Y.%M.%d") | |
58 | + elif(len(value.split("//")) > 1): | |
59 | + data = time.strptime(value, "%D/%M/%Y") | |
60 | + else: | |
61 | + data = time.strptime(value, "%Y%M%d") | |
62 | + return time.strftime("%d/%M/%Y",data) | |
63 | + | |
64 | + except(ValueError): | |
65 | + return "" | |
66 | + | |
25 | 67 | def debug(error_str): |
26 | 68 | """ |
27 | 69 | Redirects output to file, or to the terminal | ... | ... |