pyglet_test.py 2.25 KB
import pyglet
import math
import random
import time
 
window = pyglet.window.Window()
window.set_caption("Test graph")

starttime = time.time()
 
title = pyglet.text.Label('Test Graph',
                          font_name='Arkhip',
                          font_size=12,
                          x=window.width//2, y=window.height//2,
                          anchor_x='center', anchor_y='top')

                 
def round_sig(x, sig=2):
    return round(x, sig-int(math.floor(math.log10(abs(x))))-1)
                 
def drawgrid(xlines, ylines, xlimits, ylimits):
    pyglet.gl.glColor4f(0.5, 0.5, 0.5, 1.0) 
    for xpos in range(0, window.width, window.width/xlines):
        pyglet.graphics.draw(2, pyglet.gl.GL_LINES, ('v2i', (xpos, 0, xpos, window.height)))
        tagtext = str(round((xlimits[1]-xlimits[0])*(float(xpos)/window.width), 1))
        tag = pyglet.text.Label(tagtext, font_name='Arkhip', font_size=10, x=xpos-2, y=0, anchor_x='right', anchor_y='bottom')
        tag.draw()
    for ypos in range(0, window.width, window.width/ylines):
        pyglet.graphics.draw(2, pyglet.gl.GL_LINES, ('v2i', (0, ypos, window.width, ypos)))
        tagtext = str(round((ylimits[1]-ylimits[0])*(float(ypos)/window.height), 1))
        tag = pyglet.text.Label(tagtext, font_name='Arkhip', font_size=10, x=0, y=ypos-2, anchor_x='left', anchor_y='top')
        tag.draw()


def plotline(xdata, ydata):
    pyglet.gl.glColor4f(1, 0, 0, 1.0)
    points = []
    for n in range(max(len(xdata), len(ydata))):
        xpos = ((xdata[n]-min(xdata))*window.width)/(max(xdata)-min(xdata))
        xpos = int(xpos)
        ypos = ((ydata[n]-min(ydata))*window.height)/(max(ydata)-min(ydata))
        ypos = int(ypos)
        points.append([xpos, ypos])
    for n in range(len(points)-1):            
        pyglet.graphics.draw(2, pyglet.gl.GL_LINES, ('v2i', (points[n][0], points[n][1], points[n+1][0], points[n+1][1])))


xdata = [0]
ydata = [0]


@window.event
def on_draw():
    window.clear()
    ydata.append(random.uniform(0, 10))
    if len(ydata) > 100:
        del ydata[0]
    xdata.append(round(time.time() - starttime, 3))
    if len(xdata) > 100:
        del xdata[0]
    drawgrid(10, 12, [min(xdata), max(xdata)], [min(ydata), max(ydata)])
    plotline(xdata, ydata)

    
pyglet.app.run()