# 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
[docs]
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=[],
rasteriser=HALFBLOCK,
**kwargs) -> None:
TTkWidget.__init__(self, **kwargs)
self._data = data
self._rasterType = rasteriser
self._canvasImage = TTkCanvas()
if self._data:
self.setData(self._data)
[docs]
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()
[docs]
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)
[docs]
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)