mettagrid 0.0.1__py3-none-any.whl → 0.0.3__py3-none-any.whl

mettagrid-0.0.3.dist-info/RECORD

@@ -0,0 +1,81 @@
+LICENSE,sha256=_0ZygdDOImEUQjXap6LgAetwhuvlOfOqcMDmNlAdjLo,1070
+README.md,sha256=464FGzZirXWaYNrQe7SlyD2gmOHmJktRnNJ9Tro2Nek,4680
+build/lib.macosx-11.0-arm64-cpython-311/mettagrid/actions/actions.cpython-311-darwin.so,sha256=ZSQxWXe3SlkA_QYOeBCzvrj4IivyWvDwjYEE_Q2osAE,218336
+build/lib.macosx-11.0-arm64-cpython-311/mettagrid/actions/attack.cpython-311-darwin.so,sha256=LdsKoO8hdXrmtYXrV3ax_rnVljTQg5HEUM_bvH9EHB8,217696
+build/lib.macosx-11.0-arm64-cpython-311/mettagrid/actions/gift.cpython-311-darwin.so,sha256=xwvpz-lM9z6_GIxrphR2TNXNEM10Y4kxw5-ZRppfPBs,217064
+build/lib.macosx-11.0-arm64-cpython-311/mettagrid/actions/move.cpython-311-darwin.so,sha256=zLNYBMHescRTGANyqueyI7FoihpFRG2sSuVwCg6OIAg,217064
+build/lib.macosx-11.0-arm64-cpython-311/mettagrid/actions/noop.cpython-311-darwin.so,sha256=YRWcWJ4s_841bPxvmKhb8C_Gx7xHTWY9GK_lnHe6KPA,217064
+build/lib.macosx-11.0-arm64-cpython-311/mettagrid/actions/rotate.cpython-311-darwin.so,sha256=bXV286F_B1oGEl1qBB7x-uMaciZ4en_p97Gh5cjat7Y,217152
+build/lib.macosx-11.0-arm64-cpython-311/mettagrid/actions/shield.cpython-311-darwin.so,sha256=sB9KwHQANcqozw6RFhzTz1MHsrs869TVa4SL5dPtcHY,217152
+build/lib.macosx-11.0-arm64-cpython-311/mettagrid/actions/use.cpython-311-darwin.so,sha256=OLTTen8vVonsh7Z55SlmNt6wGQp-h6tbOd_NdVxvz88,217224
+build/lib.macosx-11.0-arm64-cpython-311/mettagrid/mettagrid_c.cpython-311-darwin.so,sha256=qKkjft1j7TYrux1XKr29j17gtdsUJavqTy4V3ZqKyuk,277600
+build/lib.macosx-11.0-arm64-cpython-311/mettagrid/objects.cpython-311-darwin.so,sha256=RFWTbLIn5QNIjxCVG6Z7jBYNbsecsXEOsyuxXnHGNk4,257136
+build/mettagrid/actions/actions.cpp,sha256=o1Y5QWJNQdSVbUU5JYzEIm_QrY608gyjKzE2PwYwBuw,1277005
+build/mettagrid/actions/attack.cpp,sha256=6VNZJxBwxj7B2_5hhaAX3FZWMbv0Up-EhCkVgbcoFb8,1274448
+build/mettagrid/actions/gift.cpp,sha256=jWvdEHgX5NVLXjtRg4t6-qJBvuloT3jzUqXVD7tisY4,1226668
+build/mettagrid/actions/move.cpp,sha256=xzlDbe5TMSNULthlE4H2oQXr7uVD1-NEtrwiNimnrB8,1233507
+build/mettagrid/actions/noop.cpp,sha256=WDCzGjmi7VnBFO6Z4aix6YQYBiq8eDX_BCFoqvUTNcc,1226691
+build/mettagrid/actions/rotate.cpp,sha256=O6QSCSn5POQxYP-73d0f7iWj6vkeKJIK0-FbBYYtmY0,1227751
+build/mettagrid/actions/shield.cpp,sha256=kT7DNr4LqUNhNC5UnNvIOYhXhmG8EW6prVMhAyNIM84,1228950
+build/mettagrid/actions/use.cpp,sha256=KRV2sE00e4m-rPOgfhfe8hkZijL-WG18TiJ_o16tK8I,1247620
+build/mettagrid/mettagrid.cpp,sha256=fBtYYDaJbdZp6uKO_gNWxdAIA2etMiKAIZT87cKc1Wg,1464344
+build/mettagrid/objects.cpp,sha256=H0FZERGFMfDZuTbU43Y_YvcpgAqhMJRBihreXBNC0jI,1395699
+mettagrid/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+mettagrid/actions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+mettagrid/actions/actions.cpython-311-darwin.so,sha256=ZSQxWXe3SlkA_QYOeBCzvrj4IivyWvDwjYEE_Q2osAE,218336
+mettagrid/actions/actions.pxd,sha256=R7F4wL8T8EMcmsNgUvUE_YHhSCQoC2RwzDDzmT7tSXE,765
+mettagrid/actions/actions.pyx,sha256=EUOWA7UZ78e0IwRyvIWVqxyVLp1C6RHAuB-w2SGgDzU,2182
+mettagrid/actions/attack.cpython-311-darwin.so,sha256=LdsKoO8hdXrmtYXrV3ax_rnVljTQg5HEUM_bvH9EHB8,217696
+mettagrid/actions/attack.pxd,sha256=2VQADMYLj7-EhzVoK1_4w_rJFTnj7l7-5fMHseBqGP8,135
+mettagrid/actions/attack.pyx,sha256=dFbQkpyemfKAAVf7eMuMTAz9EmrJo4CTKHJU29iy8vw,3090
+mettagrid/actions/gift.cpython-311-darwin.so,sha256=xwvpz-lM9z6_GIxrphR2TNXNEM10Y4kxw5-ZRppfPBs,217064
+mettagrid/actions/gift.pxd,sha256=f4Y8mQtBhBlD0z_auZQrRs3StYO54TV2fUw5WSDFnG4,105
+mettagrid/actions/gift.pyx,sha256=7s-nQPpuMxBvG5rrcReB9eho7_3FxebBUEhb26tXpRM,804
+mettagrid/actions/move.cpython-311-darwin.so,sha256=zLNYBMHescRTGANyqueyI7FoihpFRG2sSuVwCg6OIAg,217064
+mettagrid/actions/move.pxd,sha256=GN_4ieZrI6RoY92igO4PmayRIS9O-zQl1oeOfImneeE,105
+mettagrid/actions/move.pyx,sha256=Ucq05-gUooU4zia6a5C_4qrlirbPQYxTAVmfkuhZbV0,1595
+mettagrid/actions/noop.cpython-311-darwin.so,sha256=YRWcWJ4s_841bPxvmKhb8C_Gx7xHTWY9GK_lnHe6KPA,217064
+mettagrid/actions/noop.pxd,sha256=KiEvVu-SsvTa3WZ6guNCWvFf12tJEb3GnjzmmpZJB-o,105
+mettagrid/actions/noop.pyx,sha256=6uy8lsnMl7k_sNwJfA10eXcYy9BcGOBeVZ2KSLSv9z0,798
+mettagrid/actions/rotate.cpython-311-darwin.so,sha256=bXV286F_B1oGEl1qBB7x-uMaciZ4en_p97Gh5cjat7Y,217152
+mettagrid/actions/rotate.pxd,sha256=z2D17nFORP9lw2eZLZB96KE18sy_d914x0pe854O7MU,107
+mettagrid/actions/rotate.pyx,sha256=nHwovy43-n-LTq5lTm3a2Aa7LQmo5rMHwwoSc5JeWHs,894
+mettagrid/actions/shield.cpython-311-darwin.so,sha256=sB9KwHQANcqozw6RFhzTz1MHsrs869TVa4SL5dPtcHY,217152
+mettagrid/actions/shield.pxd,sha256=9lf9UOWml8koteofGGTTUyLR20dhE8YTFtZZlMAI7hA,107
+mettagrid/actions/shield.pyx,sha256=YX_oLCrP2rF_nQZSslNbAwq8pGLK1BPkWDmiUg6IAyU,949
+mettagrid/actions/use.cpython-311-darwin.so,sha256=OLTTen8vVonsh7Z55SlmNt6wGQp-h6tbOd_NdVxvz88,217224
+mettagrid/actions/use.pxd,sha256=bFx5IlWzX8mZemv1A5tAg6h97Vt6ev9SSJXewqK33CQ,104
+mettagrid/actions/use.pyx,sha256=LbQKaEkNChFE9Y8KDXKfRygWGRLb49xQL4x-19TEyEw,2786
+mettagrid/config/game_builder.py,sha256=f9fS38WaXepiwL9qlWqMBI4aMRjKN11P18Q6mFl2yAM,3360
+mettagrid/config/sample_config.py,sha256=BbRqT5hrPY5VKGIC7bVbEjwd_G59defru573KxvCtEA,822
+mettagrid/mettagrid.pyx,sha256=kobCT-gjBhRjyzy0Lq9ss7M3wCA4JdIw0TFZw2ep8jM,3971
+mettagrid/mettagrid_c.cpython-311-darwin.so,sha256=qKkjft1j7TYrux1XKr29j17gtdsUJavqTy4V3ZqKyuk,277600
+mettagrid/mettagrid_env.py,sha256=hJ2lRkGl2dztLAQlZqgW9xOBYLrz1_KflIvT-BZ6gn8,5192
+mettagrid/objects.cpython-311-darwin.so,sha256=RFWTbLIn5QNIjxCVG6Z7jBYNbsecsXEOsyuxXnHGNk4,257136
+mettagrid/objects.pxd,sha256=wt53DlwO2TqjJrdjcp9JA8b_-LZRzpU40jlSVrUDI7w,6464
+mettagrid/objects.pyx,sha256=ZjaanEUZ9vfc9i5uVrRj-HqYWQkRO2o_j4LW4lZAhgg,2319
+mettagrid/renderer/assets/agent.png,sha256=-Lt-cyIxZMDX_I7vG6w20MInah8CEorE5WGN0rmVP_k,427
+mettagrid/renderer/assets/altar.png,sha256=Nhn_6IEmPuq9t0rDaCvyPCvpdzX3vZB-X1EoMSpvNvc,425
+mettagrid/renderer/assets/arial.ttf,sha256=NcDzVZ2NtWnjbDEJW4pg1EFkPZX1kTneQOI_ragZuDM,275572
+mettagrid/renderer/assets/converter.png,sha256=HT--nGPU05Wdk1JKNP1rmBYttPbOkWKOI6J-VDs3jwk,384
+mettagrid/renderer/assets/generator.png,sha256=_vHlTbCy_NZYzYnkLxdRwF-zp1D2H4Md3yI8-ZX8QGA,376
+mettagrid/renderer/assets/items.png,sha256=zRfEkwGZ-BnzzCqyse6wyWIGTcteVt_eJVRkTlS4XBA,24116
+mettagrid/renderer/assets/monsters.png,sha256=Aw2Ktj-1BtugmjMI9Wnyb4xxAxq6rzD0mgk9IBdkCyM,32763
+mettagrid/renderer/assets/ore-0.png,sha256=_vHlTbCy_NZYzYnkLxdRwF-zp1D2H4Md3yI8-ZX8QGA,376
+mettagrid/renderer/assets/ore-1.png,sha256=HT--nGPU05Wdk1JKNP1rmBYttPbOkWKOI6J-VDs3jwk,384
+mettagrid/renderer/assets/ore-2.png,sha256=Nhn_6IEmPuq9t0rDaCvyPCvpdzX3vZB-X1EoMSpvNvc,425
+mettagrid/renderer/assets/ore-3.png,sha256=-Lt-cyIxZMDX_I7vG6w20MInah8CEorE5WGN0rmVP_k,427
+mettagrid/renderer/assets/ore-4.png,sha256=vySYN37jPtFeTXXmpyJCCalkR0K2RfutIHp8CXgImjg,410
+mettagrid/renderer/assets/ore-5.png,sha256=KSvduI7HnGD4ky3Cq27FwbwHe9pO4-WY1yFoh7mAzCM,410
+mettagrid/renderer/assets/ore-6.png,sha256=6GWdbQJ4P5WLsTz8rWpGtXbc6P7JJ-bO7ze3qHQ4bH0,418
+mettagrid/renderer/assets/puffer_chars.png,sha256=6FaTd_z6JkMldwlYAW94BajgaaGz30ScWBTmUggxu78,71193
+mettagrid/renderer/assets/wall.png,sha256=MONt_xbscHfODf9T9hiy8poHX_TkMXbk_dC6T0ora7I,1155
+mettagrid/renderer/json_renderer.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+mettagrid/renderer/raylib_renderer.py,sha256=MaCT7R_9IG8zPcQkWjdO0-2QFFguTwt60o6iwanx79M,14977
+mettagrid/renderer/render_code_example.py,sha256=eQJyOK9j-hBiIfQnmyPRSw1n9sDZ2dIQNGzV3m_j_WY,2350
+setup.py,sha256=KBVLqjWKFh_5AiG8GNbn7oXBSGA2cCXQqc_nTStNnPY,2527
+test_perf.py,sha256=GiLNt7eyM3NNyXdx2DZ9CI0LrwRAS_cKU8aV3bt7yiA,3221
+mettagrid-0.0.3.dist-info/LICENSE,sha256=_0ZygdDOImEUQjXap6LgAetwhuvlOfOqcMDmNlAdjLo,1070
+mettagrid-0.0.3.dist-info/METADATA,sha256=8pFkU72N4n3B6qUXA-Hq4YwXrIy8OSihPOrYQvzoBoc,5437
+mettagrid-0.0.3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+mettagrid-0.0.3.dist-info/RECORD,,

