batplot 1.4.0__tar.gz → 1.4.2__tar.gz

{batplot-1.4.0 → batplot-1.4.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: batplot
-Version: 1.4.0
+Version: 1.4.2
 Summary: Interactive plotting for XRD, PDF, and XAS data (.xye, .xy, .qye, .dat, .csv, .gr, .nor, .chik, .chir)
 Author-email: Tian Dai <tianda@uio.no>
 License: MIT License

{batplot-1.4.0 → batplot-1.4.2}/batplot/args.py

@@ -104,14 +104,16 @@ def _print_general_help() -> None:
         "  • Batch styling: apply .bps/.bpsg files to all exports (use --all flag)\n"
         "  • Format option: use --format png/pdf/jpg/etc to change export format\n\n"
     
-        "More help:\n"
-        "  batplot -h xy   # XY file plotting guide\n"
-        "  batplot -h ec   # Electrochemistry (GC/dQdV/CV/CPC) guide\n"
-        "  batplot -h op   # Operando guide\n\n"
-        "Contact & Updates:\n"
-        "  Subscribe to batplot-lab@kjemi.uio.no for updates and feedback\n"
-        "  GitHub: https://github.com/tiandai-chem/batplot\n"
-        "  Email: tianda@uio.no\n"
+    "More help:\n"
+    "  batplot -h xy   # XY file plotting guide\n"
+    "  batplot -h ec   # Electrochemistry (GC/dQdV/CV/CPC) guide\n"
+    "  batplot -h op   # Operando guide\n"
+    "  Manual: https://github.com/tiandai-chem/batplot/blob/main/USER_MANUAL.md\n\n"
+    "Contact & Updates:\n"
+    "  Subscribe to batplot-lab@kjemi.uio.no for updates\n"
+    "  (If you are not from UiO, send an email to sympa@kjemi.uio.no with the subject line \"subscribe batplot-lab@kjemi.uio.no your-name\")\n"
+    "  GitHub: https://github.com/tiandai-chem/batplot\n"
+    "  Email: tianda@uio.no\n"
     )
     _print_help(msg)
 
@@ -143,6 +145,10 @@ def _print_xy_help() -> None:
     "  --interactive             : open interactive menu for styling, ranges, fonts, export, sessions\n"
     "  --delta/-d <float>        : spacing between curves, e.g. --delta 0.1\n"
     "  --norm                    : normalize intensity to 0-1 range. Stack mode (--stack) auto-normalizes\n"
+    "  --chik                    : EXAFS χ(k) plot (sets labels to k (Å⁻¹) vs χ(k))\n"
+    "  --kchik                   : multiply y by x for EXAFS kχ(k) plots (sets labels to k (Å⁻¹) vs kχ(k) (Å⁻¹))\n"
+    "  --k2chik                  : multiply y by x² for EXAFS k²χ(k) plots (sets labels to k (Å⁻¹) vs k²χ(k) (Å⁻²))\n"
+    "  --k3chik                  : multiply y by x³ for EXAFS k³χ(k) plots (sets labels to k (Å⁻¹) vs k³χ(k) (Å⁻³))\n"
     "  --xrange/-r <min> <max>   : set x-axis range, e.g. --xrange 0 10\n"
     "  --out/-o <filename>       : save figure to file, e.g. --out file.svg\n"
     "  --xaxis <type>            : set x-axis type (Q, 2theta, r, k, energy, rft, time, or user defined)\n"
@@ -234,6 +240,10 @@ def build_parser() -> argparse.ArgumentParser:
     parser.add_argument("--wl", type=float, help=argparse.SUPPRESS)
     parser.add_argument("--fullprof", nargs="+", type=float, help=argparse.SUPPRESS)
     parser.add_argument("--norm", action="store_true", help=argparse.SUPPRESS)
+    parser.add_argument("--chik", action="store_true", help=argparse.SUPPRESS)
+    parser.add_argument("--kchik", action="store_true", help=argparse.SUPPRESS)
+    parser.add_argument("--k2chik", action="store_true", help=argparse.SUPPRESS)
+    parser.add_argument("--k3chik", action="store_true", help=argparse.SUPPRESS)
     parser.add_argument("--interactive", action="store_true", help=argparse.SUPPRESS)
     parser.add_argument("--savefig", type=str, help=argparse.SUPPRESS)
     parser.add_argument("--stack", action="store_true", help=argparse.SUPPRESS)

{batplot-1.4.0 → batplot-1.4.2}/batplot/batch.py

@@ -285,7 +285,7 @@ def batch_process(directory: str, args):
                         args._batch_warned_extensions.add(ext)
                         print(f"  Note: Reading '{ext}' files as 2-column (x, y) data with x-axis = {args.xaxis}")
                 else:
-                    raise ValueError(f"Cannot determine X-axis type for file {fname} (need .qye/.gr/.nor/.chik/.chir or --xaxis).")
+                    raise ValueError(f"Unknown file type: {fname}. Use --xaxis [Q|2theta|r|k|energy|rft] or batplot -h for help.")
 
             # Convert to Q if needed
             if axis_mode == 'Q' and ext not in ('.qye', '.gr', '.nor'):

{batplot-1.4.0 → batplot-1.4.2}/batplot/batplot.py

@@ -1587,6 +1587,12 @@ def batplot_main() -> int:
             fig._stack_label_at_bottom = stack_label_at_bottom
         except Exception:
             pass
+        # Restore grid state
+        try:
+            grid_state = bool(sess.get('grid', False))
+            ax.grid(grid_state, color='0.85', linestyle='-', linewidth=0.5, alpha=0.7)
+        except Exception:
+            pass
         # CIF tick series (optional)
         cif_tick_series = sess.get('cif_tick_series') or []
         cif_hkl_map = {k: [tuple(v) for v in val] for k,val in sess.get('cif_hkl_map', {}).items()}
@@ -1879,7 +1885,7 @@ def batplot_main() -> int:
             if args.xaxis:
                 axis_mode = args.xaxis
             else:
-                raise ValueError("Cannot determine X-axis type for .txt files. Please specify --xaxis (Q, 2theta, r, k, energy, rft, or 'user defined').")
+                raise ValueError("Unknown file type. Use: batplot file.txt --xaxis [Q|2theta|r|k|energy|rft] or batplot -h for help.")
         elif any_lambda or any_cif:
             if args.xaxis and args.xaxis.lower() in ("2theta","two_theta","tth"):
                 axis_mode = "2theta"
@@ -1890,7 +1896,7 @@ def batplot_main() -> int:
         elif args.xaxis:
             axis_mode = args.xaxis
         else:
-            raise ValueError("Cannot determine X-axis type (need .qye / .gr / .nor / .chik / .chir / .cif / wavelength / --xaxis). For .txt or unknown file types, use --xaxis Q, 2theta, r, k, energy, rft, or 'user defined'.")
+            raise ValueError("Unknown file type. Use: batplot file.csv --xaxis [Q|2theta|r|k|energy|rft] or batplot -h for help.")
 
     use_Q   = axis_mode == "Q"
     use_2th = axis_mode == "2theta"
@@ -2125,6 +2131,24 @@ def batplot_main() -> int:
         else:
             x_plot = x_full
 
