add templates for layer-specific PSD and update the Drive-Dipole-Spec…

…trogram (3x1) tempate to point to the layer-specific PSDs

hnn_core/gui/_viz_manager.py

@@ -26,6 +26,8 @@
     'input histogram',
     'spikes',
     'PSD',
+    'layer 2 PSD',
+    'layer 5 PSD',
     'spectrogram',
     'network',
 ]
@@ -130,8 +132,11 @@
             "gridspec_kw": {"height_ratios": [1, 1, 1]}
         },
         "mosaic": "0\n1\n2",
-        "ax_plots": [("ax0", "layer2 dipole"), ("ax1", "layer5 dipole"),
-                     ("ax2", "PSD")]
+        "ax_plots": [
+            ("ax0", "layer 2 PSD"),
+            ("ax1", "layer 5 PSD"),
+            ("ax2", "PSD"),
+        ]
     }
 }
 
@@ -296,6 +301,36 @@ def _update_ax(fig, ax, single_simulation, sim_name, plot_type, plot_config):
             dpls_copied[0].plot_psd(fmin=min_f, fmax=max_f, ax=ax, color=color,
                                     label=sim_name, show=False)
 
+    elif plot_type == 'layer 2 PSD':
+        if len(dpls_copied) > 0:
+            min_f = plot_config['min_spectral_frequency']
+            max_f = plot_config['max_spectral_frequency']
+            color = ax._get_lines.get_next_color()
+            dpls_copied[0].plot_psd(
+                fmin=min_f,
+                fmax=max_f,
+                layer='L2',
+                color=color,
+                label=sim_name,
+                ax=ax,
+                show=False
+            )
+
+    elif plot_type == 'layer 5 PSD':
+        if len(dpls_copied) > 0:
+            min_f = plot_config['min_spectral_frequency']
+            max_f = plot_config['max_spectral_frequency']
+            color = ax._get_lines.get_next_color()
+            dpls_copied[0].plot_psd(
+                fmin=min_f,
+                fmax=max_f,
+                layer='L5',
+                color=color,
+                label=sim_name,
+                ax=ax,
+                show=False
+            )
+
     elif plot_type == 'spectrogram':
         if len(dpls_copied) > 0:
             min_f = plot_config['min_spectral_frequency']