cardiac-geometriesx 0.4.6__py3-none-any.whl → 0.4.8__py3-none-any.whl

cardiac_geometries/cli.py

@@ -92,6 +92,14 @@ def app():
     help="Function space for fibers of the form family_degree",
     show_default=True,
 )
+@click.option(
+    "--clipped/--no-clipped",
+    default=False,
+    is_flag=True,
+    type=bool,
+    help="If True create clip away the outflow tracts",
+    show_default=True,
+)
 def ukb(
     outdir: Path | str,
     mode: int = -1,
@@ -102,6 +110,7 @@ def ukb(
     fiber_angle_endo: float = 60,
     fiber_angle_epi: float = -60,
     fiber_space: str = "P_1",
+    clipped: bool = False,
 ):
     outdir = Path(outdir)
     outdir.mkdir(exist_ok=True)
@@ -116,6 +125,7 @@ def ukb(
         fiber_angle_endo=fiber_angle_endo,
         fiber_angle_epi=fiber_angle_epi,
         fiber_space=fiber_space,
+        clipped=clipped,
     )
     geo.save(outdir / "ukb.bp")
 
@@ -199,7 +209,6 @@ def ukb(
     "--create-fibers",
     default=False,
     is_flag=True,
-    type=bool,
     help="If True create analytic fibers",
     show_default=True,
 )
@@ -421,7 +430,6 @@ def lv_ellipsoid(
     "--create-fibers",
     default=False,
     is_flag=True,
-    type=bool,
     help="If True create analytic fibers",
     show_default=True,
 )
@@ -689,7 +697,6 @@ def biv_ellipsoid(
     "--create-fibers",
     default=False,
     is_flag=True,
-    type=bool,
     help="If True create analytic fibers",
     show_default=True,
 )
@@ -827,7 +834,6 @@ def biv_ellipsoid_torso(
     "--create-fibers",
     default=False,
     is_flag=True,
-    type=bool,
     help="If True create analytic fibers",
     show_default=True,
 )

cardiac_geometries/geometry.py

@@ -98,7 +98,12 @@ class Geometry:
         self.mesh.comm.barrier()
 
     @classmethod
-    def from_file(cls, comm: MPI.Intracomm, path: str | Path) -> "Geometry":
+    def from_file(
+        cls,
+        comm: MPI.Intracomm,
+        path: str | Path,
+        function_space_data: dict[str, np.ndarray] | None = None,
+    ) -> "Geometry":
         path = Path(path)
 
         mesh = adios4dolfinx.read_mesh(comm=comm, filename=path)
@@ -118,10 +123,12 @@ class Geometry:
                 tags[name] = None
 
         functions = {}
-        function_space = adios4dolfinx.read_attributes(
-            comm=comm, filename=path, name="function_space"
-        )
-        for name, el in function_space.items():
+        if function_space_data is None:
+            function_space_data = adios4dolfinx.read_attributes(
+                comm=comm, filename=path, name="function_space"
+            )
+        assert isinstance(function_space_data, dict), "function_space_data must be a dictionary"
+        for name, el in function_space_data.items():
             element = utils.array2element(el)
             V = dolfinx.fem.functionspace(mesh, element)
             f = dolfinx.fem.Function(V, name=name)

cardiac_geometries/mesh.py

@@ -32,6 +32,7 @@ def ukb(
     fiber_angle_endo: float = 60,
     fiber_angle_epi: float = -60,
     fiber_space: str = "P_1",
+    clipped: bool = False,
     comm: MPI.Comm = MPI.COMM_WORLD,
 ) -> Geometry:
     try:
@@ -44,26 +45,29 @@ def ukb(
         raise ImportError(msg) from e
 
     if comm.rank == 0:
+        ukb.cli.main(["surf", str(outdir), "--mode", str(mode), "--std", str(std), "--case", case])
+        mesh_args = [
+            "mesh",
+            str(outdir),
+            "--case",
+            case,
+            "--char_length_max",
+            str(char_length_max),
+            "--char_length_min",
+            str(char_length_min),
+        ]
+        if clipped:
+            ukb.cli.main(["clip", str(outdir), "--case", case, "--smooth"])
+            mesh_args.append("--clipped")
         print(comm.rank)
-        ukb.cli.main(
-            [
-                str(outdir),
-                "--mode",
-                str(mode),
-                "--std",
-                str(std),
-                "--case",
-                case,
-                "--mesh",
-                "--char_length_max",
-                str(char_length_max),
-                "--char_length_min",
-                str(char_length_min),
-            ]
-        )
+
+        ukb.cli.main(mesh_args)
     comm.barrier()
     outdir = Path(outdir)
-    mesh_name = outdir / f"{case}.msh"
+    if clipped:
+        mesh_name = outdir / f"{case}_clipped.msh"
+    else:
+        mesh_name = outdir / f"{case}.msh"
 
     geometry = utils.gmsh2dolfin(comm=comm, msh_file=mesh_name)
 
@@ -98,24 +102,25 @@ def ukb(
         )
         raise ImportError(msg)
 
-    # base_marker = 3
-    # indices = []
-    # for k in ["PV", "TV", "AV", "MV"]:
-    #     indices.append(geometry.ffun.find(geometry.markers[k][0]))
-    # indices = np.hstack(comm.allreduce(indices, op=MPI.SUM))
-    # values = np.full(len(indices), base_marker, dtype=np.int32)
-
-    markers = {
-        "lv": [geometry.markers["LV"][0]],
-        "rv": [geometry.markers["RV"][0]],
-        "epi": [geometry.markers["EPI"][0]],
-        "base": [
-            geometry.markers["PV"][0],
-            geometry.markers["TV"][0],
-            geometry.markers["AV"][0],
-            geometry.markers["MV"][0],
-        ],
-    }
+    if clipped:
+        markers = {
+            "lv": [geometry.markers["LV"][0]],
+            "rv": [geometry.markers["RV"][0]],
+            "epi": [geometry.markers["EPI"][0]],
+            "base": [geometry.markers["BASE"][0]],
+        }
+    else:
+        markers = {
+            "lv": [geometry.markers["LV"][0]],
+            "rv": [geometry.markers["RV"][0]],
+            "epi": [geometry.markers["EPI"][0]],
+            "base": [
+                geometry.markers["PV"][0],
+                geometry.markers["TV"][0],
+                geometry.markers["AV"][0],
+                geometry.markers["MV"][0],
+            ],
+        }
     system = ldrb.dolfinx_ldrb(
         mesh=geometry.mesh,
         ffun=geometry.ffun,

{cardiac_geometriesx-0.4.6.dist-info → cardiac_geometriesx-0.4.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
-Metadata-Version: 2.2
+Metadata-Version: 2.4
 Name: cardiac-geometriesx
-Version: 0.4.6
+Version: 0.4.8
 Summary: A python library for cardiac geometries
 Author-email: Henrik Finsberg <henriknf@simula.no>
 License: MIT
@@ -38,7 +38,7 @@ Provides-Extra: test
 Requires-Dist: pre-commit; extra == "test"
 Requires-Dist: pytest; extra == "test"
 Requires-Dist: pytest-cov; extra == "test"
-Requires-Dist: ukb-atlas; extra == "test"
+Requires-Dist: ukb-atlas[pyvista]; extra == "test"
 Requires-Dist: fenicsx-ldrb; extra == "test"
 Provides-Extra: gui
 Requires-Dist: streamlit; extra == "gui"
@@ -47,6 +47,7 @@ Requires-Dist: pyvista[all]>=0.43.0; extra == "gui"
 Requires-Dist: trame-vuetify; extra == "gui"
 Requires-Dist: ipywidgets; extra == "gui"
 Requires-Dist: fenicsx-ldrb; extra == "gui"
+Dynamic: license-file
 
 ![_](docs/_static/logo.png)
 

cardiac_geometriesx-0.4.8.dist-info/RECORD

@@ -0,0 +1,16 @@
+cardiac_geometries/__init__.py,sha256=2W_ywAeLjyRk5MqSPAodHa4UN2lOnW1h8tmGLQ3gaJ0,150
+cardiac_geometries/cli.py,sha256=SBg2qFC1jgLoWUK5eke6grNyrYOklwi2QwLUgFW7VJs,21947
+cardiac_geometries/geometry.py,sha256=Qdbd2npBe3QrwknV4XWts9HprElGufasX0IhZWEDCx0,7051
+cardiac_geometries/gui.py,sha256=9WYR850wLrqsUrVUC37E2SaO0OWA_oagSe-YNrsxz3k,8376
+cardiac_geometries/mesh.py,sha256=jLHTujz_38SQ5loN94nuN6pMG8IVPmlCdwOJhzuMxy4,30922
+cardiac_geometries/utils.py,sha256=nJE-gyV7NIAmgl5JGlG5Vl7XsIcYS8Y20bVjYcd4R6c,21964
+cardiac_geometries/fibers/__init__.py,sha256=WpRrn9Iakl-3m8IGtFkqP0LXGjw5EZHZ8Eg9JCnCdrg,137
+cardiac_geometries/fibers/lv_ellipsoid.py,sha256=LZNhxzTsn-h88xXGP5bNfnWKhdvBROUjWjt8yrQFW48,6010
+cardiac_geometries/fibers/slab.py,sha256=TYQhckJ8mwXz_08Cx3QsPQMtemkaxZ955SnSMrDfBPE,3898
+cardiac_geometries/fibers/utils.py,sha256=pCoXkaUbidLGPY0Ty8uWCU1siDkHcv2qP8rZtdxMf0k,3475
+cardiac_geometriesx-0.4.8.dist-info/licenses/LICENSE,sha256=lo5K2rJPZOSv6luutGHbzzi3IpXNaB9E2UWq60qvNx0,1111
+cardiac_geometriesx-0.4.8.dist-info/METADATA,sha256=pw7N1Amzyn_xrXkt1o3I1ftA_fwIhazcpm7xy3mKNWA,4320
+cardiac_geometriesx-0.4.8.dist-info/WHEEL,sha256=DnLRTWE75wApRYVsjgc6wsVswC54sMSJhAEd4xhDpBk,91
+cardiac_geometriesx-0.4.8.dist-info/entry_points.txt,sha256=xOBnlc6W-H9oCDYLNz3kpki26OmpfYSoFSrmi_4V-Ec,52
+cardiac_geometriesx-0.4.8.dist-info/top_level.txt,sha256=J0gQxkWR2my5Vf7Qt8buDY8ZOjYdVfIweVunCGXWKNE,19
+cardiac_geometriesx-0.4.8.dist-info/RECORD,,

{cardiac_geometriesx-0.4.6.dist-info → cardiac_geometriesx-0.4.8.dist-info}/WHEEL

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

cardiac_geometriesx-0.4.6.dist-info/RECORD

@@ -1,16 +0,0 @@
-cardiac_geometries/__init__.py,sha256=2W_ywAeLjyRk5MqSPAodHa4UN2lOnW1h8tmGLQ3gaJ0,150
-cardiac_geometries/cli.py,sha256=DvUxq3sXcrFrqJs9gKpNRosgtalASTRS-YAM_71MaxM,21777
-cardiac_geometries/geometry.py,sha256=y-V4DyksNCqPwK7-NQIjcS6ldlvh1ub2p7lj9FZJHWw,6795
-cardiac_geometries/gui.py,sha256=9WYR850wLrqsUrVUC37E2SaO0OWA_oagSe-YNrsxz3k,8376
-cardiac_geometries/mesh.py,sha256=3Uk2U_zSWQ6Ae45w6bEMsvaRM77js37er1QSMa3pLlY,30696
-cardiac_geometries/utils.py,sha256=nJE-gyV7NIAmgl5JGlG5Vl7XsIcYS8Y20bVjYcd4R6c,21964
-cardiac_geometries/fibers/__init__.py,sha256=WpRrn9Iakl-3m8IGtFkqP0LXGjw5EZHZ8Eg9JCnCdrg,137
-cardiac_geometries/fibers/lv_ellipsoid.py,sha256=LZNhxzTsn-h88xXGP5bNfnWKhdvBROUjWjt8yrQFW48,6010
-cardiac_geometries/fibers/slab.py,sha256=TYQhckJ8mwXz_08Cx3QsPQMtemkaxZ955SnSMrDfBPE,3898
-cardiac_geometries/fibers/utils.py,sha256=pCoXkaUbidLGPY0Ty8uWCU1siDkHcv2qP8rZtdxMf0k,3475
-cardiac_geometriesx-0.4.6.dist-info/LICENSE,sha256=lo5K2rJPZOSv6luutGHbzzi3IpXNaB9E2UWq60qvNx0,1111
-cardiac_geometriesx-0.4.6.dist-info/METADATA,sha256=Z3JiJw3b5EfAhIvZBWhYm0XP9lcrNgodWH0y3vWvdMs,4289
-cardiac_geometriesx-0.4.6.dist-info/WHEEL,sha256=jB7zZ3N9hIM9adW7qlTAyycLYW9npaWKLRzaoVcLKcM,91
-cardiac_geometriesx-0.4.6.dist-info/entry_points.txt,sha256=xOBnlc6W-H9oCDYLNz3kpki26OmpfYSoFSrmi_4V-Ec,52
-cardiac_geometriesx-0.4.6.dist-info/top_level.txt,sha256=J0gQxkWR2my5Vf7Qt8buDY8ZOjYdVfIweVunCGXWKNE,19
-cardiac_geometriesx-0.4.6.dist-info/RECORD,,