Function main

tools/geom2graph.rs:66–117

()

}

fn main() -> eyre::Result<()> {
    color_eyre::install()?;
    let args = CmdlineOptions::parse();

    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 strategy = match args.snap_strategy {
        CliSnappingStrategy::ClosestPoint => {
            SnappingStrategy::ClosestPoint(args.tolerance.unwrap_or_default())
        }
        CliSnappingStrategy::RegularGrid => {
            SnappingStrategy::RegularGrid(args.tolerance.unwrap_or_default())
        }
    };

    if args.geom2graph || !args.graph2geom {
        let geometries = read_geometries(reader);
        let graph = node::<_, petgraph::Undirected>(geometries);

        let graph = if args.tolerance.is_some() {
            snap_graph(graph, strategy)
        } else {
            graph
        };

        write_graph(writer, &graph, &args.graph_format)
    } else {
        let graph: GeometryGraph<petgraph::Undirected> = read_tgf_graph(reader);
        let (polygons, dangles) = polygonize(&graph);
        let polygons = polygons.into_iter().map(Geometry::Polygon);
        let dangles = dangles.into_iter().map(Geometry::LineString);
        let geometries = polygons.chain(dangles);

        let geometries = if args.tolerance.is_some() {
            snap_geoms(geometries, strategy)
        } else {
            Box::new(geometries)
        };

        write_geometries(writer, geometries)
    }
}