[u/mrichter/AliRoot.git] / PWG3 / hfe / AliHFEpostAnalysis.cxx

/**************************************************************************
* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
*                                                                        *
* Author: The ALICE Off-line Project.                                    *
* Contributors are mentioned in the code where appropriate.              *
*                                                                        *
* Permission to use, copy, modify and distribute this software and its   *
* documentation strictly for non-commercial purposes is hereby granted   *
* without fee, provided that the above copyright notice appears in all   *
* copies and that both the copyright notice and this permission notice   *
* appear in the supporting documentation. The authors make no claims     *
* about the suitability of this software for any purpose. It is          *
* provided "as is" without express or implied warranty.                  *
**************************************************************************/

/* $Id$ */

//
// Post analysis code
// Drawing nice pictures containing
//  - Efficiency
//  - Signal/Background
//  - PID Performance
//  More Post Analysis code will be added in time
//
//  Autor: Markus Fasel
//

#include <TAxis.h>
#include <TCanvas.h>
#include <TFile.h>
#include <TH1D.h>
#include <TLegend.h>
#include <TList.h>
#include "AliCFContainer.h"
#include "AliCFEffGrid.h"

#include "AliHFEcontainer.h"
#include "AliHFEcuts.h"
#include "AliHFEpostAnalysis.h"

ClassImp(AliHFEpostAnalysis)

//____________________________________________________________
AliHFEpostAnalysis::AliHFEpostAnalysis():
  TObject(),
  fResults(NULL),
  fAnalysisObjects(0),
  fEfficiencyContainer(NULL),
  fPIDperformance(NULL),
  fSignalToBackgroundMC(NULL)
{
  //
  // Default Constructor
  //
}

//____________________________________________________________
AliHFEpostAnalysis::AliHFEpostAnalysis(const AliHFEpostAnalysis &ref):
  TObject(ref),
  fResults(ref.fResults),
  fAnalysisObjects(ref.fAnalysisObjects),
  fEfficiencyContainer(ref.fEfficiencyContainer),
  fPIDperformance(ref.fPIDperformance),
  fSignalToBackgroundMC(ref.fSignalToBackgroundMC)
{
  //
  // Copy Constructor
  //
}

//____________________________________________________________
AliHFEpostAnalysis& AliHFEpostAnalysis::operator=(const AliHFEpostAnalysis &ref){
  //
  // Assignment Operator
  //
  TObject::operator=(ref);
  fResults = ref.fResults;
  fAnalysisObjects = ref.fAnalysisObjects;
  fPIDperformance = ref.fPIDperformance;
  fSignalToBackgroundMC = ref.fSignalToBackgroundMC;

  return *this;
}

//____________________________________________________________
AliHFEpostAnalysis::~AliHFEpostAnalysis(){
  //
  // Do not delete objects where we are not owner
  //
  if(fResults) delete fResults;
}

//____________________________________________________________
Int_t AliHFEpostAnalysis::SetTaskQA(const TList *input){
  //
  // Publish the results to the post analysis
  //
  Int_t nFound = 0;
  fPIDperformance = dynamic_cast<THnSparseF *>(input->FindObject("PIDperformance"));
  if(!fPIDperformance){
    AliError("Histogram fPIDperformance not found in the List of Outputs");
  } else {
    SETBIT(fAnalysisObjects, kPIDperf);
    nFound++;
  }
  fSignalToBackgroundMC = dynamic_cast<THnSparseF *>(input->FindObject("SignalToBackgroundMC"));
  if(!fSignalToBackgroundMC){
    AliError("Histogram fSignalToBackgroundMC not found in the list of outputs");
  } else {
    SETBIT(fAnalysisObjects, kSigBackg);
    nFound++;
  }
  AliInfo(Form("Found %d analysis objects", nFound));
  return nFound;
}

//____________________________________________________________
void AliHFEpostAnalysis::StoreOutput(const char *filename){
  //
  // Save the results produced in a rootfile
  //
  if(fResults){
    TFile *outfile = new TFile(filename, "RECREATE");
    outfile->cd();
    fResults->Write("HFEresults", TObject::kSingleKey);
    outfile->Close();
    delete outfile;
  }
}

//____________________________________________________________
void AliHFEpostAnalysis::DrawMCSignal2Background(){
  //
  // Draw the MC signal/background plots
  //
  if(!fSignalToBackgroundMC) return;

  // First Select everything within the first ITS Layer
  fSignalToBackgroundMC->GetAxis(5)->SetRange(2,2);
  TH1 *hEff[3], *hSB[3];
  // Select for different charge
  hEff[0] = CreateHistoSignalToBackgroundMC(0, 0);
  hEff[1] = CreateHistoSignalToBackgroundMC(0, 1);
  hEff[2] = CreateHistoSignalToBackgroundMC(0, 2);

  hSB[0] = CreateHistoSignalToBackgroundMC(1, 0);
  hSB[1] = CreateHistoSignalToBackgroundMC(1, 1);
  hSB[2] = CreateHistoSignalToBackgroundMC(1, 2);
 
  // Undo projections
  fSignalToBackgroundMC->GetAxis(5)->SetRange(0, fSignalToBackgroundMC->GetAxis(4)->GetNbins());

  // Prepare Canvas
  TCanvas *cMCSB = new TCanvas("cMCSB", "MC Sig/Backg studies", 800, 400);
  cMCSB->Divide(2);
  TLegend *leg;
  TH1 **sample[2] = {&hEff[0], &hSB[0]};
  const char *chargename[3] = {"All Charges", "Negative Charge", "Positive Charge"};
  for(Int_t isample = 0; isample < 2; isample++){
    leg = new TLegend(0.7, 0.1, 0.89, 0.3);
    leg->SetBorderSize(1); 
    leg->SetFillColor(kWhite);
    cMCSB->cd(isample + 1);
    for(Int_t icharge = 0; icharge < 3; icharge++){
      sample[isample][icharge]->Draw(icharge > 0?  "psame" : "p");
      leg->AddEntry(sample[isample][icharge], chargename[icharge], "p");
    }
    leg->Draw();
    gPad->Update();
  }
}

