pyimcom 1.2.1__py3-none-any.whl

pyimcom/diagnostics/noise/stability.py

@@ -0,0 +1,126 @@
+import os
+import re
+import sys
+
+import matplotlib.pyplot as plt
+import numpy as np
+from astropy.io import fits
+from mpl_toolkits.axes_grid1 import make_axes_locatable
+
+
+def load_row_profiles(directory, name_pattern):
+    """
+    Load row median profiles from FITS files in a directory matching a name pattern.
+    Parameters
+    ----------
+    directory : str
+        Path to the directory containing FITS files.
+    name_pattern : str
+        Regex pattern to match filenames.
+
+    Returns
+    -------
+    row_profiles : np.ndarray
+        2D array where each row corresponds to the row median profile of an image.
+    obsnames : list of str
+        List of observation names extracted from filenames.
+    """
+    file_pattern = re.compile(name_pattern)
+    row_profiles = []
+    obsnames = []
+
+    for filename in sorted(os.listdir(directory)):
+        if m := file_pattern.match(filename):
+            obs = m.group(1)
+            with fits.open(os.path.join(directory, filename)) as hdul:
+                image = hdul[0].data
+            row_medians = np.median(image, axis=1)
+            row_profiles.append(row_medians)
+            obsnames.append(obs)
+    return np.array(row_profiles), obsnames
+
+
+def plot_row_stability_summary(row_profiles, SCA):
+    """
+    Plot a summary of row stability including a heatmap of row median profiles
+    and a plot of mean ± std deviation across rows.
+    Parameters
+    ----------
+    row_profiles : np.ndarray
+        2D array where each row corresponds to the row median profile of an image.
+    SCA : str
+        SCA identifier for labeling the plots.
+    Returns
+    -------
+    None
+    """
+    n_images, n_rows = row_profiles.shape
+
+    # Compute per-row statistics
+    mean_profile = np.mean(row_profiles, axis=0)
+    std_profile = np.std(row_profiles, axis=0)
+
+    fig, (ax1, ax2) = plt.subplots(2, 1, figsize=(10, 6), sharex=True, gridspec_kw={"height_ratios": [3, 1]})
+
+    # Top: Heatmap of row median profiles
+    im = ax1.imshow(
+        row_profiles,
+        aspect="auto",
+        cmap="viridis",
+        origin="lower",
+        vmin=np.percentile(row_profiles, 10),
+        vmax=np.percentile(row_profiles, 90),
+    )
+    ax1.set_ylabel("Observation Index")
+    ax1.set_title(f"Row Median Profiles Across Images: SCA {SCA}")
+
+    # Use make_axes_locatable to keep colorbar from shifting the axis
+    divider = make_axes_locatable(ax1)
+    cax = divider.append_axes("right", size="5%", pad=0.05)
+    plt.colorbar(im, cax=cax, label="Row Median (DN/fr)")
+
+    # Bottom: Mean ± std profile
+    x = np.arange(n_rows)
+
+    ax2.fill_between(
+        x,
+        mean_profile - std_profile,
+        mean_profile + std_profile,
+        color="yellowgreen",
+        label="±1 Sigma",
+        alpha=0.6,
+    )
+    ax2.fill_between(
+        x,
+        mean_profile - 2 * std_profile,
+        mean_profile + 2 * std_profile,
+        color="yellowgreen",
+        label="±2 Sigma",
+        alpha=0.2,
+    )
+
+    ax2.plot(x, mean_profile, color="black", label="Mean", linewidth=1)
+
+    ax2.set_ylim(-0.06, 0.06)
+    # ax2.set_yscale('symlog')  # Optional symlog toggle
+    ax2.set_ylabel("Median Value (DN/fr)")
+    ax2.set_xlabel("Row Index")
+    ax2.set_title("Row-wise Mean ± Std")
+
+    ax2.legend()
+    plt.tight_layout()
+    plt.savefig(f"plots/row_stability_summary_{SCA}.png", bbox_inches="tight")
+    plt.close()
+
+
+if __name__ == "__main__":
+    # --- Configuration ---
+    directory = sys.argv[1]  # Set this
+    output_csv = "stripe_stability.csv"
+
+    # --- Run Analysis ---
+    for i in range(18):
+        SCA = str(i + 1)
+        name_pattern = r"slope_(\d*)_(" + SCA + ").fits"
+        row_profiles, filenames = load_row_profiles(directory, name_pattern, SCA=SCA)
+        plot_row_stability_summary(row_profiles, SCA=SCA)