Function main

tools/pack.rs:44–105

()

}

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 mut geometries: Vec<_> = read_geometries(reader).collect();
    let padding = Coord {
        x: args.padding,
        y: args.padding,
    };

    let mut rects = GroupedRectsToPlace::<usize, ()>::new();
    for (idx, geometry) in geometries.iter().enumerate() {
        if let Some(mut bbox) = geometry.bounding_rect() {
            // Apply a little bit of padding around each bounding box. This is not just to generate
            // a more appealing result; it's so that POINTs don't get "packed" all on top of each
            // other because they have 0x0 bounding boxes.
            let min = bbox.min();
            let new_min = min - padding;
            bbox.set_min(new_min);

            let max = bbox.max();
            let new_max = max + padding;
            bbox.set_max(new_max);
            let rect =
                RectToInsert::new(bbox.width().ceil() as u32, bbox.height().ceil() as u32, 1);
            rects.push_rect(idx, None, rect);
        }
    }

    let mut bins = BTreeMap::new();
    bins.insert(0, TargetBin::new(args.width, args.height, 1));

    let packing = pack_rects(&rects, &mut bins, &volume_heuristic, &contains_smallest_box)
        .map_err(|e| eyre::eyre!("Failed to pack rectangles: {e:?}"))?;
    let packing = packing.packed_locations();
    for (idx, (_bin, location)) in packing.iter() {
        let geometry = &mut geometries[*idx];
        let bbox = geometry.bounding_rect().ok_or_else(|| {
            eyre::eyre!("Geometry at index {idx} has no bounding box, cannot pack")
        })?;
        let source = bbox.center();
        let target = Coord {
            x: location.x() as f64,
            y: location.y() as f64,
        };
        let offset = target - source;
        geometry.translate_mut(offset.x, offset.y);
    }