sonolus.py 0.1.5__py3-none-any.whl → 0.1.7__py3-none-any.whl

sonolus/build/cli.py

@@ -2,6 +2,7 @@ import argparse
 import contextlib
 import http.server
 import importlib
+import json
 import shutil
 import socket
 import socketserver
@@ -11,7 +12,7 @@ from time import perf_counter
 
 from sonolus.build.engine import package_engine
 from sonolus.build.level import package_level_data
-from sonolus.build.project import build_project_to_collection
+from sonolus.build.project import build_project_to_collection, get_project_schema
 from sonolus.script.project import Project
 
 
@@ -147,6 +148,14 @@ def main():
     dev_parser.add_argument("--build-dir", type=str, default="./build")
     dev_parser.add_argument("--port", type=int, default=8000)
 
+    schema_parser = subparsers.add_parser("schema")
+    schema_parser.add_argument(
+        "module",
+        type=str,
+        nargs="?",
+        help="Module path (e.g., 'module.name'). If omitted, will auto-detect if only one module exists.",
+    )
+
     args = parser.parse_args()
 
     if not args.module:
@@ -161,16 +170,18 @@ def main():
     if project is None:
         sys.exit(1)
 
-    build_dir = Path(args.build_dir)
-
     if args.command == "build":
+        build_dir = Path(args.build_dir)
         start_time = perf_counter()
         build_project(project, build_dir)
         end_time = perf_counter()
         print(f"Project built successfully to '{build_dir.resolve()}' in {end_time - start_time:.2f}s")
     elif args.command == "dev":
+        build_dir = Path(args.build_dir)
         start_time = perf_counter()
         build_collection(project, build_dir)
         end_time = perf_counter()
         print(f"Build finished in {end_time - start_time:.2f}s")
         run_server(build_dir / "site", port=args.port)
+    elif args.command == "schema":
+        print(json.dumps(get_project_schema(project), indent=2))

sonolus/build/project.py

@@ -5,7 +5,7 @@ from sonolus.build.engine import package_engine
 from sonolus.build.level import package_level_data
 from sonolus.script.engine import Engine
 from sonolus.script.level import Level
