freealg 0.1.5__tar.gz → 0.1.6__tar.gz

{freealg-0.1.5/freealg.egg-info → freealg-0.1.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: freealg
-Version: 0.1.5
+Version: 0.1.6
 Summary: Free probability for large matrices
 Keywords: leaderboard bot chat
 Platform: Linux

freealg-0.1.6/freealg/__version__.py

@@ -0,0 +1 @@
+__version__ = "0.1.6"

{freealg-0.1.5 → freealg-0.1.6}/freealg/_decompress.py

@@ -11,8 +11,9 @@
 # =======
 
 import numpy
+# from scipy.integrate import solve_ivp
 
-__all__ = ['decompress']
+__all__ = ['decompress', 'reverse_characteristics']
 
 
 # ==========
@@ -134,3 +135,46 @@ def decompress(matrix, size, x=None, delta=1e-4, iterations=500, step_size=0.1,
     rho = rho.reshape(*x.shape)
 
     return rho, x, (lb, ub)
+
+
+# =======================
+# reverse characteristics
+# =======================
+
+def reverse_characteristics(matrix, z_inits, T, iterations=500, step_size=0.1,
+                            tolerance=1e-8):
+    """
+    """
+
+    t_span = (0, T)
+    t_eval = numpy.linspace(t_span[0], t_span[1], 50)
+
+    m = matrix._eval_stieltjes
+
+    def _char_z(z, t):
+        return z + (1 / m(z)[1]) * (1 - numpy.exp(t))
+
+    target_z, target_t = numpy.meshgrid(z_inits, t_eval)
+
+    z = numpy.full(target_z.shape, numpy.mean(matrix.support) - .1j,
+                   dtype=numpy.complex128)
+
+    # Broken Newton steps can produce a lot of warnings. Removing them for now.
+    with numpy.errstate(all='ignore'):
+        for _ in range(iterations):
+            objective = _char_z(z, target_t) - target_z
+            mask = numpy.abs(objective) >= tolerance
+            if not numpy.any(mask):
+                break
+            z_m = z[mask]
+            t_m = target_t[mask]
+
+            # Perform finite difference approximation
+            dfdz = _char_z(z_m+tolerance, t_m) - _char_z(z_m-tolerance, t_m)
+            dfdz /= 2*tolerance
+            dfdz[dfdz == 0] = 1.0
+
+            # Perform Newton step
+            z[mask] = z_m - step_size * objective[mask] / dfdz
+
+    return z

{freealg-0.1.5 → freealg-0.1.6}/freealg/_plot_util.py

@@ -18,6 +18,7 @@ import matplotlib
 import colorsys
 import matplotlib.ticker as ticker
 import matplotlib.gridspec as gridspec
+from ._decompress import reverse_characteristics
 
 __all__ = ['plot_fit', 'plot_density', 'plot_hilbert', 'plot_stieltjes',
            'plot_stieltjes_on_disk']
@@ -243,11 +244,42 @@ def _value_formatter(v, pos):
         return f"{m_val:.1f}"
 
 
+# ================
+# plot char curves
+# ================
+def _plot_char_curves(ax, char_curves):
+    """
+    """
+
+    curves = reverse_characteristics(char_curves['matrix'],
+                                     char_curves['z'], 4)
+    lw = 2
+    for idx in range(curves.shape[1]):
+
+        creal, cimag = curves[:, idx].real, curves[:, idx].imag
+
+        ax.plot(creal, cimag, ':', color='white', linewidth=lw,
+                alpha=0.75)
+
+        ax.annotate(
+            '',                             # no text
+            xy=(creal[-1], cimag[-1]),      # arrow tip at final point
+            xytext=(creal[-2], cimag[-2]),  # tail at penultimate point
+            arrowprops=dict(
+                arrowstyle='-|>',           # simple arrow head
+                mutation_scale=5,           # size of the head
+                color='white',
+                lw=lw, alpha=0.75,          # arrow shaft line width
+            )
+        )
+
+
 # ==============
 # plot stieltjes
 # ==============
 
-def plot_stieltjes(x, y, m1, m2, support, latex=False, save=False):
+def plot_stieltjes(x, y, m1, m2, support, latex=False, char_curves=None,
+                   save=False):
     """
     """
 
@@ -297,6 +329,10 @@ def plot_stieltjes(x, y, m1, m2, support, latex=False, save=False):
         ax1.set_xlim([x_min, x_max])
         ax1.set_ylim([y_min, y_max])
 
+        # Plot characteristic curves
+        if char_curves is not None:
+            _plot_char_curves(ax1, char_curves)
+
         pos = ax1.get_position()
         cbar_width = 0.013
         pad = 0.013

{freealg-0.1.5 → freealg-0.1.6}/freealg/freeform.py

@@ -631,7 +631,8 @@ class FreeForm(object):
     # stieltjes
     # =========
 
-    def stieltjes(self, x, y, plot=False, latex=False, save=False):
+    def stieltjes(self, x, y, plot=False, latex=False, char_curves=None,
+                  save=False):
         """
         Compute Stieltjes transform of the spectral density over a 2D Cartesian
         grid on the complex plane.
@@ -655,6 +656,10 @@ class FreeForm(object):
             If `True`, the plot is rendered using LaTeX. This option is
             relevant only if ``plot=True``.
 
+        char_curves : numpy.array, default=None
+            If ``plot=True``, also plot characteristic curves starting from
+            these locations.
+
         save : bool, default=False
             If not `False`, the plot is saved. If a string is given, it is
             assumed to the save filename (with the file extension). This option
@@ -742,7 +747,9 @@ class FreeForm(object):
         m2[mask_m, :] = -m1[mask_m, :] + self._glue(z[mask_m, :])
 
         if plot:
-            plot_stieltjes(x, y, m1, m2, self.support, latex=latex, save=save)
+            plot_stieltjes(x, y, m1, m2, self.support, latex=latex,
+                           char_curves={'matrix': self, 'z': char_curves},
+                           save=save)
 
         return m1, m2
 

{freealg-0.1.5 → freealg-0.1.6/freealg.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: freealg
-Version: 0.1.5
+Version: 0.1.6
 Summary: Free probability for large matrices
 Keywords: leaderboard bot chat
 Platform: Linux

freealg-0.1.5/freealg/__version__.py

@@ -1 +0,0 @@
-__version__ = "0.1.5"