nil-core 0.7.18

Multiplayer strategy game
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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117

// Copyright (C) Call of Nil contributors
// SPDX-License-Identifier: AGPL-3.0-only

#[cfg(test)]
mod tests;

use crate::continent::index::ContinentKey;
use crate::error::Result;
use crate::resources::diff::ResourcesDiff;
use crate::resources::maintenance::MaintenanceBalance;
use crate::resources::prelude::Maintenance;
use crate::resources::{Food, Resources};
use crate::ruler::Ruler;
use crate::world::World;
use nil_num::mul_ceil::MulCeil;

impl World {
  /// Increases the resources of a ruler, ensuring their storage do not overflow.
  pub(crate) fn add_resources_within_capacity<R>(
    &mut self,
    ruler: R,
    resources: Resources,
  ) -> Result<()>
  where
    R: Into<Ruler>,
  {
    let ruler: Ruler = ruler.into();
    let capacity = self.get_storage_capacity(ruler.clone())?;
    let resources_diff = ResourcesDiff::try_from(resources)?;

    self
      .ruler_mut(&ruler)?
      .resources_mut()
      .add_within_capacity(resources_diff, capacity);

    Ok(())
  }

  pub(crate) fn get_maintenance<R>(&self, ruler: R) -> Result<Maintenance>
  where
    R: Into<Ruler>,
  {
    let ruler: Ruler = ruler.into();
    let stats = self.stats().infrastructure();
    let mut maintenance = self.military().maintenance_of(ruler.clone());

    for city in self.continent().cities_of(ruler) {
      maintenance += city.maintenance(&stats)?;
    }

    Ok(maintenance)
  }

  pub(crate) fn get_maintenance_balance<R>(&self, ruler: R) -> Result<MaintenanceBalance>
  where
    R: Into<Ruler>,
  {
    let ruler: Ruler = ruler.into();
    let stats = self.stats().infrastructure();

    let mut production = Food::new(0);
    let mut maintenance = self.military().maintenance_of(ruler.clone());

    for city in self.continent().cities_of(ruler) {
      maintenance += city.maintenance(&stats)?;
      production += city.round_production(&stats)?.food;
    }

    Ok(MaintenanceBalance { maintenance, production })
  }

  pub fn get_weighted_resources<K>(&self, key: K) -> Result<Resources>
  where
    K: ContinentKey,
  {
    let city = key
      .into_coord(self.continent.size())
      .and_then(|coord| self.city(coord))?;

    let capacity_weight = self.get_storage_capacity_weight(city.coord())?;
    let mut resources = self.ruler(city.owner())?.resources();

    macro_rules! apply {
      ($res:ident, $storage:ident) => {
        let weight = f64::from(capacity_weight.$storage);
        resources.$res = resources.$res.mul_ceil(weight).into();
      };
    }

    apply!(food, silo);
    apply!(iron, warehouse);
    apply!(stone, warehouse);
    apply!(wood, warehouse);

    Ok(resources)
  }

  /// Withdraws up to the requested resources, storing the amount withdrawn in `resources`.
  pub(crate) fn withdraw_resources_up_to<R>(
    &mut self,
    ruler: R,
    resources: &mut Resources,
  ) -> Result<()>
  where
    R: Into<Ruler>,
  {
    let mut ruler = self.ruler_mut(&ruler.into())?;
    let current_resources = ruler.take_resources();
    if let Some(result) = current_resources.checked_sub(*resources) {
      *ruler.resources_mut() = result;
    } else {
      *resources = current_resources;
    }

    Ok(())
  }
}