plots ench

opal.py

@@ -50,16 +50,42 @@ class LineData:
     поэтому не имеет собственного значения названия. Вместо этого
     название берется из графического компонента.
     """
-    def __init__(self, type, mdata, columns, colour = None, style = None):
+    def __init__(self, ums_ptr, plots_ptr,
-        self.type       = type      # тип графика
+        type, columns, colour = None, style = None):
-        self.mdata      = mdata     # указатель на данные модели
+
-        self.title      = ''
+        self.ums_ptr = ums_ptr      # (ptr, item) указатель на компонент с моделями
-        self.columns    = columns   # пара (x, y)
+                                    # и индекс в этом компоненте
-        self.colour     = colour    # цвет: если не задан выбирается из списка
+        self.plots_ptr = plots_ptr  # (ptr, item) указатель на компонент с графиками
-        self.style      = style     # стиль: если не задан, еспользуется по умолчанию
+                                    # и индекс в этом компоненте
+
+        self.type = type            # тип графика
+        self.columns = columns      # пара (x, y)
+        self.colour = colour        # цвет: если не задан выбирается из списка
+        self.style = style          # стиль: если не задан, еспользуется по умолчанию
+
+    def GetTitle(self):
+        # если есть указатель на компонент с графиками,
+        # извлекаем оттуда название
+        if self.plots_ptr:
+            container, item = self.plots_ptr
+            return container.GetItemText(item)
+        # иначе формируем название на основе имени модели и
+        # имен выбранных столбцов
+        else:
+            container, item = self.ums_ptr
+            title = container.GetItemText(item)
+            data = container.GetPyData(item)
+            assert data.res # если результата нет, то а-та-та
+            colx, coly = self.columns
+            title_x = data.res.columns[colx].GetTitle()
+            title_y = data.res.columns[coly].GetTitle()
+            return "{}: {}({})".format(title, title_y, title_x)
 
     def GetPoints(self):
-        return self.mdata.res.Zip(*self.columns)
+        container, item = self.ums_ptr
+        data = container.GetPyData(item)
+        assert data.res # если результата нет, то а-та-та
+        return data.res.Zip(*self.columns)
 
 class ItemError(Exception):
     pass
@@ -601,47 +627,73 @@ class MainFrame(forms.MainFrame):
 
         Возвращает список экземпляров LineData
         """
-        um = self.m_user_models
-        item, data = self.GetSelectedItemData(um)
-        title = um.GetItemText(item)
-        if not data.res:
-            wx.MessageBox('There is no any result data', 'Warning', wx.OK | wx.ICON_EXCLAMATION)
-            return []
-        f = LineSelectDialog(self, 'Select lines for "{}"'.format(title))
-        for index, col in enumerate(data.res.columns):
-            row_title = col.GetTitle()
-            row_data  = index
-            f.Add(row_title, row_data)
-        f.SetSelections()
 
+        def CreateLineSelectDialog(parent, title, model_data):
+            f = LineSelectDialog(parent, title)
+            for index, col in enumerate(model_data.res.columns):
+                row_title = col.GetTitle()
+                row_data  = index
+                f.Add(row_title, row_data)
+            f.SetSelections()
+            return f
+
+        def GetLinesFromUser(select_dialog):
+            lines = []
+            self.do_nothing = True
+            try:
+                if select_dialog.ShowModal() == wx.ID_OK:
+                    for xy in select_dialog.GetLineColumns():
+                        line_data = LineData(
+                            (um, item), # указатель на модель
+                            None,       # указатель на график
+                            line_type, xy) # тип линии, колонки и прочее
+                        lines.append(line_data)
+                f.Destroy()
+            finally:
+                self.do_nothing = False
+            return lines
+
+        um = self.m_user_models
         lines = []
-        self.do_nothing = True
+        items = um.GetSelections()
-        try:
+        count = len(items)
-            if f.ShowModal() == wx.ID_OK:
+        for index, item in enumerate(items, 1):
-                lines = [ LineData(line_type, data, xy)
+            data = um.GetPyData(item)
-                            for xy in f.GetLineColumns() ]
+            title = um.GetItemText(item)
-        finally:
+
-            self.do_nothing = False
+            msg = 'Line(s) for "{}" ({}/{})'.format(title, index, count)
+
+            if not data.res:
+                wx.MessageBox(
+                    'There is no any result data for model!', 
+                    msg, wx.OK | wx.ICON_EXCLAMATION)
+            else:
+                f = CreateLineSelectDialog(self, msg, data)
+                lines += GetLinesFromUser(f)
 
         return lines
 
     @item_protector
     def AddLines(self, line_type):
+        """
+        Добавляет линии в выделенный график компонента с графиками
+        (m_plots)
+        """
+        # получаем указатель на индекс и данные элемента, который выделен
         item, data = self.GetSelectedItemData(self.m_plots)
+        # если это на контейнер с графиками, то выходим
         if data != 'plot':
             return
+        # получаем указанные пользователем линии
         lines = self.GetLines(line_type)
-        if not lines:
+        # if not lines:
-            return
+        #     return
-        it = self.GetSelectedItem(self.m_user_models)
         for line in lines:
-            x, y = line.columns
+            title = line.GetTitle()
-            data = line.mdata
+            child = self.m_plots.AppendItem(item, title)
-            model_name = self.m_user_models.GetItemText(it)
+            # указываем на только что созданный новый элемент
-            x_name = data.res.columns[x].GetTitle()
+            line.plots_ptr = (self.m_plots, child)
-            y_name = data.res.columns[y].GetTitle()
+            # заполняем элемент данными
-            title  = "{}: {}({})".format(model_name, y_name, x_name)
-            child  = self.m_plots.AppendItem(item, title)
             self.m_plots.SetPyData(child, line)
             self.m_plots.SetItemImage(child, self.icons.pline)
             self.m_plots.Expand(item)
@@ -663,14 +715,9 @@ class MainFrame(forms.MainFrame):
 
     def OnQuickShowPlot(self, event):
         lines = self.GetLines(LINE_CURVE)
-        um    = self.m_user_models
+        um = self.m_user_models
-        item, data = self.GetSelectedItemData(um)
+        item = self.GetSelectedItem(um)
         title = um.GetItemText(item)
-        for line in lines:
-            colx, coly = line.columns
-            title_x = data.res.columns[colx].GetTitle()
-            title_y = data.res.columns[coly].GetTitle()
-            line.title = "{}: {}({})".format(title, title_y, title_x)
         self.ShowPlot(lines, title)
 
     def OnPlotProcess(self, event):
@@ -853,6 +900,7 @@ class LineSelectDialog(forms.LineSelectDialog):
 class PlotFrame(forms.PlotFrame):
     def __init__(self, parent, title, lines):
         forms.PlotFrame.__init__(self, parent, title)
+        self.lines = lines      # список объектор LineData
 
         self.Bind(wx.EVT_MENU, self.OnSaveImage,
             id = forms.ID_SAVE_PLOT)
@@ -868,7 +916,7 @@ class PlotFrame(forms.PlotFrame):
                 handle = wxplot.PolyLine
             points = line.GetPoints()
             attr['colour'] = line.colour or colours[i % len(colours)]
-            attr['legend'] = line.title or 'Unknown line'
+            attr['legend'] = line.GetTitle() or 'Unknown line'
             plot_lines.append(handle(points, **attr))
 
         graph = wxplot.PlotGraphics(plot_lines)