batplot 1.7.23__tar.gz → 1.7.24__tar.gz

{batplot-1.7.23/batplot.egg-info → batplot-1.7.24}/PKG-INFO

@@ -1,9 +1,30 @@
 Metadata-Version: 2.4
 Name: batplot
-Version: 1.7.23
+Version: 1.7.24
 Summary: Interactive plotting tool for material science (1D plot) and electrochemistry (GC, CV, dQ/dV, CPC, operando) with batch processing
 Author-email: Tian Dai <tianda@uio.no>
-License-Expression: MIT
+License: MIT License
+        
+        Copyright (c) 2025 Tian Dai
+        
+        Permission is hereby granted, free of charge, to any person obtaining a copy
+        of this software and associated documentation files (the "Software"), to deal
+        in the Software without restriction, including without limitation the rights
+        to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+        copies of the Software, and to permit persons to whom the Software is
+        furnished to do so, subject to the following conditions:
+        
+        The above copyright notice and this permission notice shall be included in all
+        copies or substantial portions of the Software.
+        
+        THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+        IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+        FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+        AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+        LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+        OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+        SOFTWARE.
+        
 Project-URL: Homepage, https://github.com/TianDai1729/batplot
 Project-URL: Repository, https://github.com/TianDai1729/batplot
 Project-URL: Issues, https://github.com/TianDai1729/batplot/issues

{batplot-1.7.23 → batplot-1.7.24}/batplot/__init__.py

@@ -1,5 +1,5 @@
 """batplot: Interactive plotting for battery data visualization."""
 
-__version__ = "1.7.23"
+__version__ = "1.7.24"
 
 __all__ = ["__version__"]

{batplot-1.7.23 → batplot-1.7.24}/batplot/cpc_interactive.py

@@ -79,6 +79,46 @@ def _colorize_menu(text):
     return f"\033[96m{cmd}\033[0m: {desc}"  # Cyan for command, default for description
 
 
+def _color_of(artist):
+    """Return a representative color for a Line2D/PathCollection."""
+    try:
+        if artist is None:
+            return None
+        if hasattr(artist, 'get_color'):
+            c = artist.get_color()
+            if isinstance(c, (list, tuple)) and c and not isinstance(c, str):
+                return c[0]
+            return c
+        if hasattr(artist, 'get_facecolors'):
+            arr = artist.get_facecolors()
+            if arr is not None and len(arr):
+                from matplotlib.colors import to_hex
+                return to_hex(arr[0])
+    except Exception:
+        return None
+    return None
+
+
+def _get_legend_title(fig, default: str = "Legend") -> str:
+    """Fetch stored legend title, falling back to current legend text or default."""
+    try:
+        title = getattr(fig, '_cpc_legend_title', None)
+        if isinstance(title, str) and title:
+            return title
+    except Exception:
+        pass
+    try:
+        for ax in getattr(fig, 'axes', []):
+            leg = ax.get_legend()
+            if leg is not None:
+                t = leg.get_title().get_text()
+                if t:
+                    return t
+    except Exception:
+        pass
+    return default
+
+
 def _colorize_prompt(text):
     """Colorize commands within input prompts. Handles formats like (s=size, f=family, q=return) or (y/n)."""
     import re
@@ -234,7 +274,7 @@ def _rebuild_legend(ax, ax2, file_data):
                 h_all.append(h)
                 l_all.append(l)
         if h_all:
-            _legend_no_frame(ax, h_all, l_all, loc='best', borderaxespad=1.0)
+            _legend_no_frame(ax, h_all, l_all, loc='best', borderaxespad=1.0, title=_get_legend_title(ax.figure))
         else:
             leg = ax.get_legend()
             if leg:
