[u/mrichter/AliRoot.git] / PWG3 / vertexingHF / AliAnalysisTaskSEBkgLikeSignJPSI.cxx

/**************************************************************************
 * Copyright(c) 1998-2009, 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.                  *
 **************************************************************************/

///////////////////////////////////////////////////////////////////////////
//
// AliAnalysisTaskSE for reading both reconstructed JPSI -> ee candidates
// and like sign pairs and for drawing corresponding distributions
//
// Author: C.Di Giglio, carmelo.digiglio@ba.infn.it
///////////////////////////////////////////////////////////////////////////

#include <TSystem.h>
#include <TROOT.h>
#include <TClonesArray.h>

#include "AliAODEvent.h"
#include "AliAODVertex.h"
#include "AliAODTrack.h"
#include "AliAODRecoDecayHF2Prong.h"
#include "AliAnalysisTaskSE.h"
#include "AliAnalysisTaskSEBkgLikeSignJPSI.h"

ClassImp(AliAnalysisTaskSEBkgLikeSignJPSI)

//________________________________________________________________________
AliAnalysisTaskSEBkgLikeSignJPSI::AliAnalysisTaskSEBkgLikeSignJPSI():
AliAnalysisTaskSE(),
fOutput(0), 
fHistMassJPSI(0),
fHistMassLS(0),
fHistCtsJPSI(0),           
fHistCtsLS(0),
fHistCtsLSpos(0),
fHistCtsLSneg(0),
fHistCPtaJPSI(0),          
fHistCPtaLS(0),
fHistd0d0JPSI(0),          
fHistd0d0LS(0),
fHistDCAJPSI(0),           
fHistDCALS(0),
fVHF(0),
fTotPosPairs(0),
fTotNegPairs(0),
fLsNormalization(1.)
{
  //
  // Default constructor
  //
  // Output slot #1 writes into a TList container
  DefineOutput(1,TList::Class());  //My private output
}

//________________________________________________________________________
AliAnalysisTaskSEBkgLikeSignJPSI::AliAnalysisTaskSEBkgLikeSignJPSI(const char *name):
AliAnalysisTaskSE(name),
fOutput(0),
fHistMassJPSI(0),
fHistMassLS(0),
fHistCtsJPSI(0),
fHistCtsLS(0),
fHistCtsLSpos(0),
fHistCtsLSneg(0),
fHistCPtaJPSI(0),
fHistCPtaLS(0),
fHistd0d0JPSI(0),
fHistd0d0LS(0),
fHistDCAJPSI(0),
fHistDCALS(0),
fVHF(0),
fTotPosPairs(0),
fTotNegPairs(0),
fLsNormalization(1.)
{
  //
  // Default constructor
  //
  // Output slot #1 writes into a TList container
  DefineOutput(1,TList::Class());  //My private output
}

//________________________________________________________________________
AliAnalysisTaskSEBkgLikeSignJPSI::~AliAnalysisTaskSEBkgLikeSignJPSI()
{
  // Destructor
  if (fOutput) {
    delete fOutput;
    fOutput = 0;
  }

  if (fVHF) {
    delete fVHF;
    fVHF = 0;
  }

}
//________________________________________________________________________
void AliAnalysisTaskSEBkgLikeSignJPSI::Init()
{
  // Initialization

  if(fDebug > 1) printf("AnalysisTaskSEBkgLikeSignJPSI::Init() \n");

  gROOT->LoadMacro("ConfigVertexingHF.C");

  fVHF = (AliAnalysisVertexingHF*)gROOT->ProcessLine("ConfigVertexingHF()");
  fVHF->PrintStatus();

  return;

  return;
}

