#
# A PyGtk-based Python Trace Collector window
#
# Copyright (C) 2007 TK Soh <teekaysoh@gmail.com>
#

import pygtk
pygtk.require("2.0")
import gtk
import gobject
import pango
import threading
import Queue
import win32trace

try:
    from hggtk.hglib import toutf
except ImportError:
    import locale
    _encoding = locale.getpreferredencoding()
    def toutf(s):
        return s.decode(_encoding, 'replace').encode('utf-8')

class TraceLog():
    def __init__(self):
        self.window = gtk.Window(gtk.WINDOW_TOPLEVEL)
        self.window.set_title("Python Trace Collector")
        
        # construct window
        self.window.set_default_size(700, 400)
        self.main_area = gtk.VBox()
        self.window.add(self.main_area)
        
        # mimic standard dialog widgets
        self.action_area = gtk.HBox()
        self.main_area.pack_end(self.action_area, False, False, 5)
        sep = gtk.HSeparator()
        self.main_area.pack_end(sep, False, False, 0)
        self.vbox = gtk.VBox()
        self.main_area.pack_end(self.vbox)        
        
        # add python trace ouput window
        scrolledwindow = gtk.ScrolledWindow()
        scrolledwindow.set_policy(gtk.POLICY_AUTOMATIC, gtk.POLICY_AUTOMATIC)
        self.textview = gtk.TextView(buffer=None)
        self.textview.set_editable(False)
        self.textview.modify_font(pango.FontDescription("Monospace"))
        scrolledwindow.add(self.textview)
        self.textview.set_editable(False)
        self.textbuffer = self.textview.get_buffer()
        self.vbox.pack_start(scrolledwindow, True, True)
        self.vbox.show_all()

        # add buttons
        self._button_quit = gtk.Button("Quit")
        self._button_quit.connect('clicked', self._on_ok_clicked)
        self.action_area.pack_end(self._button_quit, False, False, 5)        

        self._button_clear = gtk.Button("Clear")
        self._button_clear.connect('clicked', self._on_clear_clicked)
        self.action_area.pack_end(self._button_clear, False, False, 5)

        # add assorted window event handlers
        self.window.connect('map_event', self._on_window_map_event)
        self.window.connect('delete_event', self._on_window_close_clicked)

    def _on_ok_clicked(self, button):
        self._stop_read_thread()
        gtk.main_quit()
        
    def _on_clear_clicked(self, button):
        self.write("", False)
        
    def _on_window_close_clicked(self, event, param):
        self._stop_read_thread()
        gtk.main_quit()
        
    def _on_window_map_event(self, event, param):
        self._begin_trace()
    
    def _begin_trace(self):
        self.queue = Queue.Queue()
        win32trace.InitRead()
        self.write("Collecting Python Trace Output...\n")
        gobject.timeout_add(10, self._process_queue)
        self._start_read_thread()
        
    def _start_read_thread(self):
        self._read_trace = True
        self.thread1 = threading.Thread(target=self._do_read_trace)
        self.thread1.start()

    def _stop_read_thread(self):
        self._read_trace = False

        # wait for worker thread to to fix Unhandled exception in thread
        self.thread1.join() 
        
    def _process_queue(self):
        """
        Handle all the messages currently in the queue (if any).
        """
        while self.queue.qsize():
            try:
                msg = self.queue.get(0)
                self.write(msg)
            except Queue.Empty:
                pass
                
        return True
        
    def _do_read_trace(self):
        """
        print buffer collected in win32trace
        """
        while self._read_trace:
            msg = win32trace.read()
            if msg:
                self.queue.put(msg)
        
    def write(self, msg, append=True):
        msg = toutf(msg)
        if append:
            enditer = self.textbuffer.get_end_iter()
            self.textbuffer.insert(enditer, msg)
        else:
            self.textbuffer.set_text(msg)

    def main(self):
        self.window.show_all()
        gtk.main()
        
def run():
    dlg = TraceLog()
    dlg.main()
    
if __name__ == "__main__":
    run()