#!/usr/bin/env python

import pyglet
#import math
#import time

import serial

from colours import *

datafeed = serial.Serial(
port='/dev/ttyUSB0',
baudrate = 9600,
parity=serial.PARITY_NONE,
stopbits=serial.STOPBITS_ONE,
bytesize=serial.EIGHTBITS,
timeout=1
)

class Series:
    def __init__(self, points=100, title="Series title", xname="x-axis name", yname="y-axis name"):
        self.title = title
        self.xname = xname
        self.yname = yname
        self.data = []
        self.points = points
    def addpoint(self, point):
        self.data.append(point)
        if len(self.data) > self.points:
            del self.points[-1]

class Plot():

    def __init__(self, series, size=(640, 480)):

        """Setup a the details of a plot, and create a corresponding window"""

        self.series = series
        self.title = self.series.title

        self.size = size
        self.font = 'Arkhip'

        self.margins = (0.02, 0.02) # Fractions of window size
        self.lines = (12, 8)

        self.window = pyglet.window.Window(self.size[0], self.size[1], resizable=True)

        self.window.set_caption(self.title)

        self.window.on_resize = self.resize
        self.window.on_draw = self.draw

        self.bounds = ((int(self.window.width * self.margins[0]), int(self.window.width * (1 - self.margins[0]))),
                       (int(self.window.height * self.margins[1]), int(self.window.height * (1 - self.margins[1]))))

    def resize(self, width, height):

        """Handle a pyglet resize event, then give control back to the event loop"""

        self.size = (width, height)

        super(pyglet.window.Window, self.window).on_resize(width, height)

    def draw(self):

        """Draw all the components of the graph"""

        self.drawBackground()
        self.drawHeading()

        self.drawXAxis()

    def drawBackground(self):

        """Draw the graph background, currently a plain colour"""

        pyglet.image.SolidColorImagePattern(WHITE).create_image(self.size[0], self.size[1]).blit(0, 0)
        
    def drawHeading(self):

        """Draw a title for the graph (duplicated in the window titlebar, if present"""

        heading = pyglet.text.Label(self.title, color=BLACK,

                            font_name=self.font, font_size=self.size[0]*self.margins[0], x=self.size[0]/2, y=self.size[1],

                            anchor_x='center', anchor_y='top')
        heading.draw()

     
    def drawXAxis(self):
        pyglet.graphics.draw(2, pyglet.gl.GL_LINES, ('v2i', (self.bounds[0][0], self.bounds[1][0],
                                                             self.bounds[1][0], self.bounds[1][1])),
                             ('c3B', (0, 0, 0, 0, 0, 0)))
        
        

testseries = Series()

plots = []         

plots.append(Plot(testseries))

def pollSerial(elapsed):

    """Check serial port for incoming data"""

    # Note, elapsed is time since last call of this function

    values = datafeed.readline().strip().split(", ")
    testseries.addpoint(values)

pyglet.clock.schedule_interval(pollSerial, 0.1)

pyglet.app.run()