Exemple #1
0
    def configure(self, parent, filepath):
        self.filepath = filepath
        self.parent = parent
        # Init panel controls

        def bind_btn(btnid, fn):
            self.Bind(wx.EVT_BUTTON, fn, id=xrc.get(btnid))

        bind_btn("btn_load_cancel", self.on_cancel)

        # Start loading thread
        self.thread_load = LoadThread(self, self.on_load_result, False)

        fn = paply(self.thread_load.default_progress_fn, wxgauge=xrc.get("progress_gauge", self))

        self.thread_load.configure(filepath, fn)
        self.thread_load.start()

        self.Bind(wx.EVT_WINDOW_DESTROY, self.on_destroy)
Exemple #2
0
class LoadPanel(wx.Panel):
    def __init__(self, *args, **kwargs):
        pre = wx.PrePanel()
        # create step is handled by XRC
        self.PostCreate(pre)

    def configure(self, parent, filepath):
        self.filepath = filepath
        self.parent = parent
        # Init panel controls

        def bind_btn(btnid, fn):
            self.Bind(wx.EVT_BUTTON, fn, id=xrc.get(btnid))

        bind_btn("btn_load_cancel", self.on_cancel)

        # Start loading thread
        self.thread_load = LoadThread(self, self.on_load_result, False)

        fn = paply(self.thread_load.default_progress_fn, wxgauge=xrc.get("progress_gauge", self))

        self.thread_load.configure(filepath, fn)
        self.thread_load.start()

        self.Bind(wx.EVT_WINDOW_DESTROY, self.on_destroy)

    def on_cancel(self, event=None):
        self.thread_load.abort()
        wx.CallAfter(self.parent.DeletePage, self.parent.GetSelection())

    def on_load_result(self, event=None):
        logging.debug("Load result")
        if event.data is None:
            logging.debug("Load failure")
            dial = wx.MessageDialog(self, "Error loading file, check console", "Error", wx.OK | wx.ICON_ERROR)
            dial.ShowModal()
            dial.Destroy()
            self.on_cancel()
        else:
            # plot
            logging.info("Data loaded, plotting")
            self.plot(event.data)

            self.thread_update = UpdateThread(self, self.on_update, False)
            self.thread_update.configure(
                self.filepath,
                event.data["_last_red"],
                event.data["_last_mappings"],
                event.data["_bounds"],
                event.data["_time_bounds"],
            )
            self.thread_update.start()

    def on_update(self, event=None):
        # event.data = data to be merged with current plot data
        self.plot.data["_bounds"] = event.data.pop("_bounds")
        self.plot.data["_time_bounds"] = event.data.pop("_time_bounds")
        self.plot.data["_times"] += event.data.pop("_times")

        for dtype, value in event.data.iteritems():
            self.plot.data[dtype] += value

            if hasattr(self.plot, dtype):
                line = getattr(self.plot, dtype)
                for index, _line in enumerate(self.plot.axes.lines):
                    if _line == line:
                        break

                self.plot.axes.lines[index].set_xdata(self.plot.data["_times"])
                self.plot.axes.lines[index].set_ydata(self.plot.data[dtype])

        live_mode = self.plot.controls.live_mode.GetSelection()
        xmax = self.plot.data["_times"][-1]
        if live_mode == 1:  # follow
            # TODO (normal): configurable live window size
            xmin = xmax - timedelta(minutes=10)
            self.plot.axes.set_xbound(lower=xmin, upper=xmax)
        elif live_mode == 2:  # scale x
            xmin = self.plot.data["_times"][0]
            self.plot.axes.set_xbound(lower=xmin, upper=xmax)

        self.plot.redraw()

    def on_destroy(self, event=None):
        """
            End our threads gracefully
        """
        if hasattr(self, "thread_load") and self.thread_load is not None:
            self.thread_load.abort()
        if hasattr(self, "thread_update") and self.thread_update is not None:
            self.thread_update.abort()

    def plot(self, data):
        xrc.get("load_progress", self).Hide()
        chart = xrc.get("chart_panel", self)
        if data["_times"] != []:
            title = data["_times"][int(len(data["_times"]) / 2)].strftime("%d %b %Y")
        else:
            title = os.path.basename(self.filepath)
        self.plot = PlotPanel(chart, data, title, self)

        sizer = wx.BoxSizer(wx.HORIZONTAL)
        sizer.Add(self.plot, 1, wx.EXPAND)

        chart.SetSizer(sizer)

        self.Fit()