cardiac-geometriesx 0.3.1__py3-none-any.whl → 0.4.0__py3-none-any.whl

cardiac_geometries/fibers/utils.py

@@ -1,3 +1,4 @@
+import json
 from pathlib import Path
 from typing import NamedTuple, Sequence
 
@@ -19,16 +20,21 @@ class Microstructure(NamedTuple):
 def save_microstructure(
     mesh: dolfinx.mesh.Mesh, functions: Sequence[dolfinx.fem.Function], outdir: str | Path
 ) -> None:
-    from ..utils import element2array
+    from ..utils import create_xdmf_pointcloud, element2array
 
+    if len(functions) == 0:
+        return
     # Save for paraview visualization
-    try:
-        with dolfinx.io.VTXWriter(
-            mesh.comm, Path(outdir) / "microstructure-viz.bp", functions, engine="BP4"
-        ) as file:
-            file.write(0.0)
-    except RuntimeError:
-        pass
+    if functions[0].function_space.ufl_element().family_name == "quadrature":
+        create_xdmf_pointcloud(us=functions, filename=Path(outdir) / "microstructure-viz.xdmf")
+    else:
+        try:
+            with dolfinx.io.VTXWriter(
+                mesh.comm, Path(outdir) / "microstructure-viz.bp", functions, engine="BP4"
+            ) as file:
+                file.write(0.0)
+        except RuntimeError:
+            pass
 
     # Save with proper function space
     filename = Path(outdir) / "microstructure.bp"
@@ -43,6 +49,16 @@ def save_microstructure(
         attributes=attributes,
     )
 
+    def json_serial(obj):
+        if isinstance(obj, (np.ndarray)):
+            return obj.tolist()
+        raise TypeError("Type %s not serializable" % type(obj))
+
+    if mesh.comm.rank == 0:
+        (Path(outdir) / "microstructure.json").write_text(
+            json.dumps(attributes, indent=4, default=json_serial)
+        )
+
 
 def normalize(u):
     return u / np.linalg.norm(u, axis=0)

cardiac_geometries/geometry.py

@@ -147,13 +147,22 @@ class Geometry:
         else:
             markers = {}
 
+        if (folder / "microstructure.json").exists():
+            if comm.rank == 0:
+                microstructure = json.loads((folder / "microstructure.json").read_text())
+            else:
+                microstructure = {}
+            microstructure = comm.bcast(microstructure, root=0)
+        else:
+            microstructure = {}
+
         functions = {}
         microstructure_path = folder / "microstructure.bp"
         if microstructure_path.exists():
-            function_space = adios4dolfinx.read_attributes(
-                comm=MPI.COMM_WORLD, filename=microstructure_path, name="function_space"
-            )
-            for name, el in function_space.items():
+            # function_space = adios4dolfinx.read_attributes(
+            #     comm=MPI.COMM_WORLD, filename=microstructure_path, name="function_space"
+            # )
+            for name, el in microstructure.items():
                 element = utils.array2element(el)
                 V = dolfinx.fem.functionspace(mesh, element)
                 f = dolfinx.fem.Function(V, name=name)

cardiac_geometries/utils.py

@@ -1,5 +1,6 @@
 import tempfile
 import typing
+import xml.etree.ElementTree as ET
 from enum import Enum
 from pathlib import Path
 from typing import Iterable, NamedTuple
@@ -25,6 +26,23 @@ class GMshModel(NamedTuple):
     vertex_tags: dolfinx.mesh.MeshTags
 
 
