206 lines
7.5 KiB
Python
206 lines
7.5 KiB
Python
"""
|
||
CompareViews：
|
||
对比Storylines基线和最新Storylines编译结果差异对比
|
||
"""
|
||
|
||
import sys
|
||
import os
|
||
from typing import List, Dict, Tuple
|
||
from PyQt5.QtCore import QModelIndex
|
||
from PyQt5.QtGui import QStandardItemModel, QStandardItem
|
||
from PyQt5.QtWidgets import QWidget, QVBoxLayout, QTableView, QSplitter, QApplication, QMainWindow
|
||
|
||
sys.path.append(os.path.dirname(os.path.dirname(__file__)))
|
||
from frame.MergeView import LinesMergeView, MergeDestination
|
||
from manage.MileStone import base_store_path, current_store_path
|
||
from parse.StoryMap import FragmentSlice, StoryMap, XAST_ParseTool, storyline_list2map
|
||
from parse.StorylineCmp import ModifyReason, CmpTool
|
||
|
||
|
||
class ResultList(QTableView):
|
||
def __init__(self, parent):
|
||
QTableView.__init__(self, parent)
|
||
self.model = QStandardItemModel(self)
|
||
self.model.setHorizontalHeaderLabels(["故事名称", "情节名称", "状态"])
|
||
self.setModel(self.model)
|
||
self.setSelectionMode(QTableView.SelectionMode.SingleSelection)
|
||
self.setSelectionBehavior(QTableView.SelectionBehavior.SelectRows)
|
||
pass
|
||
|
||
def present_affected_fragments(self, graph_new: Dict[str, StoryMap], cmp_result: Dict[ModifyReason, List[FragmentSlice]]):
|
||
"""
|
||
展示影响节点
|
||
:return:
|
||
"""
|
||
old_remove = cmp_result[ModifyReason.Removed]
|
||
affected_list = []
|
||
for node in old_remove:
|
||
affected_list = affected_list + self.__find_remove_affected_slice_within_new(graph_new, node)
|
||
pass
|
||
|
||
fragm_items = self.__filter_and_related_defines_seek(affected_list)
|
||
self.__append_fragments_list(fragm_items, "REMOVE")
|
||
|
||
curr_append = cmp_result[ModifyReason.Append]
|
||
affected_list = self.__find_append_and_changed_affected_slice(curr_append)
|
||
fragm_items = self.__filter_and_related_defines_seek(affected_list)
|
||
self.__append_fragments_list(fragm_items, "APPEND")
|
||
|
||
curr_changed = cmp_result[ModifyReason.Changed]
|
||
affected_list = self.__find_append_and_changed_affected_slice(curr_changed)
|
||
fragm_items = self.__filter_and_related_defines_seek(affected_list)
|
||
self.__append_fragments_list(fragm_items, "UPDATE")
|
||
|
||
pass
|
||
|
||
def __find_remove_affected_slice_within_new(self, graph_new: Dict[str, StoryMap], peers_old: FragmentSlice) -> List[FragmentSlice]:
|
||
"""
|
||
获取受到移除节点直接影响的节点
|
||
:param graph_new: 新图
|
||
:param peers_old: 旧图节点
|
||
:return:
|
||
"""
|
||
retvs = []
|
||
story_name = peers_old.parent_story
|
||
|
||
# 移除了情节定义
|
||
if peers_old.is_define_node[0]:
|
||
refer_list = peers_old.refers_slice
|
||
|
||
for refer_slice_prev in refer_list.values():
|
||
retvs = retvs + self.__find_remove_affect_within_new(graph_new, refer_slice_prev)
|
||
pass
|
||
|
||
story_curr = graph_new[story_name]
|
||
# 新图内故事线存在，非最后一个节点，计算影响
|
||
if story_curr is not None and peers_old.next_node is not None:
|
||
o_frag_def = peers_old.next_node
|
||
frag_curr_def = story_curr.get_fragment_defined(o_frag_def.is_define_node[1])
|
||
if frag_curr_def is not None:
|
||
retvs.append(frag_curr_def)
|
||
pass
|
||
pass
|
||
else:
|
||
refer_story_new = graph_new[story_name]
|
||
slice_affected = peers_old.next_node
|
||
|
||
# 非最后一个，需要计算影响，新图中的故事线存在，计算影响
|
||
if slice_affected is not None and refer_story_new is not None:
|
||
v_s_name = slice_affected.story_refer
|
||
v_f_name = slice_affected.fragm_refer
|
||
|
||
# 获取收到影响的情节片段
|
||
slice_refer_now = CmpTool.__locate_fragment_refer__(refer_story_new.slice_list[0], v_s_name, v_f_name)
|
||
if slice_refer_now is not None:
|
||
retvs.append(slice_refer_now)
|
||
pass
|
||
pass
|
||
|
||
return retvs
|
||
|
||
def __find_append_and_changed_affected_slice(self, peers_new: List[FragmentSlice]) -> List[FragmentSlice]:
|
||
retvs = []
|
||
|
||
for curr_node in peers_new:
|
||
next_slice = curr_node.next_node
|
||
if next_slice is not None:
|
||
retvs.append(next_slice)
|
||
|
||
return retvs
|
||
|
||
def __filter_and_related_defines_seek(self, affect_directed: List[FragmentSlice]) -> List[Tuple[str, str]]:
|
||
story_fragment_names: Dict[str, Tuple[str, str]] = {}
|
||
for node in affect_directed:
|
||
# 情节定义节点
|
||
if node.is_define_node[0]:
|
||
fragm_key = f"{node.parent_story}#{node.is_define_node[1]}"
|
||
story_fragment_names[fragm_key] = (node.parent_story, node.is_define_node[1])
|
||
|
||
# 情节引用节点
|
||
else:
|
||
fragm_key = f"{node.story_refer}#{node.fragm_refer}"
|
||
story_fragment_names[fragm_key] = (node.story_refer, node.fragm_refer)
|
||
pass
|
||
pass
|
||
|
||
return list(story_fragment_names.values())
|
||
|
||
def __append_fragments_list(self, fragment_keys: List[Tuple[str, str]], comp_type: str) -> None:
|
||
for key_def in fragment_keys:
|
||
row = [
|
||
QStandardItem(key_def[0]),
|
||
QStandardItem(key_def[1]),
|
||
QStandardItem(comp_type)
|
||
]
|
||
for cell in row:
|
||
cell.setEditable(False)
|
||
self.model.appendRow(row)
|
||
pass
|
||
pass
|
||
|
||
pass
|
||
|
||
|
||
class CompareWindow(QWidget):
|
||
def __init__(self, graph_new: Dict[str, StoryMap], cmp_result: Dict[ModifyReason, List[FragmentSlice]], parent):
|
||
QWidget.__init__(self, parent)
|
||
|
||
self.graph_new_refer = graph_new
|
||
|
||
layout = QVBoxLayout(self)
|
||
self.splitter = QSplitter(self)
|
||
layout.addWidget(self.splitter)
|
||
|
||
self.result_list = ResultList(self)
|
||
self.splitter.addWidget(self.result_list)
|
||
self.splitter.addWidget(QWidget(self))
|
||
|
||
self.result_list.present_affected_fragments(graph_new, cmp_result)
|
||
|
||
self.result_list.clicked.connect(self.present_fragment_merge_content)
|
||
pass
|
||
|
||
def present_fragment_merge_content(self, index: QModelIndex):
|
||
if not index.isValid():
|
||
return
|
||
|
||
story_name = self.result_list.model.item(index.row(), 0).text()
|
||
fragm_name = self.result_list.model.item(index.row(), 1).text()
|
||
story_node = self.graph_new_refer[story_name]
|
||
fragm_node = story_node.get_fragment_defined(fragm_name)
|
||
|
||
main_point = MergeDestination(story_name, fragm_node.prev_node, fragm_node)
|
||
elist = []
|
||
for refstory in fragm_node.refers_slice.keys():
|
||
refslice = fragm_node.refers_slice[refstory]
|
||
elist.append(MergeDestination(refstory, refslice.prev_node, refslice))
|
||
|
||
new_view = LinesMergeView(main_point, elist, self)
|
||
new_view.load_from_mem()
|
||
self.splitter.replaceWidget(1, new_view)
|
||
pass
|
||
|
||
pass
|
||
|
||
|
||
if __name__ == "__main__":
|
||
app = QApplication(sys.argv)
|
||
win = QMainWindow()
|
||
|
||
ast_old = XAST_ParseTool(base_store_path)
|
||
ast_new = XAST_ParseTool(current_store_path)
|
||
map_old = storyline_list2map(ast_old.story_list)
|
||
map_new = storyline_list2map(ast_new.story_list)
|
||
|
||
ast_old.storylines_plait(map_old)
|
||
ast_new.storylines_plait(map_new)
|
||
|
||
cmp_tool = CmpTool()
|
||
all_changed = cmp_tool.graph_compare(map_new, map_old)
|
||
|
||
view = CompareWindow(map_new, all_changed, win)
|
||
win.setCentralWidget(view)
|
||
win.show()
|
||
|
||
app.exec()
|