Function main

scripts/add_element_attribute.py:73–86

()

    return parser.parse_args();

def main():
    args = parse_args();
    mesh = pymesh.load_mesh(args.input_mesh);

    formula = Formula(mesh);
    values = formula.eval_formula(args.formula);
    if isinstance(values, list):
        values = np.array(values, dtype=float);
    elif isinstance(values, (int, float)):
        values = np.ones(mesh.num_elements, dtype=float) * values;
    mesh.add_attribute(args.name);
    mesh.set_attribute(args.name, values);

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

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