Differential D11076 Diff 29164 plugins/python/comics_project_management_tools/comics_project_page_viewer.py

Changeset View

Standalone View

View Options

plugins/python/comics_project_management_tools/comics_project_page_viewer.py

1""" 1​"""
2Copyright (c) 2017 Wolthera van Hövell tot Westerflier <griffinvalley@gmail.com> 2​Copyright (c) 2017 Wolthera van Hövell tot Westerflier <griffinvalley@gmail.com>
3​Copyright (c) 2018 Ragnar Brynúlfsson <me@ragnarb.com>
3 4​
4This file is part of the Comics Project Management Tools(CPMT). 5​This file is part of the Comics Project Management Tools(CPMT).
5 6​
6CPMT is free software: you can redistribute it and/or modify 7​CPMT is free software: you can redistribute it and/or modify
7it under the terms of the GNU General Public License as published by 8​it under the terms of the GNU General Public License as published by
8the Free Software Foundation, either version 3 of the License, or 9​the Free Software Foundation, either version 3 of the License, or
9(at your option) any later version. 10​(at your option) any later version.
10 11​
11CPMT is distributed in the hope that it will be useful, 12​CPMT is distributed in the hope that it will be useful,
12but WITHOUT ANY WARRANTY; without even the implied warranty of 13​but WITHOUT ANY WARRANTY; without even the implied warranty of
13MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 14​MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14GNU General Public License for more details. 15​GNU General Public License for more details.
15 16​
16You should have received a copy of the GNU General Public License 17​You should have received a copy of the GNU General Public License
17along with the CPMT. If not, see <http://www.gnu.org/licenses/>. 18​along with the CPMT. If not, see <http://www.gnu.org/licenses/>.
18""" 19​"""
19 20​
20""" 21​"""
21This is a docker that shows your comic pages. 22​This is a docker that shows your comic pages.
22""" 23​"""
23 24​
25​import os
26​import sys
27​import json
28​import zipfile
24from PyQt5.QtGui import QImage, QPainter 29​from PyQt5.QtGui import QImage, QPainter
25from PyQt5.QtWidgets import QDialog, QWidget, QVBoxLayout, QSizePolicy, QDialogButtonBox 30​from PyQt5.QtWidgets import QDialog, QWidget, QHBoxLayout, QVBoxLayout, QPushButton, QSizePolicy, QDialogButtonBox, QShortcut, QLabel
26from PyQt5.QtCore import QSize, Qt 31​from PyQt5.QtCore import QSize, Qt
27from krita import * ​
28 32​
33​# To run standalon
34​from PyQt5.QtWidgets import QApplication
29 35​
30class page_viewer(QWidget): 36​
37​class page_viewer(QPushButton):
31 38​
32 def __init__(self, parent=None, flags=None): 39​ def __init__(self, parent=None, flags=None):
33 super(page_viewer, self).__init__(parent) 40​ super(page_viewer, self).__init__(parent)
41​ self.alignment = 'left'
34 self.image = QImage() 42​ self.image = QImage()
35 self.setSizePolicy(QSizePolicy.MinimumExpanding, QSizePolicy.MinimumExpanding) 43​ self.setSizePolicy(QSizePolicy.MinimumExpanding, QSizePolicy.MinimumExpanding)
36 44​
45​ def align_left(self):
46​ self.alignment = 'left'
47​
48​ def align_right(self):
49​ self.alignment = 'right'
50​
51​ def align_center(self):
52​ self.alignment = 'center'
53​
37 def set_image(self, image=QImage()): 54​ def set_image(self, image=QImage()):
38 self.image = image 55​ self.image = image
39 self.update() 56​ self.update()
40 57​
41 def paintEvent(self, event): 58​ def paintEvent(self, event):
42 painter = QPainter(self) 59​ painter = QPainter(self)
43 image = self.image.scaled(self.size(), Qt.KeepAspectRatio, Qt.SmoothTransformation) 60​ image = self.image.scaled(self.size(), Qt.KeepAspectRatio, Qt.SmoothTransformation)
44 painter.drawImage(0, 0, image) 61​ if self.alignment == 'right':
62​ x_offset = int(self.width() - image.width())
63​ elif self.alignment == 'center':
64​ x_offset = int((self.width() - image.width()) / 2)
65​ else:
66​ x_offset = 0
67​ painter.drawImage(x_offset, 0, image)
45 68​
46 def sizeHint(self): 69​ def sizeHint(self):
47 return QSize(256, 256) 70​ return QSize(256, 256)
48 71​
49 72​
50class comics_project_page_viewer(QDialog): 73​class comics_project_page_viewer(QDialog):
51 currentPageNumber = 0 74​ pageIndex = 0
75​ spread = True
52 76​
53 def __init__(self): 77​ def __init__(self):
54 super().__init__() 78​ super().__init__()
79​
55 self.setModal(False) 80​ self.setModal(False)
56 self.setWindowTitle(i18n("Comics page viewer.")) 81​ self.setWindowTitle('Untitled')
57 self.setMinimumSize(200, 200) 82​ self.setWindowFlags(
58 self.listOfImages = [QImage()] 83​ Qt.WindowTitleHint |
84​ Qt.WindowMinimizeButtonHint |
85​ Qt.WindowMaximizeButtonHint |
86​ Qt.WindowCloseButtonHint
87​ )
88​ self.resize(1024, 768)
59 self.setLayout(QVBoxLayout()) 89​ self.setLayout(QVBoxLayout())
60 90​
61 self.viewer = page_viewer() 91​ self.left_viewer = page_viewer()
62 self.layout().addWidget(self.viewer) 92​ self.left_viewer.align_right()
63 buttonBox = QDialogButtonBox(QDialogButtonBox.Close) 93​ self.left_viewer.clicked.connect(self.prev_page)
64 buttonBox.rejected.connect(self.close) 94​ self.right_viewer = page_viewer()
65 self.layout().addWidget(buttonBox) 95​ self.right_viewer.align_left()
96​ self.right_viewer.clicked.connect(self.next_page)
97​ self.single_viewer = page_viewer()
98​ self.single_viewer.align_center()
99​ self.single_viewer.clicked.connect(self.next_page)
100​ self.single_viewer.hide()
101​
102​ self.page_layout = QHBoxLayout()
103​ self.layout().addLayout(self.page_layout)
104​ self.page_layout.addWidget(self.left_viewer)
105​ self.page_layout.addWidget(self.right_viewer)
106​ self.page_layout.addWidget(self.single_viewer)
107​
108​ # Keyboard shortcuts
109​ self.home_shortcut = QShortcut('Home', self)
110​ self.home_shortcut.activated.connect(self.first_page)
111​ self.left_shortcut = QShortcut('Left', self)
112​ self.left_shortcut.activated.connect(self.prev_page)
113​ self.right_shortcut = QShortcut('Right', self)
114​ self.right_shortcut.activated.connect(self.next_page)
115​ self.space_shortcut = QShortcut('Space', self)
116​ self.space_shortcut.activated.connect(self.next_page)
117​ self.end_shortcut = QShortcut('End', self)
118​ self.end_shortcut.activated.connect(self.last_page)
119​
120​ self.first_btn = QPushButton('First')
121​ self.first_btn.clicked.connect(self.first_page)
woltheravUnsubmitted
Not Done

