StoryCheckTools/frame/ContentView.py

import sys
from typing import Dict, List

from PyQt5.QtCore import QPoint
from PyQt5.QtWidgets import QApplication, QMainWindow
from PyQt5.QtWidgets import QMenu

from graph.DataType import Arrow, Point
from graph.directed_acyclic_graph.DAGPresent import DAGActiveView
from parse.StoryMap import StoryMap, XAST_ParseTool
from parse.ast_load import global_ast_path


class FragmentPoint(Point):
    def __init__(self, name: str):
        Point.__init__(self, name)
        pass


class ContentWindow(QMainWindow):
    def __init__(self, parent):
        QMainWindow.__init__(self, parent)
        self.fragment_view = DAGActiveView(self)
        self.setCentralWidget(self.fragment_view)
        self.fragment_view.nodes_clicked.connect(self.print_node_list)
        self.present_graph: Dict[str, StoryMap] = None
        pass
    pass

    def present_stories_graph(self, graph: Dict[str, StoryMap]) -> None:
        self.present_graph = graph

        arrows: List[Arrow] = []
        for story_name in graph:
            fragments = graph[story_name].slice_list
            for fragi in range(1, len(fragments)):
                start_frag = fragments[fragi - 1]
                start_node_label = f"{start_frag.parent_story}&{start_frag.is_define_node[1]}"
                if not start_frag.is_define_node[0]:
                    start_node_label = f"{start_frag.story_refer}&{start_frag.fragm_refer}"
                if fragi == 1:
                    start_node_label = story_name

                end_frag = fragments[fragi]
                end_node_label = f"{end_frag.parent_story}&{end_frag.is_define_node[1]}"
                if not end_frag.is_define_node[0]:
                    end_node_label = f"{end_frag.story_refer}&{end_frag.fragm_refer}"

                arrows.append(Arrow(
                    FragmentPoint(start_node_label),
                    FragmentPoint(end_node_label))
                )
                pass
            pass

        self.fragment_view.update_with_edges(arrows)
        pass

    def print_node_list(self, xpos, ypos, list):
        if len(list) == 0:
            return

        if len(list) > 1:
            pass
        else:
            node_x = list[0]
            if node_x[0].startswith("node"):
                node_sections = [s for s in node_x[1].split("&") if s != ""]
                if len(node_sections) == 1:
                    story = self.present_graph[node_sections[0]]
                    node_names = [node_sections[0]]
                    for slice in story.slice_list[1:]:
                        if slice.is_define_node[0]:
                            node_names.append(f"{slice.parent_story}&{slice.is_define_node[1]}")
                        else:
                            node_names.append(f"{slice.story_refer}&{slice.fragm_refer}")
                            pass
                        pass
                    self.fragment_view.highlight_graph_link(node_names)
                    pass
                elif len(node_sections) > 1:
                    story_node = self.present_graph[node_sections[0]]
                    fragm_node = story_node.get_fragment_defined(node_sections[1])

                    story_list = [node_sections[0]]
                    for vsname in fragm_node.refers_slice:
                        if vsname not in story_list:
                            story_list.append(vsname)
                            pass
                        pass

                    if len(story_list) == 1:
                        self.print_node_list(xpos, ypos, [("node", story_list[0])])
                    elif len(story_list) > 1:
                        menu = QMenu()
                        for story in story_list:
                            def trigger(story_name:str):
                                return lambda : self.print_node_list(xpos, ypos, [("node", story_name)])

                            menu.addAction(f"story/{story}", trigger(story))
                            pass

                        menu.exec(self.fragment_view.mapToGlobal(QPoint(xpos, ypos)))
                        pass
                    else:
                        print(story_list)
                        pass
                    pass
                pass
            pass
        pass


if __name__ == "__main__":
    app = QApplication(sys.argv)
    view = ContentWindow(None)
    view.show()

    tool = XAST_ParseTool(global_ast_path)
    view.present_stories_graph(tool.get_story_graph())

#    view.fragment_view.highlight_graph_link(["血脉的源头", "血脉的源头&待续"])

    app.exec()