#include "AliCFManager.h"
+#include "TMath.h"
#include "TObjArray.h"
#include "AliFlowCandidateTrack.h"
#include "AliFlowEventCuts.h"
AliAnalysisTaskSE(),
fCutsEvent(NULL),
fCuts(NULL),
- fList(NULL),
- fMass(NULL),
- fDCA(NULL),
- fDL(NULL),
- fCTP(NULL),
- fMassF(NULL),
- fDCAF(NULL),
- fDLF(NULL),
- fCTPF(NULL),
+ fQAList(NULL),
fMassMin(0),
- fMassMax(0)
+ fMassMax(0),
+ fDLcut(0),
+ tEvent(NULL),
+ tMulti(NULL)
{
+ for(int i=0; i!=4; ++i) {
+ tMass[i] = NULL;
+ tDCA[i] = NULL;
+ tDL[i] = NULL;
+ tCTP[i] = NULL;
+ td0d0[i] = NULL;
+ tPhi[i] = NULL;
+ tEta[i] = NULL;
+ tPt[i] = NULL;
+ tAPhi[i] = NULL;
+ tAEta[i] = NULL;
+ tAPt[i] = NULL;
+ tBPhi[i] = NULL;
+ tBEta[i] = NULL;
+ tBPt[i] = NULL;
+ }
}
//_____________________________________________________________________________
AliAnalysisTaskSE(name),
fCutsEvent(cutsEvent),
fCuts(cuts),
- fList(NULL),
- fMass(NULL),
- fDCA(NULL),
- fDL(NULL),
- fCTP(NULL),
- fMassF(NULL),
- fDCAF(NULL),
- fDLF(NULL),
- fCTPF(NULL),
+ fQAList(NULL),
fMassMin(MassMin),
- fMassMax(MassMax)
+ fMassMax(MassMax),
+ fDLcut(0),
+ tEvent(NULL),
+ tMulti(NULL)
{
+ for(int i=0; i!=4; ++i) {
+ tMass[i] = NULL;
+ tDCA[i] = NULL;
+ tDL[i] = NULL;
+ tCTP[i] = NULL;
+ td0d0[i] = NULL;
+ tPhi[i] = NULL;
+ tEta[i] = NULL;
+ tPt[i] = NULL;
+ tAPhi[i] = NULL;
+ tAEta[i] = NULL;
+ tAPt[i] = NULL;
+ tBPhi[i] = NULL;
+ tBEta[i] = NULL;
+ tBPt[i] = NULL;
+ }
DefineInput( 0, TChain::Class() );
DefineOutput( 1, TObjArray::Class() );
DefineOutput( 2, TList::Class() );
//_____________________________________________________________________________
AliAnalysisTaskFlowK0Candidates::~AliAnalysisTaskFlowK0Candidates()
{
- if(fList) {
- fList->Delete();
- delete fList;
- }
+ if(fQAList) delete fQAList;
}
//_____________________________________________________________________________
void AliAnalysisTaskFlowK0Candidates::UserCreateOutputObjects()
{
- fList = new TList();
- fList->SetOwner();
- fMass = new TH1D( "fMass","Mass;M_{#pi#pi} [GeV];Counts per MeV", 180, 0.41, 0.59); fList->Add( fMass );
- fDCA = new TH1D( "fDCA" ,"DCA;[cm];Counts per 10 um", 180, 0.00, 0.18); fList->Add( fDCA );
- fDL = new TH1D( "fDL" ,"Decay Length;[cm];Counts per 0.1 mm", 180, 0.00, 1.80); fList->Add( fDL );
- fCTP = new TH1D( "fCTP" ,"Cos#theta_{p}", 180,-1.10, 1.10); fList->Add( fCTP );
-
- fMassF= new TH1D( "fMassF","Mass after Cuts;M_{#pi#pi} [GeV];Counts per MeV", 180, 0.41, 0.59); fList->Add( fMassF );
- fDCAF = new TH1D( "fDCAF" ,"DCA after Cuts;[cm];Counts per 10 um", 180, 0.00, 0.18); fList->Add( fDCAF );
- fDLF = new TH1D( "fDLF" ,"Decay Length after Cuts;[cm];Counts per 0.1 mm", 180, 0.00, 1.80); fList->Add( fDLF );
- fCTPF = new TH1D( "fCTPF" ,"Cos#theta_{p} after Cuts", 180,-1.10, 1.10); fList->Add( fCTPF );
-
- PostData( 2, fList);
+ fQAList = new TList();
+ fQAList->SetOwner();
+ AddQAEvents();
+ AddQACandidates();
+ PostData( 2, fQAList);
+}
+//_____________________________________________________________________________
+void AliAnalysisTaskFlowK0Candidates::AddQAEvents()
+{
+ TList *tQAEvents = new TList();
+ tQAEvents->SetName("Events");
+ tQAEvents->SetOwner();
+ tEvent = new TH1D("Event", "Number of Events", 2, 0, 2);
+ tQAEvents->Add(tEvent);
+ tMulti = new TH1D("Multiplicity", "Multiplicity", 180, 0, 10000);
+ tQAEvents->Add(tMulti);
+ fQAList->Add(tQAEvents);
+}
+//_____________________________________________________________________________
+void AliAnalysisTaskFlowK0Candidates::AddQACandidates()
+{
+ TList *tQACandidates[4];
+ TList *tQADaughters[4];
+ for(int i=0; i!=4; ++i) {
+ tQACandidates[i] = new TList();
+ tQACandidates[i]->SetOwner();
+ tQACandidates[i]->SetName(Form("Candidates%d",i));
+ tMass[i] = new TH1D( Form("Mass%i",i), "Mass;M_{#pi#pi} [GeV];Counts per MeV", 180, 0.41, 0.59); tQACandidates[i]->Add( tMass[i] );
+ tDCA[i] = new TH1D( Form("DCA%i" ,i), "DCA;[cm];Counts per 10 um", 180, 0.00, 0.18); tQACandidates[i]->Add( tDCA[i] );
+ tDL[i] = new TH1D( Form("DL%i" ,i), "Decay Length;[cm];Counts per 0.1 mm", 180, 0.00, 1.80); tQACandidates[i]->Add( tDL[i] );
+ tCTP[i] = new TH1D( Form("CTP%i" ,i), "Cos#theta_{p}", 180,-1.10, 1.10); tQACandidates[i]->Add( tCTP[i] );
+ td0d0[i] = new TH1D( Form("d0d0%i",i), "d_{0}xd_{0};[cm^{2}];Cnts 0.01 mm^{2}",180,-0.009,0.009);tQACandidates[i]->Add( td0d0[i] );
+ tPhi[i] = new TH1D( Form("Phi%i" ,i), "Phi;[rad];Counts per degree", 180,0,TMath::TwoPi()); tQACandidates[i]->Add( tPhi[i] );
+ tEta[i] = new TH1D( Form("Eta%i" ,i), "Eta;;Counts per 0.04", 180,-3.60, 3.60); tQACandidates[i]->Add( tEta[i] );
+ tPt[i] = new TH1D( Form("Pt%i" ,i), "Pt;[GeV];Counts per 0.1 GeV", 180, 0.00,18.00); tQACandidates[i]->Add( tPt[i] );
+ tQADaughters[i] = new TList();
+ tQADaughters[i]->SetOwner();
+ tQADaughters[i]->SetName(Form("Daughters%d",i));
+ tAPhi[i] = new TH1D( Form("PhiBef%i",i), "Phi prePropagation;[rad];Cnts per degree",180,0,TMath::TwoPi()); tQADaughters[i]->Add( tAPhi[i] );
+ tAEta[i] = new TH1D( Form("EtaBef%i",i), "Eta prePropagation;;Counts per 0.04" ,180,-3.6,3.6); tQADaughters[i]->Add( tAEta[i] );
+ tAPt[i] = new TH1D( Form("PtBef%i" ,i), "Pt prePropagation;[GeV];Counts per 0.1 GeV",180, 0, 18); tQADaughters[i]->Add( tAPt[i] );
+ tBPhi[i] = new TH1D( Form("PhiAft%i",i), "Phi posPropagation;[rad];Cnts per degree",180,0,TMath::TwoPi()); tQADaughters[i]->Add( tBPhi[i] );
+ tBEta[i] = new TH1D( Form("EtaAft%i",i), "Eta posPropagation;;Counts per 0.04" ,180,-3.6,3.6); tQADaughters[i]->Add( tBEta[i] );
+ tBPt[i] = new TH1D( Form("PtAft%i" ,i), "Pt posPropagation;[GeV];Counts per 0.1 GeV",180, 0, 18); tQADaughters[i]->Add( tBPt[i] );
+ tQACandidates[i]->Add(tQADaughters[i]);
+ fQAList->Add(tQACandidates[i]);
+ }
}
-
-
//_____________________________________________________________________________
void AliAnalysisTaskFlowK0Candidates::NotifyRun()
{
- AliESDEvent *fESD = dynamic_cast<AliESDEvent*>(InputEvent());
- if(!fESD) return;
+// AliESDEvent *fESD = dynamic_cast<AliESDEvent*>(InputEvent());
+// if(!fESD) return;
}
//_____________________________________________________________________________
void AliAnalysisTaskFlowK0Candidates::UserExec(Option_t *)
{
- AliESDEvent *fESD = dynamic_cast<AliESDEvent*>(fInputEvent);
- if(!fESD) return;
+ AliESDEvent *fESD = dynamic_cast<AliESDEvent*>(InputEvent());
+ AliAODEvent *fAOD = dynamic_cast<AliAODEvent*>(InputEvent());
+ if( (!fESD)&&(!fAOD) )
+ return;
+// printf("\nEvent found");
+ tEvent->Fill( 0 );
if(fCutsEvent)
- if( !(fCutsEvent->IsSelected(InputEvent())) ) return;
-
+ if( !(fCutsEvent->IsSelected(InputEvent())) )
+ return;
+ tEvent->Fill( 1 );
+ if(fESD)
+ ReadFromESD(fESD);
+ else if(fAOD)
+ ReadFromAOD(fAOD);
+}
+//_____________________________________________________________________________
+void AliAnalysisTaskFlowK0Candidates::ReadFromESD(AliESDEvent *fESD)
+{
TObjArray *POIselection = new TObjArray();
POIselection->SetOwner();
-
int nTracks = fESD->GetNumberOfTracks();
- for(int i=0; i!=nTracks; ++i) { // first particle
+ tMulti->Fill( nTracks );
+ for(int i=0; i!=nTracks; ++i) {
AliESDtrack *ioT = fESD->GetTrack(i);
if(fCuts)
if( !(fCuts->IsSelected(ioT)) )
continue;
- for(int j=i+1; j!=nTracks; ++j) { // second particle
+ // printf("\n first particle OK...");
+ for(int j=i+1; j!=nTracks; ++j) {
AliESDtrack *joT = fESD->GetTrack(j);
- if( (ioT->Charge()*joT->Charge()) > 0 ) // only keeping opposite signs
+ if( (ioT->Charge()*joT->Charge()) > 0 )
continue;
if(fCuts)
if( !(fCuts->IsSelected(joT)) )
continue;
+ // printf("\n second also...");
AliESDtrack *iT = new AliESDtrack(*ioT);
AliESDtrack *jT = new AliESDtrack(*joT);
// getting distance of closest approach
double DCA = iT->PropagateToDCA(jT,fESD->GetMagneticField());
- fDCA->Fill( DCA );
+ // printf(" propagated...");
// getting decay length
+ TVector3 vp, vv, vl;
+ vp = TVector3( (iT->Xv()+jT->Xv())/2, (iT->Yv()+jT->Yv())/2, (iT->Zv()+jT->Zv())/2 ); // candidate position
+ vv = TVector3( fESD->GetPrimaryVertex()->GetX(), fESD->GetPrimaryVertex()->GetY(), fESD->GetPrimaryVertex()->GetZ() ); // vertex position
+ vl = vp - vv; // flight line
double DL;
- double vx = fESD->GetPrimaryVertex()->GetX();
- double vy = fESD->GetPrimaryVertex()->GetY();
- double vz = fESD->GetPrimaryVertex()->GetZ();
- double px = (iT->Xv()+jT->Xv())/2;
- double py = (iT->Yv()+jT->Yv())/2;
- double pz = (iT->Zv()+jT->Zv())/2;
- DL = (vx-px)*(vx-px);
- DL += (vy-py)*(vy-py);
- DL += (vz-pz)*(vz-pz);
- DL = sqrt(DL);
- fDL->Fill( DL );
- // cos pointing angle
+ DL = vl.Mag();
+ // getting cos pointing angle
double CTP;
- TVector3 vi, vj, vs, vp;
+ TVector3 vi, vj, vs;
vi = TVector3( iT->Px(), iT->Py(), iT->Pz() );
vj = TVector3( jT->Px(), jT->Py(), jT->Pz() );
vs = vi + vj;
- vp = TVector3( px, py, pz );
- CTP = TMath::Cos( vp.Angle(vs) );
- fCTP->Fill(CTP);
-
+ CTP = TMath::Cos( vl.Angle(vs) ); // Marta says: "Carlos, it is faster if you calculate this by hand!"
+ double D0D0;
+ D0D0 = iT->GetD(vv.X(),vv.Y(),fESD->GetMagneticField())*jT->GetD(vv.X(),vv.Y(),fESD->GetMagneticField());
// getting invariant mass
double sum12 = iT->P()*iT->P()+jT->P()*jT->P();
double pro12 = iT->P()*iT->P()*jT->P()*jT->P();
InvMass -= ( iT->Px()*jT->Px()+iT->Py()*jT->Py()+iT->Pz()*jT->Pz() );
InvMass *= 2;
InvMass = sqrt(InvMass);
- fMass->Fill( InvMass );
-
- // evaluating cuts
- bool bW = (InvMass>fMassMin) && (InvMass<fMassMax);
- bool bCTP = (fabs(CTP)>0.95);
- bool bDCA = (DCA<0.05);
- bool bDL = (DL>2.5);
- if( (!bW) || (!bDCA) || (!bDL) || (!bCTP) ) {
- delete iT;
- delete jT;
- continue;
+ // filtering
+ int iLevel = 3;
+ if( CTP<0.8 )
+ iLevel = 2;
+ if( DL<fDLcut ) // 0.5
+ iLevel = 1;
+ if( DCA>0.05 )
+ iLevel = 0;
+ // printf(" candidate at level %d...",iLevel);
+ for(int h=0; h!=iLevel+1; ++h) {
+ // printf(" %d",h);
+ // candidate
+ tDCA[h]->Fill( DCA );
+ tDL[h]->Fill( DL );
+ tCTP[h]->Fill( CTP );
+ td0d0[h]->Fill( D0D0 );
+ tMass[h]->Fill( InvMass );
+ tPhi[h]->Fill( vs.Phi()+TMath::Pi() );
+ tEta[h]->Fill( vs.Eta() );
+ tPt[h]->Fill( vs.Pt() );
+ // daughters
+ tAPhi[h]->Fill( iT->Phi() );
+ tAEta[h]->Fill( iT->Eta() );
+ tAPt[h]->Fill( iT->Pt() );
+ tAPhi[h]->Fill( jT->Phi() );
+ tAEta[h]->Fill( jT->Eta() );
+ tAPt[h]->Fill( jT->Pt() );
+ tBPhi[h]->Fill( ioT->Phi());
+ tBEta[h]->Fill( ioT->Eta());
+ tBPt[h]->Fill( ioT->Pt() );
+ tBPhi[h]->Fill( joT->Phi());
+ tBEta[h]->Fill( joT->Eta());
+ tBPt[h]->Fill( joT->Pt() );
+ }
+ // building candidate
+ if( (InvMass>fMassMin) && (InvMass<fMassMax) && (iLevel==3) ) {
+ AliFlowCandidateTrack *sTrack = new AliFlowCandidateTrack();
+ sTrack->SetMass( InvMass );
+ sTrack->SetPt( vs.Pt() );
+ sTrack->SetPhi( vs.Phi()+TMath::Pi() );
+ sTrack->SetEta( vs.Eta() );
+ sTrack->SetCharge( 0 );
+ sTrack->AddDaughter( iT->GetID() );
+ sTrack->AddDaughter( jT->GetID() );
+ POIselection->AddLast( sTrack );
}
- fDCAF->Fill( DCA );
- fDLF->Fill( DL );
- fCTPF->Fill(CTP);
- fMassF->Fill( InvMass );
-
- AliFlowCandidateTrack *sTrack = new AliFlowCandidateTrack();
- //booking pt of selected candidates
- sTrack->SetMass( InvMass );
- sTrack->SetPt( vs.Pt() );
- sTrack->SetPhi( vs.Phi() );
- sTrack->SetEta( vs.Eta() );
- sTrack->SetCharge( 0 );
- sTrack->AddDaughter( i );
- sTrack->AddDaughter( j );
- //saving selected candidate
- POIselection->AddLast( sTrack );
delete iT;
delete jT;
+ } // endfor j
+ } // endfor i
+ PostData( 1, POIselection );
+ PostData( 2, fQAList);
+}
+
+//_____________________________________________________________________________
+void AliAnalysisTaskFlowK0Candidates::ReadFromAOD(AliAODEvent *fAOD)
+{
+ TObjArray *POIselection = new TObjArray();
+ POIselection->SetOwner();
+ int nTracks = fAOD->GetNumberOfTracks();
+ tMulti->Fill( nTracks );
+ for(int i=0; i!=fAOD->GetNumberOfV0s(); ++i) {
+ AliAODv0 *iV0 = (AliAODv0*) fAOD->GetV0(i);
+ if ( iV0->GetOnFlyStatus() ) continue;
+ if ( (iV0->ChargeProng(0)*iV0->ChargeProng(1))>0 ) continue;
+ // getting distance of closest approach
+ double DCA = iV0->DcaV0Daughters();
+ // getting decay length
+ double DL;
+ double vv[3];
+ vv[0] = fAOD->GetPrimaryVertex()->GetX();
+ vv[1] = fAOD->GetPrimaryVertex()->GetY();
+ vv[2] = fAOD->GetPrimaryVertex()->GetZ();
+ DL = iV0->DecayLengthV0(vv);
+ // cos pointing angle
+ double CTP;
+ CTP = iV0->CosPointingAngle(vv);
+ double D0D0;
+ D0D0 = iV0->Prodd0d0();
+ // getting invariant mass
+ double InvMass = iV0->MassK0Short();
+ // filtering
+ int iLevel = 3;
+ if( CTP<0.8 )
+ iLevel = 2;
+ if( DL<fDLcut ) // 0.5
+ iLevel = 1;
+ if( DCA>0.05 )
+ iLevel = 0;
+ for(int h=0; h!=iLevel+1; ++h) {
+ // candidate
+ tDCA[h]->Fill( DCA );
+ tDL[h]->Fill( DL );
+ tCTP[h]->Fill( CTP );
+ td0d0[h]->Fill( D0D0 );
+ tMass[h]->Fill( InvMass );
+ tPhi[h]->Fill( iV0->Phi() );
+ tEta[h]->Fill( iV0->Eta() );
+ tPt[h]->Fill( iV0->Pt() );
+ // daughters
+ tAPhi[h]->Fill( ( (AliAODTrack*) iV0->GetDaughter(0) )->Phi() );
+ tAEta[h]->Fill( ( (AliAODTrack*) iV0->GetDaughter(0) )->Eta() );
+ tAPt[h]->Fill( ( (AliAODTrack*) iV0->GetDaughter(0) )->Pt() );
+ tAPhi[h]->Fill( ( (AliAODTrack*) iV0->GetDaughter(1) )->Phi() );
+ tAEta[h]->Fill( ( (AliAODTrack*) iV0->GetDaughter(1) )->Eta() );
+ tAPt[h]->Fill( ( (AliAODTrack*) iV0->GetDaughter(1) )->Pt() );
+ tBPhi[h]->Fill( iV0->PhiProng(0) );
+ tBEta[h]->Fill( iV0->EtaProng(0) );
+ tBPt[h]->Fill( iV0->PtProng(0) );
+ tBPhi[h]->Fill( iV0->PhiProng(1) );
+ tBEta[h]->Fill( iV0->EtaProng(1) );
+ tBPt[h]->Fill( iV0->PtProng(1) );
}
+ AliFlowCandidateTrack *sTrack = new AliFlowCandidateTrack();
+ sTrack->SetMass( InvMass );
+ sTrack->SetPt( iV0->Pt() );
+ sTrack->SetPhi( iV0->Phi() );
+ sTrack->SetEta( iV0->Eta() );
+ sTrack->SetCharge( 0 );
+ sTrack->AddDaughter( iV0->GetPosID() );
+ sTrack->AddDaughter( iV0->GetNegID() );
+ POIselection->AddLast( sTrack );
}
PostData( 1, POIselection );
- PostData( 2, fList);
+ PostData( 2, fQAList);
}
//_____________________________________________________________________________
git://git.uio.no / u/mrichter/AliRoot.git / commitdiff