MvtxQA.C

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file MvtxQA.C

#include "../CommonTools.h"
#include <sPhenixStyle.C>

TCanvas* Draw( TFile* qa_file_new, TFile* qa_file_ref, const TString& hist_name_prefix, const TString& tag );
static constexpr int first_layer_mvtx = 0;
static constexpr int nlayers_mvtx = 3;
const char *hist_name_prefix = "QAG4SimulationMvtx";

void MvtxQA(std::string reffile, std::string newfile, std::string outfile)
{
  SetsPhenixStyle();
  TFile *reff = TFile::Open(reffile.c_str());
  TFile *newf = TFile::Open(newfile.c_str());
  
  auto c1 = Draw( newf, reff, hist_name_prefix, "drphi" );
  auto c2 = Draw( newf, reff, hist_name_prefix, "rphi_error" );
  auto c3 = Draw( newf, reff, hist_name_prefix, "phi_pulls" );
  auto c4 = Draw( newf, reff, hist_name_prefix, "dz" );
  auto c5 = Draw( newf, reff, hist_name_prefix, "z_error" );
  auto c6 = Draw( newf, reff, hist_name_prefix, "z_pulls" );
  auto c7 = Draw( newf, reff, hist_name_prefix, "clus_size" );
  auto c8 = Draw( newf, reff, hist_name_prefix, "clus_size_phi" );
  auto c9 = Draw( newf, reff, hist_name_prefix, "clus_size_z" );
  
  TFile *outfilef = new TFile(outfile.c_str(), "recreate");
  c1->Write();
  c2->Write();
  c3->Write();
  c4->Write();
  c5->Write();
  c6->Write();
  c7->Write();
  c8->Write();
  c9->Write();
  outfilef->Close();
  
}

TCanvas* Draw( TFile* qa_file_new, TFile* qa_file_ref, const TString& hist_name_prefix, const TString& tag )
{

    const TString prefix = TString("h_") + hist_name_prefix + TString("_");

    auto cv = new TCanvas(
      TString("QA_Draw_Mvtx_") + tag + TString("_") + hist_name_prefix,
      TString("QA_Draw_Mvtx_") + tag + TString("_") + hist_name_prefix,
      1800, 1000);

    cv->Divide( nlayers_mvtx, 1 );
    for( int ilayer = 0; ilayer < nlayers_mvtx; ++ilayer )
    {

      const int layer = ilayer + first_layer_mvtx;

      // get histograms
      auto hnew = static_cast<TH1*>( qa_file_new->GetObjectChecked( Form( "%s%s_%i", prefix.Data(), tag.Data(), layer ), "TH1" ) );
      hnew->Scale( 1./hnew->GetEntries() );
      hnew->SetMinimum(0);
      hnew->GetXaxis()->SetMaxDigits(2);

      // reference
      auto href = qa_file_ref ? static_cast<TH1*>( qa_file_ref->GetObjectChecked( Form( "%s%s_%i", prefix.Data(), tag.Data(), layer ), "TH1" ) ) : nullptr;
      if( href )
      {
        href->Scale( 1./href->GetEntries() );
        href->SetMinimum(0);
        href->GetXaxis()->SetMaxDigits(2);
      }

      // draw
      cv->cd( ilayer+1 );
      DrawReference(hnew, href);
      gPad->SetRightMargin(0.15);
        
      auto line = VerticalLine( gPad, 0 );
      line->Draw();
    }

    return cv;

}