foscat 3.8.0__py3-none-any.whl → 3.9.0__py3-none-any.whl

foscat/scat_cov2D.py

@@ -1,6 +1,8 @@
-import foscat.scat_cov as scat
 import numpy as np
 
+import foscat.scat_cov as scat
+
+
 class scat_cov2D:
     def __init__(self, s0, s2, s3, s4, s1=None, s3p=None, backend=None):
 
@@ -11,12 +13,13 @@ class scat_cov2D:
     def fill(self, im, nullval=0):
         return self.fill_2d(im, nullval=nullval)
 
+
 class funct(scat.funct):
     def __init__(self, *args, **kwargs):
         # Impose que use_2D=True pour la classe scat
-        super().__init__(use_2D=True,KERNELSZ=5, *args, **kwargs)
-     
-    def spectrum(self,image):
+        super().__init__(use_2D=True, KERNELSZ=5, *args, **kwargs)
+
+    def spectrum(self, image):
         """
         Compute the 1D power spectrum of a 2D image by averaging the 2D power spectrum
         over concentric frequency rings (radial averaging), using np.bincount for efficiency.
@@ -29,7 +32,7 @@ class funct(scat.funct):
         - spectrum_1d : corresponding 1D power spectrum
         """
         import numpy as np
-        
+
         # Compute the 2D Fourier Transform and shift the zero frequency to the center
         F = np.fft.fftshift(np.fft.fft2(image))
         power_spectrum = np.abs(F) ** 2
@@ -37,7 +40,7 @@ class funct(scat.funct):
         # Create coordinate grids and compute the radial distance from the center
         y, x = np.indices(power_spectrum.shape)
         center = np.array(power_spectrum.shape) // 2
-        r = np.sqrt((x - center[1])**2 + (y - center[0])**2).astype(int)
+        r = np.sqrt((x - center[1]) ** 2 + (y - center[0]) ** 2).astype(int)
 
         # Use np.bincount for fast summation and counting
         sum_power = np.bincount(r.ravel(), weights=power_spectrum.ravel())
@@ -48,71 +51,161 @@ class funct(scat.funct):
 
         return spectrum_1d
 
-    def plot_results(self,in_image,out_image,vmin=None,vmax=None,cmap='coolwarm',spec_range=None):
+    def plot_results(
+        self,
+        in_image,
+        out_image,
+        vmin=None,
+        vmax=None,
+        cmap="coolwarm",
+        spec_range=None,
+    ):
         import matplotlib.pyplot as plt
-        
-        if len(out_image.shape)>2:
-            nimage=out_image.shape[0]
-            ndraw=np.min([3,nimage])
-            plt.figure(figsize=(16,12))
-            plt.subplot(2,ndraw+1,1)
-            plt.title('Original field') 
-            plt.imshow(in_image,cmap=cmap,vmin=vmin,vmax=vmax,origin='lower')
+
+        if len(out_image.shape) > 2:
+            nimage = out_image.shape[0]
+            ndraw = np.min([3, nimage])
+            plt.figure(figsize=(16, 12))
+            plt.subplot(2, ndraw + 1, 1)
+            plt.title("Original field")
+            plt.imshow(in_image, cmap=cmap, vmin=vmin, vmax=vmax, origin="lower")
             plt.xticks([])
             plt.yticks([])
             for k in range(ndraw):
-                plt.subplot(2,ndraw+1,2+k)
-                plt.title('Modeled field #%d'%(k)) 
-                plt.imshow(out_image[k],cmap=cmap,vmin=vmin,vmax=vmax,origin='lower')
+                plt.subplot(2, ndraw + 1, 2 + k)
+                plt.title("Modeled field #%d" % (k))
+                plt.imshow(
+                    out_image[k], cmap=cmap, vmin=vmin, vmax=vmax, origin="lower"
+                )
                 plt.xticks([])
                 plt.yticks([])
-            plt.subplot(2,2,3)
-            plt.title('Histogram') 
+            plt.subplot(2, 2, 3)
+            plt.title("Histogram")
             for k in range(nimage):
-                if k==0:
-                    plt.hist(out_image[k].flatten(),bins=100,label='modeled',color='b',histtype='step',log=True,alpha=0.5)
+                if k == 0:
+                    plt.hist(
+                        out_image[k].flatten(),
+                        bins=100,
+                        label="modeled",
+                        color="b",
+                        histtype="step",
+                        log=True,
+                        alpha=0.5,
+                    )
                 else:
