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

//! Automated scenario conversions.
//!
//! This module implements conversions between different scenario formats and game versions.
mod aoc_to_wk;
mod hd_to_wk;

use crate::{Scenario, ScenarioObject};

pub use aoc_to_wk::AoCToWK;
pub use hd_to_wk::HDToWK;

/// Error indicating scenario conversion failure.
#[derive(Debug, thiserror::Error)]
pub enum ConvertError {
    /// The input scenario version is not supported by the converter.
    #[error("invalid version")]
    InvalidVersion,
}

/// Convert an AoC or HD Edition scenario file to a WololoKingdoms one.
///
/// It will auto-detect the version of the file, and output a WK compatible scenario.
/// AoC scenarios will have their unit and terrain IDs switched around so they have the correct
/// look in WK.
/// HD Edition scenarios will have all the new unit and terrain IDs mapped to WK IDs.
///
/// ## Usage
///
/// ```rust,ignore
/// use genie_scx::convert::AutoToWK;
/// AutoToWK::default().convert(&mut scenario)?
/// ```
pub struct AutoToWK {}

impl Default for AutoToWK {
    fn default() -> Self {
        AutoToWK {}
    }
}

/// Check if a scenario object is likely a WololoKingdoms one.
fn is_wk_object(object: &ScenarioObject) -> bool {
    // Stormy Dog is the highest ID in AoC 1.0c.
    const STORMY_DOG: i32 = 862;
    object.id > STORMY_DOG
}

impl AutoToWK {
    pub fn convert(&self, scen: &mut Scenario) -> Result<(), ConvertError> {
        if scen.version().is_hd_edition() {
            HDToWK::default().convert(scen)
        } else if scen.version().is_aok() || scen.version().is_aoc() {
            if scen.objects().any(is_wk_object) {
                // No conversion necessary.
                Ok(())
            } else {
                AoCToWK::default().convert(scen)
            }
        } else {
            Err(ConvertError::InvalidVersion)
        }
    }
}