From ea74dc7f882aacd2ef0af5d33a0fb40a64bc25b9 Mon Sep 17 00:00:00 2001
From: Dawith <cephalopodmaster@protonmail.com>
Date: Mon, 27 Oct 2025 13:45:03 -0400
Subject: [PATCH] Spectra/Spectrogram plotting basic code added to plot.py

---
 visualize/plot.py | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/visualize/plot.py b/visualize/plot.py
index 065c90c..1fcf3a6 100644
--- a/visualize/plot.py
+++ b/visualize/plot.py
@@ -1,9 +1,13 @@
 # --*- coding: utf-8 -*-
 
 import matplotlib.pyplot as plt
+import numpy as np
 import seaborn as sns
 
 def lineplot(data=None, x=None, y=None, hue=None):
+    """
+    """
+
     if data is None or x is None or y is None:
         raise ValueError("Data, x, and y parameters must be provided.")
 
@@ -19,5 +23,21 @@ def lineplot(data=None, x=None, y=None, hue=None):
         plt.savefig(f"/app/workdir/figures/lineplot_{y}_by_{x}.png")
     plt.close()
 
+def spectra_plot(spec_array, name=None):
+    """
+    """
+
+    spec_array = spec_array[:,:130]
+    print(spec_array.shape)
+    plt.pcolor(spec_array, cmap="bwr", vmin=-1, vmax=1)
+    plt.title(f"{name} spectra")
+    plt.savefig(f"/app/workdir/figures/{name}_spectra_plot.png")
+    plt.close()
+
+    spec_array -= np.mean(spec_array[:6,:], axis=0)
+    plt.pcolor(spec_array, cmap="bwr", vmin=-1, vmax=1)
+    plt.title(f"{name} spectrogram")
+    plt.savefig(f"/app/workdir/figures/{name}_spectrogram_plot.png")
+    plt.close()
 
 # EOF