# Copyright\r
\r
-"""Defines :class:`GUI` providing a wxWindows interface to Hooke.\r
+"""Defines :class:`GUI` providing a wxWidgets interface to Hooke.\r
"""\r
\r
-from ..command import CommandExit, Exit, Command, Argument, StoreValue\r
-from ..interaction import Request, BooleanRequest, ReloadUserInterfaceConfig\r
-from ..ui import UserInterface, CommandMessage\r
-from ..util.encoding import get_input_encoding, get_output_encoding\r
+WX_GOOD=['2.8']\r
+\r
+import wxversion\r
+wxversion.select(WX_GOOD)\r
+\r
+import copy\r
+import os\r
+import os.path\r
+import platform\r
+import shutil\r
+import time\r
+\r
+import wx.html\r
+import wx.aui as aui\r
+import wx.lib.evtmgr as evtmgr\r
+\r
+\r
+# wxPropertyGrid is included in wxPython >= 2.9.1, see\r
+# http://wxpropgrid.sourceforge.net/cgi-bin/index?page=download\r
+# until then, we'll avoid it because of the *nix build problems.\r
+#import wx.propgrid as wxpg\r
+\r
+from matplotlib.ticker import FuncFormatter\r
+\r
+from ...command import CommandExit, Exit, Success, Failure, Command, Argument\r
+from ...config import Setting\r
+from ...interaction import Request, BooleanRequest, ReloadUserInterfaceConfig\r
+from ...ui import UserInterface, CommandMessage\r
+from .dialog.selection import Selection as SelectionDialog\r
+from .dialog.save_file import select_save_file\r
+from . import menu as menu\r
+from . import navbar as navbar\r
+from . import panel as panel\r
+from .panel.propertyeditor2 import prop_from_argument, prop_from_setting\r
+from . import prettyformat as prettyformat\r
+from . import statusbar as statusbar\r
+\r
+\r
+class HookeFrame (wx.Frame):\r
+ """The main Hooke-interface window. \r
+ """\r
+ def __init__(self, gui, commands, inqueue, outqueue, *args, **kwargs):\r
+ super(HookeFrame, self).__init__(*args, **kwargs)\r
+ self.gui = gui\r
+ self.commands = commands\r
+ self.inqueue = inqueue\r
+ self.outqueue = outqueue\r
+ self._perspectives = {} # {name: perspective_str}\r
+ self._c = {}\r
+\r
+ self.SetIcon(wx.Icon(self.gui.config['icon image'], wx.BITMAP_TYPE_ICO))\r
+\r
+ # setup frame manager\r
+ self._c['manager'] = aui.AuiManager()\r
+ self._c['manager'].SetManagedWindow(self)\r
+\r
+ # set the gradient and drag styles\r
+ self._c['manager'].GetArtProvider().SetMetric(\r
+ aui.AUI_DOCKART_GRADIENT_TYPE, aui.AUI_GRADIENT_NONE)\r
+ self._c['manager'].SetFlags(\r
+ self._c['manager'].GetFlags() ^ aui.AUI_MGR_TRANSPARENT_DRAG)\r
+\r
+ # Min size for the frame itself isn't completely done. See\r
+ # the end of FrameManager::Update() for the test code. For\r
+ # now, just hard code a frame minimum size.\r
+ self.SetMinSize(wx.Size(500, 500))\r
+\r
+ self._setup_panels()\r
+ self._setup_toolbars()\r
+ self._c['manager'].Update() # commit pending changes\r
+\r
+ # Create the menubar after the panes so that the default\r
+ # perspective is created with all panes open\r
+ self._c['menu bar'] = menu.HookeMenuBar(\r
+ parent=self,\r
+ callbacks={\r
+ 'close': self._on_close,\r
+ 'about': self._on_about,\r
+ 'view_panel': self._on_panel_visibility,\r
+ 'save_perspective': self._on_save_perspective,\r
+ 'delete_perspective': self._on_delete_perspective,\r
+ 'select_perspective': self._on_select_perspective,\r
+ })\r
+ self.SetMenuBar(self._c['menu bar'])\r
+\r
+ self._c['status bar'] = statusbar.StatusBar(\r
+ parent=self,\r
+ style=wx.ST_SIZEGRIP)\r
+ self.SetStatusBar(self._c['status bar'])\r
+\r
+ self._setup_perspectives()\r
+ self._bind_events()\r
+\r
+ name = self.gui.config['active perspective']\r
+ return # TODO: cleanup\r
+ self.playlists = self._c['playlists'].Playlists\r
+ self._displayed_plot = None\r
+ #load default list, if possible\r
+ self.do_loadlist(self.GetStringFromConfig('core', 'preferences', 'playlist'))\r
+\r
+\r
+ # GUI maintenance\r
+\r
+ def _setup_panels(self):\r
+ client_size = self.GetClientSize()\r
+ for label,p,style in [\r
+# ('folders', wx.GenericDirCtrl(\r
+# parent=self,\r
+# dir=self.gui.config['folders-workdir'],\r
+# size=(200, 250),\r
+# style=wx.DIRCTRL_SHOW_FILTERS,\r
+# filter=self.gui.config['folders-filters'],\r
+# defaultFilter=int(self.gui.config['folders-filter-index'])), 'left'), #HACK: config should convert\r
+# ('playlists', panel.PANELS['playlist'](\r
+# callbacks={},\r
+# config=self.gui.config,\r
+# parent=self,\r
+# style=wx.WANTS_CHARS|wx.NO_BORDER,\r
+# # WANTS_CHARS so the panel doesn't eat the Return key.\r
+# size=(160, 200)), 'left'),\r
+# ('note', panel.note.Note(\r
+# parent=self\r
+# style=wx.WANTS_CHARS|wx.NO_BORDER,\r
+# size=(160, 200)), 'left'),\r
+# ('notebook', Notebook(\r
+# parent=self,\r
+# pos=wx.Point(client_size.x, client_size.y),\r
+# size=wx.Size(430, 200),\r
+# style=aui.AUI_NB_DEFAULT_STYLE\r
+# | aui.AUI_NB_TAB_EXTERNAL_MOVE | wx.NO_BORDER), 'center'),\r
+ ('commands', panel.PANELS['commands'](\r
+ commands=self.commands,\r
+ selected=self.gui.config['selected command'],\r
+ callbacks={\r
+ 'execute': self.execute_command,\r
+ 'select_plugin': self.select_plugin,\r
+ 'select_command': self.select_command,\r
+# 'selection_changed': self.panelProperties.select(self, method, command), #SelectedTreeItem = selected_item,\r
+ },\r
+ parent=self,\r
+ style=wx.WANTS_CHARS|wx.NO_BORDER,\r
+ # WANTS_CHARS so the panel doesn't eat the Return key.\r
+# size=(160, 200)\r
+ ), 'center'),\r
+ ('property', panel.PANELS['propertyeditor2'](\r
+ callbacks={},\r
+ parent=self,\r
+ style=wx.WANTS_CHARS,\r
+ # WANTS_CHARS so the panel doesn't eat the Return key.\r
+ ), 'center'),\r
+# ('assistant', wx.TextCtrl(\r
+# parent=self,\r
+# pos=wx.Point(0, 0),\r
+# size=wx.Size(150, 90),\r
+# style=wx.NO_BORDER|wx.TE_MULTILINE), 'right'),\r
+ ('output', panel.PANELS['output'](\r
+ parent=self,\r
+ pos=wx.Point(0, 0),\r
+ size=wx.Size(150, 90),\r
+ style=wx.TE_READONLY|wx.NO_BORDER|wx.TE_MULTILINE),\r
+ 'bottom'),\r
+# ('results', panel.results.Results(self), 'bottom'),\r
+ ]:\r
+ self._add_panel(label, p, style)\r
+ #self._c['assistant'].SetEditable(False)\r
+\r
+ def _add_panel(self, label, panel, style):\r
+ self._c[label] = panel\r
+ cap_label = label.capitalize()\r
+ info = aui.AuiPaneInfo().Name(cap_label).Caption(cap_label)\r
+ info.PaneBorder(False).CloseButton(True).MaximizeButton(False)\r
+ if style == 'top':\r
+ info.Top()\r
+ elif style == 'center':\r
+ info.CenterPane()\r
+ elif style == 'left':\r
+ info.Left()\r
+ elif style == 'right':\r
+ info.Right()\r
+ else:\r
+ assert style == 'bottom', style\r
+ info.Bottom()\r
+ self._c['manager'].AddPane(panel, info)\r
+\r
+ def _setup_toolbars(self):\r
+ self._c['navigation bar'] = navbar.NavBar(\r
+ callbacks={\r
+ 'next': self._next_curve,\r
+ 'previous': self._previous_curve,\r
+ },\r
+ parent=self,\r
+ style=wx.TB_FLAT | wx.TB_NODIVIDER)\r
+ self._c['manager'].AddPane(\r
+ self._c['navigation bar'],\r
+ aui.AuiPaneInfo().Name('Navigation').Caption('Navigation'\r
+ ).ToolbarPane().Top().Layer(1).Row(1).LeftDockable(False\r
+ ).RightDockable(False))\r
+\r
+ def _bind_events(self):\r
+ # TODO: figure out if we can use the eventManager for menu\r
+ # ranges and events of 'self' without raising an assertion\r
+ # fail error.\r
+ self.Bind(wx.EVT_ERASE_BACKGROUND, self._on_erase_background)\r
+ self.Bind(wx.EVT_SIZE, self._on_size)\r
+ self.Bind(wx.EVT_CLOSE, self._on_close)\r
+ self.Bind(aui.EVT_AUI_PANE_CLOSE, self.OnPaneClose)\r
+ self.Bind(aui.EVT_AUINOTEBOOK_PAGE_CLOSE, self._on_notebook_page_close)\r
+\r
+ return # TODO: cleanup\r
+ for value in self._c['menu bar']._c['view']._c.values():\r
+ self.Bind(wx.EVT_MENU_RANGE, self._on_view, value)\r
+\r
+ self.Bind(wx.EVT_MENU, self._on_save_perspective,\r
+ self._c['menu bar']._c['perspective']._c['save'])\r
+ self.Bind(wx.EVT_MENU, self._on_delete_perspective,\r
+ self._c['menu bar']._c['perspective']._c['delete'])\r
+\r
+ treeCtrl = self._c['folders'].GetTreeCtrl()\r
+ treeCtrl.Bind(wx.EVT_LEFT_DCLICK, self._on_dir_ctrl_left_double_click)\r
+ \r
+ # TODO: playlist callbacks\r
+ return # TODO: cleanup\r
+ evtmgr.eventManager.Register(self.OnUpdateNote, wx.EVT_BUTTON, self.panelNote.UpdateButton)\r
+ #property editor\r
+ self.panelProperties.pg.Bind(wxpg.EVT_PG_CHANGED, self.OnPropGridChanged)\r
+ #results panel\r
+ self.panelResults.results_list.OnCheckItem = self.OnResultsCheck\r
+\r
+ def _on_about(self, *args):\r
+ dialog = wx.MessageDialog(\r
+ parent=self,\r
+ message=self.gui._splash_text(),\r
+ caption='About Hooke',\r
+ style=wx.OK|wx.ICON_INFORMATION)\r
+ dialog.ShowModal()\r
+ dialog.Destroy()\r
+\r
+ def _on_close(self, *args):\r
+ # apply changes\r
+ self.gui.config['main height'] = str(self.GetSize().GetHeight())\r
+ self.gui.config['main left'] = str(self.GetPosition()[0])\r
+ self.gui.config['main top'] = str(self.GetPosition()[1])\r
+ self.gui.config['main width'] = str(self.GetSize().GetWidth())\r
+ # push changes back to Hooke.config?\r
+ self._c['manager'].UnInit()\r
+ del self._c['manager']\r
+ self.Destroy()\r
+\r
+\r
+\r
+ # Command handling\r
+\r
+ def _command_by_name(self, name):\r
+ cs = [c for c in self.commands if c.name == name]\r
+ if len(cs) == 0:\r
+ raise KeyError(name)\r
+ elif len(cs) > 1:\r
+ raise Exception('Multiple commands named "%s"' % name)\r
+ return cs[0]\r
+\r
+ def execute_command(self, _class=None, method=None,\r
+ command=None, args=None):\r
+ if args == None:\r
+ args = {}\r
+ if ('property' in self._c\r
+ and self.gui.config['selected command'] == command):\r
+ arg_names = [arg.name for arg in command.arguments]\r
+ for name,value in self._c['property'].get_values().items():\r
+ if name in arg_names:\r
+ args[name] = value\r
+ print 'executing', command.name, args\r
+ self.inqueue.put(CommandMessage(command, args))\r
+ results = []\r
+ while True:\r
+ msg = self.outqueue.get()\r
+ results.append(msg)\r
+ if isinstance(msg, Exit):\r
+ self._on_close()\r
+ break\r
+ elif isinstance(msg, CommandExit):\r
+ # TODO: display command complete\r
+ break\r
+ elif isinstance(msg, ReloadUserInterfaceConfig):\r
+ self.gui.reload_config(msg.config)\r
+ continue\r
+ elif isinstance(msg, Request):\r
+ h = handler.HANDLERS[msg.type]\r
+ h.run(self, msg) # TODO: pause for response?\r
+ continue\r
+ pp = getattr(\r
+ self, '_postprocess_%s' % command.name.replace(' ', '_'),\r
+ self._postprocess_text)\r
+ pp(command=command, results=results)\r
+ return results\r
+\r
+ def _handle_request(self, msg):\r
+ """Repeatedly try to get a response to `msg`.\r
+ """\r
+ if prompt == None:\r
+ raise NotImplementedError('_%s_request_prompt' % msg.type)\r
+ prompt_string = prompt(msg)\r
+ parser = getattr(self, '_%s_request_parser' % msg.type, None)\r
+ if parser == None:\r
+ raise NotImplementedError('_%s_request_parser' % msg.type)\r
+ error = None\r
+ while True:\r
+ if error != None:\r
+ self.cmd.stdout.write(''.join([\r
+ error.__class__.__name__, ': ', str(error), '\n']))\r
+ self.cmd.stdout.write(prompt_string)\r
+ value = parser(msg, self.cmd.stdin.readline())\r
+ try:\r
+ response = msg.response(value)\r
+ break\r
+ except ValueError, error:\r
+ continue\r
+ self.inqueue.put(response)\r
+\r
+\r
+\r
+ # Command-specific postprocessing\r
+\r
+ def _postprocess_text(self, command, results):\r
+ """Print the string representation of the results to the Results window.\r
+\r
+ This is similar to :class:`~hooke.ui.commandline.DoCommand`'s\r
+ approach, except that :class:`~hooke.ui.commandline.DoCommand`\r
+ doesn't print some internally handled messages\r
+ (e.g. :class:`~hooke.interaction.ReloadUserInterfaceConfig`).\r
+ """\r
+ for result in results:\r
+ if isinstance(result, CommandExit):\r
+ self._c['output'].write(result.__class__.__name__+'\n')\r
+ self._c['output'].write(str(result).rstrip()+'\n')\r
+\r
+ def _postprocess_get_curve(self, command, results):\r
+ """Update `self` to show the curve.\r
+ """\r
+ if not isinstance(results[-1], Success):\r
+ return # error executing 'get curve'\r
+ assert len(results) == 2, results\r
+ curve = results[0]\r
+ print curve\r
+\r
+ selected_item = self._c['playlists']._c['tree'].GetSelection()\r
+ if self._c['playlists']._c['tree'].ItemHasChildren(selected_item):\r
+ #GetFirstChild returns a tuple\r
+ #we only need the first element\r
+ next_item = self._c['playlists']._c['tree'].GetFirstChild(selected_item)[0]\r
+ else:\r
+ next_item = self._c['playlists']._c['tree'].GetNextSibling(selected_item)\r
+ if not next_item.IsOk():\r
+ parent_item = self._c['playlists']._c['tree'].GetItemParent(selected_item)\r
+ #GetFirstChild returns a tuple\r
+ #we only need the first element\r
+ next_item = self._c['playlists']._c['tree'].GetFirstChild(parent_item)[0]\r
+ self._c['playlists']._c['tree'].SelectItem(next_item, True)\r
+ if not self._c['playlists']._c['tree'].ItemHasChildren(selected_item):\r
+ playlist = self.GetActivePlaylist()\r
+ if playlist.count > 1:\r
+ playlist.next()\r
+ self._c['status bar'].set_playlist(playlist)\r
+ self.UpdateNote()\r
+ self.UpdatePlot()\r
+\r
+\r
+\r
+ # TODO: cruft\r
+\r
+ def _GetActiveFileIndex(self):\r
+ lib.playlist.Playlist = self.GetActivePlaylist()\r
+ #get the selected item from the tree\r
+ selected_item = self._c['playlists']._c['tree'].GetSelection()\r
+ #test if a playlist or a curve was double-clicked\r
+ if self._c['playlists']._c['tree'].ItemHasChildren(selected_item):\r
+ return -1\r
+ else:\r
+ count = 0\r
+ selected_item = self._c['playlists']._c['tree'].GetPrevSibling(selected_item)\r
+ while selected_item.IsOk():\r
+ count += 1\r
+ selected_item = self._c['playlists']._c['tree'].GetPrevSibling(selected_item)\r
+ return count\r
+\r
+ def _GetPlaylistTab(self, name):\r
+ for index, page in enumerate(self._c['notebook']._tabs._pages):\r
+ if page.caption == name:\r
+ return index\r
+ return -1\r
+\r
+ def select_plugin(self, _class=None, method=None, plugin=None):\r
+ pass\r
+\r
+ def AddPlaylistFromFiles(self, files=[], name='Untitled'):\r
+ if files:\r
+ playlist = lib.playlist.Playlist(self, self.drivers)\r
+ for item in files:\r
+ playlist.add_curve(item)\r
+ if playlist.count > 0:\r
+ playlist.name = self._GetUniquePlaylistName(name)\r
+ playlist.reset()\r
+ self.AddTayliss(playlist)\r
+\r
+ def AppliesPlotmanipulator(self, name):\r
+ '''\r
+ Returns True if the plotmanipulator 'name' is applied, False otherwise\r
+ name does not contain 'plotmanip_', just the name of the plotmanipulator (e.g. 'flatten')\r
+ '''\r
+ return self.GetBoolFromConfig('core', 'plotmanipulators', name)\r
+\r
+ def ApplyPlotmanipulators(self, plot, plot_file):\r
+ '''\r
+ Apply all active plotmanipulators.\r
+ '''\r
+ if plot is not None and plot_file is not None:\r
+ manipulated_plot = copy.deepcopy(plot)\r
+ for plotmanipulator in self.plotmanipulators:\r
+ if self.GetBoolFromConfig('core', 'plotmanipulators', plotmanipulator.name):\r
+ manipulated_plot = plotmanipulator.method(manipulated_plot, plot_file)\r
+ return manipulated_plot\r
+\r
+ def GetActiveFigure(self):\r
+ playlist_name = self.GetActivePlaylistName()\r
+ figure = self.playlists[playlist_name].figure\r
+ if figure is not None:\r
+ return figure\r
+ return None\r
+\r
+ def GetActiveFile(self):\r
+ playlist = self.GetActivePlaylist()\r
+ if playlist is not None:\r
+ return playlist.get_active_file()\r
+ return None\r
+\r
+ def GetActivePlot(self):\r
+ playlist = self.GetActivePlaylist()\r
+ if playlist is not None:\r
+ return playlist.get_active_file().plot\r
+ return None\r
+\r
+ def GetDisplayedPlot(self):\r
+ plot = copy.deepcopy(self.displayed_plot)\r
+ #plot.curves = []\r
+ #plot.curves = copy.deepcopy(plot.curves)\r
+ return plot\r
+\r
+ def GetDisplayedPlotCorrected(self):\r
+ plot = copy.deepcopy(self.displayed_plot)\r
+ plot.curves = []\r
+ plot.curves = copy.deepcopy(plot.corrected_curves)\r
+ return plot\r
+\r
+ def GetDisplayedPlotRaw(self):\r
+ plot = copy.deepcopy(self.displayed_plot)\r
+ plot.curves = []\r
+ plot.curves = copy.deepcopy(plot.raw_curves)\r
+ return plot\r
+\r
+ def GetDockArt(self):\r
+ return self._c['manager'].GetArtProvider()\r
+\r
+ def GetPlotmanipulator(self, name):\r
+ '''\r
+ Returns a plot manipulator function from its name\r
+ '''\r
+ for plotmanipulator in self.plotmanipulators:\r
+ if plotmanipulator.name == name:\r
+ return plotmanipulator\r
+ return None\r
+\r
+ def HasPlotmanipulator(self, name):\r
+ '''\r
+ returns True if the plotmanipulator 'name' is loaded, False otherwise\r
+ '''\r
+ for plotmanipulator in self.plotmanipulators:\r
+ if plotmanipulator.command == name:\r
+ return True\r
+ return False\r
+\r
+\r
+ def _on_dir_ctrl_left_double_click(self, event):\r
+ file_path = self.panelFolders.GetPath()\r
+ if os.path.isfile(file_path):\r
+ if file_path.endswith('.hkp'):\r
+ self.do_loadlist(file_path)\r
+ event.Skip()\r
+\r
+ def _on_erase_background(self, event):\r
+ event.Skip()\r
+\r
+ def _on_notebook_page_close(self, event):\r
+ ctrl = event.GetEventObject()\r
+ playlist_name = ctrl.GetPageText(ctrl._curpage)\r
+ self.DeleteFromPlaylists(playlist_name)\r
+\r
+ def OnPaneClose(self, event):\r
+ event.Skip()\r
+\r
+ def OnPropGridChanged (self, event):\r
+ prop = event.GetProperty()\r
+ if prop:\r
+ item_section = self.panelProperties.SelectedTreeItem\r
+ item_plugin = self._c['commands']._c['tree'].GetItemParent(item_section)\r
+ plugin = self._c['commands']._c['tree'].GetItemText(item_plugin)\r
+ config = self.gui.config[plugin]\r
+ property_section = self._c['commands']._c['tree'].GetItemText(item_section)\r
+ property_key = prop.GetName()\r
+ property_value = prop.GetDisplayedString()\r
+\r
+ config[property_section][property_key]['value'] = property_value\r
+\r
+ def OnResultsCheck(self, index, flag):\r
+ results = self.GetActivePlot().results\r
+ if results.has_key(self.results_str):\r
+ results[self.results_str].results[index].visible = flag\r
+ results[self.results_str].update()\r
+ self.UpdatePlot()\r
+\r
+\r
+ def _on_size(self, event):\r
+ event.Skip()\r
+\r
+ def OnUpdateNote(self, event):\r
+ '''\r
+ Saves the note to the active file.\r
+ '''\r
+ active_file = self.GetActiveFile()\r
+ active_file.note = self.panelNote.Editor.GetValue()\r
+\r
+ def UpdateNote(self):\r
+ #update the note for the active file\r
+ active_file = self.GetActiveFile()\r
+ if active_file is not None:\r
+ self.panelNote.Editor.SetValue(active_file.note)\r
+\r
+ def UpdatePlaylistsTreeSelection(self):\r
+ playlist = self.GetActivePlaylist()\r
+ if playlist is not None:\r
+ if playlist.index >= 0:\r
+ self._c['status bar'].set_playlist(playlist)\r
+ self.UpdateNote()\r
+ self.UpdatePlot()\r
+\r
+ def UpdatePlot(self, plot=None):\r
+\r
+ def add_to_plot(curve, set_scale=True):\r
+ if curve.visible and curve.x and curve.y:\r
+ #get the index of the subplot to use as destination\r
+ destination = (curve.destination.column - 1) * number_of_rows + curve.destination.row - 1\r
+ #set all parameters for the plot\r
+ axes_list[destination].set_title(curve.title)\r
+ if set_scale:\r
+ axes_list[destination].set_xlabel(curve.prefix.x + curve.units.x)\r
+ axes_list[destination].set_ylabel(curve.prefix.y + curve.units.y)\r
+ #set the formatting details for the scale\r
+ formatter_x = lib.curve.PrefixFormatter(curve.decimals.x, curve.prefix.x, use_zero)\r
+ formatter_y = lib.curve.PrefixFormatter(curve.decimals.y, curve.prefix.y, use_zero)\r
+ axes_list[destination].xaxis.set_major_formatter(formatter_x)\r
+ axes_list[destination].yaxis.set_major_formatter(formatter_y)\r
+ if curve.style == 'plot':\r
+ axes_list[destination].plot(curve.x, curve.y, color=curve.color, label=curve.label, lw=curve.linewidth, zorder=1)\r
+ if curve.style == 'scatter':\r
+ axes_list[destination].scatter(curve.x, curve.y, color=curve.color, label=curve.label, s=curve.size, zorder=2)\r
+ #add the legend if necessary\r
+ if curve.legend:\r
+ axes_list[destination].legend()\r
+\r
+ if plot is None:\r
+ active_file = self.GetActiveFile()\r
+ if not active_file.driver:\r
+ #the first time we identify a file, the following need to be set\r
+ active_file.identify(self.drivers)\r
+ for curve in active_file.plot.curves:\r
+ curve.decimals.x = self.GetIntFromConfig('core', 'preferences', 'x_decimals')\r
+ curve.decimals.y = self.GetIntFromConfig('core', 'preferences', 'y_decimals')\r
+ curve.legend = self.GetBoolFromConfig('core', 'preferences', 'legend')\r
+ curve.prefix.x = self.GetStringFromConfig('core', 'preferences', 'x_prefix')\r
+ curve.prefix.y = self.GetStringFromConfig('core', 'preferences', 'y_prefix')\r
+ if active_file.driver is None:\r
+ self.AppendToOutput('Invalid file: ' + active_file.filename)\r
+ return\r
+ self.displayed_plot = copy.deepcopy(active_file.plot)\r
+ #add raw curves to plot\r
+ self.displayed_plot.raw_curves = copy.deepcopy(self.displayed_plot.curves)\r
+ #apply all active plotmanipulators\r
+ self.displayed_plot = self.ApplyPlotmanipulators(self.displayed_plot, active_file)\r
+ #add corrected curves to plot\r
+ self.displayed_plot.corrected_curves = copy.deepcopy(self.displayed_plot.curves)\r
+ else:\r
+ active_file = None\r
+ self.displayed_plot = copy.deepcopy(plot)\r
+\r
+ figure = self.GetActiveFigure()\r
+ figure.clear()\r
+\r
+ #use '0' instead of e.g. '0.00' for scales\r
+ use_zero = self.GetBoolFromConfig('core', 'preferences', 'use_zero')\r
+ #optionally remove the extension from the title of the plot\r
+ hide_curve_extension = self.GetBoolFromConfig('core', 'preferences', 'hide_curve_extension')\r
+ if hide_curve_extension:\r
+ title = lh.remove_extension(self.displayed_plot.title)\r
+ else:\r
+ title = self.displayed_plot.title\r
+ figure.suptitle(title, fontsize=14)\r
+ #create the list of all axes necessary (rows and columns)\r
+ axes_list =[]\r
+ number_of_columns = max([curve.destination.column for curve in self.displayed_plot.curves])\r
+ number_of_rows = max([curve.destination.row for curve in self.displayed_plot.curves])\r
+ for index in range(number_of_rows * number_of_columns):\r
+ axes_list.append(figure.add_subplot(number_of_rows, number_of_columns, index + 1))\r
+\r
+ #add all curves to the corresponding plots\r
+ for curve in self.displayed_plot.curves:\r
+ add_to_plot(curve)\r
+\r
+ #make sure the titles of 'subplots' do not overlap with the axis labels of the 'main plot'\r
+ figure.subplots_adjust(hspace=0.3)\r
+\r
+ #display results\r
+ self.panelResults.ClearResults()\r
+ if self.displayed_plot.results.has_key(self.results_str):\r
+ for curve in self.displayed_plot.results[self.results_str].results:\r
+ add_to_plot(curve, set_scale=False)\r
+ self.panelResults.DisplayResults(self.displayed_plot.results[self.results_str])\r
+ else:\r
+ self.panelResults.ClearResults()\r
+ #refresh the plot\r
+ figure.canvas.draw()\r
+\r
+ def _on_curve_select(self, playlist, curve):\r
+ #create the plot tab and add playlist to the dictionary\r
+ plotPanel = panel.plot.PlotPanel(self, ID_FirstPlot + len(self.playlists))\r
+ notebook_tab = self._c['notebook'].AddPage(plotPanel, playlist.name, True)\r
+ #tab_index = self._c['notebook'].GetSelection()\r
+ playlist.figure = plotPanel.get_figure()\r
+ self.playlists[playlist.name] = playlist\r
+ #self.playlists[playlist.name] = [playlist, figure]\r
+ self._c['status bar'].set_playlist(playlist)\r
+ self.UpdateNote()\r
+ self.UpdatePlot()\r
+\r
+\r
+ def _on_playlist_left_doubleclick(self):\r
+ index = self._c['notebook'].GetSelection()\r
+ current_playlist = self._c['notebook'].GetPageText(index)\r
+ if current_playlist != playlist_name:\r
+ index = self._GetPlaylistTab(playlist_name)\r
+ self._c['notebook'].SetSelection(index)\r
+ self._c['status bar'].set_playlist(playlist)\r
+ self.UpdateNote()\r
+ self.UpdatePlot()\r
+\r
+ def _on_playlist_delete(self, playlist):\r
+ notebook = self.Parent.plotNotebook\r
+ index = self.Parent._GetPlaylistTab(playlist.name)\r
+ notebook.SetSelection(index)\r
+ notebook.DeletePage(notebook.GetSelection())\r
+ self.Parent.DeleteFromPlaylists(playlist_name)\r
+\r
+\r
+\r
+ # Command panel interface\r
+\r
+ def select_command(self, _class, method, command):\r
+ #self.select_plugin(plugin=command.plugin)\r
+ if 'assistant' in self._c:\r
+ self._c['assitant'].ChangeValue(command.help)\r
+ self._c['property'].clear()\r
+ for argument in command.arguments:\r
+ if argument.name == 'help':\r
+ continue\r
+ p = prop_from_argument(\r
+ argument, curves=[], playlists=[]) # TODO: lookup playlists/curves\r
+ if p == None:\r
+ continue # property intentionally not handled (yet)\r
+ self._c['property'].append_property(p)\r
+\r
+ self.gui.config['selected command'] = command # TODO: push to engine\r
+\r
+\r
+\r
+ # Navbar interface\r
+\r
+ def _next_curve(self, *args):\r
+ """Call the `next curve` command.\r
+ """\r
+ results = self.execute_command(\r
+ command=self._command_by_name('next curve'))\r
+ if isinstance(results[-1], Success):\r
+ self.execute_command(\r
+ command=self._command_by_name('get curve'))\r
+\r
+ def _previous_curve(self, *args):\r
+ """Call the `previous curve` command.\r
+ """\r
+ self.execute_command(\r
+ command=self._command_by_name('previous curve'))\r
+ if isinstance(results[-1], Success):\r
+ self.execute_command(\r
+ command=self._command_by_name('get curve'))\r
+\r
+\r
+\r
+ # Panel display handling\r
+\r
+ def _on_panel_visibility(self, _class, method, panel_name, visible):\r
+ pane = self._c['manager'].GetPane(panel_name)\r
+ print visible\r
+ pane.Show(visible)\r
+ #if we don't do the following, the Folders pane does not resize properly on hide/show\r
+ if pane.caption == 'Folders' and pane.IsShown() and pane.IsDocked():\r
+ #folders_size = pane.GetSize()\r
+ self.panelFolders.Fit()\r
+ self._c['manager'].Update()\r
+\r
+ def _setup_perspectives(self):\r
+ """Add perspectives to menubar and _perspectives.\r
+ """\r
+ self._perspectives = {\r
+ 'Default': self._c['manager'].SavePerspective(),\r
+ }\r
+ path = self.gui.config['perspective path']\r
+ if os.path.isdir(path):\r
+ files = sorted(os.listdir(path))\r
+ for fname in files:\r
+ name, extension = os.path.splitext(fname)\r
+ if extension != self.gui.config['perspective extension']:\r
+ continue\r
+ fpath = os.path.join(path, fname)\r
+ if not os.path.isfile(fpath):\r
+ continue\r
+ perspective = None\r
+ with open(fpath, 'rU') as f:\r
+ perspective = f.readline()\r
+ if perspective:\r
+ self._perspectives[name] = perspective\r
+\r
+ selected_perspective = self.gui.config['active perspective']\r
+ if not self._perspectives.has_key(selected_perspective):\r
+ self.gui.config['active perspective'] = 'Default' # TODO: push to engine's Hooke\r
+\r
+ self._restore_perspective(selected_perspective)\r
+ self._update_perspective_menu()\r
+\r
+ def _update_perspective_menu(self):\r
+ self._c['menu bar']._c['perspective'].update(\r
+ sorted(self._perspectives.keys()),\r
+ self.gui.config['active perspective'])\r
+\r
+ def _save_perspective(self, perspective, perspective_dir, name,\r
+ extension=None):\r
+ path = os.path.join(perspective_dir, name)\r
+ if extension != None:\r
+ path += extension\r
+ if not os.path.isdir(perspective_dir):\r
+ os.makedirs(perspective_dir)\r
+ with open(path, 'w') as f:\r
+ f.write(perspective)\r
+ self._perspectives[name] = perspective\r
+ self._restore_perspective(name)\r
+ self._update_perspective_menu()\r
+\r
+ def _delete_perspectives(self, perspective_dir, names,\r
+ extension=None):\r
+ print 'pop', names\r
+ for name in names:\r
+ path = os.path.join(perspective_dir, name)\r
+ if extension != None:\r
+ path += extension\r
+ os.remove(path)\r
+ del(self._perspectives[name])\r
+ self._update_perspective_menu()\r
+ if self.gui.config['active perspective'] in names:\r
+ self._restore_perspective('Default')\r
+ # TODO: does this bug still apply?\r
+ # Unfortunately, there is a bug in wxWidgets for win32 (Ticket #3258\r
+ # http://trac.wxwidgets.org/ticket/3258 \r
+ # ) that makes the radio item indicator in the menu disappear.\r
+ # The code should be fine once this issue is fixed.\r
+\r
+ def _restore_perspective(self, name):\r
+ if name != self.gui.config['active perspective']:\r
+ print 'restoring perspective:', name\r
+ self.gui.config['active perspective'] = name # TODO: push to engine's Hooke\r
+ self._c['manager'].LoadPerspective(self._perspectives[name])\r
+ self._c['manager'].Update()\r
+ for pane in self._c['manager'].GetAllPanes():\r
+ if pane.name in self._c['menu bar']._c['view']._c.keys():\r
+ pane.Check(pane.window.IsShown())\r
+\r
+ def _on_save_perspective(self, *args):\r
+ perspective = self._c['manager'].SavePerspective()\r
+ name = self.gui.config['active perspective']\r
+ if name == 'Default':\r
+ name = 'New perspective'\r
+ name = select_save_file(\r
+ directory=self.gui.config['perspective path'],\r
+ name=name,\r
+ extension=self.gui.config['perspective extension'],\r
+ parent=self,\r
+ message='Enter a name for the new perspective:',\r
+ caption='Save perspective')\r
+ if name == None:\r
+ return\r
+ self._save_perspective(\r
+ perspective, self.gui.config['perspective path'], name=name,\r
+ extension=self.gui.config['perspective extension'])\r
+\r
+ def _on_delete_perspective(self, *args, **kwargs):\r
+ options = sorted([p for p in self._perspectives.keys()\r
+ if p != 'Default'])\r
+ dialog = SelectionDialog(\r
+ options=options,\r
+ message="\nPlease check the perspectives\n\nyou want to delete and click 'Delete'.\n",\r
+ button_id=wx.ID_DELETE,\r
+ selection_style='multiple',\r
+ parent=self,\r
+ title='Delete perspective(s)',\r
+ style=wx.DEFAULT_DIALOG_STYLE | wx.RESIZE_BORDER)\r
+ dialog.CenterOnScreen()\r
+ dialog.ShowModal()\r
+ names = [options[i] for i in dialog.selected]\r
+ dialog.Destroy()\r
+ self._delete_perspectives(\r
+ self.gui.config['perspective path'], names=names,\r
+ extension=self.gui.config['perspective extension'])\r
+\r
+ def _on_select_perspective(self, _class, method, name):\r
+ self._restore_perspective(name)\r
+\r
+\r
+\r
+class HookeApp (wx.App):\r
+ """A :class:`wx.App` wrapper around :class:`HookeFrame`.\r
+\r
+ Tosses up a splash screen and then loads :class:`HookeFrame` in\r
+ its own window.\r
+ """\r
+ def __init__(self, gui, commands, inqueue, outqueue, *args, **kwargs):\r
+ self.gui = gui\r
+ self.commands = commands\r
+ self.inqueue = inqueue\r
+ self.outqueue = outqueue\r
+ super(HookeApp, self).__init__(*args, **kwargs)\r
+\r
+ def OnInit(self):\r
+ self.SetAppName('Hooke')\r
+ self.SetVendorName('')\r
+ self._setup_splash_screen()\r
+\r
+ height = int(self.gui.config['main height']) # HACK: config should convert\r
+ width = int(self.gui.config['main width'])\r
+ top = int(self.gui.config['main top'])\r
+ left = int(self.gui.config['main left'])\r
+\r
+ # Sometimes, the ini file gets confused and sets 'left' and\r
+ # 'top' to large negative numbers. Here we catch and fix\r
+ # this. Keep small negative numbers, the user might want\r
+ # those.\r
+ if left < -width:\r
+ left = 0\r
+ if top < -height:\r
+ top = 0\r
+\r
+ self._c = {\r
+ 'frame': HookeFrame(\r
+ self.gui, self.commands, self.inqueue, self.outqueue,\r
+ parent=None, title='Hooke',\r
+ pos=(left, top), size=(width, height),\r
+ style=wx.DEFAULT_FRAME_STYLE|wx.SUNKEN_BORDER|wx.CLIP_CHILDREN),\r
+ }\r
+ self._c['frame'].Show(True)\r
+ self.SetTopWindow(self._c['frame'])\r
+ return True\r
+\r
+ def _setup_splash_screen(self):\r
+ if self.gui.config['show splash screen'] == 'True': # HACK: config should decode\r
+ print 'splash', self.gui.config['show splash screen']\r
+ path = self.gui.config['splash screen image']\r
+ if os.path.isfile(path):\r
+ duration = int(self.gui.config['splash screen duration']) # HACK: config should decode types\r
+ wx.SplashScreen(\r
+ bitmap=wx.Image(path).ConvertToBitmap(),\r
+ splashStyle=wx.SPLASH_CENTRE_ON_SCREEN|wx.SPLASH_TIMEOUT,\r
+ milliseconds=duration,\r
+ parent=None)\r
+ wx.Yield()\r
+ # For some reason splashDuration and sleep do not\r
+ # correspond to each other at least not on Windows.\r
+ # Maybe it's because duration is in milliseconds and\r
+ # sleep in seconds. Thus we need to increase the\r
+ # sleep time a bit. A factor of 1.2 seems to work.\r
+ sleepFactor = 1.2\r
+ time.sleep(sleepFactor * duration / 1000)\r
\r
\r
class GUI (UserInterface):\r
\r
Setting(section=self.setting_section, help=self.__doc__)\r
"""\r
- return []\r
+ return [\r
+ Setting(section=self.setting_section, help=self.__doc__),\r
+ Setting(section=self.setting_section, option='icon image',\r
+ value=os.path.join('doc', 'img', 'microscope.ico'),\r
+ help='Path to the hooke icon image.'),\r
+ Setting(section=self.setting_section, option='show splash screen',\r
+ value=True,\r
+ help='Enable/disable the splash screen'),\r
+ Setting(section=self.setting_section, option='splash screen image',\r
+ value=os.path.join('doc', 'img', 'hooke.jpg'),\r
+ help='Path to the Hooke splash screen image.'),\r
+ Setting(section=self.setting_section, option='splash screen duration',\r
+ value=1000,\r
+ help='Duration of the splash screen in milliseconds.'),\r
+ Setting(section=self.setting_section, option='perspective path',\r
+ value=os.path.join('resources', 'gui', 'perspective'),\r
+ help='Directory containing perspective files.'), # TODO: allow colon separated list, like $PATH.\r
+ Setting(section=self.setting_section, option='perspective extension',\r
+ value='.txt',\r
+ help='Extension for perspective files.'),\r
+ Setting(section=self.setting_section, option='hide extensions',\r
+ value=False,\r
+ help='Hide file extensions when displaying names.'),\r
+ Setting(section=self.setting_section, option='folders-workdir',\r
+ value='.',\r
+ help='This should probably go...'),\r
+ Setting(section=self.setting_section, option='folders-filters',\r
+ value='.',\r
+ help='This should probably go...'),\r
+ Setting(section=self.setting_section, option='active perspective',\r
+ value='Default',\r
+ help='Name of active perspective file (or "Default").'),\r
+ Setting(section=self.setting_section, option='folders-filter-index',\r
+ value='0',\r
+ help='This should probably go...'),\r
+ Setting(section=self.setting_section, option='main height',\r
+ value=500,\r
+ help='Height of main window in pixels.'),\r
+ Setting(section=self.setting_section, option='main width',\r
+ value=500,\r
+ help='Width of main window in pixels.'),\r
+ Setting(section=self.setting_section, option='main top',\r
+ value=0,\r
+ help='Pixels from screen top to top of main window.'),\r
+ Setting(section=self.setting_section, option='main left',\r
+ value=0,\r
+ help='Pixels from screen left to left of main window.'), \r
+ Setting(section=self.setting_section, option='selected command',\r
+ value='load playlist',\r
+ help='Name of the initially selected command.'),\r
+ ]\r
\r
- def reload_config(self):\r
- pass\r
+ def _app(self, commands, ui_to_command_queue, command_to_ui_queue):\r
+ redirect = True\r
+ if __debug__:\r
+ redirect=False\r
+ app = HookeApp(gui=self,\r
+ commands=commands,\r
+ inqueue=ui_to_command_queue,\r
+ outqueue=command_to_ui_queue,\r
+ redirect=redirect)\r
+ return app\r
\r
def run(self, commands, ui_to_command_queue, command_to_ui_queue):\r
- self._initialize()\r
- cmd = self._cmd(commands, ui_to_command_queue, command_to_ui_queue)\r
- cmd.cmdloop(self._splash_text())\r
-\r
- def run_lines(self, commands, ui_to_command_queue, command_to_ui_queue,\r
- lines):\r
- raise NotImplementedError(\r
- 'Use the command line interface for run_lines()')\r
+ app = self._app(commands, ui_to_command_queue, command_to_ui_queue)\r
+ app.MainLoop()\r
projects / hooke.git / blobdiff