Commit 16a741dedefd74ed919a4e7c1aa1edb501002079
1 parent
1a0875bc
Exists in
master
and in
6 other branches
ENH: Starting the sync between the treecontrol and preview
Showing
2 changed files
with
346 additions
and
368 deletions
Show diff stats
invesalius/gui/dicom_preview_panel.py
@@ -47,6 +47,49 @@ STR_LOCAL = _("Location: %.2f") | @@ -47,6 +47,49 @@ STR_LOCAL = _("Location: %.2f") | ||
47 | STR_PATIENT = "%s\n%s" | 47 | STR_PATIENT = "%s\n%s" |
48 | STR_ACQ = _("%s %s\nMade in InVesalius") | 48 | STR_ACQ = _("%s %s\nMade in InVesalius") |
49 | 49 | ||
50 | +myEVT_PREVIEW_CLICK = wx.NewEventType() | ||
51 | +EVT_PREVIEW_CLICK = wx.PyEventBinder(myEVT_PREVIEW_CLICK, 1) | ||
52 | + | ||
53 | +myEVT_PREVIEW_DBLCLICK = wx.NewEventType() | ||
54 | +EVT_PREVIEW_DBLCLICK = wx.PyEventBinder(myEVT_PREVIEW_DBLCLICK, 1) | ||
55 | + | ||
56 | +myEVT_CLICK_SLICE = wx.NewEventType() | ||
57 | +# This event occurs when the user select a preview | ||
58 | +EVT_CLICK_SLICE = wx.PyEventBinder(myEVT_CLICK_SLICE, 1) | ||
59 | + | ||
60 | +myEVT_CLICK_SERIE = wx.NewEventType() | ||
61 | +# This event occurs when the user select a preview | ||
62 | +EVT_CLICK_SERIE = wx.PyEventBinder(myEVT_CLICK_SERIE, 1) | ||
63 | + | ||
64 | +myEVT_CLICK = wx.NewEventType() | ||
65 | +EVT_CLICK = wx.PyEventBinder(myEVT_CLICK, 1) | ||
66 | + | ||
67 | +class SelectionEvent(wx.PyCommandEvent): | ||
68 | + pass | ||
69 | + | ||
70 | + | ||
71 | +class PreviewEvent(wx.PyCommandEvent): | ||
72 | + def __init__(self , evtType, id): | ||
73 | + super(PreviewEvent, self).__init__(evtType, id) | ||
74 | + | ||
75 | + def GetSelectID(self): | ||
76 | + return self.SelectedID | ||
77 | + | ||
78 | + def SetSelectedID(self, id): | ||
79 | + self.SelectedID = id | ||
80 | + | ||
81 | + def GetItemData(self): | ||
82 | + return self.data | ||
83 | + | ||
84 | + def SetItemData(self, data): | ||
85 | + self.data = data | ||
86 | + | ||
87 | + | ||
88 | +class SerieEvent(PreviewEvent): | ||
89 | + def __init__(self , evtType, id): | ||
90 | + super(SerieEvent, self).__init__(evtType, id) | ||
91 | + | ||
92 | + | ||
50 | class DicomInfo(object): | 93 | class DicomInfo(object): |
51 | """ | 94 | """ |
52 | Keep the informations and the image used by preview. | 95 | Keep the informations and the image used by preview. |
@@ -57,25 +100,14 @@ class DicomInfo(object): | @@ -57,25 +100,14 @@ class DicomInfo(object): | ||
57 | self.title = title | 100 | self.title = title |
58 | self.subtitle = subtitle | 101 | self.subtitle = subtitle |
59 | self._preview = None | 102 | self._preview = None |
60 | - self._size = (70, 70) | ||
61 | self.selected = False | 103 | self.selected = False |
62 | - self.resized = False | ||
63 | - | ||
64 | - @property | ||
65 | - def size(self): | ||
66 | - return self._size | ||
67 | - | ||
68 | - @size.setter | ||
69 | - def size(self, size): | ||
70 | - if size != self._size: | ||
71 | - self._size = size | ||
72 | - self.resized = True | ||
73 | 104 | ||
74 | @property | 105 | @property |
75 | def preview(self): | 106 | def preview(self): |
76 | if self._preview: | 107 | if self._preview: |
77 | return self._preview | 108 | return self._preview |
78 | else: | 109 | else: |
110 | + print "First time!" | ||
79 | colorer = vtk.vtkImageMapToWindowLevelColors() | 111 | colorer = vtk.vtkImageMapToWindowLevelColors() |
80 | colorer.SetInput(self.dicom.image.imagedata) | 112 | colorer.SetInput(self.dicom.image.imagedata) |
81 | colorer.SetWindow(float(self.dicom.image.window)) | 113 | colorer.SetWindow(float(self.dicom.image.window)) |
@@ -118,7 +150,6 @@ class DicomPaintPanel(wx.Panel): | @@ -118,7 +150,6 @@ class DicomPaintPanel(wx.Panel): | ||
118 | else: | 150 | else: |
119 | return image.Scale(*self.last_size) | 151 | return image.Scale(*self.last_size) |
120 | 152 | ||
121 | - | ||
122 | def SetImage(self, image): | 153 | def SetImage(self, image): |
123 | self.image = image | 154 | self.image = image |
124 | r_img = self._image_resize(image) | 155 | r_img = self._image_resize(image) |
@@ -193,10 +224,9 @@ class Preview(wx.Panel): | @@ -193,10 +224,9 @@ class Preview(wx.Panel): | ||
193 | #self.subtitle.Bind(wx.EVT_LEAVE_WINDOW, self.OnLeave) | 224 | #self.subtitle.Bind(wx.EVT_LEAVE_WINDOW, self.OnLeave) |
194 | 225 | ||
195 | self.Bind(wx.EVT_LEFT_DOWN, self.OnSelect) | 226 | self.Bind(wx.EVT_LEFT_DOWN, self.OnSelect) |
196 | - #self.interactor.Bind(wx.EVT_LEFT_DOWN, self.OnSelect) | ||
197 | - #self.panel.Bind(wx.EVT_LEFT_DOWN, self.OnSelect) | ||
198 | - #self.title.Bind(wx.EVT_LEFT_DOWN, self.OnSelect) | ||
199 | - #self.subtitle.Bind(wx.EVT_LEFT_DOWN, self.OnSelect) | 227 | + self.title.Bind(wx.EVT_LEFT_DOWN, self.OnSelect) |
228 | + self.subtitle.Bind(wx.EVT_LEFT_DOWN, self.OnSelect) | ||
229 | + self.image_viewer.Bind(wx.EVT_LEFT_DOWN, self.OnSelect) | ||
200 | 230 | ||
201 | #self.Bind(wx.EVT_SIZE, self.OnSize) | 231 | #self.Bind(wx.EVT_SIZE, self.OnSize) |
202 | 232 | ||
@@ -252,6 +282,13 @@ class Preview(wx.Panel): | @@ -252,6 +282,13 @@ class Preview(wx.Panel): | ||
252 | #self.SetBackgroundColour(c) | 282 | #self.SetBackgroundColour(c) |
253 | self.Select() | 283 | self.Select() |
254 | 284 | ||
285 | + # Generating a EVT_PREVIEW_CLICK event | ||
286 | + my_evt = SerieEvent(myEVT_PREVIEW_CLICK, self.GetId()) | ||
287 | + my_evt.SetSelectedID(self.dicom_info.id) | ||
288 | + my_evt.SetItemData(self.dicom_info.dicom) | ||
289 | + print "patient", self.dicom_info.dicom.patient | ||
290 | + self.GetEventHandler().ProcessEvent(my_evt) | ||
291 | + | ||
255 | def OnSize(self, evt): | 292 | def OnSize(self, evt): |
256 | if self.dicom_info: | 293 | if self.dicom_info: |
257 | self.SetDicomToPreview(self.dicom_info) | 294 | self.SetDicomToPreview(self.dicom_info) |
@@ -266,313 +303,10 @@ class Preview(wx.Panel): | @@ -266,313 +303,10 @@ class Preview(wx.Panel): | ||
266 | self.Refresh() | 303 | self.Refresh() |
267 | 304 | ||
268 | def OnDClick(self, evt): | 305 | def OnDClick(self, evt): |
269 | - evt = PreviewEvent(myEVT_SELECT, self.GetId()) | ||
270 | - evt.SetSelectedID(self.ID) | ||
271 | - evt.SetItemData(self.data) | ||
272 | - self.GetEventHandler().ProcessEvent(evt) | ||
273 | - | ||
274 | - def ShowShadow(self): | ||
275 | - self._nImgSize = 16 | ||
276 | - nPadding = 4 | ||
277 | - print "ShowShadow" | ||
278 | - dc = wx.BufferedPaintDC(self) | ||
279 | - style = self.GetParent().GetWindowStyleFlag() | ||
280 | - | ||
281 | - backBrush = wx.WHITE_BRUSH | ||
282 | - if 1: #style & INB_BORDER: | ||
283 | - borderPen = wx.Pen(wx.SystemSettings_GetColour(wx.SYS_COLOUR_3DSHADOW)) | ||
284 | - #else: | ||
285 | - # borderPen = wx.TRANSPARENT_PEN | ||
286 | - | ||
287 | - size = self.GetSize() | ||
288 | - | ||
289 | - # Background | ||
290 | - dc.SetBrush(backBrush) | ||
291 | - | ||
292 | - borderPen.SetWidth(1) | ||
293 | - dc.SetPen(borderPen) | ||
294 | - dc.DrawRectangle(0, 0, size.x, size.y) | ||
295 | - #bUsePin = (style & INB_USE_PIN_BUTTON and [True] or [False])[0] | ||
296 | - | ||
297 | - borderPen = wx.BLACK_PEN | ||
298 | - borderPen.SetWidth(1) | ||
299 | - dc.SetPen(borderPen) | ||
300 | - dc.DrawLine(0, size.y, size.x, size.y) | ||
301 | - dc.DrawPoint(0, size.y) | ||
302 | - | ||
303 | - clientSize = 0 | ||
304 | - #bUseYcoord = (style & INB_RIGHT or style & INB_LEFT) | ||
305 | - bUseYcoord = 1 | ||
306 | - | ||
307 | - if bUseYcoord: | ||
308 | - clientSize = size.GetHeight() | ||
309 | - else: | ||
310 | - clientSize = size.GetWidth() | ||
311 | - | ||
312 | - if 1: | ||
313 | - # Default values for the surronounding rectangle | ||
314 | - # around a button | ||
315 | - rectWidth = self._nImgSize * 2 # To avoid the recangle to 'touch' the borders | ||
316 | - rectHeight = self._nImgSize * 2 | ||
317 | - | ||
318 | - # Incase the style requires non-fixed button (fit to text) | ||
319 | - # recalc the rectangle width | ||
320 | - if 1: | ||
321 | - #if style & INB_FIT_BUTTON and \ | ||
322 | - # not ((style & INB_LEFT) or (style & INB_RIGHT)) and \ | ||
323 | - # not self._pagesInfoVec[i].GetCaption() == "" and \ | ||
324 | - # not (style & INB_SHOW_ONLY_IMAGES): | ||
325 | - | ||
326 | - | ||
327 | - #rectWidth = ((textWidth + nPadding * 2) > rectWidth and [nPadding * 2 + textWidth] or [rectWidth])[0] | ||
328 | - | ||
329 | - rectWidth = ((nPadding * 2) > rectWidth and [nPadding * 2] or [rectWidth])[0] | ||
330 | - # Make the width an even number | ||
331 | - if rectWidth % 2 != 0: | ||
332 | - rectWidth += 1 | ||
333 | - | ||
334 | - # If Pin button is used, consider its space as well (applicable for top/botton style) | ||
335 | - # since in the left/right, its size is already considered in 'pos' | ||
336 | - #pinBtnSize = (bUsePin and [20] or [0])[0] | ||
337 | - | ||
338 | - #if pos + rectWidth + pinBtnSize > clientSize: | ||
339 | - # break | ||
340 | - | ||
341 | - # Calculate the button rectangle | ||
342 | - modRectWidth = rectWidth - 2# or [rectWidth])[0] | ||
343 | - modRectHeight = rectHeight# or [rectHeight - 2])[0] | ||
344 | - | ||
345 | - pos = rectWidth | ||
346 | - | ||
347 | - if bUseYcoord: | ||
348 | - buttonRect = wx.Rect(1, pos, modRectWidth, modRectHeight) | ||
349 | - else: | ||
350 | - buttonRect = wx.Rect(pos , 1, modRectWidth, modRectHeight) | ||
351 | - | ||
352 | - def ShowShadow2(self): | ||
353 | - pass | ||
354 | - | ||
355 | - | ||
356 | -class SingleImagePreview(wx.Panel): | ||
357 | - def __init__(self, parent): | ||
358 | - wx.Panel.__init__(self, parent, -1) | ||
359 | - self.__init_gui() | ||
360 | - self.__init_vtk() | ||
361 | - self.__bind_evt_gui() | ||
362 | - self.dicom_list = [] | ||
363 | - self.nimages = 1 | ||
364 | - self.current_index = 0 | ||
365 | - self.window_width = const.WINDOW_LEVEL[_("Bone")][0] | ||
366 | - self.window_level = const.WINDOW_LEVEL[_("Bone")][1] | ||
367 | - | ||
368 | - def __init_vtk(self): | ||
369 | - actor = vtk.vtkImageActor() | ||
370 | - self.actor = actor | ||
371 | - | ||
372 | - text_image_size = vtku.Text() | ||
373 | - text_image_size.SetPosition(const.TEXT_POS_LEFT_UP) | ||
374 | - text_image_size.SetValue(_("image size")) | ||
375 | - self.text_image_size = text_image_size | ||
376 | - | ||
377 | - text_image_location = vtku.Text() | ||
378 | - text_image_location.SetVerticalJustificationToBottom() | ||
379 | - text_image_location.SetPosition(const.TEXT_POS_LEFT_DOWN) | ||
380 | - text_image_location.SetValue("localization") | ||
381 | - self.text_image_location = text_image_location | ||
382 | - | ||
383 | - value = _("id\nprotocol") | ||
384 | - text_patient = vtku.Text() | ||
385 | - text_patient.SetJustificationToRight() | ||
386 | - text_patient.SetPosition(const.TEXT_POS_RIGHT_UP) | ||
387 | - text_patient.SetValue(value) | ||
388 | - self.text_patient = text_patient | ||
389 | - | ||
390 | - value = _("date time\n Made in InVesalius") | ||
391 | - text_acquisition = vtku.Text() | ||
392 | - text_acquisition.SetJustificationToRight() | ||
393 | - text_acquisition.SetVerticalJustificationToBottom() | ||
394 | - text_acquisition.SetPosition(const.TEXT_POS_RIGHT_DOWN) | ||
395 | - text_acquisition.SetValue(value) | ||
396 | - self.text_acquisition = text_acquisition | ||
397 | - | ||
398 | - renderer = vtk.vtkRenderer() | ||
399 | - renderer.AddActor(actor) | ||
400 | - renderer.AddActor(text_image_size.actor) | ||
401 | - renderer.AddActor(text_image_location.actor) | ||
402 | - renderer.AddActor(text_patient.actor) | ||
403 | - renderer.AddActor(text_acquisition.actor) | ||
404 | - self.renderer = renderer | ||
405 | - | ||
406 | - style = vtk.vtkInteractorStyleImage() | ||
407 | - | ||
408 | - interactor = wxVTKRenderWindowInteractor(self.panel, -1, | ||
409 | - size=wx.Size(340,340)) | ||
410 | - interactor.GetRenderWindow().AddRenderer(renderer) | ||
411 | - interactor.SetInteractorStyle(style) | ||
412 | - interactor.Render() | ||
413 | - self.interactor = interactor | ||
414 | - | ||
415 | - sizer = wx.BoxSizer(wx.VERTICAL) | ||
416 | - sizer.Add(interactor, 1, wx.GROW|wx.EXPAND) | ||
417 | - sizer.Fit(self.panel) | ||
418 | - self.panel.SetSizer(sizer) | ||
419 | - self.Layout() | ||
420 | - self.Update() | ||
421 | - | ||
422 | - | ||
423 | - def __init_gui(self): | ||
424 | - self.panel = wx.Panel(self, -1) | ||
425 | - | ||
426 | - slider = wx.Slider(self, | ||
427 | - id=-1, | ||
428 | - value=0, | ||
429 | - minValue=0, | ||
430 | - maxValue=99, | ||
431 | - style=wx.SL_HORIZONTAL|wx.SL_AUTOTICKS) | ||
432 | - slider.SetWindowVariant(wx.WINDOW_VARIANT_SMALL) | ||
433 | - slider.SetTickFreq(1, 1) | ||
434 | - self.slider = slider | ||
435 | - | ||
436 | - checkbox = wx.CheckBox(self, -1, _("Auto-play")) | ||
437 | - self.checkbox = checkbox | ||
438 | - | ||
439 | - in_sizer = wx.BoxSizer(wx.HORIZONTAL) | ||
440 | - in_sizer.Add(slider, 1, wx.GROW|wx.EXPAND) | ||
441 | - in_sizer.Add(checkbox, 0) | ||
442 | - | ||
443 | - sizer = wx.BoxSizer(wx.VERTICAL) | ||
444 | - sizer.Add(self.panel, 20, wx.GROW|wx.EXPAND) | ||
445 | - sizer.Add(in_sizer, 1, wx.GROW|wx.EXPAND) | ||
446 | - sizer.Fit(self) | ||
447 | - | ||
448 | - self.SetSizer(sizer) | ||
449 | - self.Layout() | ||
450 | - self.Update() | ||
451 | - self.SetAutoLayout(1) | ||
452 | - | ||
453 | - def __bind_evt_gui(self): | ||
454 | - self.slider.Bind(wx.EVT_SLIDER, self.OnSlider) | ||
455 | - self.checkbox.Bind(wx.EVT_CHECKBOX, self.OnCheckBox) | ||
456 | - | ||
457 | - def OnSlider(self, evt): | ||
458 | - pos = evt.GetInt() | ||
459 | - self.ShowSlice(pos) | ||
460 | - evt.Skip() | ||
461 | - | ||
462 | - def OnCheckBox(self, evt): | ||
463 | - self.ischecked = evt.IsChecked() | ||
464 | - if evt.IsChecked(): | ||
465 | - wx.CallAfter(self.OnRun) | ||
466 | - evt.Skip() | ||
467 | - | ||
468 | - def OnRun(self): | ||
469 | - pos = self.slider.GetValue() | ||
470 | - pos += 1 | ||
471 | - if not (self.nimages- pos): | ||
472 | - pos = 0 | ||
473 | - self.slider.SetValue(pos) | ||
474 | - self.ShowSlice(pos) | ||
475 | - time.sleep(0.2) | ||
476 | - if self.ischecked: | ||
477 | - try: | ||
478 | - wx.Yield() | ||
479 | - #TODO: temporary fix necessary in the Windows XP 64 Bits | ||
480 | - #BUG in wxWidgets http://trac.wxwidgets.org/ticket/10896 | ||
481 | - except(wx._core.PyAssertionError): | ||
482 | - print "wx._core.PyAssertionError" | ||
483 | - finally: | ||
484 | - wx.CallAfter(self.OnRun) | ||
485 | - | ||
486 | - def SetDicomGroup(self, group): | ||
487 | - self.dicom_list = group.GetHandSortedList() | ||
488 | - self.current_index = 0 | ||
489 | - self.nimages = len(self.dicom_list) | ||
490 | - # GUI | ||
491 | - self.slider.SetMax(self.nimages-1) | ||
492 | - print self.nimages | ||
493 | - self.slider.SetValue(0) | ||
494 | - self.ShowSlice() | ||
495 | - | ||
496 | - def ShowSlice(self, index = 0): | ||
497 | - print "ShowSlice" | ||
498 | - dicom = self.dicom_list[index] | ||
499 | - | ||
500 | - # UPDATE GUI | ||
501 | - ## Text related to size | ||
502 | - value = STR_SIZE %(dicom.image.size[0], dicom.image.size[1]) | ||
503 | - self.text_image_size.SetValue(value) | ||
504 | - | ||
505 | - ## Text related to slice position | ||
506 | - value1 = STR_SPC %(dicom.image.spacing[2]) | ||
507 | - value2 = STR_LOCAL %(dicom.image.position[2]) | ||
508 | - value = "%s\n%s" %(value1, value2) | ||
509 | - self.text_image_location.SetValue(value) | ||
510 | - | ||
511 | - ## Text related to patient/ acquisiiton data | ||
512 | - value = STR_PATIENT %(dicom.patient.id,\ | ||
513 | - dicom.acquisition.protocol_name) | ||
514 | - self.text_patient.SetValue(value) | ||
515 | - | ||
516 | - ## Text related to acquisition date and time | ||
517 | - value = STR_ACQ % (dicom.acquisition.date, | ||
518 | - dicom.acquisition.time) | ||
519 | - self.text_acquisition.SetValue(value) | ||
520 | - | ||
521 | - # READ FILE | ||
522 | - #filename = dicom.image.file | ||
523 | - #reader = vtkgdcm.vtkGDCMImageReader() | ||
524 | - #reader.SetFileName(filename) | ||
525 | - | ||
526 | - # ADJUST CONTRAST | ||
527 | - window_level = dicom.image.level | ||
528 | - window_width = dicom.image.window | ||
529 | - colorer = vtk.vtkImageMapToWindowLevelColors() | ||
530 | - colorer.SetInput(dicom.image.imagedata) | ||
531 | - colorer.SetWindow(float(window_width)) | ||
532 | - colorer.SetLevel(float(window_level)) | ||
533 | - | ||
534 | - # PLOT IMAGE INTO VIEWER | ||
535 | - self.actor.SetInput(colorer.GetOutput()) | ||
536 | - self.renderer.ResetCamera() | ||
537 | - self.interactor.Render() | ||
538 | - | ||
539 | - def __del__(self): | ||
540 | - print "---------> morri" | ||
541 | - | ||
542 | - | ||
543 | - | ||
544 | -myEVT_SELECT = wx.NewEventType() | ||
545 | -# This event occurs when the user select a preview | ||
546 | -EVT_SELECT = wx.PyEventBinder(myEVT_SELECT, 1) | ||
547 | - | ||
548 | -myEVT_SELECT_SERIE = wx.NewEventType() | ||
549 | -# This event occurs when the user select a preview | ||
550 | -EVT_SELECT_SERIE = wx.PyEventBinder(myEVT_SELECT_SERIE, 1) | ||
551 | - | ||
552 | -myEVT_CLICK = wx.NewEventType() | ||
553 | -EVT_CLICK = wx.PyEventBinder(myEVT_CLICK, 1) | ||
554 | - | ||
555 | -class PreviewEvent(wx.PyCommandEvent): | ||
556 | - def __init__(self , evtType, id): | ||
557 | - wx.PyCommandEvent.__init__(self, evtType, id) | ||
558 | - | ||
559 | - def GetSelectID(self): | ||
560 | - return self.SelectedID | ||
561 | - | ||
562 | - def SetSelectedID(self, id): | ||
563 | - self.SelectedID = id | ||
564 | - | ||
565 | - def GetItemData(self): | ||
566 | - return self.data | ||
567 | - | ||
568 | - def SetItemData(self, data): | ||
569 | - self.data = data | ||
570 | - | ||
571 | - | ||
572 | -class SerieEvent(PreviewEvent): | ||
573 | - def __init__(self , evtType, id): | ||
574 | - super(SerieEvent, self).__init__(evtType, id) | ||
575 | - | 306 | + my_evt = SerieEvent(myEVT_PREVIEW_DBLCLICK, self.GetId()) |
307 | + my_evt.SetSelectedID(self.dicom_info.id) | ||
308 | + my_evt.SetItemData(self.dicom_info.dicom) | ||
309 | + self.GetEventHandler().ProcessEvent(my_evt) | ||
576 | 310 | ||
577 | 311 | ||
578 | class DicomPreviewSeries(wx.Panel): | 312 | class DicomPreviewSeries(wx.Panel): |
@@ -588,7 +322,6 @@ class DicomPreviewSeries(wx.Panel): | @@ -588,7 +322,6 @@ class DicomPreviewSeries(wx.Panel): | ||
588 | self._init_ui() | 322 | self._init_ui() |
589 | 323 | ||
590 | def _init_ui(self): | 324 | def _init_ui(self): |
591 | - | ||
592 | scroll = wx.ScrollBar(self, -1, style=wx.SB_VERTICAL) | 325 | scroll = wx.ScrollBar(self, -1, style=wx.SB_VERTICAL) |
593 | self.scroll = scroll | 326 | self.scroll = scroll |
594 | 327 | ||
@@ -603,7 +336,6 @@ class DicomPreviewSeries(wx.Panel): | @@ -603,7 +336,6 @@ class DicomPreviewSeries(wx.Panel): | ||
603 | self.SetSizer(background_sizer) | 336 | self.SetSizer(background_sizer) |
604 | background_sizer.Fit(self) | 337 | background_sizer.Fit(self) |
605 | 338 | ||
606 | - | ||
607 | self.Layout() | 339 | self.Layout() |
608 | self.Update() | 340 | self.Update() |
609 | self.SetAutoLayout(1) | 341 | self.SetAutoLayout(1) |
@@ -618,6 +350,7 @@ class DicomPreviewSeries(wx.Panel): | @@ -618,6 +350,7 @@ class DicomPreviewSeries(wx.Panel): | ||
618 | for i in xrange(NROWS): | 350 | for i in xrange(NROWS): |
619 | for j in xrange(NCOLS): | 351 | for j in xrange(NCOLS): |
620 | p = Preview(self) | 352 | p = Preview(self) |
353 | + p.Bind(EVT_PREVIEW_CLICK, self.OnSelect) | ||
621 | #if (i == j == 0): | 354 | #if (i == j == 0): |
622 | #self._show_shadow(p) | 355 | #self._show_shadow(p) |
623 | #p.Hide() | 356 | #p.Hide() |
@@ -627,16 +360,15 @@ class DicomPreviewSeries(wx.Panel): | @@ -627,16 +360,15 @@ class DicomPreviewSeries(wx.Panel): | ||
627 | #def _show_shadow(self, preview): | 360 | #def _show_shadow(self, preview): |
628 | # preview.ShowShadow() | 361 | # preview.ShowShadow() |
629 | 362 | ||
630 | - | ||
631 | def _bind_events(self): | 363 | def _bind_events(self): |
632 | # When the user scrolls the window | 364 | # When the user scrolls the window |
633 | self.Bind(wx.EVT_SCROLL, self.OnScroll) | 365 | self.Bind(wx.EVT_SCROLL, self.OnScroll) |
634 | - self.Bind(EVT_SELECT, self.OnSelect) | ||
635 | 366 | ||
636 | def OnSelect(self, evt): | 367 | def OnSelect(self, evt): |
637 | - my_evt = SerieEvent(myEVT_SELECT_SERIE, self.GetId()) | 368 | + print dir(evt) |
369 | + my_evt = SerieEvent(myEVT_CLICK_SERIE, self.GetId()) | ||
638 | my_evt.SetSelectedID(evt.GetSelectID()) | 370 | my_evt.SetSelectedID(evt.GetSelectID()) |
639 | - my_evt.SetItemData(self.group_list) | 371 | + my_evt.SetItemData(evt.GetItemData()) |
640 | self.GetEventHandler().ProcessEvent(my_evt) | 372 | self.GetEventHandler().ProcessEvent(my_evt) |
641 | 373 | ||
642 | def SetPatientGroups(self, patient): | 374 | def SetPatientGroups(self, patient): |
@@ -647,14 +379,6 @@ class DicomPreviewSeries(wx.Panel): | @@ -647,14 +379,6 @@ class DicomPreviewSeries(wx.Panel): | ||
647 | self.group_list = group_list | 379 | self.group_list = group_list |
648 | n = 0 | 380 | n = 0 |
649 | for group in group_list: | 381 | for group in group_list: |
650 | - #info = (group.dicom.image, | ||
651 | - # float(group.dicom.image.window), | ||
652 | - # float(group.dicom.image.level), | ||
653 | - # group.title, | ||
654 | - # _("%d Images") %(group.nslices), | ||
655 | - # n, | ||
656 | - # group_list, | ||
657 | - # group.dicom) | ||
658 | info = DicomInfo(n, group.dicom, | 382 | info = DicomInfo(n, group.dicom, |
659 | group.title, | 383 | group.title, |
660 | _("%d Images") %(group.nslices), | 384 | _("%d Images") %(group.nslices), |
@@ -690,8 +414,6 @@ class DicomPreviewSeries(wx.Panel): | @@ -690,8 +414,6 @@ class DicomPreviewSeries(wx.Panel): | ||
690 | pass | 414 | pass |
691 | self.nhidden_last_display = 0 | 415 | self.nhidden_last_display = 0 |
692 | 416 | ||
693 | - | ||
694 | - | ||
695 | for f, p in zip(self.files[initial:final], self.previews): | 417 | for f, p in zip(self.files[initial:final], self.previews): |
696 | #print "f", f | 418 | #print "f", f |
697 | p.SetDicomToPreview(f) | 419 | p.SetDicomToPreview(f) |
@@ -700,23 +422,22 @@ class DicomPreviewSeries(wx.Panel): | @@ -700,23 +422,22 @@ class DicomPreviewSeries(wx.Panel): | ||
700 | for f, p in zip(self.files[initial:final], self.previews): | 422 | for f, p in zip(self.files[initial:final], self.previews): |
701 | p.Show() | 423 | p.Show() |
702 | 424 | ||
703 | - | ||
704 | def OnScroll(self, evt): | 425 | def OnScroll(self, evt): |
705 | if self.displayed_position != evt.GetPosition(): | 426 | if self.displayed_position != evt.GetPosition(): |
706 | self.displayed_position = evt.GetPosition() | 427 | self.displayed_position = evt.GetPosition() |
707 | self._display_previews() | 428 | self._display_previews() |
708 | 429 | ||
709 | -class DicomPreview(wx.Panel): | 430 | + |
431 | +class DicomPreviewSlice(wx.Panel): | ||
710 | """A dicom preview panel""" | 432 | """A dicom preview panel""" |
711 | def __init__(self, parent): | 433 | def __init__(self, parent): |
712 | - super(DicomPreview, self).__init__(parent) | 434 | + super(DicomPreviewSlice, self).__init__(parent) |
713 | # TODO: 3 pixels between the previews is a good idea? | 435 | # TODO: 3 pixels between the previews is a good idea? |
714 | # I have to test. | 436 | # I have to test. |
715 | self.displayed_position = 0 | 437 | self.displayed_position = 0 |
716 | self.nhidden_last_display = 0 | 438 | self.nhidden_last_display = 0 |
717 | self._init_ui() | 439 | self._init_ui() |
718 | 440 | ||
719 | - | ||
720 | def _init_ui(self): | 441 | def _init_ui(self): |
721 | scroll = wx.ScrollBar(self, -1, style=wx.SB_VERTICAL) | 442 | scroll = wx.ScrollBar(self, -1, style=wx.SB_VERTICAL) |
722 | self.scroll = scroll | 443 | self.scroll = scroll |
@@ -732,7 +453,6 @@ class DicomPreview(wx.Panel): | @@ -732,7 +453,6 @@ class DicomPreview(wx.Panel): | ||
732 | self.SetSizer(background_sizer) | 453 | self.SetSizer(background_sizer) |
733 | background_sizer.Fit(self) | 454 | background_sizer.Fit(self) |
734 | 455 | ||
735 | - | ||
736 | self.Layout() | 456 | self.Layout() |
737 | self.Update() | 457 | self.Update() |
738 | self.SetAutoLayout(1) | 458 | self.SetAutoLayout(1) |
@@ -747,6 +467,7 @@ class DicomPreview(wx.Panel): | @@ -747,6 +467,7 @@ class DicomPreview(wx.Panel): | ||
747 | for i in xrange(NROWS): | 467 | for i in xrange(NROWS): |
748 | for j in xrange(NCOLS): | 468 | for j in xrange(NCOLS): |
749 | p = Preview(self) | 469 | p = Preview(self) |
470 | + p.Bind(EVT_PREVIEW_CLICK, self.OnPreviewClick) | ||
750 | #p.Hide() | 471 | #p.Hide() |
751 | self.previews.append(p) | 472 | self.previews.append(p) |
752 | self.grid.Add(p, 1, flag=wx.EXPAND) | 473 | self.grid.Add(p, 1, flag=wx.EXPAND) |
@@ -839,8 +560,201 @@ class DicomPreview(wx.Panel): | @@ -839,8 +560,201 @@ class DicomPreview(wx.Panel): | ||
839 | for f, p in zip(self.files[initial:final], self.previews): | 560 | for f, p in zip(self.files[initial:final], self.previews): |
840 | p.Show() | 561 | p.Show() |
841 | 562 | ||
842 | - | ||
843 | def OnScroll(self, evt): | 563 | def OnScroll(self, evt): |
844 | if self.displayed_position != evt.GetPosition(): | 564 | if self.displayed_position != evt.GetPosition(): |
845 | self.displayed_position = evt.GetPosition() | 565 | self.displayed_position = evt.GetPosition() |
846 | self._display_previews() | 566 | self._display_previews() |
567 | + | ||
568 | + def OnPreviewClick(self, evt): | ||
569 | + print "Hey man, you've clicked over me" | ||
570 | + my_evt = SerieEvent(myEVT_CLICK_SLICE, self.GetId()) | ||
571 | + my_evt.SetSelectedID(evt.GetSelectID()) | ||
572 | + my_evt.SetItemData(evt.GetItemData()) | ||
573 | + self.GetEventHandler().ProcessEvent(my_evt) | ||
574 | + | ||
575 | + | ||
576 | +class SingleImagePreview(wx.Panel): | ||
577 | + def __init__(self, parent): | ||
578 | + wx.Panel.__init__(self, parent, -1) | ||
579 | + self.__init_gui() | ||
580 | + self.__init_vtk() | ||
581 | + self.__bind_evt_gui() | ||
582 | + self.dicom_list = [] | ||
583 | + self.nimages = 1 | ||
584 | + self.current_index = 0 | ||
585 | + self.window_width = const.WINDOW_LEVEL[_("Bone")][0] | ||
586 | + self.window_level = const.WINDOW_LEVEL[_("Bone")][1] | ||
587 | + | ||
588 | + def __init_vtk(self): | ||
589 | + actor = vtk.vtkImageActor() | ||
590 | + self.actor = actor | ||
591 | + | ||
592 | + text_image_size = vtku.Text() | ||
593 | + text_image_size.SetPosition(const.TEXT_POS_LEFT_UP) | ||
594 | + text_image_size.SetValue(_("image size")) | ||
595 | + self.text_image_size = text_image_size | ||
596 | + | ||
597 | + text_image_location = vtku.Text() | ||
598 | + text_image_location.SetVerticalJustificationToBottom() | ||
599 | + text_image_location.SetPosition(const.TEXT_POS_LEFT_DOWN) | ||
600 | + text_image_location.SetValue("localization") | ||
601 | + self.text_image_location = text_image_location | ||
602 | + | ||
603 | + value = _("id\nprotocol") | ||
604 | + text_patient = vtku.Text() | ||
605 | + text_patient.SetJustificationToRight() | ||
606 | + text_patient.SetPosition(const.TEXT_POS_RIGHT_UP) | ||
607 | + text_patient.SetValue(value) | ||
608 | + self.text_patient = text_patient | ||
609 | + | ||
610 | + value = _("date time\n Made in InVesalius") | ||
611 | + text_acquisition = vtku.Text() | ||
612 | + text_acquisition.SetJustificationToRight() | ||
613 | + text_acquisition.SetVerticalJustificationToBottom() | ||
614 | + text_acquisition.SetPosition(const.TEXT_POS_RIGHT_DOWN) | ||
615 | + text_acquisition.SetValue(value) | ||
616 | + self.text_acquisition = text_acquisition | ||
617 | + | ||
618 | + renderer = vtk.vtkRenderer() | ||
619 | + renderer.AddActor(actor) | ||
620 | + renderer.AddActor(text_image_size.actor) | ||
621 | + renderer.AddActor(text_image_location.actor) | ||
622 | + renderer.AddActor(text_patient.actor) | ||
623 | + renderer.AddActor(text_acquisition.actor) | ||
624 | + self.renderer = renderer | ||
625 | + | ||
626 | + style = vtk.vtkInteractorStyleImage() | ||
627 | + | ||
628 | + interactor = wxVTKRenderWindowInteractor(self.panel, -1, | ||
629 | + size=wx.Size(340,340)) | ||
630 | + interactor.GetRenderWindow().AddRenderer(renderer) | ||
631 | + interactor.SetInteractorStyle(style) | ||
632 | + interactor.Render() | ||
633 | + self.interactor = interactor | ||
634 | + | ||
635 | + sizer = wx.BoxSizer(wx.VERTICAL) | ||
636 | + sizer.Add(interactor, 1, wx.GROW|wx.EXPAND) | ||
637 | + sizer.Fit(self.panel) | ||
638 | + self.panel.SetSizer(sizer) | ||
639 | + self.Layout() | ||
640 | + self.Update() | ||
641 | + | ||
642 | + def __init_gui(self): | ||
643 | + self.panel = wx.Panel(self, -1) | ||
644 | + | ||
645 | + slider = wx.Slider(self, | ||
646 | + id=-1, | ||
647 | + value=0, | ||
648 | + minValue=0, | ||
649 | + maxValue=99, | ||
650 | + style=wx.SL_HORIZONTAL|wx.SL_AUTOTICKS) | ||
651 | + slider.SetWindowVariant(wx.WINDOW_VARIANT_SMALL) | ||
652 | + slider.SetTickFreq(1, 1) | ||
653 | + self.slider = slider | ||
654 | + | ||
655 | + checkbox = wx.CheckBox(self, -1, _("Auto-play")) | ||
656 | + self.checkbox = checkbox | ||
657 | + | ||
658 | + in_sizer = wx.BoxSizer(wx.HORIZONTAL) | ||
659 | + in_sizer.Add(slider, 1, wx.GROW|wx.EXPAND) | ||
660 | + in_sizer.Add(checkbox, 0) | ||
661 | + | ||
662 | + sizer = wx.BoxSizer(wx.VERTICAL) | ||
663 | + sizer.Add(self.panel, 20, wx.GROW|wx.EXPAND) | ||
664 | + sizer.Add(in_sizer, 1, wx.GROW|wx.EXPAND) | ||
665 | + sizer.Fit(self) | ||
666 | + | ||
667 | + self.SetSizer(sizer) | ||
668 | + self.Layout() | ||
669 | + self.Update() | ||
670 | + self.SetAutoLayout(1) | ||
671 | + | ||
672 | + def __bind_evt_gui(self): | ||
673 | + self.slider.Bind(wx.EVT_SLIDER, self.OnSlider) | ||
674 | + self.checkbox.Bind(wx.EVT_CHECKBOX, self.OnCheckBox) | ||
675 | + | ||
676 | + def OnSlider(self, evt): | ||
677 | + pos = evt.GetInt() | ||
678 | + self.ShowSlice(pos) | ||
679 | + evt.Skip() | ||
680 | + | ||
681 | + def OnCheckBox(self, evt): | ||
682 | + self.ischecked = evt.IsChecked() | ||
683 | + if evt.IsChecked(): | ||
684 | + wx.CallAfter(self.OnRun) | ||
685 | + evt.Skip() | ||
686 | + | ||
687 | + def OnRun(self): | ||
688 | + pos = self.slider.GetValue() | ||
689 | + pos += 1 | ||
690 | + if not (self.nimages- pos): | ||
691 | + pos = 0 | ||
692 | + self.slider.SetValue(pos) | ||
693 | + self.ShowSlice(pos) | ||
694 | + time.sleep(0.2) | ||
695 | + if self.ischecked: | ||
696 | + try: | ||
697 | + wx.Yield() | ||
698 | + #TODO: temporary fix necessary in the Windows XP 64 Bits | ||
699 | + #BUG in wxWidgets http://trac.wxwidgets.org/ticket/10896 | ||
700 | + except(wx._core.PyAssertionError): | ||
701 | + print "wx._core.PyAssertionError" | ||
702 | + finally: | ||
703 | + wx.CallAfter(self.OnRun) | ||
704 | + | ||
705 | + def SetDicomGroup(self, group): | ||
706 | + self.dicom_list = group.GetHandSortedList() | ||
707 | + self.current_index = 0 | ||
708 | + self.nimages = len(self.dicom_list) | ||
709 | + # GUI | ||
710 | + self.slider.SetMax(self.nimages-1) | ||
711 | + print self.nimages | ||
712 | + self.slider.SetValue(0) | ||
713 | + self.ShowSlice() | ||
714 | + | ||
715 | + def ShowSlice(self, index = 0): | ||
716 | + print "ShowSlice" | ||
717 | + dicom = self.dicom_list[index] | ||
718 | + | ||
719 | + # UPDATE GUI | ||
720 | + ## Text related to size | ||
721 | + value = STR_SIZE %(dicom.image.size[0], dicom.image.size[1]) | ||
722 | + self.text_image_size.SetValue(value) | ||
723 | + | ||
724 | + ## Text related to slice position | ||
725 | + value1 = STR_SPC %(dicom.image.spacing[2]) | ||
726 | + value2 = STR_LOCAL %(dicom.image.position[2]) | ||
727 | + value = "%s\n%s" %(value1, value2) | ||
728 | + self.text_image_location.SetValue(value) | ||
729 | + | ||
730 | + ## Text related to patient/ acquisiiton data | ||
731 | + value = STR_PATIENT %(dicom.patient.id,\ | ||
732 | + dicom.acquisition.protocol_name) | ||
733 | + self.text_patient.SetValue(value) | ||
734 | + | ||
735 | + ## Text related to acquisition date and time | ||
736 | + value = STR_ACQ % (dicom.acquisition.date, | ||
737 | + dicom.acquisition.time) | ||
738 | + self.text_acquisition.SetValue(value) | ||
739 | + | ||
740 | + # READ FILE | ||
741 | + #filename = dicom.image.file | ||
742 | + #reader = vtkgdcm.vtkGDCMImageReader() | ||
743 | + #reader.SetFileName(filename) | ||
744 | + | ||
745 | + # ADJUST CONTRAST | ||
746 | + window_level = dicom.image.level | ||
747 | + window_width = dicom.image.window | ||
748 | + colorer = vtk.vtkImageMapToWindowLevelColors() | ||
749 | + colorer.SetInput(dicom.image.imagedata) | ||
750 | + colorer.SetWindow(float(window_width)) | ||
751 | + colorer.SetLevel(float(window_level)) | ||
752 | + | ||
753 | + # PLOT IMAGE INTO VIEWER | ||
754 | + self.actor.SetInput(colorer.GetOutput()) | ||
755 | + self.renderer.ResetCamera() | ||
756 | + self.interactor.Render() | ||
757 | + | ||
758 | + def __del__(self): | ||
759 | + print "---------> morri" | ||
760 | + |
invesalius/gui/import_panel.py
@@ -24,6 +24,29 @@ import wx.lib.splitter as spl | @@ -24,6 +24,29 @@ import wx.lib.splitter as spl | ||
24 | import dicom_preview_panel as dpp | 24 | import dicom_preview_panel as dpp |
25 | import reader.dicom_grouper as dcm | 25 | import reader.dicom_grouper as dcm |
26 | 26 | ||
27 | +myEVT_SELECT_SERIE = wx.NewEventType() | ||
28 | +EVT_SELECT_SERIE = wx.PyEventBinder(myEVT_SELECT_SERIE, 1) | ||
29 | + | ||
30 | +myEVT_SELECT_SLICE = wx.NewEventType() | ||
31 | +EVT_SELECT_SLICE = wx.PyEventBinder(myEVT_SELECT_SLICE, 1) | ||
32 | + | ||
33 | +class SelectEvent(wx.PyCommandEvent): | ||
34 | + def __init__(self , evtType, id): | ||
35 | + super(SelectEvent, self).__init__(evtType, id) | ||
36 | + | ||
37 | + def GetSelectID(self): | ||
38 | + return self.SelectedID | ||
39 | + | ||
40 | + def SetSelectedID(self, id): | ||
41 | + self.SelectedID = id | ||
42 | + | ||
43 | + def GetItemData(self): | ||
44 | + return self.data | ||
45 | + | ||
46 | + def SetItemData(self, data): | ||
47 | + self.data = data | ||
48 | + | ||
49 | + | ||
27 | class Panel(wx.Panel): | 50 | class Panel(wx.Panel): |
28 | def __init__(self, parent): | 51 | def __init__(self, parent): |
29 | wx.Panel.__init__(self, parent, pos=wx.Point(5, 5))#, | 52 | wx.Panel.__init__(self, parent, pos=wx.Point(5, 5))#, |
@@ -74,14 +97,25 @@ class InnerPanel(wx.Panel): | @@ -74,14 +97,25 @@ class InnerPanel(wx.Panel): | ||
74 | self.image_panel = ImagePanel(splitter) | 97 | self.image_panel = ImagePanel(splitter) |
75 | splitter.AppendWindow(self.image_panel, 250) | 98 | splitter.AppendWindow(self.image_panel, 250) |
76 | 99 | ||
77 | - self.__bind_evt() | 100 | + self._bind_events() |
101 | + self._bind_pubsubevt() | ||
78 | 102 | ||
79 | - def __bind_evt(self): | 103 | + def _bind_pubsubevt(self): |
80 | ps.Publisher().subscribe(self.ShowDicomPreview, "Load import panel") | 104 | ps.Publisher().subscribe(self.ShowDicomPreview, "Load import panel") |
105 | + | ||
106 | + def _bind_events(self): | ||
107 | + self.Bind(EVT_SELECT_SERIE, self.OnSelectSerie) | ||
108 | + self.Bind(EVT_SELECT_SLICE, self.OnSelectSlice) | ||
81 | 109 | ||
82 | def ShowDicomPreview(self, pubsub_evt): | 110 | def ShowDicomPreview(self, pubsub_evt): |
83 | dicom_groups = pubsub_evt.data | 111 | dicom_groups = pubsub_evt.data |
84 | self.text_panel.Populate(dicom_groups) | 112 | self.text_panel.Populate(dicom_groups) |
113 | + | ||
114 | + def OnSelectSerie(self, evt): | ||
115 | + print "You've selected the serie", evt.GetSelectID() | ||
116 | + | ||
117 | + def OnSelectSlice(self, evt): | ||
118 | + print "You've selected the slice", evt.GetSelectID() | ||
85 | 119 | ||
86 | 120 | ||
87 | class TextPanel(wx.Panel): | 121 | class TextPanel(wx.Panel): |
@@ -141,6 +175,7 @@ class TextPanel(wx.Panel): | @@ -141,6 +175,7 @@ class TextPanel(wx.Panel): | ||
141 | 175 | ||
142 | def Populate(self, patient_list): | 176 | def Populate(self, patient_list): |
143 | tree = self.tree | 177 | tree = self.tree |
178 | + self.idserie_treeitem = {} | ||
144 | 179 | ||
145 | first = 0 | 180 | first = 0 |
146 | for patient in patient_list: | 181 | for patient in patient_list: |
@@ -172,7 +207,7 @@ class TextPanel(wx.Panel): | @@ -172,7 +207,7 @@ class TextPanel(wx.Panel): | ||
172 | tree.SetItemText(parent, "%s" % dicom.patient.physician, 11) | 207 | tree.SetItemText(parent, "%s" % dicom.patient.physician, 11) |
173 | 208 | ||
174 | group_list = patient.GetGroups() | 209 | group_list = patient.GetGroups() |
175 | - for group in group_list: | 210 | + for n, group in enumerate(group_list): |
176 | dicom = group.GetDicomSample() | 211 | dicom = group.GetDicomSample() |
177 | 212 | ||
178 | child = tree.AppendItem(parent, group.title) | 213 | child = tree.AppendItem(parent, group.title) |
@@ -184,14 +219,16 @@ class TextPanel(wx.Panel): | @@ -184,14 +219,16 @@ class TextPanel(wx.Panel): | ||
184 | tree.SetItemText(child, "%s" % date_time, 6) | 219 | tree.SetItemText(child, "%s" % date_time, 6) |
185 | tree.SetItemText(child, "%s" % group.nslices, 7) | 220 | tree.SetItemText(child, "%s" % group.nslices, 7) |
186 | 221 | ||
222 | + self.idserie_treeitem[n] = child | ||
223 | + | ||
187 | tree.Expand(self.root) | 224 | tree.Expand(self.root) |
188 | 225 | ||
189 | tree.SelectItem(parent_select) | 226 | tree.SelectItem(parent_select) |
227 | + print "parent select", parent_select | ||
190 | 228 | ||
191 | tree.Bind(wx.EVT_TREE_ITEM_ACTIVATED, self.OnActivate) | 229 | tree.Bind(wx.EVT_TREE_ITEM_ACTIVATED, self.OnActivate) |
192 | tree.Bind(wx.EVT_TREE_SEL_CHANGED, self.OnSelChanged) | 230 | tree.Bind(wx.EVT_TREE_SEL_CHANGED, self.OnSelChanged) |
193 | 231 | ||
194 | - | ||
195 | def OnSelChanged(self, evt): | 232 | def OnSelChanged(self, evt): |
196 | item = self.tree.GetSelection() | 233 | item = self.tree.GetSelection() |
197 | group = self.tree.GetItemPyData(item) | 234 | group = self.tree.GetItemPyData(item) |
@@ -202,7 +239,6 @@ class TextPanel(wx.Panel): | @@ -202,7 +239,6 @@ class TextPanel(wx.Panel): | ||
202 | ps.Publisher().sendMessage('Load patient into import panel', | 239 | ps.Publisher().sendMessage('Load patient into import panel', |
203 | group) | 240 | group) |
204 | 241 | ||
205 | - | ||
206 | def OnActivate(self, evt): | 242 | def OnActivate(self, evt): |
207 | item = evt.GetItem() | 243 | item = evt.GetItem() |
208 | group = self.tree.GetItemPyData(item) | 244 | group = self.tree.GetItemPyData(item) |
@@ -218,15 +254,23 @@ class TextPanel(wx.Panel): | @@ -218,15 +254,23 @@ class TextPanel(wx.Panel): | ||
218 | def OnSize(self, evt): | 254 | def OnSize(self, evt): |
219 | self.tree.SetSize(self.GetSize()) | 255 | self.tree.SetSize(self.GetSize()) |
220 | 256 | ||
257 | + def SelectSerie(self, serie): | ||
258 | + item = self.idserie_treeitem[serie] | ||
259 | + self.tree.SelectItem(item) | ||
260 | + | ||
221 | 261 | ||
222 | class ImagePanel(wx.Panel): | 262 | class ImagePanel(wx.Panel): |
223 | def __init__(self, parent): | 263 | def __init__(self, parent): |
224 | wx.Panel.__init__(self, parent, -1) | 264 | wx.Panel.__init__(self, parent, -1) |
225 | - #self.SetBackgroundColour((0,255,0)) | ||
226 | - | 265 | + self._init_ui() |
266 | + self._bind_events() | ||
267 | + | ||
268 | + def _init_ui(self): | ||
227 | splitter = spl.MultiSplitterWindow(self, style=wx.SP_LIVE_UPDATE) | 269 | splitter = spl.MultiSplitterWindow(self, style=wx.SP_LIVE_UPDATE) |
228 | splitter.SetOrientation(wx.HORIZONTAL) | 270 | splitter.SetOrientation(wx.HORIZONTAL) |
229 | self.splitter = splitter | 271 | self.splitter = splitter |
272 | + | ||
273 | + splitter.ContainingSizer = wx.BoxSizer(wx.HORIZONTAL) | ||
230 | 274 | ||
231 | sizer = wx.BoxSizer(wx.HORIZONTAL) | 275 | sizer = wx.BoxSizer(wx.HORIZONTAL) |
232 | sizer.Add(splitter, 1, wx.EXPAND) | 276 | sizer.Add(splitter, 1, wx.EXPAND) |
@@ -238,13 +282,27 @@ class ImagePanel(wx.Panel): | @@ -238,13 +282,27 @@ class ImagePanel(wx.Panel): | ||
238 | self.image_panel = SlicePanel(splitter) | 282 | self.image_panel = SlicePanel(splitter) |
239 | splitter.AppendWindow(self.image_panel, 250) | 283 | splitter.AppendWindow(self.image_panel, 250) |
240 | 284 | ||
241 | - | ||
242 | self.SetSizer(sizer) | 285 | self.SetSizer(sizer) |
243 | sizer.Fit(self) | 286 | sizer.Fit(self) |
244 | 287 | ||
245 | self.Layout() | 288 | self.Layout() |
246 | self.Update() | 289 | self.Update() |
247 | self.SetAutoLayout(1) | 290 | self.SetAutoLayout(1) |
291 | + | ||
292 | + def _bind_events(self): | ||
293 | + self.text_panel.Bind(EVT_SELECT_SERIE, self.OnSelectSerie) | ||
294 | + self.text_panel.Bind(EVT_SELECT_SLICE, self.OnSelectSlice) | ||
295 | + | ||
296 | + def OnSelectSerie(self, evt): | ||
297 | + print "Hi, You selected Serie" | ||
298 | + evt.Skip() | ||
299 | + | ||
300 | + def OnSelectSlice(self, evt): | ||
301 | + print "Hi, You selected slice" | ||
302 | + print "Selected ID", evt.GetSelectID() | ||
303 | + self.image_panel.dicom_preview.ShowSlice(evt.GetSelectID()) | ||
304 | + evt.Skip() | ||
305 | + | ||
248 | 306 | ||
249 | class SeriesPanel(wx.Panel): | 307 | class SeriesPanel(wx.Panel): |
250 | def __init__(self, parent): | 308 | def __init__(self, parent): |
@@ -252,7 +310,7 @@ class SeriesPanel(wx.Panel): | @@ -252,7 +310,7 @@ class SeriesPanel(wx.Panel): | ||
252 | #self.SetBackgroundColour((0,0,0)) | 310 | #self.SetBackgroundColour((0,0,0)) |
253 | 311 | ||
254 | self.serie_preview = dpp.DicomPreviewSeries(self) | 312 | self.serie_preview = dpp.DicomPreviewSeries(self) |
255 | - self.dicom_preview = dpp.DicomPreview(self) | 313 | + self.dicom_preview = dpp.DicomPreviewSlice(self) |
256 | self.dicom_preview.Show(0) | 314 | self.dicom_preview.Show(0) |
257 | 315 | ||
258 | self.sizer = wx.BoxSizer(wx.HORIZONTAL) | 316 | self.sizer = wx.BoxSizer(wx.HORIZONTAL) |
@@ -260,10 +318,8 @@ class SeriesPanel(wx.Panel): | @@ -260,10 +318,8 @@ class SeriesPanel(wx.Panel): | ||
260 | self.sizer.Add(self.dicom_preview, 1, wx.EXPAND | wx.ALL, 5) | 318 | self.sizer.Add(self.dicom_preview, 1, wx.EXPAND | wx.ALL, 5) |
261 | self.sizer.Fit(self) | 319 | self.sizer.Fit(self) |
262 | 320 | ||
263 | - | ||
264 | self.SetSizer(self.sizer) | 321 | self.SetSizer(self.sizer) |
265 | 322 | ||
266 | - | ||
267 | self.Layout() | 323 | self.Layout() |
268 | self.Update() | 324 | self.Update() |
269 | self.SetAutoLayout(1) | 325 | self.SetAutoLayout(1) |
@@ -277,7 +333,8 @@ class SeriesPanel(wx.Panel): | @@ -277,7 +333,8 @@ class SeriesPanel(wx.Panel): | ||
277 | ps.Publisher().subscribe(self.SetPatientSeries, 'Load patient into import panel') | 333 | ps.Publisher().subscribe(self.SetPatientSeries, 'Load patient into import panel') |
278 | 334 | ||
279 | def _bind_gui_evt(self): | 335 | def _bind_gui_evt(self): |
280 | - self.Bind(dpp.EVT_SELECT_SERIE, self.OnSelectSerie) | 336 | + self.serie_preview.Bind(dpp.EVT_CLICK_SERIE, self.OnSelectSerie) |
337 | + self.dicom_preview.Bind(dpp.EVT_CLICK_SLICE, self.OnSelectSlice) | ||
281 | 338 | ||
282 | def SetDicomSeries(self, pubsub_evt): | 339 | def SetDicomSeries(self, pubsub_evt): |
283 | group = pubsub_evt.data | 340 | group = pubsub_evt.data |
@@ -286,7 +343,6 @@ class SeriesPanel(wx.Panel): | @@ -286,7 +343,6 @@ class SeriesPanel(wx.Panel): | ||
286 | self.serie_preview.Show(0) | 343 | self.serie_preview.Show(0) |
287 | self.sizer.Layout() | 344 | self.sizer.Layout() |
288 | self.Update() | 345 | self.Update() |
289 | - | ||
290 | 346 | ||
291 | def SetPatientSeries(self, pubsub_evt): | 347 | def SetPatientSeries(self, pubsub_evt): |
292 | patient = pubsub_evt.data | 348 | patient = pubsub_evt.data |
@@ -299,26 +355,36 @@ class SeriesPanel(wx.Panel): | @@ -299,26 +355,36 @@ class SeriesPanel(wx.Panel): | ||
299 | 355 | ||
300 | self.Update() | 356 | self.Update() |
301 | 357 | ||
302 | - | ||
303 | def OnSelectSerie(self, evt): | 358 | def OnSelectSerie(self, evt): |
304 | - serie = evt.GetSelectID() | ||
305 | - self.dicom_preview.SetDicomSerie(serie) | ||
306 | - | 359 | + print "Hey, You selected a serie" |
360 | + serie = evt.GetItemData() | ||
307 | data = evt.GetItemData() | 361 | data = evt.GetItemData() |
308 | 362 | ||
363 | + my_evt = SelectEvent(myEVT_SELECT_SERIE, self.GetId()) | ||
364 | + my_evt.SetSelectedID(evt.GetSelectID()) | ||
365 | + my_evt.SetItemData(evt.GetItemData()) | ||
366 | + self.GetEventHandler().ProcessEvent(my_evt) | ||
367 | + | ||
368 | + self.dicom_preview.SetDicomSerie(serie) | ||
309 | self.dicom_preview.Show(1) | 369 | self.dicom_preview.Show(1) |
310 | self.serie_preview.Show(0) | 370 | self.serie_preview.Show(0) |
311 | self.sizer.Layout() | 371 | self.sizer.Layout() |
312 | #self.Show() | 372 | #self.Show() |
313 | self.Update() | 373 | self.Update() |
314 | 374 | ||
375 | + def OnSelectSlice(self, evt): | ||
376 | + print "Hey, Ho, Let's go", evt.GetSelectID() | ||
377 | + | ||
378 | + my_evt = SelectEvent(myEVT_SELECT_SLICE, self.GetId()) | ||
379 | + my_evt.SetSelectedID(evt.GetSelectID()) | ||
380 | + my_evt.SetItemData(evt.GetItemData()) | ||
381 | + self.GetEventHandler().ProcessEvent(my_evt) | ||
315 | 382 | ||
316 | def ShowDicomSeries(self, pubsub_evt): | 383 | def ShowDicomSeries(self, pubsub_evt): |
317 | patient = pubsub_evt.data | 384 | patient = pubsub_evt.data |
318 | if isinstance(patient, dcm.PatientGroup): | 385 | if isinstance(patient, dcm.PatientGroup): |
319 | self.serie_preview.SetPatientGroups(patient) | 386 | self.serie_preview.SetPatientGroups(patient) |
320 | self.dicom_preview.SetPatientGroups(patient) | 387 | self.dicom_preview.SetPatientGroups(patient) |
321 | - | ||
322 | 388 | ||
323 | 389 | ||
324 | class SlicePanel(wx.Panel): | 390 | class SlicePanel(wx.Panel): |
@@ -352,15 +418,13 @@ class SlicePanel(wx.Panel): | @@ -352,15 +418,13 @@ class SlicePanel(wx.Panel): | ||
352 | self.dicom_preview.SetDicomGroup(group) | 418 | self.dicom_preview.SetDicomGroup(group) |
353 | self.sizer.Layout() | 419 | self.sizer.Layout() |
354 | self.Update() | 420 | self.Update() |
355 | - | ||
356 | 421 | ||
357 | - def SetDicomSeries(self, pubsub_evt): | ||
358 | - group = pubsub_evt.data | 422 | + def SetDicomSeries(self, evt): |
423 | + group = evt.data | ||
359 | self.dicom_preview.SetDicomGroup(group) | 424 | self.dicom_preview.SetDicomGroup(group) |
360 | self.sizer.Layout() | 425 | self.sizer.Layout() |
361 | self.Update() | 426 | self.Update() |
362 | 427 | ||
363 | - | ||
364 | def ShowDicomSeries(self, pubsub_evt): | 428 | def ShowDicomSeries(self, pubsub_evt): |
365 | patient = pubsub_evt.data | 429 | patient = pubsub_evt.data |
366 | group = patient.GetGroups()[0] | 430 | group = patient.GetGroups()[0] |