From fa6ed5b3b37b0a2e130a813508596e6485ae5a1a Mon Sep 17 00:00:00 2001
From: Till Raab <till.raab@uni-tuebingen.de>
Date: Fri, 23 Jun 2023 10:37:48 +0200
Subject: [PATCH] included temporal correlation of chirps cented on rises
 win/lose.

---
 event_time_correlations.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/event_time_correlations.py b/event_time_correlations.py
index 80ab703..98ddd09 100644
--- a/event_time_correlations.py
+++ b/event_time_correlations.py
@@ -256,6 +256,7 @@ def main(base_path):
     lose_chrips_centered_on_contact_t = []
     lose_chrips_centered_on_win_rises = []
     lose_chrips_centered_on_win_chirp = []
+    lose_chirps_centered_on_lose_rises = []
     lose_chirp_count = []
 
     win_chrips_centered_on_ag_off_t = []
@@ -263,6 +264,7 @@ def main(base_path):
     win_chrips_centered_on_contact_t = []
     win_chrips_centered_on_lose_rises = []
     win_chrips_centered_on_lose_chirp = []
+    win_chirps_centered_on_win_rises = []
     win_chirp_count = []
 
     lose_rises_centered_on_ag_off_t = []
@@ -320,6 +322,7 @@ def main(base_path):
             lose_chrips_centered_on_contact_t.append(event_centered_times(contact_t_GRID, chirp_times[1]))
             lose_chrips_centered_on_win_rises.append(event_centered_times(rise_times[0], chirp_times[1]))
             lose_chrips_centered_on_win_chirp.append(event_centered_times(chirp_times[0], chirp_times[1]))
+            lose_chirps_centered_on_lose_rises.append(event_centered_times(rise_times[1], chirp_times[1]))
             lose_chirp_count.append(len(chirp_times[1]))
 
             win_chrips_centered_on_ag_off_t.append(event_centered_times(ag_on_off_t_GRID[:, 1], chirp_times[0]))
@@ -327,6 +330,7 @@ def main(base_path):
             win_chrips_centered_on_contact_t.append(event_centered_times(contact_t_GRID, chirp_times[0]))
             win_chrips_centered_on_lose_rises.append(event_centered_times(rise_times[1], chirp_times[0]))
             win_chrips_centered_on_lose_chirp.append(event_centered_times(chirp_times[1], chirp_times[0]))
+            win_chirps_centered_on_win_rises.append(event_centered_times(rise_times[0], chirp_times[0]))
 
             win_chirp_count.append(len(chirp_times[0]))
         else:
@@ -335,6 +339,7 @@ def main(base_path):
             lose_chrips_centered_on_contact_t.append(np.array([]))
             lose_chrips_centered_on_win_rises.append(np.array([]))
             lose_chrips_centered_on_win_chirp.append(np.array([]))
+            lose_chirps_centered_on_lose_rises.append(np.array([]))
             lose_chirp_count.append(np.nan)
 
             win_chrips_centered_on_ag_off_t.append(np.array([]))
@@ -377,12 +382,14 @@ def main(base_path):
              [lose_chrips_centered_on_contact_t, lose_chirp_count, r'chirp$_{lose}$ on contact'],
              [lose_chrips_centered_on_win_rises, lose_chirp_count, r'chirp$_{lose}$ on rise$_{win}$'],
              [lose_chrips_centered_on_win_chirp, lose_chirp_count, r'chirp$_{lose}$ on chirp$_{win}$'],
+             [lose_chirps_centered_on_lose_rises, lose_chirp_count, r'chirp$_{lose}$ on rises$_{lose}$'],
 
              [win_chrips_centered_on_ag_off_t, win_chirp_count, r'chirp$_{win}$ on chase$_{off}$'],
              [win_chrips_centered_on_ag_on_t, win_chirp_count, r'chirp$_{win}$ on chase$_{on}$'],
              [win_chrips_centered_on_contact_t, win_chirp_count, r'chirp$_{win}$ on contact'],
              [win_chrips_centered_on_lose_rises, win_chirp_count, r'chirp$_{win}$ on rise$_{lose}$'],
              [win_chrips_centered_on_lose_chirp, win_chirp_count, r'chirp$_{win}$ on chirp$_{lose}$'],
+             [win_chirps_centered_on_win_rises, win_chirp_count, r'chirp$_{win}$ on rises$_{win}$'],
 
              [lose_rises_centered_on_ag_off_t, lose_rises_count, r'rise$_{lose}$ on chase$_{off}$'],
              [lose_rises_centered_on_ag_on_t, lose_rises_count, r'rise$_{lose}$ on chase$_{on}$'],