basilisk-engine 0.1.3__py3-none-any.whl → 0.1.4__py3-none-any.whl

basilisk/render/shader_handler.py

@@ -1,80 +1,80 @@
-import moderngl as mgl
-import glm
-from .shader import Shader
-
-
-class ShaderHandler:
-    engine: ...
-    """Back reference to the parent engine"""
-    scene: ...
-    """Back reference to the parent scene"""
-    ctx: mgl.Context
-    """Back reference to the parent context"""
-    shaders: set
-    """Dictionary containing all the shaders"""
-    uniform_values: dict = {}
-    """Dictionary containing uniform values"""    
-
-    def __init__(self, scene) -> None:
-        """
-        Handles all the shader programs in a basilisk scene
-        """
-        
-        # Back references
-        self.scene  = scene
-        self.engine = scene.engine
-        self.ctx    = scene.engine.ctx
-
-        # Initalize dictionaries
-        self.shaders = set()
-        self.add(self.engine.shader)
-
-    def add(self, shader: Shader) -> None:
-        """
-        Creates a shader program from a file name.
-        Parses through shaders to identify uniforms and save for writting
-        """
-
-
-        if not shader: return None
-        if shader in self.shaders: return shader
-
-        self.shaders.add(shader)
-        
-        if self.scene.material_handler:
-            self.scene.light_handler.write()
-            self.scene.material_handler.write()
-            self.scene.material_handler.image_handler.write()
-
-        return shader
-
-    def get_uniforms_values(self) -> None:
-        """
-        Gets uniforms from various parts of the scene.
-        These values are stored and used in write_all_uniforms and update_uniforms.
-        This is called by write_all_uniforms and update_uniforms, so there is no need to call this manually.
-        """
-        
-        self.uniform_values = {
-            'projectionMatrix' : self.scene.camera.m_proj,
-            'viewMatrix' : self.scene.camera.m_view,
-            'cameraPosition' : self.scene.camera.position,
-        }
-
-    def write(self) -> None:
-        """
-        Writes all of the uniforms in every shader program.
-        """
-
-        self.get_uniforms_values()
-        for uniform in self.uniform_values:
-            for shader in self.shaders:
-                if not uniform in shader.uniforms: continue  # Does not write uniforms not in the shader
-                shader.write(uniform, self.uniform_values[uniform])
-
-    def release(self) -> None:
-        """
-        Releases all shader programs in handler
-        """
-        
+import moderngl as mgl
+import glm
+from .shader import Shader
+
+
+class ShaderHandler:
+    engine: ...
+    """Back reference to the parent engine"""
+    scene: ...
+    """Back reference to the parent scene"""
+    ctx: mgl.Context
+    """Back reference to the parent context"""
+    shaders: set
+    """Dictionary containing all the shaders"""
+    uniform_values: dict = {}
+    """Dictionary containing uniform values"""    
+
+    def __init__(self, scene) -> None:
+        """
+        Handles all the shader programs in a basilisk scene
+        """
+        
+        # Back references
+        self.scene  = scene
+        self.engine = scene.engine
+        self.ctx    = scene.engine.ctx
+
+        # Initalize dictionaries
+        self.shaders = set()
+        self.add(self.engine.shader)
+
+    def add(self, shader: Shader) -> None:
+        """
+        Creates a shader program from a file name.
+        Parses through shaders to identify uniforms and save for writting
+        """
+
+
+        if not shader: return None
+        if shader in self.shaders: return shader
+
+        self.shaders.add(shader)
+        
+        if self.scene.material_handler:
+            self.scene.light_handler.write()
+            self.scene.material_handler.write()
+            self.scene.material_handler.image_handler.write()
+
+        return shader
+
+    def get_uniforms_values(self) -> None:
+        """
+        Gets uniforms from various parts of the scene.
+        These values are stored and used in write_all_uniforms and update_uniforms.
+        This is called by write_all_uniforms and update_uniforms, so there is no need to call this manually.
+        """
+        
+        self.uniform_values = {
+            'projectionMatrix' : self.scene.camera.m_proj,
+            'viewMatrix' : self.scene.camera.m_view,
+            'cameraPosition' : self.scene.camera.position,
+        }
+
+    def write(self) -> None:
+        """
+        Writes all of the uniforms in every shader program.
+        """
+
+        self.get_uniforms_values()
+        for uniform in self.uniform_values:
+            for shader in self.shaders:
+                if not uniform in shader.uniforms: continue  # Does not write uniforms not in the shader
+                shader.write(uniform, self.uniform_values[uniform])
+
+    def release(self) -> None:
+        """
+        Releases all shader programs in handler
+        """
+        
         [shader.__del__() for shader in self.shaders]

basilisk/render/sky.py

