From 07b858c96b61272d9b5416100c644940ec80c95a Mon Sep 17 00:00:00 2001
From: Helmut Grohne <helmut@subdivi.de>
Date: Sat, 26 Feb 2011 13:35:23 +0100
Subject: initial checkin

---
 tcvt.py | 449 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 449 insertions(+)
 create mode 100755 tcvt.py

diff --git a/tcvt.py b/tcvt.py
new file mode 100755
index 0000000..389ec14
--- /dev/null
+++ b/tcvt.py
@@ -0,0 +1,449 @@
+#!/usr/bin/python
+#
+# Tow Column Virtual Terminal.
+#
+# Copyright 2011 Helmut Grohne <helmut@subdivi.de>. All rights reserved.
+# 
+# Redistribution and use in source and binary forms, with or without
+# modification, are permitted provided that the following conditions are met:
+# 
+#    1. Redistributions of source code must retain the above copyright notice,
+#       this list of conditions and the following disclaimer.
+# 
+#    2. Redistributions in binary form must reproduce the above copyright
+#       notice, this list of conditions and the following disclaimer in the
+#       documentation and/or other materials provided with the distribution.
+# 
+# THIS SOFTWARE IS PROVIDED BY HELMUT GROHNE ``AS IS'' AND ANY EXPRESS OR
+# IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
+# MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO
+# EVENT SHALL HELMUT GROHNE OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
+# INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
+# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA,
+# OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
+# LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
+# NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE,
+# EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+# 
+# The views and conclusions contained in the software and documentation are
+# those of the authors and should not be interpreted as representing official
+# policies, either expressed or implied, of Helmut Grohne.
+
+import pty
+import sys
+import os
+import select
+import fcntl
+import termios
+import struct
+import curses
+import errno
+
+def init_color_pairs():
+    for bi, bc in enumerate((curses.COLOR_BLACK, curses.COLOR_RED,
+                             curses.COLOR_GREEN, curses.COLOR_YELLOW,
+                             curses.COLOR_BLUE, curses.COLOR_MAGENTA,
+                             curses.COLOR_CYAN, curses.COLOR_WHITE)):
+        for fi, fc in enumerate((curses.COLOR_WHITE, curses.COLOR_BLACK,
+                                 curses.COLOR_RED, curses.COLOR_GREEN,
+                                 curses.COLOR_YELLOW, curses.COLOR_BLUE,
+                                 curses.COLOR_MAGENTA, curses.COLOR_CYAN)):
+            if fi != 0 or bi != 0:
+                curses.init_pair(fi*8+bi, fc, bc)
+
+def get_color(fg=1, bg=0):
+    return curses.color_pair(((fg + 1) % 8) * 8 + bg)
+
+class Simple:
+    def __init__(self, curseswindow):
+        self.screen = curseswindow
+        self.screen.scrollok(1)
+
+    def getmaxyx(self):
+        return self.screen.getmaxyx()
+
+    def move(self, ypos, xpos):
+        ym, xm = self.getmaxyx()
+        self.screen.move(max(0, min(ym - 1, ypos)), max(0, min(xm - 1, xpos)))
+
+    def relmove(self, yoff, xoff):
+        y, x = self.getyx()
+        self.move(y + yoff, x + xoff)
+
+    def addch(self, char):
+        self.screen.addch(char)
+
+    def refresh(self):
+        self.screen.refresh()
+
+    def getyx(self):
+        return self.screen.getyx()
+
+    def scroll(self):
+        self.screen.scroll()
+
+    def clrtobot(self):
+        self.screen.clrtobot()
+
+    def attron(self, attr):
+        self.screen.attron(attr)
+
+    def clrtoeol(self):
+        self.screen.clrtoeol()
+
+    def delch(self):
+        self.screen.delch()
+
+    def attrset(self, attr):
+        self.screen.attrset(attr)
+
+    def insertln(self):
+        self.screen.insertln()
+
+    def insch(self, char):
+        self.screen.insch(char)
+
+    def deleteln(self):
+        self.screen.deleteln()
+
+    def inch(self):
+        return self.screen.inch()
+
+class TwoColumn:
+    def __init__(self, curseswindow):
+        self.screen = curseswindow
+        self.height, width = self.screen.getmaxyx()
+        self.halfwidth = (width - 1) // 2
+        self.left = self.screen.derwin(self.height, self.halfwidth, 0, 0)
+        self.left.scrollok(1)
+        self.right = self.screen.derwin(self.height, self.halfwidth, 0,
+                                        self.halfwidth + 1)
+        self.right.scrollok(1)
+        self.ypos, self.xpos = 0, 0
+        self.screen.vline(0, self.halfwidth, curses.ACS_VLINE, self.height)
+        self.attrs = 0
+
+    @property
+    def curwin(self):
+        return self.left if self.ypos < self.height else self.right
+
+    def getmaxyx(self):
+        return (self.height * 2, self.halfwidth)
+
+    def move(self, ypos, xpos):
+        self.ypos = max(0, min(self.height * 2 - 1, ypos))
+        self.xpos = max(0, min(self.halfwidth - 1, xpos))
+        self.do_move()
+
+    def do_move(self):
+        self.curwin.move(self.ypos % self.height, self.xpos)
+
+    def relmove(self, yoff, xoff):
+        self.move(self.ypos + yoff, self.xpos + xoff)
+
+    def addch(self, char):
+        if self.xpos == self.halfwidth - 1:
+            self.curwin.insch(self.ypos % self.height, self.xpos, char,
+                              self.attrs)
+            if self.ypos + 1 == 2 * self.height:
+                self.scroll()
+                self.move(self.ypos, 0)
+            else:
+                self.move(self.ypos + 1, 0)
+        else:
+            self.curwin.addch(self.ypos % self.height, self.xpos, char,
+                              self.attrs)
+            self.xpos += 1
+
+    def refresh(self):
+        self.screen.refresh()
+        if self.ypos < self.height:
+            self.right.refresh()
+            self.left.refresh()
+        else:
+            self.left.refresh()
+            self.right.refresh()
+
+    def getyx(self):
+        return (self.ypos, self.xpos)
+
+    def scroll_up_right(self):
+        """Copy first line of right window to last line of left window and
+        scroll up the right window."""
+        self.left.move(self.height - 1, 0)
+        for x in range(self.halfwidth - 1):
+            self.left.addch(self.right.inch(0, x))
+        self.left.insch(self.right.inch(0, self.halfwidth - 1))
+        self.do_move() # fix cursor
+        self.right.scroll()
+
+    def scroll_down_right(self):
+        """Scroll down the right window and copy the last line of the left
+        window to the first line of the right window."""
+        self.right.scroll(-1)
+        self.right.move(0, 0)
+        for x in range(self.halfwidth - 1):
+            self.right.addch(self.left.inch(self.height - 1, x))
+        self.right.insch(self.left.inch(self.height - 1, self.halfwidth - 1))
+        self.do_move() # fix cursor
+
+    def scroll(self):
+        self.left.scroll()
+        self.scroll_up_right()
+
+    def clrtobot(self):
+        if self.ypos < self.height:
+            self.right.clear()
+            self.left.clrtobot()
+        else:
+            self.right.clrtobot()
+
+    def attron(self, attr):
+        self.attrs |= attr
+
+    def clrtoeol(self):
+        self.curwin.clrtoeol()
+
+    def delch(self):
+        self.curwin.delch(self.ypos % self.height, self.xpos)
+
+    def attrset(self, attr):
+        self.attrs = attr
+
+    def insertln(self):
+        if self.ypos >= self.height:
+            self.right.insertln()
+        else:
+            self.scroll_down_right()
+            self.left.insertln()
+
+    def insch(self, char):
+        self.curwin.insch(self.ypos % self.height, self.xpos, char, self.attrs)
+
+    def deleteln(self):
+        if self.ypos >= self.height:
+            self.right.deleteln()
+        else:
+            self.left.deleteln()
+            self.scroll_up_right()
+
+    def inch(self):
+        return self.curwin.inch(self.ypos % self.height, self.xpos)
+
+class Terminal:
+    def __init__(self):
+        self.mode = (self.feed_simple,)
+        self.realscreen = None
+        self.screen = None
+        self.fg = self.bg = 0
+
+    def switchmode(self):
+        if isinstance(self.screen, TwoColumn):
+            self.screen = Simple(self.realscreen)
+        else:
+            self.screen = TwoColumn(self.realscreen)
+        self.screen.refresh()
+
+    def resized(self):
+        self.screen = Simple(self.realscreen)
+        self.screen.refresh()
+
+    def resizepty(self, ptyfd):
+        ym, xm = self.screen.getmaxyx()
+        fcntl.ioctl(ptyfd, termios.TIOCSWINSZ,
+                    struct.pack("HHHH", ym, xm, 0, 0))
+
+    def start(self):
+        self.realscreen = curses.initscr()
+        curses.start_color()
+        init_color_pairs()
+        self.screen = TwoColumn(self.realscreen)
+        curses.noecho()
+        curses.raw()
+
+    def stop(self):
+        curses.noraw()
+        curses.echo()
+        curses.endwin()
+
+    def feed(self, char):
+        self.mode[0](char, *self.mode[1:])
+
+    def feed_simple(self, char):
+        if char in 'abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ':
+            self.screen.addch(ord(char))
+        elif char in '0123456789@:~$ .#!/_(),[]=-+*\'"|<>%&\\?;`^{}':
+            self.screen.addch(ord(char))
+        elif char in '\xb6\xb7\xc3\xfc\xf6':
+            self.screen.addch(ord(char))
+        elif char == '\xc4':
+            self.screen.addch(curses.ACS_HLINE)
+        elif char == '\r':
+            self.screen.relmove(0, -9999)
+        elif char == '\n':
+            y, _ = self.screen.getyx()
+            ym, _ = self.screen.getmaxyx()
+            if y + 1 == ym:
+                self.screen.scroll()
+                self.screen.move(y, 0)
+            else:
+                self.screen.move(y+1, 0)
+        elif char == '\x1b':
+            self.mode = (self.feed_esc,)
+        elif char == '\a':
+            curses.beep()
+        elif char == '\b':
+            self.screen.relmove(0, -1)
+        else:
+            raise ValueError("feed %r" % char)
+
+    def feed_esc(self, char):
+        if char == '[':
+            self.mode = (self.feed_esc_opbr,)
+        else:
+            raise ValueError("feed esc %r" % char)
+
+    def feed_esc_opbr(self, char):
+        self.mode = (self.feed_simple,) # default
+        if char == 'H':
+            self.feed_esc_opbr_next('H', "0;0")
+        elif char == 'J':
+            self.screen.clrtobot()
+        elif char == 'm':
+            self.feed_esc_opbr_next('m', '0')
+        elif char in '0123456789':
+            self.mode = (self.feed_esc_opbr_next, char)
+        elif char == 'K':
+            self.screen.clrtoeol()
+        elif char in 'ABCDLMP':
+            self.feed_esc_opbr_next(char, '1')
+        else:
+            raise ValueError("feed esc [ %r" % char)
+
+    def feed_color(self, code):
+        if code == 0:
+            self.fg = self.bg = 0
+            self.screen.attrset(0)
+        elif code == 1:
+            self.screen.attron(curses.A_BOLD)
+        elif code == 4:
+            self.screen.attron(curses.A_UNDERLINE)
+        elif code == 7:
+            self.screen.attron(curses.A_REVERSE)
+        elif 10 <= code <= 19:
+            pass # set something about fonts. we cannot change fonts...
+        elif 30 <= code <= 37:
+            self.fg = code - 30
+            self.screen.attron(get_color(self.fg, self.bg))
+        elif code == 39:
+            self.fg = 7
+            self.screen.attron(get_color(self.fg, self.bg))
+        elif 40 <= code <= 47:
+            self.bg = code - 40
+            self.screen.attron(get_color(self.fg, self.bg))
+        elif code == 49:
+            self.bg = 0
+            self.screen.attron(get_color(self.fg, self.bg))
+        else:
+            raise ValueError("feed esc [ %r m" % code)
+
+    def feed_esc_opbr_next(self, char, prev):
+        self.mode = (self.feed_simple,) # default
+        if char in '0123456789;':
+            self.mode = (self.feed_esc_opbr_next, prev + char)
+        elif char == 'm':
+            parts = prev.split(';')
+            for p in parts:
+                self.feed_color(int(p))
+        elif char == 'H':
+            parts = prev.split(';')
+            if len(parts) != 2:
+                raise ValueError("feed esc [ %r H" % parts)
+            self.screen.move(*map((-1).__add__, map(int, parts)))
+        elif prev == '2' and char == 'J':
+            self.screen.move(0, 0)
+            self.screen.clrtobot()
+        elif char == 'C' and prev.isdigit():
+            self.screen.relmove(0, int(prev))
+        elif char == 'P' and prev.isdigit():
+            for _ in range(int(prev)):
+                self.screen.delch()
+        elif char == '@' and prev.isdigit():
+            for _ in range(int(prev)):
+                self.screen.insch(ord(' '))
+        elif char == 'A' and prev.isdigit():
+            self.screen.relmove(-int(prev), 0)
+        elif char == 'M' and prev.isdigit():
+            for _ in range(int(prev)):
+                self.screen.deleteln()
+        elif char == 'L' and prev.isdigit():
+            for _ in range(int(prev)):
+                self.screen.insertln()
+        elif char == 'D' and prev.isdigit():
+            self.screen.relmove(0, -int(prev))
+        elif char == 'd' and prev.isdigit():
+            _, x = self.screen.getyx()
+            self.screen.move(int(prev) - 1, x)
+        elif char == 'B' and prev.isdigit():
+            self.screen.relmove(int(prev), 0)
+        elif char == 'b' and prev.isdigit():
+            repchar = self.screen.inch()
+            for _ in range(int(prev)):
+                self.screen.addch(repchar)
+        elif char == 'G' and prev.isdigit():
+            y, _ = self.screen.getyx()
+            self.screen.move(y, int(prev) - 1)
+        elif char == 'X' and prev.isdigit():
+            for _ in range(int(prev)):
+                self.screen.addch(ord(' '))
+        elif char == 'K' and prev == '1':
+            y, x = self.screen.getyx()
+            self.screen.move(y, 0)
+            for _ in range(x):
+                self.screen.addch(ord(' '))
+        else:
+            raise ValueError("feed esc [ %r %r" % (prev, char))
+
+def main():
+    t = Terminal()
+
+    pid, masterfd = pty.fork()
+    if pid == 0: # child
+        os.environ["TERM"] = "ansi"
+        os.execvp(sys.argv[1], sys.argv[1:])
+        sys.exit(1)
+
+    try:
+        t.start()
+        t.resizepty(masterfd)
+        while True:
+            try:
+                res, _, _ = select.select([0, masterfd], [], [])
+            except select.error, err:
+                if err.args[0] == errno.EINTR:
+                    t.resized()
+                    t.resizepty(masterfd)
+                    continue
+                raise
+            if 0 in res:
+                data = os.read(0, 1)
+                if ord(data) == 0xb3:
+                    t.switchmode()
+                    t.resizepty(masterfd)
+                else:
+                    os.write(masterfd, data)
+            elif masterfd in res:
+                try:
+                    data = os.read(masterfd, 1024)
+                except OSError:
+                    break
+                if not data:
+                    break
+                for char in data:
+                    t.feed(char)
+                t.screen.refresh()
+    finally:
+        t.stop()
+
+if __name__ == '__main__':
+    main()
-- 
cgit v1.2.3