| 571 | pane.data[0:grad_img_resize.shape[0], 0:grad_img_resize.shape[1], :] = grad_img_resize |
| 572 | |
| 573 | def _draw_jpgvis_pane(self, pane): |
| 574 | pane.data[:] = to_255(self.settings.window_background) |
| 575 | |
| 576 | with self.state.lock: |
| 577 | state_layer, state_selected_unit, cursor_area, show_unit_jpgs = self.state.layer, self.state.selected_unit, self.state.cursor_area, self.state.show_unit_jpgs |
| 578 | |
| 579 | try: |
| 580 | # Some may be missing this setting |
| 581 | self.settings.caffevis_jpgvis_layers |
| 582 | except: |
| 583 | print '\n\nNOTE: you need to upgrade your settings.py and settings_local.py files. See README.md.\n\n' |
| 584 | raise |
| 585 | |
| 586 | if self.settings.caffevis_jpgvis_remap and state_layer in self.settings.caffevis_jpgvis_remap: |
| 587 | img_key_layer = self.settings.caffevis_jpgvis_remap[state_layer] |
| 588 | else: |
| 589 | img_key_layer = state_layer |
| 590 | |
| 591 | if self.settings.caffevis_jpgvis_layers and img_key_layer in self.settings.caffevis_jpgvis_layers and cursor_area == 'bottom' and show_unit_jpgs: |
| 592 | img_key = (img_key_layer, state_selected_unit, pane.data.shape) |
| 593 | img_resize = self.img_cache.get(img_key, None) |
| 594 | if img_resize is None: |
| 595 | # If img_resize is None, loading has not yet been attempted, so show stale image and request load by JPGVisLoadingThread |
| 596 | with self.state.lock: |
| 597 | self.state.jpgvis_to_load_key = img_key |
| 598 | pane.data[:] = to_255(self.settings.stale_background) |
| 599 | elif img_resize.nbytes == 0: |
| 600 | # This is the sentinal value when the image is not |
| 601 | # found, i.e. loading was already attempted but no jpg |
| 602 | # assets were found. Just display disabled. |
| 603 | pane.data[:] = to_255(self.settings.window_background) |
| 604 | else: |
| 605 | # Show image |
| 606 | pane.data[:img_resize.shape[0], :img_resize.shape[1], :] = img_resize |
| 607 | else: |
| 608 | # Will never be available |
| 609 | pane.data[:] = to_255(self.settings.window_background) |
| 610 | |
| 611 | def handle_key(self, key, panes): |
| 612 | return self.state.handle_key(key) |