Source code for TermTk.TTkWidgets.image

#!/usr/bin/env python3

# 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.

from TermTk.TTkCore.log import TTkLog
from TermTk.TTkCore.color import TTkColor
from TermTk.TTkCore.string import TTkString
from TermTk.TTkWidgets.widget import TTkWidget

[docs]class TTkImage(TTkWidget): __slots__ = ('_data') def __init__(self, *args, **kwargs): TTkWidget.__init__(self, *args, **kwargs) self._name = kwargs.get('name' , 'TTkImage' ) self._data = kwargs.get('data' , [] ) if self._data: w = min([len(i) for i in self._data]) h = len(self._data) self.resize(w//2,h//2) def _reduce(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 # 0x00 0x01 0x02 0x03 quad = [ ' ', '▘', '▝', '▀', # 0x04 0x05 0x06 0x07 '▖', '▌', '▞', '▛', # 0x08 0x09 0x0A 0x0B '▗', '▚', '▐', '▜', # 0x0C 0x0D 0x0E 0x0F '▄', '▙', '▟', '█'] 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}')) + \ quad[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 paintEvent(self): img = self._data 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): self._canvas.drawText( \ pos=(x//2,y//2), \ text=self._reduce( img[y][x] , img[y][x+1] , img[y+1][x] , img[y+1][x+1] ))