plotastrodata 1.8.8__tar.gz → 1.8.10__tar.gz

{plotastrodata-1.8.8/plotastrodata.egg-info → plotastrodata-1.8.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: plotastrodata
-Version: 1.8.8
+Version: 1.8.10
 Summary: plotastrodata is a tool for astronomers to create figures from FITS files and perform fundamental data analyses with ease.
 Home-page: https://github.com/yusukeaso-astron/plotastrodata
 Download-URL: https://github.com/yusukeaso-astron/plotastrodata

{plotastrodata-1.8.8 → plotastrodata-1.8.10}/plotastrodata/__init__.py

@@ -1,4 +1,4 @@
 import warnings
 
 warnings.simplefilter('ignore', FutureWarning)
-__version__ = '1.8.8'
+__version__ = '1.8.10'

{plotastrodata-1.8.8 → plotastrodata-1.8.10}/plotastrodata/plot_utils.py

@@ -95,36 +95,6 @@ def logcbticks(vmin: float = -3.01, vmax: float = 3.01
     return ticks, ticklabels
 
 
-def pow10(x: np.ndarray, stretchpower: float) -> np.ndarray:
-    """A power-law function scaled by xmin. This function is used for the case of stretch='power' in PlotAstroData.add_color().
-
-    Args:
-        x (np.ndarray): Input in the linear scale.
-        stretchpower (float, optional): The output is (data**stretchpower - 1) / stretchpower / ln(10). 1 means the linear scale, while 0 means the logarithmic scale.
-
-    Returns:
-        np.ndarray: Output values.
-    """
-    p = 1e-6 if stretchpower == 0 else stretchpower
-    y = (x**p - 1) / p / np.log(10)
-    return y
-
-
-def ipow10(x: np.ndarray, stretchpower: float) -> np.ndarray:
-    """The inverse function of pow10. This function is used for the case of stretch='power' in PlotAstroData.add_color().
-
-    Args:
-        x (np.ndarray): Input values.
-        stretchpower (float, optional): The input is (data**stretchpower - 1) / stretchpower / ln(10). 1 means the linear scale, while 0 means the logarithmic scale.
-
-    Returns:
-        np.ndarray: Output values in the linear scale.
-    """
-    p = 1e-6 if stretchpower == 0 else stretchpower
-    y = (1 + p * np.log(10) * x)**(1 / p)
-    return y
-
-
 def do_stretch(x: list | np.ndarray,
                stretch: str, stretchscale: float,
                stretchpower: float) -> np.ndarray:
@@ -133,8 +103,8 @@ def do_stretch(x: list | np.ndarray,
     Args:
         x (list | np.ndarray): Input array in the linear scale.
         stretch (str): 'log', 'asinh', 'power', or 'linear'. Any other means 'linear'. 'log' means the mapped data are logarithmic. 'asinh' means the mapped data are arc sin hyperbolic. 'power' means the mapped data are power-law (see also stretchpower). Defaults to 'linear'.
-        stretchscale (float, optional): The output is asinh(data / stretchscale). Defaults to None.
-        stretchpower (float, optional): The output is ((data / stretchscale)**stretchpower - 1) / stretchpower / ln(10). 1 means the linear scale, while 0 means the logarithmic scale.
+        stretchscale (float): The output is asinh(data / stretchscale).
+        stretchpower (float): The output is data**stretchpower / stretchpower. 1 means the linear scale, while 0 means the logarithmic scale.
 
     Returns:
         np.ndarray: Output stretched array.
@@ -146,7 +116,8 @@ def do_stretch(x: list | np.ndarray,
         case 'asinh':
             t = np.arcsinh(t / stretchscale)
         case 'power':
-            t = pow10(t / stretchscale, stretchpower)
+            p = 1e-6 if stretchpower == 0 else stretchpower
+            t = t**p / p
     return t
 
 
@@ -158,8 +129,8 @@ def undo_stretch(x: list | np.ndarray,
     Args:
         x (list | np.ndarray): Input stretched array.
         stretch (str): 'log', 'asinh', 'power', or 'linear'. Any other means 'linear'. 'log' means the mapped data are logarithmic. 'asinh' means the mapped data are arc sin hyperbolic. 'power' means the mapped data are power-law (see also stretchpower). Defaults to 'linear'.
-        stretchscale (float, optional): The input is asinh(data / stretchscale). Defaults to None.
-        stretchpower (float, optional): The input is ((data / stretchscale)**stretchpower - 1) / stretchpower / ln(10). 1 means the linear scale, while 0 means the logarithmic scale.
+        stretchscale (float): The input is asinh(data / stretchscale).
+        stretchpower (float): The input is data**stretchpower / stretchpower. 1 means the linear scale, while 0 means the logarithmic scale.
 
     Returns:
         np.ndarray: Output array in the linear scale.
@@ -171,7 +142,8 @@ def undo_stretch(x: list | np.ndarray,
         case 'asinh':
             t = np.sinh(t) * stretchscale
         case 'power':
-            t = ipow10(t, stretchpower) * stretchscale
+            p = 1e-6 if stretchpower == 0 else stretchpower
+            t = (t * p)**(1 / p)
     return t
 
 
@@ -285,9 +257,9 @@ def set_minmax(data: np.ndarray, stretch: str, stretchscale: float,
 
     Args:
         data (np.ndarray): Plotted data.
-        stretch (str): 'log', 'asinh', 'power'. Any other means linear.
-        stretchscale (float): For the arcsinh strech and the power stretch.
-        stretchpower (float): For the power strech.
+        stretch (str): 'log', 'asinh', 'power', or 'linear'. Any other means 'linear'. 'log' means the mapped data are logarithmic. 'asinh' means the mapped data are arc sin hyperbolic. 'power' means the mapped data are power-law (see also stretchpower). Defaults to 'linear'.
+        stretchscale (float): The input is asinh(data / stretchscale).
+        stretchpower (float): The input is data**stretchpower / stretchpower. 1 means the linear scale, while 0 means the logarithmic scale.
         sigma (float): Noise level.
         kw (dict): Probably like {'vmin':0, 'vmax':1}.
 
@@ -308,8 +280,10 @@ def set_minmax(data: np.ndarray, stretch: str, stretchscale: float,
     for k in ['vmin', 'vmax']:
         if k not in kw:
             kw[k] = [None] * n
-    stretch = np.where(np.equal(stretch, None), sigma, stretch)
-    cmin = np.where(np.equal(kw['vmin'], None), sigma, kw['vmin'])
+    isnan = np.equal(stretchscale, None)
+    stretchscale = np.where(isnan, sigma, stretchscale)
+    isnan = np.equal(kw['vmin'], None)
+    cmin = np.where(isnan, sigma, kw['vmin'])
 
     argslist = (stretch, stretchscale, stretchpower)
     for i, stretch_args in enumerate(zip(*argslist)):
@@ -344,8 +318,7 @@ def kwargs2AstroData(kw: dict) -> AstroData:
     d = AstroData(data=np.zeros((2, 2)))
     for k in vars(d):
         if k in kw:
-            tmp[k] = kw[k]
-            del kw[k]
+            tmp[k] = kw.pop(k)
     if tmp == {}:
         print('No argument given.')
         return None
@@ -387,8 +360,7 @@ def kwargs2PlotAxes2D(kw: dict) -> PlotAxes2D:
     d = PlotAxes2D()
     for k in vars(d):
         if k in kw:
-            tmp[k] = kw[k]
-            del kw[k]
+            tmp[k] = kw.pop(k)
     d = PlotAxes2D(**tmp)
     return d
 
@@ -405,11 +377,8 @@ def kwargs2beamargs(kw: dict) -> dict:
     tmp = {}
     for k in ['show_beam', 'beamcolor', 'beampos']:
         if k in kw:
-            tmp[k] = kw[k]
-            del kw[k]
-    if 'beam_kwargs' in kw:
-        tmp.update(kw['beam_kwargs'])
-        del kw['beam_kwargs']
+            tmp[k] = kw.pop(k)
+    tmp.update(kw.pop('beam_kwargs', {}))
     return tmp
 
 
@@ -596,16 +565,8 @@ class PlotAstroData(AstroFrame):
         """
         beam_kwargs = kwargs2beamargs(kw)
         self._kw.update(kw)
-        if 'xskip' in self._kw:
-            xskip = self._kw['xskip']
-            del self._kw['xskip']
-        else:
-            xskip = 1
-        if 'yskip' in self._kw:
-            yskip = self._kw['yskip']
-            del self._kw['yskip']
-        else:
-            yskip = 1
+        xskip = self._kw.pop('xskip', 1)
+        yskip = self._kw.pop('yskip', 1)
         d = kwargs2AstroData(self._kw)
         self.read(d, xskip, yskip)
         self.beam = d.beam
@@ -734,6 +695,11 @@ class PlotAstroData(AstroFrame):
         """
         _kw = {'color': 'gray', 'fontsize': 15, 'ha': 'center',
                'va': 'center', 'zorder': 10}
+        subkeys = {'ha': 'horizontalalignment',
+                   'va': 'verticalalignment'}
+        for short, long in subkeys.items():
+            if long in kwargs:
+                kwargs[short] = kwargs.pop(long)
         _kw.update(kwargs)
         if include_chan is None:
             include_chan = self.allchan
@@ -818,7 +784,7 @@ class PlotAstroData(AstroFrame):
             linewidth (float, optional): Width of the bar. Defaults to 3.
         """
         if length == 0:
-            print('Please input length to add_scalebar().')
+            print('No length is given. Skip add_scalebar().')
             return
         if fontsize is None:
             fontsize = 20 if len(self.ax) == 1 else 15
@@ -861,20 +827,21 @@ class PlotAstroData(AstroFrame):
             cbticks = do_stretch(cbticks, *stretch_args)
         if cbticks is None and stretch == 'log':
             cbticks, cbticklabels = logcbticks()
-        if cbticks is not None:
-            cond = (vmin < cbticks) * (cbticks < vmax)
-            cb.set_ticks(cbticks[cond])
-            if cbticklabels is not None:
-                tl = np.array(cbticklabels)[cond]
-            else:
-                t = undo_stretch(cb.get_ticks(), *stretch_args)
-                tl = [f'{d:{cbformat[1:]}}' for d in t]
-            cb.set_ticklabels(tl)
+        if cbticks is None:
+            cbticks = cb.get_ticks()
+        cond = (vmin < cbticks) * (cbticks < vmax)
+        cb.set_ticks(cbticks[cond])
+        if cbticklabels is not None:
+            tl = np.array(cbticklabels)[cond]
+        else:
+            t = undo_stretch(cb.get_ticks(), *stretch_args)
+            tl = [f'{d:{cbformat[1:]}}' for d in t]
+        cb.set_ticklabels(tl)
 
     def add_color(self,
                   stretch: str = 'linear',
                   stretchscale: float | None = None,
-                  stretchpower: float = 1,
+                  stretchpower: float = 0.5,
                   show_cbar: bool = True,
                   cblabel: str | None = None,
                   cbformat: float = '%.1e',
@@ -887,7 +854,7 @@ class PlotAstroData(AstroFrame):
         Args:
             stretch (str, optional): 'log', 'asinh', 'power', or 'linear'. Any other means 'linear'. 'log' means the mapped data are logarithmic. 'asinh' means the mapped data are arc sin hyperbolic. 'power' means the mapped data are power-law (see also stretchpower). Defaults to 'linear'.
             stretchscale (float, optional): Color scale is asinh(data / stretchscale). Defaults to None.
-            stretchpower (float, optional): Color scale is ((data / stretchscale)**stretchpower - 1) / stretchpower / ln(10). 1 means the linear scale, while 0 means the logarithmic scale. Defaults to 1.
+            stretchpower (float, optional): Color scale is data**stretchpower / stretchpower. 1 means the linear scale, while 0 means the logarithmic scale. Defaults to 0.5.
             show_cbar (bool, optional): Show color bar. Defaults to True.
             cblabel (str, optional): Colorbar label. Defaults to None.
             cbformat (float, optional): Format for ticklabels of colorbar. Defaults to '%.1e'.
@@ -905,7 +872,9 @@ class PlotAstroData(AstroFrame):
 
         if cblabel is None:
             cblabel = bunit
-        
+        if stretchscale is None:
+            stretchscale = sigma
+
         stretch_args = (stretch, stretchscale, stretchpower)
         c = set_minmax(c, *stretch_args, sigma, _kw)
         c = self.vskipfill(c, v)
@@ -1018,7 +987,7 @@ class PlotAstroData(AstroFrame):
         Args:
             stretch (str, optional): 'log', 'asinh', 'power', or 'linear'. Any other means 'linear'. 'log' means the mapped data are logarithmic. 'asinh' means the mapped data are arc sin hyperbolic. 'power' means the mapped data are power-law (see also stretchpower). Defaults to 'linear'.
             stretchscale (float, optional): Color scale is asinh(data / stretchscale). Defaults to None.
-            stretchpower (float, optional): Color scale is ((data / stretchscale)**stretchpower - 1) / stretchpower / ln(10). 1 means the linear scale, while 0 means the logarithmic scale. Defaults to 1.
+            stretchpower (float, optional): Color scale is data**stretchpower / stretchpower. 1 means the linear scale, while 0 means the logarithmic scale. Defaults to 1.
         """
         from PIL import Image
 
@@ -1248,7 +1217,8 @@ class PlotAstroData(AstroFrame):
             tuple: (fig, ax)
         """
         if len(self.ax) > 1:
-            print('get_figax is not supported with channel maps')
+            print('PlotAstroData.get_figax() is not supported'
+                  + ' with channel maps')
             return
 
         fig = plt.figure(0) if self.fig is None else self.fig

{plotastrodata-1.8.8 → plotastrodata-1.8.10/plotastrodata.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: plotastrodata
-Version: 1.8.8
+Version: 1.8.10
 Summary: plotastrodata is a tool for astronomers to create figures from FITS files and perform fundamental data analyses with ease.
 Home-page: https://github.com/yusukeaso-astron/plotastrodata
 Download-URL: https://github.com/yusukeaso-astron/plotastrodata