modified: pipelines/idr2_preprocessing/preprocess_data.py

pipelines/idr2_preprocessing/preprocess_data.py

@@ -66,6 +66,9 @@
 else:
     M = map
 
+# change to working dir
+os.chdir(work_dir)
+
 #-------------------------------------------------------------------------------
 # Reformat Data by Baseline Type
 #-------------------------------------------------------------------------------
@@ -91,6 +94,11 @@
         if angs[i] < 0:
             angs[i] = (angs[i] + 180) % 360
 
+    # put in autocorrs
+    reds = [zip(uvd.antenna_numbers, uvd.antenna_numbers)] + reds
+    lens = [0] + lens
+    angs = [0] + angs
+
     # iterate over polarization group
     for i, dfs in enumerate(datafiles):
 
@@ -196,6 +204,11 @@ def run_xrfi(i, datafiles=datafiles, p=cf['algorithm']['xrfi']):
     angs = [np.arctan2(*(antpos_dict[r[0][0]] - antpos_dict[r[0][1]])[:2][::-1]) * 180 / np.pi for r in reds]
     angs = [(a + 180) % 360 if a < 0 else a for a in angs]
 
+    # put in autocorrs
+    reds = [zip(uvd.antenna_numbers, uvd.antenna_numbers)] + reds
+    lens = [0] + lens
+    angs = [0] + angs
+
     # iterate over pols
     for i, dfs in enumerate(datafiles):
         pol = datapols[i][0]
@@ -329,6 +342,11 @@ def tavg_sub(j, dfs=dfs, pol=pol, tavg_file=tavg_out, p=cf['algorithm']['timeavg
     angs = [np.arctan2(*(antpos_dict[r[0][0]] - antpos_dict[r[0][1]])[:2][::-1]) * 180 / np.pi for r in reds]
     angs = [(a + 180) % 360 if a < 0 else a for a in angs]
 
+    # put in autocorrs
+    reds = [zip(uvd.antenna_numbers, uvd.antenna_numbers)] + reds
+    lens = [0] + lens
+    angs = [0] + angs
+
     # iterate over pol groups
     for i, dfs in enumerate(datafiles):
         pol = datapols[i][0]