//____________________________________________________________
TH1 *AliHFEpostAnalysis::CreateHistoSignalToBackgroundMC(Int_t mode, Int_t charge){ 
  //
  // Make Efficiency / SB histograms for different charges
  //
  TH1 *hNom = NULL, *hDenom = NULL;
  if(charge) fSignalToBackgroundMC->GetAxis(3)->SetRange(charge, charge);
  // For Signal electrons we project axis 4 to everything > 0
  fSignalToBackgroundMC->GetAxis(4)->SetRange(2,3);
  hNom = fSignalToBackgroundMC->Projection(0);
  hNom->SetName("nom");
  fSignalToBackgroundMC->GetAxis(4)->SetRange(0, fSignalToBackgroundMC->GetAxis(4)->GetLast() + 1);
  if(mode == 1) fSignalToBackgroundMC->GetAxis(4)->SetRange(1,1);
  hDenom =  fSignalToBackgroundMC->Projection(0);
  hDenom->SetName("denom");
  if(mode == 1)  fSignalToBackgroundMC->GetAxis(4)->SetRange(0, fSignalToBackgroundMC->GetAxis(4)->GetLast() + 1);
  if(charge) fSignalToBackgroundMC->GetAxis(3)->SetRange(0, fSignalToBackgroundMC->GetAxis(3)->GetLast() + 1);

  TH1 *hEff = dynamic_cast<TH1D *>(hNom->Clone());
  if(hEff){
    TString hname, cname;
    hname = mode ? "sigToBack" : "sigEff";
    Color_t mycolor = kBlack;
    switch(charge){
      case 0: mycolor = kBlue; cname = "All"; break;
      case 1: mycolor = kBlack; cname = "Neg"; break;
      case 2: mycolor = kRed; cname ="Pos"; break;
      default: break;
    }
    hname += cname;
    hEff->SetName(hname);
    hEff->SetTitle(mode ? "Signal/Background" : "Signal/(Signal+Background)");
    hEff->Divide(hDenom);

    // Make nice plots
    hEff->GetXaxis()->SetTitle("p_{T} / GeV/c");
    hEff->GetYaxis()->SetTitle("Efficiency");
    hEff->SetStats(kFALSE);
    hEff->SetLineColor(kBlack);
    hEff->SetLineWidth(1);
    hEff->SetMarkerStyle(22);
    hEff->SetMarkerColor(mycolor);
  }

  delete hNom; delete hDenom;
  return hEff;
}

//____________________________________________________________
void AliHFEpostAnalysis::DrawEfficiency(){
  //
  // Draw the Efficiency
  // We show: 
  // + InAcceptance / Generated
  // + Signal / Generated
  // + Selected / Generated
  // + Selected / InAcceptance (Reconstructible)
  //
  TCanvas *cEff = new TCanvas("cEff", "Efficiency", 800, 600);
  cEff->Divide(2,2);
  if(!fEfficiencyContainer) return;
  AliCFContainer *tracks = fEfficiencyContainer->MakeMergedCFContainer("trackContCombined", "MC + Rec(reco) Track Information", "MCTrackCont:recTrackContReco");
  AliCFEffGrid *effCalc = new AliCFEffGrid("effCalc", "Efficiency Calculation Grid", *tracks);
  effCalc->CalculateEfficiency(AliHFEcuts::kStepMCInAcceptance, AliHFEcuts::kStepMCGenerated);
  TH1 *effReconstructibleP = effCalc->Project(0);
  effReconstructibleP->SetName("effReconstructibleP");
  effReconstructibleP->SetTitle("Efficiency of reconstructible tracks");
  effReconstructibleP->GetXaxis()->SetTitle("p_{T} / GeV/c");
  effReconstructibleP->GetYaxis()->SetTitle("Efficiency");
  effReconstructibleP->GetYaxis()->SetRangeUser(0.,1.);
  effReconstructibleP->SetMarkerStyle(22);
  effReconstructibleP->SetMarkerColor(kBlue);
  effReconstructibleP->SetLineColor(kBlack);
  effReconstructibleP->SetStats(kFALSE);
  cEff->cd(1);
  effReconstructibleP->Draw("e");
  effCalc->CalculateEfficiency(AliHFEcuts::kStepMCGeneratedZOutNoPileUp, AliHFEcuts::kStepMCGenerated);
  TH1 *effSignal = effCalc->Project(0);
  effSignal->SetName("effSignal");
  effSignal->SetTitle("Efficiency of Signal Electrons");
  effSignal->GetXaxis()->SetTitle("p_{T} / GeV/c");
  effSignal->GetYaxis()->SetTitle("Efficiency");
  effSignal->GetYaxis()->SetRangeUser(0., 1.);
  effSignal->SetMarkerStyle(22);
  effSignal->SetMarkerColor(kBlue);
  effSignal->SetLineColor(kBlack);
  effSignal->SetStats(kFALSE);
  cEff->cd(2);
  effSignal->Draw("e");
  effCalc->CalculateEfficiency(tracks->GetNStep() - 1, AliHFEcuts::kStepMCGenerated);
  TH1 *effPIDP = effCalc->Project(0);
  effPIDP->SetName("effPIDP");
  effPIDP->SetTitle("Efficiency of selected tracks");
  effPIDP->GetXaxis()->SetTitle("p_{T} / GeV/c");
  effPIDP->GetYaxis()->SetTitle("Efficiency");
  effPIDP->GetYaxis()->SetRangeUser(0.,1.);
  effPIDP->SetMarkerStyle(22);
  effPIDP->SetMarkerColor(kBlue);
  effPIDP->SetLineColor(kBlack);
  effPIDP->SetStats(kFALSE);
  cEff->cd(3);
  effPIDP->Draw("e");
  effCalc->CalculateEfficiency(tracks->GetNStep() - 1, AliHFEcuts::kStepMCInAcceptance);
  TH1 *effPIDAcc = effCalc->Project(0);
  effPIDAcc->SetName("effPIDAcc");
  effPIDAcc->SetTitle("Efficiency of selected tracks in acceptance");
  effPIDAcc->GetXaxis()->SetTitle("p_{T} / GeV/c");
  effPIDAcc->GetYaxis()->SetTitle("Efficiency");
  effPIDAcc->GetYaxis()->SetRangeUser(0.,1.);
  effPIDAcc->SetMarkerStyle(22);
  effPIDAcc->SetMarkerColor(kBlue);
  effPIDAcc->SetLineColor(kBlack);
  effPIDAcc->SetStats(kFALSE);
  cEff->cd(4);
  effPIDAcc->Draw("e");
  delete effCalc;
}

