From dd965d4da1c67b0f1d4beb8350aad14679954ff4 Mon Sep 17 00:00:00 2001
From: Guillaume Cerutti <guillaume.cerutti@inria.fr>
Date: Thu, 28 Mar 2024 15:43:09 +0100
Subject: [PATCH] add voxelsize argument to python function
---
src/pybind_example/labelled_tetrahedral_mesh.py | 6 ++++--
tests/test_cgal_image.py | 4 +++-
2 files changed, 7 insertions(+), 3 deletions(-)
diff --git a/src/pybind_example/labelled_tetrahedral_mesh.py b/src/pybind_example/labelled_tetrahedral_mesh.py
index ce7b8b7..4699c0f 100644
--- a/src/pybind_example/labelled_tetrahedral_mesh.py
+++ b/src/pybind_example/labelled_tetrahedral_mesh.py
@@ -1,8 +1,10 @@
+from typing import Iterable, Optional
+
import numpy as np
from ._wrp import LabelledTetrahedralMesh, make_tetrahedron
from ._wrp import build_cgal_image as _build_cgal_image
-def build_cgal_image(img: np.ndarray) -> bool:
- return _build_cgal_image(img.ravel().astype(np.uint16), img.shape, [1., 1., 1.])
+def build_cgal_image(img: np.ndarray, voxelsize: Optional[Iterable[float]]=(1., 1., 1.)) -> bool:
+ return _build_cgal_image(img.ravel().astype(np.uint16), img.shape, voxelsize)
diff --git a/tests/test_cgal_image.py b/tests/test_cgal_image.py
index 4f60f14..9506048 100644
--- a/tests/test_cgal_image.py
+++ b/tests/test_cgal_image.py
@@ -13,9 +13,11 @@ class TestCgalImage(unittest.TestCase):
self.image[:, :4, :4] = 2
self.image[:, 4:, :4] = 3
+ self.voxelsize = [0.5, 0.5, 1]
+
def tearDown(self):
pass
def test_cgal_image(self):
- res = build_cgal_image(self.image)
+ res = build_cgal_image(self.image, voxelsize=self.voxelsize)
assert res == 0
--
GitLab