+        # ---- Apply EXAFS k-weighting transformation if requested ----
+        if getattr(args, 'k3chik', False):
+            # Multiply y by x³ for EXAFS k³χ(k) plots
+            y_plot = y_plot * (x_plot ** 3)
+            y_full_raw = y_full_raw * (x_full ** 3)
+            raw_y_full_list[-1] = y_full_raw
+        elif getattr(args, 'k2chik', False):
+            # Multiply y by x² for EXAFS k²χ(k) plots
+            y_plot = y_plot * (x_plot ** 2)
+            y_full_raw = y_full_raw * (x_full ** 2)
+            raw_y_full_list[-1] = y_full_raw
+        elif getattr(args, 'kchik', False):
+            # Multiply y by x for EXAFS kχ(k) plots
+            y_plot = y_plot * x_plot
+            y_full_raw = y_full_raw * x_full
+            raw_y_full_list[-1] = y_full_raw
+        # elif getattr(args, 'chik', False): no multiplication needed, just label change
+
         # ---- Normalize (display subset) ----
         # Auto-normalize for --stack mode, or explicit --norm flag
         should_normalize = args.stack or getattr(args, 'norm', False)
@@ -2499,26 +2523,43 @@ def batplot_main() -> int:
         ax._cif_extend_func = extend_cif_tick_series
         ax._cif_draw_func = draw_cif_ticks
 
-    if use_E: x_label = "Energy (eV)"
-    elif use_r: x_label = r"r (Å)"
-    elif use_k: x_label = r"k ($\mathrm{\AA}^{-1}$)"
-    elif use_rft: x_label = "Radial distance (Å)"
-    elif use_Q: x_label = r"Q ($\mathrm{\AA}^{-1}$)"
-    elif use_2th: x_label = r"$2\theta$ (deg)"
-    elif use_time: x_label = "Time (h)"
-    elif args.xaxis:
-        x_label = str(args.xaxis)
+    # Handle EXAFS k-weighted χ(k) mode labels
+    if getattr(args, 'k3chik', False):
+        x_label = r"k ($\mathrm{\AA}^{-1}$)"
+        y_label = r"k$^3$χ(k) ($\mathrm{\AA}^{-3}$)"
+    elif getattr(args, 'k2chik', False):
+        x_label = r"k ($\mathrm{\AA}^{-1}$)"
+        y_label = r"k$^2$χ(k) ($\mathrm{\AA}^{-2}$)"
+    elif getattr(args, 'kchik', False):
+        x_label = r"k ($\mathrm{\AA}^{-1}$)"
+        y_label = r"kχ(k) ($\mathrm{\AA}^{-1}$)"
+    elif getattr(args, 'chik', False):
+        x_label = r"k ($\mathrm{\AA}^{-1}$)"
+        y_label = r"χ(k)"
     else:
-        x_label = "X"
+        if use_E: x_label = "Energy (eV)"
+        elif use_r: x_label = r"r (Å)"
+        elif use_k: x_label = r"k ($\mathrm{\AA}^{-1}$)"
+        elif use_rft: x_label = "Radial distance (Å)"
+        elif use_Q: x_label = r"Q ($\mathrm{\AA}^{-1}$)"
+        elif use_2th: x_label = r"$2\theta$ (deg)"
+        elif use_time: x_label = "Time (h)"
+        elif args.xaxis:
+            x_label = str(args.xaxis)
+        else:
+            x_label = "X"
+        
+        # Y-axis label: normalized if --stack or --norm, or voltage for time mode
+        should_normalize = args.stack or getattr(args, 'norm', False)
+        if use_time:
+            y_label = "Voltage (V)"
+        elif should_normalize:
+            y_label = "Normalized intensity (a.u.)"
+        else:
+            y_label = "Intensity"
+    
     ax.set_xlabel(x_label, fontsize=16)
-    # Y-axis label: normalized if --stack or --norm, or voltage for time mode
-    should_normalize = args.stack or getattr(args, 'norm', False)
-    if use_time:
-        ax.set_ylabel("Voltage (V)", fontsize=16)
-    elif should_normalize:
-        ax.set_ylabel("Normalized intensity (a.u.)", fontsize=16)
-    else:
-        ax.set_ylabel("Intensity", fontsize=16)
+    ax.set_ylabel(y_label, fontsize=16)
 
     # Store originals for axis-title toggle restoration (t menu bn/ln)
     try:

{batplot-1.4.0 → batplot-1.4.2}/batplot/cli.py

@@ -27,6 +27,10 @@ def main(argv: Optional[list] = None) -> int:
 		# Import the main batplot function (now refactored to be safe)
 		from .batplot import batplot_main
 		return batplot_main()
+	except ValueError as e:
+		# Print clean error message without traceback
+		print(f"Error: {e}", file=sys.stderr)
+		return 1
 	finally:
 		if argv is not None:
 			sys.argv = old_argv

{batplot-1.4.0 → batplot-1.4.2}/batplot/cpc_interactive.py

