| """Shared ACM-style plotting configuration for figures_v2.""" |
| from __future__ import annotations |
|
|
| from pathlib import Path |
|
|
| import matplotlib as mpl |
| import matplotlib.pyplot as plt |
|
|
| SINGLE_COL = 3.35 |
| DOUBLE_COL = 6.95 |
|
|
| TITLE_SIZE = 10 |
| SUBTITLE_SIZE = 9 |
| LABEL_SIZE = 8.5 |
| TICK_SIZE = 7.5 |
| LEGEND_SIZE = 7.5 |
| ANNOTATION_SIZE = 7 |
| PANEL_LABEL_SIZE = 9 |
|
|
| LINEWIDTH = 1.6 |
| MARKER_SIZE = 4.5 |
| AXIS_LINEWIDTH = 0.8 |
| GRID_LINEWIDTH = 0.4 |
| BAR_EDGEWIDTH = 0.4 |
|
|
| COLORS = { |
| "blue": "#4C72B0", |
| "orange": "#DD8452", |
| "green": "#55A868", |
| "red": "#C44E52", |
| "purple": "#8172B2", |
| "gray": "#7F7F7F", |
| "dark": "#2B2B2B", |
| "light_gray": "#EAEAEA", |
| "pale_blue": "#EAF1FB", |
| "pale_green": "#ECF6EF", |
| "pale_orange": "#FBF0E9", |
| "pale_red": "#F9EDEE", |
| } |
|
|
|
|
| def apply_style() -> None: |
| """Apply compact, print-friendly defaults for ACM two-column figures.""" |
| mpl.rcParams.update( |
| { |
| "font.family": "DejaVu Sans", |
| "font.size": LABEL_SIZE, |
| "axes.titlesize": SUBTITLE_SIZE, |
| "axes.labelsize": LABEL_SIZE, |
| "xtick.labelsize": TICK_SIZE, |
| "ytick.labelsize": TICK_SIZE, |
| "legend.fontsize": LEGEND_SIZE, |
| "axes.linewidth": AXIS_LINEWIDTH, |
| "axes.spines.top": False, |
| "axes.spines.right": False, |
| "axes.grid": True, |
| "grid.color": "#D8D8D8", |
| "grid.alpha": 0.55, |
| "grid.linewidth": GRID_LINEWIDTH, |
| "lines.linewidth": LINEWIDTH, |
| "lines.markersize": MARKER_SIZE, |
| "figure.dpi": 150, |
| "savefig.dpi": 300, |
| "savefig.bbox": "tight", |
| "savefig.pad_inches": 0.03, |
| "pdf.fonttype": 42, |
| "ps.fonttype": 42, |
| "svg.fonttype": "none", |
| "legend.frameon": False, |
| } |
| ) |
|
|
|
|
| def output_dirs(root: Path) -> dict[str, Path]: |
| base = root / "figures_v2" |
| dirs = {} |
| for name in ("pdf", "png", "svg", "data"): |
| p = base / name |
| p.mkdir(parents=True, exist_ok=True) |
| dirs[name] = p |
| return dirs |
|
|
|
|
| def save_all(fig: plt.Figure, name: str, dirs: dict[str, Path]) -> list[str]: |
| """Save PDF/SVG as vector output and PNG as a 300 dpi preview.""" |
| if not fig.get_constrained_layout(): |
| fig.tight_layout() |
| outputs = [] |
| for ext in ("pdf", "png", "svg"): |
| path = dirs[ext] / f"{name}.{ext}" |
| if ext == "png": |
| fig.savefig(path, dpi=300, bbox_inches="tight", pad_inches=0.03) |
| else: |
| fig.savefig(path, bbox_inches="tight", pad_inches=0.03) |
| outputs.append(str(path)) |
| plt.close(fig) |
| return outputs |
|
|
|
|
| def panel_label(ax, label: str) -> None: |
| ax.text( |
| -0.10, |
| 1.04, |
| label, |
| transform=ax.transAxes, |
| fontsize=PANEL_LABEL_SIZE, |
| fontweight="bold", |
| va="bottom", |
| ha="left", |
| color=COLORS["dark"], |
| ) |
|
|
