Difficult-Rocket/libs/pyglet_rs/__init__.py

#  -------------------------------
#  Difficult Rocket
#  Copyright © 2020-2023 by shenjackyuanjie 3695888@qq.com
#  All rights reserved
#  -------------------------------

from .lib import *

from typing import TYPE_CHECKING

if TYPE_CHECKING:
    from pyglet.event import EventDispatcher

    def get_version_str() -> str: ...


    class Sprite_rs(EventDispatcher): ...

    class Vector2_rs: ...
    class Vector3_rs: ...
    class Vector4_rs: ...

    class Matrix3_rs: ...
    class Matrix4_rs: ...


def patch_sprite():
    from pyglet import sprite
    sprite.Sprite = Sprite_rs


def patch_vector():
    from pyglet import math
    math.Vector2 = Vector2_rs
    math.Vector3 = Vector3_rs
    math.Vector4 = Vector4_rs


def patch_matrix():
    from pyglet import math
    math.Matrix3 = Matrix3_rs
    math.Matrix4 = Matrix4_rs


def patch_all():
    patch_sprite()
    patch_vector()
    patch_matrix()