It might be an idea to, instead of using seperate shortcuts and pushbuttons, make QActions, assign a shortcut to those, and then make a QToolbutton that the QAction becomes the default action off. OR, even simpler, make a QMenu or QToolbar where you just set the layoutDirection instead of making the long list of removes and adds.

https://doc.qt.io/qt-5/qaction.html#shortcut-prop
https://doc.qt.io/qt-5/qwidget.html#layoutDirection-prop
https://doc.qt.io/qt-5/qmenu.html

woltherav: It might be an idea to, instead of using seperate shortcuts and pushbuttons, make QActions…
122​ self.prev_btn = QPushButton('Previous')
123​ self.prev_btn.clicked.connect(self.prev_page)
124​ self.spread_btn = QPushButton('Single Page')
125​ self.spread_btn.clicked.connect(self.toggle_spread)
126​ self.next_btn = QPushButton('Next')
127​ self.next_btn.clicked.connect(self.next_page)
128​ self.last_btn = QPushButton('Last')
129​ self.last_btn.clicked.connect(self.last_page)
130​
131​ self.buttons_layout = QHBoxLayout()
132​ self.layout().addLayout(self.buttons_layout)
133​ self.buttons_layout.addWidget(self.first_btn)
134​ self.buttons_layout.addWidget(self.prev_btn)
135​ self.buttons_layout.addWidget(self.spread_btn)
136​ self.buttons_layout.addWidget(self.next_btn)
137​ self.buttons_layout.addWidget(self.last_btn)
138​
139​ def load_comic(self, path_to_config):
140​ self.path_to_config = path_to_config
141​ configFile = open(self.path_to_config, "r", newline="", encoding="utf-16")
142​ self.setupDictionary = json.load(configFile)
143​ self.projecturl = os.path.dirname(str(self.path_to_config))
144​ if 'readingDirection' in self.setupDictionary:
145​ if self.setupDictionary['readingDirection'] == "leftToRight":
146​ self.setLayoutDirection(Qt.LeftToRight)
147​ self.left_shortcut.disconnect()
148​ self.right_shortcut.disconnect()
149​ self.left_shortcut.activated.connect(self.prev_page)
150​ self.right_shortcut.activated.connect(self.next_page)
151​ else:
152​ self.left_shortcut.disconnect()
153​ self.right_shortcut.disconnect()
154​ self.setLayoutDirection(Qt.RightToLeft)
155​ self.left_shortcut.activated.connect(self.next_page)
156​ self.right_shortcut.activated.connect(self.prev_page)
157​ else:
158​ self.left_shortcut.disconnect()
159​ self.right_shortcut.disconnect()
160​ self.setLayoutDirection(Qt.LeftToRight)
161​ self.left_shortcut.activated.connect(self.prev_page)
162​ self.right_shortcut.activated.connect(self.next_page)
163​ configFile.close()
164​
165​ def go_to_page_index(self, index):
166​ if index >= 0 and index < len(self.setupDictionary['pages']):
167​ self.pageIndex = index
168​ else:
169​ self.pageIndex = 0
170​ self.flip_page()
171​
172​ def toggle_spread(self):
173​ if self.spread:
174​ self.spread = False
175​ self.spread_btn.setText('Double Spread')
176​ self.update_single_page(self.pageIndex)
177​ self.left_viewer.hide()
178​ self.right_viewer.hide()
179​ self.single_viewer.show()
180​ self.flip_page()
181​ else:
182​ self.spread = True
183​ self.spread_btn.setText('Single Page')
184​ self.left_viewer.show()
185​ self.right_viewer.show()
186​ self.single_viewer.hide()
187​ self.flip_page()
188​
189​ def update_single_page(self, index):
190​ image = self.get_mergedimage(index)
191​ self.single_viewer.set_image(image)
192​
193​ def update_left_page(self, index):
194​ image = self.get_mergedimage(index)
195​ self.left_viewer.set_image(image)
196​
197​ def update_right_page(self, index):
198​ image = self.get_mergedimage(index)
199​ self.right_viewer.set_image(image)
200​
201​ def first_page(self):
202​ self.pageIndex = 0
203​ self.flip_page()
204​
205​ def prev_page(self):
206​ if self.pageIndex <= 0:
207​ self.pageIndex = len(self.setupDictionary['pages']) - 1
208​ else:
209​ if self.spread:
210​ self.pageIndex -= 2
211​ else:
212​ self.pageIndex -= 1
213​ self.flip_page()
214​
215​ def next_page(self):
216​ if self.pageIndex >= len(self.setupDictionary['pages']) - 1:
217​ self.pageIndex = 0
218​ else:
219​ if self.spread:
220​ self.pageIndex += 2
221​ else:
222​ self.pageIndex += 1
223​ self.flip_page()
224​
225​ def last_page(self):
226​ self.pageIndex = len(self.setupDictionary['pages']) - 1
227​ self.flip_page()
228​
229​ def flip_page(self):
230​ if self.spread:
231​ if self.pageIndex % 2 == 0: # Even/Left
232​ left_page_number = self.pageIndex - 1
233​ right_page_number = self.pageIndex
234​ else: # Odd/Right
235​ left_page_number = self.pageIndex
236​ right_page_number = self.pageIndex + 1
237​ self.update_left_page(left_page_number)
238​ self.update_right_page(right_page_number)
239​ if left_page_number < 0:
240​ page_numbers = str(right_page_number + 1)
241​ elif right_page_number >= len(self.setupDictionary['pages']):
242​ page_numbers = str(left_page_number + 1)
243​ else:
244​ page_numbers = '{left} / {right}'.format(
245​ left = str(left_page_number + 1),
246​ right = str(right_page_number + 1))
247​ self.setWindowTitle('{name} - {page_numbers}'.format(
248​ name = self.setupDictionary['projectName'],
249​ page_numbers = page_numbers))
250​ else:
251​ if self.pageIndex >= len(self.setupDictionary['pages']):
252​ self.pageIndex = len(self.setupDictionary['pages']) - 1
253​ if self.pageIndex < 0:
254​ self.pageIndex = 0
255​ self.update_single_page(self.pageIndex)
256​ self.setWindowTitle('{name} - {page_numbers}'.format(
257​ name = self.setupDictionary['projectName'],
258​ page_numbers = str(self.pageIndex + 1)))
259​
260​
261​ def get_mergedimage(self, index):
262​ if index < len(self.setupDictionary['pages']) and index > -1:
263​ image_url = os.path.join(self.projecturl, self.setupDictionary['pages'][index])
264​ if os.path.exists(image_url):
265​ page = zipfile.ZipFile(image_url, "r")
266​ image = QImage.fromData(page.read("mergedimage.png"))
267​ page.close()
268​ return image
269​ image = QImage(QSize(10, 10), QImage.Format_ARGB32)
270​ image.fill(Qt.GlobalColor(19))
271​ return image
272​
273​if __name__ == '__main__':
274​ ''' Run the page viewer outside Krita '''
275​ app = QApplication(sys.argv)
276​ if sys.argv:
277​ print(sys.argv)
278​ path_to_config = sys.argv[1]
279​ if os.path.exists(path_to_config):
280​ print('Run comics viewer')
281​ page_viewer_dialog = comics_project_page_viewer()
282​ page_viewer_dialog.load_comic(path_to_config)
283​ page_viewer_dialog.go_to_page_index(0)
284​ page_viewer_dialog.show()
285​ else:
286​ print('No comic found in {comic}'.format(comic=comic))
287​ else:
288​ print('Pass the path to a Krita comicConfig.json file to this script, to view the comic.')
289​
290​ sys.exit(app.exec_())
291​
292​
66 293​
67 def update_image(self, image=QImage()): ​
68 self.viewer.set_image(image) ​