plotastrodata 1.9.20__tar.gz → 1.10.0__tar.gz

{plotastrodata-1.9.20/plotastrodata.egg-info → plotastrodata-1.10.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: plotastrodata
-Version: 1.9.20
+Version: 1.10.0
 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
@@ -14,12 +14,13 @@ License-File: LICENSE
 Requires-Dist: astropy>=7.2
 Requires-Dist: corner
 Requires-Dist: dynesty
-Requires-Dist: emcee
+Requires-Dist: emcee>=3.1.5
 Requires-Dist: matplotlib
 Requires-Dist: numpy>=2.0
 Requires-Dist: pillow
 Requires-Dist: plotly
 Requires-Dist: ptemcee
+Requires-Dist: pydantic>=2
 Requires-Dist: scikit-image
 Requires-Dist: scipy
 Requires-Dist: tqdm
@@ -84,13 +85,14 @@ plotastrodata can do the following things.
 * astropy >= 7.2
 * corner (only for fitting)
 * dynesty (only for fitting)
-* emcee >= 3.0 (only for fitting)
+* emcee >= 3.1.5 (only for fitting)
 * ffmpeg (only for movie)
 * matplotlib
 * numpy >= 2.0
 * pillow (only for RGB figures)
 * plotly (only for html cube)
 * ptemcee (only for fitting)
+* pydantic >= 2
 * scikit-image (only for html cube)
 * scipy
 * tqdm (only for fitting)

{plotastrodata-1.9.20 → plotastrodata-1.10.0}/README.md

@@ -56,13 +56,14 @@ plotastrodata can do the following things.
 * astropy >= 7.2
 * corner (only for fitting)
 * dynesty (only for fitting)
-* emcee >= 3.0 (only for fitting)
+* emcee >= 3.1.5 (only for fitting)
 * ffmpeg (only for movie)
 * matplotlib
 * numpy >= 2.0
 * pillow (only for RGB figures)
 * plotly (only for html cube)
 * ptemcee (only for fitting)
+* pydantic >= 2
 * scikit-image (only for html cube)
 * scipy
 * tqdm (only for fitting)

{plotastrodata-1.9.20 → plotastrodata-1.10.0}/plotastrodata/__init__.py

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

{plotastrodata-1.9.20 → plotastrodata-1.10.0}/plotastrodata/analysis_utils.py

@@ -3,7 +3,8 @@ import warnings
 from dataclasses import dataclass
 from scipy.interpolate import RegularGridInterpolator as RGI
 from scipy.signal import convolve
-from typing import Any, Callable
+from pydantic.dataclasses import dataclass as pydantic_dataclass
+from typing import Any, Callable, Literal
 
 from plotastrodata import const_utils as cu
 from plotastrodata.coord_utils import coord2xy, rel2abs, xy2coord
@@ -621,7 +622,7 @@ ASTRODATA_ARGS = ['fitsimage', 'data', 'Tb', 'sigma', 'center', 'restfreq',
                   'beam_org', 'fitsheader', 'pv', 'pvpa']
 
 
-@dataclass
+@pydantic_dataclass
 class AstroFrame():
     """Parameter set to limit and reshape the data in the AstroData format.
 
@@ -662,7 +663,7 @@ class AstroFrame():
     yflip: bool = False
     swapxy: bool = False
     pv: bool = False
-    quadrants: str | None = None
+    quadrants: Literal['13', '24'] | None = None
 
     def __post_init__(self) -> None:
         self.xdir = -1 if self.xflip else 1

{plotastrodata-1.9.20 → plotastrodata-1.10.0}/plotastrodata/fitting_utils.py

@@ -2,9 +2,11 @@ import corner
 import emcee
 import matplotlib.pyplot as plt
 import numpy as np
+import pickle
 import ptemcee
 import warnings
 from dynesty import DynamicNestedSampler as DNS
+from functools import partial
 from multiprocessing import Pool
 from tqdm import tqdm
 from typing import Any, Callable
@@ -18,6 +20,25 @@ bar = None
 global_progressbar = True
 
 
+def _gaussian_log_likelihood(x: np.ndarray, model: Callable,
+                             xdata: np.ndarray, ydata: np.ndarray,
+                             sigma: float | np.ndarray) -> float:
+    """Return a Gaussian log likelihood using pickleable arguments."""
+    chi2 = np.sum((ydata - model(xdata, *x))**2 / sigma**2)
+    return chi2 / (-2)
+
+
+def _bounded_log_probability(x: np.ndarray, log_likelihood: Callable,
+                             bounds: np.ndarray,
+                             update_progress: bool = False) -> float:
+    """Combine a bounded uniform prior with a log likelihood."""
+    if update_progress:
+        bar.update(1)
+    if np.all((bounds[:, 0] < x) & (x < bounds[:, 1])):
+        return log_likelihood(x)
+    return -np.inf
+
+
 def logp(x: np.ndarray) -> float:
     """Log prior function made from the boundary (global_bounds) of fitting parameters.
 
@@ -95,9 +116,8 @@ class EmceeCorner():
         if logl is None and (model is not None
                              and xdata is not None
                              and ydata is not None):
-            def logl(x: np.ndarray) -> float:
-                chi2 = np.sum((ydata - model(xdata, *x))**2 / sigma**2)
-                return chi2 / (-2)
+            logl = partial(_gaussian_log_likelihood, model=model,
+                           xdata=xdata, ydata=ydata, sigma=sigma)
         self.bounds = global_bounds
         self.dim = len(self.bounds)
         self.logl = logl
@@ -125,21 +145,27 @@ class EmceeCorner():
                               'logl': self.logl, 'logp': self.logp}
         else:
             if ncores > 1:
