// -----------------------------------------------------------------------------
//                                    Tweak
// -----------------------------------------------------------------------------

// Input parameters
pub const CORPUS_FILE_NAME: &str = "corpus.txt";
pub const STATS_FILE_NAME: &str = "shai.json";
pub const LOAD_STATS: bool = true;

pub const STARTING_LAYOUT_STRING: &str = "
A B C D E  F G H I J
K L M N O  P Q R S T
U V W X Y  Z . , / '
";

// Tournament parameters
pub const NUM_CONTESTANTS: usize = 256;
pub const NUM_PERSIST: usize = 8;
pub const MAX_NUM_TRANSPOSITIONS: usize = 15;

// Fitness parameters
pub const STANDARD_DSFB_WEIGHT: f32 = 1.0 / 3.0;
pub const INDEX_DSFB_WEIGHT: f32 = 1.0 / 3.0;

pub const MIN_INDEX_USAGE_PERCENT: f32 = 0.10;
pub fn index_usage_fitness(index_perc: f32) -> f32 {
    if index_perc < MIN_INDEX_USAGE_PERCENT {
        MIN_INDEX_USAGE_PERCENT - index_perc
    } else {
        0.0
    }
}

pub const MAX_PINKY_USAGE_PERCENT: f32 = 0.085;
pub fn pinky_col_penalty(lesser_perc: f32, greater_perc: f32) -> f32 {
    if greater_perc > MAX_PINKY_USAGE_PERCENT {
        if lesser_perc > MAX_PINKY_USAGE_PERCENT {
            lesser_perc + greater_perc - MAX_PINKY_USAGE_PERCENT * 2.0
        } else {
            greater_perc - MAX_PINKY_USAGE_PERCENT
        }
    } else {
        0.0
    }
}

// -----------------------------------------------------------------------------
//     Do not change the following, the code makes assumptions about these
// -----------------------------------------------------------------------------
pub const NUM_KEYS: usize = 30;
pub const ROW_LENGTH: usize = 10;

pub const KEY_CHARS: [char; NUM_KEYS] = [
    'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J', 'K', 'L', 'M', 'N', 'O', 'P', 'Q', 'R', 'S',
    'T', 'U', 'V', 'W', 'X', 'Y', 'Z', '/', '.', ',', '\'',
];

pub const KEY_TO_FINGER: [usize; NUM_KEYS] = [
    0, 1, 2, 3, 3, 4, 4, 5, 6, 7, 0, 1, 2, 3, 3, 4, 4, 5, 6, 7, 0, 1, 2, 3, 3, 4, 4, 5, 6, 7,
];

pub fn canonicalise(inp: char) -> Option<u8> {
    if inp.is_ascii_alphabetic() {
        return Some(inp.to_ascii_uppercase() as u8);
    } else {
        match inp {
            '.' => Some('.' as u8),
            '>' => Some('.' as u8),
            ',' => Some(',' as u8),
            '<' => Some(',' as u8),
            '/' => Some('/' as u8),
            '?' => Some('/' as u8),
            '\'' => Some('\'' as u8),
            '"' => Some('\'' as u8),
            _ => None,
        }
    }
}