plot_interaction_position.cpp

Go to the documentation of this file.

 
#include <filesystem>
 
#include "util.h"
// CERN ROOT
#include "ROOT/RDataFrame.hxx"
#include "TH2.h"
#include "TVector3.h"
#include "TCanvas.h"
 
// NiceMC
// #include "Event.h"
 
int main(int argc, char *argv[]){
 
  std::unique_ptr<TChain> allTree_chain_ptr;
  std::unique_ptr<TChain> passTree_chain_ptr;
  std::filesystem::path PUEO_MC_DATA = argv[1];
  (void) argc; // suppresses unused variable warning
 
  Util::prepare_icefinal_chain(PUEO_MC_DATA, passTree_chain_ptr, allTree_chain_ptr);
 
  TH2D int_pos_hist("",
                    "All-Event Interaction Positions; x (1e6, meters?); y (1e6, meters?)",
                    150, -0.2,  0.2,  // horizontal bin number and bounds
                    100, -0.2,  0.2);
 
 
  // a lambda expression that fills the above histogram
  auto fill_histogram = [&int_pos_hist](
    TVector3 & int_pos){
      int_pos_hist.Fill(int_pos.X()/1e6, int_pos.Y()/1e6);
  };
 
  // use a dataframe to loop through each entry in the passTree
  ROOT::RDataFrame df(*allTree_chain_ptr);
  df.Foreach(
    fill_histogram, 
    {
      "eventSummary.interaction.position"
    }
  );
 
  TCanvas canvas("", "");
  int_pos_hist.Draw("colz");
  int_pos_hist.SetStats(0);
  canvas.SaveAs("Interaction_position_XY_histogram.svg");
 
 
  return 0;
}