scitex 2.8.1__py3-none-any.whl → 2.10.2__py3-none-any.whl

scitex/io/_save_modules/__init__.py

@@ -1,5 +1,4 @@
 #!/usr/bin/env python3
-# -*- coding: utf-8 -*-
 # Timestamp: "2025-06-12 13:05:00 (ywatanabe)"
 # File: /ssh:sp:/home/ywatanabe/proj/.claude-worktree/scitex_repo/src/scitex/io/_save_modules/__init__.py
 # ----------------------------------------
@@ -18,37 +17,85 @@ to that specific format.
 """
 
 # Import save functions from individual modules
+from ._bibtex import save_bibtex
+from ._catboost import _save_catboost as save_catboost
 from ._csv import _save_csv as save_csv
 from ._excel import save_excel
-from ._numpy import _save_npy as save_npy, _save_npz as save_npz
-from ._pickle import (
-    _save_pickle as save_pickle,
-    _save_pickle_gz as save_pickle_compressed,
-)
-from ._joblib import _save_joblib as save_joblib
-from ._torch import _save_torch as save_torch
+from ._figure_utils import get_figure_with_data
+
+# Optional: image (requires plotly)
+try:
+    from ._image import save_image
+except ImportError:
+    save_image = None
+
+# Optional: HTML (requires plotly)
+try:
+    from ._html import save_html
+except ImportError:
+    save_html = None
+
+# Optional: HDF5 (requires h5py)
+try:
+    from ._hdf5 import _save_hdf5 as save_hdf5
+except ImportError:
+    save_hdf5 = None
+
+# Import extracted utilities
+from ._image_csv import handle_image_with_csv
 from ._json import _save_json as save_json
-from ._yaml import _save_yaml as save_yaml
-from ._hdf5 import _save_hdf5 as save_hdf5
-from ._matlab import _save_matlab as save_matlab
-from ._catboost import _save_catboost as save_catboost
-from ._text import _save_text as save_text
-from ._tex import _save_tex as save_tex
-from ._html import save_html
-from ._image import save_image
-from ._mp4 import _mk_mp4 as save_mp4
-from ._zarr import _save_zarr as save_zarr
-from ._bibtex import save_bibtex
+from ._legends import save_separate_legends
 
 # Import additional save utilities
 from ._listed_dfs_as_csv import _save_listed_dfs_as_csv as save_listed_dfs_as_csv
 from ._listed_scalars_as_csv import (
     _save_listed_scalars_as_csv as save_listed_scalars_as_csv,
 )
+from ._mp4 import _mk_mp4 as save_mp4
+from ._numpy import _save_npy as save_npy
+from ._numpy import _save_npz as save_npz
 from ._optuna_study_as_csv_and_pngs import save_optuna_study_as_csv_and_pngs
+from ._pickle import (
+    _save_pickle as save_pickle,
+)
+from ._pickle import (
+    _save_pickle_gz as save_pickle_compressed,
+)
+from ._pltz_bundle import save_pltz_bundle
+from ._pltz_stx import save_pltz_as_stx
+from ._stx_bundle import save_stx_bundle
+from ._symlink import symlink, symlink_to
+from ._tex import _save_tex as save_tex
+from ._text import _save_text as save_text
+from ._yaml import _save_yaml as save_yaml
+
+# Optional: joblib (requires joblib)
+try:
+    from ._joblib import _save_joblib as save_joblib
+except ImportError:
+    save_joblib = None
+
+# Optional: matlab (requires scipy)
+try:
+    from ._matlab import _save_matlab as save_matlab
+except ImportError:
+    save_matlab = None
+
+# Optional: torch (requires torch)
+try:
+    from ._torch import _save_torch as save_torch
+except ImportError:
+    save_torch = None
+
+# Optional: zarr (requires zarr)
+try:
+    from ._zarr import _save_zarr as save_zarr
+except ImportError:
+    save_zarr = None
 
 # Define what gets imported with "from scitex.io._save_modules import *"
 __all__ = [
+    # Core save functions
     "save_csv",
     "save_excel",
     "save_npy",
@@ -72,6 +119,16 @@ __all__ = [
     "save_listed_dfs_as_csv",
     "save_listed_scalars_as_csv",
     "save_optuna_study_as_csv_and_pngs",
+    # Bundle save functions
+    "save_stx_bundle",
+    "save_pltz_bundle",
+    "save_pltz_as_stx",
+    "save_separate_legends",
+    "handle_image_with_csv",
+    # Utilities
+    "get_figure_with_data",
+    "symlink",
+    "symlink_to",
 ]
 
 # EOF

scitex/io/_save_modules/_figure_utils.py

@@ -0,0 +1,90 @@
+#!/usr/bin/env python3
+# Timestamp: 2025-12-19
+# File: /home/ywatanabe/proj/scitex-code/src/scitex/io/_save_modules/_figure_utils.py
+
+"""Utility functions for extracting figure data for CSV export."""
+
+
+def get_figure_with_data(obj):
+    """
+    Extract figure or axes object that may contain plotting data for CSV export.
+
+    Parameters
+    ----------
+    obj : various matplotlib objects
+        Could be Figure, Axes, FigWrapper, AxisWrapper, or other matplotlib objects
+
+    Returns
+    -------
+    object or None
+        Figure or axes object that has export_as_csv methods, or None if not found
+    """
+    import matplotlib.axes
+    import matplotlib.figure
+    import matplotlib.pyplot as plt
+
+    # Check if object already has export methods (SciTeX wrapped objects)
+    if hasattr(obj, "export_as_csv"):
+        return obj
+
+    # Handle matplotlib Figure objects
+    if isinstance(obj, matplotlib.figure.Figure):
+        # Get the current axes that might be wrapped with SciTeX functionality
+        current_ax = plt.gca()
+        if hasattr(current_ax, "export_as_csv"):
+            return current_ax
+
+        # Check all axes in the figure
+        for ax in obj.axes:
+            if hasattr(ax, "export_as_csv"):
+                return ax
+
+        return None
+
+    # Handle matplotlib Axes objects
+    if isinstance(obj, matplotlib.axes.Axes):
+        if hasattr(obj, "export_as_csv"):
+            return obj
+        return None
+
+    # Handle FigWrapper or similar SciTeX objects
+    if hasattr(obj, "figure") and hasattr(obj.figure, "axes"):
+        # Check if the wrapper itself has export methods
+        if hasattr(obj, "export_as_csv"):
+            return obj
+
+        # Check the underlying figure's axes
+        for ax in obj.figure.axes:
+            if hasattr(ax, "export_as_csv"):
+                return ax
+
+        return None
+
+    # Handle AxisWrapper or similar SciTeX objects
+    if hasattr(obj, "_axis_mpl") or hasattr(obj, "_ax"):
+        if hasattr(obj, "export_as_csv"):
+            return obj
+        return None
+
+    # Try to get the current figure and its axes as fallback
+    try:
+        current_fig = plt.gcf()
+        current_ax = plt.gca()
+
+        if hasattr(current_ax, "export_as_csv"):
+            return current_ax
+        elif hasattr(current_fig, "export_as_csv"):
+            return current_fig
+
+        # Check all axes in current figure
+        for ax in current_fig.axes:
+            if hasattr(ax, "export_as_csv"):
+                return ax
+
+    except:
+        pass
+
+    return None
+
+
+# EOF