//________________________________________________________________________
void AliAnalysisTaskSEBkgLikeSignJPSI::UserCreateOutputObjects()
{
  // Create the output container
  //
  if(fDebug > 1) printf("AnalysisTaskSEBkgLikeSignJPSI::UserCreateOutputObjects() \n");

  // Several histograms are more conveniently managed in a TList
  fOutput = new TList();
  fOutput->SetOwner();

  fHistMassJPSI = new TH1F("fHistMassJPSI", "J/#Psi invariant mass; M [GeV]; Entries",200,2.8,3.25);
  fHistMassJPSI->Sumw2();
  fHistMassJPSI->SetMinimum(0);
  fOutput->Add(fHistMassJPSI);

  fHistMassLS = new TH1F("fHistMassLS", "Like sign pairs invariant mass; M [GeV]; Entries",200,2.8,3.25);
  fHistMassLS->Sumw2();
  fHistMassLS->SetMinimum(0);
  fOutput->Add(fHistMassLS);

  fHistCtsJPSI = new TH1F("fHistCtsJPSI", "J/#Psi cosine of decay angle; Cos#Theta*; Entries",200,-1.,1.);
  fHistCtsJPSI->Sumw2();
  fHistCtsJPSI->SetMinimum(0);
  fOutput->Add(fHistCtsJPSI);

  fHistCtsLS = new TH1F("fHistCtsLS", "Like sign pairs cosine of decay angle; Cos#Theta*; Entries",200,-1.,1.);
  fHistCtsLS->Sumw2();
  fHistCtsLS->SetMinimum(0);
  fOutput->Add(fHistCtsLS);

  fHistCtsLSpos = new TH1F("fHistCtsLSpos", "Like sign ++ pairs cosine of decay angle; Cos#Theta*; Entries",200,-1.,1.);
  fHistCtsLSpos->Sumw2();
  fHistCtsLSpos->SetMinimum(0);
  fOutput->Add(fHistCtsLSpos);

  fHistCtsLSneg = new TH1F("fHistCtsLSneg", "Like sign -- pairs cosine of decay angle; Cos#Theta*; Entries",200,-1.,1.);
  fHistCtsLSneg->Sumw2();
  fHistCtsLSneg->SetMinimum(0);
  fOutput->Add(fHistCtsLSneg);

  fHistCPtaJPSI = new TH1F("fHistCPtaJPSI", "J/#Psi cosine of pointing angle; Cos#Theta_{point}; Entries",200,-1.,1.);
  fHistCPtaJPSI->Sumw2();
  fHistCPtaJPSI->SetMinimum(0);
  fOutput->Add(fHistCPtaJPSI);

  fHistCPtaLS = new TH1F("fHistCPtaLS", "Like sign pairs cosine of pointing angle; Cos#Theta_{point}; Entries",200,-1.,1.);
  fHistCPtaLS->Sumw2();
  fHistCPtaLS->SetMinimum(0);
  fOutput->Add(fHistCPtaLS);

  fHistd0d0JPSI = new TH1F("fHistd0d0JPSI", "J/#Psi product of impact parameters; d0xd0 [#mu m^{2}]; Entries",200,-100000.,100000.);
  fHistd0d0JPSI->Sumw2(); 
  fHistd0d0JPSI->SetMinimum(0);
  fOutput->Add(fHistd0d0JPSI);

  fHistd0d0LS = new TH1F("fHistd0d0LS", "Like sign pairs product of impact parameters; d0xd0 [#mu m^{2}]; Entries",200,-100000.,100000.);
  fHistd0d0LS->Sumw2();
  fHistd0d0LS->SetMinimum(0);
  fOutput->Add(fHistd0d0LS);

  fHistDCAJPSI = new TH1F("fHistDCAJPSI", "J/#Psi distance of closest approach; dca [10^{2}#mu m]; Entries",100,0.,5.);
  fHistDCAJPSI->Sumw2(); 
  fHistDCAJPSI->SetMinimum(0);
  fOutput->Add(fHistDCAJPSI);

  fHistDCALS = new TH1F("fHistDCALS", "Like sign pairs distance of closest approach; dca [10^{2}#mu m]; Entries",100,0.,5.);
  fHistDCALS->Sumw2(); 
  fHistDCALS->SetMinimum(0);
  fOutput->Add(fHistDCALS);

  return;
}

