makePlots.C

Go to the documentation of this file.

//Currently, script is setup to make average correlation graph vs number of iterations for pi^{0} kinetic energy.

#include <vector>
#include <string>

#include "TF1.h"
#include "TH1.h"
#include "TFile.h"
#include "TGraph.h"
#include "TGraphErrors.h"
#include "TTree.h"
#include "TCanvas.h"
#include "TLegend.h"
#include "rootlogon.C"


void makePlots()
{
  
  TFile* file = new TFile("/nova/ana/users/kalra/tmva/SystOptimisation/Unfolding/RootFilesFor_UnfoldingMetrics_UseInMakePlotsMacro/ForPi0KE/UnfoldingMetricsEnergy.root");


  int iIter = 100;
  int iSyst = 79;

  float fChiSq[iIter];
  float fCor[iIter];
  float fBias[iIter];
  float fCoverage[iIter];

  float fChiSqErr[iIter];
  float fCorErr[iIter];
  float fBiasErr[iIter];
  float fCoverageErr[iIter];


  TGraph* hChiSqCov[iSyst];
  TGraph* hAverageCor[iSyst];
  TGraph* hBias[iSyst];
  TGraph* hCoverage[iSyst];


  for(int i = 0; i < iSyst; i++){
    char biasname[50];
    char coveragename[50];
    char chi2name[50];
    char correlationname[50];
    sprintf(biasname, "%s_%i_bias", "unf_RecopiKE",  i);
    sprintf(coveragename, "%s_%i_coverage",  "unf_RecopiKE", i);
    sprintf(chi2name, "%s_%i_chi", "unf_RecopiKE", i);
    sprintf(correlationname, "%s_%i_correlation", "unf_RecopiKE", i);

    // sprintf(biasname, "%s_%i_bias", "unf_RecopiCOS",  i);
    // sprintf(coveragename, "%s_%i_coverage",  "unf_RecopiCOS", i);
    //sprintf(chi2name, "%s_%i_chi", "unf_RecopiCOS", i);
    //sprintf(correlationname, "%s_%i_correlation", "unf_RecopiCOS", i);
   
    hChiSqCov[i]   = (TGraph*)file->Get(chi2name);
    hAverageCor[i] = (TGraph*)file->Get(correlationname);
    hBias[i]       = (TGraph*)file->Get(biasname);
    hCoverage[i]   = (TGraph*)file->Get(coveragename);
  }

  for(int j = 0; j < iIter; j++){
    TH1F* hChi2Holder = new TH1F("hChi2Holder",";#chi^{2};Count",100,0,200);
    TH1F* hCorrel = new TH1F("hCorrel",";Correlation;Count",100,0,1);
    TH1F* hBiasHolder = new TH1F("hBiasHolder",";Bias;Count",100,-10,10);
    TH1F* hCoverHolder = new TH1F("hCoverHolder",";Coverage;Count",100,0,1);

    for(int i = 0; i < iSyst; i++){
      Double_t x;
      Double_t y;

      hChiSqCov[i]->GetPoint(j,x,y);
      hChi2Holder->Fill(y);

      hAverageCor[i]->GetPoint(j,x,y);
      //std::cout << x << "\t" << y << std::endl;
      hCorrel->Fill(y);

      hBias[i]->GetPoint(j,x,y);
      hBiasHolder->Fill(y);   

      hCoverage[i]->GetPoint(j,x,y);
      hCoverHolder->Fill(y);   
    }

    fChiSq[j] = (float)hChi2Holder->GetMean();
    fCor[j]   = (float)hCorrel->GetMean();
    fBias[j]  = (float)hBiasHolder->GetMean();
    fCoverage[j] = (float)hCoverHolder->GetMean();

    fChiSqErr[j] = (float)hChi2Holder->GetRMS();
    fCorErr[j]   = (float)hCorrel->GetRMS();
    fBiasErr[j]  = (float)hBiasHolder->GetRMS();
    fCoverageErr[j] = (float)hCoverHolder->GetRMS();

  }

  float xArray[100] = {0};
  for(int i = 0; i < iIter; i++){
    xArray[i] = i+1;
  }
  float xArrayErr[100] = {0};

  for(int i = 0; i < iIter; i++){
    std::cout << i << ": " << fChiSq[i] << "\t"
              << fCor[i] << "\t" << fBias[i] << "\t" 
              << fCoverage[i] << "\t" << std::endl;
  }
  TGraphErrors* hCoverageFinal = 
    new TGraphErrors(100, xArray, fCoverage, xArrayErr, fCoverageErr);

  TGraphErrors* hChiSqFinal = 
    new TGraphErrors(100, xArray, fChiSq, xArrayErr, fChiSqErr);

  TGraphErrors* hCorrelationFinal = 
    new TGraphErrors(100, xArray, fCor, xArrayErr, fCorErr);

  TGraphErrors* hBiasFinal = 
    new TGraphErrors(100, xArray, fBias, xArrayErr, fBiasErr);

 
 
  TH1F* hFrame1 = new TH1F("hFrame1", ";Iterations; #chi^{2}",
                           500,0,100);
  for(int ibin = 0; ibin < hFrame1->GetSize(); ibin++)
    hFrame1->SetBinContent(ibin,-100);
              

  TH1F* hFrame2 = new TH1F("hFrame2", ";Iterations; Average Correlation",
                           500,0,100);
  for(int ibin = 0; ibin < hFrame1->GetSize(); ibin++)
    hFrame2->SetBinContent(ibin,-100);
    
  TH1F* hFrame3 = new TH1F("hFrame3", ";Iterations; Average Bias",
                           500,0,100);
  for(int ibin = 0; ibin < hFrame1->GetSize(); ibin++)
    hFrame3->SetBinContent(ibin,-100);
  
  TH1F* hFrame4 = new TH1F("hFrame4", ";Iterations; Average Coverage",
                           500,0,100);
  for(int ibin = 0; ibin < hFrame1->GetSize(); ibin++)
    hFrame4->SetBinContent(ibin,-100);


  hFrame1->GetXaxis()->CenterTitle();
  hFrame1->GetYaxis()->CenterTitle();

  hFrame2->GetXaxis()->CenterTitle();
  hFrame2->GetYaxis()->CenterTitle();
  hFrame3->GetXaxis()->CenterTitle();
  hFrame3->GetYaxis()->CenterTitle();
  hFrame4->GetXaxis()->CenterTitle();
  hFrame4->GetYaxis()->CenterTitle();

 
  TCanvas* c1 = new TCanvas("c1","c1");
  hChiSqFinal->SetMarkerStyle(7);
  hFrame1->GetYaxis()->SetRangeUser(0,110);
  hFrame1->Draw();
  hChiSqFinal->Draw("PL same");
  c1->SaveAs("unf_RecopiKETuned_chisq_averaged.pdf");

  TCanvas* c2 = new TCanvas("c2","c2");
  hBiasFinal->SetMarkerStyle(7);
  hFrame3->GetYaxis()->SetRangeUser(0,0.03);
  hFrame3->Draw();
  hBiasFinal->Draw("PL same");
  c2->SaveAs("unf_RecopiKETuned_bias_averaged.pdf");
  
  TCanvas* c3 = new TCanvas("c3","c3");
  hCorrelationFinal->SetMarkerStyle(7);
  hFrame2->GetYaxis()->SetRangeUser(0,1.0);
  hFrame2->Draw();
  hCorrelationFinal->Draw("PL same");
  c3->SaveAs("unf_RecopiKETuned_correlation_averaged.pdf");

  TCanvas* c4 = new TCanvas("c4","c4");
  hCoverageFinal->SetMarkerStyle(7);
  hFrame4->GetYaxis()->SetRangeUser(0,1.0);
  hFrame4->Draw();
  hCoverageFinal->Draw("PL same");
  c4->SaveAs("unf_RecopiKETuned_coverage_averaged.pdf");


  // file->Close();

}