parabellum 0.2.16

Parabellum environment for parallel warfare simulation

parabellum

Parabellum is an ultra-scalable, high-performance warfare simulation engine. It is based on JaxMARL's SMAX environment, but has been heavily modified to support a wide range of new features and improvements.

Installation

Parabellum is written in Python 3.11 and can be installed using pip:

pip install parabellum

Usage

Parabellum is designed to be used in conjunction with JAX, a high-performance numerical computing library. Here is a simple example of how to use Parabellum to simulate a game with 10 agents and 10 enemies, each taking random actions:

import parabellum as pb
from jax import random

# define the scenario
kwargs = dict(obstacle_coords=[(7, 7)], obstacle_deltas=[(10, 0)])
scenario = pb.Scenario(**kwargs)  # <- Scenario is an important part of parabellum

# create the environment
kwargs = dict(map_width=256, map_height=256, num_agents=10, num_enemies=10)
env = pb.Parabellum(**kwargs)  # <- Parabellum is the central class of parabellum

# initiate stochasticity
rng = random.PRNGKey(0)
rng, key = random.split(rng)

# initialize the environment state
obs, state = env.reset(key)
state_sequence = []

for _ in range(1000):

    # manage stochasticity
    rng, rng_act, key_step = random.split(key)
    key_act = random.split(rng_act, len(env.agents))

    # sample actions and append to state sequence
    act = {a: env.action_space(a).sample(k)
        for a, k in zip(env.agents, key_act)}

    # step the environment
    state_sequence.append((key_act, state, act))
    obs, state, reward, done, info = env.step(key_step, act, state)


# save visualization of the state sequence
vis = pb.Visualizer(env, state_sequence)  # <- Visualizer is a nice to have class
vis.animate()

Features

• Obstacles — can be inserted in

TODO

• Parallel pygame vis
  • Parallel bullet renderings
  • Combine parallell plots into one (maybe out of parabellum scope)
• Color for health?
• Add the ability to see ongoing game.
• Bug test friendly fire.
• Start sim from arbitrary state.
• Save when the episode ends in some state/obs variable
• Look for the source of the bug when using more Allies than Enemies
• Y inversed axis for parabellum visualization
• Units see through obstacles?