mirror of
https://github.com/olivierkes/manuskript.git
synced 2024-06-22 16:20:19 +12:00
90 lines
2.5 KiB
Python
90 lines
2.5 KiB
Python
|
#!/usr/bin/env python
|
||
|
|
# --!-- coding: utf8 --!--
|
||
|
|
from PyQt5.QtCore import Qt
|
||
|
|
from PyQt5.QtGui import QBrush, QPen
|
||
|
|
from PyQt5.QtWidgets import QWidget, QGraphicsScene, QGraphicsSimpleTextItem, QMenu, QAction
|
||
|
|
|
||
|
|
from manuskript.functions import randomColor
|
||
|
|
from manuskript.ui.views.storylineView_ui import Ui_storylineView
|
||
|
|
|
||
|
|
|
||
|
|
class storylineView(QWidget, Ui_storylineView):
|
||
|
|
def __init__(self, parent=None):
|
||
|
|
QWidget.__init__(self, parent)
|
||
|
|
self.setupUi(self)
|
||
|
|
self._mdlPlots = None
|
||
|
|
self.btnRefresh.clicked.connect(self.refresh)
|
||
|
|
self.scene = QGraphicsScene()
|
||
|
|
self.view.setScene(self.scene)
|
||
|
|
|
||
|
|
self.generateMenu()
|
||
|
|
|
||
|
|
def generateMenu(self):
|
||
|
|
m = QMenu()
|
||
|
|
|
||
|
|
for i in [
|
||
|
|
self.tr("Show Plots"),
|
||
|
|
self.tr("Show Characters"),
|
||
|
|
self.tr("Show Objects"),
|
||
|
|
]:
|
||
|
|
a = QAction(i, m)
|
||
|
|
a.setCheckable(True)
|
||
|
|
a.setEnabled(False)
|
||
|
|
m.addAction(a)
|
||
|
|
|
||
|
|
self.btnSettings.setMenu(m)
|
||
|
|
|
||
|
|
def setModel(self, mdlPlots):
|
||
|
|
self._mdlPlots = mdlPlots
|
||
|
|
self._mdlPlots.dataChanged.connect(self.refresh)
|
||
|
|
self._mdlPlots.rowsInserted.connect(self.refresh)
|
||
|
|
|
||
|
|
def refresh(self):
|
||
|
|
if not self._mdlPlots:
|
||
|
|
pass
|
||
|
|
|
||
|
|
LINE_HEIGHT = 32
|
||
|
|
LINE_SPACING = 6
|
||
|
|
RECT_WIDTH = 200
|
||
|
|
TOTAL_WIDTH = 4000
|
||
|
|
|
||
|
|
s = self.scene
|
||
|
|
s.clear()
|
||
|
|
plotsID = self._mdlPlots.getPlotsByImportance()
|
||
|
|
i = 0
|
||
|
|
|
||
|
|
plots = []
|
||
|
|
|
||
|
|
# Add Plots
|
||
|
|
for importance in plotsID:
|
||
|
|
for ID in importance:
|
||
|
|
name = self._mdlPlots.getPlotNameByID(ID)
|
||
|
|
print(ID, name)
|
||
|
|
color = randomColor()
|
||
|
|
|
||
|
|
# Rect
|
||
|
|
r = s.addRect(0, 0, RECT_WIDTH, LINE_HEIGHT)
|
||
|
|
r.setPen(QPen(Qt.NoPen))
|
||
|
|
r.setBrush(QBrush(color))
|
||
|
|
r.setPos(0, i * LINE_HEIGHT + i * LINE_SPACING)
|
||
|
|
i += 1
|
||
|
|
|
||
|
|
# Text
|
||
|
|
txt = QGraphicsSimpleTextItem(name, r)
|
||
|
|
txt.setPos(r.boundingRect().center() - txt.boundingRect().center())
|
||
|
|
|
||
|
|
# Line
|
||
|
|
line = s.addLine(10, LINE_HEIGHT / 2, TOTAL_WIDTH, LINE_HEIGHT / 2)
|
||
|
|
line.setParentItem(r)
|
||
|
|
line.setPen(QPen(color, 5))
|
||
|
|
line.setZValue(-10)
|
||
|
|
|
||
|
|
plots.append((ID, r))
|
||
|
|
|
||
|
|
|
||
|
|
# Add Folders and Texts
|
||
|
|
|
||
|
|
|
||
|
|
# self.view.fitInView(0, 0, TOTAL_WIDTH, i * LINE_HEIGHT, Qt.KeepAspectRatioByExpanding) # KeepAspectRatio
|
||
|
|
self.view.setSceneRect()
|