//________________________________________________________________________
void AliAnalysisTaskSEBkgLikeSignJPSI::UserExec(Option_t */*option*/)
{
  // Execute analysis for current event:
  // heavy flavor candidates association to MC truth
  
  AliAODEvent *aod = dynamic_cast<AliAODEvent*> (InputEvent());

  // load heavy flavour vertices
  TClonesArray *arrayVerticesHF =
    (TClonesArray*)aod->GetList()->FindObject("VerticesHF");
  if(!arrayVerticesHF) {
    printf("AliAnalysisTaskSEBkgLikeSignJPSI::UserExec: VerticesHF branch not found!\n");
    return;
  }

  // load JPSI->ee candidates                                                   
  TClonesArray *arrayJPSItoEle =
    (TClonesArray*)aod->GetList()->FindObject("JPSItoEle");
  if(!arrayJPSItoEle) {
    printf("AliAnalysisTaskSEBkgLikeSignJPSI::UserExec: JPSItoEle branch not found!\n");
    return;
  }

  // load like sign candidates
  TClonesArray *arrayLikeSign =
    (TClonesArray*)aod->GetList()->FindObject("LikeSign2Prong");
  if(!arrayLikeSign) {
    printf("AliAnalysisTaskSEBkgLikeSignJPSI::UserExec: LikeSign2Prong branch not found!\n");
    return;
  }

  // AOD primary vertex
  AliAODVertex *vtx1 = (AliAODVertex*)aod->GetPrimaryVertex();

  // make trkIDtoEntry register (temporary)
  Int_t trkIDtoEntry[100000];
  for(Int_t it=0;it<aod->GetNumberOfTracks();it++) {
    AliAODTrack *track = aod->GetTrack(it);
    trkIDtoEntry[track->GetID()]=it;
  }

  // loop over Like sign candidates
  Int_t nPosPairs=0,nNegPairs=0;
  Int_t nLikeSign = arrayLikeSign->GetEntriesFast();
  printf("+++\n+++Number of like sign pairs ---> %d \n+++\n", nLikeSign);

  for(Int_t iLikeSign = 0; iLikeSign < nLikeSign; iLikeSign++) {
    AliAODRecoDecayHF2Prong *d = (AliAODRecoDecayHF2Prong*)arrayLikeSign->UncheckedAt(iLikeSign);
    Bool_t unsetvtx=kFALSE;
    if(!d->GetOwnPrimaryVtx()) {
        d->SetOwnPrimaryVtx(vtx1); // needed to compute all variables
        unsetvtx=kTRUE;
    }
    Int_t okBtoJPSIls=0;
    if(d->SelectBtoJPSI(fVHF->GetBtoJPSICuts(),okBtoJPSIls)) {
       fHistMassLS->Fill(d->InvMassJPSIee());
       fHistCPtaLS->Fill(d->CosPointingAngle());
       fHistd0d0LS->Fill(1e8*d->Prodd0d0());
       fHistCtsLS->Fill(d->CosThetaStarJPSI());
       fHistDCALS->Fill(100*d->GetDCA());
       PostData(1,fOutput);
       AliAODTrack *trk0 = (AliAODTrack*)d->GetDaughter(0);
       if(!trk0) {
          trk0=aod->GetTrack(trkIDtoEntry[d->GetProngID(0)]);
          printf("references to standard AOD not available \n");
       }
       if((trk0->Charge())==1) {
          nPosPairs++;
          fHistCtsLSpos->Fill(d->CosThetaStarJPSI());
          PostData(1,fOutput);
        } else {
          nNegPairs++;
          fHistCtsLSneg->Fill(d->CosThetaStarJPSI());
          PostData(1,fOutput);
        }
      PostData(1,fOutput);
      }
    if(unsetvtx) d->UnsetOwnPrimaryVtx();
  }

  printf("------------ N. of positive pairs in Event ----- %d \n", nPosPairs);
  printf("------------ N. of negative pairs in Event ----- %d \n", nNegPairs);

  fTotPosPairs += nPosPairs;
  fTotNegPairs += nNegPairs;

  // loop over JPSI candidates
  Int_t nBtoJpsiToEle = arrayJPSItoEle->GetEntriesFast();
  printf("Number of like JPSI -> ee candidates ---> %d \n", nBtoJpsiToEle);

  for (Int_t iBtoJpsiToEle = 0; iBtoJpsiToEle < nBtoJpsiToEle; iBtoJpsiToEle++) {
    AliAODRecoDecayHF2Prong *d = (AliAODRecoDecayHF2Prong*)arrayJPSItoEle->UncheckedAt(iBtoJpsiToEle);
    Bool_t unsetvtx=kFALSE;
    if(!d->GetOwnPrimaryVtx()) {
      d->SetOwnPrimaryVtx(vtx1); // needed to compute all variables
      unsetvtx=kTRUE;
    }
    Int_t okBtoJPSI=0;
    if(d->SelectBtoJPSI(fVHF->GetBtoJPSICuts(),okBtoJPSI)) {
      fHistMassJPSI->Fill(d->InvMassJPSIee());
      fHistCtsJPSI->Fill(d->CosThetaStarJPSI());
      fHistd0d0JPSI->Fill(1e8*d->Prodd0d0());
      fHistCPtaJPSI->Fill(d->CosPointingAngle());
      fHistDCAJPSI->Fill(100*d->GetDCA());
      PostData(1,fOutput);
    }
   if(unsetvtx) d->UnsetOwnPrimaryVtx();
  }

  return;
}