-                print('Use logl as log_prob_fn to avoid function-in-function.')
-                log_prob_fn = self.logl
-            else:
-                def log_prob_fn(x: np.ndarray) -> float:
-                    return self.logp(x) + self.logl(x)
+                try:
+                    pickle.dumps(self.logl)
+                except (pickle.PicklingError, AttributeError, TypeError) as exc:
+                    raise TypeError(
+                        'logl and model must be pickleable when ncores > 1. '
+                        'Define them at module scope instead of inside another '
+                        'function.') from exc
 
             sampler_cls = emcee.EnsembleSampler
             sampler_kwargs = {'nwalkers': nwalkers, 'ndim': self.dim,
-                              'log_prob_fn': log_prob_fn}
+                              'log_prob_fn': _bounded_log_probability,
+                              'args': (self.logl, self.bounds,
+                                       global_progressbar and ncores == 1)}
         if ncores > 1:
             with Pool(ncores) as pool:
                 sampler = sampler_cls(**sampler_kwargs, pool=pool)
+                # This run_mcmc is duplicated so sampling finishes before the pool closes.
+                sampler.run_mcmc(pos0, nsteps)
         else:
             sampler = sampler_cls(**sampler_kwargs, pool=None)
-        sampler.run_mcmc(pos0, nsteps)
+            sampler.run_mcmc(pos0, nsteps)
         return sampler
 
     def _get_samples(self, sampler: Any, nburnin: int,
@@ -190,7 +216,7 @@ class EmceeCorner():
             ntry (int, optional): Number of trials for the Gelman-Rubin check. Defaults to 1.
             pos0 (np.nparray, optional): Initial parameter set in the shape of (ntemps, nwalkers, dim). Defaults to None.
             savechain (str, optional): File name of the chain in format of .npy. Existing files with the same name are overwritten by ``numpy.save``. Defaults to None.
-            ncores (int, optional): Number of cores for multiprocessing.Pool. ncores=1 does not use multiprocessing. Defaults to 1.
+            ncores (int, optional): Number of cores for multiprocessing.Pool. ncores=1 does not use multiprocessing. For ncores > 1, user-supplied logl and model functions must be pickleable, such as functions defined at module scope. Defaults to 1.
             grcheck (bool, optional): Whether to check Gelman-Rubin statistics. Defaults to False.
             pt (bool, optional): Whether to use ptemcee; otherwise, emcee is used. Defaults to False.
         """
@@ -243,13 +269,14 @@ class EmceeCorner():
             labels = [f'Par {i:d}' for i in range(self.dim)]
         if cornerrange is None:
             cornerrange = self.bounds
-        corner.corner(np.reshape(self.samples, (-1, self.dim)),
-                      truths=self.popt,
-                      quantiles=[self.percent[0] / 100,
-                                 0.5,
-                                 self.percent[1] / 100],
-                      show_titles=True, labels=labels, range=cornerrange)
-        close_figure(plt, savefig, show, tight=False)
+        fig = corner.corner(np.reshape(self.samples, (-1, self.dim)),
+                            truths=self.popt,
+                            quantiles=[self.percent[0] / 100,
+                                       0.5,
+                                       self.percent[1] / 100],
+                            show_titles=True, labels=labels,
+                            range=cornerrange)
+        close_figure(fig, savefig, show, tight=False)
 
     def plotchain(self, labels: list | None = None, ylim: list | None = None,
                   savefig: dict | str | None = None,

{plotastrodata-1.9.20 → plotastrodata-1.10.0}/plotastrodata/other_utils.py

@@ -296,7 +296,7 @@ def close_figure(fig: object, savefig: dict | str | None = None,
                  show: bool = False, tight: bool = True) -> None:
     """Save, show, and close the figure.
 
-    If ``savefig`` is provided, the figure is saved with Matplotlib ``Figure.savefig``. Existing files with the same name are overwritten by Matplotlib. After optional saving/showing, the figure is closed with ``plt.close()``.
+    If ``savefig`` is provided, the figure is saved with Matplotlib ``Figure.savefig``. Existing files with the same name are overwritten by Matplotlib. After optional saving/showing, the figure is closed with ``plt.close(fig)``.
 
     Default keyword values:
         Figure.savefig: ``bbox_inches='tight'`` and ``transparent=True``. Values in ``savefig`` override these defaults.
@@ -316,4 +316,4 @@ def close_figure(fig: object, savefig: dict | str | None = None,
         fig.savefig(**savefig0)
     if show:
         plt.show()
-    plt.close()
+    plt.close(fig)

{plotastrodata-1.9.20 → plotastrodata-1.10.0}/plotastrodata/plot_utils.py

@@ -1,9 +1,10 @@
 import matplotlib as mpl
 import matplotlib.pyplot as plt
 import numpy as np
-from dataclasses import dataclass
 from matplotlib.patches import Ellipse, Rectangle
-from typing import Any, TypeVar, Callable
+from pydantic import Field, field_validator
+from pydantic.dataclasses import dataclass as pydantic_dataclass
+from typing import Any, Callable, Literal, TypeVar
 
 from plotastrodata.analysis_utils import AstroData, AstroFrame
 from plotastrodata.coord_utils import (coord2xy, xy2coord,
@@ -17,6 +18,13 @@ from plotastrodata.other_utils import (close_figure, listing,
 plt.ioff()  # force to turn off interactive mode
 
 T = TypeVar('T')
+Stretch = Literal['linear', 'log', 'asinh', 'power']
+AxisScale = Literal['linear', 'log', 'symlog', 'asinh', 'logit']
+FloatOrList = float | list[float]
+OptionalFloatOrList = float | list[float | None] | None
+BeamTriple = list[float | None]
+BeamValue = BeamTriple | list[BeamTriple]
+BeamPosition = list[float] | list[list[float] | None] | None
 
 
 def set_rcparams(fontsize: int = 18, nancolor: str = 'w',
@@ -187,24 +195,24 @@ def _get_vskipfill(nv: int, v_org: np.ndarray, vskip: int,
     return vskipfill
 
 
-@dataclass
+@pydantic_dataclass
 class Stretcher():
     """Arguments and methods related to the stretch in PlotAstroData.add_color() and add_rgb().
 
         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): The output is asinh(data / stretchscale). Defaults to None.
-            stretchpower (float, optional): The output is data**stretchpower / stretchpower. 1 means the linear scale, while 0 means the logarithmic scale. Defaults to 0.5.
-            vmin (float, optional): The minimum value for Axes.pcolormesh() of matplotlib. Defaults to None.
-            vmax (float, optional): The maximum value for Axes.pcolormesh() of matplotlib. Defaults to None.
-            sigma (float, optional): Noise level. Defaults to 0.
+            stretchscale (float or list, optional): The output is asinh(data / stretchscale). Defaults to None.
+            stretchpower (float or list, optional): The output is data**stretchpower / stretchpower. 1 means the linear scale, while 0 means the logarithmic scale. Defaults to 0.5.
+            vmin (float or list, optional): The minimum value for Axes.pcolormesh() of matplotlib. Defaults to None.
+            vmax (float or list, optional): The maximum value for Axes.pcolormesh() of matplotlib. Defaults to None.
+            sigma (float, list, or None, optional): Noise level. None means no noise level is used. Defaults to 0.
     """
-    stretch: str = 'linear'
-    stretchscale: float | None = None
-    stretchpower: float = 0.5
-    vmin: float | None = None
-    vmax: float | None = None
-    sigma: float = 0
+    stretch: Stretch | list[Stretch] = 'linear'
+    stretchscale: OptionalFloatOrList = None
+    stretchpower: FloatOrList = 0.5
+    vmin: OptionalFloatOrList = None
+    vmax: OptionalFloatOrList = None
+    sigma: OptionalFloatOrList = 0
 
     def __post_init__(self) -> None:
         self.n = 1 if isinstance(self.stretch, str) else len(self.stretch)
@@ -301,27 +309,88 @@ class Stretcher():
         return dataout, vminout, vmaxout
 
 
+@pydantic_dataclass
 class Beam():
     """Arguments for PlotAstroData.add_beam().
 
         Args:
             show_beam (bool, optional): Defaults to True.
-            beam (list, optional): [bmaj, bmin, bpa]. This may be a list of list. Defaults to [None, None, None].
-            beamcolor (str, optional): matplotlib color. This may be a list of str. Defaults to 'gray'.
-            beampos (list, optional): Relative position. This may be a list of list or a list of None. Defaults to None.
-            beam_kwargs (dict, optional): Additional arguments for matplotlib.patches. Defaults to {}.
+            beam (list or np.ndarray, optional): One ``[bmaj, bmin, bpa]`` triple or a list of triples. ``bmaj`` and ``bmin`` must be positive; ``bpa`` is in degrees. ``[None, None, None]`` means no beam is available. Partial triples are not supported. Defaults to ``[None, None, None]``.
+            beamcolor (str or list of str, optional): Matplotlib color. One color is used for every beam; when a list is given, provide one color per beam. Defaults to ``'gray'``.
+            beampos (list or list of list, optional): One relative ``[x, y]`` position is used for every beam; when a list of positions is given, provide one position per beam. Each coordinate must be from 0 (left or bottom) to 1 (right or top). None selects the automatic position. Defaults to None.
+            beam_kwargs (dict, optional): Additional Matplotlib patch arguments. Defaults to {}.
     """
-    def __init__(self,
-                 show_beam: bool = True,
-                 beam: list[float | None] = [None] * 3,
-                 beamcolor: str = 'gray',
-                 beampos: list[float] | None = None,
-                 beam_kwargs: dict = {}) -> None:
-        self.show_beam = show_beam
-        self.beam = beam
-        self.beamcolor = beamcolor
-        self.beampos = beampos
-        self.beam_kwargs = beam_kwargs
+    show_beam: bool = True
+    beam: BeamValue = Field(default_factory=lambda: [None] * 3)
+    beamcolor: str | list[str] = 'gray'
+    beampos: BeamPosition = None
+    beam_kwargs: dict[str, Any] = Field(default_factory=dict)
+
+    @field_validator('beam', mode='before')
+    @classmethod
+    def _convert_beam_arrays(cls, value: Any) -> Any:
+        """Convert documented NumPy beam inputs to the canonical list form.
+
+        A single NumPy triple becomes a list, and NumPy triples inside a
+        list of beams are converted individually. This keeps subsequent
+        validation and plotting code independent of the input container.
+        """
+        if isinstance(value, np.ndarray):
+            value = value.tolist()
+        if isinstance(value, list):
+            return [item.tolist() if isinstance(item, np.ndarray) else item
+                    for item in value]
+        return value
+
+    @field_validator('beam')
+    @classmethod
+    def _validate_beam(cls, value: BeamValue) -> BeamValue:
+        beams = value if value and isinstance(value[0], list) else [value]
+        for beam in beams:
+            if len(beam) != 3:
+                raise ValueError('Each beam must contain [bmaj, bmin, bpa].')
+            if all(component is None for component in beam):
+                continue
+            if any(component is None for component in beam):
+                raise ValueError('A beam must be complete or all None.')
+            if not all(np.isfinite(component) for component in beam):
+                raise ValueError('Beam values must be finite.')
+            if beam[0] <= 0 or beam[1] <= 0:
+                raise ValueError('bmaj and bmin must be positive.')
+        return value
+
+    @field_validator('beampos')
+    @classmethod
+    def _validate_beampos(cls, value: BeamPosition) -> BeamPosition:
+        if value is None:
+            return value
+        positions = [value] if cls._is_position(value) else value
+        for position in positions:
+            if position is None:
+                continue
+            if len(position) != 2:
+                raise ValueError('Each beam position must contain [x, y].')
+            if not all(0 <= coordinate <= 1 for coordinate in position):
+                raise ValueError('Beam positions must be between 0 and 1.')
+        return value
+
+    @staticmethod
+    def _is_position(value: Any) -> bool:
+        return (isinstance(value, list) and len(value) == 2
+                and all(isinstance(coordinate, float)
+                        for coordinate in value))
+
+    def validate_display(self) -> None:
+        """Validate color and position lists against the number of beams."""
+        multi = self.beam and isinstance(self.beam[0], list)
+        beams = self.beam if multi else [self.beam]
+        nbeams = len(beams)
+        if isinstance(self.beamcolor, list) and len(self.beamcolor) != nbeams:
+            raise ValueError('Provide one beamcolor for each beam.')
+        if (self.beampos is not None
+                and not self._is_position(self.beampos)
+                and len(self.beampos) != nbeams):
+            raise ValueError('Provide one beampos for each beam.')
 
     def todict(self) -> dict[str, Any]:
         """Return beam display settings as a dictionary.
@@ -337,13 +406,13 @@ class Beam():
         return tmp
 
 
-@dataclass
+@pydantic_dataclass
 class PlotAxes2D():
     """Use Axes.set_* to adjust x and y axes.
 
     Args:
         samexy (bool, optional): True supports same ticks between x and y. Defaults to True.
-        loglog (float, optional): If a float is given, plot on a log-log plane, and xim=(xmax / loglog, xmax) and so does ylim. Defaults to None.
+        loglog (float, optional): A positive value plots on a log-log plane, with xim=(xmax / loglog, xmax) and similarly for ylim. Defaults to None.
         xscale (str, optional): ``'log'`` labels decade ticks and ticks near the limits; other intermediate ticks are minor and unlabeled. Defaults to ``'linear'``.
         yscale (str, optional): ``'log'`` labels decade ticks and ticks near the limits; other intermediate ticks are minor and unlabeled. Defaults to ``'linear'``.
         xlim (list, optional): Defaults to None.
@@ -360,9 +429,9 @@ class PlotAxes2D():
         aspect (dict or float, optional): Defaults to None.
     """
     samexy: bool = True
-    loglog: float | None = None
-    xscale: str = 'linear'
-    yscale: str = 'linear'
+    loglog: float | None = Field(default=None, gt=0)
+    xscale: AxisScale = 'linear'
+    yscale: AxisScale = 'linear'
     xlim: list | None = None
     ylim: list | None = None
     xlabel: str | None = None
@@ -564,13 +633,18 @@ class PlotAstroData(AstroFrame):
                               figsize=figsize,
                               ncols=ncols, nrows=nrows, nchan=nchan)
         need_vlabel = nchan > 1 or animation
+        figs = []
         for ch in range(nchan):
             n, i, j = ch2nij(ch)
-            if internalfig and n not in plt.get_fignums():
-                fig = plt.figure(n, figsize=figsize)
+            if n == len(figs):
+                if internalfig:
+                    fig = plt.figure(figsize=figsize)
+                figs.append(fig)
                 if need_vlabel:
                     fig.subplots_adjust(hspace=0, wspace=0,
                                         right=0.87, top=0.87)
+            else:
+                fig = figs[n]
             if internalax:
                 sharex = ax[nij2ch(n, i - 1, j)] if i > 0 else None
                 sharey = ax[nij2ch(n, i, j - 1)] if j > 0 else None
@@ -582,6 +656,7 @@ class PlotAstroData(AstroFrame):
                             rf'${vlabel:.{veldigit}f}$', color='black',
                             backgroundcolor='white', zorder=20)
         self.fig = None if internalfig else fig
+        self.figs = figs
         self.ax = ax
         self.rowcol = nrows * ncols
         self.npages = npages
@@ -612,9 +687,12 @@ class PlotAstroData(AstroFrame):
         self.sigma = d.sigma
         singlepix = d.dx is None or d.dy is None
         if len(d.beam) == 4:
-            b.beam = self.beam = next(b for b in d.beam if None not in b)
+            beam = next((onebeam for onebeam in d.beam
+                         if all(a is not None for a in onebeam)),
+                        [None, None, None],)
         else:
-            b.beam = self.beam = d.beam
+            beam = d.beam
+        b.beam = self.beam = Beam(beam=beam).beam
         self.add_beam(**b.todict())
         return (d.data, d.x, d.y, d.v, d.sigma, d.bunit,
                 self._kw, singlepix)
@@ -657,8 +735,6 @@ class PlotAstroData(AstroFrame):
             for ch, axnow in enumerate(self.ax):
                 if ch not in self._validchan(include_chan):
                     continue
-                if self.fig is None:
-                    plt.figure(ch // self.rowcol)
                 if patch == 'rectangle':
                     a = np.radians(angle)
                     xp = x - (width*np.cos(a) + height*np.sin(a)) / 2.
@@ -674,7 +750,7 @@ class PlotAstroData(AstroFrame):
     def add_beam(self, **kwargs: Any) -> None:
         """Use add_region() to plot the beam.
 
-        kwargs may include the arguments of Beam, except for beam_kwargs, to specify the beam appearance. Those arguments may be a list of each format.
+        kwargs may include the arguments of Beam, except for beam_kwargs, to specify the beam appearance. A single ``beamcolor`` or ``beampos`` value applies to every beam; per-beam lists must provide one value for each beam. A ``[None, None, None]`` beam is skipped.
 
         Default keyword values:
             Beam patch: ``facecolor=beamcolor`` and ``edgecolor=None``. Other keyword arguments override these values and are passed to ``add_region``.
@@ -683,6 +759,7 @@ class PlotAstroData(AstroFrame):
         show_beam, beamcolor, beampos = b.show_beam, b.beamcolor, b.beampos
         beam = b.beam
         del kwargs['beam']
+        b.validate_display()
         if not show_beam:
             return
 
@@ -691,7 +768,7 @@ class PlotAstroData(AstroFrame):
         blist = [beam] if np.ndim(beam) == 1 else beam
         n = len(blist)
         bclist = beamcolor if isinstance(beamcolor, list) else [beamcolor] * n
-        islist = beampos == [None] * 3 or np.ndim(beampos) == 2
+        islist = beampos is not None and not b._is_position(beampos)
         bplist = beampos if islist else [beampos] * n
         for (bmaj, bmin, bpa), bc, bp in zip(blist, bclist, bplist):
             if None in [bmaj, bmin, bpa]:
@@ -857,16 +934,13 @@ class PlotAstroData(AstroFrame):
         if not show_cbar:
             return
 
-        if self.fig is None:
-            fig = plt.figure(ch // self.rowcol)
-        else:
-            fig = self.fig
+        fig = self.figs[ch // self.rowcol]
         if len(self.ax) == 1:
             ax = self.ax[ch]
             cb = fig.colorbar(mappable[ch], ax=ax, label=cblabel,
                               format=cbformat, location=cblocation)
         else:
-            cax = plt.axes([0.88, 0.105, 0.015, 0.77])
+            cax = fig.add_axes([0.88, 0.105, 0.015, 0.77])
             cb = fig.colorbar(mappable[ch], cax=cax, label=cblabel,
                               format=cbformat)
         cb.ax.tick_params(labelsize=cbtickfontsize)
@@ -1104,14 +1178,13 @@ class PlotAstroData(AstroFrame):
                 axnow.set_ylabel('')
             if len(self.ax) == 1:
                 if self.fig is None:
-                    plt.figure(0).tight_layout()
+                    self.figs[0].tight_layout()
         if title is not None:
             if len(self.ax) > 1:
                 t = {'y': 0.9}
                 t_in = {'t': title} if isinstance(title, str) else title
                 t.update(t_in)
-                for i in range(self.npages):
-                    fig = plt.figure(i)
+                for fig in self.figs:
                     fig.suptitle(**t)
             else:
                 t = {'label': title} if isinstance(title, str) else title
@@ -1249,13 +1322,13 @@ class PlotAstroData(AstroFrame):
                 show: bool = False, **kwargs: Any) -> None:
         """Use savefig of matplotlib.
 
-        If ``filename`` is provided, existing files with the same name are overwritten by Matplotlib. This method closes all Matplotlib figures with ``plt.close('all')`` after optional saving/showing.
+        If ``filename`` is provided, existing files with the same name are overwritten by Matplotlib. After optional saving/showing, figures managed by this instance are closed.
 
         Default keyword values:
             Figure.savefig: ``transparent=True`` and ``bbox_inches='tight'``. User-supplied keyword arguments override these values.
 
         Args:
-            filename (str, optional): Output image file name. Existing files may be overwritten, and all Matplotlib figures are closed after saving/showing. Defaults to None.
+            filename (str, optional): Output image file name. Existing files may be overwritten, and figures managed by this instance are closed after saving/showing. Defaults to None.
             show (bool, optional): True means doing plt.show(). Defaults to False.
         """
         _kw = {'transparent': True, 'bbox_inches': 'tight'}
@@ -1265,15 +1338,15 @@ class PlotAstroData(AstroFrame):
             axnow.set_ylim(*self.Ylim)
         if isinstance(filename, str):
             ext = filename.split('.')[-1]
-            for i in range(self.npages):
+            for i, fig in enumerate(self.figs):
                 ver = '' if self.npages == 1 else f'_{i:d}'
-                fig = plt.figure(i)
                 fig.patch.set_alpha(0)
                 fname = filename.replace(f'.{ext}', f'{ver}.{ext}')
                 fig.savefig(fname, **_kw)
         if show:
             plt.show()
-        plt.close('all')
+        for fig in self.figs:
+            plt.close(fig)
 
     def get_figax(self) -> tuple[object, object] | None:
         """Output the external fig and ax after plotting.
@@ -1286,8 +1359,7 @@ class PlotAstroData(AstroFrame):
                   + ' with channel maps')
             return
 
-        fig = plt.figure(0) if self.fig is None else self.fig
-        return fig, self.ax[0]
+        return self.figs[0], self.ax[0]
 
 
 def _get_ylabel_profile(_kw: dict, Tb: bool, flux: bool, bunit: str

{plotastrodata-1.9.20 → plotastrodata-1.10.0/plotastrodata.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: plotastrodata
-Version: 1.9.20
+Version: 1.10.0
 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
@@ -14,12 +14,13 @@ License-File: LICENSE
 Requires-Dist: astropy>=7.2
 Requires-Dist: corner
 Requires-Dist: dynesty
-Requires-Dist: emcee
+Requires-Dist: emcee>=3.1.5
 Requires-Dist: matplotlib
 Requires-Dist: numpy>=2.0
 Requires-Dist: pillow
 Requires-Dist: plotly
 Requires-Dist: ptemcee
+Requires-Dist: pydantic>=2
 Requires-Dist: scikit-image
 Requires-Dist: scipy
 Requires-Dist: tqdm
@@ -84,13 +85,14 @@ plotastrodata can do the following things.
 * astropy >= 7.2
 * corner (only for fitting)
 * dynesty (only for fitting)
-* emcee >= 3.0 (only for fitting)
+* emcee >= 3.1.5 (only for fitting)
 * ffmpeg (only for movie)
 * matplotlib
 * numpy >= 2.0
 * pillow (only for RGB figures)
 * plotly (only for html cube)
 * ptemcee (only for fitting)
+* pydantic >= 2
 * scikit-image (only for html cube)
 * scipy
 * tqdm (only for fitting)

{plotastrodata-1.9.20 → plotastrodata-1.10.0}/plotastrodata.egg-info/requires.txt

@@ -1,12 +1,13 @@
 astropy>=7.2
 corner
 dynesty
-emcee
+emcee>=3.1.5
 matplotlib
 numpy>=2.0
 pillow
 plotly
 ptemcee
+pydantic>=2
 scikit-image
 scipy
 tqdm

{plotastrodata-1.9.20 → plotastrodata-1.10.0}/setup.cfg

@@ -19,12 +19,13 @@ install_requires =
 	astropy >= 7.2
 	corner
 	dynesty
-	emcee
+	emcee >= 3.1.5
 	matplotlib
 	numpy >= 2.0
 	pillow
 	plotly
 	ptemcee
+	pydantic >= 2
 	scikit-image
 	scipy
 	tqdm