@@ -1,121 +1,121 @@
-import numpy as np
-from PIL import Image as PIL_Image
-from .shader import Shader
-
-class Sky:
-    texture_cube=None
-    vbo          = None
-    vao          = None
-    def __init__(self, engine, sky_texture: str | list=None):
-        """
-        Handler for all skybox rendering
-        """
-        
-        self.scene = engine.scene
-        self.ctx   = engine.ctx
-        
-        if not sky_texture: sky_texture = engine.root + '/bsk_assets/skybox.png'
-
-        self.set_renderer()
-        self.set_texture(sky_texture)
-
-    def render(self):
-        """
-        Render the skybox to current render destination
-        """
-        self.vao.render()
-
-    def write(self):
-        # Write the texture cube to the sky shader
-        self.shader.program['skyboxTexture']  = 8
-        self.texture_cube.use(location = 8)
-
-        shader = self.scene.engine.shader
-        if 'skyboxTexture' not in shader.uniforms: return
-        shader.program['skyboxTexture'] = 8
-        self.texture_cube.use(location = 8)
-
-
-    def set_texture(self, skybox_images: list):
-        """
-        Sets the skybox texture. Can either be set with 6 images for each skybox side or a single skybox image.
-        List items should be string paths.
-        The six images should be should be in the following order: right, left, top, bottom, front, back
-        """
-
-        # Release any existing memory
-        if self.texture_cube: self.texture_cube.release()
-
-        # Function-Scoped data
-        images = None
-
-        # Given a sinlge image for the skybox        
-        if isinstance(skybox_images, str) or ((isinstance(skybox_images, list) or isinstance(skybox_images, tuple)) and len(skybox_images) == 1):
-            path = skybox_images if isinstance(skybox_images, str) else skybox_images[0]
-            
-            # Verify the path type
-            if not isinstance(path, str): raise ValueError(f"Skybox: Invalid image path type {type(path)}")
-
-            image = PIL_Image.open(path).convert('RGB')
-            width, height = image.size[0] // 4, image.size[1] // 3
-
-            images = [image.crop((x * width, y * height, (x + 1) * width, (y + 1) * height)) for x, y in [(2, 1), (0, 1), (1, 0), (1, 2), (1, 1), (3, 1)]]
-
-        # Given a list of images for the skybox
-        elif isinstance(skybox_images, list) or isinstance(skybox_images, tuple):
-            # Verify the correct number of images was given
-            if len(skybox_images) != 6: raise ValueError("Skybox: Invalid number of images for skybox. Expected 1 or 6")
-            # Verify the all image path types            
-            if not all([isinstance(path, str) for path in skybox_images]): raise ValueError(f"Skybox: Invalid image path type {type(path)}")
-
-            images = [PIL_Image.open(path).convert('RGB') for path in skybox_images]
-
-        else:
-            raise ValueError(f"Skybox: Invalid skybox type {type(skybox_images)}. Expected list of string paths or a single image")
-        
-        # Create a texture map from the images
-        size = min(images[0].size)
-        size = (size, size)
-        images = [img.resize(size) for img in images]
-        images = [img.tobytes() for img in images]
-        self.texture_cube = self.ctx.texture_cube(size=size, components=3, data=None)
-        for i, data in enumerate(images):
-            self.texture_cube.write(face=i, data=data)
-
-    def set_renderer(self):
-        """
-        
-        """
-        
-        # Release any existing memory
-        if self.vbo: self.vbo.release()
-        if self.vao: self.vao.release()
-        
-        # Get the cube vertex data
-        vertices = [(-1, -1, 1), ( 1, -1,  1), (1,  1,  1), (-1, 1,  1),
-                    (-1, 1, -1), (-1, -1, -1), (1, -1, -1), ( 1, 1, -1)]
-
-        indices = [(0, 2, 3), (0, 1, 2),
-                   (1, 7, 2), (1, 6, 7),
-                   (6, 5, 4), (4, 7, 6),
-                   (3, 4, 5), (3, 5, 0),
-                   (3, 7, 4), (3, 2, 7),
-                   (0, 6, 1), (0, 5, 6)]
-        
-        vertex_data = np.array([vertices[ind] for trigangle in indices for ind in trigangle], dtype='f4')
-        vertex_data = np.flip(vertex_data, 1).copy(order='C')
-
-        # Create a renderable vao
-        self.vbo     = self.ctx.buffer(vertex_data)
-        root = self.scene.engine.root
-        self.shader = self.scene.shader_handler.add(Shader(self.scene.engine, root + '/shaders/sky.vert', root + '/shaders/sky.frag'))
-        self.vao     = self.ctx.vertex_array(self.shader.program, [(self.vbo, '3f', 'in_position')], skip_errors=True)
-
-    def __del__(self):
-        """
-        Releases all data used by the skybox
-        """
-        
-        if self.texture_cube: self.texture_cube.release()
-        if self.vbo: self.vbo.release()
+import numpy as np
+from PIL import Image as PIL_Image
+from .shader import Shader
+
+class Sky:
+    texture_cube=None
+    vbo          = None
+    vao          = None
+    def __init__(self, engine, sky_texture: str | list=None):
+        """
+        Handler for all skybox rendering
+        """
+        
+        self.scene = engine.scene
+        self.ctx   = engine.ctx
+        
+        if not sky_texture: sky_texture = engine.root + '/bsk_assets/skybox.png'
+
+        self.set_renderer()
+        self.set_texture(sky_texture)
+
+    def render(self):
+        """
+        Render the skybox to current render destination
+        """
+        self.vao.render()
+
+    def write(self):
+        # Write the texture cube to the sky shader
+        self.shader.program['skyboxTexture']  = 8
+        self.texture_cube.use(location = 8)
+
+        shader = self.scene.engine.shader
+        if 'skyboxTexture' not in shader.uniforms: return
+        shader.program['skyboxTexture'] = 8
+        self.texture_cube.use(location = 8)
+
+
+    def set_texture(self, skybox_images: list):
+        """
+        Sets the skybox texture. Can either be set with 6 images for each skybox side or a single skybox image.
+        List items should be string paths.
+        The six images should be should be in the following order: right, left, top, bottom, front, back
+        """
+
+        # Release any existing memory
+        if self.texture_cube: self.texture_cube.release()
+
+        # Function-Scoped data
+        images = None
+
+        # Given a sinlge image for the skybox        
+        if isinstance(skybox_images, str) or ((isinstance(skybox_images, list) or isinstance(skybox_images, tuple)) and len(skybox_images) == 1):
+            path = skybox_images if isinstance(skybox_images, str) else skybox_images[0]
+            
+            # Verify the path type
+            if not isinstance(path, str): raise ValueError(f"Skybox: Invalid image path type {type(path)}")
+
+            image = PIL_Image.open(path).convert('RGB')
+            width, height = image.size[0] // 4, image.size[1] // 3
+
+            images = [image.crop((x * width, y * height, (x + 1) * width, (y + 1) * height)) for x, y in [(2, 1), (0, 1), (1, 0), (1, 2), (1, 1), (3, 1)]]
+
+        # Given a list of images for the skybox
+        elif isinstance(skybox_images, list) or isinstance(skybox_images, tuple):
+            # Verify the correct number of images was given
+            if len(skybox_images) != 6: raise ValueError("Skybox: Invalid number of images for skybox. Expected 1 or 6")
+            # Verify the all image path types            
+            if not all([isinstance(path, str) for path in skybox_images]): raise ValueError(f"Skybox: Invalid image path type {type(path)}")
+
+            images = [PIL_Image.open(path).convert('RGB') for path in skybox_images]
+
+        else:
+            raise ValueError(f"Skybox: Invalid skybox type {type(skybox_images)}. Expected list of string paths or a single image")
+        
+        # Create a texture map from the images
+        size = min(images[0].size)
+        size = (size, size)
+        images = [img.resize(size) for img in images]
+        images = [img.tobytes() for img in images]
+        self.texture_cube = self.ctx.texture_cube(size=size, components=3, data=None)
+        for i, data in enumerate(images):
+            self.texture_cube.write(face=i, data=data)
+
+    def set_renderer(self):
+        """
+        
+        """
+        
+        # Release any existing memory
+        if self.vbo: self.vbo.release()
+        if self.vao: self.vao.release()
+        
+        # Get the cube vertex data
+        vertices = [(-1, -1, 1), ( 1, -1,  1), (1,  1,  1), (-1, 1,  1),
+                    (-1, 1, -1), (-1, -1, -1), (1, -1, -1), ( 1, 1, -1)]
+
+        indices = [(0, 2, 3), (0, 1, 2),
+                   (1, 7, 2), (1, 6, 7),
+                   (6, 5, 4), (4, 7, 6),
+                   (3, 4, 5), (3, 5, 0),
+                   (3, 7, 4), (3, 2, 7),
+                   (0, 6, 1), (0, 5, 6)]
+        
+        vertex_data = np.array([vertices[ind] for trigangle in indices for ind in trigangle], dtype='f4')
+        vertex_data = np.flip(vertex_data, 1).copy(order='C')
+
+        # Create a renderable vao
+        self.vbo     = self.ctx.buffer(vertex_data)
+        root = self.scene.engine.root
+        self.shader = self.scene.shader_handler.add(Shader(self.scene.engine, root + '/shaders/sky.vert', root + '/shaders/sky.frag'))
+        self.vao     = self.ctx.vertex_array(self.shader.program, [(self.vbo, '3f', 'in_position')], skip_errors=True)
+
+    def __del__(self):
+        """
+        Releases all data used by the skybox
+        """
+        
+        if self.texture_cube: self.texture_cube.release()
+        if self.vbo: self.vbo.release()
         if self.vao: self.vao.release()