use contour_isobands::ContourBuilder;
use geojson::{FeatureCollection, GeoJson};
use std::fs::File;
use std::io::{BufWriter, Write};

fn main() {
    // First example, from a grid of points
    // on which were computed the population potentials in Metropolitan France
    let pot_pop_fr = include_str!("../tests/fixtures/pot_pop_fr.json");
    let raw_data: serde_json::Value = serde_json::from_str(pot_pop_fr).unwrap();
    let matrix: Vec<f64> = raw_data["data"]
        .as_array()
        .unwrap()
        .iter()
        .map(|x| x.as_f64().unwrap())
        .collect();
    let h = raw_data["height"].as_u64().unwrap() as usize;
    let w = raw_data["width"].as_u64().unwrap() as usize;

    let x_origin = -6.144721171428571;
    let y_origin = 51.78171334283718;
    let x_step = 0.11875873095057177;
    let y_step = -0.08993203637245273;

    let contours = ContourBuilder::new(w, h)
        .x_origin(x_origin)
        .y_origin(y_origin)
        .x_step(x_step)
        .y_step(y_step)
        .contours(
            &matrix,
            &[
                0.001, 105483.25, 310000., 527416.25, 850000., 1054832.5, 2109665., 3164497.5,
                4219330., 5274162.5, 6328995., 7383827.5, 8438660., 9704459., 10548326.,
            ],
        )
        .unwrap();

    let features = contours
        .iter()
        .map(|band| band.to_geojson())
        .collect::<Vec<geojson::Feature>>();

    let geojson_str = GeoJson::from(FeatureCollection {
        bbox: None,
        features,
        foreign_members: None,
    })
    .to_string();

    let mut file_writer = BufWriter::new(File::create("/tmp/example-pot_pop_fr.geojson").unwrap());
    file_writer.write(&geojson_str.as_bytes()).unwrap();

    // Second example from the volcano dataset
    let volcano = include_str!("../tests/fixtures/volcano.json");
    let raw_data: serde_json::Value = serde_json::from_str(volcano).unwrap();
    let matrix: Vec<f64> = raw_data["data"]
        .as_array()
        .unwrap()
        .iter()
        .map(|x| x.as_f64().unwrap())
        .collect();
    let h = raw_data["height"].as_u64().unwrap() as usize;
    let w = raw_data["width"].as_u64().unwrap() as usize;

    let contours = ContourBuilder::new(w, h)
        .contours(
            &matrix,
            &[
                90., 95., 100., 105., 110., 115., 120., 125., 130., 135., 140., 145., 150., 155.,
                160., 165., 170., 175., 180., 185., 190., 195., 200.,
            ],
        )
        .unwrap();

    let features = contours
        .iter()
        .map(|band| band.to_geojson())
        .collect::<Vec<geojson::Feature>>();

    let geojson_str = GeoJson::from(FeatureCollection {
        bbox: None,
        features,
        foreign_members: None,
    })
    .to_string();

    let mut file_writer = BufWriter::new(File::create("/tmp/example-volcano.geojson").unwrap());
    file_writer.write(&geojson_str.as_bytes()).unwrap();
}