Initial commit

1 files changed, 168 insertions, 0 deletions

diff --git a/presentty/image.py b/presentty/image.py
new file mode 100644
index 0000000..9aabee8
--- /dev/null
+++ b/presentty/image.py
@@ -0,0 +1,168 @@
+# Copyright (C) 2015 James E. Blair <corvus@gnu.org>
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+import subprocess
+import HTMLParser
+import re
+
+import PIL
+import PIL.ExifTags
+import urwid
+
+import slide
+
+def nearest_color(x):
+    if x < 0x30: return '0'
+    if x < 0x70: return '6'
+    if x < 0x98: return '8'
+    if x < 0xc0: return 'a'
+    if x < 0xe8: return 'd'
+    return 'f'
+
+class ANSIImage(urwid.Widget):
+    def __init__(self, uri, hinter=None):
+        super(ANSIImage, self).__init__()
+        self.uri = uri
+        image = self._loadImage()
+        self.htmlparser = HTMLParser.HTMLParser()
+        self.ratio = float(image.size[0])/float(image.size[1])
+        self.hinter = hinter
+
+    def _loadImage(self):
+        image = PIL.Image.open(self.uri)
+        image.load()
+        exif = image._getexif()
+        if exif:
+            orientation = exif.get(274, 1)
+            if orientation == 1:
+                pass
+            elif orientation == 3:
+                image = image.rotate(180)
+            elif orientation == 6:
+                image = image.rotate(-90)
+            elif orientation == 8:
+                image = image.rotate(90)
+            else:
+                raise Exception("unknown orientation %s" % orientation)
+        return image
+
+    def pack(self, size, focus=False):
+        cols = size[0]
+        if len(size) > 1:
+            rows = size[1]
+        elif self.hinter:
+            rows = self.hinter.getSize()[1]
+        else:
+            rows = None
+        width = cols
+        height = int(cols*(1.0/self.ratio)/2.0)
+        if rows is not None and height > rows:
+            height = rows
+            width = int(rows*self.ratio*2.0)
+        return (width, height)
+
+    def rows(self, size, focus=False):
+        r = self.pack(size)
+        return r[1]
+
+    SPAN_RE = re.compile(r"<span style='color:#(......); background-color:#(......);'>(.*)")
+    def render(self, size, focus=False):
+        spanre = self.SPAN_RE
+        htmlparser = self.htmlparser
+        width, height = self.pack(size, focus)
+        jp2a = subprocess.Popen(['jp2a', '--colors', '--fill',
+                                 '--width=%s' % width,
+                                 '--height=%s' % height,
+                                 '--html-raw', '-'],
+                                stdin=subprocess.PIPE,
+                                stdout=subprocess.PIPE,
+                                stderr=subprocess.PIPE)
+        image = self._loadImage()
+        image.save(jp2a.stdin, 'JPEG')
+        jp2a.stdin.close()
+        data = jp2a.stdout.read()
+        jp2a.stderr.read()
+        jp2a.wait()
+
+        line_list = []
+        attr_list = []
+        line_text = ''
+        line_attrs = []
+        current_attr = [None, 0]
+        current_fg = None
+        current_bg = None
+        current_props = None
+        for line in data.split('<br/>'):
+            if not line:
+                continue
+            for span in line.split('</span>'):
+                if not span:
+                    continue
+                m = spanre.match(span)
+                fg, bg, char = m.groups()
+                if '&' in char:
+                    char = htmlparser.unescape(char)
+                char = char.encode('utf8')
+                line_text += char
+                props = []
+                # TODO: if bold is set, append bold to props
+                fg = ('#'+
+                      nearest_color(int(fg[0:2], 16)) +
+                      nearest_color(int(fg[2:4], 16)) +
+                      nearest_color(int(fg[4:6], 16)))
+                bg = ('#'+
+                      nearest_color(int(bg[0:2], 16)) +
+                      nearest_color(int(bg[2:4], 16)) +
+                      nearest_color(int(bg[4:6], 16)))
+                if current_fg == fg and current_bg == bg and current_props == props:
+                    current_attr[1] += len(char)
+                else:
+                    if current_attr[0]:
+                        line_attrs.append(tuple(current_attr))
+                    fg = ', '.join(props + [fg])
+                    attr = urwid.AttrSpec(fg, bg)
+                    current_attr = [attr, len(char)]
+                    current_fg = fg
+                    current_bg = bg
+                    current_props = props
+            line_attrs.append(tuple(current_attr))
+            current_attr = [None, 0]
+            current_fg = None
+            current_bg = None
+            line_list.append(line_text)
+            line_text = ''
+            attr_list.append(line_attrs)
+            line_attrs = []
+        canvas = urwid.TextCanvas(line_list, attr_list)
+        return canvas
+
+def main():
+    import PIL.Image
+    img = PIL.Image.open('/tmp/p/8.jpg')
+    img.load()
+    hinter = slide.ScreenHinter()
+    hinter.set_cols_rows((80, 25))
+    w = ANSIImage(img, hinter)
+    slpile = slide.SlidePile([])
+    slpile.contents.append((w, slpile.options()))
+    pad = slide.SlidePadding(slpile, align='center', width='pack')
+    fill = slide.SlideFiller(pad)
+    #w.render((80,25))
+    fill.render((80,25))
+    screen = urwid.raw_display.Screen()
+    if True:
+        with screen.start():
+            screen.draw_screen((80,25), fill.render((80,25)))
+            raw_input()