invrs-opt 0.10.5__py3-none-any.whl → 0.10.6__py3-none-any.whl

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/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/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.dist-info → invrs_opt-0.10.6.dist-info}/METADATA

@@ -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.dist-info → invrs_opt-0.10.6.dist-info}/RECORD

@@ -1,4 +1,4 @@
-invrs_opt/__init__.py,sha256=b2AnOoyTVD9z6CZEjHc3_S0un9BDAXTz8vv8yWtw0AE,586
+invrs_opt/__init__.py,sha256=4uq04SwbUcX0hVAhaX0-GJ4R4tRJAAXfIzzfU-VcalA,586
 invrs_opt/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 invrs_opt/experimental/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 invrs_opt/experimental/client.py,sha256=tbtH13FrA65XmTZfTO71CxJ78jeAEj3Zf85R-MTwbiU,4909
@@ -10,11 +10,11 @@ invrs_opt/optimizers/wrapped_optax.py,sha256=781-8v_TlHsGaQF9Se9_iOEvtOLOr-BesTL
 invrs_opt/parameterization/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 invrs_opt/parameterization/base.py,sha256=jSwrEO86lGkYQG5gWsHvcIMWpZnnbdiKpn--2qaU02g,5362
 invrs_opt/parameterization/filter_project.py,sha256=XL3HTEBLrF-q_75TjhOWLNdfUOSEEjKcoM7Qj844QpQ,4590
-invrs_opt/parameterization/gaussian_levelset.py,sha256=-6foekLTFoZDtMKuoMEvdxMJt0_zTxrKNJo0Vn-Rv80,26073
+invrs_opt/parameterization/gaussian_levelset.py,sha256=PDvjdgBzklRTCUoBpo4ZMcmXeTkn0BpZEzQj7ojtYGE,24813
 invrs_opt/parameterization/pixel.py,sha256=YWkyBhfYtzI8cQ-M90PAZqRAbabwVaUh0UiYIGegQHI,1955
-invrs_opt/parameterization/transforms.py,sha256=8GzaIsUuuXvMCLiqAEEfxmi9qE9KqHzbuTj_m0GjH3w,8216
-invrs_opt-0.10.5.dist-info/LICENSE,sha256=IMF9i4xIpgCADf0U-V1cuf9HBmqWQd3qtI3FSuyW4zE,26526
-invrs_opt-0.10.5.dist-info/METADATA,sha256=JV6RdTh63uMDr2ae3PedC4BIxz54G8twEBGd38hKlZg,32816
-invrs_opt-0.10.5.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-invrs_opt-0.10.5.dist-info/top_level.txt,sha256=hOziS2uJ_NgwaW9yhtOfeuYnm1X7vobPBcp_6eVWKfM,10
-invrs_opt-0.10.5.dist-info/RECORD,,
+invrs_opt/parameterization/transforms.py,sha256=mqDKuAg4wpSL9kh0oYKxtSoH0mHOQeKG1RND2fJSYaU,9441
+invrs_opt-0.10.6.dist-info/LICENSE,sha256=IMF9i4xIpgCADf0U-V1cuf9HBmqWQd3qtI3FSuyW4zE,26526
+invrs_opt-0.10.6.dist-info/METADATA,sha256=40XUQ7i3S4nYRGUqkEVOwKrJwGtpvdwmpD9ojFpAeAM,32816
+invrs_opt-0.10.6.dist-info/WHEEL,sha256=nn6H5-ilmfVryoAQl3ZQ2l8SH5imPWFpm1A5FgEuFV4,91
+invrs_opt-0.10.6.dist-info/top_level.txt,sha256=hOziS2uJ_NgwaW9yhtOfeuYnm1X7vobPBcp_6eVWKfM,10
+invrs_opt-0.10.6.dist-info/RECORD,,

{invrs_opt-0.10.5.dist-info → invrs_opt-0.10.6.dist-info}/WHEEL

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