//________________________________________________________________________
void AliAnalysisTaskSEBkgLikeSignJPSI::Terminate(Option_t */*option*/)
{
  // Terminate analysis
  //
  if(fDebug > 1) printf("AnalysisTaskSEBkgLikeSignJPSI: Terminate() \n");

  fOutput = dynamic_cast<TList*> (GetOutputData(1));
  if (!fOutput) {     
    printf("ERROR: fOutput not available\n");
    return;
  }

  fLsNormalization = 2.*TMath::Sqrt(fTotPosPairs*fTotNegPairs); 

  fHistMassJPSI = dynamic_cast<TH1F*>(fOutput->FindObject("fHistMassJPSI"));
  fHistMassLS = dynamic_cast<TH1F*>(fOutput->FindObject("fHistMassLS"));
  fHistCtsJPSI = dynamic_cast<TH1F*>(fOutput->FindObject("fHistCtsJPSI"));
  fHistCtsLS = dynamic_cast<TH1F*>(fOutput->FindObject("fHistCtsLS"));
  fHistCtsLSpos = dynamic_cast<TH1F*>(fOutput->FindObject("fHistCtsLSpos"));
  fHistCtsLSneg = dynamic_cast<TH1F*>(fOutput->FindObject("fHistCtsLSneg"));
  fHistCPtaJPSI = dynamic_cast<TH1F*>(fOutput->FindObject("fHistCPtaJPSI"));
  fHistCPtaLS = dynamic_cast<TH1F*>(fOutput->FindObject("fHistCPtaLS"));
  fHistd0d0JPSI = dynamic_cast<TH1F*>(fOutput->FindObject("fHistd0d0JPSI"));
  fHistd0d0LS = dynamic_cast<TH1F*>(fOutput->FindObject("fHistd0d0LS"));
  fHistDCAJPSI = dynamic_cast<TH1F*>(fOutput->FindObject("fHistDCAJPSI"));
  fHistDCALS = dynamic_cast<TH1F*>(fOutput->FindObject("fHistDCALS"));

  fHistMassLS->Scale((1/fLsNormalization)*fHistMassLS->GetEntries());
  fHistCtsLS->Scale((1/fLsNormalization)*fHistCtsLS->GetEntries());
  fHistCtsLSpos->Scale((1/fLsNormalization)*fHistCtsLSpos->GetEntries());
  fHistCtsLSneg->Scale((1/fLsNormalization)*fHistCtsLSneg->GetEntries());
  fHistCPtaLS->Scale((1/fLsNormalization)*fHistCPtaLS->GetEntries());
  fHistd0d0LS->Scale((1/fLsNormalization)*fHistd0d0LS->GetEntries());
  fHistDCALS->Scale((1/fLsNormalization)*fHistDCALS->GetEntries());

  return;
}
Commit	Line	Data
adef5d4e	1	/**************************************************************************
	2	* Copyright(c) 1998-2009, 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	**************************************************************************/
	15
	16	///////////////////////////////////////////////////////////////////////////
	17	//
	18	// AliAnalysisTaskSE for reading both reconstructed JPSI -> ee candidates
	19	// and like sign pairs and for drawing corresponding distributions
	20	//
	21	// Author: C.Di Giglio, carmelo.digiglio@ba.infn.it
	22	///////////////////////////////////////////////////////////////////////////
	23
	24	#include <TSystem.h>
	25	#include <TROOT.h>
	26	#include <TClonesArray.h>
	27
	28	#include "AliAODEvent.h"
	29	#include "AliAODVertex.h"
	30	#include "AliAODTrack.h"
	31	#include "AliAODRecoDecayHF2Prong.h"
	32	#include "AliAnalysisTaskSE.h"
	33	#include "AliAnalysisTaskSEBkgLikeSignJPSI.h"
	34
	35	ClassImp(AliAnalysisTaskSEBkgLikeSignJPSI)
	36
	37	//________________________________________________________________________
	38	AliAnalysisTaskSEBkgLikeSignJPSI::AliAnalysisTaskSEBkgLikeSignJPSI():
	39	AliAnalysisTaskSE(),
	40	fOutput(0),
	41	fHistMassJPSI(0),
	42	fHistMassLS(0),
	43	fHistCtsJPSI(0),
	44	fHistCtsLS(0),
	45	fHistCtsLSpos(0),
	46	fHistCtsLSneg(0),
	47	fHistCPtaJPSI(0),
	48	fHistCPtaLS(0),
	49	fHistd0d0JPSI(0),
	50	fHistd0d0LS(0),
	51	fHistDCAJPSI(0),
	52	fHistDCALS(0),
	53	fVHF(0),
	54	fTotPosPairs(0),
	55	fTotNegPairs(0),
	56	fLsNormalization(1.)
	57	{
	58	//
	59	// Default constructor
	60	//
	61	// Output slot #1 writes into a TList container
	62	DefineOutput(1,TList::Class()); //My private output
	63	}
	64
