Function main

tools/transform.rs:277–337

()

}

fn main() -> eyre::Result<()> {
    color_eyre::install()?;
    let args = CmdlineOptions::parse();
    if !args.range1.is_empty() && args.range1.len() != 2 {
        eyre::bail!("--range1 must have 2 comma-separated values");
    }
    if !args.range2.is_empty() && args.range2.len() != 2 {
        eyre::bail!("--range2 must have 2 comma-separated values");
    }

    let filter = tracing_subscriber::EnvFilter::builder()
        .with_default_directive(args.log_level.into())
        .from_env_lossy();
    tracing_subscriber::fmt()
        .with_env_filter(filter)
        .with_ansi(true)
        .with_writer(std::io::stderr)
        .init();

    let reader = get_input_reader(&args.input)?;
    let writer = get_output_writer(&args.output)?;
    let geometries = read_geometries(reader);
    let mut transformed = affine_transform(geometries, &args);

    if args.range1.len() == 2 || args.range2.len() == 2 {
        let geometries: Vec<_> = transformed.collect();
        let bounds = bounding_box(&geometries);

        let mut x_dst = None;
        let mut y_dst = None;
        if args.range1.len() == 2 {
            let dst = [args.range1[0], args.range1[1]];
            x_dst = Some(dst);
        }
        if args.range2.len() == 2 {
            // If we're converting from polar, then the "y" coordinate is actually theta.
            // Use degrees in the CLI args, because it's waaaay easier to do "0 360" than it is
            // "0 2PI"
            let dst = if args.from_polar {
                [args.range2[0].to_radians(), args.range2[1].to_radians()]
            } else {
                [args.range2[0], args.range2[1]]
            };
            y_dst = Some(dst);
        }

        let scaled: Vec<_> = geoms_coordwise(geometries.into_iter(), |coord| {
            scale_coord_range(&bounds, x_dst.as_ref(), y_dst.as_ref(), coord)
        })
        .collect();
        transformed = Box::new(scaled.into_iter());
    }

    if args.to_polar {
        transformed = Box::new(geoms_coordwise(transformed, to_polar));
    } else if args.from_polar {
        transformed = Box::new(geoms_coordwise(transformed, from_polar));
    }