//____________________________________________________________
void AliHFEpostAnalysis::DrawPIDperformance(){
  //
  // Plotting Ratio histograms
  // + All electrons / all candidates (Purity for Electrons)
  // + All signal electrons / all electrons (Purity for signals)
  //

  if(!fPIDperformance) return;
  // Make projection
  TH1 *electronPurity[3], *signalPurity[3], *fakeContamination[3];
  electronPurity[0] = CreateHistoPIDperformance(0, 0);
  electronPurity[1] = CreateHistoPIDperformance(0, 1);
  electronPurity[2] = CreateHistoPIDperformance(0, 2);

  signalPurity[0] = CreateHistoPIDperformance(1, 0);
  signalPurity[1] = CreateHistoPIDperformance(1, 1);
  signalPurity[2] = CreateHistoPIDperformance(1, 2);

  fakeContamination[0] = CreateHistoPIDperformance(2, 0);
  fakeContamination[1] = CreateHistoPIDperformance(2, 1);
  fakeContamination[2] = CreateHistoPIDperformance(2, 2);

  // Draw output
  TCanvas *cRatios = new TCanvas("cRatios", "Ratio Plots", 800, 600);
  const char *chargename[3] = {"All Charges", "Negative Charge", "Positive Charge"};
  cRatios->Divide(2,2);
  TH1 **sample[3] = {&electronPurity[0], &signalPurity[0], &fakeContamination[0]};
  TLegend *leg;
  for(Int_t isample = 0; isample < 3; isample++){
    cRatios->cd(isample + 1);
    leg = new TLegend(0.7, 0.1, 0.89, 0.3);
    leg->SetBorderSize(1);
    leg->SetFillColor(kWhite);
    for(Int_t icharge = 0; icharge < 3; icharge++){
      leg->AddEntry(sample[isample][icharge], chargename[icharge], "p");
      sample[isample][icharge]->Draw(icharge > 0 ? "esame" : "e");
    }
    leg->Draw();
    gPad->Update();
  }
}

//____________________________________________________________
TH1 *AliHFEpostAnalysis::CreateHistoPIDperformance(Int_t mode, Int_t charge){
  //
  // Make Histograms for PID performance plots
  //
  fPIDperformance->GetAxis(4)->SetRange(0, fPIDperformance->GetAxis(4)->GetNbins()+1);
  fPIDperformance->GetAxis(3)->SetRange(0, fPIDperformance->GetAxis(3)->GetNbins() + 1);

  TH1 *hNom = NULL, *hDenom = NULL;
  TString hname, htitle, cname;
  Color_t mycolor = kBlack;
  if(charge) fPIDperformance->GetAxis(3)->SetRange(charge, charge);
  // Normalisation by all candidates - no restriction in axis 4 - only for mode == 1 
  if(mode == 1) fPIDperformance->GetAxis(4)->SetRange(2,3);
  hDenom = fPIDperformance->Projection(0);
  hDenom->Sumw2();
  hDenom->SetName("hDenom");
  if(mode == 1) fPIDperformance->GetAxis(4)->SetRange(0, fPIDperformance->GetAxis(4)->GetNbins() + 1);
  // Nominator need a restriction in the 4th axis
  switch(mode){
    case 0: // Electron purity
      fPIDperformance->GetAxis(4)->SetRange(2,3);
      hname = "electronPurity";
      htitle = "Electron Purity";
      break;
    case 1: // Signal purity
      fPIDperformance->GetAxis(4)->SetRange(3,3);   // here signal not divided into charm and beauty
      hname = "signalPurity";
      htitle = "Signal Purity";
      break;
    case 2: // Fake contamination
      fPIDperformance->GetAxis(4)->SetRange(1,1);
      hname = "fakeContamination";
      htitle = "Contamination of misidentified hadrons";
      break;
    default: break;
  }
  switch(charge){
    case 0: 
      cname = "All"; 
      mycolor = kBlue;
      break;
    case 1: 
      cname = "Neg"; 
      mycolor = kBlack;
      break;
    case 2: 
      cname = "Pos"; 
      mycolor = kRed;
      break;
  }
  hname += cname;
  hNom = fPIDperformance->Projection(0);
  hNom->Sumw2();
  hNom->SetName("hNom");
  // Reset axis
  fPIDperformance->GetAxis(4)->SetRange(0, fPIDperformance->GetAxis(4)->GetNbins()+1);
  if(charge) fPIDperformance->GetAxis(3)->SetRange(0, fPIDperformance->GetAxis(3)->GetNbins() + 1);

  // Create Efficiency histogram
  TH1 *hEff = dynamic_cast<TH1D *>(hNom->Clone());
  if(hEff){
    hEff->SetName(hname.Data());
    hEff->SetTitle(htitle.Data());
    hEff->Divide(hDenom);
    hEff->Scale(100.);
    hEff->GetXaxis()->SetTitle("p_{T} / GeV/c");
    hEff->GetYaxis()->SetTitle(mode < 2 ? "Purity / %" : "Contamination / %");
    hEff->GetYaxis()->SetRangeUser(0., 100.);
    hEff->SetStats(kFALSE);
    hEff->SetLineColor(kBlack);
    hEff->SetLineWidth(1);
    hEff->SetMarkerColor(mycolor);
    hEff->SetMarkerStyle(22);
    delete hNom; delete hDenom;
  }
  return hEff;
}