65	//________________________________________________________________________
66	AliAnalysisTaskSEBkgLikeSignJPSI::AliAnalysisTaskSEBkgLikeSignJPSI(const char *name):
67	AliAnalysisTaskSE(name),
68	fOutput(0),
69	fHistMassJPSI(0),
70	fHistMassLS(0),
71	fHistCtsJPSI(0),
72	fHistCtsLS(0),
73	fHistCtsLSpos(0),
74	fHistCtsLSneg(0),
75	fHistCPtaJPSI(0),
76	fHistCPtaLS(0),
77	fHistd0d0JPSI(0),
78	fHistd0d0LS(0),
79	fHistDCAJPSI(0),
80	fHistDCALS(0),
81	fVHF(0),
82	fTotPosPairs(0),
83	fTotNegPairs(0),
84	fLsNormalization(1.)
85	{
86	//
87	// Default constructor
88	//
89	// Output slot #1 writes into a TList container
90	DefineOutput(1,TList::Class()); //My private output
91	}
92
93	//________________________________________________________________________
94	AliAnalysisTaskSEBkgLikeSignJPSI::~AliAnalysisTaskSEBkgLikeSignJPSI()
95	{
96	// Destructor
97	if (fOutput) {
98	delete fOutput;
99	fOutput = 0;
100	}
101
102	if (fVHF) {
103	delete fVHF;
104	fVHF = 0;
105	}
106
107	}
108	//________________________________________________________________________
109	void AliAnalysisTaskSEBkgLikeSignJPSI::Init()
110	{
111	// Initialization
112
113	if(fDebug > 1) printf("AnalysisTaskSEBkgLikeSignJPSI::Init() \n");
114
115	gROOT->LoadMacro("ConfigVertexingHF.C");
116
117	fVHF = (AliAnalysisVertexingHF*)gROOT->ProcessLine("ConfigVertexingHF()");
118	fVHF->PrintStatus();
119
120	return;
121
122	return;
123	}
124
125	//________________________________________________________________________
126	void AliAnalysisTaskSEBkgLikeSignJPSI::UserCreateOutputObjects()
127	{
128	// Create the output container
129	//
130	if(fDebug > 1) printf("AnalysisTaskSEBkgLikeSignJPSI::UserCreateOutputObjects() \n");
131
132	// Several histograms are more conveniently managed in a TList
133	fOutput = new TList();
134	fOutput->SetOwner();
135
136	fHistMassJPSI = new TH1F("fHistMassJPSI", "J/#Psi invariant mass; M [GeV]; Entries",200,2.8,3.25);
137	fHistMassJPSI->Sumw2();
138	fHistMassJPSI->SetMinimum(0);
139	fOutput->Add(fHistMassJPSI);
140
141	fHistMassLS = new TH1F("fHistMassLS", "Like sign pairs invariant mass; M [GeV]; Entries",200,2.8,3.25);
142	fHistMassLS->Sumw2();
143	fHistMassLS->SetMinimum(0);
144	fOutput->Add(fHistMassLS);
145
146	fHistCtsJPSI = new TH1F("fHistCtsJPSI", "J/#Psi cosine of decay angle; Cos#Theta*; Entries",200,-1.,1.);
147	fHistCtsJPSI->Sumw2();
148	fHistCtsJPSI->SetMinimum(0);
149	fOutput->Add(fHistCtsJPSI);
150
151	fHistCtsLS = new TH1F("fHistCtsLS", "Like sign pairs cosine of decay angle; Cos#Theta*; Entries",200,-1.,1.);
152	fHistCtsLS->Sumw2();
153	fHistCtsLS->SetMinimum(0);
154	fOutput->Add(fHistCtsLS);
155
156	fHistCtsLSpos = new TH1F("fHistCtsLSpos", "Like sign ++ pairs cosine of decay angle; Cos#Theta*; Entries",200,-1.,1.);
157	fHistCtsLSpos->Sumw2();
158	fHistCtsLSpos->SetMinimum(0);
159	fOutput->Add(fHistCtsLSpos);
160
161	fHistCtsLSneg = new TH1F("fHistCtsLSneg", "Like sign -- pairs cosine of decay angle; Cos#Theta*; Entries",200,-1.,1.);
162	fHistCtsLSneg->Sumw2();
163	fHistCtsLSneg->SetMinimum(0);
164	fOutput->Add(fHistCtsLSneg);
165
166	fHistCPtaJPSI = new TH1F("fHistCPtaJPSI", "J/#Psi cosine of pointing angle; Cos#Theta_{point}; Entries",200,-1.,1.);
167	fHistCPtaJPSI->Sumw2();
168	fHistCPtaJPSI->SetMinimum(0);
169	fOutput->Add(fHistCPtaJPSI);
170
171	fHistCPtaLS = new TH1F("fHistCPtaLS", "Like sign pairs cosine of pointing angle; Cos#Theta_{point}; Entries",200,-1.,1.);
172	fHistCPtaLS->Sumw2();
173	fHistCPtaLS->SetMinimum(0);
174	fOutput->Add(fHistCPtaLS);
175
176	fHistd0d0JPSI = new TH1F("fHistd0d0JPSI", "J/#Psi product of impact parameters; d0xd0 [#mu m^{2}]; Entries",200,-100000.,100000.);
177	fHistd0d0JPSI->Sumw2();
178	fHistd0d0JPSI->SetMinimum(0);
179	fOutput->Add(fHistd0d0JPSI);
180
181	fHistd0d0LS = new TH1F("fHistd0d0LS", "Like sign pairs product of impact parameters; d0xd0 [#mu m^{2}]; Entries",200,-100000.,100000.);
182	fHistd0d0LS->Sumw2();
183	fHistd0d0LS->SetMinimum(0);
184	fOutput->Add(fHistd0d0LS);
185
186	fHistDCAJPSI = new TH1F("fHistDCAJPSI", "J/#Psi distance of closest approach; dca [10^{2}#mu m]; Entries",100,0.,5.);
187	fHistDCAJPSI->Sumw2();
188	fHistDCAJPSI->SetMinimum(0);
189	fOutput->Add(fHistDCAJPSI);
190
191	fHistDCALS = new TH1F("fHistDCALS", "Like sign pairs distance of closest approach; dca [10^{2}#mu m]; Entries",100,0.,5.);
192	fHistDCALS->Sumw2();
193	fHistDCALS->SetMinimum(0);
194	fOutput->Add(fHistDCALS);
195
196	return;
197	}
198
199	//________________________________________________________________________
200	void AliAnalysisTaskSEBkgLikeSignJPSI::UserExec(Option_t /option*/)
201	{
202	// Execute analysis for current event:
203	// heavy flavor candidates association to MC truth
204
205	AliAODEvent aod = dynamic_cast<AliAODEvent> (InputEvent());
206
207	// load heavy flavour vertices
208	TClonesArray *arrayVerticesHF =
209	(TClonesArray*)aod->GetList()->FindObject("VerticesHF");
210	if(!arrayVerticesHF) {
211	printf("AliAnalysisTaskSEBkgLikeSignJPSI::UserExec: VerticesHF branch not found!\n");
212	return;
213	}
214
215	// load JPSI->ee candidates
216	TClonesArray *arrayJPSItoEle =
217	(TClonesArray*)aod->GetList()->FindObject("JPSItoEle");
218	if(!arrayJPSItoEle) {
219	printf("AliAnalysisTaskSEBkgLikeSignJPSI::UserExec: JPSItoEle branch not found!\n");
220	return;
221	}
222
223	// load like sign candidates
224	TClonesArray *arrayLikeSign =
225	(TClonesArray*)aod->GetList()->FindObject("LikeSign2Prong");
226	if(!arrayLikeSign) {
227	printf("AliAnalysisTaskSEBkgLikeSignJPSI::UserExec: LikeSign2Prong branch not found!\n");
228	return;
229	}
230
231	// AOD primary vertex
232	AliAODVertex vtx1 = (AliAODVertex)aod->GetPrimaryVertex();
233
234	// make trkIDtoEntry register (temporary)
235	Int_t trkIDtoEntry[100000];
236	for(Int_t it=0;it<aod->GetNumberOfTracks();it++) {
237	AliAODTrack *track = aod->GetTrack(it);
238	trkIDtoEntry[track->GetID()]=it;
239	}
240
241	// loop over Like sign candidates
242	Int_t nPosPairs=0,nNegPairs=0;
243	Int_t nLikeSign = arrayLikeSign->GetEntriesFast();
244	printf("+++\n+++Number of like sign pairs ---> %d \n+++\n", nLikeSign);
245
246	for(Int_t iLikeSign = 0; iLikeSign < nLikeSign; iLikeSign++) {
247	AliAODRecoDecayHF2Prong d = (AliAODRecoDecayHF2Prong)arrayLikeSign->UncheckedAt(iLikeSign);
248	Bool_t unsetvtx=kFALSE;
249	if(!d->GetOwnPrimaryVtx()) {
250	d->SetOwnPrimaryVtx(vtx1); // needed to compute all variables
251	unsetvtx=kTRUE;
252	}
253	Int_t okBtoJPSIls=0;
254	if(d->SelectBtoJPSI(fVHF->GetBtoJPSICuts(),okBtoJPSIls)) {
255	fHistMassLS->Fill(d->InvMassJPSIee());
256	fHistCPtaLS->Fill(d->CosPointingAngle());
257	fHistd0d0LS->Fill(1e8*d->Prodd0d0());
258	fHistCtsLS->Fill(d->CosThetaStarJPSI());
259	fHistDCALS->Fill(100*d->GetDCA());
260	PostData(1,fOutput);
261	AliAODTrack trk0 = (AliAODTrack)d->GetDaughter(0);
262	if(!trk0) {
263	trk0=aod->GetTrack(trkIDtoEntry[d->GetProngID(0)]);
264	printf("references to standard AOD not available \n");
265	}
266	if((trk0->Charge())==1) {
267	nPosPairs++;
268	fHistCtsLSpos->Fill(d->CosThetaStarJPSI());
269	PostData(1,fOutput);
270	} else {
271	nNegPairs++;
272	fHistCtsLSneg->Fill(d->CosThetaStarJPSI());
273	PostData(1,fOutput);
274	}
275	PostData(1,fOutput);
276	}
277	if(unsetvtx) d->UnsetOwnPrimaryVtx();
278	}
279
280	printf("------------ N. of positive pairs in Event ----- %d \n", nPosPairs);
281	printf("------------ N. of negative pairs in Event ----- %d \n", nNegPairs);
282
283	fTotPosPairs += nPosPairs;
284	fTotNegPairs += nNegPairs;
285
286	// loop over JPSI candidates
287	Int_t nBtoJpsiToEle = arrayJPSItoEle->GetEntriesFast();
288	printf("Number of like JPSI -> ee candidates ---> %d \n", nBtoJpsiToEle);
289
290	for (Int_t iBtoJpsiToEle = 0; iBtoJpsiToEle < nBtoJpsiToEle; iBtoJpsiToEle++) {
291	AliAODRecoDecayHF2Prong d = (AliAODRecoDecayHF2Prong)arrayJPSItoEle->UncheckedAt(iBtoJpsiToEle);
292	Bool_t unsetvtx=kFALSE;
293	if(!d->GetOwnPrimaryVtx()) {
294	d->SetOwnPrimaryVtx(vtx1); // needed to compute all variables
295	unsetvtx=kTRUE;
296	}
297	Int_t okBtoJPSI=0;
298	if(d->SelectBtoJPSI(fVHF->GetBtoJPSICuts(),okBtoJPSI)) {
299	fHistMassJPSI->Fill(d->InvMassJPSIee());
300	fHistCtsJPSI->Fill(d->CosThetaStarJPSI());
301	fHistd0d0JPSI->Fill(1e8*d->Prodd0d0());
302	fHistCPtaJPSI->Fill(d->CosPointingAngle());
303	fHistDCAJPSI->Fill(100*d->GetDCA());
304	PostData(1,fOutput);
305	}
306	if(unsetvtx) d->UnsetOwnPrimaryVtx();
307	}
308
309	return;
310	}
311
312	//________________________________________________________________________
313	void AliAnalysisTaskSEBkgLikeSignJPSI::Terminate(Option_t /option*/)
314	{
315	// Terminate analysis
316	//
317	if(fDebug > 1) printf("AnalysisTaskSEBkgLikeSignJPSI: Terminate() \n");
318
319	fOutput = dynamic_cast<TList*> (GetOutputData(1));
320	if (!fOutput) {
321	printf("ERROR: fOutput not available\n");
322	return;
323	}
324
325	fLsNormalization = 2.TMath::Sqrt(fTotPosPairsfTotNegPairs);
326
327	fHistMassJPSI = dynamic_cast<TH1F*>(fOutput->FindObject("fHistMassJPSI"));
328	fHistMassLS = dynamic_cast<TH1F*>(fOutput->FindObject("fHistMassLS"));
329	fHistCtsJPSI = dynamic_cast<TH1F*>(fOutput->FindObject("fHistCtsJPSI"));
330	fHistCtsLS = dynamic_cast<TH1F*>(fOutput->FindObject("fHistCtsLS"));
331	fHistCtsLSpos = dynamic_cast<TH1F*>(fOutput->FindObject("fHistCtsLSpos"));
332	fHistCtsLSneg = dynamic_cast<TH1F*>(fOutput->FindObject("fHistCtsLSneg"));
333	fHistCPtaJPSI = dynamic_cast<TH1F*>(fOutput->FindObject("fHistCPtaJPSI"));
334	fHistCPtaLS = dynamic_cast<TH1F*>(fOutput->FindObject("fHistCPtaLS"));
335	fHistd0d0JPSI = dynamic_cast<TH1F*>(fOutput->FindObject("fHistd0d0JPSI"));
336	fHistd0d0LS = dynamic_cast<TH1F*>(fOutput->FindObject("fHistd0d0LS"));
337	fHistDCAJPSI = dynamic_cast<TH1F*>(fOutput->FindObject("fHistDCAJPSI"));
338	fHistDCALS = dynamic_cast<TH1F*>(fOutput->FindObject("fHistDCALS"));
339
340	fHistMassLS->Scale((1/fLsNormalization)*fHistMassLS->GetEntries());
341	fHistCtsLS->Scale((1/fLsNormalization)*fHistCtsLS->GetEntries());
342	fHistCtsLSpos->Scale((1/fLsNormalization)*fHistCtsLSpos->GetEntries());
343	fHistCtsLSneg->Scale((1/fLsNormalization)*fHistCtsLSneg->GetEntries());
344	fHistCPtaLS->Scale((1/fLsNormalization)*fHistCPtaLS->GetEntries());
345	fHistd0d0LS->Scale((1/fLsNormalization)*fHistd0d0LS->GetEntries());
346	fHistDCALS->Scale((1/fLsNormalization)*fHistDCALS->GetEntries());
347
348	return;
349	}