add GPU shaders demo

3d/demo_shaders.py

@@ -0,0 +1,197 @@
+
+
+import moderngl_window as mglw
+import moderngl
+#import moderngl.window
+import time
+#print(dir(moderngl))
+
+#export DISPLAY=:99.0
+#Xvfb :99 -screen 0 640x480x24 &
+#redirects screen output to invisibel framebuffer 
+# python3 main.py --window glfw --fullscreen --samples 16 --cursor false --size 800x600
+
+vertex_shader = """
+#version 430
+
+in vec3 in_position;
+
+void main(){
+    gl_Position = vec4(in_position, 0.9);
+}
+"""
+
+
+fragment_shader = """
+#version 430
+
+out vec4 fragColor;
+
+uniform vec2 resolution;
+uniform float time;
+
+vec2 rotate2D(vec2 uv, float a){
+    float s = sin(a);
+    float c = cos(a);
+    return mat2(c,-s,s,c) * uv;
+}
+vec3 ring(vec3 col,vec2 uv, int z,float ang, float speed,float scaleX, float scaleY){
+    //z =1; //count of star's
+    //ang = 270;
+    for( float i=0.0; i < z; i++){
+
+        float a = 0;
+        a = ang/z * i * (3.1415/180)  ;
+	    a += time*speed;
+
+        float dy = cos(a)/2;
+        float dx = sin(a)/2;
+
+	    dy = dy * scaleY; // *.4; // scale y
+	    dx = dx * scaleX; //*.9; // scale x
+        col += 0.01  / length(uv+vec2(dx+0.1,dy) );
+    }
+    col *=  sin(vec3(0.2,0.8,0.9)  * time) * 0.15  +0.25;
+    fragColor = vec4(col, 1.0);
+    return col;
+    //return fragColor;
+}
+
+void main() {
+    vec2 uv = (gl_FragCoord.xy - resolution.xy/2) / resolution.y;
+    vec3 col = vec3(0.0);
+    
+    //uv = rotate2D(uv, 3.14/2.0+3.2);
+    //uv = rotate2D(uv, 3.14/2.0);
+
+    float r = 0.17;
+    
+    int z = 10;
+    float ang = 360;
+    float speed = .9;
+
+    z = 50; //count of star's
+    ang = 270;
+    speed = -.1;
+    col = ring(col,uv, z, ang, speed, .9, .4);
+    
+    z = 10;
+    //col = ring(col,uv, z, 270,0.1, .7, .7);
+
+
+    z =2; //count of star's
+    ang = 90;
+    speed = 10;
+    col = ring(col,uv, z, ang, speed, .6, .2);
+
+
+
+    z = 20; //count of star's
+    ang = 360;
+    speed = 0.5;
+    col = ring(col,uv, z, ang, speed, .5, .5);
+
+
+/*
+    float i = 0;
+    float a = time+i;
+    a = ang/z * i * (3.1415/180)  ;
+    a += time/4;
+    float dx = 0.1;//cos(a)/2;
+    float dy = 0.1;//sin(a)/2;
+    //col += 0.01 / length(uv+vec2(dx+0.1,dy) );
+    col += 0.03 / length(uv+vec2(dx+0.5,dy) );
+    //col *= sin(vec3(0.8,0.2,0.9) * time) * 0.15 +0.25;
+    fragColor = vec4(col, 1.0);
+    dx = 0.2;
+    dy = 0.42;
+    col += 0.01 / length(uv+vec2(dx+0.5,dy) );
+    fragColor = vec4(col, 1.0);
+
+    uv + vec2(0,0);
+    col = vec3(0.0);
+    //fragColor = vec4(vec3(0,0,1), 1.0);
+    //fragColor[0] = [0,0,1,1];
+*/
+}
+
+
+
+"""
+
+vertex_shader2 = """
+#version 450
+
+in vec3 in_position;
+
+void main(){
+    gl_Position = vec4(in_position, 0.9);
+
+    const vec3 positions[3] = vec3[3](
+		vec3(1.f,1.f, 0.0f),
+		vec3(-1.f,1.f, 0.0f),
+		vec3(0.f,-1.f, 0.0f)
+	);
+
+	//output the position of each vertex
+	//gl_Position = vec4(positions, 1.0f);
+	//gl_Position = vec4(1.f,1.f, 0.0f, 1.0f);
+	//gl_Position = vec4(-1.f,1.f, 0.0f, 1.0f);
+	//gl_Position = vec4(1.f,-1.f, 0.0f, 1.0f);
+	//gl_Position = vec4(positions[gl_VertexIndex], 1.0f);
+}
+"""
+
+fragment_shader2 = """
+#version 450
+
+out vec4 fragColor;
+
+uniform vec2 resolution;
+uniform float time;
+
+void main(){
+    vec2 uv = (gl_FragCoord.xy - resolution.xy/2) / resolution.y;
+    vec3 col = vec3(0.0);
+	fragColor = vec4(1.f,0.f,0.f,1.0f);
+
+}
+"""
+
+
+class Screen(mglw.WindowConfig):
+    window_size = 600, 300
+    window_size = 900, 506
+    #resource_dir = 'programms'
+
+    def __init__(self,**args):
+        super().__init__(**args)
+
+        self.quad = mglw.geometry.quad_fs()
+        # load shader's as string
+        self.prog = self.ctx.program(vertex_shader=vertex_shader,
+                                      fragment_shader=fragment_shader)
+        # load shader's form file
+        #self.prog = self.load_program(vertex_shader='vertex_shader.gls1',
+        #                              #fragment_shader='fragment_shader.gls1')
+
+        self.set_uniform('resolution',self.window_size)
+        self.t = time.time()
+
+    def set_uniform(self,u_name, u_value):
+        try:
+            self.prog[u_name] = u_value
+        except KeyError:
+            print(f'uniform: {u_name} - not used in shader')
+
+    def render(self,_time,frame_time):
+        self.ctx.clear()
+        self.set_uniform('time',_time)
+        if self.t+10 < time.time():
+            print(_time)
+            self.t = time.time()
+        self.quad.render(self.prog)
+
+if __name__ == "__main__":
+    mglw.run_window_config(Screen)
+