sonolus.py 0.9.3__py3-none-any.whl → 0.10.1__py3-none-any.whl

sonolus/script/internal/math_impls.py

@@ -4,72 +4,72 @@ from sonolus.backend.ops import Op
 from sonolus.script.internal.native import native_function
 
 
-@native_function(Op.Sin)
+@native_function(Op.Sin, const_eval=True)
 def _sin(x: float) -> float:
     return math.sin(x)
 
 
-@native_function(Op.Cos)
+@native_function(Op.Cos, const_eval=True)
 def _cos(x: float) -> float:
     return math.cos(x)
 
 
-@native_function(Op.Tan)
+@native_function(Op.Tan, const_eval=True)
 def _tan(x: float) -> float:
     return math.tan(x)
 
 
-@native_function(Op.Arcsin)
+@native_function(Op.Arcsin, const_eval=True)
 def _asin(x: float) -> float:
     return math.asin(x)
 
 
-@native_function(Op.Arccos)
+@native_function(Op.Arccos, const_eval=True)
 def _acos(x: float) -> float:
     return math.acos(x)
 
 
-@native_function(Op.Arctan)
+@native_function(Op.Arctan, const_eval=True)
 def _atan(x: float) -> float:
     return math.atan(x)
 
 
-@native_function(Op.Arctan2)
+@native_function(Op.Arctan2, const_eval=True)
 def _atan2(y: float, x: float) -> float:
     return math.atan2(y, x)
 
 
-@native_function(Op.Sinh)
+@native_function(Op.Sinh, const_eval=True)
 def _sinh(x: float) -> float:
     return math.sinh(x)
 
 
-@native_function(Op.Cosh)
+@native_function(Op.Cosh, const_eval=True)
 def _cosh(x: float) -> float:
     return math.cosh(x)
 
 
-@native_function(Op.Tanh)
+@native_function(Op.Tanh, const_eval=True)
 def _tanh(x: float) -> float:
     return math.tanh(x)
 
 
-@native_function(Op.Floor)
+@native_function(Op.Floor, const_eval=True)
 def _floor(x: float) -> float:
     return math.floor(x)
 
 
-@native_function(Op.Ceil)
+@native_function(Op.Ceil, const_eval=True)
 def _ceil(x: float) -> float:
     return math.ceil(x)
 
 
-@native_function(Op.Trunc)
+@native_function(Op.Trunc, const_eval=True)
 def _trunc(x: float) -> float:
     return math.trunc(x)
 
 
-@native_function(Op.Round)
+@native_function(Op.Round, const_eval=True)
 def __round(x: float) -> float:
     return round(x)
 
@@ -80,12 +80,12 @@ def _round(x: float, n: int = 0) -> float:
     return __round(x * 10**n) / 10**n
 
 
-@native_function(Op.Frac)
+@native_function(Op.Frac, const_eval=True)
 def frac(x: float) -> float:
     return x % 1
 
 
-@native_function(Op.Log)
+@native_function(Op.Log, const_eval=True)
 def _ln(x: float) -> float:
     return math.log(x)
 
@@ -101,19 +101,19 @@ def _sqrt(x: float) -> float:
     return x**0.5
 
 
-@native_function(Op.Degree)
+@native_function(Op.Degree, const_eval=True)
 def _degrees(x: float) -> float:
     """Convert radians to degrees."""
     return math.degrees(x)
 
 
-@native_function(Op.Radian)
+@native_function(Op.Radian, const_eval=True)
 def _radians(x: float) -> float:
     """Convert degrees to radians."""
     return math.radians(x)
 
 
-@native_function(Op.Rem)
+@native_function(Op.Rem, const_eval=True)
 def _remainder(x: float, y: float) -> float:
     # This is different from math.remainder in Python's math package, which could be confusing
     return math.copysign(abs(x) % abs(y), x)

sonolus/script/internal/native.py

@@ -20,7 +20,7 @@ def native_call(op: Op, *args: int | float | bool) -> Num:
     return Num._from_place_(result)
 
 
-def native_function[**P, R](op: Op) -> Callable[[Callable[P, R]], Callable[P, R]]:
+def native_function[**P, R](op: Op, const_eval: bool = False) -> Callable[[Callable[P, R]], Callable[P, R]]:
     def decorator(fn: Callable[P, int | float | bool]) -> Callable[P, Num]:
         signature = inspect.signature(fn)
 
