You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
370 lines
16 KiB
370 lines
16 KiB
#!/usr/bin/env python3 |
|
|
|
# MIT License |
|
# |
|
# Copyright (c) 2023 Eugenio Parodi <ceccopierangiolieugenio AT googlemail DOT com> |
|
# |
|
# Permission is hereby granted, free of charge, to any person obtaining a copy |
|
# of this software and associated documentation files (the "Software"), to deal |
|
# in the Software without restriction, including without limitation the rights |
|
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell |
|
# copies of the Software, and to permit persons to whom the Software is |
|
# furnished to do so, subject to the following conditions: |
|
# |
|
# The above copyright notice and this permission notice shall be included in all |
|
# copies or substantial portions of the Software. |
|
# |
|
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR |
|
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, |
|
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE |
|
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER |
|
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, |
|
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE |
|
# SOFTWARE. |
|
|
|
import os |
|
from PIL import Image |
|
|
|
import sys, os |
|
|
|
sys.path.append(os.path.join(sys.path[0],'..')) |
|
import TermTk as ttk |
|
|
|
ttk.TTkTheme.loadTheme(ttk.TTkTheme.NERD) |
|
|
|
class Ansieditor(ttk.TTkGridLayout): |
|
def __init__(self, *args, **kwargs): |
|
super().__init__(*args, **kwargs) |
|
|
|
self._te = ttk.TTkTextEdit(lineNumber=True, readOnly=False) |
|
self._te.setText(ttk.TTkString("Ansi editor\n",ttk.TTkColor.UNDERLINE+ttk.TTkColor.BOLD+ttk.TTkColor.ITALIC)) |
|
|
|
self.addItem(wrapLayout := ttk.TTkGridLayout(), 0,0) |
|
self.addItem(fontLayout := ttk.TTkGridLayout(columnMinWidth=1), 1,0) |
|
self.addWidget(self._te,2,0,1,2) |
|
|
|
wrapLayout.addWidget(ttk.TTkLabel(text="Wrap: ", maxWidth=6),0,0) |
|
wrapLayout.addWidget(lineWrap := ttk.TTkComboBox(list=['NoWrap','WidgetWidth','FixedWidth'], maxWidth=20),0,1) |
|
wrapLayout.addWidget(ttk.TTkLabel(text=" Type: ",maxWidth=7),0,2) |
|
wrapLayout.addWidget(wordWrap := ttk.TTkComboBox(list=['WordWrap','WrapAnywhere'], maxWidth=20, enabled=False),0,3) |
|
wrapLayout.addWidget(ttk.TTkLabel(text=" FixW: ",maxWidth=7),0,4) |
|
wrapLayout.addWidget(fixWidth := ttk.TTkSpinBox(value=self._te.wrapWidth(), maxWidth=5, maximum=500, minimum=10, enabled=False),0,5) |
|
wrapLayout.addWidget(ttk.TTkSpacer(),0,10) |
|
|
|
# Empty columns/cells are 1 char wide due to "columnMinWidth=1" parameter in the GridLayout |
|
# 1 3 8 11 |
|
# 0 2 4 5 6 7 9 10 12 |
|
# 0 [ ] FG [ ] BG [ ] LineNumber |
|
# 1 ┌─────┐ ┌─────┐ ╒═══╕╒═══╕╒═══╕╒═══╕ ┌──────┐┌──────┐ |
|
# 2 │ │ │ │ │ a ││ a ││ a ││ a │ │ UNDO ││ REDO │ |
|
# 3 └─────┘ └─────┘ └───┘└───┘└───┘└───┘ ╘══════╛└──────┘ ┕━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┙ |
|
|
|
# Char Fg/Bg buttons |
|
fontLayout.addWidget(cb_fg := ttk.TTkCheckbox(text=" FG"),0,0) |
|
fontLayout.addWidget(btn_fgColor := ttk.TTkColorButtonPicker(border=True, enabled=False, maxSize=(7,3)),1,0) |
|
|
|
fontLayout.addWidget(cb_bg := ttk.TTkCheckbox(text=" BG"),0,2) |
|
fontLayout.addWidget(btn_bgColor := ttk.TTkColorButtonPicker(border=True, enabled=False, maxSize=(7 ,3)),1,2) |
|
|
|
fontLayout.addWidget(cb_linenumber := ttk.TTkCheckbox(text=" LineNumber", checked=True),0,4,1,3) |
|
|
|
# Char style buttons |
|
fontLayout.addWidget(btn_bold := ttk.TTkButton(border=True, maxSize=(5,3), checkable=True, text=ttk.TTkString( 'a' , ttk.TTkColor.BOLD) ),1,4) |
|
fontLayout.addWidget(btn_italic := ttk.TTkButton(border=True, maxSize=(5,3), checkable=True, text=ttk.TTkString( 'a' , ttk.TTkColor.ITALIC) ),1,5) |
|
fontLayout.addWidget(btn_underline := ttk.TTkButton(border=True, maxSize=(5,3), checkable=True, text=ttk.TTkString(' a ', ttk.TTkColor.UNDERLINE) ),1,6) |
|
fontLayout.addWidget(btn_strikethrough := ttk.TTkButton(border=True, maxSize=(5,3), checkable=True, text=ttk.TTkString(' a ', ttk.TTkColor.STRIKETROUGH)),1,7) |
|
|
|
# Undo/Redo buttons |
|
fontLayout.addWidget(btn_undo := ttk.TTkButton(border=True, maxSize=(8,3), enabled=self._te.isUndoAvailable(), text=' UNDO '),1,9 ) |
|
fontLayout.addWidget(btn_redo := ttk.TTkButton(border=True, maxSize=(8,3), enabled=self._te.isRedoAvailable(), text=' REDO '),1,10) |
|
# Undo/Redo events |
|
self._te.undoAvailable.connect(btn_undo.setEnabled) |
|
self._te.redoAvailable.connect(btn_redo.setEnabled) |
|
btn_undo.clicked.connect(self._te.undo) |
|
btn_redo.clicked.connect(self._te.redo) |
|
|
|
@ttk.pyTTkSlot(ttk.TTkColor) |
|
def _currentColorChangedCB(format:ttk.TTkColor): |
|
if format.hasForeground(): |
|
cb_fg.setCheckState(ttk.TTkK.Checked) |
|
btn_fgColor.setEnabled() |
|
btn_fgColor.setColor(format.foreground()) |
|
else: |
|
cb_fg.setCheckState(ttk.TTkK.Unchecked) |
|
btn_fgColor.setDisabled() |
|
|
|
if format.hasBackground(): |
|
cb_bg.setCheckState(ttk.TTkK.Checked) |
|
btn_bgColor.setEnabled() |
|
btn_bgColor.setColor(format.background()) |
|
else: |
|
cb_bg.setCheckState(ttk.TTkK.Unchecked) |
|
btn_bgColor.setDisabled() |
|
|
|
btn_bold.setChecked(format.bold()) |
|
btn_italic.setChecked(format.italic()) |
|
btn_underline.setChecked(format.underline()) |
|
btn_strikethrough.setChecked(format.strikethrough()) |
|
# ttk.TTkLog.debug(f"{fg=} {bg=} {bold=} {italic=} {underline=} {strikethrough= }") |
|
|
|
self._te.currentColorChanged.connect(_currentColorChangedCB) |
|
|
|
def _setStyle(): |
|
color = ttk.TTkColor() |
|
if cb_fg.checkState() == ttk.TTkK.Checked: |
|
color += btn_fgColor.color().invertFgBg() |
|
if cb_bg.checkState() == ttk.TTkK.Checked: |
|
color += btn_bgColor.color() |
|
if btn_bold.isChecked(): |
|
color += ttk.TTkColor.BOLD |
|
if btn_italic.isChecked(): |
|
color += ttk.TTkColor.ITALIC |
|
if btn_underline.isChecked(): |
|
color += ttk.TTkColor.UNDERLINE |
|
if btn_strikethrough.isChecked(): |
|
color += ttk.TTkColor.STRIKETROUGH |
|
cursor = self._te.textCursor() |
|
cursor.applyColor(color) |
|
cursor.setColor(color) |
|
self._te.setFocus() |
|
|
|
cb_fg.stateChanged.connect(lambda x: btn_fgColor.setEnabled(x==ttk.TTkK.Checked)) |
|
cb_bg.stateChanged.connect(lambda x: btn_bgColor.setEnabled(x==ttk.TTkK.Checked)) |
|
cb_fg.clicked.connect(lambda _: _setStyle()) |
|
cb_bg.clicked.connect(lambda _: _setStyle()) |
|
|
|
cb_linenumber.stateChanged.connect(lambda x: self._te.setLineNumber(x==ttk.TTkK.Checked)) |
|
|
|
btn_fgColor.colorSelected.connect(lambda _: _setStyle()) |
|
btn_bgColor.colorSelected.connect(lambda _: _setStyle()) |
|
|
|
btn_bold.clicked.connect(_setStyle) |
|
btn_italic.clicked.connect(_setStyle) |
|
btn_underline.clicked.connect(_setStyle) |
|
btn_strikethrough.clicked.connect(_setStyle) |
|
|
|
lineWrap.setCurrentIndex(0) |
|
wordWrap.setCurrentIndex(1) |
|
|
|
fixWidth.valueChanged.connect(self._te.setWrapWidth) |
|
|
|
@ttk.pyTTkSlot(int) |
|
def _lineWrapCallback(index): |
|
if index == 0: |
|
self._te.setLineWrapMode(ttk.TTkK.NoWrap) |
|
wordWrap.setDisabled() |
|
fixWidth.setDisabled() |
|
elif index == 1: |
|
self._te.setLineWrapMode(ttk.TTkK.WidgetWidth) |
|
wordWrap.setEnabled() |
|
fixWidth.setDisabled() |
|
else: |
|
self._te.setLineWrapMode(ttk.TTkK.FixedWidth) |
|
self._te.setWrapWidth(fixWidth.value()) |
|
wordWrap.setEnabled() |
|
fixWidth.setEnabled() |
|
|
|
lineWrap.currentIndexChanged.connect(_lineWrapCallback) |
|
|
|
@ttk.pyTTkSlot(int) |
|
def _wordWrapCallback(index): |
|
if index == 0: |
|
self._te.setWordWrapMode(ttk.TTkK.WordWrap) |
|
else: |
|
self._te.setWordWrapMode(ttk.TTkK.WrapAnywhere) |
|
|
|
wordWrap.currentIndexChanged.connect(_wordWrapCallback) |
|
|
|
def te(self): |
|
return self._te |
|
|
|
class SigmaskTool(ttk.TTkGridLayout): |
|
def __init__(self, *args, **kwargs): |
|
super().__init__(*args, **kwargs) |
|
|
|
self.addWidget(cb_c := ttk.TTkCheckbox(text="CTRL-C (VINTR) ", checked=ttk.TTkK.Checked),1,0) |
|
self.addWidget(cb_s := ttk.TTkCheckbox(text="CTRL-S (VSTOP) ", checked=ttk.TTkK.Checked),2,0) |
|
self.addWidget(cb_z := ttk.TTkCheckbox(text="CTRL-Z (VSUSP) ", checked=ttk.TTkK.Checked),3,0) |
|
self.addWidget(cb_q := ttk.TTkCheckbox(text="CTRL-Q (VSTART)", checked=ttk.TTkK.Checked),4,0) |
|
|
|
cb_c.stateChanged.connect(lambda x: ttk.TTkTerm.setSigmask(ttk.TTkTerm.Sigmask.CTRL_C,x==ttk.TTkK.Checked)) |
|
cb_s.stateChanged.connect(lambda x: ttk.TTkTerm.setSigmask(ttk.TTkTerm.Sigmask.CTRL_S,x==ttk.TTkK.Checked)) |
|
cb_z.stateChanged.connect(lambda x: ttk.TTkTerm.setSigmask(ttk.TTkTerm.Sigmask.CTRL_Z,x==ttk.TTkK.Checked)) |
|
cb_q.stateChanged.connect(lambda x: ttk.TTkTerm.setSigmask(ttk.TTkTerm.Sigmask.CTRL_Q,x==ttk.TTkK.Checked)) |
|
|
|
self.addWidget(quitBtn := ttk.TTkButton(text="QUIT", border=True, maxHeight=3),5,0) |
|
quitBtn.clicked.connect(ttk.TTkHelper.quit) |
|
|
|
|
|
|
|
root = ttk.TTk(title="Image Tool", layout=ttk.TTkGridLayout()) |
|
|
|
splitter = ttk.TTkSplitter(parent=root) |
|
# splitter.addWidget(fileTree:=ttk.TTkFileTree(path='tmp'), 15) |
|
splitter.addWidget(smt := SigmaskTool(), 25) |
|
splitter.addWidget(mainSplitter := ttk.TTkSplitter(orientation=ttk.TTkK.VERTICAL)) |
|
mainSplitter.addWidget(imageSplitter := ttk.TTkSplitter(orientation=ttk.TTkK.HORIZONTAL)) |
|
mainSplitter.addWidget(controlsWidget := ttk.TTkContainer(layout=ttk.TTkGridLayout()),6) |
|
mainSplitter.addWidget(te := ttk.TTkTextEdit(lineNumber=True, readOnly=False)) |
|
mainSplitter.addWidget(ttk.TTkLogViewer(),6) |
|
|
|
smt.addWidget(fileTree:=ttk.TTkFileTree(path='tmp'),0,0) |
|
|
|
|
|
|
|
|
|
imageSplitter.addWidget(sa := ttk.TTkScrollArea()) |
|
imageSplitter.addWidget(ansiEdit := Ansieditor()) |
|
|
|
controlsWidget.layout().addWidget(resizeFrame := ttk.TTkFrame(title='Resize',layout=ttk.TTkGridLayout())) |
|
controlsWidget.layout().addWidget(propertiesFrame := ttk.TTkFrame(title='Image Properties',layout=ttk.TTkGridLayout())) |
|
|
|
resizeFrame.layout().addWidget(ttk.TTkLabel(text='Width:' ),0,0) |
|
resizeFrame.layout().addWidget(b_width := ttk.TTkSpinBox(maximum=0x1000),0,1) |
|
resizeFrame.layout().addWidget(ttk.TTkLabel(text='Height:' ),1,0) |
|
resizeFrame.layout().addWidget(b_height := ttk.TTkSpinBox(maximum=0x1000),1,1) |
|
resizeFrame.layout().addWidget(ttk.TTkLabel(text='Resample:'),2,0) |
|
resizeFrame.layout().addWidget(cb_resample := ttk.TTkComboBox(),2,1) |
|
|
|
resizeFrame.layout().addWidget(b_quadAR := ttk.TTkButton(text='Quad A/R', border=True),0,2,3,1) |
|
resizeFrame.layout().addWidget(b_sexAR := ttk.TTkButton(text='Sex A/R', border=True), 0,3,3,1) |
|
resizeFrame.layout().addWidget(b_resize := ttk.TTkButton(text='Resize', border=True), 0,4,3,1) |
|
|
|
cb_resample.addItems(['NEAREST','BOX','BILINEAR','HAMMING','BICUBIC','LANCZOS']) |
|
cb_resample.setCurrentIndex(0) |
|
|
|
propertiesFrame.layout().addWidget(ttk.TTkLabel(text='Resolution:'), 0,0) |
|
propertiesFrame.layout().addWidget(cb_resolution := ttk.TTkComboBox(), 0,1) |
|
propertiesFrame.layout().addWidget(ttk.TTkLabel(text='BgColor:'), 1,0) |
|
propertiesFrame.layout().addWidget(b_color := ttk.TTkColorButtonPicker(color=ttk.TTkColor.fg("#000000")), 1,1) |
|
propertiesFrame.layout().addWidget(b_export := ttk.TTkButton(text="Export"),2,0,1,2) |
|
# propertiesFrame.layout().addItem(ttk.TTkLayout(),3,0,1,2) |
|
|
|
cb_resolution.addItems(['FULLBLOCK','HALFBLOCK','QUADBLOCK','SEXBLOCK']) |
|
cb_resolution.setCurrentIndex(2) |
|
|
|
# te.setLineWrapMode(ttk.TTkK.WidgetWidth) |
|
# te.setWordWrapMode(ttk.TTkK.WordWrap) |
|
|
|
te.setText("Select an Image") |
|
|
|
ttkImage = ttk.TTkImage(parent=sa.viewport(), pos=(0,0)) |
|
pilImage = None |
|
|
|
def _export(): |
|
if not ttkImage: |
|
return |
|
data = ttkImage._data |
|
te.setText("Image:") |
|
te.append("=============[Raw Data START]============") |
|
te.append(str(ttkImage._data).replace('],','],\n')) |
|
te.append("=============[Raw Data STOP]=============") |
|
# b64str = base64.b64encode(zlib.compress(bytearray(pickle.dumps(data)))).decode("ascii") |
|
b64str = ttk.TTkUtil.obj_inflate_2_base64(data) |
|
te.append('from TermTk import TTkUtil') |
|
te.append(f'# Data generated using {os.path.basename(__file__)}') |
|
te.append('data = TTkUtil.base64_deflate_2_obj(') |
|
b64list = ' "' + '" +\n "'.join([b64str[i:i+128] for i in range(0,len(b64str),128)]) + '")' |
|
te.append(b64list) |
|
te.append(f'# ANSII {os.path.basename(__file__)}') |
|
# ansi = ttkImage.getCanvas().toAnsi() |
|
# te.append(ansi.replace('\033','<ESC>')) |
|
b64str = ttk.TTkUtil.obj_inflate_2_base64(ansiEdit.te().toAnsi()) |
|
te.append('data = TTkUtil.base64_deflate_2_obj(') |
|
b64list = ' "' + '" +\n "'.join([b64str[i:i+128] for i in range(0,len(b64str),128)]) + '")' |
|
te.append(b64list) |
|
|
|
b_export.clicked.connect(_export) |
|
|
|
def _resolutionChanged(res): |
|
newRes = { |
|
'FULLBLOCK' : ttk.TTkImage.FULLBLOCK, |
|
'HALFBLOCK' : ttk.TTkImage.HALFBLOCK, |
|
'QUADBLOCK' : ttk.TTkImage.QUADBLOCK, |
|
'SEXBLOCK' : ttk.TTkImage.SEXBLOCK |
|
}.get(res, ttk.TTkImage.QUADBLOCK) |
|
ttkImage.setRasteriser(newRes) |
|
|
|
cb_resolution.currentTextChanged.connect(_resolutionChanged) |
|
|
|
|
|
def _quadAR(): |
|
if not pilImage: return |
|
width, height = pilImage.size |
|
b_width.setValue(width) |
|
b_height.setValue(height//2) |
|
|
|
b_quadAR.clicked.connect(_quadAR) |
|
|
|
def _sexAR(): |
|
if not pilImage: return |
|
width, height = pilImage.size |
|
# w/h = 4/3 |
|
b_width.setValue(width) |
|
b_height.setValue(height*3//4) |
|
|
|
b_sexAR.clicked.connect(_sexAR) |
|
|
|
oldImages = [] |
|
def _resize(): |
|
global pilImage |
|
if not pilImage: return |
|
width = b_width.value() |
|
height = b_height.value() |
|
resample = {'NEAREST' : Image.NEAREST, |
|
'BOX' : Image.BOX, |
|
'BILINEAR': Image.BILINEAR, |
|
'HAMMING' : Image.HAMMING, |
|
'BICUBIC' : Image.BICUBIC, |
|
'LANCZOS' : Image.LANCZOS}.get( |
|
cb_resample.currentText(),Image.NEAREST) |
|
pilImage = pilImage.resize((width,height),resample) |
|
data = list(pilImage.getdata()) |
|
# rgbList = [(r,g,b) for r,g,b,a in data] |
|
br,bg,bb = b_color.color().bgToRGB() |
|
rgbList = [ |
|
((r*a+(255-a)*br)//255, |
|
(g*a+(255-a)*bg)//255, |
|
(b*a+(255-a)*bb)//255) |
|
for r,g,b,a in data] |
|
|
|
imageList = [rgbList[i:i+width] for i in range(0, len(rgbList), width)] |
|
ttkImage.setData(imageList) |
|
ansiEdit.te().setText(ttkImage.getCanvas().toAnsi()) |
|
|
|
b_resize.clicked.connect(_resize) |
|
|
|
def _openFile(file): |
|
global pilImage |
|
pilImage = Image.open(file) |
|
pilImage = pilImage.convert('RGBA') |
|
te.setText(str(pilImage.size)) |
|
te.append("data") |
|
#te.append(str(list(pilImage.getdata()))) |
|
data = list(pilImage.getdata()) |
|
# rgbList = list(zip(data[::3],data[1::3],data[2::3])) |
|
br,bg,bb = b_color.color().bgToRGB() |
|
rgbList = [ |
|
((r*a+(255-a)*br)//255, |
|
(g*a+(255-a)*bg)//255, |
|
(b*a+(255-a)*bb)//255) |
|
for r,g,b,a in data] |
|
te.append("rgbList") |
|
#te.append(str(rgbList)) |
|
width, height = pilImage.size |
|
b_width.setValue(width) |
|
b_height.setValue(height) |
|
# imageList = [rgbList[i:i+width] for i in range(0, len(rgbList), width)] |
|
imageList = [rgbList[i:i+width] for i in range(0, len(rgbList), width)] |
|
te.append("imageList") |
|
#te.append(str(imageList)) |
|
|
|
ttkImage.setData(imageList) |
|
ansiEdit.te().setText(ttkImage.getCanvas().toAnsi()) |
|
|
|
#pilImage.size |
|
#pilImage.size |
|
|
|
fileTree.fileActivated.connect(lambda x: _openFile(x.path())) |
|
|
|
root.mainloop() |