1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57
|
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# vispy: gallery 1
"""
Simplest Shader Program
=======================
"""
import sys
from vispy import gloo
from vispy import app
import numpy as np
VERT_SHADER = """
attribute vec2 a_position;
uniform float u_size;
void main() {
gl_Position = vec4(a_position, 0.0, 1.0);
gl_PointSize = u_size;
}
"""
FRAG_SHADER = """
void main() {
gl_FragColor = vec4(1.0, 0.0, 0.0, 1.0);
}
"""
class Canvas(app.Canvas):
def __init__(self):
app.Canvas.__init__(self, keys='interactive')
ps = self.pixel_scale
self.program = gloo.Program(VERT_SHADER, FRAG_SHADER)
data = np.random.uniform(-0.5, 0.5, size=(20, 2))
self.program['a_position'] = data.astype(np.float32)
self.program['u_size'] = 20.*ps
self.show()
def on_resize(self, event):
width, height = event.size
gloo.set_viewport(0, 0, width, height)
def on_draw(self, event):
gloo.clear('white')
self.program.draw('points')
if __name__ == '__main__':
canvas = Canvas()
if sys.flags.interactive != 1:
app.run()
|