//____________________________________________________________
void AliHFEpostAnalysis::DrawCutEfficiency(Bool_t MC, Int_t source){
  //
  // Calculate efficiency for each cut step
  // Starting from MC steps 
  //
  TCanvas *output = new TCanvas("effCut", "Cut Step efficiency", 800, 600);
  output->cd();
  TLegend *leg = new TLegend(0.6, 0.7, 0.89, 0.89);
  leg->SetHeader("Cut Step:");
  leg->SetBorderSize(0);
  leg->SetFillColor(kWhite);
  leg->SetFillStyle(0);

  AliCFContainer *tracks = fEfficiencyContainer->MakeMergedCFContainer("mergedTracks", "Container for MC and reconstructed Track information", "MCTrackCont:recTrackContReco");
  Int_t nStepMC = fEfficiencyContainer->GetCFContainer("MCTrackCont")->GetNStep();
  AliDebug(1, Form("Number of MC Cut Steps: %d", nStepMC));
  const Int_t markerStart = 24;
  const Int_t colorStart = 1;
  TH1 *hTemp = NULL;

  if(MC){
    if(source > -1 && source < 4){
      AliInfo(Form("Setting source to %d", source))
      for(Int_t istep = 0; istep < tracks->GetNStep(); istep++) tracks->GetAxis(4, istep)->SetRange(source + 1, source + 1);
    }
  }
  AliCFEffGrid effcalc("cutEfficiency", "Cut step efficiency calculation", *tracks);
  Bool_t first = kTRUE;
  // Calculate efficiency for MC Steps
  Int_t histcounter = 0;
  if(MC){
    //
    // Draw plots related to MC values
    //
    effcalc.CalculateEfficiency(AliHFEcuts::kStepMCInAcceptance, AliHFEcuts::kStepMCGenerated);
    hTemp = effcalc.Project(0);
    hTemp->SetName(Form("hEff%d", histcounter));
    hTemp->SetTitle("Cut Step Efficiency");
    hTemp->SetMarkerColor(colorStart + histcounter);
    hTemp->SetMarkerStyle(markerStart + histcounter);
    hTemp->GetXaxis()->SetTitle("p / GeV/c");
    hTemp->GetYaxis()->SetTitle("Efficiency");
    hTemp->GetYaxis()->SetRangeUser(0., 2.);
    hTemp->SetStats(kFALSE);
    hTemp->Draw("ep");
    leg->AddEntry(hTemp, tracks->GetStepTitle(AliHFEcuts::kStepMCInAcceptance), "p");
    histcounter++;
    effcalc.CalculateEfficiency(nStepMC, AliHFEcuts::kStepMCGenerated);
    hTemp = effcalc.Project(0);
    hTemp->SetName("hEff2");
    hTemp->SetTitle("Cut Step Efficiency");
    hTemp->SetMarkerColor(colorStart + histcounter);
    hTemp->SetMarkerStyle(markerStart + histcounter);
    hTemp->GetXaxis()->SetTitle("p / GeV/c");
    hTemp->GetYaxis()->SetTitle("Efficiency");
    hTemp->GetYaxis()->SetRangeUser(0., 2.);
    hTemp->SetStats(kFALSE);
    hTemp->Draw("epsame");
    leg->AddEntry(hTemp, tracks->GetStepTitle(nStepMC), "p");
    first = kFALSE;
    histcounter++;
  }
  for(Int_t istep = nStepMC+1; istep < tracks->GetNStep(); istep++){
    effcalc.CalculateEfficiency(istep, istep - 1); 
    hTemp = effcalc.Project(0);
    hTemp->SetName(Form("hEff%d", istep));
    hTemp->SetTitle("Cut Step Efficiency");
    hTemp->SetMarkerColor(colorStart + histcounter);
    hTemp->SetMarkerStyle(markerStart + histcounter);
    hTemp->SetStats(kFALSE);
    hTemp->Draw(first ? "ep" : "epsame");
    hTemp->GetXaxis()->SetTitle("P / GeV/c");
    hTemp->GetYaxis()->SetTitle("Efficiency");
    hTemp->GetYaxis()->SetRangeUser(0., 2.);
    leg->AddEntry(hTemp, tracks->GetStepTitle(istep), "p");
    first = kFALSE;
    histcounter++;
  }
  leg->Draw();
  delete tracks;
}
Commit	Line	Data
d2af20c5	1	/**************************************************************************
	2	* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	3	* *
	4	* Author: The ALICE Off-line Project. *
	5	* Contributors are mentioned in the code where appropriate. *
	6	* *
	7	* Permission to use, copy, modify and distribute this software and its *
	8	* documentation strictly for non-commercial purposes is hereby granted *
	9	* without fee, provided that the above copyright notice appears in all *
	10	* copies and that both the copyright notice and this permission notice *
	11	* appear in the supporting documentation. The authors make no claims *
	12	* about the suitability of this software for any purpose. It is *
	13	* provided "as is" without express or implied warranty. *
	14	**************************************************************************/
27de2dfb	15
	16	/* $Id$ */
	17
d2af20c5	18	//
	19	// Post analysis code
	20	// Drawing nice pictures containing
	21	// - Efficiency
	22	// - Signal/Background
	23	// - PID Performance
	24	// More Post Analysis code will be added in time
	25	//
	26	// Autor: Markus Fasel
	27	//
3a72645a	28
d2af20c5	29	#include <TAxis.h>
	30	#include <TCanvas.h>
	31	#include <TFile.h>
	32	#include <TH1D.h>
	33	#include <TLegend.h>
	34	#include <TList.h>
	35	#include "AliCFContainer.h"
	36	#include "AliCFEffGrid.h"
	37
