freealg 0.1.5__py3-none-any.whl → 0.1.6__py3-none-any.whl

freealg/__version__.py

@@ -1 +1 @@
-__version__ = "0.1.5"
+__version__ = "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/_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/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.dist-info → freealg-0.1.6.dist-info}/METADATA

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

@@ -1,21 +1,21 @@
 freealg/__init__.py,sha256=K92neXJZ9VE1U_j_pj28Qyq1MzlMXhOuYK2ZihgwCaU,463
-freealg/__version__.py,sha256=rPSfWgIeq2YWVPyESOAwCBt8vftsTpIkuLAGDEzyRQc,22
+freealg/__version__.py,sha256=n3oM6B_EMz93NsTI18NNZd-jKFcUPzUkbIKj5VFK5ok,22
 freealg/_chebyshev.py,sha256=X6u5pKjR1HPZ-KbCfr7zT6HRwB6pZMADvVS3sT5LTkA,5638
 freealg/_damp.py,sha256=k2vtBtWOxQBf4qXaWu_En81lQBXbEO4QbxxWpvuVhdE,1802
-freealg/_decompress.py,sha256=H7ocq09gQnCY-q_8xHi6qyYj3qp239MCgj406hn0yeE,3344
+freealg/_decompress.py,sha256=7U2lL8F5z76aFuZJBsPj70jEVRuzvJHnIh5FSw-aLME,4680
 freealg/_jacobi.py,sha256=AT4ONSHGGDxVKE3MGMLyMR8uDFiO-e9u3x5udYfdJJk,5635
 freealg/_pade.py,sha256=mP96wEPfIzHLZ6PDB5OyhmSA8N1uVPVUkmJa3ebXXiU,13623
-freealg/_plot_util.py,sha256=wVx99GRdIFu_wzmG8f5JSDZ65BJohnuSBm3mZ58wElg,18426
+freealg/_plot_util.py,sha256=Ng9c_U9kxsRb16GsvVa2u4AiXUKw0oD4E3BB_e9d9h8,19575
 freealg/_sample.py,sha256=K1ZxKoiuPbEKyh-swL5X7gz1kYcQno6Mof0o1xF38tg,2323
 freealg/_util.py,sha256=alJ9s1U_sHL7dXq7hI10fa8CF_AZ6Xmy_QsoyDYPSDQ,3677
-freealg/freeform.py,sha256=kbh7UoOJkAVFKj2Zmddy803-asoslkqn-gWJ-HpLN7U,28750
+freealg/freeform.py,sha256=pZM4IUVegXjKCi9484Gz4bPkYt0tRvslTrha_yKW7y4,29042
 freealg/distributions/__init__.py,sha256=Hnk9bJi4Wy8I_1uuskRyrT2DUpPN1YmBY5uK7XI3U_o,644
 freealg/distributions/kesten_mckay.py,sha256=Oq2FCX60seojy7LDn8nYPrbqinmXv4YC-93VxlmDQ6M,15913
 freealg/distributions/marchenko_pastur.py,sha256=GwDTN-7au2h7H7PnZkQfs6bas8fNhgEnQ-hTWsBMZuE,16403
 freealg/distributions/wachter.py,sha256=2eqbJY4S1MqLjgqO6qY06m3-_s-bKTuSiryS_ZH_xvI,16136
 freealg/distributions/wigner.py,sha256=MSrB-HLMzOwnWDDzw5XPLsoL4LEIV35w5jWeL-qDn9Y,15448
-freealg-0.1.5.dist-info/licenses/LICENSE.txt,sha256=J-EEYEtxb3VVf_Bn1TYfWnpY5lMFIM15iLDDcnaDTPA,1443
-freealg-0.1.5.dist-info/METADATA,sha256=sG6aPyglynO4hXzXNnMPDliEtC5JLlrOMy84e_KMDzU,2939
-freealg-0.1.5.dist-info/WHEEL,sha256=Nw36Djuh_5VDukK0H78QzOX-_FQEo6V37m3nkm96gtU,91
-freealg-0.1.5.dist-info/top_level.txt,sha256=eR2wrgYwDdnnJ9Zf5PruPqe4kQav0GMvRsqct6y00Q8,8
-freealg-0.1.5.dist-info/RECORD,,
+freealg-0.1.6.dist-info/licenses/LICENSE.txt,sha256=J-EEYEtxb3VVf_Bn1TYfWnpY5lMFIM15iLDDcnaDTPA,1443
+freealg-0.1.6.dist-info/METADATA,sha256=eIE88q1eRccGUJdZmKYfjoP6p-XTP4y7tBkKrCb3jfU,2939
+freealg-0.1.6.dist-info/WHEEL,sha256=Nw36Djuh_5VDukK0H78QzOX-_FQEo6V37m3nkm96gtU,91
+freealg-0.1.6.dist-info/top_level.txt,sha256=eR2wrgYwDdnnJ9Zf5PruPqe4kQav0GMvRsqct6y00Q8,8
+freealg-0.1.6.dist-info/RECORD,,