+def distribute_entity_data(
+    mesh: dolfinx.mesh.Mesh,
+    tdim: int,
+    marked_entities: np.ndarray,
+    entity_values: np.ndarray,
+) -> tuple[np.ndarray, np.ndarray]:
+    if dolfinx.__version__ >= "0.9.0":
+        local_entities, local_values = dolfinx.io.utils.distribute_entity_data(
+            mesh, tdim, marked_entities, entity_values
+        )
+    else:
+        local_entities, local_values = dolfinx.io.utils.distribute_entity_data(
+            mesh._cpp_object, tdim, marked_entities, entity_values
+        )
+    return local_entities, local_values
+
+
 # copied from https://github.com/FEniCS/dolfinx/blob/main/python/dolfinx/io/gmshio.py
 def model_to_mesh(
     model,
@@ -160,9 +178,10 @@ def model_to_mesh(
     )
 
     # Create MeshTags for cells
-    local_entities, local_values = dolfinx.io.utils.distribute_entity_data(
-        mesh._cpp_object, mesh.topology.dim, cells, cell_values
+    local_entities, local_values = distribute_entity_data(
+        mesh, mesh.topology.dim, cells, cell_values
     )
+
     mesh.topology.create_connectivity(mesh.topology.dim, 0)
     adj = dolfinx.cpp.graph.AdjacencyList_int32(local_entities)
     ct = dolfinx.mesh.meshtags_from_entities(
@@ -188,11 +207,13 @@ def model_to_mesh(
         gmsh_facet_perm = dolfinx.io.gmshio.cell_perm_array(facet_type, num_facet_nodes)
         marked_facets = marked_facets[:, gmsh_facet_perm]
 
-        local_entities, local_values = dolfinx.io.utils.distribute_entity_data(
-            mesh._cpp_object, tdim - 1, marked_facets, facet_values
+        local_entities, local_values = distribute_entity_data(
+            mesh, mesh.topology.dim - 1, marked_facets, facet_values
         )
+
         mesh.topology.create_connectivity(topology.dim - 1, tdim)
         adj = dolfinx.cpp.graph.AdjacencyList_int32(local_entities)
+
         ft = dolfinx.io.gmshio.meshtags_from_entities(
             mesh, tdim - 1, adj, local_values.astype(np.int32, copy=False)
         )
@@ -210,8 +231,8 @@ def model_to_mesh(
         gmsh_edge_perm = dolfinx.io.gmshio.cell_perm_array(edge_type, num_edge_nodes)
         marked_edges = marked_edges[:, gmsh_edge_perm]
 
-        local_entities, local_values = dolfinx.io.utils.distribute_entity_data(
-            mesh._cpp_object, tdim - 2, marked_edges, edge_values
+        local_entities, local_values = distribute_entity_data(
+            mesh, tdim - 2, marked_edges, edge_values
         )
         mesh.topology.create_connectivity(topology.dim - 2, tdim)
         adj = dolfinx.cpp.graph.AdjacencyList_int32(local_entities)
@@ -232,8 +253,8 @@ def model_to_mesh(
         gmsh_vertex_perm = dolfinx.io.gmshio.cell_perm_array(vertex_type, num_vertex_nodes)
         marked_vertices = marked_vertices[:, gmsh_vertex_perm]
 
-        local_entities, local_values = dolfinx.io.utils.distribute_entity_data(
-            mesh._cpp_object, tdim - 3, marked_vertices, vertex_values
+        local_entities, local_values = distribute_entity_data(
+            mesh, tdim - 3, marked_vertices, vertex_values
         )
         mesh.topology.create_connectivity(topology.dim - 3, tdim)
         adj = dolfinx.cpp.graph.AdjacencyList_int32(local_entities)
@@ -340,6 +361,7 @@ def array2element(arr: np.ndarray) -> basix.finite_element.FiniteElement:
             degree=degree,
             discontinuous=discontinuous,
             shape=(3,),
+            lagrange_variant=basix.LagrangeVariant.unset,
         )
 
 
@@ -438,3 +460,88 @@ def gmsh2dolfin(comm: MPI.Intracomm, msh_file, rank: int = 0) -> GMshGeometry:
         )
 
     return GMshGeometry(mesh, ct, ft, et, vt, markers)
+
+
+def create_xdmf_pointcloud(filename: Path, us: typing.Sequence[dolfinx.fem.Function]) -> None:
+    # Adopted from https://gist.github.com/jorgensd/8bae61ad7a0c211570dff0116a68a356
+    if len(us) == 0:
+        return
+
+    import adios2
+
+    u = us[0]
+    points = u.function_space.tabulate_dof_coordinates()
+    h5name = filename.with_suffix(".h5")
+
+    bs = u.function_space.dofmap.index_map_bs
+    comm = u.function_space.mesh.comm
+    num_dofs_global = u.function_space.dofmap.index_map.size_global
+    num_dofs_local = u.function_space.dofmap.index_map.size_local
+    local_range = np.array(u.function_space.dofmap.index_map.local_range, dtype=np.int64)
+
+    # Write XDMF on rank 0
+    if comm.rank == 0:
+        xdmf = ET.Element("XDMF")
+        xdmf.attrib["Version"] = "3.0"
+        xdmf.attrib["xmlns:xi"] = "http://www.w3.org/2001/XInclude"
+        domain = ET.SubElement(xdmf, "Domain")
+        grid = ET.SubElement(domain, "Grid")
+        grid.attrib["GridType"] = "Uniform"
+        grid.attrib["Name"] = "Point Cloud"
+        topology = ET.SubElement(grid, "Topology")
+        topology.attrib["NumberOfElements"] = str(num_dofs_global)
+        topology.attrib["TopologyType"] = "PolyVertex"
+        topology.attrib["NodesPerElement"] = "1"
+        geometry = ET.SubElement(grid, "Geometry")
+        geometry.attrib["GeometryType"] = "XY" if points.shape[1] == 2 else "XYZ"
+        for u in us:
+            it0 = ET.SubElement(geometry, "DataItem")
+            it0.attrib["Dimensions"] = f"{num_dofs_global} {points.shape[1]}"
+            it0.attrib["Format"] = "HDF"
+            it0.text = f"{h5name.name}:/Step0/Points"
+            attrib = ET.SubElement(grid, "Attribute")
+            attrib.attrib["Name"] = u.name
+            if bs == 1:
+                attrib.attrib["AttributeType"] = "Scalar"
+            else:
+                attrib.attrib["AttributeType"] = "Vector"
+            attrib.attrib["Center"] = "Node"
+            it1 = ET.SubElement(attrib, "DataItem")
+            it1.attrib["Dimensions"] = f"{num_dofs_global} {bs}"
+            it1.attrib["Format"] = "HDF"
+            it1.text = f"{h5name.name}:/Step0/Values_{u.name}"
+        text = [
+            '<?xml version="1.0"?>\n<!DOCTYPE Xdmf SYSTEM "Xdmf.dtd" []>\n',
+            ET.tostring(xdmf, encoding="unicode"),
+        ]
+        filename.write_text("".join(text))
+    # Create ADIOS2 reader
+    # start = time.perf_counter()
+    adios = adios2.ADIOS(comm)
+    io = adios.DeclareIO("Point cloud writer")
+    io.SetEngine("HDF5")
+    outfile = io.Open(h5name.as_posix(), adios2.Mode.Write)
+    points_out = points[:num_dofs_local, :]
+
+    pointvar = io.DefineVariable(
+        "Points",
+        points_out,
+        shape=[num_dofs_global, points.shape[1]],
+        start=[local_range[0], 0],
+        count=[num_dofs_local, points.shape[1]],
+    )
+    outfile.Put(pointvar, points_out)
+    for u in us:
+        data = u.x.array[: num_dofs_local * bs].reshape(-1, bs)
+
+        valuevar = io.DefineVariable(
+            f"Values_{u.name}",
+            data,
+            shape=[num_dofs_global, bs],
+            start=[local_range[0], 0],
+            count=[num_dofs_local, bs],
+        )
+        outfile.Put(valuevar, data)
+    outfile.PerformPuts()
+    outfile.Close()
+    assert adios.RemoveIO("Point cloud writer")

{cardiac_geometriesx-0.3.1.dist-info → cardiac_geometriesx-0.4.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: cardiac-geometriesx
-Version: 0.3.1
+Version: 0.4.0
 Summary: A python library for cardiac geometries
 Author-email: Henrik Finsberg <henriknf@simula.no>
 License: MIT

{cardiac_geometriesx-0.3.1.dist-info → cardiac_geometriesx-0.4.0.dist-info}/RECORD

@@ -1,16 +1,16 @@
 cardiac_geometries/__init__.py,sha256=2W_ywAeLjyRk5MqSPAodHa4UN2lOnW1h8tmGLQ3gaJ0,150
 cardiac_geometries/cli.py,sha256=GKcsMNhVFoxWDr-o3jiDCXmTvPgCSEB9Kws8AvYIK4Q,19678
-cardiac_geometries/geometry.py,sha256=iK8lydZEqU3sz0983yucW9IzQ2VqSXSy4D5ubdl9bOs,5636
+cardiac_geometries/geometry.py,sha256=s-1BIDKvP9M3uu3Sl8JpGRnTnXZAa9Aq3FfpIIEAxhY,5982
 cardiac_geometries/gui.py,sha256=9WYR850wLrqsUrVUC37E2SaO0OWA_oagSe-YNrsxz3k,8376
 cardiac_geometries/mesh.py,sha256=Ejciny48mnKKN20RrY9DwLY7_Lh1jb6Kldj4CW1wr1o,26909
-cardiac_geometries/utils.py,sha256=EwEXAuB4XiJaqlZaGZ1cc6sIBSPJLCx-43As1NUcJrA,16488
+cardiac_geometries/utils.py,sha256=fMHlVzwxnk1lJd8XQPy8dgFB9C9ecSnJB2NzMXJGAAQ,20348
 cardiac_geometries/fibers/__init__.py,sha256=WpRrn9Iakl-3m8IGtFkqP0LXGjw5EZHZ8Eg9JCnCdrg,137
 cardiac_geometries/fibers/lv_ellipsoid.py,sha256=KuWnx9yZ5KDyoeoYyDnXLZYb82DRiNveZyNIboD3uJ8,5768
 cardiac_geometries/fibers/slab.py,sha256=5tMvOSqXQ4_nbdUUtho_tQjDrmICxZfN7SXNq4FKdlY,3836
-cardiac_geometries/fibers/utils.py,sha256=j1ERqXcdaWCwO2yjbYGlgTt2G1DxZCVzbNHuDhh_QrM,2541
-cardiac_geometriesx-0.3.1.dist-info/LICENSE,sha256=lo5K2rJPZOSv6luutGHbzzi3IpXNaB9E2UWq60qvNx0,1111
-cardiac_geometriesx-0.3.1.dist-info/METADATA,sha256=maN2Wb_hCkroXXzBdl-_UX0BxQW9jvNAQkqb38YAb40,4163
-cardiac_geometriesx-0.3.1.dist-info/WHEEL,sha256=cVxcB9AmuTcXqmwrtPhNK88dr7IR_b6qagTj0UvIEbY,91
-cardiac_geometriesx-0.3.1.dist-info/entry_points.txt,sha256=xOBnlc6W-H9oCDYLNz3kpki26OmpfYSoFSrmi_4V-Ec,52
-cardiac_geometriesx-0.3.1.dist-info/top_level.txt,sha256=J0gQxkWR2my5Vf7Qt8buDY8ZOjYdVfIweVunCGXWKNE,19
-cardiac_geometriesx-0.3.1.dist-info/RECORD,,
+cardiac_geometries/fibers/utils.py,sha256=x6MNtW6xN5NBGBD0Ubm4eqTDde0j_Vd-nh8agFT_AiU,3161
+cardiac_geometriesx-0.4.0.dist-info/LICENSE,sha256=lo5K2rJPZOSv6luutGHbzzi3IpXNaB9E2UWq60qvNx0,1111
+cardiac_geometriesx-0.4.0.dist-info/METADATA,sha256=rPU1WjGfdSGrlkgnkhIubCY3WzJ7nlMGsLfcuNdU2ME,4163
+cardiac_geometriesx-0.4.0.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
+cardiac_geometriesx-0.4.0.dist-info/entry_points.txt,sha256=xOBnlc6W-H9oCDYLNz3kpki26OmpfYSoFSrmi_4V-Ec,52
+cardiac_geometriesx-0.4.0.dist-info/top_level.txt,sha256=J0gQxkWR2my5Vf7Qt8buDY8ZOjYdVfIweVunCGXWKNE,19
+cardiac_geometriesx-0.4.0.dist-info/RECORD,,

{cardiac_geometriesx-0.3.1.dist-info → cardiac_geometriesx-0.4.0.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (74.1.2)
+Generator: setuptools (75.1.0)
 Root-Is-Purelib: true
 Tag: py3-none-any