Add files via upload

Author: JingShing

crt_shader.py

@@ -1,55 +1,88 @@
 import struct
-import pygame
+import pygame, os, sys
 from pygame.locals import *
-from settings import *
+
 import moderngl
 
+def resource_path(relative):
+	if hasattr(sys, "_MEIPASS"):
+		absolute_path = os.path.join(sys._MEIPASS, relative)
+	else:
+		absolute_path = os.path.join(relative)
+	return absolute_path
+
 class Graphic_engine:
-    # you need to give this class screen
-    # screen is a Surface you declare to replace display
-    def __init__(self, screen):
+    def __init__(self, screen, style = 1, VIRTUAL_RES=(800, 600), cpu_only=False, fullscreen=False):
         pygame.init()
+        self.VIRTUAL_RES = VIRTUAL_RES
+        self.cpu_only = cpu_only
         self.screen = screen
-        self.ctx = moderngl.create_context()
-        self.texture_coordinates = [0, 1,  1, 1,
-                                    0, 0,  1, 0]
-        self.world_coordinates = [-1, -1,  1, -1,
-                                  -1,  1,  1,  1]
-        self.render_indices = [0, 1, 2,
-                               1, 2, 3]
-
-        self.prog = self.ctx.program(
-            vertex_shader=open('shaders/VERTEX_SHADER.glsl').read(),
-            fragment_shader=open('shaders/FRAGMENT_SHADER.glsl').read(),
-        )
-
-        self.screen_texture = self.ctx.texture(
-            VIRTUAL_RES, 3,
-            pygame.image.tostring(screen, "RGB", 1)
+        self.fullscreen = fullscreen
+        if not(self.cpu_only):
+            self.ctx = moderngl.create_context()
+            self.texture_coordinates = [0, 1,  1, 1,
+                                        0, 0,  1, 0]
+            self.world_coordinates = [-1, -1,  1, -1,
+                                    -1,  1,  1,  1]
+            self.render_indices = [0, 1, 2,
+                                1, 2, 3]
+
+            self.style = style
+            # shader style : 0, no shader. 1, crt. 2, flat_crt.
+            self.prog = self.ctx.program(
+                vertex_shader=open(resource_path('shaders/VERTEX_SHADER.glsl')).read(),
+                fragment_shader=open(resource_path('shaders/FRAGMENT_SHADER.glsl')).read(),
             )
+            self.prog['mode'] = self.style
 
-        self.screen_texture.repeat_x = False
-        self.screen_texture.repeat_y = False
+            self.screen_texture = self.ctx.texture(
+                self.VIRTUAL_RES, 3,
+                pygame.image.tostring(screen, "RGB", 1)
+                )
 
-        self.vbo = self.ctx.buffer(struct.pack('8f', *self.world_coordinates))
-        self.uvmap = self.ctx.buffer(struct.pack('8f', *self.texture_coordinates))
-        self.ibo= self.ctx.buffer(struct.pack('6I', *self.render_indices))
+            self.screen_texture.repeat_x = False
+            self.screen_texture.repeat_y = False
 
-        self.vao_content = [
-            (self.vbo, '2f', 'vert'),
-            (self.uvmap, '2f', 'in_text')
-        ]
+            self.vbo = self.ctx.buffer(struct.pack('8f', *self.world_coordinates))
+            self.uvmap = self.ctx.buffer(struct.pack('8f', *self.texture_coordinates))
+            self.ibo= self.ctx.buffer(struct.pack('6I', *self.render_indices))
 
-        self.vao = self.ctx.vertex_array(self.prog, self.vao_content, self.ibo)
+            self.vao_content = [
+                (self.vbo, '2f', 'vert'),
+                (self.uvmap, '2f', 'in_text'),
+            ]
+
+            self.vao = self.ctx.vertex_array(self.prog, self.vao_content, index_buffer=self.ibo)
+        else:
+            self.diaplay = pygame.display.get_surface()
+
+    def change_shader(self):
+        if not self.cpu_only:
+            self.__init__(self.screen, (self.style + 1) % 3, self.VIRTUAL_RES)
 
     def render(self):
-        # render will do all the work and update the screen
-        texture_data = self.screen.get_view('1')
-        self.screen_texture.write(texture_data)
-        self.ctx.clear(14/255,40/255,66/255)
-        self.screen_texture.use()
-        self.vao.render()
-        pygame.display.flip()
+        if not(self.cpu_only):
+            texture_data = self.screen.get_view('1')
+            self.screen_texture.write(texture_data)
+            self.ctx.clear(14/255,40/255,66/255)
+            self.screen_texture.use()
+            self.vao.render()
+            pygame.display.flip()
+        else:
+            self.diaplay.blit(self.screen, (0, 0))
+            pygame.display.update()
+    
+    def Full_screen(self, REAL_RES):
+        if not(self.cpu_only):
+            if not(self.fullscreen):
+                pygame.display.set_mode(REAL_RES, pygame.DOUBLEBUF|pygame.OPENGL)
+            else:
+                pygame.display.set_mode(REAL_RES, pygame.DOUBLEBUF|pygame.OPENGL|pygame.FULLSCREEN)
+        else:
+            if not(self.fullscreen):
+                pygame.display.set_mode(self.VIRTUAL_RES)
+            else:
+                pygame.display.set_mode(self.VIRTUAL_RES, pygame.FULLSCREEN)
 
     def __call__(self):
         return self.render()