mirror of
https://github.com/olivierkes/manuskript.git
synced 2024-06-13 16:34:34 +12:00
277 lines
11 KiB
Python
277 lines
11 KiB
Python
#!/usr/bin/env python
|
|
#--!-- coding: utf8 --!--
|
|
|
|
from __future__ import print_function
|
|
from __future__ import unicode_literals
|
|
|
|
from PyQt4.QtCore import *
|
|
from PyQt4.QtGui import *
|
|
|
|
from ui.mainWindow import *
|
|
from loadSave import *
|
|
|
|
class MainWindow(QMainWindow, Ui_MainWindow):
|
|
def __init__(self):
|
|
QMainWindow.__init__(self)
|
|
self.setupUi(self)
|
|
|
|
self.readSettings()
|
|
|
|
# UI
|
|
self.splitterPersos.setStretchFactor(0, 25)
|
|
self.splitterPersos.setStretchFactor(1, 75)
|
|
|
|
self.splitterPlot.setStretchFactor(0, 20)
|
|
self.splitterPlot.setStretchFactor(1, 40)
|
|
self.splitterPlot.setStretchFactor(2, 40)
|
|
|
|
|
|
self.splitterPlan.setStretchFactor(0, 25)
|
|
self.splitterPlan.setStretchFactor(1, 75)
|
|
|
|
|
|
self.splitterRedac.setStretchFactor(0, 20)
|
|
self.splitterRedac.setStretchFactor(1, 60)
|
|
self.splitterRedac.setStretchFactor(2, 20)
|
|
|
|
# Word count
|
|
self.mprWordCount = QSignalMapper(self)
|
|
for t, i in [
|
|
(self.txtSummarySentance, 0),
|
|
(self.txtSummaryPara, 1),
|
|
(self.txtSummaryPage, 2),
|
|
(self.txtSummaryFull, 3)
|
|
]:
|
|
t.textChanged.connect(self.mprWordCount.map)
|
|
self.mprWordCount.setMapping(t, i)
|
|
self.mprWordCount.mapped.connect(self.wordCount)
|
|
|
|
# Snowflake Method Cycle
|
|
self.mapperCycle = QSignalMapper(self)
|
|
for t, i in [
|
|
(self.btnStepTwo, 0),
|
|
(self.btnStepThree, 1),
|
|
(self.btnStepFour, 2),
|
|
(self.btnStepFive, 3),
|
|
(self.btnStepSix, 4),
|
|
(self.btnStepSeven, 5),
|
|
(self.btnStepEight, 6)
|
|
]:
|
|
t.clicked.connect(self.mapperCycle.map)
|
|
self.mapperCycle.setMapping(t, i)
|
|
|
|
self.mapperCycle.mapped.connect(self.clickCycle)
|
|
|
|
|
|
# Données
|
|
self.mdlFlatData = QStandardItemModel(2, 8)
|
|
self.tblDebugFlatData.setModel(self.mdlFlatData)
|
|
|
|
self.mprSummary = QDataWidgetMapper()
|
|
self.mprSummary.setModel(self.mdlFlatData)
|
|
self.mprSummary.addMapping(self.txtSummarySentance, 0)
|
|
self.mprSummary.addMapping(self.txtSummarySentance_2, 0)
|
|
self.mprSummary.addMapping(self.txtSummaryPara, 1)
|
|
self.mprSummary.addMapping(self.txtSummaryPara_2, 1)
|
|
self.mprSummary.addMapping(self.txtPlotSummaryPara, 1)
|
|
self.mprSummary.addMapping(self.txtSummaryPage, 2)
|
|
self.mprSummary.addMapping(self.txtSummaryPage_2, 2)
|
|
self.mprSummary.addMapping(self.txtPlotSummaryPage, 2)
|
|
self.mprSummary.addMapping(self.txtSummaryFull, 3)
|
|
self.mprSummary.addMapping(self.txtPlotSummaryFull, 3)
|
|
self.mprSummary.setCurrentIndex(1)
|
|
|
|
self.mprInfos = QDataWidgetMapper()
|
|
self.mprInfos.setModel(self.mdlFlatData)
|
|
self.mprInfos.addMapping(self.txtGeneralTitle, 0)
|
|
self.mprInfos.addMapping(self.txtGeneralSubtitle, 1)
|
|
self.mprInfos.addMapping(self.txtGeneralSerie, 2)
|
|
self.mprInfos.addMapping(self.txtGeneralVolume, 3)
|
|
self.mprInfos.addMapping(self.txtGeneralGenre, 4)
|
|
self.mprInfos.addMapping(self.txtGeneralLicense, 5)
|
|
self.mprInfos.addMapping(self.txtGeneralAuthor, 6)
|
|
self.mprInfos.addMapping(self.txtGeneralEmail, 7)
|
|
self.mprInfos.setCurrentIndex(0)
|
|
|
|
# Persos
|
|
self.mdlPersos = QStandardItemModel(0, 10)
|
|
self.mdlPersosInfos = QStandardItemModel(1, 0)
|
|
self.mdlPersosInfos.insertColumn(0, [QStandardItem(i) for i in ["ID"]])
|
|
self.mdlPersosInfos.setHorizontalHeaderLabels(["Description"])
|
|
self.lstPersos.setModel(self.mdlPersos)
|
|
self.tblDebugPersos.setModel(self.mdlPersos)
|
|
self.tblPersoInfos.setModel(self.mdlPersosInfos)
|
|
self.tblPersoInfos.setRowHidden(0, True)
|
|
self.tblDebugPersosInfos.setModel(self.mdlPersosInfos)
|
|
|
|
self.btnAddPerso.clicked.connect(self.createPerso)
|
|
self.btnRmPerso.clicked.connect(self.removePerso)
|
|
self.btnPersoAddInfo.clicked.connect(lambda: self.mdlPersosInfos.insertRow(self.mdlPersosInfos.rowCount()))
|
|
self.mprPersos = QDataWidgetMapper()
|
|
self.mprPersos.setModel(self.mdlPersos)
|
|
|
|
mapping = [
|
|
(self.txtPersoName, 0),
|
|
(self.txtPersoMotivation, 2),
|
|
(self.txtPersoGoal, 3),
|
|
(self.txtPersoConflict, 4),
|
|
(self.txtPersoEpiphany, 5),
|
|
(self.txtPersoSummarySentance, 6),
|
|
(self.txtPersoSummaryPara, 7),
|
|
(self.txtPersoSummaryFull, 8)
|
|
]
|
|
for w, i in mapping:
|
|
self.mprPersos.addMapping(w, i)
|
|
self.mprPersos.addMapping(self.sldPersoImportance, 8, "importance")
|
|
self.sldPersoImportance.importanceChanged.connect(self.mprPersos.submit)
|
|
|
|
self.mprPersos.setCurrentIndex(0)
|
|
self.lstPersos.selectionModel().currentChanged.connect(self.mprPersos.setCurrentModelIndex)
|
|
self.lstPersos.selectionModel().currentChanged.connect(self.changeCurrentPerso)
|
|
self.tabPersos.currentChanged.connect(self.resizePersosInfos)
|
|
|
|
#Debug
|
|
self.mdlFlatData.setVerticalHeaderLabels(["Infos générales", "Summary"])
|
|
self.tblDebugFlatData.setModel(self.mdlFlatData)
|
|
|
|
|
|
self.loadProject("test_project")
|
|
|
|
####################################################################################################
|
|
# PERSOS #
|
|
####################################################################################################
|
|
|
|
def createPerso(self):
|
|
"Creates a perso by adding a row in mdlPersos and a column in mdlPersosInfos with same ID"
|
|
p = QStandardItem("Nouveau perso")
|
|
self.mdlPersos.appendRow(p)
|
|
pid = self.getPersosID()
|
|
self.checkPersosID() # Attributes a persoID (which is logically pid)
|
|
|
|
# Add column in persos infos
|
|
self.mdlPersosInfos.insertColumn(self.mdlPersosInfos.columnCount(), [QStandardItem(pid)])
|
|
self.mdlPersosInfos.setHorizontalHeaderItem(self.mdlPersosInfos.columnCount()-1, QStandardItem("Valeur"))
|
|
|
|
def getPersosID(self):
|
|
"Returns an unused perso ID (row 1)"
|
|
vals = []
|
|
for i in range(self.mdlPersos.rowCount()):
|
|
item = self.mdlPersos.item(i, 1)
|
|
if item and item.text():
|
|
vals.append(int(item.text()))
|
|
|
|
k = 0
|
|
while k in vals: k += 1
|
|
return str(k)
|
|
|
|
def checkPersosID(self):
|
|
"Checks whether some persos ID (row 1) are empty, if so, assign an ID"
|
|
empty = []
|
|
for i in range(self.mdlPersos.rowCount()):
|
|
item = self.mdlPersos.item(i, 1)
|
|
if not item:
|
|
item = QStandardItem()
|
|
item.setText(self.getPersosID())
|
|
self.mdlPersos.setItem(i, 1, item)
|
|
|
|
def removePerso(self):
|
|
i = self.lstPersos.currentIndex()
|
|
self.mdlPersos.takeRow(i.row())
|
|
self.mdlPersosInfos.takeColumn(i.row()+1)
|
|
|
|
def changeCurrentPerso(self, trash=None):
|
|
idx = self.lstPersos.currentIndex()
|
|
pid = self.mdlPersos.item(idx.row(), 1).text()
|
|
for c in range(self.mdlPersosInfos.columnCount()):
|
|
pid2 = self.mdlPersosInfos.item(0, c).text()
|
|
self.tblPersoInfos.setColumnHidden(c, c <> 0 and pid <> pid2)
|
|
|
|
self.resizePersosInfos()
|
|
|
|
def resizePersosInfos(self):
|
|
self.tblPersoInfos.resizeColumnToContents(0)
|
|
w = self.tblPersoInfos.viewport().width()
|
|
w2 = self.tblPersoInfos.columnWidth(0)
|
|
current = self.lstPersos.currentIndex().row() + 1
|
|
self.tblPersoInfos.setColumnWidth(current, w - w2)
|
|
|
|
|
|
####################################################################################################
|
|
# GENERAL #
|
|
####################################################################################################
|
|
|
|
def loadProject(self, project):
|
|
self.currentProject = project
|
|
loadStandardItemModelXML(self.mdlFlatData, "{}/flatModel.xml".format(project))
|
|
loadStandardItemModelXML(self.mdlPersos, "{}/perso.xml".format(project))
|
|
loadStandardItemModelXML(self.mdlPersosInfos, "{}/persoInfos.xml".format(project))
|
|
|
|
# Stuff
|
|
self.checkPersosID()
|
|
|
|
def readSettings(self):
|
|
# Load State and geometry
|
|
settings = QSettings(qApp.organizationName(), qApp.applicationName())
|
|
self.restoreGeometry(settings.value("geometry").toByteArray())
|
|
self.restoreState(settings.value("windowState").toByteArray())
|
|
|
|
def closeEvent(self, event):
|
|
# Save State and geometry
|
|
settings = QSettings(qApp.organizationName(), qApp.applicationName())
|
|
settings.setValue("geometry", self.saveGeometry())
|
|
settings.setValue("windowState", self.saveState())
|
|
|
|
# Save data from models
|
|
saveStandardItemModelXML(self.mdlFlatData, "{}/flatModel.xml".format(self.currentProject))
|
|
saveStandardItemModelXML(self.mdlPersos, "{}/perso.xml".format(self.currentProject))
|
|
saveStandardItemModelXML(self.mdlPersosInfos, "{}/persoInfos.xml".format(self.currentProject))
|
|
|
|
|
|
# closeEvent
|
|
QMainWindow.closeEvent(self, event)
|
|
|
|
def clickCycle(self, i):
|
|
if i == 0: # step 2 - paragraph summary
|
|
self.tabMain.setCurrentIndex(1)
|
|
self.tabSummary.setCurrentIndex(1)
|
|
if i == 1: # step 3 - characters summary
|
|
self.tabMain.setCurrentIndex(2)
|
|
self.tabPersos.setCurrentIndex(0)
|
|
if i == 2: # step 4 - page summary
|
|
self.tabMain.setCurrentIndex(1)
|
|
self.tabSummary.setCurrentIndex(2)
|
|
if i == 3: # step 5 - characters description
|
|
self.tabMain.setCurrentIndex(2)
|
|
self.tabPersos.setCurrentIndex(1)
|
|
if i == 4: # step 6 - four page synopsis
|
|
self.tabMain.setCurrentIndex(1)
|
|
self.tabSummary.setCurrentIndex(3)
|
|
if i == 5: # step 7 - full character charts
|
|
self.tabMain.setCurrentIndex(2)
|
|
self.tabPersos.setCurrentIndex(2)
|
|
if i == 6: # step 8 - scene list
|
|
self.tabMain.setCurrentIndex(3)
|
|
|
|
|
|
|
|
"Updates word counts over tabs"
|
|
def wordCount(self, i):
|
|
|
|
src= {
|
|
0:self.txtSummarySentance,
|
|
1:self.txtSummaryPara,
|
|
2:self.txtSummaryPage,
|
|
3:self.txtSummaryFull
|
|
}[i]
|
|
|
|
lbl = {
|
|
0:self.lblSummaryWCSentance,
|
|
1:self.lblSummaryWCPara,
|
|
2:self.lblSummaryWCPage,
|
|
3:self.lblSummaryWCFull
|
|
}[i]
|
|
|
|
wc = len(src.toPlainText().trimmed().split(" ")) if src.toPlainText() else 0
|
|
if i in [2, 3]: pages = " (~{} pages)".format(int(wc / 25) / 10.)
|
|
else: pages = ""
|
|
lbl.setText("Mots: {}{}".format(wc, pages)) |