3a72645a	38	#include "AliHFEcontainer.h"
d2af20c5	39	#include "AliHFEcuts.h"
	40	#include "AliHFEpostAnalysis.h"
	41
	42	ClassImp(AliHFEpostAnalysis)
	43
	44	//____________________________________________________________
	45	AliHFEpostAnalysis::AliHFEpostAnalysis():
	46	TObject(),
	47	fResults(NULL),
	48	fAnalysisObjects(0),
	49	fEfficiencyContainer(NULL),
	50	fPIDperformance(NULL),
	51	fSignalToBackgroundMC(NULL)
	52	{
	53	//
	54	// Default Constructor
	55	//
	56	}
	57
	58	//____________________________________________________________
	59	AliHFEpostAnalysis::AliHFEpostAnalysis(const AliHFEpostAnalysis &ref):
	60	TObject(ref),
	61	fResults(ref.fResults),
	62	fAnalysisObjects(ref.fAnalysisObjects),
	63	fEfficiencyContainer(ref.fEfficiencyContainer),
	64	fPIDperformance(ref.fPIDperformance),
	65	fSignalToBackgroundMC(ref.fSignalToBackgroundMC)
	66	{
	67	//
	68	// Copy Constructor
	69	//
	70	}
	71
	72	//____________________________________________________________
	73	AliHFEpostAnalysis& AliHFEpostAnalysis::operator=(const AliHFEpostAnalysis &ref){
	74	//
	75	// Assignment Operator
	76	//
	77	TObject::operator=(ref);
	78	fResults = ref.fResults;
	79	fAnalysisObjects = ref.fAnalysisObjects;
	80	fPIDperformance = ref.fPIDperformance;
	81	fSignalToBackgroundMC = ref.fSignalToBackgroundMC;
	82
	83	return *this;
	84	}
	85
	86	//____________________________________________________________
	87	AliHFEpostAnalysis::~AliHFEpostAnalysis(){
	88	//
	89	// Do not delete objects where we are not owner
	90	//
	91	if(fResults) delete fResults;
	92	}
	93
	94	//____________________________________________________________
