Function main

scripts/distortion.py:260–284

()

                orientation[i], ]);

def main():
    args = parse_args();
    mesh = pymesh.load_mesh(args.input_mesh, drop_zero_dim=True);
    if mesh.num_voxels > 0 and mesh.vertex_per_voxel != 4:
        raise RuntimeError("Only tet mesh is supported for distortion computation");
    elif mesh.num_voxels == 0 and mesh.num_faces > 0 and mesh.vertex_per_face!=3:
        raise RuntimeError("Only triangle mesh is supported for distortion computation");

    numeric_level = getattr(logging, args.log, None);
    if not isinstance(numeric_level, int):
        raise ValueError('Invalid log level: %s' % loglevel);
    logging.basicConfig(level=numeric_level);

    if mesh.num_voxels > 0:
        compute_distortion_energies_3D(mesh);
        compute_tet_quality_measures(mesh);
        if args.csv != None:
            output_to_csv_3D(mesh, args.csv);
    else:
        compute_distortion_energies_2D(mesh);
        compute_tri_quality_measures(mesh);
        if args.csv != None:
            output_to_csv_2D(mesh, args.csv);

    pymesh.save_mesh(args.output_mesh, mesh, *mesh.attribute_names);

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