fn main() {
    use data::Subject;
    use pharmsol::*;

    let subject = Subject::builder("id1")
        .bolus(0.0, 100.0, 0)
        .repeat(2, 0.5)
        .observation(0.5, 0.1, 0)
        .observation(1.0, 0.4, 0)
        .observation(2.0, 1.0, 0)
        .observation(2.5, 1.1, 0)
        .covariate("wt", 0.0, 80.0)
        .covariate("wt", 1.0, 83.0)
        .covariate("age", 0.0, 25.0)
        .build();
    println!("{subject}");
    let ode = equation::ODE::new(
        |x, p, _t, dx, _rateiv, _cov| {
            // fetch_cov!(cov, t,);
            fetch_params!(p, ka, ke, _tlag, _v);
            //Struct
            dx[0] = -ka * x[0];
            dx[1] = ka * x[0] - ke * x[1];
        },
        |p| {
            fetch_params!(p, _ka, _ke, tlag, _v);
            lag! {0=>tlag}
        },
        |_p| fa! {},
        |_p, _t, _cov, _x| {},
        |x, p, _t, _cov, y| {
            fetch_params!(p, _ka, _ke, _tlag, v);
            y[0] = x[1] / v;
        },
        (2, 1),
    );
    let net = equation::ODENet::new(
        vec![
            dmatrix![
                -1.0,0.0;
                1.0,0.0
            ],
            dmatrix![
                0.0,0.0;
                0.0,-1.0
            ],
            dmatrix![
                0.0,0.0;
                0.0,0.0
            ],
            dmatrix![
                0.0,0.0;
                0.0,0.0
            ],
        ],
        vec![],
        vec![],
        vec![Lag::new(0, Op::Equal(P(2)))],
        vec![],
        vec![],
        vec![OutEq::new(0, Op::Div(X(1), P(3)))],
        (2, 1),
    );

    let op = net.estimate_predictions(&subject, &vec![0.3, 0.5, 0.1, 70.0]);
    println!("{:#?}", op.flat_predictions());

    let op = ode.estimate_predictions(&subject, &vec![0.3, 0.5, 0.1, 70.0]);
    println!("{:#?}", op.flat_predictions());
}