@@ -398,7 +438,8 @@ def _style_snapshot(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_data=Non
         'font': {'family': fam0, 'size': fsize},
         'legend': {
             'visible': legend_visible,
-            'position_inches': legend_xy_in  # [x, y] offset from canvas center in inches
+            'position_inches': legend_xy_in,  # [x, y] offset from canvas center in inches
+            'title': _get_legend_title(fig),
         },
         'ticks': {
             'widths': {
@@ -671,6 +712,8 @@ def _apply_style(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, cfg: Dict, file_
         if leg_cfg:
             leg_visible = leg_cfg.get('visible', True)
             leg_xy_in = leg_cfg.get('position_inches')
+            if 'title' in leg_cfg:
+                fig._cpc_legend_title = leg_cfg.get('title') or _get_legend_title(fig)
             if leg_xy_in is not None:
                 fig._cpc_legend_xy_in = _sanitize_legend_offset(tuple(leg_xy_in))
             leg = ax.get_legend()
@@ -1197,7 +1240,7 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
         try:
             xy_in = _sanitize_legend_offset(getattr(fig, '_cpc_legend_xy_in', None))
             leg = ax.get_legend()
-            if xy_in is None or leg is None:
+            if xy_in is None:
                 return
             # Compute figure-fraction anchor from inches
             fw, fh = fig.get_size_inches()
@@ -1209,7 +1252,16 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
             h1, l1 = ax.get_legend_handles_labels()
             h2, l2 = ax2.get_legend_handles_labels()
             if h1 or h2:
-                _legend_no_frame(ax, h1 + h2, l1 + l2, loc='center', bbox_to_anchor=(fx, fy), bbox_transform=fig.transFigure, borderaxespad=1.0)
+                _legend_no_frame(
+                    ax,
+                    h1 + h2,
+                    l1 + l2,
+                    loc='center',
+                    bbox_to_anchor=(fx, fy),
+                    bbox_transform=fig.transFigure,
+                    borderaxespad=1.0,
+                    title=_get_legend_title(fig),
+                )
         except Exception:
             pass
 
@@ -1680,12 +1732,28 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
                         else:
                             print(f"  {i}: {fname}")
                 
-                fname = input("Export filename (default .svg if no extension) or number to overwrite (q=cancel): ").strip()
+                last_figure_path = getattr(fig, '_last_figure_export_path', None)
+                if last_figure_path:
+                    fname = input("Export filename (default .svg if no extension), number to overwrite, or o to overwrite last (q=cancel): ").strip()
+                else:
+                    fname = input("Export filename (default .svg if no extension) or number to overwrite (q=cancel): ").strip()
                 if not fname or fname.lower() == 'q':
                     _print_menu(); continue
                 
+                # Check for 'o' option
+                if fname.lower() == 'o':
+                    if not last_figure_path:
+                        print("No previous export found.")
+                        _print_menu(); continue
+                    if not os.path.exists(last_figure_path):
+                        print(f"Previous export file not found: {last_figure_path}")
+                        _print_menu(); continue
+                    yn = input(f"Overwrite '{os.path.basename(last_figure_path)}'? (y/n): ").strip().lower()
+                    if yn != 'y':
+                        _print_menu(); continue
+                    target = last_figure_path
                 # Check if user selected a number
-                if fname.isdigit() and files:
+                elif fname.isdigit() and files:
                     idx = int(fname)
                     if 1 <= idx <= len(files):
                         name = files[idx-1]
@@ -1734,6 +1802,7 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
                     # Export the figure
                     fig.savefig(target, bbox_inches='tight')
                     print(f"Exported figure to {target}")
+                    fig._last_figure_export_path = target
                     
                     # Restore original labels
                     if is_multi_file and original_labels:
@@ -1767,10 +1836,28 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
                             print(f"  {i}: {f}  ({timestamp})")
                         else:
                             print(f"  {i}: {f}")
-                prompt = "Enter new filename (no ext needed) or number to overwrite (q=cancel): "
+                last_session_path = getattr(fig, '_last_session_save_path', None)
+                if last_session_path:
+                    prompt = "Enter new filename (no ext needed), number to overwrite, or o to overwrite last (q=cancel): "
+                else:
+                    prompt = "Enter new filename (no ext needed) or number to overwrite (q=cancel): "
                 choice = input(prompt).strip()
                 if not choice or choice.lower() == 'q':
                     _print_menu(); continue
+                if choice.lower() == 'o':
+                    # Overwrite last saved session
+                    if not last_session_path:
+                        print("No previous save found.")
+                        _print_menu(); continue
+                    if not os.path.exists(last_session_path):
+                        print(f"Previous save file not found: {last_session_path}")
+                        _print_menu(); continue
+                    yn = input(f"Overwrite '{os.path.basename(last_session_path)}'? (y/n): ").strip().lower()
+                    if yn != 'y':
+                        _print_menu(); continue
+                    dump_cpc_session(last_session_path, fig=fig, ax=ax, ax2=ax2, sc_charge=sc_charge, sc_discharge=sc_discharge, sc_eff=sc_eff, file_data=file_data, skip_confirm=True)
+                    print(f"Overwritten session to {last_session_path}")
+                    _print_menu(); continue
                 if choice.isdigit() and files:
                     idx = int(choice)
                     if 1 <= idx <= len(files):
@@ -1779,10 +1866,13 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
                         if yn != 'y':
                             _print_menu(); continue
                         target = os.path.join(folder, name)
+                        dump_cpc_session(target, fig=fig, ax=ax, ax2=ax2, sc_charge=sc_charge, sc_discharge=sc_discharge, sc_eff=sc_eff, file_data=file_data, skip_confirm=True)
+                        fig._last_session_save_path = target
+                        _print_menu(); continue
                     else:
                         print("Invalid number.")
                         _print_menu(); continue
-                else:
+                if choice.lower() != 'o':
                     name = choice
                     root, ext = os.path.splitext(name)
                     if ext == '':
@@ -1792,7 +1882,8 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
                         yn = input(f"'{os.path.basename(target)}' exists. Overwrite? (y/n): ").strip().lower()
                         if yn != 'y':
                             _print_menu(); continue
-                dump_cpc_session(target, fig=fig, ax=ax, ax2=ax2, sc_charge=sc_charge, sc_discharge=sc_discharge, sc_eff=sc_eff, file_data=file_data, skip_confirm=True)
+                    dump_cpc_session(target, fig=fig, ax=ax, ax2=ax2, sc_charge=sc_charge, sc_discharge=sc_discharge, sc_eff=sc_eff, file_data=file_data, skip_confirm=True)
+                    fig._last_session_save_path = target
             except Exception as e:
                 print(f"Save failed: {e}")
             _print_menu(); continue
@@ -1916,11 +2007,44 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
                             else:
                                 print(f"  {_i}: {fname}")
                     
-                    sub = input(_colorize_prompt("Style submenu: (e=export, q=return, r=refresh): ")).strip().lower()
+                    last_style_path = getattr(fig, '_last_style_export_path', None)
+                    if last_style_path:
+                        sub = input(_colorize_prompt("Style submenu: (e=export, o=overwrite last, q=return, r=refresh): ")).strip().lower()
+                    else:
+                        sub = input(_colorize_prompt("Style submenu: (e=export, q=return, r=refresh): ")).strip().lower()
                     if sub == 'q':
                         break
                     if sub == 'r' or sub == '':
                         continue
+                    if sub == 'o':
+                        # Overwrite last exported style file
+                        if not last_style_path:
+                            print("No previous export found.")
+                            continue
+                        if not os.path.exists(last_style_path):
+                            print(f"Previous export file not found: {last_style_path}")
+                            continue
+                        yn = input(f"Overwrite '{os.path.basename(last_style_path)}'? (y/n): ").strip().lower()
+                        if yn != 'y':
+                            continue
+                        # Rebuild config based on current state
+                        snap = _style_snapshot(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_data)
+                        # Determine if last export was style-only or style+geometry
+                        try:
+                            with open(last_style_path, 'r', encoding='utf-8') as f:
+                                old_cfg = json.load(f)
+                            if old_cfg.get('kind') == 'cpc_style_geom':
+                                snap['kind'] = 'cpc_style_geom'
+                                snap['geometry'] = _get_geometry_snapshot(ax, ax2)
+                            else:
+                                snap['kind'] = 'cpc_style'
+                        except Exception:
+                            snap['kind'] = 'cpc_style'
+                        with open(last_style_path, 'w', encoding='utf-8') as f:
+                            json.dump(snap, f, indent=2)
+                        print(f"Overwritten style to {last_style_path}")
+                        style_menu_active = False
+                        break
                     if sub == 'e':
                         # Ask for ps or psg
                         print("Export options:")
@@ -1978,10 +2102,42 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
                                     print(f"  {i}: {fname}  ({timestamp})")
                                 else:
                                     print(f"  {i}: {fname}")
-                        choice = input("Enter new filename or number to overwrite (q=cancel): ").strip()
+                        if last_style_path:
+                            choice = input("Enter new filename, number to overwrite, or o to overwrite last (q=cancel): ").strip()
+                        else:
+                            choice = input("Enter new filename or number to overwrite (q=cancel): ").strip()
                         if not choice or choice.lower() == 'q':
                             print("Style export canceled.")
                             continue
+                        if choice.lower() == 'o':
+                            # Overwrite last exported style file
+                            if not last_style_path:
+                                print("No previous export found.")
+                                continue
+                            if not os.path.exists(last_style_path):
+                                print(f"Previous export file not found: {last_style_path}")
+                                continue
+                            yn = input(f"Overwrite '{os.path.basename(last_style_path)}'? (y/n): ").strip().lower()
+                            if yn != 'y':
+                                continue
+                            # Rebuild config based on current state
+                            snap = _style_snapshot(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_data)
+                            # Determine if last export was style-only or style+geometry
+                            try:
+                                with open(last_style_path, 'r', encoding='utf-8') as f:
+                                    old_cfg = json.load(f)
+                                if old_cfg.get('kind') == 'cpc_style_geom':
+                                    snap['kind'] = 'cpc_style_geom'
+                                    snap['geometry'] = _get_geometry_snapshot(ax, ax2)
+                                else:
+                                    snap['kind'] = 'cpc_style'
+                            except Exception:
+                                snap['kind'] = 'cpc_style'
+                            with open(last_style_path, 'w', encoding='utf-8') as f:
+                                json.dump(snap, f, indent=2)
+                            print(f"Overwritten style to {last_style_path}")
+                            style_menu_active = False
+                            break
                         target = None
                         if choice.isdigit() and files:
                             idx = int(choice)
@@ -2011,6 +2167,7 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
                             with open(target, 'w', encoding='utf-8') as f:
                                 json.dump(snap, f, indent=2)
                             print(f"Exported CPC style to {target}")
+                            fig._last_style_export_path = target
                         style_menu_active = False  # Exit style submenu and return to main menu
                         break
                     else:

{batplot-1.7.23 → batplot-1.7.24}/batplot/electrochem_interactive.py

@@ -468,6 +468,8 @@ def _get_legend_title(fig, default: str = "Cycle") -> str:
 def _rebuild_legend(ax):
     """Rebuild legend using only visible lines, anchoring to absolute inches from canvas center if available."""
     fig = ax.figure
+    # Capture existing title before any rebuild so it isn't lost
+    _store_legend_title(fig, ax)
     if not _get_legend_user_pref(fig):
         leg = ax.get_legend()
         if leg is not None:
@@ -1271,6 +1273,10 @@ def electrochem_interactive_menu(fig, ax, cycle_lines: Dict[int, Dict[str, Optio
                 snap['legend']['visible'] = bool(leg_obj.get_visible()) if leg_obj is not None else False
             except Exception:
                 pass
+            try:
+                snap['legend']['title'] = _get_legend_title(fig)
+            except Exception:
+                snap['legend']['title'] = None
             try:
                 legend_xy = getattr(fig, '_ec_legend_xy_in', None)
                 if legend_xy is not None:
@@ -1299,7 +1305,33 @@ def electrochem_interactive_menu(fig, ax, cycle_lines: Dict[int, Dict[str, Optio
             if len(state_history) > 40:
                 state_history.pop(0)
         except Exception:
-            pass
+            # Minimal fallback so undo still works if full snapshot fails
+            try:
+                fallback = {
+                    'note': f"{note}-fallback",
+                    'xlim': ax.get_xlim(),
+                    'ylim': ax.get_ylim(),
+                    'legend': {
+                        'visible': bool(ax.get_legend().get_visible()) if ax.get_legend() else False,
+                        'position_inches': getattr(fig, '_ec_legend_xy_in', None),
+                        'title': _get_legend_title(fig),
+                    },
+                    'lines': []
+                }
+                for i, ln in enumerate(ax.lines):
+                    try:
+                        fallback['lines'].append({
+                            'index': i,
+                            'color': ln.get_color(),
+                            'visible': ln.get_visible(),
+                        })
+                    except Exception:
+                        fallback['lines'].append({'index': i})
+                state_history.append(fallback)
+                if len(state_history) > 40:
+                    state_history.pop(0)
+            except Exception:
+                pass
 
     def restore_state():
         if not state_history:
@@ -1529,6 +1561,8 @@ def electrochem_interactive_menu(fig, ax, cycle_lines: Dict[int, Dict[str, Optio
             legend_snap = snap.get('legend', {})
             if legend_snap:
                 try:
+                    if 'title' in legend_snap:
+                        fig._ec_legend_title = legend_snap.get('title') or _get_legend_title(fig)
                     xy = legend_snap.get('position_inches')
                     fig._ec_legend_xy_in = _sanitize_legend_offset(fig, xy) if xy is not None else None
                 except Exception:
@@ -1594,14 +1628,35 @@ def electrochem_interactive_menu(fig, ax, cycle_lines: Dict[int, Dict[str, Optio
                         else:
                             print(f"  {i}: {fname}")
                 
-                fname = input("Export filename (default .svg if no extension) or number to overwrite (q=cancel): ").strip()
+                last_figure_path = getattr(fig, '_last_figure_export_path', None)
+                if last_figure_path:
+                    fname = input("Export filename (default .svg if no extension), number to overwrite, or o to overwrite last (q=cancel): ").strip()
+                else:
+                    fname = input("Export filename (default .svg if no extension) or number to overwrite (q=cancel): ").strip()
                 if not fname or fname.lower() == 'q':
                     _print_menu(len(all_cycles), is_dqdv)
                     continue
                 
+                already_confirmed = False  # Initialize for new filename case
+                # Check for 'o' option
+                if fname.lower() == 'o':
+                    if not last_figure_path:
+                        print("No previous export found.")
+                        _print_menu(len(all_cycles), is_dqdv)
+                        continue
+                    if not os.path.exists(last_figure_path):
+                        print(f"Previous export file not found: {last_figure_path}")
+                        _print_menu(len(all_cycles), is_dqdv)
+                        continue
+                    yn = input(f"Overwrite '{os.path.basename(last_figure_path)}'? (y/n): ").strip().lower()
+                    if yn != 'y':
+                        _print_menu(len(all_cycles), is_dqdv)
+                        continue
+                    target = last_figure_path
+                    already_confirmed = True
                 # Check if user selected a number
-                already_confirmed = False
-                if fname.isdigit() and files:
+                elif fname.isdigit() and files:
+                    already_confirmed = False
                     idx = int(fname)
                     if 1 <= idx <= len(files):
                         name = files[idx-1]
@@ -1671,6 +1726,7 @@ def electrochem_interactive_menu(fig, ax, cycle_lines: Dict[int, Dict[str, Optio
                         else:
                             fig.savefig(target, bbox_inches='tight')
                         print(f"Exported figure to {target}")
+                        fig._last_figure_export_path = target
                 except Exception as e:
                     print(f"Export failed: {e}")
             except Exception as e:
@@ -1854,11 +1910,45 @@ def electrochem_interactive_menu(fig, ax, cycle_lines: Dict[int, Dict[str, Optio
                             else:
                                 print(f"  {_i}: {fname}")
                     
-                    sub = input(_colorize_prompt("Style submenu: (e=export, q=return, r=refresh): ")).strip().lower()
+                    last_style_path = getattr(fig, '_last_style_export_path', None)
+                    if last_style_path:
+                        sub = input(_colorize_prompt("Style submenu: (e=export, o=overwrite last, q=return, r=refresh): ")).strip().lower()
+                    else:
+                        sub = input(_colorize_prompt("Style submenu: (e=export, q=return, r=refresh): ")).strip().lower()
                     if sub == 'q':
                         break
                     if sub == 'r' or sub == '':
                         continue
+                    if sub == 'o':
+                        # Overwrite last exported style file
+                        if not last_style_path:
+                            print("No previous export found.")
+                            continue
+                        if not os.path.exists(last_style_path):
+                            print(f"Previous export file not found: {last_style_path}")
+                            continue
+                        yn = input(f"Overwrite '{os.path.basename(last_style_path)}'? (y/n): ").strip().lower()
+                        if yn != 'y':
+                            continue
+                        # Rebuild config based on current state
+                        cfg = _get_style_snapshot(fig, ax, cycle_lines, tick_state)
+                        # Determine if last export was style-only or style+geometry
+                        try:
+                            with open(last_style_path, 'r', encoding='utf-8') as f:
+                                old_cfg = json.load(f)
+                            if old_cfg.get('kind') == 'ec_style_geom':
+                                geom = _get_geometry_snapshot(fig, ax)
+                                cfg['kind'] = 'ec_style_geom'
+                                cfg['geometry'] = geom
+                            else:
+                                cfg['kind'] = 'ec_style'
+                        except Exception:
+                            cfg['kind'] = 'ec_style'
+                        with open(last_style_path, 'w', encoding='utf-8') as f:
+                            json.dump(cfg, f, indent=2)
+                        print(f"Overwritten style to {last_style_path}")
+                        style_menu_active = False
+                        break
                     if sub == 'e':
                         # Ask for ps or psg
                         print("Export options:")
@@ -1895,7 +1985,9 @@ def electrochem_interactive_menu(fig, ax, cycle_lines: Dict[int, Dict[str, Optio
                             print("Style export canceled.")
                             continue
                         print(f"\nChosen path: {save_base}")
-                        _export_style_dialog(cfg, default_ext=default_ext, base_path=save_base)
+                        exported_path = _export_style_dialog(cfg, default_ext=default_ext, base_path=save_base)
+                        if exported_path:
+                            fig._last_style_export_path = exported_path
                         style_menu_active = False  # Exit style submenu and return to main menu
                         break
                     else:
@@ -2197,6 +2289,19 @@ def electrochem_interactive_menu(fig, ax, cycle_lines: Dict[int, Dict[str, Optio
                     except Exception as e:
                         print(f"Warning: Could not apply geometry: {e}")
                 
+                # Restore title offsets
+                try:
+                    offsets = cfg.get('title_offsets', {})
+                    if offsets:
+                        ax._top_xlabel_manual_offset_y_pts = float(offsets.get('top_y', 0.0) or 0.0)
+                        ax._top_xlabel_manual_offset_x_pts = float(offsets.get('top_x', 0.0) or 0.0)
+                        ax._bottom_xlabel_manual_offset_y_pts = float(offsets.get('bottom_y', 0.0) or 0.0)
+                        ax._left_ylabel_manual_offset_x_pts = float(offsets.get('left_x', 0.0) or 0.0)
+                        ax._right_ylabel_manual_offset_x_pts = float(offsets.get('right_x', 0.0) or 0.0)
+                        ax._right_ylabel_manual_offset_y_pts = float(offsets.get('right_y', 0.0) or 0.0)
+                except Exception:
+                    pass
+                
                 # Final label positioning - do this AFTER all style changes to prevent drift
                 # Set pending labelpad before repositioning to preserve original values
                 try:
@@ -2210,12 +2315,11 @@ def electrochem_interactive_menu(fig, ax, cycle_lines: Dict[int, Dict[str, Optio
                     font_cfg = cfg.get('font', {})
                     font_changed = (font_cfg.get('family') is not None or font_cfg.get('size') is not None)
                     
-                    if axes_position_changed or font_changed:
-                        # Reposition titles (will use _pending_xlabelpad if set, preserving original labelpad)
-                        _ui_position_top_xlabel(ax, fig, tick_state)
-                        _ui_position_bottom_xlabel(ax, fig, tick_state)
-                        _ui_position_left_ylabel(ax, fig, tick_state)
-                        _ui_position_right_ylabel(ax, fig, tick_state)
+                    # Always reposition titles to apply offsets (even if nothing else changed)
+                    _ui_position_top_xlabel(ax, fig, tick_state)
+                    _ui_position_bottom_xlabel(ax, fig, tick_state)
+                    _ui_position_left_ylabel(ax, fig, tick_state)
+                    _ui_position_right_ylabel(ax, fig, tick_state)
                     
                     # Always ensure labelpad is exactly as it was before style import
                     # This is a final safeguard against any drift
@@ -2798,6 +2902,7 @@ def electrochem_interactive_menu(fig, ax, cycle_lines: Dict[int, Dict[str, Optio
         elif key == 's':
             try:
                 from .session import dump_ec_session
+                last_session_path = getattr(fig, '_last_session_save_path', None)
                 folder = choose_save_path(source_paths, purpose="EC session save")
                 if not folder:
                     _print_menu(len(all_cycles), is_dqdv); continue
@@ -2815,10 +2920,27 @@ def electrochem_interactive_menu(fig, ax, cycle_lines: Dict[int, Dict[str, Optio
                             print(f"  {i}: {f}  ({timestamp})")
                         else:
                             print(f"  {i}: {f}")
-                prompt = "Enter new filename (no ext needed) or number to overwrite (q=cancel): "
+                if last_session_path:
+                    prompt = "Enter new filename (no ext needed), number to overwrite, or o to overwrite last (q=cancel): "
+                else:
+                    prompt = "Enter new filename (no ext needed) or number to overwrite (q=cancel): "
                 choice = input(prompt).strip()
                 if not choice or choice.lower() == 'q':
                     _print_menu(len(all_cycles), is_dqdv); continue
+                if choice.lower() == 'o':
+                    # Overwrite last saved session
+                    if not last_session_path:
+                        print("No previous save found.")
+                        _print_menu(len(all_cycles), is_dqdv); continue
+                    if not os.path.exists(last_session_path):
+                        print(f"Previous save file not found: {last_session_path}")
+                        _print_menu(len(all_cycles), is_dqdv); continue
+                    yn = input(f"Overwrite '{os.path.basename(last_session_path)}'? (y/n): ").strip().lower()
+                    if yn != 'y':
+                        _print_menu(len(all_cycles), is_dqdv); continue
+                    dump_ec_session(last_session_path, fig=fig, ax=ax, cycle_lines=cycle_lines, skip_confirm=True)
+                    print(f"Overwritten session to {last_session_path}")
+                    _print_menu(len(all_cycles), is_dqdv); continue
                 if choice.isdigit() and files:
                     idx = int(choice)
                     if 1 <= idx <= len(files):
@@ -2827,10 +2949,13 @@ def electrochem_interactive_menu(fig, ax, cycle_lines: Dict[int, Dict[str, Optio
                         if yn != 'y':
                             _print_menu(len(all_cycles), is_dqdv); continue
                         target = os.path.join(folder, name)
+                        dump_ec_session(target, fig=fig, ax=ax, cycle_lines=cycle_lines, skip_confirm=True)
+                        fig._last_session_save_path = target
+                        _print_menu(len(all_cycles), is_dqdv); continue
                     else:
                         print("Invalid number.")
                         _print_menu(len(all_cycles), is_dqdv); continue
-                else:
+                if choice.lower() != 'o':
                     name = choice
                     root, ext = os.path.splitext(name)
                     if ext == '':
@@ -2841,6 +2966,7 @@ def electrochem_interactive_menu(fig, ax, cycle_lines: Dict[int, Dict[str, Optio
                         if yn != 'y':
                             _print_menu(len(all_cycles), is_dqdv); continue
                 dump_ec_session(target, fig=fig, ax=ax, cycle_lines=cycle_lines, skip_confirm=True)
+                fig._last_session_save_path = target
             except Exception as e:
                 print(f"Save failed: {e}")
             _print_menu(len(all_cycles), is_dqdv)
@@ -4008,6 +4134,14 @@ def _get_style_snapshot(fig, ax, cycle_lines: Dict, tick_state: Dict) -> Dict:
         'ticks': {'widths': tick_widths, 'direction': tick_direction},
         'grid': grid_enabled,
         'wasd_state': wasd_state,
+        'title_offsets': {
+            'top_y': float(getattr(ax, '_top_xlabel_manual_offset_y_pts', 0.0) or 0.0),
+            'top_x': float(getattr(ax, '_top_xlabel_manual_offset_x_pts', 0.0) or 0.0),
+            'bottom_y': float(getattr(ax, '_bottom_xlabel_manual_offset_y_pts', 0.0) or 0.0),
+            'left_x': float(getattr(ax, '_left_ylabel_manual_offset_x_pts', 0.0) or 0.0),
+            'right_x': float(getattr(ax, '_right_ylabel_manual_offset_x_pts', 0.0) or 0.0),
+            'right_y': float(getattr(ax, '_right_ylabel_manual_offset_y_pts', 0.0) or 0.0),
+        },
         'curve_linewidth': curve_linewidth,
         'curve_markers': curve_marker_props,
         'rotation_angle': getattr(fig, '_ec_rotation_angle', 0),
@@ -4236,9 +4370,11 @@ def _export_style_dialog(cfg: Dict, default_ext: str = '.bpcfg', base_path: Opti
         with open(target_path, 'w', encoding='utf-8') as f:
             json.dump(cfg, f, indent=2)
         print(f"Style exported to {target_path}")
+        return target_path
 
     except Exception as e:
         print(f"Export failed: {e}")
+        return None
 def _legend_no_frame(ax, *args, title: Optional[str] = None, **kwargs):
     leg = ax.legend(*args, **kwargs)
     if leg is not None:
@@ -4258,6 +4394,8 @@ def _apply_legend_position(fig, ax):
     xy_in = _sanitize_legend_offset(fig, getattr(fig, '_ec_legend_xy_in', None))
     if xy_in is None:
         return False
+    # Preserve current title before rebuilding the legend
+    _store_legend_title(fig, ax)
     handles, labels = _visible_legend_entries(ax)
     if not handles:
         return False