use crate::{LogNormal, create_state};
use crate::standard_distributions::standard_normal;

impl LogNormal {
    /// コンストラクタ
    /// * `_seed_1` - 乱数の種
    /// * `_seed_2` - 乱数の種。`_seed_1`と同じ値の場合、コンストラクタ側で変更する。
    pub fn new(_seed_1: u32, _seed_2: u32) -> Self {
        let _seed_other = if _seed_1 != _seed_2 { _seed_2 } else { (_seed_1 as u64 + 1192u64) as u32};
        let xyzuv0: (u32, u32, u32, u32, u32) = create_state(_seed_1);
        let xyzuv1: (u32, u32, u32, u32, u32) = create_state(_seed_other);
        Self {
            x0: xyzuv0.0, y0: xyzuv0.1, z0: xyzuv0.2, u0: xyzuv0.3, v0: xyzuv0.4,
            x1: xyzuv1.0, y1: xyzuv1.1, z1: xyzuv1.2, u1: xyzuv1.3, v1: xyzuv1.4,
            mean: 0f64,
            std: 1f64,
        }
    }

    /// 対数正規分布に従う乱数を返す
    pub fn sample(&mut self) -> f64 {
        (standard_normal(&mut self.x0, &mut self.y0, &mut self.z0, &mut self.u0, &mut self.v0,
            &mut self.x1, &mut self.y1, &mut self.z1, &mut self.u1, &mut self.v1) * self.std + self.mean).exp()
    }

    /// 確率変数のパラメータを変更する
    /// * `mean` - 平均
    /// * `variance` - 分散
    pub fn try_set_params(&mut self, mean: f64, variance: f64) -> Result<(f64, f64), &str> {
        if variance <= 0f64 {
            Err("分散が0以下です。確率変数のパラメータは前回の設定を維持します。")
        }
        else {
            self.mean = mean;
            self.std = variance.sqrt();
            Ok( (mean, variance) )
        }
    }
}

#[macro_export]
/// 対数正規分布
/// * `() =>` - 乱数の種は自動生成
/// * `($seed_1: expr, $seed_2: expr) =>` - 乱数の種を指定する
/// # 使用例 1
/// ```
/// let mut log_normal = rand_simple::create_log_normal!(1192u32, 765u32);
/// println!("平均値 0, 分散 1 の標準対数正規分布乱数を生成する -> {}", log_normal.sample());
/// ```
/// # 使用例 2
/// ```
/// let mut log_normal = rand_simple::create_log_normal!();
/// println!("平均値 0, 分散 1 の標準対数正規分布乱数を生成する -> {}", log_normal.sample());
/// ```
macro_rules! create_log_normal {
    () => {{
        let seeds: (u32, u32) = $crate::create_seeds();
        $crate::LogNormal::new(seeds.0, seeds.1)
    }};
    ($seed_1: expr, $seed_2: expr) => {
        $crate::LogNormal::new($seed_1 as u32, $seed_2 as u32)
    };
}


impl std::fmt::Display for LogNormal {
    /// println!マクロなどで表示するためのフォーマッタ
    /// * 構造体の型
    /// * 平均
    /// * 分散
    fn fmt(&self, f: &mut std::fmt::Formatter) -> std::fmt::Result {
        writeln!(f, "構造体の型: {}", std::any::type_name::<Self>())?;
        writeln!(f, "平均: {}", self.mean)?;
        writeln!(f, "分散: {}", self.std.powi(2))?;
        Ok(())
    }
}