@@ -30,6 +30,12 @@ def native_function[**P, R](op: Op) -> Callable[[Callable[P, R]], Callable[P, R]
             if len(args) < sum(1 for p in signature.parameters.values() if p.default == inspect.Parameter.empty):
                 raise TypeError(f"Expected {len(signature.parameters)} arguments, got {len(args)}")
             if ctx():
+                if const_eval:
+                    args = tuple(validate_value(arg) for arg in args)
+                    if not all(_is_num(arg) for arg in args):
+                        raise RuntimeError("All arguments must be of type Num")
+                    if all(arg._is_py_() for arg in args):
+                        return Num._accept_(fn(*[arg._as_py_() for arg in args]))
                 bound_args = signature.bind(*args)
                 bound_args.apply_defaults()
                 return native_call(op, *bound_args.args)

sonolus/script/internal/range.py

@@ -41,6 +41,9 @@ class Range(Record, ArrayLike[int]):
     def __getitem__(self, index: int) -> int:
         return self.start + get_positive_index(index, len(self)) * self.step
 
+    def get_unchecked(self, index: Num) -> int:
+        return self.start + index * self.step
+
     def __setitem__(self, index: int, value: int):
         raise TypeError("Range does not support item assignment")
 

sonolus/script/interval.py

@@ -374,9 +374,10 @@ def interp(
     Returns:
         The interpolated value.
     """
-    assert len(xp) == len(fp)
-    assert len(xp) >= 2
+    assert len(xp) == len(fp), "xp and fp must have the same length"
+    assert len(xp) >= 2, "xp and fp must have at least 2 elements"
     for i in range_or_tuple(1, len(xp) - 1):
+        assert xp[i] > xp[i - 1], "xp must be in increasing order"
         # At i == 1, x may be less than x[0], but since we're extrapolating, we use the first segment regardless.
         if x <= xp[i]:
             return remap(xp[i - 1], xp[i], fp[i - 1], fp[i], x)
@@ -402,11 +403,12 @@ def interp_clamped(
     Returns:
         The interpolated value.
     """
-    assert len(xp) == len(fp)
-    assert len(xp) >= 2
+    assert len(xp) == len(fp), "xp and fp must have the same length"
+    assert len(xp) >= 2, "xp and fp must have at least 2 elements"
     if x <= xp[0]:
         return fp[0]
     for i in range_or_tuple(1, len(xp)):
+        assert xp[i] > xp[i - 1], "xp must be in increasing order"
         if x <= xp[i]:
             return remap(xp[i - 1], xp[i], fp[i - 1], fp[i], x)
     return fp[-1]

sonolus/script/options.py

@@ -191,7 +191,7 @@ class _OptionField(SonolusDescriptor):
         if sim_ctx():
             return sim_ctx().get_or_put_value((instance, self), lambda: copy(self.info.default))
         if ctx():
-            match ctx().global_state.mode:
+            match ctx().mode_state.mode:
                 case Mode.PLAY:
                     block = ctx().blocks.LevelOption
                 case Mode.WATCH:
@@ -212,7 +212,7 @@ class _OptionField(SonolusDescriptor):
         if sim_ctx():
             return sim_ctx().set_or_put_value((instance, self), lambda: copy(self.info.default), value)
         if ctx() and debug_config().unchecked_writes:
-            match ctx().global_state.mode:
+            match ctx().mode_state.mode:
                 case Mode.PLAY:
                     block = ctx().blocks.LevelOption
                 case Mode.WATCH:

sonolus/script/particle.py

@@ -5,7 +5,7 @@ from dataclasses import dataclass
 from typing import Annotated, Any, NewType, dataclass_transform, get_origin
 
 from sonolus.backend.ops import Op
-from sonolus.script.array_like import ArrayLike
+from sonolus.script.array_like import ArrayLike, check_positive_index
 from sonolus.script.debug import static_error
 from sonolus.script.internal.introspection import get_field_specifiers
 from sonolus.script.internal.native import native_function
@@ -71,7 +71,10 @@ class ParticleGroup(Record, ArrayLike[Particle]):
         return self.size
 
     def __getitem__(self, index: int) -> Particle:
-        assert 0 <= index < self.size
+        check_positive_index(index, self.size)
+        return Particle(self.start_id + index)
+
+    def get_unchecked(self, index: int) -> Particle:
         return Particle(self.start_id + index)
 
     def __setitem__(self, index: int, value: Particle) -> None:

sonolus/script/project.py

@@ -8,7 +8,6 @@ from typing import ClassVar, TypedDict
 
 from sonolus.backend.optimize import optimize
 from sonolus.backend.optimize.passes import CompilerPass
-from sonolus.build.compile import CompileCache
 from sonolus.script.archetype import ArchetypeSchema
 from sonolus.script.engine import Engine
 from sonolus.script.level import ExternalLevelData, Level, LevelData
@@ -64,13 +63,11 @@ class Project:
             port: The port of the development server.
             config: The build configuration.
         """
-        from sonolus.build.cli import build_collection, run_server
+        from sonolus.build.cli import run_server
 
         if config is None:
             config = BuildConfig()
 
-        cache = CompileCache()
-        build_collection(self, Path(build_dir), config, cache=cache)
         run_server(
             Path(build_dir) / "site",
             port=port,
@@ -78,7 +75,7 @@ class Project:
             core_module_names=None,
             build_dir=Path(build_dir),
             config=config,
-            cache=cache,
+            project=self,
         )
 
     def build(self, build_dir: PathLike, config: BuildConfig | None = None):

sonolus/script/runtime.py

@@ -343,7 +343,7 @@ class RuntimeUi(Record):
 
         Available in play, watch, preview, and tutorial mode.
         """
-        match ctx().global_state.mode:
+        match ctx().mode_state.mode:
             case Mode.PLAY:
                 return UiLayout(_PlayRuntimeUi.menu)
             case Mode.WATCH:
@@ -362,7 +362,7 @@ class RuntimeUi(Record):
 
         Available in play, watch, preview, and tutorial mode.
         """
-        match ctx().global_state.mode:
+        match ctx().mode_state.mode:
             case Mode.PLAY:
                 return UiConfig(_PlayRuntimeUiConfigs.menu)
             case Mode.WATCH:
@@ -381,7 +381,7 @@ class RuntimeUi(Record):
 
         Available in play and watch mode.
         """
-        match ctx().global_state.mode:
+        match ctx().mode_state.mode:
             case Mode.PLAY:
                 return UiLayout(_PlayRuntimeUi.judgment)
             case Mode.WATCH:
@@ -396,7 +396,7 @@ class RuntimeUi(Record):
 
         Available in play and watch mode.
         """
-        match ctx().global_state.mode:
+        match ctx().mode_state.mode:
             case Mode.PLAY:
                 return UiConfig(_PlayRuntimeUiConfigs.judgment)
             case Mode.WATCH:
@@ -411,7 +411,7 @@ class RuntimeUi(Record):
 
         Available in play and watch mode.
         """
-        match ctx().global_state.mode:
+        match ctx().mode_state.mode:
             case Mode.PLAY:
                 return UiLayout(_PlayRuntimeUi.combo_value)
             case Mode.WATCH:
@@ -426,7 +426,7 @@ class RuntimeUi(Record):
 
         Available in play and watch mode.
         """
-        match ctx().global_state.mode:
+        match ctx().mode_state.mode:
             case Mode.PLAY:
                 return UiLayout(_PlayRuntimeUi.combo_text)
             case Mode.WATCH:
@@ -441,7 +441,7 @@ class RuntimeUi(Record):
 
         Available in play and watch mode.
         """
-        match ctx().global_state.mode:
+        match ctx().mode_state.mode:
             case Mode.PLAY:
                 return UiConfig(_PlayRuntimeUiConfigs.combo)
             case Mode.WATCH:
@@ -456,7 +456,7 @@ class RuntimeUi(Record):
 
         Available in play and watch mode.
         """
-        match ctx().global_state.mode:
+        match ctx().mode_state.mode:
             case Mode.PLAY:
                 return UiLayout(_PlayRuntimeUi.primary_metric_bar)
             case Mode.WATCH:
@@ -471,7 +471,7 @@ class RuntimeUi(Record):
 
         Available in play and watch mode.
         """
-        match ctx().global_state.mode:
+        match ctx().mode_state.mode:
             case Mode.PLAY:
                 return UiLayout(_PlayRuntimeUi.primary_metric_value)
             case Mode.WATCH:
@@ -486,7 +486,7 @@ class RuntimeUi(Record):
 
         Available in play and watch mode.
         """
-        match ctx().global_state.mode:
+        match ctx().mode_state.mode:
             case Mode.PLAY:
                 return UiConfig(_PlayRuntimeUiConfigs.primary_metric)
             case Mode.WATCH:
@@ -501,7 +501,7 @@ class RuntimeUi(Record):
 
         Available in play and watch mode.
         """
-        match ctx().global_state.mode:
+        match ctx().mode_state.mode:
             case Mode.PLAY:
                 return UiLayout(_PlayRuntimeUi.secondary_metric_bar)
             case Mode.WATCH:
@@ -516,7 +516,7 @@ class RuntimeUi(Record):
 
         Available in play and watch mode.
         """
-        match ctx().global_state.mode:
+        match ctx().mode_state.mode:
             case Mode.PLAY:
                 return UiLayout(_PlayRuntimeUi.secondary_metric_value)
             case Mode.WATCH:
@@ -531,7 +531,7 @@ class RuntimeUi(Record):
 
         Available in play and watch mode.
         """
-        match ctx().global_state.mode:
+        match ctx().mode_state.mode:
             case Mode.PLAY:
                 return UiConfig(_PlayRuntimeUiConfigs.secondary_metric)
             case Mode.WATCH:
@@ -546,7 +546,7 @@ class RuntimeUi(Record):
 
         Available in watch and preview mode.
         """
-        match ctx().global_state.mode:
+        match ctx().mode_state.mode:
             case Mode.WATCH:
                 return UiLayout(_WatchRuntimeUi.progress)
             case Mode.PREVIEW:
@@ -561,7 +561,7 @@ class RuntimeUi(Record):
 
         Available in watch and preview mode.
         """
-        match ctx().global_state.mode:
+        match ctx().mode_state.mode:
             case Mode.WATCH:
                 return UiConfig(_WatchRuntimeUiConfigs.progress)
             case Mode.PREVIEW:
@@ -576,7 +576,7 @@ class RuntimeUi(Record):
 
         Available in tutorial mode.
         """
-        match ctx().global_state.mode:
+        match ctx().mode_state.mode:
             case Mode.TUTORIAL:
                 return UiLayout(_TutorialRuntimeUi.previous)
             case _:
@@ -589,7 +589,7 @@ class RuntimeUi(Record):
 
         Available in tutorial mode.
         """
-        match ctx().global_state.mode:
+        match ctx().mode_state.mode:
             case Mode.TUTORIAL:
                 return UiLayout(_TutorialRuntimeUi.next)
             case _:
@@ -602,7 +602,7 @@ class RuntimeUi(Record):
 
         Available in tutorial mode.
         """
-        match ctx().global_state.mode:
+        match ctx().mode_state.mode:
             case Mode.TUTORIAL:
                 return UiConfig(_TutorialRuntimeUiConfigs.navigation)
             case _:
@@ -615,7 +615,7 @@ class RuntimeUi(Record):
 
         Available in tutorial mode.
         """
-        match ctx().global_state.mode:
+        match ctx().mode_state.mode:
             case Mode.TUTORIAL:
                 return UiLayout(_TutorialRuntimeUi.instruction)
             case _:
@@ -628,7 +628,7 @@ class RuntimeUi(Record):
 
         Available in tutorial mode.
         """
-        match ctx().global_state.mode:
+        match ctx().mode_state.mode:
             case Mode.TUTORIAL:
                 return UiConfig(_TutorialRuntimeUiConfigs.instruction)
             case _:
@@ -841,7 +841,7 @@ def is_debug() -> bool:
     """Check if the game is running in debug mode."""
     if not ctx():
         return False
-    match ctx().global_state.mode:
+    match ctx().mode_state.mode:
         case Mode.PLAY:
             return _PlayRuntimeEnvironment.is_debug
         case Mode.WATCH:
@@ -857,25 +857,25 @@ def is_debug() -> bool:
 @meta_fn
 def is_play() -> bool:
     """Check if the game is running in play mode."""
-    return bool(ctx() and ctx().global_state.mode == Mode.PLAY)
+    return bool(ctx() and ctx().mode_state.mode == Mode.PLAY)
 
 
 @meta_fn
 def is_preview() -> bool:
     """Check if the game is running in preview mode."""
-    return bool(ctx() and ctx().global_state.mode == Mode.PREVIEW)
+    return bool(ctx() and ctx().mode_state.mode == Mode.PREVIEW)
 
 
 @meta_fn
 def is_watch() -> bool:
     """Check if the game is running in watch mode."""
-    return bool(ctx() and ctx().global_state.mode == Mode.WATCH)
+    return bool(ctx() and ctx().mode_state.mode == Mode.WATCH)
 
 
 @meta_fn
 def is_tutorial() -> bool:
     """Check if the game is running in tutorial mode."""
-    return bool(ctx() and ctx().global_state.mode == Mode.TUTORIAL)
+    return bool(ctx() and ctx().mode_state.mode == Mode.TUTORIAL)
 
 
 @meta_fn
@@ -892,7 +892,7 @@ def aspect_ratio() -> float:
     """Get the aspect ratio of the game."""
     if not ctx():
         return 16 / 9
-    match ctx().global_state.mode:
+    match ctx().mode_state.mode:
         case Mode.PLAY:
             return _PlayRuntimeEnvironment.aspect_ratio
         case Mode.WATCH:
@@ -911,7 +911,7 @@ def audio_offset() -> float:
     """
     if not ctx():
         return 0
-    match ctx().global_state.mode:
+    match ctx().mode_state.mode:
         case Mode.PLAY:
             return _PlayRuntimeEnvironment.audio_offset
         case Mode.WATCH:
@@ -930,7 +930,7 @@ def input_offset() -> float:
     """
     if not ctx():
         return 0
-    match ctx().global_state.mode:
+    match ctx().mode_state.mode:
         case Mode.PLAY:
             return _PlayRuntimeEnvironment.input_offset
         case Mode.WATCH:
@@ -947,7 +947,7 @@ def is_multiplayer() -> bool:
     """
     if not ctx():
         return False
-    match ctx().global_state.mode:
+    match ctx().mode_state.mode:
         case Mode.PLAY:
             return _PlayRuntimeEnvironment.is_multiplayer
         case _:
@@ -962,7 +962,7 @@ def is_replay() -> bool:
     """
     if not ctx():
         return False
-    match ctx().global_state.mode:
+    match ctx().mode_state.mode:
         case Mode.WATCH:
             return _WatchRuntimeEnvironment.is_replay
         case _:
@@ -977,7 +977,7 @@ def time() -> float:
     """
     if not ctx():
         return 0
-    match ctx().global_state.mode:
+    match ctx().mode_state.mode:
         case Mode.PLAY:
             return _PlayRuntimeUpdate.time
         case Mode.WATCH:
@@ -996,7 +996,7 @@ def offset_adjusted_time() -> float:
     """
     if not ctx():
         return 0
-    match ctx().global_state.mode:
+    match ctx().mode_state.mode:
         case Mode.PLAY:
             return _PlayRuntimeUpdate.time - _PlayRuntimeEnvironment.input_offset
         case Mode.WATCH:
@@ -1015,7 +1015,7 @@ def delta_time() -> float:
     """
     if not ctx():
         return 0
-    match ctx().global_state.mode:
+    match ctx().mode_state.mode:
         case Mode.PLAY:
             return _PlayRuntimeUpdate.delta_time
         case Mode.WATCH:
@@ -1034,7 +1034,7 @@ def scaled_time() -> float:
     """
     if not ctx():
         return 0
-    match ctx().global_state.mode:
+    match ctx().mode_state.mode:
         case Mode.PLAY:
             return _PlayRuntimeUpdate.scaled_time
         case Mode.WATCH:
@@ -1059,7 +1059,7 @@ def touches() -> ArrayLike[Touch]:
     """Get the current touches of the game."""
     if not ctx():
         return Array[Touch, 0]()  # type: ignore
-    match ctx().global_state.mode:
+    match ctx().mode_state.mode:
         case Mode.PLAY:
             return ArrayPointer[Touch]._raw(
                 size=Num._accept_(_PlayRuntimeUpdate.touch_count),
@@ -1078,7 +1078,7 @@ def is_skip() -> bool:
     """
     if not ctx():
         return False
-    match ctx().global_state.mode:
+    match ctx().mode_state.mode:
         case Mode.WATCH:
             return _WatchRuntimeUpdate.is_skip
         case _:
@@ -1093,7 +1093,7 @@ def navigation_direction() -> int:
     """
     if not ctx():
         return 0
-    match ctx().global_state.mode:
+    match ctx().mode_state.mode:
         case Mode.TUTORIAL:
             return _TutorialRuntimeUpdate.navigation_direction
         case _:

sonolus/script/sprite.py

@@ -4,11 +4,12 @@ from enum import StrEnum
 from typing import Annotated, Any, NewType, dataclass_transform, get_origin
 
 from sonolus.backend.ops import Op
-from sonolus.script.array_like import ArrayLike
+from sonolus.script.array_like import ArrayLike, check_positive_index
 from sonolus.script.debug import static_error
 from sonolus.script.internal.impl import perf_meta_fn
 from sonolus.script.internal.introspection import get_field_specifiers
 from sonolus.script.internal.native import native_function
+from sonolus.script.num import Num
 from sonolus.script.quad import QuadLike, flatten_quad
 from sonolus.script.record import Record
 from sonolus.script.vec import Vec2
@@ -138,7 +139,10 @@ class SpriteGroup(Record, ArrayLike[Sprite]):
         return self.size
 
     def __getitem__(self, index: int) -> Sprite:
-        assert 0 <= index < self.size
+        check_positive_index(index, self.size)
+        return Sprite(self.start_id + index)
+
+    def get_unchecked(self, index: Num) -> Sprite:
         return Sprite(self.start_id + index)
 
     def __setitem__(self, index: int, value: Sprite) -> None:

sonolus/script/stream.py

@@ -118,19 +118,19 @@ def streams[T](cls: type[T]) -> T:
 
 @meta_fn
 def _check_can_read_stream() -> None:
-    if not ctx() or ctx().global_state.mode != Mode.WATCH:
+    if not ctx() or ctx().mode_state.mode != Mode.WATCH:
         raise RuntimeError("Stream read operations are only allowed in watch mode.")
 
 
 @meta_fn
 def _check_can_write_stream() -> None:
-    if not ctx() or ctx().global_state.mode != Mode.PLAY:
+    if not ctx() or ctx().mode_state.mode != Mode.PLAY:
         raise RuntimeError("Stream write operations are only allowed in play mode.")
 
 
 @meta_fn
 def _check_can_read_or_write_stream() -> None:
-    if not ctx() or ctx().global_state.mode not in {Mode.PLAY, Mode.WATCH}:
+    if not ctx() or ctx().mode_state.mode not in {Mode.PLAY, Mode.WATCH}:
         raise RuntimeError("Stream operations are only allowed in play and watch modes.")
 
 

sonolus/script/transform.py

@@ -325,6 +325,7 @@ class Transform2d(Record):
         Returns:
             A new normalized transform.
         """
+        assert self.a22 != 0, "Cannot normalize transform with a22 == 0"
         return Transform2d(
             self.a00 / self.a22,
             self.a01 / self.a22,

sonolus/script/vec.py

@@ -158,6 +158,7 @@ class Vec2(Record):
             A new vector with magnitude 1.
         """
         magnitude = (self.x**2 + self.y**2) ** 0.5
+        assert magnitude != 0, "Cannot normalize a zero vector"
         return Vec2._quick_construct(x=self.x / magnitude, y=self.y / magnitude)
 
     @perf_meta_fn

{sonolus_py-0.9.3.dist-info → sonolus_py-0.10.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sonolus.py
-Version: 0.9.3
+Version: 0.10.1
 Summary: Sonolus engine development in Python
 Project-URL: Documentation, https://sonolus.py.qwewqa.xyz/
 Project-URL: Repository, https://github.com/qwewqa/sonolus.py