-                    plt.hist(out_image[k].flatten(),bins=100,color='b',histtype='step',log=True,alpha=0.5)
-            plt.hist(in_image.flatten(),bins=100,label='original',color='r',histtype='step',log=True)
+                    plt.hist(
+                        out_image[k].flatten(),
+                        bins=100,
+                        color="b",
+                        histtype="step",
+                        log=True,
+                        alpha=0.5,
+                    )
+            plt.hist(
+                in_image.flatten(),
+                bins=100,
+                label="original",
+                color="r",
+                histtype="step",
+                log=True,
+            )
             plt.legend(frameon=0)
-            plt.subplot(2,2,4)
-            plt.title('Powerspectra') 
+            plt.subplot(2, 2, 4)
+            plt.title("Powerspectra")
             for k in range(nimage):
-                if k==0:
-                    plt.plot(self.spectrum(out_image[k]),color='b',label='modeled',alpha=0.5)
+                if k == 0:
+                    plt.plot(
+                        self.spectrum(out_image[k]),
+                        color="b",
+                        label="modeled",
+                        alpha=0.5,
+                    )
                 else:
-                    plt.plot(self.spectrum(out_image[k]),color='b',alpha=0.5)
-            plt.plot(self.spectrum(in_image),color='r',label='original')
-            plt.xscale('log')
-            plt.yscale('log')
+                    plt.plot(self.spectrum(out_image[k]), color="b", alpha=0.5)
+            plt.plot(self.spectrum(in_image), color="r", label="original")
+            plt.xscale("log")
+            plt.yscale("log")
             plt.legend(frameon=0)
             if spec_range is not None:
-                plt.ylim(spec_range[0],spec_range[1])
+                plt.ylim(spec_range[0], spec_range[1])
         else:
-            plt.figure(figsize=(16,3))
-            plt.subplot(1,4,1)
-            plt.title('Original field') 
-            plt.imshow(in_image,cmap=cmap,vmin=vmin,vmax=vmax,origin='lower')
+            plt.figure(figsize=(16, 3))
+            plt.subplot(1, 4, 1)
+            plt.title("Original field")
+            plt.imshow(in_image, cmap=cmap, vmin=vmin, vmax=vmax, origin="lower")
             plt.xticks([])
             plt.yticks([])
-            plt.subplot(1,4,2)
-            plt.title('Modeled field') 
-            plt.imshow(out_image,cmap=cmap,vmin=vmin,vmax=vmax,origin='lower')
+            plt.subplot(1, 4, 2)
+            plt.title("Modeled field")
+            plt.imshow(out_image, cmap=cmap, vmin=vmin, vmax=vmax, origin="lower")
             plt.xticks([])
             plt.yticks([])
-            plt.subplot(1,4,3)
-            plt.title('Histogram') 
-            plt.hist(in_image.flatten(),bins=100,label='original',color='r',histtype='step',log=True)
-            plt.hist(out_image.flatten(),bins=100,label='modeled',color='b',histtype='step',log=True)
+            plt.subplot(1, 4, 3)
+            plt.title("Histogram")
+            plt.hist(
+                in_image.flatten(),
+                bins=100,
+                label="original",
+                color="r",
+                histtype="step",
+                log=True,
+            )
+            plt.hist(
+                out_image.flatten(),
+                bins=100,
+                label="modeled",
+                color="b",
+                histtype="step",
+                log=True,
+            )
             plt.legend(frameon=0)
-            plt.subplot(1,4,4)
-            plt.title('Powerspectra') 
-            plt.plot(self.spectrum(in_image),color='b',label='original')
-            plt.plot(self.spectrum(out_image),color='r',label='modeled')
-            plt.xscale('log')
-            plt.yscale('log')
+            plt.subplot(1, 4, 4)
+            plt.title("Powerspectra")
+            plt.plot(self.spectrum(in_image), color="b", label="original")
+            plt.plot(self.spectrum(out_image), color="r", label="modeled")
+            plt.xscale("log")
+            plt.yscale("log")
             if spec_range is not None:
-                plt.ylim(spec_range[0],spec_range[1])
+                plt.ylim(spec_range[0], spec_range[1])
             plt.legend(frameon=0)
