Использование графа DOT в качестве основы для дерева GUI

Question

Я хочу использовать график, который генерируется DOT (pyDot в python), в качестве основы для интерактивного древовидного графического интерфейса, в котором каждый из узлов в дереве может быть виджетом.

Дерево в основном будет двоичным деревом азбуки Морзе, которое начинается с верхнего узла и перемещается вниз по дереву к нужной букве и выбирает его. Узел, который они хотят выбрать, должен быть подсвечиваемым, а содержимое (буквы) должно быть способно изменяться в зависимости от пользовательского ввода.

По сути, я хочу, чтобы узлы были превращены в полномасштабные объекты с настраиваемыми параметрами, которые меняются по мере использования интерфейса. Кто-нибудь может указать мне правильное направление, чтобы сделать это?

Answer 1

Я начал с демонстрационного кода по адресу: http://wiki.wxpython.org/AnotherTutorial#wx.TreeCtrl. Я добавил методы build_tree, _build_tree_helper и build_conn_dict. Ключевыми интересующими методами из библиотеки dot_parser являются edge.get_source() и edge.get_destination(), которые используются для создания словаря "connection".

Точечный график хранится в переменной dot_data. Важно отметить, что точечный граф не должен зацикливаться; то есть это должно быть связующее дерево, иначе метод _build_tree_helper будет зацикливаться бесконечно (и это не имеет смысла в TreeControl).

Я также должен был исправить патч dot_parser в соответствии с https://github.com/nlhepler/pydot-py3/issues/1#issuecomment-15999052 чтобы заставить его работать.

import wx

from dot_parser import parse_dot_data

class MyFrame(wx.Frame):
    def __init__(self, parent, id, title, **kwargs):
        self.parsed_dot = kwargs.pop("parsed_dot", None)
        wx.Frame.__init__(self, parent, id, title, wx.DefaultPosition, wx.Size(450, 350))

        hbox = wx.BoxSizer(wx.HORIZONTAL)
        vbox = wx.BoxSizer(wx.VERTICAL)
        panel1 = wx.Panel(self, -1)
        panel2 = wx.Panel(self, -1)

        self.tree = wx.TreeCtrl(panel1, 1, wx.DefaultPosition, (-1,-1), wx.TR_HAS_BUTTONS | wx.TR_LINES_AT_ROOT )
        self.build_tree(self.tree)
        self.tree.Bind(wx.EVT_TREE_SEL_CHANGED, self.OnSelChanged, id=1)
        self.display = wx.StaticText(panel2, -1, "",(10,10), style=wx.ALIGN_CENTRE)
        vbox.Add(self.tree, 1, wx.EXPAND)
        hbox.Add(panel1, 1, wx.EXPAND)
        hbox.Add(panel2, 1, wx.EXPAND)
        panel1.SetSizer(vbox)
        self.SetSizer(hbox)
        self.Centre()

    def build_conn_dict(self):
        conn_dict = {}
        if(self.parsed_dot):
            for edge in self.parsed_dot.get_edges():
                conn_dict.setdefault(edge.get_source(), []).append(edge.get_destination())
        return conn_dict

    def build_tree(self, tree):
        if(self.parsed_dot):
            conn_dict = self.build_conn_dict()
            outs = set(conn_dict.keys())
            ins = reduce(lambda x, y: x | set(y), conn_dict.values(), set([]))
            roots = list(outs - ins)
            roots = dict([(root, tree.AddRoot(root)) for root in roots])
            self._build_tree_helper(tree, conn_dict, roots)

    def _build_tree_helper(self, tree, conn_dict = {}, roots = {}):
        new_roots = {}
        for root in roots:
            if(conn_dict.has_key(root)):
                for sub_root in conn_dict[root]:
                    new_roots[sub_root] = tree.AppendItem(roots[root], sub_root)
        if(new_roots):
            self._build_tree_helper(tree, conn_dict, new_roots)

    def OnSelChanged(self, event):
        item =  event.GetItem()
        self.display.SetLabel(self.tree.GetItemText(item))
        child_text = self.tree.GetItemText(item)
        parent_text = ""
        try:
            parent = self.tree.GetItemParent(item)
            parent_text = self.tree.GetItemText(parent)
        except wx._core.PyAssertionError:
            pass
        print "child: %s, parent: %s" % (child_text, parent_text)

class MyApp(wx.App):
    def OnInit(self):
        dot_data = \
        '''
        graph ""
        {
                label="(% (EXP (% (X) (% (X) (X)))) (EXP (SIN (X))))"
                n039 ;
                n039 [label="%"] ;
                n039 -> n040 ;
                n040 [label="EXP"] ;
                n040 -> n041 ;
                n041 [label="%"] ;
                n041 -> n042 ;
                n042 [label="X"] ;
                n041 -> n043 ;
                n043 [label="%"] ;
                n043 -> n044 ;
                n044 [label="X"] ;
                n043 -> n045 ;
                n045 [label="X"] ;
                n039 -> n046 ;
                n046 [label="EXP"] ;
                n046 -> n047 ;
                n047 [label="SIN"] ;
                n047 -> n048 ;
                n048 [label="X"] ;
        }
        '''        
        parsed_dot = parse_dot_data(dot_data)
        frame = MyFrame(None, -1, "treectrl.py", parsed_dot = parsed_dot)
        frame.Show(True)
        self.SetTopWindow(frame)
        return True

app = MyApp(0)
app.MainLoop()