Small changes.

debugging/alignment_utils.py

@@ -23,6 +23,7 @@ def get_entire_session_hist(recording, peaks, peak_locations, spatial_bin_edges,
     """
     TODO: assumes 1-segment recording
     """
+    # TODO: this is weird, don't return spatial_bin_edges here... amybe assert..
     entire_session_hist, temporal_bin_edges, spatial_bin_edges = \
         make_2d_motion_histogram(
         recording,
@@ -35,6 +36,7 @@ def get_entire_session_hist(recording, peaks, peak_locations, spatial_bin_edges,
         hist_margin_um=None,
         spatial_bin_edges=spatial_bin_edges,
     )
+
     entire_session_hist = entire_session_hist[0]
 
     entire_session_hist /= recording.get_duration(segment_index=0)
@@ -290,7 +292,7 @@ def prep_recording(recording, plot=False):
     :param recording:
     :return:
     """
-    peaks = detect_peaks(recording, method="by_channel")  # "locally_exclusive")
+    peaks = detect_peaks(recording, method="locally_exclusive")
 
     peak_locations = localize_peaks(recording, peaks,
                                     method="grid_convolution")

debugging/main.py

@@ -0,0 +1,66 @@
+from spikeinterface.generation.session_displacement_generator import generate_session_displacement_recordings
+import numpy as np
+import plotting
+import alignment_utils
+import matplotlib.pyplot as plt
+import pickle
+scalings = [np.ones(10), np.r_[np.zeros(3), np.ones(7)]]
+
+SAVE = True
+
+if SAVE:
+    recordings_list, _ = generate_session_displacement_recordings(
+        non_rigid_gradient=None,
+        num_units=35,
+        recording_durations=(100, 100),
+        recording_shifts=(
+            (0, 0), (0, 0),
+        ),
+        recording_amplitude_scalings=None, # {"method": "by_amplitude_and_firing_rate", "scalings": scalings},
+        seed=None,
+    )
+
+    peaks_list = []
+    peak_locations_list = []
+
+    for recording in recordings_list:
+        peaks, peak_locations = alignment_utils.prep_recording(
+            recording, plot=False,
+        )
+        peaks_list.append(peaks)
+        peak_locations_list.append(peak_locations)
+
+    # something relatively easy, only 15 units
+    with open('all_recordings.pickle', 'wb') as handle:
+        pickle.dump((recordings_list, peaks_list, peak_locations_list),
+                    handle, protocol=pickle.HIGHEST_PROTOCOL)
+
+with open('all_recordings.pickle', 'rb') as handle:
+    recordings_list, peaks_list, peak_locations_list = pickle.load(handle)
+
+bin_um = 2
+
+# TODO: own function
+min_y = np.min([np.min(locs["y"]) for locs in peak_locations_list])
+max_y = np.max([np.max(locs["y"]) for locs in peak_locations_list])
+
+spatial_bin_edges = np.arange(min_y, max_y + bin_um, bin_um)  # TODO: expose a margin...
+spatial_bin_centers = alignment_utils.get_bin_centers(spatial_bin_edges)  # TODO: own function
+
+session_histogram_list = []
+for recording, peaks, peak_locations in zip(recordings_list, peaks_list, peak_locations_list):
+
+    hist, temp, spat = alignment_utils.get_entire_session_hist(recording, peaks, peak_locations, spatial_bin_edges, log_scale=False)
+
+    session_histogram_list.append(
+        hist
+    )
+# TODO: need to get all outputs and check are same size
+plotting.SessionAlignmentWidget(
+    recordings_list,
+    peaks_list,
+    peak_locations_list,
+    session_histogram_list,
+    histogram_spatial_bin_centers=spatial_bin_centers,
+)
+plt.show()

debugging/plotting.py

@@ -143,7 +143,7 @@ def plot_matplotlib(self, data_plot, **backend_kwargs):
                 ax_top.set_title(f"Session {i + 1}")
                 ax_top.set_xlabel(None)
 
-                plot = DriftRasterMapWidget(
+                DriftRasterMapWidget(
                     dp.peaks_list[i],
                     dp.corrected_peak_locations_list[i],
                     recording=dp.recordings_list[i],
@@ -163,7 +163,7 @@ def plot_matplotlib(self, data_plot, **backend_kwargs):
 
             ax = fig.add_subplot(gs[num_rows, :])
 
-            plot = SessionAlignmentHistogramWidget(
+            SessionAlignmentHistogramWidget(
                 dp.session_histogram_list,
                 dp.histogram_spatial_bin_centers,
                 ax=ax,
@@ -242,6 +242,7 @@ def plot_matplotlib(self, data_plot, **backend_kwargs):
         if isinstance(linewidths, int):
             linewidths = [linewidths] * num_histograms
 
+        # TODO: this leads to quite unexpected behaviours, figure something else out.
         if spatial_bin_centers is None:
             num_bins = dp.session_histogram_list[0].size
             spatial_bin_centers = [np.arange(num_bins)] * num_histograms

debugging/session_alignment.py

@@ -125,7 +125,7 @@ def estimate_inter_session_displacement(
     max_y = np.max([np.max(locs["y"]) for locs in peak_locations_list])
 
     spatial_bin_edges = np.arange(min_y, max_y + bin_um, bin_um)  # TODO: expose a margin...
-    spatial_bin_centers = alignment_utils.get_bin_centers(spatial_bin_edges)
+    spatial_bin_centers = alignment_utils.get_bin_centers(spatial_bin_edges)  # TODO: own function
 
     # Estimate an activity histogram per-session
     all_session_hists = []  # TODO: probably better as a dict

debugging/test_session_alignment.py

@@ -46,8 +46,6 @@
 # handle the case where the passed recordings are not motion correction recordings.
 
 
-# 1) get all commits and PRs in order. Work on the original PR
-# 2) investigate why the expected peaks do not drop when recording_amplitude_scalings (rename) is used
 # 3) think about and add  new neurons that are introduced when shifted
 
 # 4) add interpolation of the histograms prior to cross correlation