// and comparison with the MC truth and cut variables distributions.
//
// Authors: A.Dainese, andrea.dainese@lnl.infn.it
-// and Chiara Bianchin, chiara.bianchin@pd.infn.it
+// Chiara Bianchin, chiara.bianchin@pd.infn.it (invariant mass)
+// Carmelo Di Giglio, carmelo.digiglio@ba.infn.it (like sign)
/////////////////////////////////////////////////////////////
#include <Riostream.h>
#include <TClonesArray.h>
+#include <TCanvas.h>
#include <TNtuple.h>
#include <TList.h>
#include <TH1F.h>
#include <TH2F.h>
#include <TDatabasePDG.h>
+#include <AliAnalysisDataSlot.h>
+#include <AliAnalysisDataContainer.h>
+#include "AliAnalysisManager.h"
+#include "AliESDtrack.h"
+#include "AliVertexerTracks.h"
+#include "AliAODHandler.h"
#include "AliAODEvent.h"
#include "AliAODVertex.h"
#include "AliAODTrack.h"
#include "AliAnalysisVertexingHF.h"
#include "AliAnalysisTaskSE.h"
#include "AliAnalysisTaskSED0Mass.h"
+#include "AliNormalizationCounter.h"
ClassImp(AliAnalysisTaskSED0Mass)
//________________________________________________________________________
AliAnalysisTaskSED0Mass::AliAnalysisTaskSED0Mass():
AliAnalysisTaskSE(),
-fOutputtight(0),
-fOutputloose(0),
-fDistr(0),
-fVHFtight(0),
-fVHFloose(0),
-fNentries(0)
+fOutputMass(0),
+fDistr(0),
+fNentries(0),
+fChecks(0),
+fCuts(0),
+fArray(0),
+fReadMC(0),
+fCutOnDistr(0),
+fUsePid4Distr(0),
+fCounter(0),
+fNPtBins(1),
+fTotPosPairs(0),
+fTotNegPairs(0),
+fLsNormalization(1.),
+fFillOnlyD0D0bar(0)
+
+
{
// Default constructor
}
//________________________________________________________________________
-AliAnalysisTaskSED0Mass::AliAnalysisTaskSED0Mass(const char *name):
+AliAnalysisTaskSED0Mass::AliAnalysisTaskSED0Mass(const char *name,AliRDHFCutsD0toKpi* cuts):
AliAnalysisTaskSE(name),
-fOutputtight(0),
-fOutputloose(0),
+fOutputMass(0),
fDistr(0),
-fVHFtight(0),
-fVHFloose(0),
-fNentries(0)
+fNentries(0),
+fChecks(0),
+fCuts(0),
+fArray(0),
+fReadMC(0),
+fCutOnDistr(0),
+fUsePid4Distr(0),
+fCounter(0),
+fNPtBins(1),
+fTotPosPairs(0),
+fTotNegPairs(0),
+fLsNormalization(1.),
+fFillOnlyD0D0bar(0)
+
{
// Default constructor
- // Output slot #1 writes into a TList container
+ fNPtBins=cuts->GetNPtBins();
+ fTotPosPairs=new Int_t[fNPtBins];
+ fTotNegPairs=new Int_t[fNPtBins];
+ for(Int_t i=0;i<fNPtBins;i++) {fTotPosPairs[i]=0; fTotNegPairs[i]=0;}
+
+ fCuts=cuts;
+
+ // Output slot #1 writes into a TList container (mass with cuts)
DefineOutput(1,TList::Class()); //My private output
- // Output slot #2 writes into a TList container
+ // Output slot #2 writes into a TList container (distributions)
DefineOutput(2,TList::Class()); //My private output
- // Output slot #3 writes into a TH1F container
+ // Output slot #3 writes into a TH1F container (number of events)
DefineOutput(3,TH1F::Class()); //My private output
- // Output slot #4 writes into a TList container
+ // Output slot #4 writes into a TList container (quality check)
DefineOutput(4,TList::Class()); //My private output
+ // Output slot #5 writes into a TList container (cuts)
+ DefineOutput(5,AliRDHFCutsD0toKpi::Class()); //My private output
+ // Output slot #6 writes Normalization Counter
+ DefineOutput(6,AliNormalizationCounter::Class());
}
//________________________________________________________________________
AliAnalysisTaskSED0Mass::~AliAnalysisTaskSED0Mass()
{
- if (fOutputtight) {
- delete fOutputtight;
- fOutputtight = 0;
- }
- if (fVHFtight) {
- delete fVHFtight;
- fVHFtight = 0;
- }
- if (fOutputloose) {
- delete fOutputloose;
- fOutputloose = 0;
+ if (fOutputMass) {
+ delete fOutputMass;
+ fOutputMass = 0;
}
-
if (fDistr) {
delete fDistr;
fDistr = 0;
}
-
- if (fVHFloose) {
- delete fVHFloose;
- fVHFloose = 0;
+ if (fChecks) {
+ delete fChecks;
+ fChecks = 0;
+ }
+ if (fCuts) {
+ delete fCuts;
+ fCuts = 0;
}
if (fNentries){
delete fNentries;
fNentries = 0;
}
-
-
+ if(fCounter){
+ delete fCounter;
+ fCounter=0;
+ }
+
}
//________________________________________________________________________
if(fDebug > 1) printf("AnalysisTaskSED0Mass::Init() \n");
- gROOT->LoadMacro("$ALICE_ROOT/PWG3/vertexingHF/ConfigVertexingHF.C");
-
- // 2 sets of dedidcated cuts -- defined in UserExec
- fVHFtight = (AliAnalysisVertexingHF*)gROOT->ProcessLine("ConfigVertexingHF()");
- fVHFloose = (AliAnalysisVertexingHF*)gROOT->ProcessLine("ConfigVertexingHF()");
+ AliRDHFCutsD0toKpi* copyfCuts=new AliRDHFCutsD0toKpi(*fCuts);
+ const char* nameoutput=GetOutputSlot(5)->GetContainer()->GetName();
+ copyfCuts->SetName(nameoutput);
+ // Post the data
+ PostData(5,copyfCuts);
+
+
return;
}
if(fDebug > 1) printf("AnalysisTaskSED0Mass::UserCreateOutputObjects() \n");
// Several histograms are more conveniently managed in a TList
- fOutputtight = new TList();
- fOutputtight->SetOwner();
- fOutputtight->SetName("listtight");
-
- fOutputloose = new TList();
- fOutputloose->SetOwner();
- fOutputloose->SetName("listloose");
+ fOutputMass = new TList();
+ fOutputMass->SetOwner();
+ fOutputMass->SetName("listMass");
fDistr = new TList();
fDistr->SetOwner();
fDistr->SetName("distributionslist");
- const Int_t nhist=4;
+ fChecks = new TList();
+ fChecks->SetOwner();
+ fChecks->SetName("checkHistograms");
- TString nameMass=" ", nameSgn=" ", nameBkg=" ", nameRfl=" ", namedistr=" ";
+ TString nameMass=" ",nameSgn27=" ",nameSgn=" ", nameBkg=" ", nameRfl=" ",nameMassNocutsS =" ",nameMassNocutsB =" ", namedistr=" ";
+
+ for(Int_t i=0;i<fCuts->GetNPtBins();i++){
- for(Int_t i=0;i<nhist;i++){
nameMass="histMass_";
- nameMass+=i+1;
+ nameMass+=i;
+ nameSgn27="histSgn27_";
+ nameSgn27+=i;
nameSgn="histSgn_";
- nameSgn+=i+1;
+ nameSgn+=i;
nameBkg="histBkg_";
- nameBkg+=i+1;
+ nameBkg+=i;
nameRfl="histRfl_";
- nameRfl+=i+1;
+ nameRfl+=i;
+ nameMassNocutsS="hMassS_";
+ nameMassNocutsS+=i;
+ nameMassNocutsB="hMassB_";
+ nameMassNocutsB+=i;
+
+ //histograms of cut variable distributions
+
+ // pT
+ namedistr="hptpiS_";
+ namedistr+=i;
+ TH1F *hptpiS = new TH1F(namedistr.Data(), "P_{T} distribution (pions);p_{T} [GeV/c]",200,0.,8.);
+
+ namedistr="hptKS_";
+ namedistr+=i;
+ TH1F *hptKS = new TH1F(namedistr.Data(), "P_{T} distribution (kaons);p_{T} [GeV/c]",200,0.,8.);
+
+ namedistr="hptB_";
+ namedistr+=i;
+ TH1F *hptB = new TH1F(namedistr.Data(), "P_{T} distribution;p_{T} [GeV/c]",200,0.,8.);
+
+ // pT no mass cut
+ namedistr="hptpiSnoMcut_";
+ namedistr+=i;
+ TH1F *hptpiSnoMcut = new TH1F(namedistr.Data(), "P_{T} distribution (pions);p_{T} [GeV/c]",200,0.,8.);
+
+ namedistr="hptKSnoMcut_";
+ namedistr+=i;
+ TH1F *hptKSnoMcut = new TH1F(namedistr.Data(), "P_{T} distribution (kaons);p_{T} [GeV/c]",200,0.,8.);
+
+ namedistr="hptB1prongnoMcut_";
+ namedistr+=i;
+ TH1F *hptB1pnoMcut = new TH1F(namedistr.Data(), "P_{T} distribution;p_{T} [GeV/c]",200,0.,8.);
+
+ namedistr="hptB2prongsnoMcut_";
+ namedistr+=i;
+ TH1F *hptB2pnoMcut = new TH1F(namedistr.Data(), "P_{T} distribution;p_{T} [GeV/c]",200,0.,8.);
+
+
+ // dca
+ namedistr="hdcaS_";
+ namedistr+=i;
+ TH1F *hdcaS = new TH1F(namedistr.Data(), "DCA distribution;dca [cm]",200,0.,0.1);
+ namedistr="hdcaB_";
+ namedistr+=i;
+ TH1F *hdcaB = new TH1F(namedistr.Data(), "DCA distribution;dca [cm]",200,0.,0.1);
+
+ // costhetastar
+ namedistr="hcosthetastarS_";
+ namedistr+=i;
+ TH1F *hcosthetastarS = new TH1F(namedistr.Data(), "cos#theta* distribution;cos#theta*",200,-1.,1.);
+ namedistr="hcosthetastarB_";
+ namedistr+=i;
+ TH1F *hcosthetastarB = new TH1F(namedistr.Data(), "cos#theta* distribution;cos#theta*",200,-1.,1.);
+
+ // impact parameter
+ namedistr="hd0piS_";
+ namedistr+=i;
+ TH1F *hd0piS = new TH1F(namedistr.Data(), "Impact parameter distribution (pions);d0(#pi) [cm]",200,-0.1,0.1);
+
+ namedistr="hd0KS_";
+ namedistr+=i;
+ TH1F *hd0KS = new TH1F(namedistr.Data(), "Impact parameter distribution (kaons);d0(K) [cm]",200,-0.1,0.1);
+
+ namedistr="hd0B_";
+ namedistr+=i;
+ TH1F *hd0B = new TH1F(namedistr.Data(), "Impact parameter distribution (both);d0 [cm]",200,-0.1,0.1);
+
+ namedistr="hd0p0B_";
+ namedistr+=i;
+ TH1F *hd0p0B = new TH1F(namedistr.Data(), "Impact parameter distribution (prong +);d0 [cm]",200,-0.1,0.1);
+
+ namedistr="hd0moresB_";
+ namedistr+=i;
+ TH1F *hd0moresB = new TH1F(namedistr.Data(), "Impact parameter distribution (both);d0 [cm]",200,-0.1,0.1);
+
+ namedistr="hd0d0moresB_";
+ namedistr+=i;
+ TH1F *hd0d0moresB = new TH1F(namedistr.Data(), "Impact parameter distribution (prong +);d0 [cm]",200,-0.001,0.001);
+
+ namedistr="hd0vmoresB_";
+ namedistr+=i;
+ TH1F *hd0vmoresB = new TH1F(namedistr.Data(), "Impact parameter distribution (both);d0 [cm]",200,-0.1,0.1);
+
+ namedistr="hd0d0vmoresB_";
+ namedistr+=i;
+ TH1F *hd0d0vmoresB = new TH1F(namedistr.Data(), "Impact parameter distribution (prong +);d0 [cm]",200,-0.001,0.001);
+
+ namedistr="hd0p1B_";
+ namedistr+=i;
+ TH1F *hd0p1B = new TH1F(namedistr.Data(), "Impact parameter distribution (prong -);d0 [cm]",200,-0.1,0.1);
+
+ namedistr="hd0vpiS_";
+ namedistr+=i;
+ TH1F *hd0vpiS = new TH1F(namedistr.Data(), "Impact parameter distribution (pions)(vtx w/o these tracks);d0(#pi) [cm]",200,-0.1,0.1);
+
+ namedistr="hd0vKS_";
+ namedistr+=i;
+ TH1F *hd0vKS = new TH1F(namedistr.Data(), "Impact parameter distribution (kaons) (vtx w/o these tracks);d0(K) [cm]",200,-0.1,0.1);
+ namedistr="hd0vB_";
+ namedistr+=i;
+ TH1F *hd0vB = new TH1F(namedistr.Data(), "Impact parameter distribution (vtx w/o these tracks);d0 [cm]",200,-0.1,0.1);
+
+ namedistr="hd0vp0B_";
+ namedistr+=i;
+ TH1F *hd0vp0B = new TH1F(namedistr.Data(), "Impact parameter distribution (prong + ** vtx w/o these tracks);d0 [cm]",200,-0.1,0.1);
+
+ namedistr="hd0vp1B_";
+ namedistr+=i;
+ TH1F *hd0vp1B = new TH1F(namedistr.Data(), "Impact parameter distribution (prong - ** vtx w/o these tracks);d0 [cm]",200,-0.1,0.1);
+
+ namedistr="hd0d0S_";
+ namedistr+=i;
+ TH1F *hd0d0S = new TH1F(namedistr.Data(), "d_{0}#timesd_{0} distribution;d_{0}#timesd_{0} [cm^{2}]",200,-0.001,0.001);
+ namedistr="hd0d0B_";
+ namedistr+=i;
+ TH1F *hd0d0B = new TH1F(namedistr.Data(), "d_{0}#timesd_{0} distribution;d_{0}#timesd_{0} [cm^{2}]",200,-0.001,0.001);
+
+ namedistr="hd0d0vS_";
+ namedistr+=i;
+ TH1F *hd0d0vS = new TH1F(namedistr.Data(), "d_{0}#timesd_{0} distribution (vtx w/o these tracks);d_{0}#timesd_{0} [cm^{2}]",200,-0.001,0.001);
+ namedistr="hd0d0vB_";
+ namedistr+=i;
+ TH1F *hd0d0vB = new TH1F(namedistr.Data(), "d_{0}#timesd_{0} distribution (vtx w/o these tracks);d_{0}#timesd_{0} [cm^{2}]",200,-0.001,0.001);
+
+ //decay lenght
+ namedistr="hdeclS_";
+ namedistr+=i;
+ TH1F *hdeclengthS = new TH1F(namedistr.Data(), "Decay Length distribution;Decay Length [cm]",200,0,0.15);
+
+ namedistr="hdeclB_";
+ namedistr+=i;
+ TH1F *hdeclengthB = new TH1F(namedistr.Data(), "Decay Length distribution;Decay Length [cm]",200,0,0.15);
+
+ namedistr="hnormdeclS_";
+ namedistr+=i;
+ TH1F *hnormdeclengthS = new TH1F(namedistr.Data(), "Normalized Decay Length distribution;Decay Length/Err ",200,0,6.);
+
+ namedistr="hnormdeclB_";
+ namedistr+=i;
+ TH1F *hnormdeclengthB = new TH1F(namedistr.Data(), "Normalized Decay Length distribution;Decay Length/Err ",200,0,6.);
+
+ namedistr="hdeclvS_";
+ namedistr+=i;
+ TH1F *hdeclengthvS = new TH1F(namedistr.Data(), "Decay Length distribution (vtx w/o tracks);Decay Length [cm]",200,0,0.6);
+
+ namedistr="hdeclvB_";
+ namedistr+=i;
+ TH1F *hdeclengthvB = new TH1F(namedistr.Data(), "Decay Length distribution (vtx w/o tracks);Decay Length [cm]",200,0,0.6);
+
+ namedistr="hnormdeclvS_";
+ namedistr+=i;
+ TH1F *hnormdeclengthvS = new TH1F(namedistr.Data(), "Normalized Decay Length distribution (vtx w/o tracks);Decay Length/Err ",200,0,10.);
+
+ namedistr="hnormdeclvB_";
+ namedistr+=i;
+ TH1F *hnormdeclengthvB = new TH1F(namedistr.Data(), "Normalized Decay Length distribution (vtx w/o tracks);Decay Length/Err ",200,0,10.);
+
+ // costhetapoint
+ namedistr="hcosthetapointS_";
+ namedistr+=i;
+ TH1F *hcosthetapointS = new TH1F(namedistr.Data(), "cos#theta_{Point} distribution;cos#theta_{Point}",200,0,1.);
+ namedistr="hcosthetapointB_";
+ namedistr+=i;
+ TH1F *hcosthetapointB = new TH1F(namedistr.Data(), "cos#theta_{Point} distribution;cos#theta_{Point}",200,0,1.);
+
+ namedistr="hcosthetapointmoresB_";
+ namedistr+=i;
+ TH1F *hcosthetapointmoresB = new TH1F(namedistr.Data(), "cos#theta_{Point} distribution;cos#theta_{Point}",200,0,1.);
+
+ // costhetapoint vs d0 or d0d0
+ namedistr="hcosthpointd0S_";
+ namedistr+=i;
+ TH2F *hcosthpointd0S= new TH2F(namedistr.Data(),"Correlation cos#theta_{Point}-d_{0};cos#theta_{Point};d_{0} [cm^{2}]",200,0,1.,200,-0.001,0.001);
+
+ namedistr="hcosthpointd0B_";
+ namedistr+=i;
+ TH2F *hcosthpointd0B= new TH2F(namedistr.Data(),"Correlation cos#theta_{Point}-d_{0};cos#theta_{Point};d_{0} [cm^{2}]",200,0,1.,200,-0.001,0.001);
+
+ namedistr="hcosthpointd0d0S_";
+ namedistr+=i;
+ TH2F *hcosthpointd0d0S= new TH2F(namedistr.Data(),"Correlation cos#theta_{Point}-d_{0}#timesd_{0};cos#theta_{Point};d_{0}#timesd_{0} [cm^{2}]",200,0,1.,200,-0.001,0.001);
+ namedistr="hcosthpointd0d0B_";
+ namedistr+=i;
+ TH2F *hcosthpointd0d0B= new TH2F(namedistr.Data(),"Correlation cos#theta_{Point}-d_{0}#timesd_{0};cos#theta_{Point};d_{0}#timesd_{0} [cm^{2}]",200,0,1.,200,-0.001,0.001);
+
+ fDistr->Add(hptpiS);
+ fDistr->Add(hptKS);
+ fDistr->Add(hptB);
+
+ fDistr->Add(hptpiSnoMcut);
+ fDistr->Add(hptKSnoMcut);
+ fDistr->Add(hptB1pnoMcut);
+ fDistr->Add(hptB2pnoMcut);
+
+ fDistr->Add(hdcaS);
+ fDistr->Add(hdcaB);
+
+ fDistr->Add(hd0piS);
+ fDistr->Add(hd0KS);
+ fDistr->Add(hd0B);
+ fDistr->Add(hd0p0B);
+ fDistr->Add(hd0moresB);
+ fDistr->Add(hd0p1B);
+
+ fDistr->Add(hd0vpiS);
+ fDistr->Add(hd0vKS);
+ fDistr->Add(hd0vB);
+ fDistr->Add(hd0vp0B);
+ fDistr->Add(hd0vp1B);
+ fDistr->Add(hd0vmoresB);
+
+ fDistr->Add(hd0d0S);
+ fDistr->Add(hd0d0B);
+ fDistr->Add(hd0d0moresB);
+
+ fDistr->Add(hd0d0vS);
+ fDistr->Add(hd0d0vB);
+ fDistr->Add(hd0d0vmoresB);
+
+ fDistr->Add(hcosthetastarS);
+ fDistr->Add(hcosthetastarB);
+
+ fDistr->Add(hcosthetapointS);
+ fDistr->Add(hcosthetapointB);
+ fDistr->Add(hcosthetapointmoresB);
+
+ fDistr->Add(hdeclengthS);
+ fDistr->Add(hdeclengthB);
+
+ fDistr->Add(hnormdeclengthS);
+ fDistr->Add(hnormdeclengthB);
+
+ fDistr->Add(hdeclengthvS);
+ fDistr->Add(hdeclengthvB);
+
+ fDistr->Add(hnormdeclengthvS);
+ fDistr->Add(hnormdeclengthvB);
+
+ fDistr->Add(hcosthpointd0S);
+ fDistr->Add(hcosthpointd0B);
+
+ fDistr->Add(hcosthpointd0d0S);
+ fDistr->Add(hcosthpointd0d0B);
+
//histograms of invariant mass distributions
tmpMt->Sumw2();
tmpMl->Sumw2();
+ //to compare with AliAnalysisTaskCharmFraction
+ TH1F* tmpS27t = new TH1F(nameSgn27.Data(),"D^{0} invariant mass in M(D^{0}) +/- 27 MeV - MC; M [GeV]; Entries",200,1.765,1.965);
+ TH1F *tmpS27l=(TH1F*)tmpS27t->Clone();
+ tmpS27t->Sumw2();
+ tmpS27l->Sumw2();
+
+ //distribution w/o cuts
+ // TH1F* tmpMS = new TH1F(nameMassNocutsS.Data(),"D^{0} invariant mass; M [GeV]; Entries",300,0.7,3.);
+ TH1F* tmpMS = new TH1F(nameMassNocutsS.Data(),"D^{0} invariant mass; M [GeV]; Entries",300,1.56484,2.16484); //range (MD0-300MeV, mD0 + 300MeV)
+ TH1F *tmpMB=(TH1F*)tmpMS->Clone();
+ tmpMB->SetName(nameMassNocutsB.Data());
+ tmpMS->Sumw2();
+ tmpMB->Sumw2();
+
+ //MC signal and background
TH1F* tmpSt = new TH1F(nameSgn.Data(), "D^{0} invariant mass - MC; M [GeV]; Entries",200,1.765,1.965);
TH1F *tmpSl=(TH1F*)tmpSt->Clone();
tmpSt->Sumw2();
tmpBt->Sumw2();
tmpBl->Sumw2();
- //Reflection: histo filled with D0 which pass the cut (also) as D0bar and with D0bar which pass (also) the cut as D0
+ //Reflection: histo filled with D0Mass which pass the cut (also) as D0bar and with D0bar which pass (also) the cut as D0
TH1F* tmpRt = new TH1F(nameRfl.Data(), "Reflected signal invariant mass - MC; M [GeV]; Entries",200,1.765,1.965);
TH1F *tmpRl=(TH1F*)tmpRt->Clone();
tmpRt->Sumw2();
tmpRl->Sumw2();
// printf("Created histograms %s\t%s\t%s\t%s\n",tmpM->GetName(),tmpS->GetName(),tmpB->GetName(),tmpR->GetName());
- fOutputtight->Add(tmpMt);
- fOutputtight->Add(tmpSt);
- fOutputtight->Add(tmpBt);
- fOutputtight->Add(tmpRt);
+ fOutputMass->Add(tmpMt);
+ fOutputMass->Add(tmpSt);
+ fOutputMass->Add(tmpS27t);
+ fOutputMass->Add(tmpBt);
+ fOutputMass->Add(tmpRt);
+
+ fDistr->Add(tmpMS);
+ fDistr->Add(tmpMB);
+
- fOutputloose->Add(tmpMl);
- fOutputloose->Add(tmpSl);
- fOutputloose->Add(tmpBl);
- fOutputloose->Add(tmpRl);
-
}
- //histograms of cut variable distributions
- // pT
- namedistr="hptpiS";
- TH1F *hptpiS = new TH1F(namedistr.Data(), "P_{T} distribution (pions);p_{T} [GeV/c]",200,0.,8.);
-// namedistr="hptpiR";
-// TH1F *hptpiR = new TH1F(namedistr.Data(), "P_{T} distribution (pions);p_{T} [GeV/c]",200,0.,8.);
-
- namedistr="hptKS";
- TH1F *hptKS = new TH1F(namedistr.Data(), "P_{T} distribution (kaons);p_{T} [GeV/c]",200,0.,8.);
-
- namedistr="hptB";
- TH1F *hptB = new TH1F(namedistr.Data(), "P_{T} distribution;p_{T} [GeV/c]",200,0.,8.);
-// namedistr="hptKR";
-// TH1F *hptKR = new TH1F(namedistr.Data(), "P_{T} distribution (kaons);p_{T} [GeV/c]",200,0.,8.);
-
- // dca
- namedistr="hdcaS";
- TH1F *hdcaS = new TH1F(namedistr.Data(), "DCA distribution;dca [cm]",200,0.,0.1);
- namedistr="hdcaB";
- TH1F *hdcaB = new TH1F(namedistr.Data(), "DCA distribution;dca [cm]",200,0.,0.1);
-// namedistr="hdcaR";
-// TH1F *hdcaR = new TH1F(namedistr.Data(), "DCA distribution;dca [cm]",200,0.,0.1);
-
- // costhetastar
- namedistr="costhetastarS";
- TH1F *hcosthetastarS = new TH1F(namedistr.Data(), "cos#theta* distribution;cos#theta*",200,-1.,1.);
- namedistr="costhetastarB";
- TH1F *hcosthetastarB = new TH1F(namedistr.Data(), "cos#theta* distribution;cos#theta*",200,-1.,1.);
-// namedistr="costhetastarR";
-// TH1F *hcosthetastarR = new TH1F(namedistr.Data(), "cos#theta* distribution;cos#theta*",200,-1.,1.);
-
- // impact parameter
- namedistr="hd0piS";
- TH1F *hd0piS = new TH1F(namedistr.Data(), "Impact parameter distribution (pions);d0(#pi) [cm]",200,-0.1,0.1);
-// namedistr="hd0piR";
-// TH1F *hd0piR = new TH1F(namedistr.Data(), "Impact parameter distribution (pions);d0(#pi) [cm]",200,0.,1.);
-
- namedistr="hd0KS";
- TH1F *hd0KS = new TH1F(namedistr.Data(), "Impact parameter distribution (kaons);d0(K) [cm]",200,-0.1,0.1);
- namedistr="hd0B";
- TH1F *hd0B = new TH1F(namedistr.Data(), "Impact parameter distribution;d0 [cm]",200,-0.1,0.1);
-// namedistr="hd0KR";
-// TH1F *hd0KR = new TH1F(namedistr.Data(), "Impact parameter distribution (kaons);d0(K) [cm]",200,0.,0.1);
-
- namedistr="hd0d0S";
- TH1F *hd0d0S = new TH1F(namedistr.Data(), "d_{0}#timesd_{0} distribution;d_{0}#timesd_{0} [cm^{2}]",200,-0.001,0.001);
- namedistr="hd0d0B";
- TH1F *hd0d0B = new TH1F(namedistr.Data(), "d_{0}#timesd_{0} distribution;d_{0}#timesd_{0} [cm^{2}]",200,-0.001,0.001);
-// namedistr="hd0d0R";
-// TH1F *hd0d0R = new TH1F(namedistr.Data(), "d_{0}#timesd_{0} distribution (pions);d_{0}#timesd_{0} [cm^{2}]",200,-0.001,0.001);
-
- // costhetapoint
- namedistr="hcosthetapointS";
- TH1F *hcosthetapointS = new TH1F(namedistr.Data(), "cos#theta_{Point} distribution;cos#theta_{Point}",200,0,1.);
- namedistr="hcosthetapointB";
- TH1F *hcosthetapointB = new TH1F(namedistr.Data(), "cos#theta_{Point} distribution;cos#theta_{Point}",200,0,1.);
-// namedistr="costhetapointR";
-// TH1F *hcosthetapointR = new TH1F(namedistr.Data(), "cos#theta_{Point} distribution;cos#theta_{Point}",200,-1,1.);
-
- namedistr="hcosthpointd0d0S";
- TH2F *hcosthpointd0d0S= new TH2F(namedistr.Data(),"Correlation cos#theta_{Point}-d_{0}#timesd_{0};cos#theta_{Point};d_{0}#timesd_{0} [cm^{2}]",200,0,1.,200,-0.001,0.001);
- namedistr="hcosthpointd0d0B";
- TH2F *hcosthpointd0d0B= new TH2F(namedistr.Data(),"Correlation cos#theta_{Point}-d_{0}#timesd_{0};cos#theta_{Point};d_{0}#timesd_{0} [cm^{2}]",200,0,1.,200,-0.001,0.001);
-
- fDistr->Add(hptpiS);
- //fDistr->Add(hptpiR);
- fDistr->Add(hptKS);
- fDistr->Add(hptB);
- //fDistr->Add(hptKR);
-
- fDistr->Add(hdcaS);
- fDistr->Add(hdcaB);
- //fDistr->Add(hdcaR);
-
- fDistr->Add(hd0piS);
- //fDistr->Add(hd0piR);
- fDistr->Add(hd0KS);
- fDistr->Add(hd0B);
- //fDistr->Add(hd0KR);
-
- fDistr->Add(hd0d0S);
- fDistr->Add(hd0d0B);
- //fDistr->Add(hd0d0R);
-
- fDistr->Add(hcosthetastarS);
- fDistr->Add(hcosthetastarB);
- //fDistr->Add(hcosthetastarR);
-
- fDistr->Add(hcosthetapointS);
- fDistr->Add(hcosthetapointB);
- //fDistr->Add(hcosthetapointR);
-
- fDistr->Add(hcosthpointd0d0S);
- fDistr->Add(hcosthpointd0d0B);
-
- fNentries=new TH1F("nentriesD0", "Look at the number of entries! it is = to the number of AODs", 2,1.,2.);
+ namedistr="hpospair";
+ TH1F* hpospair=new TH1F(namedistr.Data(),"Number of positive pairs",fCuts->GetNPtBins(),-0.5,fCuts->GetNPtBins()-0.5);
+ namedistr="hnegpair";
+ TH1F* hnegpair=new TH1F(namedistr.Data(),"Number of negative pairs",fCuts->GetNPtBins(),-0.5,fCuts->GetNPtBins()-0.5);
+ fDistr->Add(hpospair);
+ fDistr->Add(hnegpair);
+
+
+ //histograms for vertex checking and TOF checking
+ TString checkname="hptGoodTr";
+
+ TH1F* hptGoodTr=new TH1F(checkname.Data(),"Pt distribution of 'good' tracks;p_{t}[GeV];Number",200,0.,8.);
+ hptGoodTr->SetTitleOffset(1.3,"Y");
+ checkname="hdistrGoodTr";
+
+ TH1F* hdistrGoodTr=new TH1F(checkname.Data(),"Distribution of number of good tracks per event;no.good-tracks/ev;Entries",31,0,31);
+ hdistrGoodTr->SetTitleOffset(1.3,"Y");
+ checkname="hTOFsig";
+ TH1F* hTOFsig=new TH1F(checkname.Data(),"Distribution of TOF signal;TOF time [ps];Entries", 100, -2.e3,40.e3);
+
+ checkname="hTPCsig";
+ TH1F* hTPCsig=new TH1F(checkname.Data(),"Distribution of TPC signal;TPC sig;Entries", 100, 35.,100.);
+
+ checkname="hTOFtimeKaonHyptime";
+ TH2F* hTOFtimeKaonHyptime=new TH2F(checkname.Data(),"TOFtime - timeHypothesisForKaon;p[GeV/c];TOFtime - timeHypothesisForKaon [ps]",200,0.,4.,1000,-20000.,20000.);
+
+ checkname="hTOFtimeKaonHyptimeAC";
+ TH2F* hTOFtimeKaonHyptimeAC=new TH2F(checkname.Data(),"TOFtime - timeHypothesisForKaon;p[GeV/c];TOFtime - timeHypothesisForKaon [ps]",200,0.,4.,1000,-20000.,20000.);
+
+ checkname="hTPCsigvsp";
+ TH2F* hTPCsigvsp=new TH2F(checkname.Data(),"TPCsig vs p;p[GeV/c];TPCsig",200,0.,4.,1000,35.,100.);
+
+ checkname="hTOFtime";
+ TH1F* hTOFtime=new TH1F(checkname.Data(),"Distribution of TOF time Kaon;TOF time(Kaon) [ps];Entries", 1000, 0.,50000.);
+
+
+ fChecks->Add(hptGoodTr);
+ fChecks->Add(hdistrGoodTr);
+ fChecks->Add(hTOFsig);
+ fChecks->Add(hTPCsig);
+ fChecks->Add(hTOFtimeKaonHyptime);
+ fChecks->Add(hTOFtimeKaonHyptimeAC);
+ fChecks->Add(hTPCsigvsp);
+ fChecks->Add(hTOFtime);
+
+ const char* nameoutput=GetOutputSlot(3)->GetContainer()->GetName();
+
+ fNentries=new TH1F(nameoutput, "Integral(1,2) = number of AODs *** Integral(2,3) = number of candidates selected with cuts *** Integral(3,4) = number of D0 selected with cuts *** Integral(4,5) = events with good vertex *** Integral(5,6) = pt out of bounds", 15,-0.5,14.5);
+
+ fNentries->GetXaxis()->SetBinLabel(1,"nEventsAnal");
+ fNentries->GetXaxis()->SetBinLabel(2,"nCandSel(Cuts)");
+ fNentries->GetXaxis()->SetBinLabel(3,"nD0Selected");
+ fNentries->GetXaxis()->SetBinLabel(4,"nEventsGoodVtxS");
+ fNentries->GetXaxis()->SetBinLabel(5,"nEventsGoodVtx+>2tracks");
+ fNentries->GetXaxis()->SetBinLabel(6,"ptbin = -1");
+ fNentries->GetXaxis()->SetBinLabel(7,"no daughter");
+ fNentries->GetXaxis()->SetBinLabel(8,"nCandSel(Tr)");
+ fNentries->GetXaxis()->SetBinLabel(9,"PID=0");
+ fNentries->GetXaxis()->SetBinLabel(10,"PID=1");
+ fNentries->GetXaxis()->SetBinLabel(11,"PID=2");
+ fNentries->GetXaxis()->SetBinLabel(12,"PID=3");
+ fNentries->GetXaxis()->SetBinLabel(13,"K");
+ fNentries->GetXaxis()->SetBinLabel(14,"Lambda");
+ fNentries->GetXaxis()->SetBinLabel(15,"Pile-up Rej");
+ fNentries->GetXaxis()->SetNdivisions(1,kFALSE);
+
+ fCounter = new AliNormalizationCounter(Form("%s",GetOutputSlot(6)->GetContainer()->GetName()));
+
+ // Post the data
+ PostData(1,fOutputMass);
+ PostData(2,fDistr);
+ PostData(3,fNentries);
+ PostData(4,fChecks);
+ PostData(6,fCounter);
return;
}
// Execute analysis for current event:
// heavy flavor candidates association to MC truth
//cout<<"I'm in UserExec"<<endl;
+
+
+ //cuts order
+ // printf(" |M-MD0| [GeV] < %f\n",fD0toKpiCuts[0]);
+ // printf(" dca [cm] < %f\n",fD0toKpiCuts[1]);
+ // printf(" cosThetaStar < %f\n",fD0toKpiCuts[2]);
+ // printf(" pTK [GeV/c] > %f\n",fD0toKpiCuts[3]);
+ // printf(" pTpi [GeV/c] > %f\n",fD0toKpiCuts[4]);
+ // printf(" |d0K| [cm] < %f\n",fD0toKpiCuts[5]);
+ // printf(" |d0pi| [cm] < %f\n",fD0toKpiCuts[6]);
+ // printf(" d0d0 [cm^2] < %f\n",fD0toKpiCuts[7]);
+ // printf(" cosThetaPoint > %f\n",fD0toKpiCuts[8]);
+
+
AliAODEvent *aod = dynamic_cast<AliAODEvent*> (InputEvent());
+
+ TString bname;
+ if(fArray==0){ //D0 candidates
+ // load D0->Kpi candidates
+ //cout<<"D0 candidates"<<endl;
+ bname="D0toKpi";
+ } else { //LikeSign candidates
+ // load like sign candidates
+ //cout<<"LS candidates"<<endl;
+ bname="LikeSign2Prong";
+ }
+
+ TClonesArray *inputArray=0;
- // load D0->Kpi candidates
- TClonesArray *inputArrayD0toKpi =
- (TClonesArray*)aod->GetList()->FindObject("D0toKpi");
- if(!inputArrayD0toKpi) {
- printf("AliAnalysisTaskSECompareHFpt::UserExec: D0toKpi branch not found!\n");
+ if(!aod && AODEvent() && IsStandardAOD()) {
+ // In case there is an AOD handler writing a standard AOD, use the AOD
+ // event in memory rather than the input (ESD) event.
+ aod = dynamic_cast<AliAODEvent*> (AODEvent());
+ // in this case the braches in the deltaAOD (AliAOD.VertexingHF.root)
+ // have to taken from the AOD event hold by the AliAODExtension
+ AliAODHandler* aodHandler = (AliAODHandler*)
+ ((AliAnalysisManager::GetAnalysisManager())->GetOutputEventHandler());
+
+ if(aodHandler->GetExtensions()) {
+ AliAODExtension *ext = (AliAODExtension*)aodHandler->GetExtensions()->FindObject("AliAOD.VertexingHF.root");
+ AliAODEvent* aodFromExt = ext->GetAOD();
+ inputArray=(TClonesArray*)aodFromExt->GetList()->FindObject(bname.Data());
+ }
+ } else {
+ inputArray=(TClonesArray*)aod->GetList()->FindObject(bname.Data());
+ }
+
+
+ if(!inputArray) {
+ printf("AliAnalysisTaskSED0Mass::UserExec: input branch not found!\n");
return;
}
- // AOD primary vertex
- AliAODVertex *vtx1 = (AliAODVertex*)aod->GetPrimaryVertex();
- //vtx1->Print();
-
- // load MC particles
- TClonesArray *mcArray =
- (TClonesArray*)aod->GetList()->FindObject(AliAODMCParticle::StdBranchName());
- if(!mcArray) {
- printf("AliAnalysisTaskSED0Mass::UserExec: MC particles branch not found!\n");
- return;
+ // fix for temporary bug in ESDfilter
+ // the AODs with null vertex pointer didn't pass the PhysSel
+ if(!aod->GetPrimaryVertex() || TMath::Abs(aod->GetMagneticField())<0.001) return;
+
+
+ TClonesArray *mcArray = 0;
+ AliAODMCHeader *mcHeader = 0;
+
+ if(fReadMC) {
+ // load MC particles
+ mcArray = (TClonesArray*)aod->GetList()->FindObject(AliAODMCParticle::StdBranchName());
+ if(!mcArray) {
+ printf("AliAnalysisTaskSED0Mass::UserExec: MC particles branch not found!\n");
+ return;
+ }
+
+ // load MC header
+ mcHeader = (AliAODMCHeader*)aod->GetList()->FindObject(AliAODMCHeader::StdBranchName());
+ if(!mcHeader) {
+ printf("AliAnalysisTaskSED0Mass::UserExec: MC header branch not found!\n");
+ return;
+ }
}
- // load MC header
- AliAODMCHeader *mcHeader =
- (AliAODMCHeader*)aod->GetList()->FindObject(AliAODMCHeader::StdBranchName());
- if(!mcHeader) {
- printf("AliAnalysisTaskSED0Mass::UserExec: MC header branch not found!\n");
+ //printf("VERTEX Z %f %f\n",vtx1->GetZ(),mcHeader->GetVtxZ());
+
+ //histogram filled with 1 for every AOD
+ fNentries->Fill(0);
+ fCounter->StoreEvent(aod,fReadMC);
+ if(!fCuts->IsEventSelected(aod)) {
+ if(fCuts->GetWhyRejection()==1) // rejected for pileup
+ fNentries->Fill(14);
return;
}
+ // AOD primary vertex
+ AliAODVertex *vtx1 = (AliAODVertex*)aod->GetPrimaryVertex();
+
+ Bool_t isGoodVtx=kFALSE;
+
+ //vtx1->Print();
+ TString primTitle = vtx1->GetTitle();
+ if(primTitle.Contains("VertexerTracks") && vtx1->GetNContributors()>0) {
+ isGoodVtx=kTRUE;
+ fNentries->Fill(3);
+ }
- //histogram filled with 1 for every AOD
- fNentries->Fill(1);
- PostData(3,fNentries);
- // loop over D0->Kpi candidates
- Int_t nInD0toKpi = inputArrayD0toKpi->GetEntriesFast();
- printf("Number of D0->Kpi: %d\n",nInD0toKpi);
+ //cout<<"Start checks"<<endl;
+ Int_t ntracks=0,isGoodTrack=0;
+
+ if(aod) ntracks=aod->GetNTracks();
+
+ //cout<<"ntracks = "<<ntracks<<endl;
+
+ //loop on tracks in the event
+ for (Int_t k=0;k<ntracks;k++){
+ AliAODTrack* track=aod->GetTrack(k);
+ //cout<<"track n = "<<k<<endl;
+
+ //check TOF
+ if(!(track->GetStatus()&AliESDtrack::kTPCrefit &&
+ track->GetStatus()&AliESDtrack::kITSrefit &&
+ track->GetTPCNcls() >=70 &&
+ track->GetStatus()&AliESDtrack::kTOFpid &&
+ track->GetStatus()&AliESDtrack::kTOFout &&
+ track->GetStatus()&AliESDtrack::kTIME)) continue;
+ AliAODPid *pid = track->GetDetPid();
+ if(!pid) {if (fDebug>1)cout<<"No AliAODPid found"<<endl; continue;}
+
+ Double_t times[5];
+ pid->GetIntegratedTimes(times);
+
+ ((TH1F*)fChecks->FindObject("hTOFtime"))->Fill(times[3]);
+ ((TH1F*)fChecks->FindObject("hTOFtimeKaonHyptime"))->Fill(track->P(),pid->GetTOFsignal()-times[3]); //3 is kaon
+
+ ((TH1F*)fChecks->FindObject("hTOFsig"))->Fill(pid->GetTOFsignal());
+ ((TH1F*)fChecks->FindObject("hTPCsig"))->Fill(pid->GetTPCsignal());
+ ((TH1F*)fChecks->FindObject("hTPCsigvsp"))->Fill(track->P(),pid->GetTPCsignal());
+ if (pid->GetTOFsignal()< 0) ((TH1F*)fChecks->FindObject("hTOFsig"))->Fill(-1);
+
+ //check clusters of the tracks
+ Int_t nclsTot=0,nclsSPD=0;
+
+ for(Int_t l=0;l<6;l++) {
+ if(TESTBIT(track->GetITSClusterMap(),l)) {
+ nclsTot++; if(l<2) nclsSPD++;
+ }
+ }
+
+ if (track->Pt()>0.3 &&
+ track->GetStatus()&AliESDtrack::kTPCrefit &&
+ track->GetStatus()&AliESDtrack::kITSrefit &&
+ nclsTot>3 &&
+ nclsSPD>0) {//fill hist good tracks
+
+ ((TH1F*)fChecks->FindObject("hptGoodTr"))->Fill(track->Pt());
+ isGoodTrack++;
+ }
+ //cout<<"isGoodTrack = "<<isGoodTrack<<endl;
+ ((TH1F*)fChecks->FindObject("hdistrGoodTr"))->Fill(isGoodTrack);
+ }
+ //number of events with good vertex and at least 2 good tracks
+ if (isGoodTrack>=2 && isGoodVtx) fNentries->Fill(4);
+
+ // loop over candidates
+ Int_t nInD0toKpi = inputArray->GetEntriesFast();
+ if(fDebug>2) printf("Number of D0->Kpi: %d\n",nInD0toKpi);
+
+ // FILE *f=fopen("4display.txt","a");
+ // fprintf(f,"Number of D0->Kpi: %d\n",nInD0toKpi);
+ Int_t nSelectedloose=0,nSelectedtight=0;
for (Int_t iD0toKpi = 0; iD0toKpi < nInD0toKpi; iD0toKpi++) {
+ //Int_t nPosPairs=0, nNegPairs=0;
//cout<<"inside the loop"<<endl;
- AliAODRecoDecayHF2Prong *d = (AliAODRecoDecayHF2Prong*)inputArrayD0toKpi->UncheckedAt(iD0toKpi);
- Bool_t unsetvtx=kFALSE;
- if(!d->GetOwnPrimaryVtx()) {
- d->SetOwnPrimaryVtx(vtx1); // needed to compute all variables
- unsetvtx=kTRUE;
+ AliAODRecoDecayHF2Prong *d = (AliAODRecoDecayHF2Prong*)inputArray->UncheckedAt(iD0toKpi);
+
+ //check daughters
+ if(!(d->GetDaughter(0) || d->GetDaughter(1))) {
+ AliDebug(1,"at least one daughter not found!");
+ fNentries->Fill(6);
+ continue;
}
+
+ // Bool_t unsetvtx=kFALSE;
+ // if(!d->GetOwnPrimaryVtx()) {
+ // d->SetOwnPrimaryVtx(vtx1); // needed to compute all variables
+ // unsetvtx=kTRUE;
+ // }
+
- //cuts order
-// printf(" |M-MD0| [GeV] < %f\n",fD0toKpiCuts[0]);
-// printf(" dca [cm] < %f\n",fD0toKpiCuts[1]);
-// printf(" cosThetaStar < %f\n",fD0toKpiCuts[2]);
-// printf(" pTK [GeV/c] > %f\n",fD0toKpiCuts[3]);
-// printf(" pTpi [GeV/c] > %f\n",fD0toKpiCuts[4]);
-// printf(" |d0K| [cm] < %f\n",fD0toKpiCuts[5]);
-// printf(" |d0pi| [cm] < %f\n",fD0toKpiCuts[6]);
-// printf(" d0d0 [cm^2] < %f\n",fD0toKpiCuts[7]);
-// printf(" cosThetaPoint > %f\n",fD0toKpiCuts[8]);
-
- //add distr here
- UInt_t pdgs[2];
+ //check reco daughter in acceptance
+ /*
+ Double_t eta0=d->EtaProng(0);
+ Double_t eta1=d->EtaProng(1);
+ */
+ if ( fCuts->IsInFiducialAcceptance(d->Pt(),d->Y(421)) ) {
+ //if( TMath::Abs(eta0)<0.9 && TMath::Abs(eta1)<0.9 ){
+ //apply cuts on tracks
+ nSelectedloose++;
+ Int_t isSelected = fCuts->IsSelected(d,AliRDHFCuts::kTracks);
+ if(((AliAODTrack*)d->GetDaughter(0))->GetTPCNcls() < 70 || ((AliAODTrack*)d->GetDaughter(1))->GetTPCNcls() < 70) isSelected=kFALSE;
+ if (!isSelected) continue;
+ nSelectedtight++;
+ fNentries->Fill(7);
+ if(fDebug>2) cout<<"tracks selected"<<endl;
+
+ Int_t ptbin=fCuts->PtBin(d->Pt());
+ if(ptbin==-1) {fNentries->Fill(5); continue;} //out of bounds
+ FillVarHists(aod,d,mcArray,fCuts,fDistr);
+ FillMassHists(aod,d,mcArray,fCuts,fOutputMass);
+ }
+
+ //if(unsetvtx) d->UnsetOwnPrimaryVtx();
+ } //end for prongs
+ fCounter->StoreCandidates(aod,nSelectedloose,kTRUE);
+ fCounter->StoreCandidates(aod,nSelectedtight,kFALSE);
+ // Post the data
+ PostData(1,fOutputMass);
+ PostData(2,fDistr);
+ PostData(3,fNentries);
+ PostData(4,fChecks);
+ PostData(6,fCounter);
+ return;
+}
+
+//____________________________________________________________________________
+void AliAnalysisTaskSED0Mass::FillVarHists(AliAODEvent* aod,AliAODRecoDecayHF2Prong *part, TClonesArray *arrMC, AliRDHFCutsD0toKpi *cuts, TList *listout){
+ //
+ // function used in UserExec to fill variable histograms:
+ //
+
+ //add distr here
+ UInt_t pdgs[2];
- Double_t mPDG=TDatabasePDG::Instance()->GetParticle(421)->Mass();
- pdgs[0]=211;
- pdgs[1]=321;
- Double_t minvD0 = d->InvMassD0();
- pdgs[1]=211;
- pdgs[0]=321;
- Double_t minvD0bar = d->InvMassD0bar();
- //apply cut on invariant mass on the pair
- if(TMath::Abs(minvD0-mPDG)<0.03 || TMath::Abs(minvD0bar-mPDG)<0.03){
- Int_t pdgDgD0toKpi[2]={321,211};
- Int_t lab=d->MatchToMC(421,mcArray,2,pdgDgD0toKpi); //return MC particle label if the array corresponds to a D0, -1 if not (cf. AliAODRecoDecay.cxx)
-
- // Int_t lab=d->MatchToMC(421,mcArray); //|pdg| requested
-// AliAODMCParticle *mcmother = (AliAODMCParticle*)mcArray->At(lab);
-// cout<<"mcmother name = "<<mcmother->GetName()<<" label = "<<mcmother->GetLabel()<<endl;
- if(lab>=0){ //signal
- //cout<<"is signal"<<endl;
- for (Int_t iprong=0; iprong<2; iprong++){
- AliAODTrack *prong=(AliAODTrack*)d->GetDaughter(iprong);
- Int_t labprong=prong->GetLabel();
+ Double_t mPDG=TDatabasePDG::Instance()->GetParticle(421)->Mass();
+ pdgs[0]=211;
+ pdgs[1]=321;
+ Double_t minvD0 = part->InvMassD0();
+ pdgs[1]=211;
+ pdgs[0]=321;
+ Double_t minvD0bar = part->InvMassD0bar();
+ //cout<<"inside mass cut"<<endl;
+
+ Int_t pdgDgD0toKpi[2]={321,211};
+ Int_t lab=-9999;
+ if(fReadMC) lab=part->MatchToMC(421,arrMC,2,pdgDgD0toKpi); //return MC particle label if the array corresponds to a D0, -1 if not (cf. AliAODRecoDecay.cxx)
+ //Double_t pt = d->Pt(); //mother pt
+ Int_t isSelected=3;
+ Int_t isSelectedPID=99;
+
+ //isSelectedPID = cuts->IsSelected(part,AliRDHFCuts::kPID); //0 rejected,1 D0,2 Dobar, 3 both
+ isSelectedPID = cuts->IsSelectedPID(part); //0 rejected,1 D0,2 Dobar, 3 both
+ if (isSelectedPID==0)fNentries->Fill(8);
+ if (isSelectedPID==1)fNentries->Fill(9);
+ if (isSelectedPID==2)fNentries->Fill(10);
+ if (isSelectedPID==3)fNentries->Fill(11);
+ //fNentries->Fill(8+isSelectedPID);
+
+ if(fCutOnDistr){
+ isSelected = cuts->IsSelected(part,AliRDHFCuts::kCandidate,aod); //cuts with variables recalculated with new vertex (w/o daughters)
+ if (!isSelected){
+ //cout<<"Not Selected"<<endl;
+ return;
+ }
+ }
+
+ Double_t invmasscut=0.03;
+
+ TString fillthispi="",fillthisK="",fillthis="";
+
+ Int_t ptbin=cuts->PtBin(part->Pt());
+
+
+ //recalculate vertex
+ AliAODVertex *vtxProp=0x0;
+ vtxProp=GetPrimaryVtxSkipped(aod,part);
+ Double_t dz1[2],dz2[2],covar1[3],covar2[3];//,d0xd0proper,errd0xd0proper;
+ dz1[0]=-99; dz2[0]=-99;
+ if(vtxProp) {
+ part->SetOwnPrimaryVtx(vtxProp);
+ //Bool_t unsetvtx=kTRUE;
+ //Calculate d0 for daughter tracks with Vtx Skipped
+ AliESDtrack *esdtrack1=new AliESDtrack((AliVTrack*)part->GetDaughter(0));
+ AliESDtrack *esdtrack2=new AliESDtrack((AliVTrack*)part->GetDaughter(1));
+ esdtrack1->PropagateToDCA(vtxProp,aod->GetMagneticField(),1.,dz1,covar1);
+ esdtrack2->PropagateToDCA(vtxProp,aod->GetMagneticField(),1.,dz2,covar2);
+ delete vtxProp; vtxProp=NULL;
+ delete esdtrack1;
+ delete esdtrack2;
+ }
+ Double_t d0[2]={dz1[0],dz2[0]};
+ Double_t decl[2]={part->DecayLength(),part->NormalizedDecayLength()};
+ part->UnsetOwnPrimaryVtx();
+
+ if(!fCutOnDistr || (fCutOnDistr && isSelected)){ //if no cuts or cuts passed
+ //printf("\nif no cuts or cuts passed\n");
+
+ //disable the PID
+ if(!fUsePid4Distr) isSelectedPID=0;
+ if((lab>=0 && fReadMC) || (!fReadMC && isSelectedPID)){ //signal (from MC or PID)
+ //if(!vtxProp) fNentries->Fill(3);
+ //check pdg of the prongs
+ AliAODTrack *prong0=(AliAODTrack*)part->GetDaughter(0);
+ AliAODTrack *prong1=(AliAODTrack*)part->GetDaughter(1);
+ if(!prong0 || !prong1) {
+ return;
+ }
+ Int_t labprong[2];
+ if(fReadMC){
+ labprong[0]=prong0->GetLabel();
+ labprong[1]=prong1->GetLabel();
+ }
+ AliAODMCParticle *mcprong=0;
+ Int_t pdgProng[2]={0,0};
+
+ for (Int_t iprong=0;iprong<2;iprong++){
+ if(fReadMC && labprong[iprong]>=0) {
+ mcprong= (AliAODMCParticle*)arrMC->At(labprong[iprong]);
+ pdgProng[iprong]=mcprong->GetPdgCode();
+ }
+ }
+
+
+ //no mass cut ditributions: ptbis
+
+ fillthispi="hptpiSnoMcut_";
+ fillthispi+=ptbin;
+
+ fillthisK="hptKSnoMcut_";
+ fillthisK+=ptbin;
+
+ if ((TMath::Abs(pdgProng[0]) == 211 && TMath::Abs(pdgProng[1]) == 321)
+ || (isSelectedPID==1 || isSelectedPID==3)){
+ ((TH1F*)listout->FindObject(fillthispi))->Fill(part->PtProng(0));
+ ((TH1F*)listout->FindObject(fillthisK))->Fill(part->PtProng(1));
+ }
+
+ if ((TMath::Abs(pdgProng[0]) == 321 && TMath::Abs(pdgProng[1]) == 211)
+ || (isSelectedPID==2 || isSelectedPID==3)){
+ ((TH1F*)listout->FindObject(fillthisK))->Fill(part->PtProng(0));
+ ((TH1F*)listout->FindObject(fillthispi))->Fill(part->PtProng(1));
+ }
+
+ //no mass cut ditributions: mass
+ fillthis="hMassS_";
+ fillthis+=ptbin;
+
+ if ((fReadMC && ((AliAODMCParticle*)arrMC->At(lab))->GetPdgCode() == 421)
+ || (!fReadMC && (isSelectedPID==1 || isSelectedPID==3))){//D0
+ ((TH1F*)listout->FindObject(fillthis))->Fill(minvD0);
+ }
+ else { //D0bar
+ if(fReadMC || (!fReadMC && isSelectedPID > 1))
+ ((TH1F*)listout->FindObject(fillthis))->Fill(minvD0bar);
+ }
+
+ //apply cut on invariant mass on the pair
+ if(TMath::Abs(minvD0-mPDG)<invmasscut || TMath::Abs(minvD0bar-mPDG)<invmasscut){
+
+ if(fArray==1) cout<<"LS signal: ERROR"<<endl;
+ for (Int_t iprong=0; iprong<2; iprong++){
+ AliAODTrack *prong=(AliAODTrack*)part->GetDaughter(iprong);
+ if (fReadMC) labprong[iprong]=prong->GetLabel();
+
//cout<<"prong name = "<<prong->GetName()<<" label = "<<prong->GetLabel()<<endl;
- AliAODMCParticle *mcprong=0;
- if(labprong>=0) mcprong= (AliAODMCParticle*)mcArray->At(labprong);
- Int_t pdgprong=mcprong->GetPdgCode();
- if(TMath::Abs(pdgprong)==211) {
- //cout<<"pi"<<endl;
- ((TH1F*)fDistr->FindObject("hptpiS"))->Fill(d->PtProng(iprong));
- ((TH1F*)fDistr->FindObject("hd0piS"))->Fill(d->Getd0Prong(iprong));
+ Int_t pdgprong=0;
+ if(fReadMC && labprong[iprong]>=0) {
+ mcprong= (AliAODMCParticle*)arrMC->At(labprong[iprong]);
+ pdgprong=mcprong->GetPdgCode();
}
- if(TMath::Abs(pdgprong)==321) {
+ Bool_t isPionHere[2]={(isSelectedPID==1 || isSelectedPID==3) ? kTRUE : kFALSE,(isSelectedPID==2 || isSelectedPID==3) ? kTRUE : kFALSE};
+
+ if(TMath::Abs(pdgprong)==211 || isPionHere[iprong]) {
+ //cout<<"pi"<<endl;
+
+ fillthispi="hptpiS_";
+ fillthispi+=ptbin;
+ ((TH1F*)listout->FindObject(fillthispi))->Fill(part->PtProng(iprong));
+ fillthispi="hd0piS_";
+ fillthispi+=ptbin;
+ ((TH1F*)listout->FindObject(fillthispi))->Fill(part->Getd0Prong(iprong));
+ if(d0[iprong] != -99) {
+
+ fillthispi="hd0vpiS_";
+ fillthispi+=ptbin;
+ ((TH1F*)listout->FindObject(fillthispi))->Fill(d0[iprong]);
+ }
+
+ }
+
+ if(TMath::Abs(pdgprong)==321 || !isPionHere[iprong]) {
//cout<<"kappa"<<endl;
- ((TH1F*)fDistr->FindObject("hptKS"))->Fill(d->PtProng(iprong));
- ((TH1F*)fDistr->FindObject("hd0KS"))->Fill(d->Getd0Prong(iprong));
+
+ fillthisK="hptKS_";
+ fillthisK+=ptbin;
+ ((TH1F*)listout->FindObject(fillthisK))->Fill(part->PtProng(iprong));
+ fillthisK="hd0KS_";
+ fillthisK+=ptbin;
+ ((TH1F*)listout->FindObject(fillthisK))->Fill(part->Getd0Prong(iprong));
+ if (d0[iprong] != -99){
+ fillthisK="hd0vKS_";
+ fillthisK+=ptbin;
+ ((TH1F*)listout->FindObject(fillthisK))->Fill(d0[iprong]);
+ }
}
- ((TH1F*)fDistr->FindObject("hdcaS"))->Fill(d->GetDCA());
- }
+ fillthis="hcosthpointd0S_";
+ fillthis+=ptbin;
+ ((TH1F*)listout->FindObject(fillthis))->Fill(part->CosPointingAngle(),part->Getd0Prong(iprong));
- if (((AliAODMCParticle*)mcArray->At(lab))->GetPdgCode() == 421)
- ((TH1F*)fDistr->FindObject("costhetastarS"))->Fill(d->CosThetaStarD0());
- else ((TH1F*)fDistr->FindObject("costhetastarS"))->Fill(d->CosThetaStarD0bar());
+ } //end loop on prongs
- ((TH1F*)fDistr->FindObject("hd0d0S"))->Fill(d->Prodd0d0());
+ fillthis="hdcaS_";
+ fillthis+=ptbin;
+ ((TH1F*)listout->FindObject(fillthis))->Fill(part->GetDCA());
- ((TH1F*)fDistr->FindObject("hcosthetapointS"))->Fill(d->CosPointingAngle());
- ((TH1F*)fDistr->FindObject("hcosthpointd0d0S"))->Fill(d->CosPointingAngle(),d->Prodd0d0());
+ fillthis="hcosthetapointS_";
+ fillthis+=ptbin;
+ ((TH1F*)listout->FindObject(fillthis))->Fill(part->CosPointingAngle());
- //cout<<"ok point"<<endl;
-
- } else{ //Background
- //cout<<"is background"<<endl;
- // AliAODTrack *prong=(AliAODTrack*)d->GetDaughter(0);
- ((TH1F*)fDistr->FindObject("hptB"))->Fill(d->PtProng(0));
- //cout<<"ptok"<<endl;
- ((TH1F*)fDistr->FindObject("hd0B"))->Fill(d->Getd0Prong(0));
- //cout<<"d0ok"<<endl;
- ((TH1F*)fDistr->FindObject("hdcaB"))->Fill(d->GetDCA());
- //cout<<"dcaok"<<endl;
- ((TH1F*)fDistr->FindObject("costhetastarB"))->Fill(d->CosThetaStarD0());
- ((TH1F*)fDistr->FindObject("costhetastarB"))->Fill(d->CosThetaStarD0bar());
- ((TH1F*)fDistr->FindObject("hd0d0B"))->Fill(d->Prodd0d0());
- //cout<<"d0d0ok"<<endl;
- ((TH1F*)fDistr->FindObject("hcosthetapointB"))->Fill(d->CosPointingAngle());
- ((TH1F*)fDistr->FindObject("hcosthpointd0d0B"))->Fill(d->CosPointingAngle(),d->Prodd0d0());
+ fillthis="hcosthpointd0d0S_";
+ fillthis+=ptbin;
+ ((TH2F*)listout->FindObject(fillthis))->Fill(part->CosPointingAngle(),part->Prodd0d0());
+
+ fillthis="hcosthetastarS_";
+ fillthis+=ptbin;
+ if ((fReadMC && ((AliAODMCParticle*)arrMC->At(lab))->GetPdgCode() == 421)) ((TH1F*)listout->FindObject(fillthis))->Fill(part->CosThetaStarD0());
+ else {
+ if (fReadMC || isSelectedPID>1)((TH1F*)listout->FindObject(fillthis))->Fill(part->CosThetaStarD0bar());
+ if(isSelectedPID==1 || isSelectedPID==3)((TH1F*)listout->FindObject(fillthis))->Fill(part->CosThetaStarD0());
+ }
+ fillthis="hd0d0S_";
+ fillthis+=ptbin;
+ ((TH1F*)listout->FindObject(fillthis))->Fill(part->Prodd0d0());
- //cout<<"pointok"<<endl;
+ if(d0[0] != -99 && d0[1] != -99 ){
+ fillthis="hd0d0vS_";
+ fillthis+=ptbin;
+ ((TH1F*)listout->FindObject(fillthis))->Fill(d0[0]*d0[1]);
+ }
- }
+ fillthis="hdeclS_";
+ fillthis+=ptbin;
+ ((TH1F*)listout->FindObject(fillthis))->Fill(part->DecayLength());
- } //inv mass cut
-
+ fillthis="hnormdeclS_";
+ fillthis+=ptbin;
+ ((TH1F*)listout->FindObject(fillthis))->Fill(part->NormalizedDecayLength());
- Double_t pt = d->Pt();
+ if(vtxProp) {
+ fillthis="hdeclvS_";
+ fillthis+=ptbin;
+ ((TH1F*)listout->FindObject(fillthis))->Fill(decl[0]);
- Int_t ptbin=0;
+ fillthis="hnormdeclvS_";
+ fillthis+=ptbin;
+ ((TH1F*)listout->FindObject(fillthis))->Fill(decl[1]);
+ }
+ } //end mass cut
- //cout<<"P_t = "<<pt<<endl;
- if (pt>0. && pt<=1.) {
- ptbin=1;
- fVHFtight->SetD0toKpiCuts(0.7,0.04,0.8,0.5,0.5,0.05,0.05,-0.0003,0.7);
- fVHFloose->SetD0toKpiCuts(0.7,0.04,0.8,0.5,0.5,0.05,0.05,-0.00025,0.7);
-// fVHFtight->SetD0toKpiCuts(0.7,0.04,0.8,0.5,0.5,0.05,0.05,-0.0002,0.7);
-// fVHFloose->SetD0toKpiCuts(0.7,0.04,0.8,0.5,0.5,1,1,-0.00015,0.5);
- //printf("I'm in the bin %d\n",ptbin);
+ } else{ //Background or LS
+ //if(!fReadMC){
+ //cout<<"is background"<<endl;
+
+ //no mass cut distributions: mass, ptbis
+ fillthis="hMassB_";
+ fillthis+=ptbin;
- }
-
- if(pt>1. && pt<=3.) {
- ptbin=2;
- fVHFtight->SetD0toKpiCuts(0.7,0.02,0.8,0.7,0.7,0.05,0.05,-0.0003,0.9);
- fVHFloose->SetD0toKpiCuts(0.7,0.02,0.8,0.7,0.7,1,1,-0.00025,0.8);
- //printf("I'm in the bin %d\n",ptbin);
- }
-
- if(pt>3. && pt<=5.){
- ptbin=3;
- fVHFtight->SetD0toKpiCuts(0.7,0.015,0.8,0.7,0.7,0.05,0.05,-0.0002,0.9);
- fVHFloose->SetD0toKpiCuts(0.7,0.02,0.8,0.7,0.7,0.05,0.05,-0.00015,0.8);
- //printf("I'm in the bin %d\n",ptbin);
- }
- if(pt>5.){
- ptbin=4;
- fVHFtight->SetD0toKpiCuts(0.7,0.015,0.8,0.7,0.7,0.05,0.05,-0.0002,0.95);
- fVHFloose->SetD0toKpiCuts(0.7,0.02,0.8,0.7,0.7,0.05,0.05,-0.00015,0.9);
- }//if(pt>5)
+ if (!fCutOnDistr || (fCutOnDistr && (isSelected==1 || isSelected==3))) ((TH1F*)listout->FindObject(fillthis))->Fill(minvD0);
+ if (!fCutOnDistr || (fCutOnDistr && isSelected>1)) ((TH1F*)listout->FindObject(fillthis))->Fill(minvD0bar);
+
+ fillthis="hptB1prongnoMcut_";
+ fillthis+=ptbin;
+
+ ((TH1F*)listout->FindObject(fillthis))->Fill(part->PtProng(0));
+
+ fillthis="hptB2prongsnoMcut_";
+ fillthis+=ptbin;
+ ((TH1F*)listout->FindObject(fillthis))->Fill(part->PtProng(0));
+ ((TH1F*)listout->FindObject(fillthis))->Fill(part->PtProng(1));
+
+ //apply cut on invariant mass on the pair
+ if(TMath::Abs(minvD0-mPDG)<invmasscut || TMath::Abs(minvD0bar-mPDG)<invmasscut){
+
+
+ AliAODTrack *prongg=(AliAODTrack*)part->GetDaughter(0);
+ if(!prongg) {
+ if(fDebug>2) cout<<"No daughter found";
+ return;
+ }
+ else{
+ if(prongg->Charge()==1) {
+ fTotPosPairs[ptbin]++;
+ ((TH1F*)fDistr->FindObject("hpospair"))->Fill(ptbin);
+ } else {
+ fTotNegPairs[ptbin]++;
+ ((TH1F*)fDistr->FindObject("hnegpair"))->Fill(ptbin);
+ }
+ }
+
+ //normalise pt distr to half afterwards
+ fillthis="hptB_";
+ fillthis+=ptbin;
+ ((TH1F*)listout->FindObject(fillthis))->Fill(part->PtProng(0));
+ ((TH1F*)listout->FindObject(fillthis))->Fill(part->PtProng(1));
+
+ fillthis="hd0p0B_";
+ fillthis+=ptbin;
+ ((TH1F*)listout->FindObject(fillthis))->Fill(part->Getd0Prong(0));
+ fillthis="hd0p1B_";
+ fillthis+=ptbin;
+ ((TH1F*)listout->FindObject(fillthis))->Fill(part->Getd0Prong(1));
+ fillthis="hd0B_";
+ fillthis+=ptbin;
+ ((TH1F*)listout->FindObject(fillthis))->Fill(part->Getd0Prong(0));
+ ((TH1F*)listout->FindObject(fillthis))->Fill(part->Getd0Prong(1));
+
+ if(fReadMC){
+ Int_t pdgMother[2]={0,0};
+ Double_t factor[2]={1,1};
+
+ for(Int_t iprong=0;iprong<2;iprong++){
+ AliAODTrack *prong=(AliAODTrack*)part->GetDaughter(iprong);
+ lab=prong->GetLabel();
+ if(lab>=0){
+ AliAODMCParticle* mcprong=(AliAODMCParticle*)arrMC->At(lab);
+ if(mcprong){
+ Int_t labmom=mcprong->GetMother();
+ if(labmom>=0){
+ AliAODMCParticle* mcmother=(AliAODMCParticle*)arrMC->At(labmom);
+ if(mcmother) pdgMother[iprong]=mcmother->GetPdgCode();
+ }
+ }
+ }
+
+ fillthis="hd0moresB_";
+ fillthis+=ptbin;
+
+ if(TMath::Abs(pdgMother[iprong])==310 || TMath::Abs(pdgMother[iprong])==130 || TMath::Abs(pdgMother[iprong])==321){ //K^0_S, K^0_L, K^+-
+ if(part->PtProng(iprong)<=1)factor[iprong]=1./.7;
+ else factor[iprong]=1./.6;
+ fNentries->Fill(12);
+ }
+
+ if(TMath::Abs(pdgMother[iprong])==3122) { //Lambda
+ factor[iprong]=1./0.25;
+ fNentries->Fill(13);
+ }
+ fillthis="hd0moresB_";
+ fillthis+=ptbin;
+
+ ((TH1F*)listout->FindObject(fillthis))->Fill(part->Getd0Prong(iprong),factor[iprong]);
+
+ fillthis="hd0vmoresB_";
+ fillthis+=ptbin;
+ ((TH1F*)listout->FindObject(fillthis))->Fill(d0[iprong],factor[iprong]);
+
+ }
+
+ fillthis="hd0d0moresB_";
+ fillthis+=ptbin;
+ ((TH1F*)listout->FindObject(fillthis))->Fill(part->Prodd0d0(),factor[0]*factor[1]);
+
+ fillthis="hd0d0vmoresB_";
+ fillthis+=ptbin;
+ ((TH1F*)listout->FindObject(fillthis))->Fill(d0[0]*d0[1],factor[0]*factor[1]);
+
+ fillthis="hcosthetapointmoresB_";
+ fillthis+=ptbin;
+ ((TH1F*)listout->FindObject(fillthis))->Fill(part->CosPointingAngle(),factor[0]*factor[1]);
+ }
+
+ fillthis="hd0vp0B_";
+ fillthis+=ptbin;
+ ((TH1F*)listout->FindObject(fillthis))->Fill(d0[0]);
+ fillthis="hd0vp1B_";
+ fillthis+=ptbin;
+ ((TH1F*)listout->FindObject(fillthis))->Fill(d0[1]);
+
+ fillthis="hd0vB_";
+ fillthis+=ptbin;
+ ((TH1F*)listout->FindObject(fillthis))->Fill(d0[0]);
+ ((TH1F*)listout->FindObject(fillthis))->Fill(d0[1]);
- //printf("I'm in the bin %d\n",ptbin);
- //old
- //fVHF->SetD0toKpiCuts(0.7,0.03,0.8,0.06,0.06,0.05,0.05,-0.0002,0.6); //2.p-p vertex reconstructed
- FillHists(ptbin,d,mcArray,fVHFtight,fOutputtight);
- FillHists(ptbin,d,mcArray,fVHFloose,fOutputloose);
- if(unsetvtx) d->UnsetOwnPrimaryVtx();
+ fillthis="hcosthpointd0B_";
+ fillthis+=ptbin;
+ ((TH1F*)listout->FindObject(fillthis))->Fill(part->CosPointingAngle(),part->Getd0Prong(0));
+ ((TH1F*)listout->FindObject(fillthis))->Fill(part->CosPointingAngle(),part->Getd0Prong(1));
+
+
+ fillthis="hdcaB_";
+ fillthis+=ptbin;
+ ((TH1F*)listout->FindObject(fillthis))->Fill(part->GetDCA());
+
+ fillthis="hcosthetastarB_";
+ fillthis+=ptbin;
+ if (!fCutOnDistr || (fCutOnDistr && (isSelected==1 || isSelected==3)))((TH1F*)listout->FindObject(fillthis))->Fill(part->CosThetaStarD0());
+ if (!fCutOnDistr || (fCutOnDistr && isSelected>1))((TH1F*)listout->FindObject(fillthis))->Fill(part->CosThetaStarD0bar());
+
+ fillthis="hd0d0B_";
+ fillthis+=ptbin;
+ ((TH1F*)listout->FindObject(fillthis))->Fill(part->Prodd0d0());
+
+ if(d0[0] != -99 && d0[1] != -99 ){
+ fillthis="hd0d0vB_";
+ fillthis+=ptbin;
+ ((TH1F*)listout->FindObject(fillthis))->Fill(d0[0]*d0[1]);
+ }
+
+ fillthis="hcosthetapointB_";
+ fillthis+=ptbin;
+ ((TH1F*)listout->FindObject(fillthis))->Fill(part->CosPointingAngle());
+
+ fillthis="hcosthpointd0d0B_";
+ fillthis+=ptbin;
+ ((TH2F*)listout->FindObject(fillthis))->Fill(part->CosPointingAngle(),part->Prodd0d0());
+
+ fillthis="hdeclB_";
+ fillthis+=ptbin;
+ ((TH1F*)listout->FindObject(fillthis))->Fill(part->DecayLength());
+
+ fillthis="hnormdeclB_";
+ fillthis+=ptbin;
+ ((TH1F*)listout->FindObject(fillthis))->Fill(part->NormalizedDecayLength());
+
+ if(vtxProp) {
+ fillthis="hdeclvB_";
+ fillthis+=ptbin;
+ ((TH1F*)listout->FindObject(fillthis))->Fill(decl[0]);
+
+ fillthis="hnormdeclvB_";
+ fillthis+=ptbin;
+ ((TH1F*)listout->FindObject(fillthis))->Fill(decl[1]);
+ }
+ }//mass cut
+ }//else (background)
}
-
-
-
-
- // Post the data
- PostData(1,fOutputtight);
- PostData(2,fOutputloose);
- PostData(4,fDistr);
return;
}
-//____________________________________________________________________________*
-void AliAnalysisTaskSED0Mass::FillHists(Int_t ptbin, AliAODRecoDecayHF2Prong *part, TClonesArray *arrMC, AliAnalysisVertexingHF *vhf, TList *listout){
+//____________________________________________________________________________
+
+void AliAnalysisTaskSED0Mass::FillMassHists(AliAODEvent* aod,AliAODRecoDecayHF2Prong *part, TClonesArray *arrMC, AliRDHFCutsD0toKpi* cuts, TList *listout){
//
- // function used in UserExec:
+ // function used in UserExec to fill mass histograms:
//
- Int_t okD0=0,okD0bar=0;
- if(part->SelectD0(vhf->GetD0toKpiCuts(),okD0,okD0bar)) {//selected
- Double_t invmassD0 = part->InvMassD0(), invmassD0bar = part->InvMassD0bar();
- //printf("SELECTED\n");
- Int_t pdgDgD0toKpi[2]={321,211};
- Int_t labD0 = part->MatchToMC(421,arrMC,2,pdgDgD0toKpi); //return MC particle label if the array corresponds to a D0, -1 if not (cf. AliAODRecoDecay.cxx)
- //Int_t labD0 = part->MatchToMC(421,arrMC); //return MC particle label if the array corresponds to a D0, -1 if not (cf. AliAODRecoDecay.cxx)
- //printf("labD0 %d",labD0);
-
- TString fillthis="";
-
- if (okD0==1) {
- fillthis="histMass_";
- fillthis+=ptbin;
- //cout<<"Filling "<<fillthis<<endl;
+ Double_t mPDG=TDatabasePDG::Instance()->GetParticle(421)->Mass();
- //printf("Fill mass with D0");
- ((TH1F*)(listout->FindObject(fillthis)))->Fill(invmassD0);
- if(labD0>=0) {
- AliAODMCParticle *partD0 = (AliAODMCParticle*)arrMC->At(labD0);
- Int_t pdgD0 = partD0->GetPdgCode();
- //cout<<"pdg = "<<pdgD0<<endl;
- if (pdgD0==421){ //D0
- //cout<<"Fill S with D0"<<endl;
- fillthis="histSgn_";
- fillthis+=ptbin;
- ((TH1F*)(listout->FindObject(fillthis)))->Fill(invmassD0);
+ Int_t isSelected=cuts->IsSelected(part,AliRDHFCuts::kCandidate,aod); //selected
+
+ //cout<<"is selected = "<<isSelected<<endl;
+
+ //cout<<"check cuts = "<<endl;
+ //cuts->PrintAll();
+ if (!isSelected){
+ //cout<<"Not Selected"<<endl;
+ //cout<<"Rejected because "<<cuts->GetWhy()<<endl;
+ return;
+ }
+
+
+ if(fDebug>2) cout<<"Candidate selected"<<endl;
+
+ Double_t invmassD0 = part->InvMassD0(), invmassD0bar = part->InvMassD0bar();
+ //printf("SELECTED\n");
+ Int_t ptbin=cuts->PtBin(part->Pt());
+
+ AliAODTrack *prong=(AliAODTrack*)part->GetDaughter(0);
+ if(!prong) {
+ AliDebug(2,"No daughter found");
+ return;
+ }
+ else{
+ if(prong->Charge()==1) {fTotPosPairs[ptbin]++;} else {fTotNegPairs[ptbin]++;}
+ }
+
+ for(Int_t it=0;it<2;it++){
+ //fill TOFsig-kaontime
+ AliAODTrack* track=(AliAODTrack*)part->GetDaughter(it);
+ AliAODPid *pid = track->GetDetPid();
+ Double_t times[5];
+ pid->GetIntegratedTimes(times);
+ ((TH2F*)fChecks->FindObject("hTOFtimeKaonHyptimeAC"))->Fill(track->P(),pid->GetTOFsignal()-times[3]);
+ }
- } else{ //it was a D0bar
- fillthis="histRfl_";
+ //request on spd points to be addes
+ if(/*nSPD==2 && */part->Pt() > 5. && (TMath::Abs(invmassD0-mPDG)<0.01 || TMath::Abs(invmassD0bar-mPDG)<0.01)){
+ FILE *f=fopen("4display.txt","a");
+ fprintf(f,"pt: %f \n Rapidity: %f \t Period Number: %x \t Run Number: %d \t BunchCrossNumb: %d \t OrbitNumber: %d\n",part->Pt(),part->Y(421),aod->GetPeriodNumber(),aod->GetRunNumber(),aod->GetBunchCrossNumber(),aod->GetOrbitNumber());
+ fclose(f);
+ //printf("PrimVtx NContributors: %d \n Prongs Rel Angle: %f \n \n",ncont,relangle);
+ }
+
+
+
+ TString fillthis="";
+ Int_t pdgDgD0toKpi[2]={321,211};
+ Int_t labD0=-1;
+ if (fReadMC) labD0 = part->MatchToMC(421,arrMC,2,pdgDgD0toKpi); //return MC particle label if the array corresponds to a D0, -1 if not (cf. AliAODRecoDecay.cxx)
+
+ //count candidates selected by cuts
+ fNentries->Fill(1);
+ //count true D0 selected by cuts
+ if (fReadMC && labD0>=0) fNentries->Fill(2);
+ //PostData(3,fNentries);
+
+ if ((isSelected==1 || isSelected==3) && fFillOnlyD0D0bar<2) { //D0
+ fillthis="histMass_";
+ fillthis+=ptbin;
+ //cout<<"Filling "<<fillthis<<endl;
+
+ //printf("Fill mass with D0");
+ ((TH1F*)(listout->FindObject(fillthis)))->Fill(invmassD0);
+
+ if(labD0>=0) {
+ if(fArray==1) cout<<"LS signal ERROR"<<endl;
+
+ AliAODMCParticle *partD0 = (AliAODMCParticle*)arrMC->At(labD0);
+ Int_t pdgD0 = partD0->GetPdgCode();
+ //cout<<"pdg = "<<pdgD0<<endl;
+ if (pdgD0==421){ //D0
+ //cout<<"Fill S with D0"<<endl;
+ fillthis="histSgn_";
+ fillthis+=ptbin;
+ ((TH1F*)(listout->FindObject(fillthis)))->Fill(invmassD0);
+ if(TMath::Abs(invmassD0 - mPDG) < 0.027){
+ fillthis="histSgn27_";
fillthis+=ptbin;
((TH1F*)(listout->FindObject(fillthis)))->Fill(invmassD0);
}
- } else {//background
- fillthis="histBkg_";
+ } else{ //it was a D0bar
+ fillthis="histRfl_";
fillthis+=ptbin;
((TH1F*)(listout->FindObject(fillthis)))->Fill(invmassD0);
}
- }
- if (okD0bar==1) {
- fillthis="histMass_";
+ } else {//background
+ fillthis="histBkg_";
fillthis+=ptbin;
- //printf("Fill mass with D0bar");
- ((TH1F*)listout->FindObject(fillthis))->Fill(invmassD0bar);
+ ((TH1F*)(listout->FindObject(fillthis)))->Fill(invmassD0);
+ }
- if(labD0>=0) {
- AliAODMCParticle *partD0 = (AliAODMCParticle*)arrMC->At(labD0);
- Int_t pdgD0 = partD0->GetPdgCode();
- //cout<<" pdg = "<<pdgD0<<endl;
- if (pdgD0==-421){ //D0bar
- fillthis="histSgn_";
- fillthis+=ptbin;
- ((TH1F*)(listout->FindObject(fillthis)))->Fill(invmassD0bar);
-
- } else{
- fillthis="histRfl_";
+ }
+ if (isSelected>1 && (fFillOnlyD0D0bar==0 || fFillOnlyD0D0bar==2)) { //D0bar
+ fillthis="histMass_";
+ fillthis+=ptbin;
+ //printf("Fill mass with D0bar");
+ ((TH1F*)listout->FindObject(fillthis))->Fill(invmassD0bar);
+
+ if(labD0>=0) {
+ if(fArray==1) cout<<"LS signal ERROR"<<endl;
+ AliAODMCParticle *partD0 = (AliAODMCParticle*)arrMC->At(labD0);
+ Int_t pdgD0 = partD0->GetPdgCode();
+ //cout<<" pdg = "<<pdgD0<<endl;
+ if (pdgD0==-421){ //D0bar
+ fillthis="histSgn_";
+ fillthis+=ptbin;
+ ((TH1F*)(listout->FindObject(fillthis)))->Fill(invmassD0bar);
+ if (TMath::Abs(invmassD0bar - mPDG) < 0.027){
+ fillthis="histSgn27_";
fillthis+=ptbin;
((TH1F*)(listout->FindObject(fillthis)))->Fill(invmassD0bar);
}
- } else {//background
- fillthis="histBkg_";
+
+
+ } else{
+ fillthis="histRfl_";
fillthis+=ptbin;
((TH1F*)(listout->FindObject(fillthis)))->Fill(invmassD0bar);
}
+ } else {//background or LS
+ fillthis="histBkg_";
+ fillthis+=ptbin;
+ ((TH1F*)(listout->FindObject(fillthis)))->Fill(invmassD0bar);
}
+ }
+ return;
+}
- } //else cout<<"NOT SELECTED"<<endl;
+//__________________________________________________________________________
+AliAODVertex* AliAnalysisTaskSED0Mass::GetPrimaryVtxSkipped(AliAODEvent *aodev,AliAODRecoDecayHF2Prong *d){
+ //Calculate the primary vertex w/o the daughter tracks of the candidate
+
+ Int_t skipped[2];
+ Int_t nTrksToSkip=2;
+ AliAODTrack *dgTrack = (AliAODTrack*)d->GetDaughter(0);
+ if(!dgTrack){
+ AliDebug(2,"no daughter found!");
+ return 0x0;
+ }
+ skipped[0]=dgTrack->GetID();
+ dgTrack = (AliAODTrack*)d->GetDaughter(1);
+ if(!dgTrack){
+ AliDebug(2,"no daughter found!");
+ return 0x0;
+ }
+ skipped[1]=dgTrack->GetID();
+ AliESDVertex *vertexESD=0x0;
+ AliAODVertex *vertexAOD=0x0;
+ AliVertexerTracks *vertexer = new AliVertexerTracks(aodev->GetMagneticField());
+
+ //
+ vertexer->SetSkipTracks(nTrksToSkip,skipped);
+ vertexer->SetMinClusters(4);
+ vertexESD = (AliESDVertex*)vertexer->FindPrimaryVertex(aodev);
+ if(!vertexESD) return vertexAOD;
+ if(vertexESD->GetNContributors()<=0) {
+ AliDebug(2,"vertexing failed");
+ delete vertexESD; vertexESD=NULL;
+ return vertexAOD;
+ }
+
+ delete vertexer; vertexer=NULL;
+
+
+ // convert to AliAODVertex
+ Double_t pos[3],cov[6],chi2perNDF;
+ vertexESD->GetXYZ(pos); // position
+ vertexESD->GetCovMatrix(cov); //covariance matrix
+ chi2perNDF = vertexESD->GetChi2toNDF();
+ delete vertexESD; vertexESD=NULL;
+
+ vertexAOD = new AliAODVertex(pos,cov,chi2perNDF);
+ return vertexAOD;
+
}
+
+
//________________________________________________________________________
void AliAnalysisTaskSED0Mass::Terminate(Option_t */*option*/)
{
//
if(fDebug > 1) printf("AnalysisTaskSED0Mass: Terminate() \n");
- fOutputtight = dynamic_cast<TList*> (GetOutputData(1));
- if (!fOutputtight) {
- printf("ERROR: fOutputthight not available\n");
- return;
- }
- fOutputloose = dynamic_cast<TList*> (GetOutputData(2));
- if (!fOutputloose) {
- printf("ERROR: fOutputloose not available\n");
+
+ fOutputMass = dynamic_cast<TList*> (GetOutputData(1));
+ if (!fOutputMass) {
+ printf("ERROR: fOutputMass not available\n");
return;
}
- fDistr = dynamic_cast<TList*> (GetOutputData(3));
+ fDistr = dynamic_cast<TList*> (GetOutputData(2));
if (!fDistr) {
printf("ERROR: fDistr not available\n");
return;
}
+
+ fNentries = dynamic_cast<TH1F*>(GetOutputData(3));
+
+ if(!fNentries){
+ printf("ERROR: fNEntries not available\n");
+ return;
+ }
+
+ fChecks = dynamic_cast<TList*> (GetOutputData(4));
+ if (!fChecks) {
+ printf("ERROR: fChecks not available\n");
+ return;
+ }
+ fCounter = dynamic_cast<AliNormalizationCounter*>(GetOutputData(6));
+ if (!fCounter) {
+ printf("ERROR: fCounter not available\n");
+ return;
+ }
+ for(Int_t ipt=0;ipt<5;ipt++){ //change 5 in GetNPtBins when sure it is written and check
+
+
+ if(fArray==1){
+ fLsNormalization = 2.*TMath::Sqrt(fTotPosPairs[ipt]*fTotNegPairs[ipt]);
+
+
+ if(fLsNormalization>1e-6) {
+
+ TString massName="histMass_";
+ massName+=ipt;
+ ((TH1F*)fOutputMass->FindObject(massName))->Scale((1/fLsNormalization)*((TH1F*)fOutputMass->FindObject(massName))->GetEntries());
+
+ }
+
+
+ fLsNormalization = 2.*TMath::Sqrt(fTotPosPairs[4]*fTotNegPairs[4]);
+
+ if(fLsNormalization>1e-6) {
+
+ TString nameDistr="hptB_";
+ nameDistr+=ipt;
+ ((TH1F*)fDistr->FindObject(nameDistr))->Scale((1/fLsNormalization)*((TH1F*)fDistr->FindObject(nameDistr))->GetEntries());
+ nameDistr="hdcaB_";
+ nameDistr+=ipt;
+ ((TH1F*)fDistr->FindObject(nameDistr))->Scale((1/fLsNormalization)*((TH1F*)fDistr->FindObject(nameDistr))->GetEntries());
+ nameDistr="hcosthetastarB_";
+ nameDistr+=ipt;
+ ((TH1F*)fDistr->FindObject(nameDistr))->Scale((1/fLsNormalization)*((TH1F*)fDistr->FindObject(nameDistr))->GetEntries());
+ nameDistr="hd0B_";
+ nameDistr+=ipt;
+ ((TH1F*)fDistr->FindObject(nameDistr))->Scale((1/fLsNormalization)*((TH1F*)fDistr->FindObject(nameDistr))->GetEntries());
+ nameDistr="hd0d0B_";
+ nameDistr+=ipt;
+ ((TH1F*)fDistr->FindObject(nameDistr))->Scale((1/fLsNormalization)*((TH1F*)fDistr->FindObject(nameDistr))->GetEntries());
+ nameDistr="hcosthetapointB_";
+ nameDistr+=ipt;
+ ((TH1F*)fDistr->FindObject(nameDistr))->Scale((1/fLsNormalization)*((TH1F*)fDistr->FindObject(nameDistr))->GetEntries());
+ nameDistr="hcosthpointd0d0B_";
+ nameDistr+=ipt;
+ ((TH2F*)fDistr->FindObject(nameDistr))->Scale((1/fLsNormalization)*((TH2F*)fDistr->FindObject(nameDistr))->GetEntries());
+
+ }
+ }
+ }
+ TString cvname,cstname;
+
+ if (fArray==0){
+ cvname="D0invmass";
+ cstname="cstat0";
+ } else {
+ cvname="LSinvmass";
+ cstname="cstat1";
+ }
+
+ TCanvas *cMass=new TCanvas(cvname,cvname);
+ cMass->cd();
+ ((TH1F*)fOutputMass->FindObject("histMass_3"))->Draw();
+
+ TCanvas* cStat=new TCanvas(cstname,Form("Stat%s",fArray ? "LS" : "D0"));
+ cStat->cd();
+ cStat->SetGridy();
+ fNentries->Draw("htext0");
+ // TCanvas *ccheck=new TCanvas(Form("cc%d",fArray),Form("cc%d",fArray));
+ // ccheck->cd();
return;
}
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff