include mean and RMS in plots

Noel Dawe
1 parent 116a8618
Showing 1 changed file with 12 additions and 2 deletions Show diff stats
scripts/ftkplot
@@ -233,15 +233,25 @@ def plot(filename):
         for axis in "xy":
             hist_pix = Hist(100, -1, 1)
             diff = arr_ftk_flat['ftk_refit_track_loc{0}_pix_l{1}'.format(axis, layer)] - arr_ftk_flat['ftk_track_loc{0}_pix_l{1}'.format(axis, layer)]
+            diff = diff[~np.isnan(diff)]
+            # get stats
+            # quote means within 3*RMS of distribution
+            mean = np.mean(diff)
+            dev = np.std(diff[(diff < 10) & (diff > -10)])
+            mean = np.mean(diff[(diff < (mean + 3 * dev)) & (diff > (mean - 3 * dev))])
             fill_hist(hist_pix, diff)
             canvas = Canvas()
             draw(hist_pix, xtitle="\Delta local{0} \;\;\; Pixel \; Layer \; {1} \;\;\; (Refit - Raw) \;\;\; [mm]".format(axis.upper(), layer), ytitle="FTK Tracks")
+            stats = ROOT.TLatex(0.2, 0.8, "#splitline{{mean = {0:.4f}}}{{RMS = {1:.4f}}}".format(mean, dev))
+            stats.SetNDC()
+            stats.Draw()
             canvas.SaveAs("pixel_layer_{0}_local{1}_refit_v_raw.png".format(layer, axis.upper()))
  
     ###############################################################
     # Plot all track parameters (truth, offline, raw and refit FTK)
     # WARNING: This can be memory intensive!
-
+    
+    """
     # Truth track parameters
     tree = root_open(filename)['tracks']
     truth_branches = [b for b in tree.branchnames if b.startswith('truth_')]
@@ -263,7 +273,7 @@ def plot(filename):
     fig = plot_variable_grid(ftk_arr)
     fig.savefig('params_ftk.png')
     del ftk_arr
-
+    """
  
 if __name__ == '__main__':
     from argparse import ArgumentParser