structured-optics 0.6.2__tar.gz → 0.6.3__tar.gz

{structured_optics-0.6.2/structured_optics.egg-info → structured_optics-0.6.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: structured_optics
-Version: 0.6.2
+Version: 0.6.3
 Summary: Package for Structured Light simulation
 Author: Altilano C. Barbosa
 License-Expression: BSD-3-Clause

{structured_optics-0.6.2 → structured_optics-0.6.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "structured_optics"
-version = "0.6.2"
+version = "0.6.3"
 description = "Package for Structured Light simulation"
 readme = "README.md"
 license = "BSD-3-Clause"

{structured_optics-0.6.2 → structured_optics-0.6.3}/structured_optics/struct_opt.py

@@ -10,7 +10,7 @@ from .prop_methods import(  propagate_fresnel_conv, propagate_fresnel_fft, propa
 from .modes import (hg, hg_astigmatic, lg, nbessel, gbessel, lg_prod, frac_oam, frac_oam_qs,
                     IG_even, IG_odd, HInceG, circle, square, triangle, lp, lp_hel)
 
-from .tools import _build_from_coefs_and_basis
+from .utils import _build_from_coefs_and_basis
 from .diagnostics import BeamDiagnostics
 
 __all__ = ["Beam"]

{structured_optics-0.6.2 → structured_optics-0.6.3}/structured_optics/tools.py

@@ -257,45 +257,4 @@ def bessel_basis(Beam, Nmax, waist=None, norm1=False):
         basis = basis / np.max(np.abs(basis))
     return basis
 
-def _build_from_coefs_and_basis(Beam, coefs, basis, pol_index = 0):
-    """
-    Reconstruct a field as a weighted sum of basis modes and store it
-    into `Beam.field`.
-
-    Internal helper (not in `__all__`); used to turn a modal
-    decomposition (e.g. from `hg_proj`/`lg_proj`) plus its basis
-    (e.g. from `hg_basis`/`lg_basis`/`bessel_basis`) back into an
-    actual field.
-
-    Parameters
-    ----------
-    Beam : object
-        Beam instance whose `.field` will be overwritten with the
-        reconstructed field.
-    coefs : np.ndarray
-        2D array of complex coefficients, shape (N, N), one per basis
-        mode.
-    basis : np.ndarray
-        Array of basis mode fields, shape (N, N, Dy, Dx), matching
-        `coefs` in its first two dimensions (e.g. as returned by
-        `hg_basis` or `lg_basis`).
-    pol_index : int, optional
-        Polarization/field-component index of `Beam.field` (and of the
-        internal auxiliary field) that the reconstructed sum is
-        written into. Defaults to 0.
-
-    Returns
-    -------
-    Beam : object
-        `Beam`, with `Beam.field` set to the auxiliary field after
-        accumulating `sum(coefs[i, j] * basis[i, j])` into component
-        `pol_index`.
-    """
-    N = len(coefs)
-    aux = Beam.copy_clean()
-    for i in range(N):
-        for j in range(N):
-            aux.field[pol_index] = aux.field[pol_index] + coefs[i,j] * basis[i,j]
-    Beam.field = aux.field
-    return Beam
 

{structured_optics-0.6.2 → structured_optics-0.6.3}/structured_optics/utils.py

@@ -880,4 +880,42 @@ def zernikes_phase(beam, coefs, strengths,):
 
     for (n, m), strength in zip(coefs, strengths):
         phase += strength * zernike(n, m, r, phi)
-    return np.exp(1j * phase)
+    return np.exp(1j * phase)
+
+
+#builder util
+def _build_from_coefs_and_basis(Beam, coefs, basis, pol_index=0):
+    """
+    Reconstruct a field as a weighted sum of basis modes.
+
+    Parameters
+    ----------
+    Beam : object
+        Beam instance whose `.field` will be overwritten.
+    coefs : np.ndarray
+        Complex coefficient array, shape (N, N).
+    basis : np.ndarray
+        Basis modes, shape (N, N, Dy, Dx).
+    pol_index : int or None, optional
+        Polarization/field-component index to reconstruct. If None,
+        the reconstructed field is assigned to all polarization
+        components.
+
+    Returns
+    -------
+    Beam
+        Beam with the reconstructed field.
+    """
+    field = np.einsum(
+        "ij,ijyx->yx",
+        coefs,
+        basis,
+        optimize=True,
+    )
+
+    if pol_index is None:
+        Beam.field[...] = field
+    else:
+        Beam.field[pol_index] = field
+
+    return Beam

{structured_optics-0.6.2 → structured_optics-0.6.3/structured_optics.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: structured_optics
-Version: 0.6.2
+Version: 0.6.3
 Summary: Package for Structured Light simulation
 Author: Altilano C. Barbosa
 License-Expression: BSD-3-Clause