3a72645a	95	Int_t AliHFEpostAnalysis::SetTaskQA(const TList *input){
d2af20c5	96	//
	97	// Publish the results to the post analysis
	98	//
	99	Int_t nFound = 0;
d2af20c5	100	fPIDperformance = dynamic_cast<THnSparseF *>(input->FindObject("PIDperformance"));
	101	if(!fPIDperformance){
	102	AliError("Histogram fPIDperformance not found in the List of Outputs");
	103	} else {
	104	SETBIT(fAnalysisObjects, kPIDperf);
	105	nFound++;
	106	}
	107	fSignalToBackgroundMC = dynamic_cast<THnSparseF *>(input->FindObject("SignalToBackgroundMC"));
	108	if(!fSignalToBackgroundMC){
	109	AliError("Histogram fSignalToBackgroundMC not found in the list of outputs");
	110	} else {
	111	SETBIT(fAnalysisObjects, kSigBackg);
	112	nFound++;
	113	}
	114	AliInfo(Form("Found %d analysis objects", nFound));
	115	return nFound;
	116	}
	117
	118	//____________________________________________________________
	119	void AliHFEpostAnalysis::StoreOutput(const char *filename){
	120	//
	121	// Save the results produced in a rootfile
	122	//
	123	if(fResults){
	124	TFile *outfile = new TFile(filename, "RECREATE");
	125	outfile->cd();
	126	fResults->Write("HFEresults", TObject::kSingleKey);
	127	outfile->Close();
	128	delete outfile;
	129	}
	130	}
	131
	132	//____________________________________________________________
	133	void AliHFEpostAnalysis::DrawMCSignal2Background(){
	134	//
	135	// Draw the MC signal/background plots
	136	//
	137	if(!fSignalToBackgroundMC) return;
	138
	139	// First Select everything within the first ITS Layer
70da6c5a	140	fSignalToBackgroundMC->GetAxis(5)->SetRange(2,2);
	141	TH1 hEff[3], hSB[3];
	142	// Select for different charge
	143	hEff[0] = CreateHistoSignalToBackgroundMC(0, 0);
	144	hEff[1] = CreateHistoSignalToBackgroundMC(0, 1);
	145	hEff[2] = CreateHistoSignalToBackgroundMC(0, 2);
	146
	147	hSB[0] = CreateHistoSignalToBackgroundMC(1, 0);
	148	hSB[1] = CreateHistoSignalToBackgroundMC(1, 1);
	149	hSB[2] = CreateHistoSignalToBackgroundMC(1, 2);
	150
	151	// Undo projections
	152	fSignalToBackgroundMC->GetAxis(5)->SetRange(0, fSignalToBackgroundMC->GetAxis(4)->GetNbins());
	153
d2af20c5	154	// Prepare Canvas
70da6c5a	155	TCanvas *cMCSB = new TCanvas("cMCSB", "MC Sig/Backg studies", 800, 400);
	156	cMCSB->Divide(2);
	157	TLegend *leg;
	158	TH1 **sample[2] = {&hEff[0], &hSB[0]};
	159	const char *chargename[3] = {"All Charges", "Negative Charge", "Positive Charge"};
	160	for(Int_t isample = 0; isample < 2; isample++){
	161	leg = new TLegend(0.7, 0.1, 0.89, 0.3);
	162	leg->SetBorderSize(1);
	163	leg->SetFillColor(kWhite);
	164	cMCSB->cd(isample + 1);
	165	for(Int_t icharge = 0; icharge < 3; icharge++){
	166	sample[isample][icharge]->Draw(icharge > 0? "psame" : "p");
	167	leg->AddEntry(sample[isample][icharge], chargename[icharge], "p");
	168	}
	169	leg->Draw();
	170	gPad->Update();
	171	}
	172	}
	173
	174	//____________________________________________________________
	175	TH1 *AliHFEpostAnalysis::CreateHistoSignalToBackgroundMC(Int_t mode, Int_t charge){
	176	//
	177	// Make Efficiency / SB histograms for different charges
	178	//
	179	TH1 hNom = NULL, hDenom = NULL;
	180	if(charge) fSignalToBackgroundMC->GetAxis(3)->SetRange(charge, charge);
	181	// For Signal electrons we project axis 4 to everything > 0
	182	fSignalToBackgroundMC->GetAxis(4)->SetRange(2,3);
	183	hNom = fSignalToBackgroundMC->Projection(0);
	184	hNom->SetName("nom");
	185	fSignalToBackgroundMC->GetAxis(4)->SetRange(0, fSignalToBackgroundMC->GetAxis(4)->GetLast() + 1);
	186	if(mode == 1) fSignalToBackgroundMC->GetAxis(4)->SetRange(1,1);
	187	hDenom = fSignalToBackgroundMC->Projection(0);
	188	hDenom->SetName("denom");
	189	if(mode == 1) fSignalToBackgroundMC->GetAxis(4)->SetRange(0, fSignalToBackgroundMC->GetAxis(4)->GetLast() + 1);
	190	if(charge) fSignalToBackgroundMC->GetAxis(3)->SetRange(0, fSignalToBackgroundMC->GetAxis(3)->GetLast() + 1);
	191
	192	TH1 hEff = dynamic_cast<TH1D >(hNom->Clone());
bf892a6a	193	if(hEff){
	194	TString hname, cname;
	195	hname = mode ? "sigToBack" : "sigEff";
	196	Color_t mycolor = kBlack;
	197	switch(charge){
	198	case 0: mycolor = kBlue; cname = "All"; break;
	199	case 1: mycolor = kBlack; cname = "Neg"; break;
	200	case 2: mycolor = kRed; cname ="Pos"; break;
	201	default: break;
	202	}
	203	hname += cname;
	204	hEff->SetName(hname);
	205	hEff->SetTitle(mode ? "Signal/Background" : "Signal/(Signal+Background)");
	206	hEff->Divide(hDenom);
70da6c5a	207
bf892a6a	208	// Make nice plots
	209	hEff->GetXaxis()->SetTitle("p_{T} / GeV/c");
	210	hEff->GetYaxis()->SetTitle("Efficiency");
	211	hEff->SetStats(kFALSE);
	212	hEff->SetLineColor(kBlack);
	213	hEff->SetLineWidth(1);
	214	hEff->SetMarkerStyle(22);
	215	hEff->SetMarkerColor(mycolor);
	216	}
d2af20c5	217
70da6c5a	218	delete hNom; delete hDenom;
70da6c5a	219	return hEff;
d2af20c5	220	}
	221
	222	//____________________________________________________________
	223	void AliHFEpostAnalysis::DrawEfficiency(){
	224	//
	225	// Draw the Efficiency
	226	// We show:
	227	// + InAcceptance / Generated
	228	// + Signal / Generated
	229	// + Selected / Generated
	230	// + Selected / InAcceptance (Reconstructible)
	231	//
	232	TCanvas *cEff = new TCanvas("cEff", "Efficiency", 800, 600);
	233	cEff->Divide(2,2);
	234	if(!fEfficiencyContainer) return;
3a72645a	235	AliCFContainer *tracks = fEfficiencyContainer->MakeMergedCFContainer("trackContCombined", "MC + Rec(reco) Track Information", "MCTrackCont:recTrackContReco");
3a72645a	236	AliCFEffGrid effCalc = new AliCFEffGrid("effCalc", "Efficiency Calculation Grid", tracks);
d2af20c5	237	effCalc->CalculateEfficiency(AliHFEcuts::kStepMCInAcceptance, AliHFEcuts::kStepMCGenerated);
	238	TH1 *effReconstructibleP = effCalc->Project(0);
	239	effReconstructibleP->SetName("effReconstructibleP");
	240	effReconstructibleP->SetTitle("Efficiency of reconstructible tracks");
	241	effReconstructibleP->GetXaxis()->SetTitle("p_{T} / GeV/c");
	242	effReconstructibleP->GetYaxis()->SetTitle("Efficiency");
	243	effReconstructibleP->GetYaxis()->SetRangeUser(0.,1.);
	244	effReconstructibleP->SetMarkerStyle(22);
	245	effReconstructibleP->SetMarkerColor(kBlue);
	246	effReconstructibleP->SetLineColor(kBlack);
	247	effReconstructibleP->SetStats(kFALSE);
	248	cEff->cd(1);
	249	effReconstructibleP->Draw("e");
3a72645a	250	effCalc->CalculateEfficiency(AliHFEcuts::kStepMCGeneratedZOutNoPileUp, AliHFEcuts::kStepMCGenerated);
d2af20c5	251	TH1 *effSignal = effCalc->Project(0);
	252	effSignal->SetName("effSignal");
	253	effSignal->SetTitle("Efficiency of Signal Electrons");
	254	effSignal->GetXaxis()->SetTitle("p_{T} / GeV/c");
	255	effSignal->GetYaxis()->SetTitle("Efficiency");
	256	effSignal->GetYaxis()->SetRangeUser(0., 1.);
	257	effSignal->SetMarkerStyle(22);
	258	effSignal->SetMarkerColor(kBlue);
	259	effSignal->SetLineColor(kBlack);
	260	effSignal->SetStats(kFALSE);
	261	cEff->cd(2);
	262	effSignal->Draw("e");
3a72645a	263	effCalc->CalculateEfficiency(tracks->GetNStep() - 1, AliHFEcuts::kStepMCGenerated);
d2af20c5	264	TH1 *effPIDP = effCalc->Project(0);
	265	effPIDP->SetName("effPIDP");
	266	effPIDP->SetTitle("Efficiency of selected tracks");
	267	effPIDP->GetXaxis()->SetTitle("p_{T} / GeV/c");
	268	effPIDP->GetYaxis()->SetTitle("Efficiency");
	269	effPIDP->GetYaxis()->SetRangeUser(0.,1.);
	270	effPIDP->SetMarkerStyle(22);
	271	effPIDP->SetMarkerColor(kBlue);
	272	effPIDP->SetLineColor(kBlack);
	273	effPIDP->SetStats(kFALSE);
	274	cEff->cd(3);
	275	effPIDP->Draw("e");
3a72645a	276	effCalc->CalculateEfficiency(tracks->GetNStep() - 1, AliHFEcuts::kStepMCInAcceptance);
d2af20c5	277	TH1 *effPIDAcc = effCalc->Project(0);
	278	effPIDAcc->SetName("effPIDAcc");
	279	effPIDAcc->SetTitle("Efficiency of selected tracks in acceptance");
	280	effPIDAcc->GetXaxis()->SetTitle("p_{T} / GeV/c");
	281	effPIDAcc->GetYaxis()->SetTitle("Efficiency");
	282	effPIDAcc->GetYaxis()->SetRangeUser(0.,1.);
	283	effPIDAcc->SetMarkerStyle(22);
	284	effPIDAcc->SetMarkerColor(kBlue);
	285	effPIDAcc->SetLineColor(kBlack);
	286	effPIDAcc->SetStats(kFALSE);
	287	cEff->cd(4);
	288	effPIDAcc->Draw("e");
	289	delete effCalc;
	290	}
	291
	292	//____________________________________________________________
	293	void AliHFEpostAnalysis::DrawPIDperformance(){
	294	//
	295	// Plotting Ratio histograms
	296	// + All electrons / all candidates (Purity for Electrons)
	297	// + All signal electrons / all electrons (Purity for signals)
d2af20c5	298	//
	299
	300	if(!fPIDperformance) return;
	301	// Make projection
70da6c5a	302	TH1 electronPurity[3], signalPurity[3], *fakeContamination[3];
	303	electronPurity[0] = CreateHistoPIDperformance(0, 0);
	304	electronPurity[1] = CreateHistoPIDperformance(0, 1);
	305	electronPurity[2] = CreateHistoPIDperformance(0, 2);
d2af20c5	306
70da6c5a	307	signalPurity[0] = CreateHistoPIDperformance(1, 0);
	308	signalPurity[1] = CreateHistoPIDperformance(1, 1);
	309	signalPurity[2] = CreateHistoPIDperformance(1, 2);
d2af20c5	310
70da6c5a	311	fakeContamination[0] = CreateHistoPIDperformance(2, 0);
	312	fakeContamination[1] = CreateHistoPIDperformance(2, 1);
	313	fakeContamination[2] = CreateHistoPIDperformance(2, 2);
d2af20c5	314
d2af20c5	315	// Draw output
d2af20c5	316	TCanvas *cRatios = new TCanvas("cRatios", "Ratio Plots", 800, 600);
70da6c5a	317	const char *chargename[3] = {"All Charges", "Negative Charge", "Positive Charge"};
d2af20c5	318	cRatios->Divide(2,2);
70da6c5a	319	TH1 **sample[3] = {&electronPurity[0], &signalPurity[0], &fakeContamination[0]};
	320	TLegend *leg;
	321	for(Int_t isample = 0; isample < 3; isample++){
	322	cRatios->cd(isample + 1);
	323	leg = new TLegend(0.7, 0.1, 0.89, 0.3);
	324	leg->SetBorderSize(1);
	325	leg->SetFillColor(kWhite);
	326	for(Int_t icharge = 0; icharge < 3; icharge++){
	327	leg->AddEntry(sample[isample][icharge], chargename[icharge], "p");
	328	sample[isample][icharge]->Draw(icharge > 0 ? "esame" : "e");
	329	}
	330	leg->Draw();
	331	gPad->Update();
	332	}
d2af20c5	333	}
d2af20c5	334
70da6c5a	335	//____________________________________________________________
	336	TH1 *AliHFEpostAnalysis::CreateHistoPIDperformance(Int_t mode, Int_t charge){
	337	//
	338	// Make Histograms for PID performance plots
	339	//
faee3b18	340	fPIDperformance->GetAxis(4)->SetRange(0, fPIDperformance->GetAxis(4)->GetNbins()+1);
	341	fPIDperformance->GetAxis(3)->SetRange(0, fPIDperformance->GetAxis(3)->GetNbins() + 1);
	342
70da6c5a	343	TH1 hNom = NULL, hDenom = NULL;
bf892a6a	344	TString hname, htitle, cname;
70da6c5a	345	Color_t mycolor = kBlack;
faee3b18	346	if(charge) fPIDperformance->GetAxis(3)->SetRange(charge, charge);
70da6c5a	347	// Normalisation by all candidates - no restriction in axis 4 - only for mode == 1
faee3b18	348	if(mode == 1) fPIDperformance->GetAxis(4)->SetRange(2,3);
70da6c5a	349	hDenom = fPIDperformance->Projection(0);
	350	hDenom->Sumw2();
	351	hDenom->SetName("hDenom");
faee3b18	352	if(mode == 1) fPIDperformance->GetAxis(4)->SetRange(0, fPIDperformance->GetAxis(4)->GetNbins() + 1);
70da6c5a	353	// Nominator need a restriction in the 4th axis
	354	switch(mode){
	355	case 0: // Electron purity
	356	fPIDperformance->GetAxis(4)->SetRange(2,3);
bf892a6a	357	hname = "electronPurity";
bf892a6a	358	htitle = "Electron Purity";
70da6c5a	359	break;
	360	case 1: // Signal purity
	361	fPIDperformance->GetAxis(4)->SetRange(3,3); // here signal not divided into charm and beauty
bf892a6a	362	hname = "signalPurity";
bf892a6a	363	htitle = "Signal Purity";
70da6c5a	364	break;
	365	case 2: // Fake contamination
	366	fPIDperformance->GetAxis(4)->SetRange(1,1);
bf892a6a	367	hname = "fakeContamination";
bf892a6a	368	htitle = "Contamination of misidentified hadrons";
70da6c5a	369	break;
	370	default: break;
	371	}
	372	switch(charge){
	373	case 0:
bf892a6a	374	cname = "All";
70da6c5a	375	mycolor = kBlue;
	376	break;
	377	case 1:
bf892a6a	378	cname = "Neg";
70da6c5a	379	mycolor = kBlack;
	380	break;
	381	case 2:
bf892a6a	382	cname = "Pos";
70da6c5a	383	mycolor = kRed;
	384	break;
	385	}
bf892a6a	386	hname += cname;
70da6c5a	387	hNom = fPIDperformance->Projection(0);
	388	hNom->Sumw2();
	389	hNom->SetName("hNom");
	390	// Reset axis
faee3b18	391	fPIDperformance->GetAxis(4)->SetRange(0, fPIDperformance->GetAxis(4)->GetNbins()+1);
faee3b18	392	if(charge) fPIDperformance->GetAxis(3)->SetRange(0, fPIDperformance->GetAxis(3)->GetNbins() + 1);
70da6c5a	393
	394	// Create Efficiency histogram
	395	TH1 hEff = dynamic_cast<TH1D >(hNom->Clone());
bf892a6a	396	if(hEff){
	397	hEff->SetName(hname.Data());
	398	hEff->SetTitle(htitle.Data());
	399	hEff->Divide(hDenom);
	400	hEff->Scale(100.);
	401	hEff->GetXaxis()->SetTitle("p_{T} / GeV/c");
	402	hEff->GetYaxis()->SetTitle(mode < 2 ? "Purity / %" : "Contamination / %");
	403	hEff->GetYaxis()->SetRangeUser(0., 100.);
	404	hEff->SetStats(kFALSE);
	405	hEff->SetLineColor(kBlack);
	406	hEff->SetLineWidth(1);
	407	hEff->SetMarkerColor(mycolor);
	408	hEff->SetMarkerStyle(22);
	409	delete hNom; delete hDenom;
	410	}
70da6c5a	411	return hEff;
	412	}
	413
	414	//____________________________________________________________
