invrs-opt 0.10.5__tar.gz → 0.10.6__tar.gz

{invrs_opt-0.10.5 → invrs_opt-0.10.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: invrs_opt
-Version: 0.10.5
+Version: 0.10.6
 Summary: Algorithms for inverse design
 Author-email: "Martin F. Schubert" <mfschubert@gmail.com>
 Maintainer-email: "Martin F. Schubert" <mfschubert@gmail.com>

{invrs_opt-0.10.5 → invrs_opt-0.10.6}/pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 
 name = "invrs_opt"
-version = "v0.10.5"
+version = "v0.10.6"
 description = "Algorithms for inverse design"
 keywords = ["topology", "optimization", "jax", "inverse design"]
 readme = "README.md"

{invrs_opt-0.10.5 → invrs_opt-0.10.6}/src/invrs_opt/__init__.py

@@ -3,7 +3,7 @@
 Copyright (c) 2023 The INVRS-IO authors.
 """
 
-__version__ = "v0.10.5"
+__version__ = "v0.10.6"
 __author__ = "Martin F. Schubert <mfschubert@gmail.com>"
 
 from invrs_opt import parameterization as parameterization

{invrs_opt-0.10.5 → invrs_opt-0.10.6}/src/invrs_opt/parameterization/gaussian_levelset.py

@@ -619,49 +619,12 @@ def _levelset_threshold(
 ) -> jnp.ndarray:
     """Thresholds a level set function `phi`."""
     if mask_gradient:
-        interface = _interface_pixels(phi, periodic)
+        interface = transforms.interface_pixels(phi, periodic)
         phi = jnp.where(interface, phi, jax.lax.stop_gradient(phi))
     thresholded = (phi > 0).astype(float) + (phi - jax.lax.stop_gradient(phi))
     return thresholded
 
 
-def _interface_pixels(phi: jnp.ndarray, periodic: Tuple[bool, bool]) -> jnp.ndarray:
-    """Identifies interface pixels of a level set function `phi`."""
-    batch_shape = phi.shape[:-2]
-    phi = phi.reshape((-1,) + phi.shape[-2:])
-
-    pad_mode = (
-        "wrap" if periodic[0] else "edge",
-        "wrap" if periodic[1] else "edge",
-    )
-    pad_width = ((1, 1), (1, 1))
-
-    kernel = jnp.asarray([[0, 1, 0], [1, 0, 1], [0, 1, 0]], dtype=float)
-
-    solid = phi > 0
-    void = ~solid
-
-    solid_padded = transforms.pad2d(solid, pad_width, pad_mode)
-    num_solid_adjacent = transforms.conv(
-        x=solid_padded[:, jnp.newaxis, :, :].astype(float),
-        kernel=kernel[jnp.newaxis, jnp.newaxis, :, :],
-        padding="VALID",
-    )
-    num_solid_adjacent = jnp.squeeze(num_solid_adjacent, axis=1)
-
-    void_padded = transforms.pad2d(void, pad_width, pad_mode)
-    num_void_adjacent = transforms.conv(
-        x=void_padded[:, jnp.newaxis, :, :].astype(float),
-        kernel=kernel[jnp.newaxis, jnp.newaxis, :, :],
-        padding="VALID",
-    )
-    num_void_adjacent = jnp.squeeze(num_void_adjacent, axis=1)
-
-    interface = solid & (num_void_adjacent > 0) | void & (num_solid_adjacent > 0)
-
-    return interface.reshape(batch_shape + interface.shape[-2:])
-
-
 def _downsample_spatial_dims(x: jnp.ndarray, downsample_factor: int) -> jnp.ndarray:
     """Downsamples the two trailing axes of `x` by `downsample_factor`."""
     shape = x.shape[:-2] + (

{invrs_opt-0.10.5 → invrs_opt-0.10.6}/src/invrs_opt/parameterization/transforms.py

@@ -231,3 +231,40 @@ def box_downsample(x: jnp.ndarray, shape: Tuple[int, ...]) -> jnp.ndarray:
     axes = list(range(1, 2 * x.ndim, 2))
     x = x.reshape(shape)
     return jnp.mean(x, axis=axes)
+
+
+def interface_pixels(phi: jnp.ndarray, periodic: Tuple[bool, bool]) -> jnp.ndarray:
+    """Identifies interface pixels of a level set function `phi`."""
+    batch_shape = phi.shape[:-2]
+    phi = phi.reshape((-1,) + phi.shape[-2:])
+
+    pad_mode = (
+        "wrap" if periodic[0] else "edge",
+        "wrap" if periodic[1] else "edge",
+    )
+    pad_width = ((1, 1), (1, 1))
+
+    kernel = jnp.asarray([[0, 1, 0], [1, 0, 1], [0, 1, 0]], dtype=float)
+
+    solid = phi > 0
+    void = ~solid
+
+    solid_padded = pad2d(solid, pad_width, pad_mode)
+    num_solid_adjacent = conv(
+        x=solid_padded[:, jnp.newaxis, :, :].astype(float),
+        kernel=kernel[jnp.newaxis, jnp.newaxis, :, :],
+        padding="VALID",
+    )
+    num_solid_adjacent = jnp.squeeze(num_solid_adjacent, axis=1)
+
+    void_padded = pad2d(void, pad_width, pad_mode)
+    num_void_adjacent = conv(
+        x=void_padded[:, jnp.newaxis, :, :].astype(float),
+        kernel=kernel[jnp.newaxis, jnp.newaxis, :, :],
+        padding="VALID",
+    )
+    num_void_adjacent = jnp.squeeze(num_void_adjacent, axis=1)
+
+    interface = solid & (num_void_adjacent > 0) | void & (num_solid_adjacent > 0)
+
+    return interface.reshape(batch_shape + interface.shape[-2:])

{invrs_opt-0.10.5 → invrs_opt-0.10.6}/src/invrs_opt.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: invrs_opt
-Version: 0.10.5
+Version: 0.10.6
 Summary: Algorithms for inverse design
 Author-email: "Martin F. Schubert" <mfschubert@gmail.com>
 Maintainer-email: "Martin F. Schubert" <mfschubert@gmail.com>