Function main

scripts/svg_to_mesh.py:186–229

()

    return mesh;

def main():
    args = parse_args();
    logger = get_logger(args.log);
    logger.info("Triangulation engine: {}".format(args.engine));

    wires = pymesh.wires.WireNetwork.create_from_file(args.input_svg);
    wires = drop_zero_dim(wires);

    if wires.num_vertices == 0:
        logger.warn("Input is empty");

    if args.with_frame and args.engine != "triwild":
        wires = add_frame(wires);
    if args.resolve_self_intersection:
        wires = resolve_self_intersection(wires, logger);
    if args.with_cleanup:
        wires = cleanup(wires, logger);
    if args.with_features:
        json_file = "{}.json".format(os.path.splitext(args.input_svg)[0]);
        assert(os.path.exists(json_file));
    else:
        json_file = None;

    basename = os.path.splitext(args.output_mesh)[0];
    wire_file = basename + ".wire";
    wires.write_to_file(wire_file);

    if args.with_triangulation:
        if os.path.splitext(args.input_svg)[1] != ".svg" and args.engine == "triwild":
            # Avoid data loss from conversion.
            assert(not args.with_cleanup);
            assert(not args.resolve_self_intersection);
            with open(args.input_svg, 'r') as fin:
                data = fin.read();
            with open(wire_file, 'w') as fout:
                fout.write(data);
        mesh = triangulate(wires, args.engine, args.stage, args.epsilon, logger, wire_file, json_file);

        if mesh.num_vertices > 0 and args.with_cell_label:
            compute_cell_labels(wires, mesh, logger);
            mesh = solve_heat_equation(mesh);
            pymesh.save_mesh(args.output_mesh, mesh, "cell", "sq_dist");
        else:
            pymesh.save_mesh(args.output_mesh, mesh);

if __name__ == "__main__":
    main();