-        
-        
+
+    def plot_results(self, in_image, out_image, vmin=None, vmax=None, cmap="coolwarm"):
+        import matplotlib.pyplot as plt
+
+        plt.figure(figsize=(16, 3))
+        plt.subplot(1, 4, 1)
+        plt.title("Original field")
+        plt.imshow(in_image, cmap=cmap, vmin=vmin, vmax=vmax, origin="lower")
+        plt.xticks([])
+        plt.yticks([])
+        plt.subplot(1, 4, 2)
+        plt.title("Modeled field")
+        plt.imshow(out_image, cmap=cmap, vmin=vmin, vmax=vmax, origin="lower")
+        plt.xticks([])
+        plt.yticks([])
+        plt.subplot(1, 4, 3)
+        plt.title("Histogram")
+        plt.hist(
+            in_image.flatten(),
+            bins=100,
+            label="original",
+            color="r",
+            histtype="step",
+            log=True,
+        )
+        plt.hist(
+            out_image.flatten(),
+            bins=100,
+            label="modeled",
+            color="b",
+            histtype="step",
+            log=True,
+        )
+        plt.legend(frameon=0)
+        plt.subplot(1, 4, 4)
+        plt.title("Powerspectra")
+        plt.plot(self.spectrum(in_image), color="b", label="original")
+        plt.plot(self.spectrum(out_image), color="r", label="modeled")
+        plt.xscale("log")
+        plt.yscale("log")
+        plt.legend(frameon=0)

