Commit cdcc6888d3563c67dc9e3f3a0b539bf3dc65bf25
1 parent
f9dfc67d
Exists in
master
and in
68 other branches
ENH: Preview dicom frame
Showing
2 changed files
with
85 additions
and
41 deletions
Show diff stats
invesalius/data/imagedata_utils.py
... | ... | @@ -173,13 +173,24 @@ def Read(filename): |
173 | 173 | def View(imagedata): |
174 | 174 | viewer = vtk.vtkImageViewer() |
175 | 175 | viewer.SetInput(imagedata) |
176 | - viewer.SetZSlice(10) | |
177 | 176 | viewer.SetColorWindow(200) |
178 | 177 | viewer.SetColorLevel(100) |
179 | 178 | viewer.Render() |
180 | 179 | |
181 | 180 | import time |
182 | 181 | time.sleep(10) |
182 | + | |
183 | +def ViewGDCM(imagedata): | |
184 | + viewer = vtkgdcm.vtkImageColorViewer() | |
185 | + viewer.SetInput(reader.GetOutput()) | |
186 | + viewer.SetColorWindow(500.) | |
187 | + viewer.SetColorLevel(50.) | |
188 | + viewer.Render() | |
189 | + | |
190 | + import time | |
191 | + time.sleep(5) | |
192 | + | |
193 | + | |
183 | 194 | |
184 | 195 | def ExtractVOI(imagedata,xi,xf,yi,yf,zi,zf): |
185 | 196 | """ | ... | ... |
invesalius/gui/dicom_preview_panel.py
... | ... | @@ -37,21 +37,15 @@ NCOLS = 6 |
37 | 37 | MAX_VALUE = NCOLS*NROWS |
38 | 38 | |
39 | 39 | |
40 | -myEVT_SELECT = wx.NewEventType() | |
41 | -# This event occurs when the user select a preview | |
42 | -EVT_SELECT = wx.PyEventBinder(myEVT_SELECT, 1) | |
43 | - | |
44 | -myEVT_SELECT_SERIE = wx.NewEventType() | |
45 | -# This event occurs when the user select a preview | |
46 | -EVT_SELECT_SERIE = wx.PyEventBinder(myEVT_SELECT_SERIE, 1) | |
47 | - | |
48 | - | |
49 | -myEVT_CLICK = wx.NewEventType() | |
50 | -EVT_CLICK = wx.PyEventBinder(myEVT_CLICK, 1) | |
40 | +STR_SIZE = "Image size: %d x %d" | |
41 | +STR_LOCAL = "Location: %.2f" | |
42 | +STR_PATIENT = "%s\n%s" | |
43 | +STR_ACQ = "%s %s\nMade in InVesalius" | |
51 | 44 | |
52 | 45 | class SingleImagePreview(wx.Panel): |
53 | 46 | def __init__(self, parent): |
54 | 47 | wx.Panel.__init__(self, parent, -1) |
48 | + self.__init_gui() | |
55 | 49 | self.__init_vtk() |
56 | 50 | self.filelist = [] |
57 | 51 | self.current_index = 0 |
... | ... | @@ -65,28 +59,32 @@ class SingleImagePreview(wx.Panel): |
65 | 59 | LEFT_UP = (X, Y) = const.TEXT_POSITION |
66 | 60 | text_image_size = vtku.Text() |
67 | 61 | text_image_size.SetPosition(LEFT_UP) |
68 | - text_image_size.SetValue("IMAGE SIZE") | |
62 | + text_image_size.SetValue("image size") | |
63 | + self.text_image_size = text_image_size | |
69 | 64 | |
70 | 65 | LEFT_DOWN = (X, 1-Y) |
71 | 66 | text_image_location = vtku.Text() |
72 | 67 | text_image_location.SetVerticalJustificationToBottom() |
73 | 68 | text_image_location.SetPosition(LEFT_DOWN) |
74 | - text_image_location.SetValue("IMAGE LOCAL") | |
75 | - | |
76 | - value = "ID PATIENT\n TEST" | |
69 | + text_image_location.SetValue("localization") | |
70 | + self.text_image_location = text_image_location | |
71 | + | |
72 | + value = "id\nprotocol" | |
77 | 73 | RIGHT_UP = (1-X, Y) |
78 | 74 | text_patient = vtku.Text() |
79 | 75 | text_patient.SetJustificationToRight() |
80 | 76 | text_patient.SetPosition(RIGHT_UP) |
81 | 77 | text_patient.SetValue(value) |
82 | - | |
83 | - value = "DATE\n Made in InVesalius" | |
78 | + self.text_patient = text_patient | |
79 | + | |
80 | + value = "date time\n Made in InVesalius" | |
84 | 81 | RIGHT_DOWN = (1-X, 1-Y) |
85 | 82 | text_acquisition = vtku.Text() |
86 | 83 | text_acquisition.SetJustificationToRight() |
87 | 84 | text_acquisition.SetVerticalJustificationToBottom() |
88 | 85 | text_acquisition.SetPosition(RIGHT_DOWN) |
89 | 86 | text_acquisition.SetValue(value) |
87 | + self.text_acquisition = text_acquisition | |
90 | 88 | |
91 | 89 | renderer = vtk.vtkRenderer() |
92 | 90 | renderer.AddActor(actor) |
... | ... | @@ -98,8 +96,8 @@ class SingleImagePreview(wx.Panel): |
98 | 96 | |
99 | 97 | style = vtk.vtkInteractorStyleImage() |
100 | 98 | |
101 | - interactor = wxVTKRenderWindowInteractor(self, -1, | |
102 | - size=wx.Size(400,400)) | |
99 | + interactor = wxVTKRenderWindowInteractor(self.panel, -1, | |
100 | + size=wx.Size(380,380)) | |
103 | 101 | interactor.GetRenderWindow().AddRenderer(renderer) |
104 | 102 | interactor.SetInteractorStyle(style) |
105 | 103 | interactor.Render() |
... | ... | @@ -107,7 +105,23 @@ class SingleImagePreview(wx.Panel): |
107 | 105 | |
108 | 106 | sizer = wx.BoxSizer(wx.VERTICAL) |
109 | 107 | sizer.Add(interactor, 1, wx.GROW|wx.EXPAND) |
108 | + sizer.Fit(self.panel) | |
109 | + self.panel.SetSizer(sizer) | |
110 | + self.Layout() | |
111 | + self.Update() | |
112 | + | |
113 | + | |
114 | + def __init_gui(self): | |
115 | + self.panel = wx.Panel(self, -1) | |
116 | + self.slider = wx.Slider(self, 100, 25, 1, 100, (30, 60), (250, -1), | |
117 | + wx.SL_HORIZONTAL | wx.SL_AUTOTICKS) | |
118 | + self.slider.SetTickFreq(1, 1) | |
119 | + | |
120 | + sizer = wx.BoxSizer(wx.VERTICAL) | |
121 | + sizer.Add(self.panel, 20, wx.GROW|wx.EXPAND) | |
122 | + sizer.Add(self.slider, 1, wx.GROW|wx.EXPAND) | |
110 | 123 | sizer.Fit(self) |
124 | + | |
111 | 125 | self.SetSizer(sizer) |
112 | 126 | self.Layout() |
113 | 127 | self.Update() |
... | ... | @@ -123,34 +137,56 @@ class SingleImagePreview(wx.Panel): |
123 | 137 | index = self.current_index |
124 | 138 | dicom = self.dicom_list[index] |
125 | 139 | |
140 | + # UPDATE GUI | |
141 | + ## Text related to size | |
142 | + value = STR_SIZE %(dicom.image.size[0], dicom.image.size[1]) | |
143 | + self.text_image_size.SetValue(value) | |
144 | + | |
145 | + ## Text related to slice position | |
146 | + value = STR_LOCAL %(dicom.image.position[2]) | |
147 | + self.text_image_location.SetValue(value) | |
148 | + | |
149 | + ## Text related to patient/ acquisiiton data | |
150 | + value = STR_PATIENT %(dicom.patient.id,\ | |
151 | + dicom.acquisition.protocol_name) | |
152 | + self.text_patient.SetValue(value) | |
153 | + | |
154 | + ## Text related to acquisition date and time | |
155 | + value = STR_ACQ % (dicom.acquisition.date, | |
156 | + dicom.acquisition.time) | |
157 | + self.text_acquisition.SetValue(value) | |
158 | + | |
159 | + # READ FILE | |
126 | 160 | filename = dicom.image.file |
127 | - window_level = dicom.image.level | |
128 | - window_width = dicom.image.window | |
129 | - | |
130 | 161 | reader = vtkgdcm.vtkGDCMImageReader() |
131 | 162 | reader.SetFileName(filename) |
132 | 163 | |
133 | - #colorer = vtk.vtkImageMapToWindowLevelColors() | |
134 | - #colorer.SetInput(reader.GetOutput()) | |
135 | - #colorer.SetWindow(float(window_width)) | |
136 | - #colorer.SetLevel(float(window_level)) | |
137 | - | |
138 | - a = vtkgdcm.vtkImageColorViewer() | |
139 | - a.SetInput(reader.GetOutput()) | |
140 | - a.SetColorWindow(float(window_width)) | |
141 | - a.SetColorLevel(float(window_level)) | |
142 | - a.Render() | |
143 | - import time | |
144 | - time.sleep(5) | |
145 | - | |
164 | + # ADJUST CONTRAST | |
165 | + window_level = dicom.image.level | |
166 | + window_width = dicom.image.window | |
167 | + colorer = vtk.vtkImageMapToWindowLevelColors() | |
168 | + colorer.SetInput(reader.GetOutput()) | |
169 | + colorer.SetWindow(float(window_width)) | |
170 | + colorer.SetLevel(float(window_level)) | |
146 | 171 | |
147 | - #self.actor.SetInput(colorer.GetOutput()) | |
148 | - self.actor.SetInput(reader.GetOutput()) | |
172 | + # PLOT IMAGE INTO VIEWER | |
173 | + self.actor.SetInput(colorer.GetOutput()) | |
149 | 174 | self.renderer.ResetCamera() |
150 | 175 | self.interactor.Render() |
151 | 176 | |
152 | 177 | |
153 | 178 | |
179 | +myEVT_SELECT = wx.NewEventType() | |
180 | +# This event occurs when the user select a preview | |
181 | +EVT_SELECT = wx.PyEventBinder(myEVT_SELECT, 1) | |
182 | + | |
183 | +myEVT_SELECT_SERIE = wx.NewEventType() | |
184 | +# This event occurs when the user select a preview | |
185 | +EVT_SELECT_SERIE = wx.PyEventBinder(myEVT_SELECT_SERIE, 1) | |
186 | + | |
187 | +myEVT_CLICK = wx.NewEventType() | |
188 | +EVT_CLICK = wx.PyEventBinder(myEVT_CLICK, 1) | |
189 | + | |
154 | 190 | class PreviewEvent(wx.PyCommandEvent): |
155 | 191 | def __init__(self , evtType, id): |
156 | 192 | wx.PyCommandEvent.__init__(self, evtType, id) |
... | ... | @@ -508,7 +544,6 @@ class DicomPreviewSeries(wx.Panel): |
508 | 544 | self.nhidden_last_display = 0 |
509 | 545 | group_list = patient.GetGroups() |
510 | 546 | self.group_list = group_list |
511 | - print "LEN:", len(group_list) | |
512 | 547 | n = 0 |
513 | 548 | for group in group_list: |
514 | 549 | info = (group.dicom.image.file, |
... | ... | @@ -530,11 +565,9 @@ class DicomPreviewSeries(wx.Panel): |
530 | 565 | def _display_previews(self): |
531 | 566 | initial = self.displayed_position * NCOLS |
532 | 567 | final = initial + MAX_VALUE |
533 | - print "len:", len(self.files) | |
534 | 568 | |
535 | 569 | if len(self.files) < final: |
536 | 570 | for i in xrange(final-len(self.files)): |
537 | - print "hide ", i | |
538 | 571 | try: |
539 | 572 | self.previews[-i-1].Hide() |
540 | 573 | except IndexError: | ... | ... |