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.
260 lines
9.9 KiB
260 lines
9.9 KiB
# MIT License |
|
# |
|
# Copyright (c) 2021 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. |
|
|
|
__all__ = ['TTkImage'] |
|
|
|
from TermTk.TTkCore.color import TTkColor |
|
from TermTk.TTkCore.canvas import TTkCanvas |
|
from TermTk.TTkCore.string import TTkString |
|
from TermTk.TTkWidgets.widget import TTkWidget |
|
|
|
class TTkImage(TTkWidget): |
|
FULLBLOCK = 0x00 |
|
HALFBLOCK = 0x01 |
|
QUADBLOCK = 0x02 |
|
SEXBLOCK = 0x03 |
|
|
|
_quadMap =[ |
|
# 0x00 0x01 0x02 0x03 |
|
' ', '▘', '▝', '▀', |
|
# 0x04 0x05 0x06 0x07 |
|
'▖', '▌', '▞', '▛', |
|
# 0x08 0x09 0x0A 0x0B |
|
'▗', '▚', '▐', '▜', |
|
# 0x0C 0x0D 0x0E 0x0F |
|
'▄', '▙', '▟', '█'] |
|
|
|
_sexMap = [ |
|
# 0x00 0x01 0x02 0x03 0x04 0x05 0x06 0x07 |
|
' ', '🬀', '🬁', '🬂', '🬃', '🬄', '🬅', '🬆', |
|
# 0x08 0x09 0x0A 0x0B 0x0C 0x0D 0x0E 0x0F |
|
'🬇', '🬈', '🬉', '🬊', '🬋', '🬌', '🬍', '🬎', |
|
|
|
# 0x10 0x11 0x12 0x13 0x14 0x15 0x16 0x17 |
|
'🬏', '🬐', '🬑', '🬒', '🬓', '▌', '🬔', '🬕', |
|
# 0x18 0x19 0x1A 0x1B 0x1C 0x1D 0x1E 0x1F |
|
'🬖', '🬗', '🬘', '🬙', '🬚', '🬛', '🬜', '🬝', |
|
|
|
# 0x20 0x21 0x22 0x23 0x24 0x25 0x26 0x27 |
|
'🬞', '🬟', '🬠', '🬡', '🬢', '🬣', '🬤', '🬥', |
|
# 0x28 0x29 0x2A 0x2B 0x2C 0x2D 0x2E 0x2F |
|
'🬦', '🬧', '▐', '🬨', '🬩', '🬪', '🬫', '🬬', |
|
|
|
# 0x30 0x31 0x32 0x33 0x34 0x35 0x36 0x37 |
|
'🬭', '🬮', '🬯', '🬰', '🬱', '🬲', '🬳', '🬴', |
|
# 0x38 0x39 0x3A 0x3B 0x3C 0x3D 0x3E 0x3F |
|
'🬵', '🬶', '🬷', '🬸', '🬹', '🬺', '🬻', '█'] |
|
|
|
__slots__ = ('_data', '_rasterType', '_canvasImage') |
|
def __init__(self, *, |
|
data=None, |
|
rasteriser=HALFBLOCK, |
|
**kwargs) -> None: |
|
TTkWidget.__init__(self, **kwargs) |
|
self._data = data if data else [] |
|
self._rasterType = rasteriser |
|
self._canvasImage = TTkCanvas() |
|
if self._data: |
|
self.setData(self._data) |
|
|
|
def setData(self, data): |
|
self._data = data |
|
w = min(len(i) for i in self._data) |
|
h = len(self._data) |
|
if self._rasterType == TTkImage.FULLBLOCK: |
|
w,h = w,h |
|
elif self._rasterType == TTkImage.HALFBLOCK: |
|
w,h = w,h//2 |
|
elif self._rasterType == TTkImage.QUADBLOCK: |
|
w,h = w//2,h//2 |
|
elif self._rasterType == TTkImage.SEXBLOCK: |
|
w,h = w//2,h//3 |
|
self._canvasImage.resize(*(self.size())) |
|
self.resize(w,h) |
|
self._canvasImage.resize(w,h) |
|
self._canvasImage.updateSize() |
|
self._drawImage(self._canvasImage) |
|
self.update() |
|
|
|
def setRasteriser(self, rasteriser): |
|
if self._rasterType == rasteriser: return |
|
self._rasterType = rasteriser |
|
if self._data: |
|
self.setData(self._data) |
|
|
|
def _reduceQuad(self, a,b,c,d): |
|
# quadblitter notcurses like |
|
l = (a,b,c,d) |
|
def delta(i): |
|
return max(v[i] for v in l) - min(v[i] for v in l) |
|
deltaR = delta(0) |
|
deltaG = delta(1) |
|
deltaB = delta(2) |
|
|
|
def midColor(c1,c2): |
|
return ((c1[0]+c2[0])//2,(c1[1]+c2[1])//2,(c1[2]+c2[2])//2) |
|
|
|
def closer(a,b,c): |
|
return \ |
|
( (a[0]-c[0])**2 + (a[1]-c[1])**2 + (a[2]-c[2])**2 ) > \ |
|
( (b[0]-c[0])**2 + (b[1]-c[1])**2 + (b[2]-c[2])**2 ) |
|
|
|
def splitReduce(i): |
|
s = sorted(l,key=lambda x:x[i]) |
|
mid = (s[3][i]+s[0][i])//2 |
|
if s[1][i] < mid: |
|
if s[2][i] > mid: |
|
c1 = midColor(s[0],s[1]) |
|
c2 = midColor(s[2],s[3]) |
|
else: |
|
c1 = midColor(s[0],s[1]) |
|
c1 = midColor(c1,s[2]) |
|
c2 = s[3] |
|
else: |
|
c1 = s[0] |
|
c2 = midColor(s[1],s[2]) |
|
c2 = midColor(c1,s[3]) |
|
|
|
|
|
ch = 0x01 if closer(c1,c2,l[0]) else 0 |
|
ch |= 0x02 if closer(c1,c2,l[1]) else 0 |
|
ch |= 0x04 if closer(c1,c2,l[2]) else 0 |
|
ch |= 0x08 if closer(c1,c2,l[3]) else 0 |
|
|
|
return TTkString() + \ |
|
(TTkColor.bg(f'#{c1[0]:02X}{c1[1]:02X}{c1[2]:02X}') + |
|
TTkColor.fg(f'#{c2[0]:02X}{c2[1]:02X}{c2[2]:02X}')) + \ |
|
TTkImage._quadMap[ch] |
|
|
|
if deltaR >= deltaG and deltaR >= deltaB: |
|
# Use Red as splitter |
|
return splitReduce(0) |
|
elif deltaG >= deltaB and deltaG >= deltaR: |
|
# Use Green as splitter |
|
return splitReduce(1) |
|
else: |
|
# Use Blue as splitter |
|
return splitReduce(2) |
|
|
|
def _reduceSex(self, a,b,c,d,e,f): |
|
# sexblitter notcurses like |
|
l = (a,b,c,d,e,f) |
|
def delta(i): |
|
return max(v[i] for v in l) - min(v[i] for v in l) |
|
deltaR = delta(0) |
|
deltaG = delta(1) |
|
deltaB = delta(2) |
|
|
|
def midColor(c1,c2): |
|
return ((c1[0]+c2[0])//2,(c1[1]+c2[1])//2,(c1[2]+c2[2])//2) |
|
|
|
def closer(a,b,c): |
|
return \ |
|
( (a[0]-c[0])**2 + (a[1]-c[1])**2 + (a[2]-c[2])**2 ) > \ |
|
( (b[0]-c[0])**2 + (b[1]-c[1])**2 + (b[2]-c[2])**2 ) |
|
|
|
def splitReduce(i): |
|
s = sorted(l,key=lambda x:x[i]) |
|
mid = (s[5][i]+s[0][i])//2 |
|
if s[1][i] < mid: |
|
if s[2][i] > mid: |
|
c1 = midColor(s[0],s[1]) |
|
c2 = midColor(s[2],s[3]) |
|
else: |
|
c1 = midColor(s[0],s[1]) |
|
c1 = midColor(c1,s[2]) |
|
c2 = s[3] |
|
else: |
|
c1 = s[0] |
|
c2 = midColor(s[1],s[2]) |
|
c2 = midColor(c1,s[3]) |
|
|
|
|
|
ch = 0x01 if closer(c1,c2,l[0]) else 0 |
|
ch |= 0x02 if closer(c1,c2,l[1]) else 0 |
|
ch |= 0x04 if closer(c1,c2,l[2]) else 0 |
|
ch |= 0x08 if closer(c1,c2,l[3]) else 0 |
|
ch |= 0x10 if closer(c1,c2,l[4]) else 0 |
|
ch |= 0x20 if closer(c1,c2,l[5]) else 0 |
|
|
|
return TTkString() + \ |
|
(TTkColor.bg(f'#{c1[0]:02X}{c1[1]:02X}{c1[2]:02X}') + |
|
TTkColor.fg(f'#{c2[0]:02X}{c2[1]:02X}{c2[2]:02X}')) + \ |
|
TTkImage._sexMap[ch] |
|
|
|
if deltaR >= deltaG and deltaR >= deltaB: |
|
# Use Red as splitter |
|
return splitReduce(0) |
|
elif deltaG >= deltaB and deltaG >= deltaR: |
|
# Use Green as splitter |
|
return splitReduce(1) |
|
else: |
|
# Use Blue as splitter |
|
return splitReduce(2) |
|
|
|
def rotHue(self, deg): |
|
old = self._data |
|
self._data = [[p for p in l ] for l in old] |
|
for row in self._data: |
|
for i,pixel in enumerate(row): |
|
h,s,l = TTkColor.rgb2hsl(pixel) |
|
row[i] = TTkColor.hsl2rgb(((h+deg)%360,s,l)) |
|
self.setData(self._data) |
|
|
|
def _drawImage(self, canvas): |
|
img = self._data |
|
if self._rasterType == TTkImage.FULLBLOCK: |
|
for y in range(0, len(img)): |
|
for x in range(0, len(img[y])): |
|
c1 = img[y][x] |
|
color = TTkColor.fg(f'#{c1[0]:02X}{c1[1]:02X}{c1[2]:02X}') |
|
canvas.drawChar(pos=(x,y), char='█', color=color) |
|
elif self._rasterType == TTkImage.HALFBLOCK: |
|
for y in range(0, len(img)&(~1), 2): |
|
for x in range(0, len(img[y])): |
|
c1, c2 = img[y][x] ,img[y+1][x] |
|
color = ( TTkColor.fg(f'#{c1[0]:02X}{c1[1]:02X}{c1[2]:02X}') + |
|
TTkColor.bg(f'#{c2[0]:02X}{c2[1]:02X}{c2[2]:02X}') ) |
|
canvas.drawChar(pos=(x,y//2), char='▀', color=color) |
|
elif self._rasterType == TTkImage.QUADBLOCK: |
|
for y in range(0, len(img)&(~1), 2): |
|
for x in range(0, min(len(img[y])&(~1),len(img[y+1])&(~1)), 2): |
|
canvas.drawText( |
|
pos=(x//2,y//2), |
|
text=self._reduceQuad( |
|
img[y][x] , img[y][x+1] , |
|
img[y+1][x] , img[y+1][x+1] )) |
|
elif self._rasterType == TTkImage.SEXBLOCK: |
|
for y in range(0, len(img)-2, 3): |
|
for x in range(0, min(len(img[y])-1,len(img[y+1])-1,len(img[y+2])-1), 2): |
|
canvas.drawText( |
|
pos=(x//2,y//3), |
|
text=self._reduceSex( |
|
img[y][x] , img[y][x+1] , |
|
img[y+1][x] , img[y+1][x+1] , |
|
img[y+2][x] , img[y+2][x+1] )) |
|
|
|
def paintEvent(self, canvas: TTkCanvas): |
|
w,h=self.size() |
|
s = (0,0,w,h) |
|
canvas.paintCanvas(self._canvasImage,s,s,s) |
|
|
|
|