| """This implements an ANSI terminal emulator as a subclass of screen. |
| |
| $Id: ANSI.py 491 2007-12-16 20:04:57Z noah $ |
| """ |
| # references: |
| # http://www.retards.org/terminals/vt102.html |
| # http://vt100.net/docs/vt102-ug/contents.html |
| # http://vt100.net/docs/vt220-rm/ |
| # http://www.termsys.demon.co.uk/vtansi.htm |
| |
| import screen |
| import FSM |
| import copy |
| import string |
| |
| |
| def Emit(fsm): |
| |
| screen = fsm.memory[0] |
| screen.write_ch(fsm.input_symbol) |
| |
| |
| def StartNumber(fsm): |
| |
| fsm.memory.append(fsm.input_symbol) |
| |
| |
| def BuildNumber(fsm): |
| |
| ns = fsm.memory.pop() |
| ns = ns + fsm.input_symbol |
| fsm.memory.append(ns) |
| |
| |
| def DoBackOne(fsm): |
| |
| screen = fsm.memory[0] |
| screen.cursor_back() |
| |
| |
| def DoBack(fsm): |
| |
| count = int(fsm.memory.pop()) |
| screen = fsm.memory[0] |
| screen.cursor_back(count) |
| |
| |
| def DoDownOne(fsm): |
| |
| screen = fsm.memory[0] |
| screen.cursor_down() |
| |
| |
| def DoDown(fsm): |
| |
| count = int(fsm.memory.pop()) |
| screen = fsm.memory[0] |
| screen.cursor_down(count) |
| |
| |
| def DoForwardOne(fsm): |
| |
| screen = fsm.memory[0] |
| screen.cursor_forward() |
| |
| |
| def DoForward(fsm): |
| |
| count = int(fsm.memory.pop()) |
| screen = fsm.memory[0] |
| screen.cursor_forward(count) |
| |
| |
| def DoUpReverse(fsm): |
| |
| screen = fsm.memory[0] |
| screen.cursor_up_reverse() |
| |
| |
| def DoUpOne(fsm): |
| |
| screen = fsm.memory[0] |
| screen.cursor_up() |
| |
| |
| def DoUp(fsm): |
| |
| count = int(fsm.memory.pop()) |
| screen = fsm.memory[0] |
| screen.cursor_up(count) |
| |
| |
| def DoHome(fsm): |
| |
| c = int(fsm.memory.pop()) |
| r = int(fsm.memory.pop()) |
| screen = fsm.memory[0] |
| screen.cursor_home(r, c) |
| |
| |
| def DoHomeOrigin(fsm): |
| |
| c = 1 |
| r = 1 |
| screen = fsm.memory[0] |
| screen.cursor_home(r, c) |
| |
| |
| def DoEraseDown(fsm): |
| |
| screen = fsm.memory[0] |
| screen.erase_down() |
| |
| |
| def DoErase(fsm): |
| |
| arg = int(fsm.memory.pop()) |
| screen = fsm.memory[0] |
| if arg == 0: |
| screen.erase_down() |
| elif arg == 1: |
| screen.erase_up() |
| elif arg == 2: |
| screen.erase_screen() |
| |
| |
| def DoEraseEndOfLine(fsm): |
| |
| screen = fsm.memory[0] |
| screen.erase_end_of_line() |
| |
| |
| def DoEraseLine(fsm): |
| |
| screen = fsm.memory[0] |
| if arg == 0: |
| screen.end_of_line() |
| elif arg == 1: |
| screen.start_of_line() |
| elif arg == 2: |
| screen.erase_line() |
| |
| |
| def DoEnableScroll(fsm): |
| |
| screen = fsm.memory[0] |
| screen.scroll_screen() |
| |
| |
| def DoCursorSave(fsm): |
| |
| screen = fsm.memory[0] |
| screen.cursor_save_attrs() |
| |
| |
| def DoCursorRestore(fsm): |
| |
| screen = fsm.memory[0] |
| screen.cursor_restore_attrs() |
| |
| |
| def DoScrollRegion(fsm): |
| |
| screen = fsm.memory[0] |
| r2 = int(fsm.memory.pop()) |
| r1 = int(fsm.memory.pop()) |
| screen.scroll_screen_rows(r1, r2) |
| |
| |
| def DoMode(fsm): |
| |
| screen = fsm.memory[0] |
| mode = fsm.memory.pop() # Should be 4 |
| # screen.setReplaceMode () |
| |
| |
| def Log(fsm): |
| |
| screen = fsm.memory[0] |
| fsm.memory = [screen] |
| fout = open('log', 'a') |
| fout.write(fsm.input_symbol + ',' + fsm.current_state + '\n') |
| fout.close() |
| |
| |
| class term (screen.screen): |
| """This is a placeholder. |
| In theory I might want to add other terminal types. |
| """ |
| |
| def __init__(self, r=24, c=80): |
| screen.screen.__init__(self, r, c) |
| |
| |
| class ANSI (term): |
| |
| """This class encapsulates a generic terminal. It filters a stream and |
| maintains the state of a screen object. """ |
| |
| def __init__(self, r=24, c=80): |
| |
| term.__init__(self, r, c) |
| |
| #self.screen = screen (24,80) |
| self.state = FSM.FSM('INIT', [self]) |
| self.state.set_default_transition(Log, 'INIT') |
| self.state.add_transition_any('INIT', Emit, 'INIT') |
| self.state.add_transition('\x1b', 'INIT', None, 'ESC') |
| self.state.add_transition_any('ESC', Log, 'INIT') |
| self.state.add_transition('(', 'ESC', None, 'G0SCS') |
| self.state.add_transition(')', 'ESC', None, 'G1SCS') |
| self.state.add_transition_list('AB012', 'G0SCS', None, 'INIT') |
| self.state.add_transition_list('AB012', 'G1SCS', None, 'INIT') |
| self.state.add_transition('7', 'ESC', DoCursorSave, 'INIT') |
| self.state.add_transition('8', 'ESC', DoCursorRestore, 'INIT') |
| self.state.add_transition('M', 'ESC', DoUpReverse, 'INIT') |
| self.state.add_transition('>', 'ESC', DoUpReverse, 'INIT') |
| self.state.add_transition('<', 'ESC', DoUpReverse, 'INIT') |
| # Selects application keypad. |
| self.state.add_transition('=', 'ESC', None, 'INIT') |
| self.state.add_transition('#', 'ESC', None, 'GRAPHICS_POUND') |
| self.state.add_transition_any('GRAPHICS_POUND', None, 'INIT') |
| self.state.add_transition('[', 'ESC', None, 'ELB') |
| # ELB means Escape Left Bracket. That is ^[[ |
| self.state.add_transition('H', 'ELB', DoHomeOrigin, 'INIT') |
| self.state.add_transition('D', 'ELB', DoBackOne, 'INIT') |
| self.state.add_transition('B', 'ELB', DoDownOne, 'INIT') |
| self.state.add_transition('C', 'ELB', DoForwardOne, 'INIT') |
| self.state.add_transition('A', 'ELB', DoUpOne, 'INIT') |
| self.state.add_transition('J', 'ELB', DoEraseDown, 'INIT') |
| self.state.add_transition('K', 'ELB', DoEraseEndOfLine, 'INIT') |
| self.state.add_transition('r', 'ELB', DoEnableScroll, 'INIT') |
| self.state.add_transition('m', 'ELB', None, 'INIT') |
| self.state.add_transition('?', 'ELB', None, 'MODECRAP') |
| self.state.add_transition_list( |
| string.digits, 'ELB', StartNumber, 'NUMBER_1') |
| self.state.add_transition_list( |
| string.digits, 'NUMBER_1', BuildNumber, 'NUMBER_1') |
| self.state.add_transition('D', 'NUMBER_1', DoBack, 'INIT') |
| self.state.add_transition('B', 'NUMBER_1', DoDown, 'INIT') |
| self.state.add_transition('C', 'NUMBER_1', DoForward, 'INIT') |
| self.state.add_transition('A', 'NUMBER_1', DoUp, 'INIT') |
| self.state.add_transition('J', 'NUMBER_1', DoErase, 'INIT') |
| self.state.add_transition('K', 'NUMBER_1', DoEraseLine, 'INIT') |
| self.state.add_transition('l', 'NUMBER_1', DoMode, 'INIT') |
| # It gets worse... the 'm' code can have infinite number of |
| # number;number;number before it. I've never seen more than two, |
| # but the specs say it's allowed. crap! |
| self.state.add_transition('m', 'NUMBER_1', None, 'INIT') |
| # LED control. Same problem as 'm' code. |
| self.state.add_transition('q', 'NUMBER_1', None, 'INIT') |
| |
| # \E[?47h appears to be "switch to alternate screen" |
| # \E[?47l restores alternate screen... I think. |
| self.state.add_transition_list( |
| string.digits, 'MODECRAP', StartNumber, 'MODECRAP_NUM') |
| self.state.add_transition_list( |
| string.digits, |
| 'MODECRAP_NUM', |
| BuildNumber, |
| 'MODECRAP_NUM') |
| self.state.add_transition('l', 'MODECRAP_NUM', None, 'INIT') |
| self.state.add_transition('h', 'MODECRAP_NUM', None, 'INIT') |
| |
| # RM Reset Mode Esc [ Ps l none |
| self.state.add_transition(';', 'NUMBER_1', None, 'SEMICOLON') |
| self.state.add_transition_any('SEMICOLON', Log, 'INIT') |
| self.state.add_transition_list( |
| string.digits, 'SEMICOLON', StartNumber, 'NUMBER_2') |
| self.state.add_transition_list( |
| string.digits, 'NUMBER_2', BuildNumber, 'NUMBER_2') |
| self.state.add_transition_any('NUMBER_2', Log, 'INIT') |
| self.state.add_transition('H', 'NUMBER_2', DoHome, 'INIT') |
| self.state.add_transition('f', 'NUMBER_2', DoHome, 'INIT') |
| self.state.add_transition('r', 'NUMBER_2', DoScrollRegion, 'INIT') |
| # It gets worse... the 'm' code can have infinite number of |
| # number;number;number before it. I've never seen more than two, |
| # but the specs say it's allowed. crap! |
| self.state.add_transition('m', 'NUMBER_2', None, 'INIT') |
| # LED control. Same problem as 'm' code. |
| self.state.add_transition('q', 'NUMBER_2', None, 'INIT') |
| |
| def process(self, c): |
| |
| self.state.process(c) |
| |
| def process_list(self, l): |
| |
| self.write(l) |
| |
| def write(self, s): |
| |
| for c in s: |
| self.process(c) |
| |
| def flush(self): |
| |
| pass |
| |
| def write_ch(self, ch): |
| """This puts a character at the current cursor position. cursor |
| position if moved forward with wrap-around, but no scrolling is done if |
| the cursor hits the lower-right corner of the screen. """ |
| |
| #\r and \n both produce a call to crlf(). |
| ch = ch[0] |
| |
| if ch == '\r': |
| # self.crlf() |
| return |
| if ch == '\n': |
| self.crlf() |
| return |
| if ch == chr(screen.BS): |
| self.cursor_back() |
| self.put_abs(self.cur_r, self.cur_c, ' ') |
| return |
| |
| if ch not in string.printable: |
| fout = open('log', 'a') |
| fout.write('Nonprint: ' + str(ord(ch)) + '\n') |
| fout.close() |
| return |
| self.put_abs(self.cur_r, self.cur_c, ch) |
| old_r = self.cur_r |
| old_c = self.cur_c |
| self.cursor_forward() |
| if old_c == self.cur_c: |
| self.cursor_down() |
| if old_r != self.cur_r: |
| self.cursor_home(self.cur_r, 1) |
| else: |
| self.scroll_up() |
| self.cursor_home(self.cur_r, 1) |
| self.erase_line() |
| |
| # def test (self): |
| # |
| # import sys |
| # write_text = 'I\'ve got a ferret sticking up my nose.\n' + \ |
| # '(He\'s got a ferret sticking up his nose.)\n' + \ |
| # 'How it got there I can\'t tell\n' + \ |
| # 'But now it\'s there it hurts like hell\n' + \ |
| # 'And what is more it radically affects my sense of smell.\n' + \ |
| # '(His sense of smell.)\n' + \ |
| # 'I can see a bare-bottomed mandril.\n' + \ |
| # '(Slyly eyeing his other nostril.)\n' + \ |
| # 'If it jumps inside there too I really don\'t know what to do\n' + \ |
| # 'I\'ll be the proud posessor of a kind of nasal zoo.\n' + \ |
| # '(A nasal zoo.)\n' + \ |
| # 'I\'ve got a ferret sticking up my nose.\n' + \ |
| # '(And what is worst of all it constantly explodes.)\n' + \ |
| # '"Ferrets don\'t explode," you say\n' + \ |
| # 'But it happened nine times yesterday\n' + \ |
| # 'And I should know for each time I was standing in the way.\n' + \ |
| # 'I\'ve got a ferret sticking up my nose.\n' + \ |
| # '(He\'s got a ferret sticking up his nose.)\n' + \ |
| # 'How it got there I can\'t tell\n' + \ |
| # 'But now it\'s there it hurts like hell\n' + \ |
| # 'And what is more it radically affects my sense of smell.\n' + \ |
| # '(His sense of smell.)' |
| # self.fill('.') |
| # self.cursor_home() |
| # for c in write_text: |
| # self.write_ch (c) |
| # print str(self) |
| # |
| # if __name__ == '__main__': |
| # t = ANSI(6,65) |
| # t.test() |
