-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathScaleNxGUI.py
410 lines (309 loc) · 14.3 KB
/
ScaleNxGUI.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
#!/usr/bin/env python3
"""
ScaleNx GUI - Common shell for ScaleNx single file and batch PNG, PPM and PGM rescaling.
---
Created by: Ilya Razmanov <ilyarazmanov@gmail.com> aka Ilyich the Toad <amphisoft@gmail.com>
History:
---
25.01.17.00 Initial.
25.01.17.21 Fully operational.
25.03.01.01 PNM batch processing added. GUI simplified to reduce imports.
---
Main site: <https://dnyarri.github.io>
Project at Github: </~https://github.com/Dnyarri/PixelArtScaling>
"""
__author__ = 'Ilya Razmanov'
__copyright__ = '(c) 2025 Ilya Razmanov'
__credits__ = 'Ilya Razmanov'
__license__ = 'unlicense'
__version__ = '25.03.01.01'
__maintainer__ = 'Ilya Razmanov'
__email__ = 'ilyarazmanov@gmail.com'
__status__ = 'Production'
from multiprocessing import Pool, freeze_support
from pathlib import Path
from tkinter import Button, Frame, Label, Tk, filedialog
from pypng import pnglpng
from pypnm import pnmlpnm
from scalenx import scalenx, scalenxsfx
def DisMiss():
"""Kill dialog and continue"""
sortir.destroy()
def UINormal():
"""Normal UI state, buttons enabled"""
for widget in frame_left.winfo_children():
if widget.winfo_class() in ('Label', 'Button'):
widget.config(state='normal')
for widget in frame_right.winfo_children():
if widget.winfo_class() in ('Label', 'Button'):
widget.config(state='normal')
info_string.config(text=info_normal['txt'], foreground=info_normal['fg'], background=info_normal['bg'])
def UIWaiting():
"""Waiting UI state, buttons disabled"""
for widget in frame_left.winfo_children():
if widget.winfo_class() in ('Label', 'Button'):
widget.config(state='disabled')
for widget in frame_right.winfo_children():
if widget.winfo_class() in ('Label', 'Button'):
widget.config(state='disabled')
info_string.config(text=info_waiting['txt'], foreground=info_waiting['fg'], background=info_waiting['bg'])
sortir.update()
def UIBusy():
"""Busy UI state, buttons disabled"""
for widget in frame_left.winfo_children():
if widget.winfo_class() in ('Label', 'Button'):
widget.config(state='disabled')
for widget in frame_right.winfo_children():
if widget.winfo_class() in ('Label', 'Button'):
widget.config(state='disabled')
info_string.config(text=info_busy['txt'], foreground=info_busy['fg'], background=info_busy['bg'])
sortir.update()
def FileNx(size: int, sfx: bool) -> None:
"""Single file ScaleNx with variable N and method.
Arguments:
size: scale size, either 2 or 3;
sfx: use either sfx or classic scaler version.
"""
UIWaiting()
# Open source file
sourcefilename = filedialog.askopenfilename(title='Open image file to rescale', filetypes=[('Supported formats', '.png .ppm .pgm .pbm'), ('PNG', '.png'), ('PNM', '.ppm .pgm .pbm')])
if sourcefilename == '':
UINormal()
return None
UIBusy()
if Path(sourcefilename).suffix == '.png':
# Reading image as list
X, Y, Z, maxcolors, image3d, info = pnglpng.png2list(sourcefilename)
elif Path(sourcefilename).suffix in ('.ppm', '.pgm', '.pbm'):
# Reading image as list
X, Y, Z, maxcolors, image3d = pnmlpnm.pnm2list(sourcefilename)
# Creating dummy info
info = {}
# Fixing color mode. The rest is fixed with pnglpng v. 25.01.07.
if maxcolors > 255:
info['bitdepth'] = 16
else:
info['bitdepth'] = 8
else:
raise ValueError('Extension not recognized')
# Choosing working scaler from the list of imported scalers
if sfx:
if size == 2:
chosen_scaler = scalenxsfx.scale2x
if size == 3:
chosen_scaler = scalenxsfx.scale3x
else:
if size == 2:
chosen_scaler = scalenx.scale2x
if size == 3:
chosen_scaler = scalenx.scale3x
# Scaling using scaler chosen above
scaled_image = chosen_scaler(image3d)
# --------------------------------------------------------------
# Fixing resolution to match original print size.
# If no pHYs found in original, 96 ppi is assumed as original value.
if 'physical' in info:
res = info['physical'] # Reading resolution as tuple
x_pixels_per_unit = res[0]
y_pixels_per_unit = res[1]
unit_is_meter = res[2]
else:
x_pixels_per_unit = y_pixels_per_unit = 3780
# 3780 px/meter = 96 px/inch, 2834 px/meter = 72 px/inch
unit_is_meter = True
x_pixels_per_unit = size * x_pixels_per_unit # Change resolution to keep print size
y_pixels_per_unit = size * y_pixels_per_unit # Change resolution to keep print size
info['physical'] = [x_pixels_per_unit, y_pixels_per_unit, unit_is_meter]
# Resolution changed
# --------------------------------------------------------------
# Explicitly setting compression
info['compression'] = 9
# Adjusting "Save to" formats to be displayed according to bitdepth
if Z < 3:
format = [('PNG', '.png'), ('Portable grey map', '.pgm')]
else:
format = [('PNG', '.png'), ('Portable pixel map', '.ppm')]
UIWaiting()
# Open export file
resultfilename = filedialog.asksaveasfilename(
title='Save image file',
filetypes=format,
defaultextension=('PNG file', '.png'),
)
if resultfilename == '':
UINormal()
return None
UIBusy()
if Path(resultfilename).suffix == '.png':
pnglpng.list2png(resultfilename, scaled_image, info)
elif Path(resultfilename).suffix in ('.ppm', '.pgm'):
pnmlpnm.list2pnm(resultfilename, scaled_image, maxcolors)
UINormal()
return None
def scale_file_png(runningfilename: str, size: int, sfx: bool) -> None:
"""Function upscales one PNG file and keeps quite.
Arguments:
runningfilename: name of file to process;
size: scale size, either 2 or 3;
sfx: use either sfx or classic scaler version.
"""
oldfile = str(runningfilename)
newfile = oldfile # Previous version used backup newfile = oldfile + '.2x.png'
# Reading image as list
X, Y, Z, maxcolors, image3d, info = pnglpng.png2list(oldfile)
# Choosing working scaler from the list of imported scalers
if sfx:
if size == 2:
chosen_scaler = scalenxsfx.scale2x
if size == 3:
chosen_scaler = scalenxsfx.scale3x
else:
if size == 2:
chosen_scaler = scalenx.scale2x
if size == 3:
chosen_scaler = scalenx.scale3x
# Scaling using scaler chosen above
scaled_image = chosen_scaler(image3d)
# --------------------------------------------------------------
# Fixing resolution to match original print size.
# If no pHYs found in original, 96 ppi is assumed as original value.
if 'physical' in info:
res = info['physical'] # Reading resolution as tuple
x_pixels_per_unit = res[0]
y_pixels_per_unit = res[1]
unit_is_meter = res[2]
else:
x_pixels_per_unit = y_pixels_per_unit = 3780
# 3780 px/meter = 96 px/inch, 2834 px/meter = 72 px/inch
unit_is_meter = True
x_pixels_per_unit = size * x_pixels_per_unit # Change resolution to keep print size
y_pixels_per_unit = size * y_pixels_per_unit # Change resolution to keep print size
info['physical'] = [x_pixels_per_unit, y_pixels_per_unit, unit_is_meter]
# Resolution changed
# --------------------------------------------------------------
# Explicitly setting compression
info['compression'] = 9
# Writing PNG file
pnglpng.list2png(newfile, scaled_image, info)
return None
def scale_file_pnm(runningfilename: str, size: int, sfx: bool) -> None:
"""Function upscales one PNM file and keeps quite.
Arguments:
runningfilename: name of file to process;
size: scale size, either 2 or 3;
sfx: use either sfx or classic scaler version.
"""
oldfile = str(runningfilename)
newfile = oldfile # Overwrite!
# Reading image as list
X, Y, Z, maxcolors, image3d = pnmlpnm.pnm2list(oldfile)
# Choosing working scaler from the list of imported scalers
if sfx:
if size == 2:
chosen_scaler = scalenxsfx.scale2x
if size == 3:
chosen_scaler = scalenxsfx.scale3x
else:
if size == 2:
chosen_scaler = scalenx.scale2x
if size == 3:
chosen_scaler = scalenx.scale3x
# Scaling using scaler chosen above
scaled_image = chosen_scaler(image3d)
# Writing PNM file
pnmlpnm.list2pnm(newfile, scaled_image, maxcolors)
return None
def FolderNx(size: int, sfx: bool) -> None:
"""Multiprocessing pool to feed `scale_file_*` processes to.
Arguments:
size: scale size, either 2 or 3;
sfx: use either sfx or classic scaler version.
"""
UIWaiting()
# Open source dir
sourcedir = filedialog.askdirectory(title='Open folder to rescale images')
if sourcedir == '':
UINormal()
return None
path = Path(sourcedir)
UIBusy()
# Creating pool
scalepool = Pool()
# Feeding the pool (no pun!)
for runningfilename in path.rglob('*.*'):
if runningfilename.suffix == '.png':
scalepool.apply_async(
scale_file_png,
args=(
runningfilename,
size,
sfx,
),
)
if runningfilename.suffix in ('.ppm', '.pgm'):
scalepool.apply_async(
scale_file_pnm,
args=(
runningfilename,
size,
sfx,
),
)
# Everything fed into the pool, waiting and closing
scalepool.close()
scalepool.join()
UINormal()
return None
""" ╔═══════════╗
║ Main body ║
╚═══════════╝ """
if __name__ == '__main__':
freeze_support() # Freezing for exe
sortir = Tk()
sortir.title('ScaleNx')
sortir.geometry('+200+100')
sortir.minsize(560, 370)
iconpath = Path(__file__).resolve().parent / '32.ico'
if iconpath.exists():
sortir.iconbitmap(str(iconpath))
# Info statuses dictionaries
info_normal = {'txt': f'ScaleNx ver. {__version__} at your command', 'fg': 'grey', 'bg': 'light grey'}
info_waiting = {'txt': 'Waiting for input', 'fg': 'green', 'bg': 'light grey'}
info_busy = {'txt': 'BUSY, PLEASE WAIT', 'fg': 'red', 'bg': 'yellow'}
butt99 = Button(sortir, text='Exit', font=('helvetica', 14), cursor='hand2', justify='center', state='normal', command=DisMiss)
butt99.pack(side='bottom', padx=4, pady=2, fill='both')
info_string = Label(sortir, text=info_normal['txt'], font=('courier', 10), foreground=info_normal['fg'], background=info_normal['bg'], relief='groove')
info_string.pack(side='bottom', padx=2, pady=(6, 1), fill='both')
frame_left = Frame(sortir, borderwidth=2, relief='groove')
frame_left.pack(side='left', anchor='nw', padx=(2, 6), pady=0)
frame_right = Frame(sortir, borderwidth=2, relief='groove')
frame_right.pack(side='right', anchor='ne', padx=(6, 2), pady=0)
label00 = Label(frame_left, text='ScaleNx', font=('helvetica', 24), justify='center', borderwidth=2, relief='groove', foreground='brown', background='light grey')
label00.pack(side='top', pady=(0, 6), fill='both')
label01 = Label(frame_left, text='Single image rescaling (PNG, PPM, PGM)'.center(42, ' '), font=('helvetica', 10), justify='center', borderwidth=2, relief='flat', foreground='dark blue')
label01.pack(side='top', pady=(12, 0))
butt01 = Button(frame_left, text='Open file ➔ 2x', font=('helvetica', 14), cursor='hand2', justify='center', state='normal', command=lambda: FileNx(2, False))
butt01.pack(side='top', padx=4, pady=2, fill='both')
butt02 = Button(frame_left, text='Open file ➔ 3x', font=('helvetica', 14), cursor='hand2', justify='center', state='normal', command=lambda: FileNx(3, False))
butt02.pack(side='top', padx=4, pady=2, fill='both')
label02 = Label(frame_left, text='Folder batch process (PNG, PPM, PGM)', font=('helvetica', 10), justify='center', borderwidth=2, relief='flat', foreground='dark blue')
label02.pack(side='top', pady=(12, 0))
butt03 = Button(frame_left, text='Select folder ➔ 2x', font=('helvetica', 14), cursor='hand2', justify='center', state='normal', command=lambda: FolderNx(2, False))
butt03.pack(side='top', padx=4, pady=2, fill='both')
butt04 = Button(frame_left, text='Select folder ➔ 3x', font=('helvetica', 14), cursor='hand2', justify='center', state='normal', command=lambda: FolderNx(3, False))
butt04.pack(side='top', padx=4, pady=2, fill='both')
label10 = Label(frame_right, text='ScaleNxSFX', font=('helvetica', 24), justify='center', borderwidth=2, relief='groove', foreground='brown', background='light grey')
label10.pack(side='top', pady=(0, 6), fill='both')
label11 = Label(frame_right, text='Single image rescaling (PNG, PPM, PGM)'.center(42, ' '), font=('helvetica', 10), justify='center', borderwidth=2, relief='flat', foreground='dark blue')
label11.pack(side='top', pady=(12, 0))
butt11 = Button(frame_right, text='Open file ➔ 2xSFX', font=('helvetica', 14), cursor='hand2', justify='center', command=lambda: FileNx(2, True))
butt11.pack(side='top', padx=4, pady=2, fill='both')
butt12 = Button(frame_right, text='Open file ➔ 3xSFX', font=('helvetica', 14), cursor='hand2', justify='center', state='normal', command=lambda: FileNx(3, True))
butt12.pack(side='top', padx=4, pady=2, fill='both')
label12 = Label(frame_right, text='Folder batch process (PNG, PPM, PGM)', font=('helvetica', 10), justify='center', borderwidth=2, relief='flat', foreground='dark blue')
label12.pack(side='top', pady=(12, 0))
butt13 = Button(frame_right, text='Select folder ➔ 2xSFX', font=('helvetica', 14), cursor='hand2', justify='center', state='normal', command=lambda: FolderNx(2, True))
butt13.pack(side='top', padx=4, pady=2, fill='both')
butt14 = Button(frame_right, text='Select folder ➔ 3xSFX', font=('helvetica', 14), cursor='hand2', justify='center', state='normal', command=lambda: FolderNx(3, True))
butt14.pack(side='top', padx=4, pady=2, fill='both')
sortir.mainloop()