s2protocol 3.5.3

A parser for Starcraft II - Replay format, exports to different target formats
Documentation 
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

use crate::game_events::VersionedBalanceUnits;

// Some colors I really liked from a Freya Holmer presentation:
// https://www.youtube.com/watch?v=kfM-yu0iQBk
pub const FREYA_ORANGE: [u8; 4] = [0xeb, 0x79, 0x07, 0xff];
pub const FREYA_GOLD: [u8; 4] = [0xea, 0x9e, 0x36, 0xff];
pub const FREYA_RED: [u8; 4] = [0xf8, 0x10, 0x53, 0xff];
pub const FREYA_BLUE: [u8; 4] = [0x30, 0xb5, 0xf7, 0xff];
pub const FREYA_GREEN: [u8; 4] = [0x0a, 0xeb, 0x9f, 0xff];
pub const FREYA_LIGHT_BLUE: [u8; 4] = [0x72, 0xc5, 0xdd, 0xff];
pub const FREYA_GRAY: [u8; 4] = [0xb2, 0xc5, 0xc5, 0xff];
pub const FREYA_PINK: [u8; 4] = [0xea, 0xa4, 0x83, 0xff];
pub const FREYA_LIGHT_GRAY: [u8; 4] = [0xf4, 0xf5, 0xf8, 0xff];
pub const FREYA_DARK_BLUE: [u8; 4] = [0x4d, 0xa7, 0xc2, 0xff];
pub const FREYA_DARK_GREEN: [u8; 4] = [0x37, 0xbd, 0xa9, 0xff];
pub const FREYA_DARK_RED: [u8; 4] = [0xae, 0x20, 0x44, 0xff];
pub const FREYA_VIOLET: [u8; 4] = [0xa4, 0x01, 0xed, 0xff];
pub const FREYA_WHITE: [u8; 4] = [0xfa, 0xf8, 0xfb, 0xff];
pub const FREYA_YELLOW: [u8; 4] = [0xf7, 0xd4, 0x54, 0xff];
pub const FREYA_LIGHT_YELLOW: [u8; 4] = [0xea, 0xd8, 0xad, 0xff];
pub const FREYA_LIGHT_GREEN: [u8; 4] = [0x6e, 0xc2, 0x9c, 0xff];

// Returns the expected size of units depending on their type
pub fn get_unit_sized_color(
    unit_name: &str,
    user_id: i64,
    balance_units: &VersionedBalanceUnits,
) -> (f32, [u8; 4]) {
    let mut unit_size: f32 = 0.45f32; // Some default value
    if let Some(balance_unit) = balance_units.get(unit_name)
        && let Some(size) = balance_unit.misc.radius
    {
        // Use a color based on the user id:
        unit_size = size;
    };
    let color = match unit_name {
        "VespeneGeyser" => FREYA_LIGHT_GREEN,
        "SpacePlatformGeyser" => FREYA_LIGHT_GREEN,
        "LabMineralField" => FREYA_LIGHT_BLUE,
        "LabMineralField750" => FREYA_LIGHT_BLUE,
        "MineralField" => FREYA_LIGHT_BLUE,
        "MineralField450" => FREYA_LIGHT_BLUE,
        "MineralField750" => FREYA_LIGHT_BLUE,
        "XelNagaTower" => {
            // This should be super transparent
            FREYA_WHITE
        }
        "RichMineralField" => FREYA_GOLD,
        "RichMineralField750" => FREYA_ORANGE,
        "DestructibleRockEx1DiagonalHugeBLUR" => FREYA_GRAY,
        "DestructibleDebris6x6" => FREYA_GRAY,
        "UnbuildablePlatesDestructible" => FREYA_LIGHT_GRAY,
        "Overlord" | "Pylon" | "SupplyDepot" => FREYA_YELLOW,
        "SCV" | "Drone" | "Probe" | "Larva" | "Egg" => FREYA_LIGHT_GRAY,
        "Hatchery" | "CommandCenter" | "Nexus" => FREYA_PINK,
        "Broodling" => FREYA_LIGHT_GRAY,
        "Infestor" => FREYA_VIOLET,
        _ => {
            // Ignore the Beacons for now.
            if !unit_name.starts_with("Beacon") {
                // tracing::warn!("Unknown unit name: '{}'", unit_name);
            }
            // Fallback to user color
            user_color(user_id)
        }
    };
    (unit_size, color)
}

pub fn user_color(user_id: i64) -> [u8; 4] {
    match user_id {
        0 => FREYA_LIGHT_GREEN,
        1 => FREYA_LIGHT_BLUE,
        2 => FREYA_LIGHT_GRAY,
        3 => FREYA_ORANGE,
        _ => FREYA_WHITE,
    }
}