setup.py

@@ -0,0 +1,91 @@
+from sympy import li
+from setuptools import Extension, setup, find_packages, Command
+import subprocess
+from Cython.Build import cythonize
+import numpy
+import os
+
+
+def build_ext(srcs, module_name=None):
+    if module_name is None:
+        module_name = srcs[0].replace("/", ".").replace(".pyx", "").replace(".cpp", "")
+    return Extension(
+        module_name,
+        srcs,
+        # include_dirs=["third_party/puffergrid"],
+        define_macros=[('NPY_NO_DEPRECATED_API', 'NPY_1_7_API_VERSION')],
+    )
+
+ext_modules = [
+    build_ext(["mettagrid/objects.pyx"]),
+    build_ext(["mettagrid/actions/actions.pyx"]),
+    build_ext(["mettagrid/actions/attack.pyx"]),
+    build_ext(["mettagrid/actions/gift.pyx"]),
+    build_ext(["mettagrid/actions/move.pyx"]),
+    build_ext(["mettagrid/actions/noop.pyx"]),
+    build_ext(["mettagrid/actions/rotate.pyx"]),
+    build_ext(["mettagrid/actions/shield.pyx"]),
+    build_ext(["mettagrid/actions/use.pyx"]),
+    build_ext(["mettagrid/mettagrid.pyx"], "mettagrid.mettagrid_c"),
+]
+
+debug = os.getenv('DEBUG', '0') == '1'
+annotate = os.getenv('ANNOTATE', '0') == '1'
+
+build_dir = 'build'
+if debug:
+    build_dir = 'build_debug'
+
+os.makedirs(build_dir, exist_ok=True)
+
+setup(
+    name='metta',
+    version='0.1',
+    packages=find_packages(),
+    install_requires=[
+        "hydra-core",
+        "jmespath",
+        "matplotlib",
+        # "numpy==2.0.0",
+        "pettingzoo",
+	    "pufferlib",
+        "puffergrid",
+        "pynvml",
+        "pytest",
+        "PyYAML",
+        "raylib",
+        "rich",
+        "scipy",
+        "tabulate",
+        "tensordict",
+        "torchrl",
+    ],
+    entry_points={
+        'console_scripts': [
+            # If you want to create any executable scripts in your package
+            # For example: 'script_name = module:function'
+        ]
+    },
+    include_dirs=[numpy.get_include()],
+    ext_modules=cythonize(
+        ext_modules,
+        build_dir='build',
+        compiler_directives={
+            "language_level": "3",
+            "embedsignature": debug,
+            "annotation_typing": debug,
+            "cdivision": debug,
+            "boundscheck": debug,
+            "wraparound": debug,
+            "initializedcheck": debug,
+            "nonecheck": debug,
+            "overflowcheck": debug,
+            "overflowcheck.fold": debug,
+            "profile": debug,
+            "linetrace": debug,
+            "c_string_encoding": "utf-8",
+            "c_string_type": "str",
+        },
+        annotate=debug or annotate,
+    ),
+)

