Fix indexing for vis_baseline_hist. (#312)

EoRImaging · Aug 14, 2024 · 000baba · 000baba
1 parent 604a1f8
commit 000baba
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/fhd_core/visibility_manipulation/vis_baseline_hist.pro b/fhd_core/visibility_manipulation/vis_baseline_hist.pro
@@ -18,11 +18,11 @@ dist_hist = histogram(dist_arr, min=obs.min_baseline, binsize=5, max=obs.max_bas
             wh_noflag = where(Abs(model_vals) GT 0, count_noflag)
             IF count_noflag EQ 0 THEN CONTINUE ELSE inds = inds[wh_noflag]
             if Keyword_Set(calibration_visibilities_subtract) THEN BEGIN
-                vis_res_ratio_mean[i] = mean(abs((*vis_arr[pol_i])[inds]))/mean(abs(model_vals))
-                vis_res_sigma[i] = sqrt(variance(abs((*vis_arr[pol_i])[inds])))/mean(abs(model_vals))
+                vis_res_ratio_mean[pol_i, i] = mean(abs((*vis_arr[pol_i])[inds]))/mean(abs(model_vals))
+                vis_res_sigma[pol_i, i] = sqrt(variance(abs((*vis_arr[pol_i])[inds])))/mean(abs(model_vals))
             ENDIF ELSE BEGIN
-                vis_res_ratio_mean[i] = mean(abs((*vis_arr[pol_i])[inds]-model_vals))/mean(abs(model_vals))
-                vis_res_sigma[i] = sqrt(variance(abs((*vis_arr[pol_i])[inds]-model_vals)))/mean(abs(model_vals))
+                vis_res_ratio_mean[pol_i, i] = mean(abs((*vis_arr[pol_i])[inds]-model_vals))/mean(abs(model_vals))
+                vis_res_sigma[pol_i, i] = sqrt(variance(abs((*vis_arr[pol_i])[inds]-model_vals)))/mean(abs(model_vals))
             ENDELSE
         ENDIF
     ENDFOR