Differential D11076 Diff 28873 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

Context not available.
21This is a docker that shows your comic pages. 21​This is a docker that shows your comic pages.
22""" 22​"""
23 23​
24from PyQt5.QtGui import QImage, QPainter 24​import os
25from PyQt5.QtWidgets import QDialog, QWidget, QVBoxLayout, QSizePolicy, QDialogButtonBox 25​import sys
26​import json
27​import zipfile
28​from PyQt5.QtGui import QImage, QPainter, QKeySequence
29​from PyQt5.QtWidgets import QDialog, QWidget, QHBoxLayout, QVBoxLayout, QPushButton, QSizePolicy, QDialogButtonBox, QShortcut, QLabel
26from PyQt5.QtCore import QSize, Qt 30​from PyQt5.QtCore import QSize, Qt
27from krita import * ​
28 31​
32​# To run standalon
33​from PyQt5.QtWidgets import QApplication
29 34​
30class page_viewer(QWidget): 35​
36​class page_viewer(QPushButton):
31 37​
32 def __init__(self, parent=None, flags=None): 38​ def __init__(self, parent=None, flags=None):
33 super(page_viewer, self).__init__(parent) 39​ super(page_viewer, self).__init__(parent)
40​ self.alignment = 'left'
34 self.image = QImage() 41​ self.image = QImage()
35 self.setSizePolicy(QSizePolicy.MinimumExpanding, QSizePolicy.MinimumExpanding) 42​ self.setSizePolicy(QSizePolicy.MinimumExpanding, QSizePolicy.MinimumExpanding)
36 43​
44​ def align_left(self):
45​ self.alignment = 'left'
46​
47​ def align_right(self):
48​ self.alignment = 'right'
49​
50​ def align_center(self):
51​ self.alignment = 'center'
52​
37 def set_image(self, image=QImage()): 53​ def set_image(self, image=QImage()):
38 self.image = image 54​ self.image = image
39 self.update() 55​ self.update()
Context not available.
41 def paintEvent(self, event): 57​ def paintEvent(self, event):
42 painter = QPainter(self) 58​ painter = QPainter(self)
43 image = self.image.scaled(self.size(), Qt.KeepAspectRatio, Qt.SmoothTransformation) 59​ image = self.image.scaled(self.size(), Qt.KeepAspectRatio, Qt.SmoothTransformation)
44 painter.drawImage(0, 0, image) 60​ if self.alignment == 'right':
61​ x_offset = int(self.width() - image.width())
62​ elif self.alignment == 'center':
63​ x_offset = int((self.width() - image.width()) / 2)
64​ else:
65​ x_offset = 0
66​ painter.drawImage(x_offset, 0, image)
45 67​
46 def sizeHint(self): 68​ def sizeHint(self):
47 return QSize(256, 256) 69​ return QSize(256, 256)
48 70​
49 71​
50class comics_project_page_viewer(QDialog): 72​class comics_project_page_viewer(QDialog):
51 currentPageNumber = 0 73​ pageIndex = 0
74​ spread = True
52 75​
53 def __init__(self): 76​ def __init__(self):
54 super().__init__() 77​ super().__init__()
78​
55 self.setModal(False) 79​ self.setModal(False)
56 self.setWindowTitle(i18n("Comics page viewer.")) 80​ self.setWindowTitle('Untitled')
57 self.setMinimumSize(200, 200) 81​ self.setWindowFlags(
58 self.listOfImages = [QImage()] 82​ Qt.WindowTitleHint |
83​ Qt.WindowMinimizeButtonHint |
84​ Qt.WindowMaximizeButtonHint |
85​ Qt.WindowCloseButtonHint
86​ )
87​ self.resize(1024, 768)
59 self.setLayout(QVBoxLayout()) 88​ self.setLayout(QVBoxLayout())
60 89​
61 self.viewer = page_viewer() 90​ self.left_viewer = page_viewer()
62 self.layout().addWidget(self.viewer) 91​ self.left_viewer.align_right()
63 buttonBox = QDialogButtonBox(QDialogButtonBox.Close) 92​ self.left_viewer.clicked.connect(self.prev_page)
64 buttonBox.rejected.connect(self.close) 93​ self.right_viewer = page_viewer()
65 self.layout().addWidget(buttonBox) 94​ self.right_viewer.align_left()
95​ self.right_viewer.clicked.connect(self.next_page)
96​ self.single_viewer = page_viewer()
97​ self.single_viewer.align_center()
98​ self.single_viewer.clicked.connect(self.next_page)
99​ self.single_viewer.hide()
100​
101​ self.page_layout = QHBoxLayout()
102​ self.layout().addLayout(self.page_layout)
103​ self.page_layout.addWidget(self.left_viewer)
104​ self.page_layout.addWidget(self.right_viewer)
105​ self.page_layout.addWidget(self.single_viewer)
106​
107​ # Keyboard shortcuts
108​ self.home_shortcut = QShortcut('Home', self)
109​ self.home_shortcut.activated.connect(self.first_page)
110​ self.left_shortcut = QShortcut('Left', self)
111​ self.left_shortcut.activated.connect(self.prev_page)
112​ self.right_shortcut = QShortcut('Right', self)
113​ self.right_shortcut.activated.connect(self.next_page)
114​ self.space_shortcut = QShortcut('Space', self)
115​ self.space_shortcut.activated.connect(self.next_page)
116​ self.end_shortcut = QShortcut(QKeySequence(Qt.Key_End), self)
117​ self.end_shortcut.activated.connect(self.last_page)
118​
119​ self.first_btn = QPushButton('First')
120​ self.first_btn.clicked.connect(self.first_page)
121​ self.prev_btn = QPushButton('Previous')
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.clicked.connect(self.prev_page)
123​ self.spread_btn = QPushButton('Single Page')
124​ self.spread_btn.clicked.connect(self.toggle_spread)
125​ self.next_btn = QPushButton('Next')
126​ self.next_btn.clicked.connect(self.next_page)
127​ self.last_btn = QPushButton('Last')
128​ self.last_btn.clicked.connect(self.last_page)
129​
130​ self.buttons_layout = QHBoxLayout()
131​ self.layout().addLayout(self.buttons_layout)
132​ self.buttons_layout.addWidget(self.first_btn)
133​ self.buttons_layout.addWidget(self.prev_btn)
134​ self.buttons_layout.addWidget(self.spread_btn)
135​ self.buttons_layout.addWidget(self.next_btn)
136​ self.buttons_layout.addWidget(self.last_btn)
137​
138​ def load_comic(self, path_to_config):
139​ self.path_to_config = path_to_config
140​ configFile = open(self.path_to_config, "r", newline="", encoding="utf-16")
141​ self.setupDictionary = json.load(configFile)
142​ self.projecturl = os.path.dirname(str(self.path_to_config))
143​ if 'readingDirection' in self.setupDictionary:
144​ self.set_reading_direction(self.setupDictionary['readingDirection'])
145​ else:
146​ self.set_reading_direction("leftToRight")
147​ configFile.close()
148​
149​ def set_reading_direction(self, direction):
150​ if direction == "rightToLeft":
151​ self.page_layout.removeWidget(self.right_viewer)
152​ self.page_layout.insertWidget(0, self.right_viewer)
153​ self.left_viewer.align_left()
154​ self.right_viewer.align_right()
155​ self.left_shortcut.activated.disconnect()
156​ self.left_shortcut.activated.connect(self.next_page)
157​ self.right_shortcut.activated.disconnect()
158​ self.right_shortcut.activated.connect(self.prev_page)
159​ self.buttons_layout.removeWidget(self.first_btn)
160​ self.buttons_layout.removeWidget(self.prev_btn)
161​ self.buttons_layout.removeWidget(self.spread_btn)
162​ self.buttons_layout.removeWidget(self.next_btn)
163​ self.buttons_layout.removeWidget(self.last_btn)
164​ self.buttons_layout.addWidget(self.last_btn)
165​ self.buttons_layout.addWidget(self.next_btn)
166​ self.buttons_layout.addWidget(self.spread_btn)
167​ self.buttons_layout.addWidget(self.prev_btn)
168​ self.buttons_layout.addWidget(self.first_btn)
169​ else: # rightToLeft
170​ self.page_layout.removeWidget(self.left_viewer)
171​ self.page_layout.insertWidget(0, self.left_viewer)
172​ self.left_viewer.align_right()
173​ self.right_viewer.align_left()
174​ self.left_shortcut.activated.disconnect()
175​ self.left_shortcut.activated.connect(self.prev_page)
176​ self.right_shortcut.activated.disconnect()
177​ self.right_shortcut.activated.connect(self.next_page)
178​ self.buttons_layout.removeWidget(self.first_btn)
179​ self.buttons_layout.removeWidget(self.prev_btn)
180​ self.buttons_layout.removeWidget(self.spread_btn)
181​ self.buttons_layout.removeWidget(self.next_btn)
182​ self.buttons_layout.removeWidget(self.last_btn)
183​ self.buttons_layout.addWidget(self.first_btn)
184​ self.buttons_layout.addWidget(self.prev_btn)
185​ self.buttons_layout.addWidget(self.spread_btn)
186​ self.buttons_layout.addWidget(self.next_btn)
187​ self.buttons_layout.addWidget(self.last_btn)
188​
189​ def go_to_page_index(self, index):
190​ if index >= 0 and index < len(self.setupDictionary['pages']):
191​ self.pageIndex = index
192​ else:
193​ self.pageIndex = 0
194​ self.flip_page()
195​
196​ def toggle_spread(self):
197​ if self.spread:
198​ self.spread = False
199​ self.spread_btn.setText('Double Spread')
200​ self.update_single_page(self.pageIndex)
201​ self.left_viewer.hide()
202​ self.right_viewer.hide()
203​ self.single_viewer.show()
204​ self.flip_page()
205​ else:
206​ self.spread = True
207​ self.spread_btn.setText('Single Page')
208​ self.left_viewer.show()
209​ self.right_viewer.show()
210​ self.single_viewer.hide()
211​ self.flip_page()
212​
213​ def update_single_page(self, index):
214​ image = self.get_mergedimage(index)
215​ self.single_viewer.set_image(image)
216​
217​ def update_left_page(self, index):
218​ image = self.get_mergedimage(index)
219​ self.left_viewer.set_image(image)
220​
221​ def update_right_page(self, index):
222​ image = self.get_mergedimage(index)
223​ self.right_viewer.set_image(image)
224​
225​ def first_page(self):
226​ self.pageIndex = 0
227​ self.flip_page()
228​
229​ def prev_page(self):
230​ if self.pageIndex <= 0:
231​ self.pageIndex = len(self.setupDictionary['pages']) - 1
232​ else:
233​ if self.spread:
234​ self.pageIndex -= 2
235​ else:
236​ self.pageIndex -= 1
237​ self.flip_page()
238​
239​ def next_page(self):
240​ if self.pageIndex >= len(self.setupDictionary['pages']) - 1:
241​ self.pageIndex = 0
242​ else:
243​ if self.spread:
244​ self.pageIndex += 2
245​ else:
246​ self.pageIndex += 1
247​ self.flip_page()
248​
249​ def last_page(self):
250​ self.pageIndex = len(self.setupDictionary['pages']) - 1
251​ self.flip_page()
252​
253​ def flip_page(self):
254​ if self.spread:
255​ if self.pageIndex % 2 == 0: # Even/Left
256​ left_page_number = self.pageIndex - 1
257​ right_page_number = self.pageIndex
258​ else: # Odd/Right
259​ left_page_number = self.pageIndex
260​ right_page_number = self.pageIndex + 1
261​ self.update_left_page(left_page_number)
262​ self.update_right_page(right_page_number)
263​ if left_page_number < 0:
264​ page_numbers = str(right_page_number + 1)
265​ elif right_page_number >= len(self.setupDictionary['pages']):
266​ page_numbers = str(left_page_number + 1)
267​ else:
268​ page_numbers = '{left} / {right}'.format(
269​ left = str(left_page_number + 1),
270​ right = str(right_page_number + 1))
271​ self.setWindowTitle('{name} - {page_numbers}'.format(
272​ name = self.setupDictionary['projectName'],
273​ page_numbers = page_numbers))
274​ else:
275​ self.update_single_page(self.pageIndex)
276​ self.setWindowTitle('{name} - {page_numbers}'.format(
277​ name = self.setupDictionary['projectName'],
278​ page_numbers = str(self.pageIndex + 1)))
279​
280​
281​ def get_mergedimage(self, index):
282​ if index < len(self.setupDictionary['pages']) and index > -1:
283​ image_url = os.path.join(self.projecturl, self.setupDictionary['pages'][index])
284​ if os.path.exists(image_url):
285​ page = zipfile.ZipFile(image_url, "r")
286​ image = QImage.fromData(page.read("mergedimage.png"))
287​ page.close()
288​ return image
289​ image = QImage(QSize(10, 10), QImage.Format_ARGB32)
290​ image.fill(Qt.GlobalColor(19))
291​ return image
292​
293​if __name__ == '__main__':
294​ ''' Run the page viewer outside Krita '''
295​ app = QApplication(sys.argv)
296​ if sys.argv:
297​ print(sys.argv)
298​ path_to_config = sys.argv[1]
299​ if os.path.exists(path_to_config):
300​ print('Run comics viewer')
301​ page_viewer_dialog = comics_project_page_viewer()
302​ page_viewer_dialog.load_comic(path_to_config)
303​ page_viewer_dialog.go_to_page_index(0)
304​ page_viewer_dialog.show()
305​ else:
306​ print('No comic found in {comic}'.format(comic=comic))
307​ else:
308​ print('Pass the path to a Krita comicConfig.json file to this script, to view the comic.')
309​
310​ sys.exit(app.exec_())
311​
312​
66 313​
67 def update_image(self, image=QImage()): ​
68 self.viewer.set_image(image)​
Context not available.