67fe7bd0	415	void AliHFEpostAnalysis::DrawCutEfficiency(Bool_t MC, Int_t source){
70da6c5a	416	//
	417	// Calculate efficiency for each cut step
	418	// Starting from MC steps
	419	//
	420	TCanvas *output = new TCanvas("effCut", "Cut Step efficiency", 800, 600);
	421	output->cd();
	422	TLegend *leg = new TLegend(0.6, 0.7, 0.89, 0.89);
	423	leg->SetHeader("Cut Step:");
	424	leg->SetBorderSize(0);
	425	leg->SetFillColor(kWhite);
	426	leg->SetFillStyle(0);
3a72645a	427
	428	AliCFContainer *tracks = fEfficiencyContainer->MakeMergedCFContainer("mergedTracks", "Container for MC and reconstructed Track information", "MCTrackCont:recTrackContReco");
	429	Int_t nStepMC = fEfficiencyContainer->GetCFContainer("MCTrackCont")->GetNStep();
	430	AliDebug(1, Form("Number of MC Cut Steps: %d", nStepMC));
	431	const Int_t markerStart = 24;
	432	const Int_t colorStart = 1;
70da6c5a	433	TH1 *hTemp = NULL;
67fe7bd0	434
	435	if(MC){
	436	if(source > -1 && source < 4){
	437	AliInfo(Form("Setting source to %d", source))
3a72645a	438	for(Int_t istep = 0; istep < tracks->GetNStep(); istep++) tracks->GetAxis(4, istep)->SetRange(source + 1, source + 1);
67fe7bd0	439	}
67fe7bd0	440	}
3a72645a	441	AliCFEffGrid effcalc("cutEfficiency", "Cut step efficiency calculation", *tracks);
67fe7bd0	442	Bool_t first = kTRUE;
70da6c5a	443	// Calculate efficiency for MC Steps
3a72645a	444	Int_t histcounter = 0;
67fe7bd0	445	if(MC){
	446	//
	447	// Draw plots related to MC values
	448	//
	449	effcalc.CalculateEfficiency(AliHFEcuts::kStepMCInAcceptance, AliHFEcuts::kStepMCGenerated);
	450	hTemp = effcalc.Project(0);
3a72645a	451	hTemp->SetName(Form("hEff%d", histcounter));
67fe7bd0	452	hTemp->SetTitle("Cut Step Efficiency");
3a72645a	453	hTemp->SetMarkerColor(colorStart + histcounter);
	454	hTemp->SetMarkerStyle(markerStart + histcounter);
	455	hTemp->GetXaxis()->SetTitle("p / GeV/c");
67fe7bd0	456	hTemp->GetYaxis()->SetTitle("Efficiency");
	457	hTemp->GetYaxis()->SetRangeUser(0., 2.);
	458	hTemp->SetStats(kFALSE);
	459	hTemp->Draw("ep");
3a72645a	460	leg->AddEntry(hTemp, tracks->GetStepTitle(AliHFEcuts::kStepMCInAcceptance), "p");
	461	histcounter++;
	462	effcalc.CalculateEfficiency(nStepMC, AliHFEcuts::kStepMCGenerated);
67fe7bd0	463	hTemp = effcalc.Project(0);
	464	hTemp->SetName("hEff2");
	465	hTemp->SetTitle("Cut Step Efficiency");
3a72645a	466	hTemp->SetMarkerColor(colorStart + histcounter);
	467	hTemp->SetMarkerStyle(markerStart + histcounter);
	468	hTemp->GetXaxis()->SetTitle("p / GeV/c");
67fe7bd0	469	hTemp->GetYaxis()->SetTitle("Efficiency");
	470	hTemp->GetYaxis()->SetRangeUser(0., 2.);
	471	hTemp->SetStats(kFALSE);
	472	hTemp->Draw("epsame");
3a72645a	473	leg->AddEntry(hTemp, tracks->GetStepTitle(nStepMC), "p");
67fe7bd0	474	first = kFALSE;
3a72645a	475	histcounter++;
67fe7bd0	476	}
3a72645a	477	for(Int_t istep = nStepMC+1; istep < tracks->GetNStep(); istep++){
3a72645a	478	effcalc.CalculateEfficiency(istep, istep - 1);
70da6c5a	479	hTemp = effcalc.Project(0);
70da6c5a	480	hTemp->SetName(Form("hEff%d", istep));
67fe7bd0	481	hTemp->SetTitle("Cut Step Efficiency");
3a72645a	482	hTemp->SetMarkerColor(colorStart + histcounter);
3a72645a	483	hTemp->SetMarkerStyle(markerStart + histcounter);
70da6c5a	484	hTemp->SetStats(kFALSE);
67fe7bd0	485	hTemp->Draw(first ? "ep" : "epsame");
70da6c5a	486	hTemp->GetXaxis()->SetTitle("P / GeV/c");
	487	hTemp->GetYaxis()->SetTitle("Efficiency");
	488	hTemp->GetYaxis()->SetRangeUser(0., 2.);
3a72645a	489	leg->AddEntry(hTemp, tracks->GetStepTitle(istep), "p");
67fe7bd0	490	first = kFALSE;
3a72645a	491	histcounter++;
70da6c5a	492	}
70da6c5a	493	leg->Draw();
3a72645a	494	delete tracks;
70da6c5a	495	}