@@ -23,6 +23,52 @@ from .ui import (
 from .utils import _confirm_overwrite
 
 
+def _colorize_menu(text):
+    """Colorize menu items: command in cyan, colon in white, description in default."""
+    if ':' not in text:
+        return text
+    parts = text.split(':', 1)
+    cmd = parts[0].strip()
+    desc = parts[1].strip() if len(parts) > 1 else ''
+    return f"\033[96m{cmd}\033[0m: {desc}"  # Cyan for command, default for description
+
+
+def _colorize_prompt(text):
+    """Colorize commands within input prompts. Handles formats like (s=size, f=family, q=return) or (y/n)."""
+    import re
+    pattern = r'\(([a-z]+=[^,)]+(?:,\s*[a-z]+=[^,)]+)*|[a-z]+(?:/[a-z]+)+)\)'
+    
+    def colorize_match(match):
+        content = match.group(1)
+        if '/' in content:
+            parts = content.split('/')
+            colored_parts = [f"\033[96m{p.strip()}\033[0m" for p in parts]
+            return f"({'/'.join(colored_parts)})"
+        else:
+            parts = content.split(',')
+            colored_parts = []
+            for part in parts:
+                part = part.strip()
+                if '=' in part:
+                    cmd, desc = part.split('=', 1)
+                    colored_parts.append(f"\033[96m{cmd.strip()}\033[0m={desc.strip()}")
+                else:
+                    colored_parts.append(part)
+            return f"({', '.join(colored_parts)})"
+    
+    return re.sub(pattern, colorize_match, text)
+
+
+def _colorize_inline_commands(text):
+    """Colorize inline command examples in help text. Colors quoted examples and specific known commands."""
+    import re
+    # Color quoted command examples (like 's2 w5 a4', 'w2 w5')
+    text = re.sub(r"'([a-z0-9\s_-]+)'", lambda m: f"'\033[96m{m.group(1)}\033[0m'", text)
+    # Color specific known commands: q, i, l, list, help, all
+    text = re.sub(r'\b(q|i|l|list|help|all)\b(?=\s*[=,]|\s*$)', lambda m: f"\033[96m{m.group(1)}\033[0m", text)
+    return text
+
+
 def _generate_similar_color(base_color):
     """Generate a similar but distinguishable color for discharge from charge color."""
     try:
@@ -83,13 +129,17 @@ def _print_menu():
     w2 = max(18, *(len(s) for s in col2))
     w3 = max(12, *(len(s) for s in col3))
     rows = max(len(col1), len(col2), len(col3))
-    print("\nCPC interactive menu:")
-    print(f"  {'(Styles)':<{w1}} {'(Geometries)':<{w2}} {'(Options)':<{w3}}")
+    print("\n\033[1mCPC interactive menu:\033[0m")  # Bold title
+    print(f"  \033[93m{'(Styles)':<{w1}}\033[0m \033[93m{'(Geometries)':<{w2}}\033[0m \033[93m{'(Options)':<{w3}}\033[0m")  # Yellow headers
     for i in range(rows):
-        p1 = col1[i] if i < len(col1) else ""
-        p2 = col2[i] if i < len(col2) else ""
-        p3 = col3[i] if i < len(col3) else ""
-        print(f"  {p1:<{w1}} {p2:<{w2}} {p3:<{w3}}")
+        p1 = _colorize_menu(col1[i]) if i < len(col1) else ""
+        p2 = _colorize_menu(col2[i]) if i < len(col2) else ""
+        p3 = _colorize_menu(col3[i]) if i < len(col3) else ""
+        # Add padding to account for ANSI escape codes
+        pad1 = w1 + (9 if i < len(col1) else 0)
+        pad2 = w2 + (9 if i < len(col2) else 0)
+        pad3 = w3 + (9 if i < len(col3) else 0)
+        print(f"  {p1:<{pad1}} {p2:<{pad2}} {p3:<{pad3}}")
 
 
 def _get_current_file_artists(file_data, current_idx):
@@ -859,7 +909,7 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
         
         if key == 'q':
             try:
-                confirm = input("Quit CPC interactive? Remember to save! Quit now? (y/n): ").strip().lower()
+                confirm = input(_colorize_prompt("Quit CPC interactive? Remember to save! Quit now? (y/n): ")).strip().lower()
             except Exception:
                 confirm = 'y'
             if confirm == 'y':
@@ -1155,9 +1205,9 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
             # Spine colors (w=top, a=left, s=bottom, d=right)
             try:
                 print("Set spine colors (with matching tick and label colors):")
-                print("  w : top spine    | a : left spine")
-                print("  s : bottom spine | d : right spine")
-                print("Example: w:red a:#4561F7 s:blue d:green")
+                print(_colorize_inline_commands("  w : top spine    | a : left spine"))
+                print(_colorize_inline_commands("  s : bottom spine | d : right spine"))
+                print(_colorize_inline_commands("Example: w:red a:#4561F7 s:blue d:green"))
                 line = input("Enter mappings (e.g., w:red a:#4561F7) or q: ").strip()
                 if line and line.lower() != 'q':
                     push_state("color-spine")
@@ -1457,7 +1507,7 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
                     print(f"Existing style files in Styles/:")
                     for i, f in enumerate(files, 1):
                         print(f"  {i}: {f}")
-                sub = input("Style submenu: (e=export, q=return): ").strip().lower()
+                sub = input(_colorize_prompt("Style submenu: (e=export, q=return): ")).strip().lower()
                 if sub == 'e':
                     choice = input("Enter new filename or number to overwrite (q=cancel): ").strip()
                     if not choice or choice.lower() == 'q':
@@ -2141,10 +2191,10 @@ def cpc_interactive_menu(fig, ax, ax2, sc_charge, sc_discharge, sc_eff, file_dat
                     print(f"  left   a1:{b(wasd['left']['spine'])} a2:{b(wasd['left']['ticks'])} a3:{b(wasd['left']['minor'])} a4:{b(wasd['left']['labels'])} a5:{b(wasd['left']['title'])}")
                     print(f"  right  d1:{b(wasd['right']['spine'])} d2:{b(wasd['right']['ticks'])} d3:{b(wasd['right']['minor'])} d4:{b(wasd['right']['labels'])} d5:{b(wasd['right']['title'])}")
 
-                print("WASD toggles: direction (w/a/s/d) x action (1..5)")
-                print("  1=spine   2=ticks   3=minor ticks   4=tick labels   5=axis title")
-                print("Examples: 'w2 w5' to toggle top ticks and top title; 'd2 d5' for right.")
-                print("Type 'i' to invert tick direction, 'l' to change tick length, 'list' to show current state, 'q' to go back.")
+                print(_colorize_inline_commands("WASD toggles: direction (w/a/s/d) x action (1..5)"))
+                print(_colorize_inline_commands("  1=spine   2=ticks   3=minor ticks   4=tick labels   5=axis title"))
+                print(_colorize_inline_commands("Examples: 'w2 w5' to toggle top ticks and top title; 'd2 d5' for right."))
+                print(_colorize_inline_commands("Type 'i' to invert tick direction, 'l' to change tick length, 'list' to show current state, 'q' to go back."))
                 while True:
                     cmd = input("t> ").strip().lower()
                     if not cmd:

{batplot-1.4.0 → batplot-1.4.2}/batplot/electrochem_interactive.py

@@ -25,6 +25,52 @@ from .plotting import update_labels as _update_labels
 from .utils import _confirm_overwrite
 
 
+def _colorize_menu(text):
+    """Colorize menu items: command in cyan, colon in white, description in default."""
+    if ':' not in text:
+        return text
+    parts = text.split(':', 1)
+    cmd = parts[0].strip()
+    desc = parts[1].strip() if len(parts) > 1 else ''
+    return f"\033[96m{cmd}\033[0m: {desc}"  # Cyan for command, default for description
+
+
+def _colorize_prompt(text):
+    """Colorize commands within input prompts. Handles formats like (s=size, f=family, q=return) or (y/n)."""
+    import re
+    pattern = r'\(([a-z]+=[^,)]+(?:,\s*[a-z]+=[^,)]+)*|[a-z]+(?:/[a-z]+)+)\)'
+    
+    def colorize_match(match):
+        content = match.group(1)
+        if '/' in content:
+            parts = content.split('/')
+            colored_parts = [f"\033[96m{p.strip()}\033[0m" for p in parts]
+            return f"({'/'.join(colored_parts)})"
+        else:
+            parts = content.split(',')
+            colored_parts = []
+            for part in parts:
+                part = part.strip()
+                if '=' in part:
+                    cmd, desc = part.split('=', 1)
+                    colored_parts.append(f"\033[96m{cmd.strip()}\033[0m={desc.strip()}")
+                else:
+                    colored_parts.append(part)
+            return f"({', '.join(colored_parts)})"
+    
+    return re.sub(pattern, colorize_match, text)
+
+
+def _colorize_inline_commands(text):
+    """Colorize inline command examples in help text. Colors quoted examples and specific known commands."""
+    import re
+    # Color quoted command examples (like 's2 w5 a4', 'w2 w5')
+    text = re.sub(r"'([a-z0-9\s_-]+)'", lambda m: f"'\033[96m{m.group(1)}\033[0m'", text)
+    # Color specific known commands: q, i, l, list, help, all
+    text = re.sub(r'\b(q|i|l|list|help|all)\b(?=\s*[=,]|\s*$)', lambda m: f"\033[96m{m.group(1)}\033[0m", text)
+    return text
+
+
 def _print_menu(n_cycles: int, is_dqdv: bool = False):
     # Three-column menu similar to operando: Styles | Geometries | Options
     # Use dynamic column widths for clean alignment.
@@ -60,13 +106,17 @@ def _print_menu(n_cycles: int, is_dqdv: bool = False):
     w2 = max(len("(Geometries)"), *(len(s) for s in col2), 12)
     w3 = max(len("(Options)"), *(len(s) for s in col3), 12)
     rows = max(len(col1), len(col2), len(col3))
-    print("\nInteractive menu:")
-    print(f"  {'(Styles)':<{w1}} {'(Geometries)':<{w2}} {'(Options)':<{w3}}")
+    print("\n\033[1mInteractive menu:\033[0m")  # Bold title
+    print(f"  \033[93m{'(Styles)':<{w1}}\033[0m \033[93m{'(Geometries)':<{w2}}\033[0m \033[93m{'(Options)':<{w3}}\033[0m")  # Yellow headers
     for i in range(rows):
-        p1 = col1[i] if i < len(col1) else ""
-        p2 = col2[i] if i < len(col2) else ""
-        p3 = col3[i] if i < len(col3) else ""
-        print(f"  {p1:<{w1}} {p2:<{w2}} {p3:<{w3}}")
+        p1 = _colorize_menu(col1[i]) if i < len(col1) else ""
+        p2 = _colorize_menu(col2[i]) if i < len(col2) else ""
+        p3 = _colorize_menu(col3[i]) if i < len(col3) else ""
+        # Add padding to account for ANSI escape codes
+        pad1 = w1 + (9 if i < len(col1) else 0)
+        pad2 = w2 + (9 if i < len(col2) else 0)
+        pad3 = w3 + (9 if i < len(col3) else 0)
+        print(f"  {p1:<{pad1}} {p2:<{pad2}} {p3:<{pad3}}")
 
 
 def _iter_cycle_lines(cycle_lines: Dict[int, Dict[str, Optional[object]]]):
@@ -863,7 +913,7 @@ def electrochem_interactive_menu(fig, ax, cycle_lines: Dict[int, Dict[str, Optio
             continue
         if key == 'q':
             try:
-                confirm = input("Quit EC interactive? Remember to save (e=export, s=save). Quit now? (y/n): ").strip().lower()
+                confirm = input(_colorize_prompt("Quit EC interactive? Remember to save (e=export, s=save). Quit now? (y/n): ")).strip().lower()
             except Exception:
                 confirm = 'y'
             if confirm == 'y':
@@ -1020,7 +1070,7 @@ def electrochem_interactive_menu(fig, ax, cycle_lines: Dict[int, Dict[str, Optio
                 xy_in = getattr(fig, '_ec_legend_xy_in', (0.0, 0.0))
                 print(f"Legend is {'ON' if vis else 'off'}; position (inches from center): x={xy_in[0]:.2f}, y={xy_in[1]:.2f}")
                 while True:
-                    sub = input("Legend: t=toggle, m=set position (x y inches), q=back: ").strip().lower()
+                    sub = input(_colorize_prompt("Legend: (t=toggle, m=set position, q=back): ")).strip().lower()
                     if not sub:
                         continue
                     if sub == 'q':
@@ -1387,13 +1437,15 @@ def electrochem_interactive_menu(fig, ax, cycle_lines: Dict[int, Dict[str, Optio
                     print(f"  Tick widths: xM={x_maj if x_maj is not None else '?'} xm={x_min if x_min is not None else '?'} yM={y_maj if y_maj is not None else '?'} ym={y_min if y_min is not None else '?'}")
                     if cur_curve_lw is not None:
                         print(f"  Curves (all): {cur_curve_lw:.3g}")
-                    print("Line submenu:")
-                    print("  c  : change curve line widths")
-                    print("  f  : change frame (axes spines) and tick widths")
-                    print("  ld : show line and dots (markers) for all curves")
-                    print("  d  : show only dots (no connecting line) for all curves")
-                    print("  q  : return")
-                    sub = input("Choose (c/f/ld/d/q): ").strip().lower()
+                    print("\033[1mLine submenu:\033[0m")
+                    print(f"  {_colorize_menu('c  : change curve line widths')}")
+                    print(f"  {_colorize_menu('f  : change frame (axes spines) and tick widths')}")
+                    print(f"  {_colorize_menu('g  : toggle grid lines')}")
+                    print(f"  {_colorize_menu('l  : show only lines (no markers) for all curves')}")
+                    print(f"  {_colorize_menu('ld : show line and dots (markers) for all curves')}")
+                    print(f"  {_colorize_menu('d  : show only dots (no connecting line) for all curves')}")
+                    print(f"  {_colorize_menu('q  : return')}")
+                    sub = input(_colorize_prompt("Choose (c/f/g/l/ld/d/q): ")).strip().lower()
                     if not sub:
                         continue
                     if sub == 'q':
@@ -1451,6 +1503,54 @@ def electrochem_interactive_menu(fig, ax, cycle_lines: Dict[int, Dict[str, Optio
                             print(f"Set frame width={frame_w}, major tick width={tick_major}, minor tick width={tick_minor}")
                         except ValueError:
                             print("Invalid numeric value(s).")
+                    elif sub == 'g':
+                        push_state("grid")
+                        # Toggle grid state - check if any gridlines are visible
+                        current_grid = False
+                        try:
+                            # Check if grid is currently on by looking at gridline visibility
+                            for line in ax.get_xgridlines() + ax.get_ygridlines():
+                                if line.get_visible():
+                                    current_grid = True
+                                    break
+                        except Exception:
+                            current_grid = ax.xaxis._gridOnMajor if hasattr(ax.xaxis, '_gridOnMajor') else False
+                        
+                        new_grid_state = not current_grid
+                        if new_grid_state:
+                            # Enable grid with light styling
+                            ax.grid(True, color='0.85', linestyle='-', linewidth=0.5, alpha=0.7)
+                        else:
+                            # Disable grid (no style parameters when disabling)
+                            ax.grid(False)
+                        fig.canvas.draw()
+                        print(f"Grid {'enabled' if new_grid_state else 'disabled'}.")
+                    elif sub == 'l':
+                        # Line-only mode: set linestyle to solid and remove markers
+                        push_state("line-only")
+                        for cyc, role, ln in _iter_cycle_lines(cycle_lines):
+                            try:
+                                # Check if already in line-only mode (has line style and no marker)
+                                current_ls = ln.get_linestyle()
+                                current_marker = ln.get_marker()
+                                # If already line-only (has line, no marker), skip
+                                if current_ls not in ['None', '', ' ', 'none'] and current_marker in ['None', '', ' ', 'none', None]:
+                                    continue
+                                # Otherwise, set to line-only
+                                ln.set_linestyle('-')
+                                ln.set_marker('None')
+                            except Exception:
+                                pass
+                        try:
+                            _rebuild_legend(ax)
+                            fig.canvas.draw()
+                        except Exception:
+                            try:
+                                _rebuild_legend(ax)
+                            except Exception:
+                                pass
+                            fig.canvas.draw_idle()
+                        print("Applied line-only style to all curves.")
                     elif sub == 'ld':
                         # Line + dots for all curves
                         push_state("line+dots")
@@ -1521,9 +1621,9 @@ def electrochem_interactive_menu(fig, ax, cycle_lines: Dict[int, Dict[str, Optio
             # Spine colors (w=top, a=left, s=bottom, d=right)
             try:
                 print("Set spine colors (with matching tick and label colors):")
-                print("  w : top spine    | a : left spine")
-                print("  s : bottom spine | d : right spine")
-                print("Example: w:red a:#4561F7 s:blue d:green")
+                print(_colorize_inline_commands("  w : top spine    | a : left spine"))
+                print(_colorize_inline_commands("  s : bottom spine | d : right spine"))
+                print(_colorize_inline_commands("Example: w:red a:#4561F7 s:blue d:green"))
                 line = input("Enter mappings (e.g., w:red a:#4561F7) or q: ").strip()
                 if line and line.lower() != 'q':
                     push_state("color-spine")
@@ -1845,10 +1945,10 @@ def electrochem_interactive_menu(fig, ax, cycle_lines: Dict[int, Dict[str, Optio
         elif key == 'c':
             print(f"Total cycles: {len(all_cycles)}")
             print("Enter one of:")
-            print("  - numbers: e.g. 1 5 10")
-            print("  - mappings: e.g. 1:red 5:#00B006 10:blue")
-            print("  - numbers + palette: e.g. 1 5 10 viridis  OR  1 5 10 3")
-            print("  - all (optionally with palette): e.g. all  OR  all viridis  OR  all 3")
+            print(_colorize_inline_commands("  - numbers: e.g. 1 5 10"))
+            print(_colorize_inline_commands("  - mappings: e.g. 1:red 5:#00B006 10:blue"))
+            print(_colorize_inline_commands("  - numbers + palette: e.g. 1 5 10 viridis  OR  1 5 10 3"))
+            print(_colorize_inline_commands("  - all (optionally with palette): e.g. all  OR  all viridis  OR  all 3"))
             print("\nRecommended palettes for scientific publications:")
             print("  1. tab10      - Distinct, colorblind-friendly (default matplotlib)")
             print("  2. Set2       - Soft, pastel colors for presentations")

{batplot-1.4.0 → batplot-1.4.2}/batplot/interactive.py

@@ -99,6 +99,54 @@ def interactive_menu(fig, ax, y_data_list, x_data_list, labels, orig_y,
     if not hasattr(fig, '_stack_label_at_bottom'):
         fig._stack_label_at_bottom = False
 
+    # ANSI color codes for menu highlighting
+    def colorize_menu(text):
+        """Colorize menu items: command in cyan, colon in white, description in default."""
+        if ':' not in text:
+            return text
+        parts = text.split(':', 1)
+        cmd = parts[0].strip()
+        desc = parts[1].strip() if len(parts) > 1 else ''
+        return f"\033[96m{cmd}\033[0m: {desc}"  # Cyan for command, default for description
+    
+    def colorize_prompt(text):
+        """Colorize commands within input prompts. Handles formats like (s=size, f=family, q=return) or (y/n) or (q=cancel)."""
+        import re
+        # Pattern to match parenthesized command lists like (s=size, f=family, q=return) or (y/n) or (m/p/s/t/q) or (q=cancel)
+        pattern = r'\(([a-z]+=[^,)]+(?:,\s*[a-z]+=[^,)]+)*|[a-z]+(?:/[a-z]+)+)\)'
+        
+        def colorize_match(match):
+            content = match.group(1)
+            # Check if it's slash-separated (like y/n or m/p/s/t/q)
+            if '/' in content:
+                parts = content.split('/')
+                colored_parts = [f"\033[96m{p.strip()}\033[0m" for p in parts]
+                return f"({'/'.join(colored_parts)})"
+            # Otherwise it's equals-separated (like s=size, f=family or q=cancel)
+            else:
+                parts = content.split(',')
+                colored_parts = []
+                for part in parts:
+                    part = part.strip()
+                    if '=' in part:
+                        cmd, desc = part.split('=', 1)
+                        colored_parts.append(f"\033[96m{cmd.strip()}\033[0m={desc.strip()}")
+                    else:
+                        colored_parts.append(part)
+                return f"({', '.join(colored_parts)})"
+        
+        return re.sub(pattern, colorize_match, text)
+    
+    def colorize_inline_commands(text):
+        """Colorize inline command examples in help text. Colors quoted examples and specific known commands."""
+        import re
+        # Color quoted command examples (like 's2 w5 a4', 'w2 w5', or 'all magma_r')
+        text = re.sub(r"'([a-z0-9\s_-]+)'", lambda m: f"'\033[96m{m.group(1)}\033[0m'", text)
+        # Color specific known single-letter commands: q, i, l, when they appear as standalone commands
+        # Pattern: word boundary + (q|i|l|list|help|all) + space/equals/comma/end
+        text = re.sub(r'\b(q|i|l|list|help|all)\b(?=\s*[=,]|\s*$)', lambda m: f"\033[96m{m.group(1)}\033[0m", text)
+        return text
+    
     # REPLACED print_main_menu with column layout (now hides 'd' and 'y' in --stack)
     is_diffraction = use_Q or (not use_r and not use_E and not use_k and not use_rft)  # 2θ or Q
     def print_main_menu():
@@ -121,18 +169,23 @@ def interactive_menu(fig, ax, y_data_list, x_data_list, labels, orig_y,
         
         if not is_diffraction:
             col3 = [item for item in col3 if not item.startswith("n:")]
-        # Dynamic widths for cleaner alignment across terminals
+        # Dynamic widths for cleaner alignment across terminals (account for ANSI codes)
+        # Use plain text length for width calculations
         w1 = max(len("(Styles)"), *(len(s) for s in col1), 16)
         w2 = max(len("(Geometries)"), *(len(s) for s in col2), 16)
         w3 = max(len("(Options)"), *(len(s) for s in col3), 16)
         rows = max(len(col1), len(col2), len(col3))
-        print("\nInteractive menu:")
-        print(f"  {'(Styles)':<{w1}} {'(Geometries)':<{w2}} {'(Options)':<{w3}}")
+        print("\n\033[1mInteractive menu:\033[0m")  # Bold title
+        print(f"  \033[93m{'(Styles)':<{w1}}\033[0m \033[93m{'(Geometries)':<{w2}}\033[0m \033[93m{'(Options)':<{w3}}\033[0m")  # Yellow headers
         for i in range(rows):
-            p1 = col1[i] if i < len(col1) else ""
-            p2 = col2[i] if i < len(col2) else ""
-            p3 = col3[i] if i < len(col3) else ""
-            print(f"  {p1:<{w1}} {p2:<{w2}} {p3:<{w3}}")
+            p1 = colorize_menu(col1[i]) if i < len(col1) else ""
+            p2 = colorize_menu(col2[i]) if i < len(col2) else ""
+            p3 = colorize_menu(col3[i]) if i < len(col3) else ""
+            # Add padding to account for ANSI escape codes (9 chars per colorized item)
+            pad1 = w1 + (9 if i < len(col1) else 0)
+            pad2 = w2 + (9 if i < len(col2) else 0)
+            pad3 = w3 + (9 if i < len(col3) else 0)
+            print(f"  {p1:<{pad1}} {p2:<{pad2}} {p3:<{pad3}}")
 
     # --- Helper for spine visibility ---
     def set_spine_visible(which, visible):
@@ -690,7 +743,8 @@ def interactive_menu(fig, ax, y_data_list, x_data_list, labels, orig_y,
                 "show_cif_hkl": (bool(getattr(_bp, 'show_cif_hkl')) if _bp is not None and hasattr(_bp, 'show_cif_hkl') else False),
                 "show_cif_titles": (bool(getattr(_bp, 'show_cif_titles')) if _bp is not None and hasattr(_bp, 'show_cif_titles') else True),
                 "rotation_angle": getattr(ax, '_rotation_angle', 0),
-                "stack_label_at_bottom": getattr(fig, '_stack_label_at_bottom', False)
+                "stack_label_at_bottom": getattr(fig, '_stack_label_at_bottom', False),
+                "grid": ax.xaxis._gridOnMajor if hasattr(ax.xaxis, '_gridOnMajor') else False
             }
             # Line + data arrays
             for i, ln in enumerate(ax.lines):
@@ -906,6 +960,16 @@ def interactive_menu(fig, ax, y_data_list, x_data_list, labels, orig_y,
             if 'stack_label_at_bottom' in snap:
                 fig._stack_label_at_bottom = bool(snap['stack_label_at_bottom'])
 
+            # Restore grid state
+            if 'grid' in snap:
+                try:
+                    if snap['grid']:
+                        ax.grid(True, color='0.85', linestyle='-', linewidth=0.5, alpha=0.7)
+                    else:
+                        ax.grid(False)
+                except Exception:
+                    pass
+
             # CIF tick sets & label visibility (write back to batplot module globals)
             if _bp is not None and snap.get("cif_tick_series") is not None and hasattr(_bp, 'cif_tick_series'):
                 try:
@@ -960,7 +1024,7 @@ def interactive_menu(fig, ax, y_data_list, x_data_list, labels, orig_y,
             continue
 
         if key == 'q':
-            confirm = input("Quit interactive? Remember to save (e=export, s=save). Quit now? (y/n): ").strip().lower()
+            confirm = input(colorize_prompt("Quit interactive? Remember to save (e=export, s=save). Quit now? (y/n): ")).strip().lower()
             if confirm == 'y':
                 break
             else:
@@ -995,11 +1059,11 @@ def interactive_menu(fig, ax, y_data_list, x_data_list, labels, orig_y,
         elif key == 'h':  # legend submenu
             try:
                 while True:
-                    print("\nLegend submenu:")
-                    print("  v: show/hide curve names")
+                    print("\n\033[1mLegend submenu:\033[0m")
+                    print(f"  {colorize_menu('v: show/hide curve names')}")
                     current_pos = "bottom-right" if getattr(fig, '_stack_label_at_bottom', False) else "top-right"
-                    print(f"  s: legend position (current: {current_pos})")
-                    print("  q: back to main menu")
+                    print(f"  {colorize_menu(f's: legend position (current: {current_pos})')}")
+                    print(f"  {colorize_menu('q: back to main menu')}")
                     sub_key = input("Choose: ").strip().lower()
                     
                     if sub_key == 'q':
@@ -1136,14 +1200,14 @@ def interactive_menu(fig, ax, y_data_list, x_data_list, labels, orig_y,
                 except Exception:
                     pass
                 while True:
-                    print("Color menu:")
-                    print("  m : manual color mapping  (e.g., 1:red 2:#00B006)")
-                    print("  p : apply colormap palette to a range (e.g., 1-3 viridis)")
-                    print("  s : spine colors (e.g., w:red a:#4561F7 for top & left)")
+                    print("\033[1mColor menu:\033[0m")
+                    print(f"  {colorize_menu('m : manual color mapping  (e.g., 1:red 2:#00B006)')}")
+                    print(f"  {colorize_menu('p : apply colormap palette to a range (e.g., 1-3 viridis)')}")
+                    print(f"  {colorize_menu('s : spine colors (e.g., w:red a:#4561F7 for top & left)')}")
                     if has_cif and (_bp is not None and getattr(_bp, 'cif_tick_series', None)):
-                        print("  t : change CIF tick set color (e.g., 1:red 2:#888888)")
-                    print("  q : return to main menu")
-                    sub = input("Choose (m/p/s/t/q): ").strip().lower()
+                        print(f"  {colorize_menu('t : change CIF tick set color (e.g., 1:red 2:#888888)')}")
+                    print(f"  {colorize_menu('q : return to main menu')}")
+                    sub = input(colorize_prompt("Choose (m/p/s/t/q): ")).strip().lower()
                     if sub == 'q':
                         break
                     if sub == '':
@@ -1177,9 +1241,9 @@ def interactive_menu(fig, ax, y_data_list, x_data_list, labels, orig_y,
                     elif sub == 's':
                         # Spine colors (w=top, a=left, s=bottom, d=right)
                         print("Set spine colors (with matching tick and label colors):")
-                        print("  w : top spine    | a : left spine")
-                        print("  s : bottom spine | d : right spine")
-                        print("Example: w:red a:#4561F7 s:blue d:green")
+                        print(colorize_inline_commands("  w : top spine    | a : left spine"))
+                        print(colorize_inline_commands("  s : bottom spine | d : right spine"))
+                        print(colorize_inline_commands("Example: w:red a:#4561F7 s:blue d:green"))
                         line = input("Enter mappings (e.g., w:red a:#4561F7) or q: ").strip()
                         if not line or line.lower() == 'q':
                             print("Canceled.")
@@ -1252,7 +1316,7 @@ def interactive_menu(fig, ax, y_data_list, x_data_list, labels, orig_y,
                             extras.append('batlowK')
                         print("Common perceptually uniform palettes:")
                         print("  " + ", ".join(base_palettes + extras[:2]))
-                        print("Example: 1-4 viridis   or: all magma_r   or: 1-3,5 plasma, _r for reverse")
+                        print(colorize_inline_commands("Example: 1-4 viridis   or: all magma_r   or: 1-3,5 plasma, _r for reverse"))
                         line = input("Enter range(s) and palette (e.g., '1-3 viridis') or q: ").strip()
                         if not line or line.lower() == 'q':
                             print("Canceled.")
@@ -1663,12 +1727,12 @@ def interactive_menu(fig, ax, y_data_list, x_data_list, labels, orig_y,
                 except Exception as e:
                     print(f"Error setting Y-axis range: {e}")
         elif key == 'd':  # <-- DELTA / OFFSET HANDLER (now only reachable if not args.stack)
-            print("\nOffset adjustment menu:")
-            print("  1-{}: adjust individual curve offset".format(len(labels)))
-            print("  a: adjust total offset (baseline shift)")
-            print("  r: reset all offsets to 0")
-            print("  d: change delta spacing (original behavior)")
-            print("  q: back to main menu")
+            print("\n\033[1mOffset adjustment menu:\033[0m")
+            print(f"  {colorize_menu('1-{}: adjust individual curve offset'.format(len(labels)))}")
+            print(f"  {colorize_menu('a: adjust total offset (baseline shift)')}")
+            print(f"  {colorize_menu('r: reset all offsets to 0')}")
+            print(f"  {colorize_menu('d: change delta spacing (original behavior)')}")
+            print(f"  {colorize_menu('q: back to main menu')}")
             
             while True:
                 offset_cmd = input("Offset> ").strip().lower()
@@ -1913,13 +1977,15 @@ def interactive_menu(fig, ax, y_data_list, x_data_list, labels, orig_y,
         elif key == 'l':
             try:
                 while True:
-                    print("Line submenu:")
-                    print("  c  : change curve line widths")
-                    print("  f  : change frame (axes spines) and tick widths")
-                    print("  ld : show line and dots (markers) for all curves")
-                    print("  d  : show only dots (no connecting line) for all curves")
-                    print("  q  : return")
-                    sub = input("Choose (c/f/ld/d/q): ").strip().lower()
+                    print("\033[1mLine submenu:\033[0m")
+                    print(f"  {colorize_menu('c  : change curve line widths')}")
+                    print(f"  {colorize_menu('f  : change frame (axes spines) and tick widths')}")
+                    print(f"  {colorize_menu('g  : toggle grid lines')}")
+                    print(f"  {colorize_menu('l  : show only lines (no markers) for all curves')}")
+                    print(f"  {colorize_menu('ld : show line and dots (markers) for all curves')}")
+                    print(f"  {colorize_menu('d  : show only dots (no connecting line) for all curves')}")
+                    print(f"  {colorize_menu('q  : return')}")
+                    sub = input(colorize_prompt("Choose (c/f/g/l/ld/d/q): ")).strip().lower()
                     if sub == 'q':
                         break
                     if sub == '':
@@ -1978,6 +2044,43 @@ def interactive_menu(fig, ax, y_data_list, x_data_list, labels, orig_y,
                                 print(f"Set frame width={frame_w}, major tick width={tick_major}, minor tick width={tick_minor}")
                             except ValueError:
                                 print("Invalid numeric value(s).")
+                    elif sub == 'g':
+                        push_state("grid")
+                        # Toggle grid state - check if any gridlines are visible
+                        current_grid = False
+                        try:
+                            # Check if grid is currently on by looking at gridline visibility
+                            for line in ax.get_xgridlines() + ax.get_ygridlines():
+                                if line.get_visible():
+                                    current_grid = True
+                                    break
+                        except Exception:
+                            current_grid = ax.xaxis._gridOnMajor if hasattr(ax.xaxis, '_gridOnMajor') else False
+                        
+                        new_grid_state = not current_grid
+                        if new_grid_state:
+                            # Enable grid with light styling
+                            ax.grid(True, color='0.85', linestyle='-', linewidth=0.5, alpha=0.7)
+                        else:
+                            # Disable grid (no style parameters when disabling)
+                            ax.grid(False)
+                        fig.canvas.draw()
+                        print(f"Grid {'enabled' if new_grid_state else 'disabled'}.")
+                    elif sub == 'l':
+                        # Line-only mode: set linestyle to solid and remove markers
+                        push_state("line-only")
+                        for ln in ax.lines:
+                            # Check if already in line-only mode (has line style and no marker)
+                            current_ls = ln.get_linestyle()
+                            current_marker = ln.get_marker()
+                            # If already line-only (has line, no marker), skip
+                            if current_ls not in ['None', '', ' ', 'none'] and current_marker in ['None', '', ' ', 'none', None]:
+                                continue
+                            # Otherwise, set to line-only
+                            ln.set_linestyle('-')
+                            ln.set_marker('None')
+                        fig.canvas.draw()
+                        print("Applied line-only style to all curves.")
                     elif sub == 'ld':
                         push_state("line+dots")
                         try:
@@ -2026,7 +2129,7 @@ def interactive_menu(fig, ax, y_data_list, x_data_list, labels, orig_y,
                 print(f"Error setting widths: {e}")
         elif key == 'f':
             while True:
-                subkey = input("Font submenu (s=size, f=family, q=return): ").strip().lower()
+                subkey = input(colorize_prompt("Font submenu (s=size, f=family, q=return): ")).strip().lower()
                 if subkey == 'q':
                     break
                 if subkey == '':
@@ -2080,7 +2183,7 @@ def interactive_menu(fig, ax, y_data_list, x_data_list, labels, orig_y,
         elif key == 'g':
             try:
                 while True:
-                    choice = input("Resize submenu: (p=plot frame, c=canvas, q=cancel): ").strip().lower()
+                    choice = input(colorize_prompt("Resize submenu: (p=plot frame, c=canvas, q=cancel): ")).strip().lower()
                     if not choice:
                         continue
                     if choice == 'q':
@@ -2157,11 +2260,11 @@ def interactive_menu(fig, ax, y_data_list, x_data_list, labels, orig_y,
         elif key == 't':
             try:
                 while True:
-                    print("Toggle help:")
-                    print("  wasd choose side: w=top, a=left, s=bottom, d=right")
-                    print("  1..5 choose what: 1=spine line, 2=major ticks, 3=minor ticks, 4=labels, 5=axis title")
-                    print("  Combine letter+number to toggle, e.g. 's2 w5 a4' (case-insensitive)")
-                    print("  i = invert tick direction, l = change tick length, list = show state, q = return")
+                    print("\033[1mToggle help:\033[0m")
+                    print(colorize_inline_commands("  wasd choose side: w=top, a=left, s=bottom, d=right"))
+                    print(colorize_inline_commands("  1..5 choose what: 1=spine line, 2=major ticks, 3=minor ticks, 4=labels, 5=axis title"))
+                    print(colorize_inline_commands("  Combine letter+number to toggle, e.g. 's2 w5 a4' (case-insensitive)"))
+                    print(colorize_inline_commands("  i = invert tick direction, l = change tick length, list = show state, q = return"))
                     cmd = input("Enter code(s): ").strip().lower()
                     if not cmd:
                         continue
@@ -2464,7 +2567,7 @@ def interactive_menu(fig, ax, y_data_list, x_data_list, labels, orig_y,
                         print("Existing style files in Styles/ (.bps/.bpsg):")
                         for _i, _f in enumerate(_bpcfg_files, 1):
                             print(f"  {_i}: {_f}")
-                    sub = input("Style submenu: (e=export, q=return, r=refresh): ").strip().lower()
+                    sub = input(colorize_prompt("Style submenu: (e=export, q=return, r=refresh): ")).strip().lower()
                     if sub == 'q':
                         break
                     if sub == 'r' or sub == '':

{batplot-1.4.0 → batplot-1.4.2}/batplot/operando_ec_interactive.py

@@ -28,6 +28,53 @@ from .ui import position_right_ylabel as _ui_position_right_ylabel
 from .ui import position_bottom_xlabel as _ui_position_bottom_xlabel
 from .ui import position_left_ylabel as _ui_position_left_ylabel
 
+
+def _colorize_menu(text):
+    """Colorize menu items: command in cyan, colon in white, description in default."""
+    if ':' not in text:
+        return text
+    parts = text.split(':', 1)
+    cmd = parts[0].strip()
+    desc = parts[1].strip() if len(parts) > 1 else ''
+    return f"\033[96m{cmd}\033[0m: {desc}"  # Cyan for command, default for description
+
+
+def _colorize_prompt(text):
+    """Colorize commands within input prompts. Handles formats like (s=size, f=family, q=return) or (y/n)."""
+    import re
+    pattern = r'\(([a-z]+=[^,)]+(?:,\s*[a-z]+=[^,)]+)*|[a-z]+(?:/[a-z]+)+)\)'
+    
+    def colorize_match(match):
+        content = match.group(1)
+        if '/' in content:
+            parts = content.split('/')
+            colored_parts = [f"\033[96m{p.strip()}\033[0m" for p in parts]
+            return f"({'/'.join(colored_parts)})"
+        else:
+            parts = content.split(',')
+            colored_parts = []
+            for part in parts:
+                part = part.strip()
+                if '=' in part:
+                    cmd, desc = part.split('=', 1)
+                    colored_parts.append(f"\033[96m{cmd.strip()}\033[0m={desc.strip()}")
+                else:
+                    colored_parts.append(part)
+            return f"({', '.join(colored_parts)})"
+    
+    return re.sub(pattern, colorize_match, text)
+
+
+def _colorize_inline_commands(text):
+    """Colorize inline command examples in help text. Colors quoted examples and specific known commands."""
+    import re
+    # Color quoted command examples (like 's2 w5 a4', 'w2 w5')
+    text = re.sub(r"'([a-z0-9\s_-]+)'", lambda m: f"'\033[96m{m.group(1)}\033[0m'", text)
+    # Color specific known commands: q, i, l, list, help, all
+    text = re.sub(r'\b(q|i|l|list|help|all)\b(?=\s*[=,]|\s*$)', lambda m: f"\033[96m{m.group(1)}\033[0m", text)
+    return text
+
+
 # ============================================================================
 # Constants
 # ============================================================================
@@ -375,14 +422,19 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax):
             w3 = max(len("(EC)"), *(len(s) for s in col3), 14)
             w4 = max(len("(Options)"), *(len(s) for s in col4), 16)
             rows = max(len(col1), len(col2), len(col3), len(col4))
-            print("\nInteractive menu:")
-            print(f"  {'(Styles)':<{w1}} {'(Operando)':<{w2}} {'(EC)':<{w3}} {'(Options)':<{w4}}")
+            print("\n\033[1mInteractive menu:\033[0m")  # Bold title
+            print(f"  \033[93m{'(Styles)':<{w1}}\033[0m \033[93m{'(Operando)':<{w2}}\033[0m \033[93m{'(EC)':<{w3}}\033[0m \033[93m{'(Options)':<{w4}}\033[0m")  # Yellow headers
             for i in range(rows):
-                p1 = col1[i] if i < len(col1) else ""
-                p2 = col2[i] if i < len(col2) else ""
-                p3 = col3[i] if i < len(col3) else ""
-                p4 = col4[i] if i < len(col4) else ""
-                print(f"  {p1:<{w1}} {p2:<{w2}} {p3:<{w3}} {p4:<{w4}}")
+                p1 = _colorize_menu(col1[i]) if i < len(col1) else ""
+                p2 = _colorize_menu(col2[i]) if i < len(col2) else ""
+                p3 = _colorize_menu(col3[i]) if i < len(col3) else ""
+                p4 = _colorize_menu(col4[i]) if i < len(col4) else ""
+                # Add padding to account for ANSI escape codes
+                pad1 = w1 + (9 if i < len(col1) else 0)
+                pad2 = w2 + (9 if i < len(col2) else 0)
+                pad3 = w3 + (9 if i < len(col3) else 0)
+                pad4 = w4 + (9 if i < len(col4) else 0)
+                print(f"  {p1:<{pad1}} {p2:<{pad2}} {p3:<{pad3}} {p4:<{pad4}}")
         else:
             # Operando-only menu (no EC panel)
             col1 = [
@@ -415,13 +467,17 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax):
             w2 = max(len("(Operando)"), *(len(s) for s in col2), 14)
             w3 = max(len("(Options)"), *(len(s) for s in col3), 16)
             rows = max(len(col1), len(col2), len(col3))
-            print("\nInteractive menu:")
-            print(f"  {'(Styles)':<{w1}} {'(Operando)':<{w2}} {'(Options)':<{w3}}")
+            print("\n\033[1mInteractive menu:\033[0m")  # Bold title
+            print(f"  \033[93m{'(Styles)':<{w1}}\033[0m \033[93m{'(Operando)':<{w2}}\033[0m \033[93m{'(Options)':<{w3}}\033[0m")  # Yellow headers
             for i in range(rows):
-                p1 = col1[i] if i < len(col1) else ""
-                p2 = col2[i] if i < len(col2) else ""
-                p3 = col3[i] if i < len(col3) else ""
-                print(f"  {p1:<{w1}} {p2:<{w2}} {p3:<{w3}}")
+                p1 = _colorize_menu(col1[i]) if i < len(col1) else ""
+                p2 = _colorize_menu(col2[i]) if i < len(col2) else ""
+                p3 = _colorize_menu(col3[i]) if i < len(col3) else ""
+                # Add padding to account for ANSI escape codes
+                pad1 = w1 + (9 if i < len(col1) else 0)
+                pad2 = w2 + (9 if i < len(col2) else 0)
+                pad3 = w3 + (9 if i < len(col3) else 0)
+                print(f"  {p1:<{pad1}} {p2:<{pad2}} {p3:<{pad3}}")
         print()
     def print_menu_old():
         col1 = [
@@ -1615,11 +1671,11 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax):
         elif cmd == 'l':
             # Line widths submenu for both operando and EC panes
             print("Line widths: set frame (spines) and tick widths for both operando and EC")
-            print("Enter frame/tick width (e.g., '1.5' or 'f t' for frame/tick separately)")
+            print(_colorize_inline_commands("Enter frame/tick width (e.g., '1.5' or 'f t' for frame/tick separately)"))
             print("Format examples:")
-            print("  1.5      - set both frame and ticks to 1.5")
-            print("  1.5 2.5  - set frame=1.5, ticks=2.5")
-            print("  q        - cancel")
+            print(_colorize_inline_commands("  1.5      - set both frame and ticks to 1.5"))
+            print(_colorize_inline_commands("  1.5 2.5  - set frame=1.5, ticks=2.5"))
+            print(_colorize_inline_commands("  q        - cancel"))
             
             inp = input("Line widths> ").strip().lower()
             if not inp or inp == 'q':
@@ -1756,9 +1812,9 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax):
                     pass
             while True:
                 if ec_ax is not None:
-                    print("Choose pane: o=operando, e=ec, q=back")
+                    print(_colorize_inline_commands("Choose pane: o=operando, e=ec, q=back"))
                 else:
-                    print("Choose pane: o=operando, q=back")
+                    print(_colorize_inline_commands("Choose pane: o=operando, q=back"))
                 pane = input("ot> ").strip().lower()
                 if not pane:
                     continue
@@ -2288,7 +2344,7 @@ def operando_ec_interactive_menu(fig, ax, im, cbar, ec_ax):
             print("  4. cividis    - Perceptually uniform, optimized for color vision deficiency")
             print("  5. magma      - Perceptually uniform (black→white), excellent for grayscale")
             print("\nOther available: " + ", ".join(base + extras))
-            print("Append _r to reverse (e.g., viridis_r or 1_r). Blank to cancel.")
+            print(_colorize_inline_commands("Append _r to reverse (e.g., viridis_r or 1_r). Blank to cancel."))
             choice = input("Palette name or number (1-5): ").strip()
             if not choice:
                 print_menu(); continue

{batplot-1.4.0 → batplot-1.4.2}/batplot/session.py

@@ -231,6 +231,8 @@ def dump_session(
         sess['curve_names_visible'] = bool(getattr(fig, '_curve_names_visible', True))
         # Save stack label position preference
         sess['stack_label_at_bottom'] = bool(getattr(fig, '_stack_label_at_bottom', False))
+        # Save grid state
+        sess['grid'] = ax.xaxis._gridOnMajor if hasattr(ax.xaxis, '_gridOnMajor') else False
         if skip_confirm:
             target = filename
         else:

{batplot-1.4.0 → batplot-1.4.2}/batplot/style.py

@@ -286,6 +286,7 @@ def export_style_config(
                 "x": ax.xaxis.label.get_color(),
                 "y": ax.yaxis.label.get_color(),
             },
+            "grid": ax.xaxis._gridOnMajor if hasattr(ax.xaxis, '_gridOnMajor') else False,
             "lines": [
                 {
                     "index": i,
@@ -777,6 +778,16 @@ def apply_style_config(
             except Exception as e:
                 print(f"Warning: Could not restore rotation angle: {e}")
 
+        # Restore grid state
+        if "grid" in cfg:
+            try:
+                if bool(cfg["grid"]):
+                    ax.grid(True, color='0.85', linestyle='-', linewidth=0.5, alpha=0.7)
+                else:
+                    ax.grid(False)
+            except Exception as e:
+                print(f"Warning: Could not restore grid state: {e}")
+
         # Re-run label placement with current mode (no mode changes via Styles)
         stack_label_bottom = getattr(fig, '_stack_label_at_bottom', False)
         update_labels_func(ax, y_data_list, label_text_objects, args.stack, stack_label_bottom)

{batplot-1.4.0 → batplot-1.4.2}/batplot.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: batplot
-Version: 1.4.0
+Version: 1.4.2
 Summary: Interactive plotting for XRD, PDF, and XAS data (.xye, .xy, .qye, .dat, .csv, .gr, .nor, .chik, .chir)
 Author-email: Tian Dai <tianda@uio.no>
 License: MIT License

{batplot-1.4.0 → batplot-1.4.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "batplot"
-version = "1.4.0"
+version = "1.4.2"
 description = "Interactive plotting for XRD, PDF, and XAS data (.xye, .xy, .qye, .dat, .csv, .gr, .nor, .chik, .chir)"
 authors = [
     { name = "Tian Dai", email = "tianda@uio.no" }