Function compute_cell_labels

scripts/svg_to_mesh.py:154–171

(wires, mesh, logger)

    return mesh;

def compute_cell_labels(wires, mesh, logger):
    start_time = time();
    arrangement = pymesh.Arrangement2();
    arrangement.points = wires.vertices;
    arrangement.segments = wires.edges;
    arrangement.run();
    mesh.add_attribute("face_centroid");
    centroids = mesh.get_face_attribute("face_centroid");
    r = arrangement.query(centroids);
    finish_time = time();
    t = finish_time - start_time;
    logger.info("Arrangement running time: {}".format(t));

    cell_type = np.array([item[0] for item in r]);
    cell_ids = np.array([item[1] for item in r]);
    cell_ids[cell_type != pymesh.Arrangement2.ElementType.CELL] = -1;
    mesh.add_attribute("cell");
    mesh.set_attribute("cell", cell_ids);

def solve_heat_equation(mesh):
    cell_ids = mesh.get_attribute("cell").ravel().astype(int);