elseware/drops/src/generic_unit.rs

use std::collections::BTreeMap;
use serde::{Serialize, Deserialize};
use rand::Rng;
use rand::seq::IteratorRandom;

use entity::character::SectionID;
use entity::item::unit::{UnitType, Unit, UnitModifier};
use maps::room::{Difficulty, Episode};
use maps::area::MapArea;
use crate::{ItemDropType, load_data_file};
use stats::items::{unit_stats, UnitStats};



#[derive(Debug, Serialize, Deserialize)]
struct UnitLevels {
    area1: u32,
    area2: u32,
    area3: u32,
    area4: u32,
    area5: u32,
    area6: u32,
    area7: u32,
    area8: u32,
    area9: u32,
    area10: u32,
}

impl UnitLevels {
    fn level_by_area(&self, map_area: &MapArea) -> u32 {
        match map_area.drop_area_value().unwrap() {
            0 => self.area1,
            1 => self.area2,
            2 => self.area3,
            3 => self.area1,
            4 => self.area1,
            5 => self.area6,
            6 => self.area7,
            7 => self.area8,
            8 => self.area9,
            9 => self.area10,
            _ => panic!()
        }
    }
}

pub struct GenericUnitTable {
    unit_levels: UnitLevels,
    unit_stats: BTreeMap<UnitType, UnitStats>,
}



impl GenericUnitTable {
    pub fn new(episode: Episode, difficulty: Difficulty, section_id: SectionID) -> GenericUnitTable {
        GenericUnitTable {
            unit_levels: load_data_file(episode, difficulty, section_id, "unit_rate.toml"),
            unit_stats: unit_stats(),
        }
    }

    fn unit_type_and_modifier<R: Rng>(&self, area_map: &MapArea, rng: &mut R) -> Option<(UnitType, Option<UnitModifier>)> {
        let level = self.unit_levels.level_by_area(area_map) as i32;
        if level == 0 {
            return None;
        }
        let units = self.unit_stats
            .iter()
            .filter(|(_, stats)| {
                stats.stars < 9
            })
            .filter_map(|(utype, stats)| {
                match level - stats.stars as i32 {
                    -1 if stats.modifier != 0 => Some(vec![(*utype, Some(UnitModifier::Minus)), (*utype, Some(UnitModifier::MinusMinus))]),
                    0  => Some(vec![(*utype, None)]),
                    1 if stats.modifier != 0 => Some(vec![(*utype, Some(UnitModifier::Plus)), (*utype, Some(UnitModifier::PlusPlus))]),
                    _ => None,
                }
            })
            .flatten();

        Some(units.choose(rng).unwrap())
    }

    pub fn get_drop<R: Rng>(&self, area_map: &MapArea, rng: &mut R) -> Option<ItemDropType> {
        let unit_type_modifier = self.unit_type_and_modifier(area_map, rng);

        unit_type_modifier.map(|(unit_type, unit_modifier)| {
            ItemDropType::Unit(Unit {
                unit: unit_type,
                modifier: unit_modifier,
            })
        })
    }
}


#[cfg(test)]
mod test {
    use super::*;
    use rand::{SeedableRng};
    #[test]
    fn test_unit_drops() {
        let mut rng = rand_chacha::ChaCha20Rng::from_seed([23;32]);

        let gut = GenericUnitTable::new(Episode::One, Difficulty::Normal, SectionID::Skyly);
        assert!(gut.get_drop(&MapArea::Forest1, &mut rng) == None);

        let gut = GenericUnitTable::new(Episode::One, Difficulty::Hard, SectionID::Skyly);

        let unit_tests = vec![(MapArea::Forest1, UnitType::ResistFreeze, Some(UnitModifier::PlusPlus)),
                              (MapArea::Caves3, UnitType::GeneralTp, None),
                              (MapArea::Mines2, UnitType::ResistEvil, Some(UnitModifier::PlusPlus)),
                              (MapArea::DarkFalz, UnitType::DragonHp, Some(UnitModifier::Minus))];
        for (area, unit, umod) in unit_tests {
            assert!(gut.get_drop(&area, &mut rng) == Some(ItemDropType::Unit(Unit {
                unit: unit,
                modifier: umod,
            })));
        }
    }
}