polis_cli 0.1.0

Command line runner for Polis Genesis
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95

//! Interactive console dashboard that displays a live ASCII map and telemetry.

use polis_core::{World, WorldConfig};
use polis_mind::speaker::{RuleSpeaker, Speaker};
use std::thread::sleep;
use std::time::Duration;

pub fn run(config: WorldConfig, ticks: u64, delay_ms: u64) {
    let mut world = World::new(config);
    let mut voice = RuleSpeaker::new();
    let width = 36;
    let height = 18;

    println!("\x1B[2J\x1B[H");
    for _ in 0..ticks {
        world.step();

        // Build grid
        let mut grid = vec![vec!['.'; width]; height];

        // Draw outer river on eastern border
        for row in &mut grid {
            row[width - 1] = '~';
            row[width - 2] = '~';
        }

        // Draw district fountains
        for dy in 0..4 {
            for dx in 0..4 {
                let fx = (dx * width / 4) + (width / 8);
                let fy = (dy * height / 4) + (height / 8);
                if fy < height && fx < width {
                    grid[fy][fx] = 'o';
                }
            }
        }

        // Draw living agents
        for agent in world.agents.iter().filter(|a| a.is_alive()) {
            let gx =
                ((agent.pos.0 as usize * (width - 3)) / (world.width as usize)).clamp(0, width - 3);
            let gy =
                ((agent.pos.1 as usize * height) / (world.height as usize)).clamp(0, height - 1);
            let ch = match agent.action {
                polis_core::Action::Eat => 'E',
                polis_core::Action::Drink => 'D',
                polis_core::Action::Rest => 'Z',
                polis_core::Action::Work => 'W',
                polis_core::Action::Socialize => 'S',
                polis_core::Action::Wander => '*',
            };
            grid[gy][gx] = ch;
        }

        let s = world.stats();
        let hour = (world.tick % 24) as u32;
        let day = (world.tick / 24) as u32;

        print!("\x1B[H");
        println!("============================================================");
        println!("  POLIS GENESIS: Autonomous Multi Agent Society Simulator   ");
        println!("============================================================");
        println!(
            " Day {:<3} Hour {:02}:00 | Pop: {:<3} Alive | Treasury: {:<5} Coins",
            day, hour, s.alive, s.treasury
        );
        println!(
            " Jobs: {:<3} | Pairs: {:<3} | Crimes: {:<3} | Gini: {:.2} | Starved: {}",
            s.employed, s.couples, s.crimes, s.gini, s.starved
        );
        println!("------------------------------------------------------------");

        for row in &grid {
            let line: String = row.iter().collect();
            println!("  |{line}|");
        }

        println!("------------------------------------------------------------");
        println!(" Actions: [E]at  [D]rink  [Z]Rest  [W]ork  [S]ocialize  [*]Roam");

        if let Some(u) = world.utterances.back() {
            let speaker = &world.agents[u.speaker as usize].name;
            let listener = &world.agents[u.listener as usize].name;
            let words = voice.say(&world, u);
            println!(" Talk: {speaker} to {listener}: \"{words}\"");
        } else {
            println!(" Talk: (The town is quiet)");
        }
        println!("============================================================");

        if delay_ms > 0 {
            sleep(Duration::from_millis(delay_ms));
        }
    }
}