test_perf.py

@@ -0,0 +1,107 @@
+import time
+
+import hydra
+import numpy as np
+from tqdm import tqdm
+import pandas as pd
+
+global actions
+global env
+
+def test_performance(env, actions, duration):
+    tick = 0
+    num_actions = actions.shape[0]
+    start = time.time()
+    with tqdm(total=duration, desc="Running performance test") as pbar:
+        while time.time() - start < duration:
+            atns = actions[tick % num_actions]
+            obs, rewards, terminated, truncated, infos = env.step(atns)
+            tick += 1
+            if tick % 100 == 0:
+                pbar.update(time.time() - start - pbar.n)
+
+    print_stats(env._c_env.get_episode_stats())
+    print(f'SPS: {atns.shape[0] * tick / (time.time() - start):.2f}')
+
+actions = {}
+env = {}
+@hydra.main(version_base=None, config_path="configs", config_name="a20_40x40")
+def main(cfg):
+    # Run with c profile
+    from cProfile import run
+    global env
+
+    cfg.env.game.max_steps = 999999999
+    env = hydra.utils.instantiate(cfg.env, render_mode="human")
+    env.reset()
+    global actions
+    num_agents = cfg.env.game.num_agents
+    actions = np.random.randint(0, env.action_space.nvec, (1024, num_agents, 2), dtype=np.uint32)
+
+    env._c_env.render()
+    test_performance(env, actions, 5)
+    # env._c_env.render()
+    exit(0)
+
+    run("test_performance(env, actions, 10)", 'stats.profile')
+    import pstats
+    from pstats import SortKey
+    p = pstats.Stats('stats.profile')
+    p.sort_stats(SortKey.TIME).print_stats(25)
+    exit(0)
+
+
+def print_stats(stats):
+    # Extract game_stats
+    game_stats = stats["game"]
+
+    # Extract agent_stats
+    agent_stats = stats["agent"]
+
+    # Calculate total, average, min, and max for each agent stat
+    total_agent_stats = {}
+    avg_agent_stats = {}
+    min_agent_stats = {}
+    max_agent_stats = {}
+    num_agents = len(agent_stats)
+
+    for agent_stat in agent_stats:
+        for key, value in agent_stat.items():
+            if key not in total_agent_stats:
+                total_agent_stats[key] = 0
+                min_agent_stats[key] = float('inf')
+                max_agent_stats[key] = float('-inf')
+            total_agent_stats[key] += value
+            if value < min_agent_stats[key]:
+                min_agent_stats[key] = value
+            if value > max_agent_stats[key]:
+                max_agent_stats[key] = value
+
+    for key, value in total_agent_stats.items():
+        avg_agent_stats[key] = value / num_agents
+
+    # Sort the keys alphabetically
+    sorted_keys = sorted(total_agent_stats.keys())
+
+    # Create DataFrame for game_stats
+    game_stats_df = pd.DataFrame(sorted(game_stats.items()), columns=["Stat", "Value"])
+
+    # Create DataFrame for agent stats
+    agent_stats_df = pd.DataFrame({
+        "Stat": sorted_keys,
+        "Total": [total_agent_stats[key] for key in sorted_keys],
+        "Average": [avg_agent_stats[key] for key in sorted_keys],
+        "Min": [min_agent_stats[key] for key in sorted_keys],
+        "Max": [max_agent_stats[key] for key in sorted_keys]
+    })
+
+    # Print the DataFrames
+    print("\nGame Stats:")
+    print(game_stats_df.to_string(index=False))
+
+    print("\nAgent Stats:")
+    print(agent_stats_df.to_string(index=False))
+
+
+if __name__ == "__main__":
+    main()