-from sonolus.script.project import Project
+from sonolus.script.project import Project, ProjectSchema
 
 BLANK_PNG = (
     b"\x89PNG\r\n\x1a\n\x00\x00\x00\rIHDR\x00\x00\x00\x01\x00\x00\x00\x01\x01\x00\x00\x00\x007n\xf9$"
@@ -92,3 +92,32 @@ def load_resources_files_to_collection(base_path: Path) -> Collection:
         collection.load_from_scp(path)
     collection.load_from_source(base_path)
     return collection
+
+
+def get_project_schema(project: Project) -> ProjectSchema:
+    by_archetype: dict[str, dict[str, bool]] = {}
+    for archetype in project.engine.data.play.archetypes:
+        fields = by_archetype.setdefault(archetype.name, {})
+        # If a field is exported, we should exclude it if it's imported in watch mode
+        for field in archetype._exported_keys_:
+            fields[field] = False
+        for field in archetype._imported_keys_:
+            fields[field] = True
+    for archetype in project.engine.data.watch.archetypes:
+        fields = by_archetype.setdefault(archetype.name, {})
+        for field in archetype._imported_keys_:
+            if field not in fields:
+                fields[field] = True
+    for archetype in project.engine.data.preview.archetypes:
+        fields = by_archetype.setdefault(archetype.name, {})
+        for field in archetype._imported_keys_:
+            fields[field] = True
+    return {
+        "archetypes": [
+            {
+                "name": name,
+                "fields": [*fields],
+            }
+            for name, fields in by_archetype.items()
+        ]
+    }

sonolus/script/archetype.py

@@ -5,7 +5,7 @@ from collections.abc import Callable
 from dataclasses import dataclass
 from enum import Enum, StrEnum
 from types import FunctionType
-from typing import Annotated, Any, ClassVar, Self, get_origin
+from typing import Annotated, Any, ClassVar, Self, TypedDict, get_origin
 
 from sonolus.backend.ir import IRConst, IRInstr
 from sonolus.backend.mode import Mode
@@ -307,6 +307,11 @@ class _ArchetypeLevelData:
 type _ArchetypeData = _ArchetypeSelfData | _ArchetypeReferenceData | _ArchetypeLevelData
 
 
+class ArchetypeSchema(TypedDict):
+    name: str
+    fields: list[str]
+
+
 class _BaseArchetype:
     _is_comptime_value_ = True
 
@@ -401,6 +406,10 @@ class _BaseArchetype:
             data.extend(field.type._accept_(bound.arguments[field.name] or zeros(field.type))._to_list_())
         native_call(Op.Spawn, archetype_id, *(Num(x) for x in data))
 
+    @classmethod
+    def schema(cls) -> ArchetypeSchema:
+        return {"name": cls.name or "unnamed", "fields": list(cls._imported_fields_)}
+
     def _level_data_entries(self, level_refs: dict[Any, int] | None = None):
         if not isinstance(self._data_, _ArchetypeLevelData):
             raise RuntimeError("Entity is not level data")

sonolus/script/array.py

@@ -60,7 +60,7 @@ class Array[T, Size](GenericValue, ArrayLike[T]):
                 raise ValueError(f"{cls.__name__} constructor should be used with {cls.size()} values, got {len(args)}")
             parameterized_cls = cls
         if ctx():
-            place = ctx().alloc(size=parameterized_cls.size())
+            place = ctx().alloc(size=parameterized_cls._size_())
             result: parameterized_cls = parameterized_cls._from_place_(place)
             result._copy_from_(parameterized_cls._with_value(values))
             return result

sonolus/script/internal/context.py

@@ -182,10 +182,18 @@ class Context:
             return
         assert len(self.outgoing) == 0
         self.outgoing[None] = header
+        values = {}
+        # First do a pass through and copy every value
         for name, target_value in header.loop_variables.items():
             with using_ctx(self):
                 if type(target_value)._is_value_type_():
                     value = self.scope.get_value(name)
+                    values[name] = value._get_()  # _get_() will make a copy on value types
+        # Then actually set them
+        for name, target_value in header.loop_variables.items():
+            with using_ctx(self):
+                if type(target_value)._is_value_type_():
+                    value = values[name]
                     value = type(target_value)._accept_(value)
                     target_value._set_(value)
                 else:

sonolus/script/project.py

@@ -1,6 +1,10 @@
+from __future__ import annotations
+
 from os import PathLike
 from pathlib import Path
+from typing import Self, TypedDict
 
+from sonolus.script.archetype import ArchetypeSchema
 from sonolus.script.engine import Engine
 from sonolus.script.level import Level
 
@@ -23,3 +27,50 @@ class Project:
         self.engine = engine
         self.levels = levels or []
         self.resources = Path(resources or "resources")
+
+    def with_levels(self, levels: list[Level]) -> Self:
+        """Create a new project with the specified levels.
+
+        Args:
+            levels: The levels of the project.
+
+        Returns:
+            The new project.
+        """
+        return Project(self.engine, levels, self.resources)
+
+    def dev(self, build_dir: PathLike, port: int = 8080):
+        """Start a development server for the project.
+
+        Args:
+            build_dir: The path to the build directory.
+            port: The port of the development server.
+        """
+        from sonolus.build.cli import build_collection, run_server
+
+        build_collection(self, Path(build_dir))
+        run_server(Path(build_dir) / "site", port=port)
+
+    def build(self, build_dir: PathLike):
+        """Build the project.
+
+        Args:
+            build_dir: The path to the build directory.
+        """
+        from sonolus.build.cli import build_project
+
+        build_project(self, Path(build_dir))
+
+    def schema(self) -> ProjectSchema:
+        """Generate the schema of the project.
+
+        Returns:
+            The schema of the project.
+        """
+        from sonolus.build.project import get_project_schema
+
+        return get_project_schema(self)
+
+
+class ProjectSchema(TypedDict):
+    archetypes: list[ArchetypeSchema]

sonolus/script/quad.py

@@ -3,7 +3,7 @@ from __future__ import annotations
 from typing import Protocol, Self
 
 from sonolus.script.record import Record
-from sonolus.script.vec import Vec2
+from sonolus.script.vec import Vec2, pnpoly
 
 
 class Quad(Record):
@@ -95,6 +95,17 @@ class Quad(Record):
         """Rotate the quad by the given angle about its center and return a new quad."""
         return self.rotate_about(angle, self.center)
 
+    def contains_point(self, point: Vec2, /) -> bool:
+        """Check if the quad contains the given point.
+
+        Args:
+            point: The point to check.
+
+        Returns:
+            True if the point is inside the quad, False otherwise.
+        """
+        return pnpoly((self.bl, self.tl, self.tr, self.br), point)
+
 
 class Rect(Record):
     """A rectangle defined by its top, right, bottom, and left edges.
@@ -225,6 +236,17 @@ class Rect(Record):
             l=self.l + shrinkage.x,
         )
 
+    def contains_point(self, point: Vec2, /) -> bool:
+        """Check if the rectangle contains the given point.
+
+        Args:
+            point: The point to check.
+
+        Returns:
+            True if the point is inside the rectangle, False otherwise.
+        """
+        return self.l <= point.x <= self.r and self.b <= point.y <= self.t
+
 
 class QuadLike(Protocol):
     """A protocol for types that can be used as quads."""

sonolus/script/sprite.py

@@ -353,9 +353,7 @@ class StandardSprite:
     SIMULTANEOUS_CONNECTION_PURPLE = Annotated[Sprite, sprite("#SIMULTANEOUS_CONNECTION_PURPLE")]
     SIMULTANEOUS_CONNECTION_CYAN = Annotated[Sprite, sprite("#SIMULTANEOUS_CONNECTION_CYAN")]
 
-    SIMULTANEOUS_CONNECTION_NEUTRAL_SEAMLESS = Annotated[
-        Sprite, sprite("#SIMULTANEOUS_CONNECTION_NEUTRAL_SEAMLESS")
-    ]
+    SIMULTANEOUS_CONNECTION_NEUTRAL_SEAMLESS = Annotated[Sprite, sprite("#SIMULTANEOUS_CONNECTION_NEUTRAL_SEAMLESS")]
     SIMULTANEOUS_CONNECTION_RED_SEAMLESS = Annotated[Sprite, sprite("#SIMULTANEOUS_CONNECTION_RED_SEAMLESS")]
     SIMULTANEOUS_CONNECTION_GREEN_SEAMLESS = Annotated[Sprite, sprite("#SIMULTANEOUS_CONNECTION_GREEN_SEAMLESS")]
     SIMULTANEOUS_CONNECTION_BLUE_SEAMLESS = Annotated[Sprite, sprite("#SIMULTANEOUS_CONNECTION_BLUE_SEAMLESS")]

sonolus/script/vec.py

@@ -1,6 +1,8 @@
 from math import atan2, cos, sin
 from typing import Self
 
+from sonolus.script.array import Array
+from sonolus.script.array_like import ArrayLike
 from sonolus.script.num import Num
 from sonolus.script.record import Record
 
@@ -194,3 +196,30 @@ class Vec2(Record):
             A new vector with inverted direction.
         """
         return Vec2(x=-self.x, y=-self.y)
+
+
+def pnpoly(vertices: ArrayLike[Vec2] | tuple[Vec2, ...], test: Vec2) -> bool:
+    """Check if a point is inside a polygon.
+
+    No guaranteed behavior for points on the edges or very close to the edges.
+
+    Args:
+        vertices: The vertices of the polygon.
+        test: The point to test.
+
+    Returns:
+        Whether the point is inside the polygon.
+    """
+    if isinstance(vertices, tuple):
+        vertices = Array(*vertices)
+    i = 0
+    j = len(vertices) - 1
+    c = False
+    while i < len(vertices):
+        if (vertices[i].y > test.y) != (vertices[j].y > test.y) and test.x < (vertices[j].x - vertices[i].x) * (
+            test.y - vertices[i].y
+        ) / (vertices[j].y - vertices[i].y) + vertices[i].x:
+            c = not c
+        j = i
+        i += 1
+    return c

{sonolus_py-0.1.5.dist-info → sonolus_py-0.1.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: sonolus.py
-Version: 0.1.5
+Version: 0.1.7
 Summary: Sonolus engine development in Python
 Requires-Python: >=3.13
 Description-Content-Type: text/markdown

{sonolus_py-0.1.5.dist-info → sonolus_py-0.1.7.dist-info}/RECORD

@@ -26,16 +26,16 @@ sonolus/backend/optimize/passes.py,sha256=OPzcpFUU7qbQFSrgr84UuzDg_vB1GYkZj2K2aq
 sonolus/backend/optimize/simplify.py,sha256=Tz4RftjH5TpIxoIOwiZuLchsrwzI9GaS_exdjAW6HKk,7927
 sonolus/backend/optimize/ssa.py,sha256=D3CQm3s3gcuU0_k_0pXVrwirm4xjAZsX6corrTDS1Co,9013
 sonolus/build/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-sonolus/build/cli.py,sha256=YlWg3Jfe1gi2wnlIJDh8LCMLSPV3l9AiuDWk2Bbphmg,5812
+sonolus/build/cli.py,sha256=12sFalCUJmn0Dg3f8ipNIkPmZ_djG15gldqWzDG7AYo,6241
 sonolus/build/collection.py,sha256=IsDgedsAJ-foHzQ4LnQ9zSVXSz5wVVt4sgqiUICBvCU,10573
 sonolus/build/compile.py,sha256=wBiRX6rnq8Op4_vVXJ-bPLtffgV4ppNBa9M9hIe9twI,3579
 sonolus/build/engine.py,sha256=TAJCdSWOsEcTq9o6jfR3A2JZ8gOoc9YQx6FC_qW-dFw,6766
 sonolus/build/level.py,sha256=3sdGPvIZ4a15u3-JdITnB6rznp6a2E3k6A0gB8A15JA,610
 sonolus/build/node.py,sha256=jwsVWt6Brh4M9MypUt3Nqnxfm7fXrCMRWYQGwBTAszI,1162
-sonolus/build/project.py,sha256=x350tPcwKCYakdbrP796b545hVUaBwrzY0yEcQR7ut4,4023
+sonolus/build/project.py,sha256=QukkCVQBGxnp1F2jQjaAkTg3TutwhWqKGXAfQZDMk1c,5162
 sonolus/script/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-sonolus/script/archetype.py,sha256=gYRkUD_hI_MhXA-diIObIp0xqpX6aNiPqWLhiw-t094,33887
-sonolus/script/array.py,sha256=KJxyL4I7A-CNFQW1RbZYQGVI527XZmoesDR-q6BurB8,9339
+sonolus/script/archetype.py,sha256=cQDoNVO7dUHYQ7RlAHiyoEnUYt0seCnFPD_bPzo0BNE,34114
+sonolus/script/array.py,sha256=rPAK3kdeL6A_a82DSnxDe6Cl99vjS2UVU97n1r8arsA,9341
 sonolus/script/array_like.py,sha256=OVOLMadS-Jj941S-EgoxlFPdaQLT6ZrHzaMnKtB6PfU,8346
 sonolus/script/bucket.py,sha256=ZNlNeBZ1FzRLQ49bTuuJDmnYZ-_sjQEvosySx3ESxQc,6944
 sonolus/script/containers.py,sha256=tf0-UUSq8NH6k9jIKq6qgXRxanFQ0f5olH6uHghTynk,13171
@@ -53,22 +53,22 @@ sonolus/script/options.py,sha256=nTOuoRC2p_qoMx4ngOVDY2WZDooTxwwGTGKyb994v2A,751
 sonolus/script/particle.py,sha256=K06ArT9tstRNdbuGIviDmDDWcK3-ieA53LHg0Xvizow,8304
 sonolus/script/pointer.py,sha256=pOh5vCSl6xR7mDm5PMOhBc_wf4O-b938DE0LtZWBmPc,1119
 sonolus/script/print.py,sha256=mNYu9QWiacBBGZrnePZQMVwbbguoelUps9GiOK_aVRU,2096
-sonolus/script/project.py,sha256=iGdMEpT6mUcyTmfGwmIW25ucHB51ETd1_SyQ6UC7AYg,612
-sonolus/script/quad.py,sha256=SVMcfRYx8Bv5t7A2vhus9QUiw-1V-qtRhldr3DnnNWw,7693
+sonolus/script/project.py,sha256=DhmBuH0CkFKmF27o1V9XhIOsywXDi57i-Wpwyd-70LM,2040
+sonolus/script/quad.py,sha256=e2kXKSp9K46Q9qstLsGLFPxaW7Z2kfVfignA8Ka2-Pw,8375
 sonolus/script/record.py,sha256=rOVlo-eUUvFG_Ee-wKrgX_ZwOqWKWtS7s8m-juM70yE,11216
 sonolus/script/runtime.py,sha256=xeiLJFeqkCev1x6pBiM5afyvi3GWdpLJ5OYRg9M9nwo,18562
-sonolus/script/sprite.py,sha256=-hURLrLrxtu73qSQ4x4_gJqnJ3XiORUmNojh5fdm2E8,15732
+sonolus/script/sprite.py,sha256=FOozsPtaQnInnm-QshtYznnC66_8jb1D2sHmo7nFEsY,15718
 sonolus/script/text.py,sha256=IsoINZJXefjReYDjJFwVaFsUCdgeQvPBDeywljM2dWo,13025
 sonolus/script/timing.py,sha256=ZR0ypV2PIoDCMHHGOMfCeezStCsBQdzomdqaz5VKex0,2981
 sonolus/script/transform.py,sha256=hH6KSRQC8vV-Z10CRCrGewMYqQwUMH3mQIEmniuC2Zw,10760
 sonolus/script/ui.py,sha256=kyuP88sLRJPT-Yx-7fx8Xu9Wdegyw_CJNslcP4WnDUs,7268
 sonolus/script/values.py,sha256=JuvJknskuY6FPprUp9R-6Gj2TDJLu4ppbVcYedG5dG0,1049
-sonolus/script/vec.py,sha256=Uph6Z9xofwbKGZu0H7rl6fzlO4bj5i0cnWu4LQeIL4I,5270
+sonolus/script/vec.py,sha256=looCsmYpt6jlasdBqddtkC0hA1exZ3lxPP3_BL8cPXw,6148
 sonolus/script/internal/__init__.py,sha256=T6rzLoiOUaiSQtaHMZ88SNO-ijSjSSv33TKtUwu-Ms8,136
 sonolus/script/internal/builtin_impls.py,sha256=w27aKMhLX4Ivd7uKz1ZgVy1lC5scSSJ23-VfsbCfIb0,7731
 sonolus/script/internal/callbacks.py,sha256=vWzJG8uiJoEtsNnbeZPqOHogCwoLpz2D1MnHY2wVV8s,2801
 sonolus/script/internal/constant.py,sha256=lIBBR84Rfw0rBSZU2reCrUnB3yEMEdOX6A7_LgptYz8,3776
-sonolus/script/internal/context.py,sha256=evLOXcOqskK5Gh5Je86ghKlMaon-Opi7dDoMtkCjEh8,13533
+sonolus/script/internal/context.py,sha256=I9X89aTWgoY-_jgnJHEnzzVLAvILACtPfgdE4ipxjOc,13930
 sonolus/script/internal/descriptor.py,sha256=XRFey-EjiAm_--KsNl-8N0Mi_iyQwlPh68gDp0pKf3E,392
 sonolus/script/internal/dict_impl.py,sha256=alu_wKGSk1kZajNf64qbe7t71shEzD4N5xNIATH8Swo,1885
 sonolus/script/internal/error.py,sha256=ZNnsvQVQAnFKzcvsm6-sste2lo-tP5pPI8sD7XlAZWc,490
@@ -82,8 +82,8 @@ sonolus/script/internal/range.py,sha256=lrTanQFHU7RuQxSSPwDdoC30Y8FnHGxcP1Ahditu
 sonolus/script/internal/transient.py,sha256=pSDFGu0m26zVsp6owmNTeSPue-osDd1JahjJSijW8t0,1520
 sonolus/script/internal/tuple_impl.py,sha256=vjXmScLVdeTkDn3t9fgIRqtW31iwngnaP2rmA6nlsLw,3431
 sonolus/script/internal/value.py,sha256=ik9sMKl0TbsH_C6QNxD4WfpAnmBFISgmmlazWwh3kY0,4308
-sonolus_py-0.1.5.dist-info/METADATA,sha256=HHVoPymf7iwqOCBJqIEMETzGgzgPIwEA3Owrjudj_a4,216
-sonolus_py-0.1.5.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
-sonolus_py-0.1.5.dist-info/entry_points.txt,sha256=oTYspY_b7SA8TptEMTDxh4-Aj-ZVPnYC9f1lqH6s9G4,54
-sonolus_py-0.1.5.dist-info/licenses/LICENSE,sha256=JEKpqVhQYfEc7zg3Mj462sKbKYmO1K7WmvX1qvg9IJk,1067
-sonolus_py-0.1.5.dist-info/RECORD,,
+sonolus_py-0.1.7.dist-info/METADATA,sha256=R9hAvB7RLZb6duTGJWpmquqnfLZ4gpWJDGOcsm175ls,216
+sonolus_py-0.1.7.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
+sonolus_py-0.1.7.dist-info/entry_points.txt,sha256=oTYspY_b7SA8TptEMTDxh4-Aj-ZVPnYC9f1lqH6s9G4,54
+sonolus_py-0.1.7.dist-info/licenses/LICENSE,sha256=JEKpqVhQYfEc7zg3Mj462sKbKYmO1K7WmvX1qvg9IJk,1067
+sonolus_py-0.1.7.dist-info/RECORD,,