Veronica K. B. Olsen
226 lines
7.0 KiB
Python
226 lines
7.0 KiB
Python
# -*- coding: utf-8 -*-
|
||
"""novelWriter GUI Project Outline
|
||
|
||
novelWriter – GUI Project Outline
|
||
===================================
|
||
Class holding the project outline view
|
||
|
||
File History:
|
||
Created: 2019-11-16 [0.4.1]
|
||
|
||
"""
|
||
|
||
import logging
|
||
import nw
|
||
|
||
from os import path
|
||
|
||
from PyQt5.QtCore import Qt
|
||
from PyQt5.QtWidgets import (
|
||
QWidget, QVBoxLayout, QTreeWidget, QTreeWidgetItem
|
||
)
|
||
|
||
from nw.constants import nwItemLayout, nwKeyWords, nwLabels
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
class GuiProjectOutline(QWidget):
|
||
|
||
I_TITLE = 0
|
||
I_LEVEL = 1
|
||
I_LABEL = 2
|
||
I_LINE = 3
|
||
I_WCOUNT = 4
|
||
I_CCOUNT = 5
|
||
I_PCOUNT = 6
|
||
I_SYNOP = 7
|
||
I_POV = 8
|
||
I_CHAR = 9
|
||
I_PLOT = 10
|
||
I_TIME = 11
|
||
I_WORLD = 12
|
||
I_OBJECT = 13
|
||
I_ENTITY = 14
|
||
I_CUSTOM = 15
|
||
|
||
COL_ORDER = [
|
||
I_TITLE, I_LEVEL, I_LABEL, I_LINE,
|
||
I_WCOUNT, I_CCOUNT, I_PCOUNT, I_SYNOP,
|
||
I_POV, I_CHAR, I_PLOT, I_TIME,
|
||
I_WORLD, I_OBJECT, I_ENTITY, I_CUSTOM,
|
||
]
|
||
|
||
COL_LABELS = {
|
||
I_TITLE : "Title",
|
||
I_LEVEL : "Level",
|
||
I_LABEL : "Document",
|
||
I_LINE : "Line",
|
||
I_WCOUNT : "Words",
|
||
I_CCOUNT : "Chars",
|
||
I_PCOUNT : "Pars",
|
||
I_SYNOP : "Synopsis",
|
||
I_POV : nwLabels.KEY_NAME[nwKeyWords.POV_KEY],
|
||
I_CHAR : nwLabels.KEY_NAME[nwKeyWords.CHAR_KEY],
|
||
I_PLOT : nwLabels.KEY_NAME[nwKeyWords.PLOT_KEY],
|
||
I_TIME : nwLabels.KEY_NAME[nwKeyWords.TIME_KEY],
|
||
I_WORLD : nwLabels.KEY_NAME[nwKeyWords.WORLD_KEY],
|
||
I_OBJECT : nwLabels.KEY_NAME[nwKeyWords.OBJECT_KEY],
|
||
I_ENTITY : nwLabels.KEY_NAME[nwKeyWords.ENTITY_KEY],
|
||
I_CUSTOM : nwLabels.KEY_NAME[nwKeyWords.CUSTOM_KEY],
|
||
}
|
||
|
||
def __init__(self, theParent, theProject):
|
||
QWidget.__init__(self, theParent)
|
||
|
||
logger.debug("Initialising ProjectOutline ...")
|
||
|
||
self.mainConf = nw.CONFIG
|
||
self.theParent = theParent
|
||
self.theProject = theProject
|
||
self.theIndex = self.theParent.theIndex
|
||
|
||
self.showWords = True
|
||
self.showSynopsis = True
|
||
self.showFilePath = False
|
||
|
||
self.outerBox = QVBoxLayout()
|
||
self.mainTree = QTreeWidget()
|
||
self.treeOrder = self.COL_ORDER
|
||
|
||
self.outerBox.addWidget(self.mainTree)
|
||
self.outerBox.setContentsMargins(0,0,0,0)
|
||
self.setLayout(self.outerBox)
|
||
|
||
logger.debug("ProjectOutline initialisation complete")
|
||
|
||
return
|
||
|
||
def populateTree(self):
|
||
|
||
theLabels = []
|
||
for n in self.treeOrder:
|
||
theLabels.append(self.COL_LABELS[n])
|
||
|
||
self.mainTree.clear()
|
||
self.mainTree.setHeaderLabels(theLabels)
|
||
|
||
treeHead = self.mainTree.headerItem()
|
||
treeHead.setTextAlignment(self.I_CCOUNT,Qt.AlignRight)
|
||
treeHead.setTextAlignment(self.I_WCOUNT,Qt.AlignRight)
|
||
treeHead.setTextAlignment(self.I_PCOUNT,Qt.AlignRight)
|
||
|
||
currTitle = None
|
||
currChapter = None
|
||
currScene = None
|
||
|
||
for tHandle in self.theProject.treeOrder:
|
||
|
||
if tHandle not in self.theIndex.novelIndex:
|
||
continue
|
||
|
||
nwItem = self.theProject.getItem(tHandle)
|
||
if nwItem.itemLayout == nwItemLayout.NOTE:
|
||
continue
|
||
|
||
for tEntry in self.theIndex.novelIndex[tHandle]:
|
||
|
||
nTitle = str(tEntry[0])
|
||
tTitle = tEntry[2]
|
||
tLevel = str(tEntry[1])
|
||
tLabel = nwItem.itemName
|
||
tItem = self._createTreeItem(tHandle, nTitle, tTitle, tLevel, tLabel)
|
||
|
||
if tEntry[1] == 1:
|
||
currTitle = tItem
|
||
self.mainTree.addTopLevelItem(tItem)
|
||
elif tEntry[1] == 2:
|
||
if currTitle is None:
|
||
self.mainTree.addTopLevelItem(tItem)
|
||
else:
|
||
currTitle.addChild(tItem)
|
||
currChapter = tItem
|
||
elif tEntry[1] == 3:
|
||
if currChapter is None:
|
||
if currTitle is None:
|
||
self.mainTree.addTopLevelItem(tItem)
|
||
else:
|
||
currTitle.addChild(tItem)
|
||
else:
|
||
currChapter.addChild(tItem)
|
||
currScene = tItem
|
||
elif tEntry[1] == 4:
|
||
if currScene is None:
|
||
if currChapter is None:
|
||
if currTitle is None:
|
||
self.mainTree.addTopLevelItem(tItem)
|
||
else:
|
||
currTitle.addChild(tItem)
|
||
else:
|
||
currChapter.addChild(tItem)
|
||
else:
|
||
currScene.addChild(tItem)
|
||
|
||
tItem.setExpanded(True)
|
||
|
||
return
|
||
|
||
##
|
||
# Internal Functions
|
||
##
|
||
|
||
def _createTreeItem(self, tHandle, nTitle, tTitle, tLevel, tLabel):
|
||
|
||
newItem = QTreeWidgetItem()
|
||
newItem.setText(self.I_TITLE, tTitle)
|
||
newItem.setText(self.I_LEVEL, tLevel)
|
||
newItem.setText(self.I_LABEL, tLabel)
|
||
newItem.setText(self.I_LINE, nTitle)
|
||
|
||
cC, wC, pC = self.theIndex.getCounts(tHandle, nTitle)
|
||
newItem.setText(self.I_CCOUNT, str(cC))
|
||
newItem.setText(self.I_WCOUNT, str(wC))
|
||
newItem.setText(self.I_PCOUNT, str(pC))
|
||
newItem.setTextAlignment(self.I_CCOUNT,Qt.AlignRight)
|
||
newItem.setTextAlignment(self.I_WCOUNT,Qt.AlignRight)
|
||
newItem.setTextAlignment(self.I_PCOUNT,Qt.AlignRight)
|
||
|
||
povList = []
|
||
charList = []
|
||
plotList = []
|
||
timeList = []
|
||
worldList = []
|
||
objectList = []
|
||
entityList = []
|
||
customList = []
|
||
for tKey, tTag in self.theIndex.getReferences(tHandle, nTitle):
|
||
if tKey == nwKeyWords.POV_KEY:
|
||
povList.append(tTag)
|
||
elif tKey == nwKeyWords.CHAR_KEY:
|
||
charList.append(tTag)
|
||
elif tKey == nwKeyWords.PLOT_KEY:
|
||
plotList.append(tTag)
|
||
elif tKey == nwKeyWords.TIME_KEY:
|
||
timeList.append(tTag)
|
||
elif tKey == nwKeyWords.WORLD_KEY:
|
||
worldList.append(tTag)
|
||
elif tKey == nwKeyWords.OBJECT_KEY:
|
||
objectList.append(tTag)
|
||
elif tKey == nwKeyWords.ENTITY_KEY:
|
||
entityList.append(tTag)
|
||
elif tKey == nwKeyWords.CUSTOM_KEY:
|
||
customList.append(tTag)
|
||
|
||
newItem.setText(self.I_POV, ", ".join(povList))
|
||
newItem.setText(self.I_CHAR, ", ".join(charList))
|
||
newItem.setText(self.I_PLOT, ", ".join(plotList))
|
||
newItem.setText(self.I_TIME, ", ".join(timeList))
|
||
newItem.setText(self.I_WORLD, ", ".join(worldList))
|
||
newItem.setText(self.I_OBJECT, ", ".join(objectList))
|
||
newItem.setText(self.I_ENTITY, ", ".join(entityList))
|
||
newItem.setText(self.I_CUSTOM, ", ".join(customList))
|
||
|
||
return newItem
|
||
|
||
|
||
# END Class GuiProjectOutline
|