mettagrid/renderer/raylib_client.py

@@ -1,180 +0,0 @@
-from pdb import set_trace as T
-import numpy as np
-import os
-
-import pettingzoo
-import gymnasium
-
-import pufferlib
-from pufferlib.environments.ocean import render
-
-class MettaRaylibClient:
-    def __init__(self, width, height, tile_size=32):
-        self.width = width
-        self.height = height
-        self.tile_size = tile_size
-
-
-        sprite_sheet_path = os.path.join(
-            *self.__module__.split('.')[:-1], './puffer_chars.png')
-        self.asset_map = {
-            1: (0, 0, 128, 128),
-            3: (128, 0, 128, 128),
-            4: (256, 0, 128, 128),
-            # 5: (384, 0, 128, 128),
-            5: (512, 0, 128, 128), #star
-        }
-
-        from raylib import rl, colors
-        rl.InitWindow(width*tile_size, height*tile_size,
-            "PufferLib Ray Grid".encode())
-        rl.SetTargetFPS(10)
-        self.puffer = rl.LoadTexture(sprite_sheet_path.encode())
-        self.rl = rl
-        self.colors = colors
-
-        import pyray as ray
-        camera = ray.Camera2D()
-        camera.target = ray.Vector2(0.0, 0.0)
-        camera.rotation = 0.0
-        camera.zoom = 1.0
-        self.camera = camera
-
-        from cffi import FFI
-        self.ffi = FFI()
-
-    def _cdata_to_numpy(self):
-        image = self.rl.LoadImageFromScreen()
-        width, height, channels = image.width, image.height, 4
-        cdata = self.ffi.buffer(image.data, width*height*channels)
-        return np.frombuffer(cdata, dtype=np.uint8
-            ).reshape((height, width, channels))[:, :, :3]
-
-    def render(self, grid):
-        rl = self.rl
-        colors = self.colors
-        ay, ax = None, None
-
-        ts = self.tile_size
-
-        pos = rl.GetMousePosition()
-        raw_mouse_x = pos.x
-        raw_mouse_y = pos.y
-        mouse_x = int(raw_mouse_x // ts)
-        mouse_y = int(raw_mouse_y // ts)
-        ay = int(np.clip((pos.y - ts*self.height//2) / 50, -3, 3)) + 3
-        ax = int(np.clip((pos.x - ts*self.width//2) / 50, -3, 3)) + 3
-
-        if rl.IsKeyDown(rl.KEY_ESCAPE):
-            exit(0)
-
-        action_id = 0
-        action_arg = 0
-
-        if rl.IsKeyDown(rl.KEY_E):
-            action_id = 0
-            action_arg = 0
-        elif rl.IsKeyDown(rl.KEY_Q):
-            action_id = 0
-            action_arg = 1
-
-        elif rl.IsKeyDown(rl.KEY_W):
-            action_id = 1
-            action_arg = 0
-        elif rl.IsKeyDown(rl.KEY_S):
-            action_id = 1
-            action_arg = 1
-        elif rl.IsKeyDown(rl.KEY_A):
-            action_id = 1
-            action_arg = 2
-        elif rl.IsKeyDown(rl.KEY_R):
-            action_id = 1
-            action_arg = 3
-
-        # if rl.IsKeyDown(rl.KEY_LEFT_SHIFT):
-        #     target_heros = 2
-
-        action = (action_id, action_arg)
-
-        rl.BeginDrawing()
-        rl.BeginMode2D(self.camera)
-        rl.ClearBackground([6, 24, 24, 255])
-        for y in range(self.height):
-            for x in range(self.width):
-                tile = grid[y, x]
-                tx = x*ts
-                ty = y*ts
-                if tile == 0:
-                    continue
-                elif tile == 2:
-                    # Wall
-                    rl.DrawRectangle(x*ts, y*ts, ts, ts, [0, 0, 0, 255])
-                    continue
-                else:
-                    # Player
-                    source_rect = self.asset_map[tile]
-                    dest_rect = (tx, ty, ts, ts)
-                    rl.DrawTexturePro(self.puffer, source_rect, dest_rect,
-                        (0, 0), 0, colors.WHITE)
-
-        # Draw circle at mouse x, y
-        rl.DrawCircle(ts*mouse_x + ts//2, ts*mouse_y + ts//8, ts//8, [255, 0, 0, 255])
-
-        rl.EndMode2D()
-
-        # Draw HUD
-        # player = entities[0]
-        # hud_y = self.height*ts - 2*ts
-        # draw_bars(rl, player, 2*ts, hud_y, 10*ts, 24, draw_text=True)
-
-        # off_color = [255, 255, 255, 255]
-        # on_color = [0, 255, 0, 255]
-
-        # q_color = on_color if skill_q else off_color
-        # w_color = on_color if skill_w else off_color
-        # e_color = on_color if skill_e else off_color
-
-        # q_cd = player.q_timer
-        # w_cd = player.w_timer
-        # e_cd = player.e_timer
-
-        # rl.DrawText(f'Q: {q_cd}'.encode(), 13*ts, hud_y - 20, 40, q_color)
-        # rl.DrawText(f'W: {w_cd}'.encode(), 17*ts, hud_y - 20, 40, w_color)
-        # rl.DrawText(f'E: {e_cd}'.encode(), 21*ts, hud_y - 20, 40, e_color)
-        # rl.DrawText(f'Stun: {player.stun_timer}'.encode(), 25*ts, hud_y - 20, 20, e_color)
-        # rl.DrawText(f'Move: {player.move_timer}'.encode(), 25*ts, hud_y, 20, e_color)
-
-        rl.EndDrawing()
-        return self._cdata_to_numpy(), action
-
-def draw_bars(rl, entity, x, y, width, height=4, draw_text=False):
-    health_bar = entity.health / entity.max_health
-    mana_bar = entity.mana / entity.max_mana
-    if entity.max_health == 0:
-        health_bar = 2
-    if entity.max_mana == 0:
-        mana_bar = 2
-    rl.DrawRectangle(x, y, width, height, [255, 0, 0, 255])
-    rl.DrawRectangle(x, y, int(width*health_bar), height, [0, 255, 0, 255])
-
-    if entity.entity_type == 0:
-        rl.DrawRectangle(x, y - height - 2, width, height, [255, 0, 0, 255])
-        rl.DrawRectangle(x, y - height - 2, int(width*mana_bar), height, [0, 255, 255, 255])
-
-    if draw_text:
-        health = int(entity.health)
-        mana = int(entity.mana)
-        max_health = int(entity.max_health)
-        max_mana = int(entity.max_mana)
-        rl.DrawText(f'Health: {health}/{max_health}'.encode(),
-            x+8, y+2, 20, [255, 255, 255, 255])
-        rl.DrawText(f'Mana: {mana}/{max_mana}'.encode(),
-            x+8, y+2 - height - 2, 20, [255, 255, 255, 255])
-
-        #rl.DrawRectangle(x, y - 2*height - 4, int(width*mana_bar), height, [255, 255, 0, 255])
-        rl.DrawText(f'Experience: {entity.xp}'.encode(),
-            x+8, y - 2*height - 4, 20, [255, 255, 255, 255])
-
-    elif entity.entity_type == 0:
-        rl.DrawText(f'Level: {entity.level}'.encode(),
-            x+4, y -2*height - 12, 12, [255, 255, 255, 255])

mettagrid-0.0.1.dist-info/METADATA

@@ -1,23 +0,0 @@
-Metadata-Version: 2.1
-Name: mettagrid
-Version: 0.0.1
-Summary: A fast grid-based open-ended MARL environment
-Home-page: https://daveey.github.io
-License: MIT
-Keywords: puffergrid,gridworld,minigrid,rl,reinforcement-learning,environment,gym
-Author: David Bloomin
-Author-email: daveey@gmail.com
-Requires-Python: >=3.10,<4.0
-Classifier: License :: OSI Approved :: MIT License
-Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.10
-Classifier: Programming Language :: Python :: 3.11
-Classifier: Programming Language :: Python :: 3.12
-Requires-Dist: cython (>=3.0.11,<4.0.0)
-Requires-Dist: numpy (>=1.21.0,<2.0.0)
-Project-URL: Repository, https://github.com/Metta-AI/mettagrid
-Description-Content-Type: text/markdown
-
-# mettagrid
-A fast grid-based open-ended MARL environment
-

mettagrid-0.0.1.dist-info/RECORD

@@ -1,38 +0,0 @@
-mettagrid/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-mettagrid/actions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-mettagrid/actions/actions.pxd,sha256=Zd9CpuIEWytp1z-KCcuAUoev7dgR16kgea92p8yodps,765
-mettagrid/actions/actions.pyx,sha256=wRg4-3lANoUAIWo6cIAB33AOiPNFg8-EfdCwn-riGP0,1622
-mettagrid/actions/attack.pxd,sha256=2VQADMYLj7-EhzVoK1_4w_rJFTnj7l7-5fMHseBqGP8,135
-mettagrid/actions/attack.pyx,sha256=dnYTMqNEDa5QX6Btz2shPzW3DBB6Y91Uehr_IbfNQP8,2561
-mettagrid/actions/gift.pxd,sha256=f4Y8mQtBhBlD0z_auZQrRs3StYO54TV2fUw5WSDFnG4,105
-mettagrid/actions/gift.pyx,sha256=YXRCP3dYATIWVYDroRjVip1wGyi4ZQx1f9Jfr9k6uZQ,748
-mettagrid/actions/move.pxd,sha256=GN_4ieZrI6RoY92igO4PmayRIS9O-zQl1oeOfImneeE,105
-mettagrid/actions/move.pyx,sha256=8onrsk0jplTOe6OykYw0OdQM9aWzffVyv3pw1Wb2HCg,1206
-mettagrid/actions/rotate.pxd,sha256=z2D17nFORP9lw2eZLZB96KE18sy_d914x0pe854O7MU,107
-mettagrid/actions/rotate.pyx,sha256=fUaWcLnw0s_utXGfmPWg-SzUDw27ma_tsUso23fpayQ,891
-mettagrid/actions/shield.pxd,sha256=9lf9UOWml8koteofGGTTUyLR20dhE8YTFtZZlMAI7hA,107
-mettagrid/actions/shield.pyx,sha256=D4h-vsbVNbG87n5eiIYs2BsXIKQIWgZe3geKEd7j-ZA,834
-mettagrid/actions/use.pxd,sha256=bFx5IlWzX8mZemv1A5tAg6h97Vt6ev9SSJXewqK33CQ,104
-mettagrid/actions/use.pyx,sha256=tHuNx4JdTf3-50_0S_A4Walgo2PZnRaEBuy9CTYA5YY,2623
-mettagrid/config/game_builder.py,sha256=f9fS38WaXepiwL9qlWqMBI4aMRjKN11P18Q6mFl2yAM,3360
-mettagrid/config/sample_config.py,sha256=BbRqT5hrPY5VKGIC7bVbEjwd_G59defru573KxvCtEA,822
-mettagrid/mettagrid.pyx,sha256=jgzXf6cO7IjHsDn2d_0Za9rgn_SngQNGmOwk7FzMyYE,2667
-mettagrid/mettagrid_env.py,sha256=NrP8ZuX75JDO-UMa_BOdNKvdz-lqPogEbICRO30Ru4Q,6677
-mettagrid/objects.pxd,sha256=_71dLM1W9K5WtwMZj35XpzzjDWZkMCiZep75EOAoXbc,5842
-mettagrid/objects.pyx,sha256=CTfc4CQ_NeXkO0LjH_OY8mC4FAluyThKfzn0wzmIHvQ,2197
-mettagrid/renderer/assets/ore-0.png,sha256=_vHlTbCy_NZYzYnkLxdRwF-zp1D2H4Md3yI8-ZX8QGA,376
-mettagrid/renderer/assets/ore-1.png,sha256=HT--nGPU05Wdk1JKNP1rmBYttPbOkWKOI6J-VDs3jwk,384
-mettagrid/renderer/assets/ore-2.png,sha256=Nhn_6IEmPuq9t0rDaCvyPCvpdzX3vZB-X1EoMSpvNvc,425
-mettagrid/renderer/assets/ore-3.png,sha256=-Lt-cyIxZMDX_I7vG6w20MInah8CEorE5WGN0rmVP_k,427
-mettagrid/renderer/assets/ore-4.png,sha256=vySYN37jPtFeTXXmpyJCCalkR0K2RfutIHp8CXgImjg,410
-mettagrid/renderer/assets/ore-5.png,sha256=KSvduI7HnGD4ky3Cq27FwbwHe9pO4-WY1yFoh7mAzCM,410
-mettagrid/renderer/assets/ore-6.png,sha256=6GWdbQJ4P5WLsTz8rWpGtXbc6P7JJ-bO7ze3qHQ4bH0,418
-mettagrid/renderer/assets/puffer_chars.png,sha256=6FaTd_z6JkMldwlYAW94BajgaaGz30ScWBTmUggxu78,71193
-mettagrid/renderer/assets/tiny_galaxy_items.png,sha256=zRfEkwGZ-BnzzCqyse6wyWIGTcteVt_eJVRkTlS4XBA,24116
-mettagrid/renderer/assets/tiny_galaxy_monsters.png,sha256=Aw2Ktj-1BtugmjMI9Wnyb4xxAxq6rzD0mgk9IBdkCyM,32763
-mettagrid/renderer/assets/wall1-0.png,sha256=MONt_xbscHfODf9T9hiy8poHX_TkMXbk_dC6T0ora7I,1155
-mettagrid/renderer/raylib_client.py,sha256=LVZVwHUFlFzDBty2FSWHe5RmR0FPFm-MqA49yTgx81M,6011
-mettagrid-0.0.1.dist-info/LICENSE,sha256=_0ZygdDOImEUQjXap6LgAetwhuvlOfOqcMDmNlAdjLo,1070
-mettagrid-0.0.1.dist-info/METADATA,sha256=ya2_hp-_kRW07IY0CXD5gvIiyK-l5OpzoaCwGIjJR5Q,815
-mettagrid-0.0.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-mettagrid-0.0.1.dist-info/RECORD,,