{foscat-3.8.0.dist-info → foscat-3.9.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: foscat
-Version: 3.8.0
+Version: 3.9.0
 Summary: Generate synthetic Healpix or 2D data using Cross Scattering Transform
 Author-email: Jean-Marc DELOUIS <jean.marc.delouis@ifremer.fr>
 Maintainer-email: Theo Foulquier <theo.foulquier@ifremer.fr>

{foscat-3.8.0.dist-info → foscat-3.9.0.dist-info}/RECORD

@@ -1,30 +1,30 @@
-foscat/BkBase.py,sha256=2ZdoxDAtr_7JWjIijdijryJ8sX7P5cecQXFCQJ-4KYk,21157
-foscat/BkNumpy.py,sha256=uaUCKrlLUB10yWDn2Q814Pa6Cuj03IBgkH6uazrJopQ,10409
-foscat/BkTensorflow.py,sha256=qWqLsyf3AaSpsdxj4fBWAfuxzfE2jG6fGCP3PEVl7M0,15660
-foscat/BkTorch.py,sha256=BrReEQZ1bmcCfweAWl6Qhthnm3KjgJAqKhUvz5CEiTc,13800
+foscat/BkBase.py,sha256=_iszgMdVIVEB47EBxNt5xemsdaKzsNFPStDF00M_-Ng,21281
+foscat/BkNumpy.py,sha256=zRldS_-L6A7y1zDzEPZXQntuw3Paw2zHZowhD43FHRs,10589
+foscat/BkTensorflow.py,sha256=N5TBacuyFB1-qGTi2kOc8zbgWzj5lVRRN47uZJpJJ10,15713
+foscat/BkTorch.py,sha256=mW0hyLWoCoGlt34a2587zqbAv1FuVgMq32Yu4AjmUpU,14306
 foscat/CNN.py,sha256=j0F2a4Xf3LijhyD_WVZ6Eg_IjGuXw3ddH6Iudj1xVaw,4874
 foscat/CircSpline.py,sha256=CXi49FxF8ZoeZ17Ua8c1AZXe2B5ICEC9aCXb97atB3s,4028
-foscat/FoCUS.py,sha256=jqIYqgKz-F8Wq6ciiQugCoVDYkqNcFTTSFx44dtxStc,103330
+foscat/FoCUS.py,sha256=h3QCu6if8fh9gEBrprEr3mG07TZ5O7y9eCc4qHbsCYk,103551
 foscat/GCNN.py,sha256=5RV-FKuvqbD-k99TwiM4CttM2LMZE21WD0IK0j5Mkko,7599
 foscat/Softmax.py,sha256=aBLQauoG0q2SJYPotV6U-cxAhsJcspWHNRWdnA_nAiQ,2854
 foscat/Spline1D.py,sha256=rKzzenduaZZ-yBDJd35it6Gyrj1spqb7hoIaUgISPzY,2983
-foscat/Synthesis.py,sha256=3oL-WIwOzjO6jYn-7J3TfAQSiPaQTqGtrmxcGBh1Gvs,13787
+foscat/Synthesis.py,sha256=NJI1laestYfqTz9iqPlqwTmwPdQWRZoe_-Sp61gozF4,13822
 foscat/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-foscat/alm.py,sha256=5M87wcAIQiPJ9xjDdmKtJoVDfqZn8I0Xr6QZ7HaxMK4,33516
-foscat/backend.py,sha256=CcHtYS1j2dmQJyJDcm13ksS5Ya6V8bpp0DnKOt4uwCs,45580
+foscat/alm.py,sha256=qZlsYj5HzV1EY9Fdzt0U8bemrZHZziaMOKZ55FU8foM,33806
+foscat/backend.py,sha256=l3aMwDyXP6jURMIvratFMGWCTcQpaR68KnUuuGDezqE,45418
 foscat/backend_tens.py,sha256=9Dp136m9frkclkwifJQLLbIpl3ETI3_txdPUZcKfuMw,1618
 foscat/loss_backend_tens.py,sha256=dCOVN6faDtIpN3VO78HTmYP2i5fnFAf-Ddy5qVBlGrM,1783
 foscat/loss_backend_torch.py,sha256=k3z18Dj3SaLKK6ZIKcm7GO4U_YKYVP6LtHG1aIbxkYk,1627
 foscat/scat.py,sha256=qGYiBIysPt65MdmF07WWA4piVlTfA9-lFDTaicnqC2w,72822
 foscat/scat1D.py,sha256=W5Uu6wdQ4ZsFKXpof0f1OBl-1wjJmW7ruvddRWxe7uM,53726
 foscat/scat2D.py,sha256=boKj0ASqMMSy7uQLK6hPniG87m3hZGJBYBiq5v8F9IQ,532
-foscat/scat_cov.py,sha256=5_ym9AfdYQQJS-HL90r6HiTm8yqKBW1EjZAoU_gKa2U,271517
+foscat/scat_cov.py,sha256=UL-1rkdGKios33gw6IYnhFqQ_hbQPJeURQra13QzV0Y,253970
 foscat/scat_cov1D.py,sha256=XOxsZZ5TYq8f34i2tUgIfzyaqaTDlICB3HzD2l_puro,531
-foscat/scat_cov2D.py,sha256=GC26Yop7Y_Db3hNRQJYV8B7TxERgoH7TYr77wygmonQ,4813
+foscat/scat_cov2D.py,sha256=pAm0fKw8wyXram0TFbtw8tGcc8QPKuPXpQk0kh10r4U,7078
 foscat/scat_cov_map.py,sha256=Swt39-nYEaQkBzyX4EOAQBvUuYQpERzJ-uVxSWS2b-Y,2911
 foscat/scat_cov_map2D.py,sha256=FqF45FBcoiQbvuVsrLWUIPRUc95GsKsrnH6fKzB3GlE,2841
-foscat-3.8.0.dist-info/LICENSE,sha256=i0ukIr8ZUpkSY2sZaE9XZK-6vuSU5iG6IgX_3pjatP8,1505
-foscat-3.8.0.dist-info/METADATA,sha256=mWWdEHezTXBN7bb9sX3fwpedVbbBR9FWuUl2TztutLc,7190
-foscat-3.8.0.dist-info/WHEEL,sha256=52BFRY2Up02UkjOa29eZOS2VxUrpPORXg1pkohGGUS8,91
-foscat-3.8.0.dist-info/top_level.txt,sha256=AGySXBBAlJgb8Tj8af6m_F-aiNg2zNTcybCUPVOKjAg,7
-foscat-3.8.0.dist-info/RECORD,,
+foscat-3.9.0.dist-info/LICENSE,sha256=i0ukIr8ZUpkSY2sZaE9XZK-6vuSU5iG6IgX_3pjatP8,1505
+foscat-3.9.0.dist-info/METADATA,sha256=NZTyxhm4YIy44NAAp1NhvTRAIRdy__1esRG594Sp94c,7190
+foscat-3.9.0.dist-info/WHEEL,sha256=beeZ86-EfXScwlR_HKu4SllMC9wUEj_8Z_4FJ3egI2w,91
+foscat-3.9.0.dist-info/top_level.txt,sha256=AGySXBBAlJgb8Tj8af6m_F-aiNg2zNTcybCUPVOKjAg,7
+foscat-3.9.0.dist-info/RECORD,,

{foscat-3.8.0.dist-info → foscat-3.9.0.dist-info}/WHEEL

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