#include <TH2F.h>
#include <TH3F.h>
#include <TRandom.h>
+#include <TParameter.h>
#include "AliAnalysisTaskPhiCorrelations.h"
#include "AliAnalyseLeadingTrackUE.h"
#include "AliAODMCHeader.h"
#include "AliGenCocktailEventHeader.h"
#include "AliGenEventHeader.h"
+#include "AliCollisionGeometry.h"
#include "AliEventPoolManager.h"
#include "AliESDtrackCuts.h"
#include "AliHelperPID.h"
+#include "AliAnalysisUtils.h"
+#include "TMap.h"
////////////////////////////////////////////////////////////////////////
//
fReduceMemoryFootprint(kFALSE),
fFillMixed(kTRUE),
fMixingTracks(50000),
-fCompareCentralities(kFALSE),
fTwoTrackEfficiencyStudy(kFALSE),
fTwoTrackEfficiencyCut(0),
+fTwoTrackCutMinRadius(0.8),
fUseVtxAxis(kFALSE),
fCourseCentralityBinning(kFALSE),
fSkipTrigger(kFALSE),
fInjectedSignals(kFALSE),
-// pointers to UE classes
+fRandomizeReactionPlane(kFALSE),
fHelperPID(0x0),
+fAnalysisUtils(0x0),
+fMap(0x0),
+// pointers to UE classes
fAnalyseUE(0x0),
fHistos(0x0),
fHistosMixed(0),
fEfficiencyCorrectionTriggers(0),
fEfficiencyCorrectionAssociated(0),
+fCentralityWeights(0),
// handlers and events
fAOD(0x0),
fESD(0x0),
fCentralityMethod("V0M"),
// track cuts
fTrackEtaCut(0.8),
+fTrackEtaCutMin(-1.),
fOnlyOneEtaSide(0),
fPtMin(0.5),
+fDCAXYCut(0),
+fSharedClusterCut(-1),
+fCrossedRowsCut(-1),
+fFoundFractionCut(-1),
fFilterBit(0xFF),
fTrackStatus(0),
fSelectBit(AliVEvent::kMB|AliVEvent::kUserDefined),
fEtaOrdering(kFALSE),
fCutConversions(kFALSE),
fCutResonances(kFALSE),
+fRejectResonanceDaughters(-1),
fFillOnlyStep0(kFALSE),
fSkipStep6(kFALSE),
fRejectCentralityOutliers(kFALSE),
+fRejectZeroTrackEvents(kFALSE),
fRemoveWeakDecays(kFALSE),
fRemoveDuplicates(kFALSE),
fSkipFastCluster(kFALSE),
fWeightPerEvent(kFALSE),
fCustomBinning(),
+fPtOrder(kTRUE),
+fTriggersFromDetector(0),
+fMCUseUncheckedCentrality(kFALSE),
fFillpT(kFALSE)
{
// Default constructor
// Initialize class with main algorithms, event and track selection.
fAnalyseUE = new AliAnalyseLeadingTrackUE();
- fAnalyseUE->SetParticleSelectionCriteria(fFilterBit, fUseChargeHadrons, fTrackEtaCut, fPtMin);
+ fAnalyseUE->SetParticleSelectionCriteria(fFilterBit, fUseChargeHadrons, fTrackEtaCut, fTrackEtaCutMin, fPtMin);
+ fAnalyseUE->SetDCAXYCut(fDCAXYCut);
+ fAnalyseUE->SetSharedClusterCut(fSharedClusterCut);
+ fAnalyseUE->SetCrossedRowsCut(fCrossedRowsCut);
+ fAnalyseUE->SetFoundFractionCut(fFoundFractionCut);
fAnalyseUE->SetTrackStatus(fTrackStatus);
fAnalyseUE->SetCheckMotherPDG(fCheckMotherPDG);
fAnalyseUE->SetDebug(fDebug);
fHistos->SetPairCuts(fCutConversions, fCutResonances);
fHistosMixed->SetPairCuts(fCutConversions, fCutResonances);
+ fHistos->SetRejectResonanceDaughters(fRejectResonanceDaughters);
+ fHistosMixed->SetRejectResonanceDaughters(fRejectResonanceDaughters);
+
fHistos->SetTrackEtaCut(fTrackEtaCut);
fHistosMixed->SetTrackEtaCut(fTrackEtaCut);
fHistos->SetWeightPerEvent(fWeightPerEvent);
fHistosMixed->SetWeightPerEvent(fWeightPerEvent);
+ fHistos->SetPtOrder(fPtOrder);
+ fHistosMixed->SetPtOrder(fPtOrder);
+
+ fHistos->SetTwoTrackCutMinRadius(fTwoTrackCutMinRadius);
+ fHistosMixed->SetTwoTrackCutMinRadius(fTwoTrackCutMinRadius);
+
if (fEfficiencyCorrectionTriggers)
{
fHistos->SetEfficiencyCorrectionTriggers(fEfficiencyCorrectionTriggers);
// add HelperPID to list
if (fHelperPID)
fListOfHistos->Add(fHelperPID);
+ // add TMap to list
+ if (fMap)
+ fListOfHistos->Add(fMap);
- fListOfHistos->Add(new TH2F("trackletsVsV0Cent", ";L1 clusters;v0 centrality", 100, -0.5, 9999.5, 101, 0, 101));
fListOfHistos->Add(new TH2F("processIDs", ";#Delta#phi;process id", 100, -0.5 * TMath::Pi(), 1.5 * TMath::Pi(), kPNoProcess + 1, -0.5, kPNoProcess + 0.5));
fListOfHistos->Add(new TH1F("eventStat", ";;events", 4, -0.5, 3.5));
fListOfHistos->Add(new TH2F("mixedDist", ";centrality;tracks;events", 101, 0, 101, 200, 0, fMixingTracks * 1.5));
- fListOfHistos->Add(new TH1F("pids", ";pdg;tracks", 2001, -1000.5, 1000.5));
fListOfHistos->Add(new TH2F("referenceMultiplicity", ";centrality;tracks;events", 101, 0, 101, 200, 0, 200));
+ if (fCentralityMethod == "V0A_MANUAL")
+ {
+ fListOfHistos->Add(new TH2F("V0AMult", "V0A multiplicity;V0A multiplicity;V0A multiplicity (scaled)", 1000, -.5, 999.5, 1000, -.5, 999.5));
+ fListOfHistos->Add(new TH2F("V0AMultCorrelation", "V0A multiplicity;V0A multiplicity;SPD tracklets", 1000, -.5, 999.5, 1000, -.5, 999.5));
+ }
+ if (fTriggersFromDetector == 1 || fTriggersFromDetector == 2)
+ fListOfHistos->Add(new TH1F("V0SingleCells", "V0 single cell multiplicity;multiplicity;events", 100, -0.5, 99.5));
PostData(0,fListOfHistos);
//____________________________________________________________________
void AliAnalysisTaskPhiCorrelations::Exec(Option_t */*option*/)
{
+ // exec (per event)
+ fAnalyseUE->NextEvent();
+
// receive ESD pointer if we are not running AOD analysis
if (!fAOD)
{
settingsTree->Branch("fZVertex", &fZVertex,"ZVertex/D");
//settingsTree->Branch("fCentralityMethod", fCentralityMethod.Data(),"CentralityMethod/C");
settingsTree->Branch("fTrackEtaCut", &fTrackEtaCut, "TrackEtaCut/D");
+ settingsTree->Branch("fTrackEtaCutMin", &fTrackEtaCutMin, "TrackEtaCutMin/D");
settingsTree->Branch("fOnlyOneEtaSide", &fOnlyOneEtaSide,"OnlyOneEtaSide/I");
settingsTree->Branch("fPtMin", &fPtMin, "PtMin/D");
settingsTree->Branch("fFilterBit", &fFilterBit,"FilterBit/I");
+ settingsTree->Branch("fSharedClusterCut", &fSharedClusterCut,"SharedClusterCut/D");
+ settingsTree->Branch("fCrossedRowsCut", &fCrossedRowsCut,"CrossedRowsCut/I");
+ settingsTree->Branch("fFoundFractionCut", &fFoundFractionCut,"FoundFractionCut/D");
settingsTree->Branch("fTrackStatus", &fTrackStatus,"TrackStatus/I");
settingsTree->Branch("fSelectBit", &fSelectBit,"EventSelectionBit/I");
settingsTree->Branch("fUseChargeHadrons", &fUseChargeHadrons,"UseChHadrons/O");
settingsTree->Branch("fEtaOrdering", &fEtaOrdering,"EtaOrdering/O");
settingsTree->Branch("fCutConversions", &fCutConversions,"CutConversions/O");
settingsTree->Branch("fCutResonances", &fCutResonances,"CutResonances/O");
+ settingsTree->Branch("fRejectResonanceDaughters", &fRejectResonanceDaughters,"RejectResonanceDaughters/I");
settingsTree->Branch("fFillpT", &fFillpT,"FillpT/O");
settingsTree->Branch("fMixingTracks", &fMixingTracks,"MixingTracks/I");
settingsTree->Branch("fSkipTrigger", &fSkipTrigger,"SkipTrigger/O");
- settingsTree->Branch("fInjectedSignals", &fInjectedSignals,"SkipTrigger/O");
+ settingsTree->Branch("fInjectedSignals", &fInjectedSignals,"InjectedSignals/O");
+ settingsTree->Branch("fRandomizeReactionPlane", &fRandomizeReactionPlane,"RandomizeReactionPlane/O");
settingsTree->Branch("fRejectCentralityOutliers", &fRejectCentralityOutliers,"RejectCentralityOutliers/O");
+ settingsTree->Branch("fRejectZeroTrackEvents", &fRejectZeroTrackEvents,"RejectZeroTrackEvents/O");
settingsTree->Branch("fRemoveWeakDecays", &fRemoveWeakDecays,"RemoveWeakDecays/O");
settingsTree->Branch("fRemoveDuplicates", &fRemoveDuplicates,"RemoveDuplicates/O");
settingsTree->Branch("fSkipFastCluster", &fSkipFastCluster,"SkipFastCluster/O");
settingsTree->Branch("fWeightPerEvent", &fWeightPerEvent,"WeightPerEvent/O");
+ settingsTree->Branch("fPtOrder", &fPtOrder,"PtOrder/O");
+ settingsTree->Branch("fTriggersFromDetector", &fTriggersFromDetector,"TriggersFromDetector/I");
+ settingsTree->Branch("fMCUseUncheckedCentrality", &fMCUseUncheckedCentrality,"MCUseUncheckedCentrality/O");
+ settingsTree->Branch("fTwoTrackEfficiencyCut", &fTwoTrackEfficiencyCut,"TwoTrackEfficiencyCut/D");
+ settingsTree->Branch("fTwoTrackCutMinRadius", &fTwoTrackCutMinRadius,"TwoTrackCutMinRadius/D");
+
//fCustomBinning
settingsTree->Fill();
if (fCentralityMethod.Length() > 0)
{
- AliCentrality *centralityObj = 0;
- if (fAOD)
- centralityObj = fAOD->GetHeader()->GetCentralityP();
- else if (fESD)
- centralityObj = fESD->GetCentrality();
-
- if (centralityObj)
+ if (fCentralityMethod == "MC_b")
{
- centrality = centralityObj->GetCentralityPercentileUnchecked(fCentralityMethod);
- AliInfo(Form("Centrality is %f", centrality));
+ AliGenEventHeader* eventHeader = GetFirstHeader();
+ if (!eventHeader)
+ {
+ // We avoid AliFatal here, because the AOD productions sometimes have events where the MC header is missing
+ // (due to unreadable Kinematics) and we don't want to loose the whole job because of a few events
+ AliError("Event header not found. Skipping this event.");
+ fHistos->FillEvent(0, AliUEHist::kCFStepAnaTopology);
+ return;
+ }
+
+ AliCollisionGeometry* collGeometry = dynamic_cast<AliCollisionGeometry*> (eventHeader);
+ if (!collGeometry)
+ {
+ eventHeader->Dump();
+ AliFatal("Asking for MC_b centrality, but event header has no collision geometry information");
+ }
+
+ centrality = collGeometry->ImpactParameter();
}
else
{
- Printf("WARNING: Centrality object is 0");
- centrality = -1;
- }
+ AliCentrality *centralityObj = 0;
+ if (fAOD)
+ centralityObj = fAOD->GetHeader()->GetCentralityP();
+ else if (fESD)
+ centralityObj = fESD->GetCentrality();
+
+ if (centralityObj)
+ {
+ if (fMCUseUncheckedCentrality)
+ centrality = centralityObj->GetCentralityPercentileUnchecked(fCentralityMethod);
+ else
+ centrality = centralityObj->GetCentralityPercentile(fCentralityMethod);
+ }
+ else
+ {
+ Printf("WARNING: Centrality object is 0");
+ centrality = -1;
+ }
+ }
+
+ AliInfo(Form("Centrality is %f", centrality));
}
// Support for ESD and AOD based analysis
headers = cocktailHeader->GetHeaders()->GetEntries();
eventHeader = dynamic_cast<AliGenEventHeader*> (cocktailHeader->GetHeaders()->First());
+
+ if (fDebug > 4)
+ {
+ for (Int_t i=0; i<cocktailHeader->GetHeaders()->GetEntries(); i++)
+ {
+ AliGenEventHeader* headerTmp = dynamic_cast<AliGenEventHeader*> (cocktailHeader->GetHeaders()->At(i));
+ if (headerTmp)
+ Printf("%d particles in header:", headerTmp->NProduced());
+ cocktailHeader->GetHeaders()->At(i)->Dump();
+ }
+ }
}
else
{
}
skipParticlesAbove = eventHeader->NProduced();
- AliInfo(Form("Injected signals in this event (%d headers). Keeping events of %s. Will skip particles/tracks above %d.", headers, eventHeader->ClassName(), skipParticlesAbove));
+ AliInfo(Form("Injected signals in this event (%d headers). Keeping particles/tracks of %s. Will skip particles/tracks above %d.", headers, eventHeader->ClassName(), skipParticlesAbove));
+ }
+
+ if (fCentralityWeights && !AcceptEventCentralityWeight(centrality))
+ {
+ AliInfo(Form("Rejecting event because of centrality weighting: %f", centrality));
+ fHistos->FillEvent(centrality, AliUEHist::kCFStepAnaTopology);
+ return;
}
// Get MC primaries
TObjArray* tracksCorrelateMC = tracksMC;
if (fParticleSpeciesAssociated != fParticleSpeciesTrigger)
{
- // TODO for MC this uses to PDG of the mother of the particle
- tracksCorrelateMC = fAnalyseUE->GetAcceptedParticles(mc, 0, kTRUE, fParticleSpeciesAssociated, kTRUE);
- CleanUp(tracksCorrelateMC, mc, skipParticlesAbove);
+ tmpList = fAnalyseUE->GetAcceptedParticles(mc, 0, kTRUE, fParticleSpeciesAssociated, kTRUE);
+ CleanUp(tmpList, mc, skipParticlesAbove);
+ tracksCorrelateMC = CloneAndReduceTrackList(tmpList);
+ delete tmpList;
}
- /*
- if (fAOD)
+ if (fRandomizeReactionPlane)
{
- for (Int_t i=0; i<fArrayMC->GetEntriesFast(); i++)
- ((TH1F*) fListOfHistos->FindObject("pids"))->Fill(((AliAODMCParticle*) fArrayMC->At(i))->PdgCode());
+ Double_t centralityDigits = centrality*1000. - (Int_t)(centrality*1000.);
+ Double_t angle = TMath::TwoPi() * centralityDigits;
+ AliInfo(Form("Shifting phi of all tracks by %f (digits %f)", angle, centralityDigits));
+ ShiftTracks(tracksMC, angle);
+ if (tracksCorrelateMC != tracksMC)
+ ShiftTracks(tracksCorrelateMC, angle);
}
- else
- {
- for (Int_t i=0; i<fMcEvent->GetNumberOfTracks(); i++)
- ((TH1F*) fListOfHistos->FindObject("pids"))->Fill(fMcEvent->GetTrack(i)->PdgCode());
- }
- */
if (fFillOnlyStep0)
zVtx = 0;
if (fFillMixed)
{
AliEventPool* pool = fPoolMgr->GetEventPool(centrality, zVtx);
+ if (fFillOnlyStep0)
+ ((TH2F*) fListOfHistos->FindObject("mixedDist"))->Fill(centrality, pool->NTracksInPool());
if (pool->IsReady())
for (Int_t jMix=0; jMix<pool->GetCurrentNEvents(); jMix++)
fHistosMixed->FillCorrelations(centrality, zVtx, AliUEHist::kCFStepAll, tracksMC, pool->GetEvent(jMix), 1.0 / pool->GetCurrentNEvents(), (jMix == 0));
// Get MC primaries that match reconstructed track
// triggers
- TObjArray* tracksRecoMatchedPrim = fAnalyseUE->GetAcceptedParticles(inputEvent, mc, kTRUE, fParticleSpeciesTrigger, kTRUE);
- CleanUp(tracksRecoMatchedPrim, mc, skipParticlesAbove);
+ tmpList = fAnalyseUE->GetAcceptedParticles(inputEvent, mc, kTRUE, fParticleSpeciesTrigger, kTRUE);
+ CleanUp(tmpList, mc, skipParticlesAbove);
+ TObjArray* tracksRecoMatchedPrim = CloneAndReduceTrackList(tmpList);
+ delete tmpList;
+
// associated
TObjArray* tracksCorrelateRecoMatchedPrim = tracksRecoMatchedPrim;
if (fParticleSpeciesAssociated != fParticleSpeciesTrigger)
{
- tracksCorrelateRecoMatchedPrim = fAnalyseUE->GetAcceptedParticles(inputEvent, mc, kTRUE, fParticleSpeciesAssociated, kTRUE);
- CleanUp(tracksCorrelateRecoMatchedPrim, mc, skipParticlesAbove);
+ tmpList = fAnalyseUE->GetAcceptedParticles(inputEvent, mc, kTRUE, fParticleSpeciesAssociated, kTRUE);
+ CleanUp(tmpList, mc, skipParticlesAbove);
+ tracksCorrelateRecoMatchedPrim = CloneAndReduceTrackList(tmpList);
+ delete tmpList;
}
// (RECO-matched (quantities from MC particle) primary particles)
// Get MC primaries + secondaries that match reconstructed track
// triggers
- TObjArray* tracksRecoMatchedAll = fAnalyseUE->GetAcceptedParticles(inputEvent, mc, kFALSE, fParticleSpeciesTrigger, kTRUE);
- CleanUp(tracksRecoMatchedAll, mc, skipParticlesAbove);
+ tmpList = fAnalyseUE->GetAcceptedParticles(inputEvent, mc, kFALSE, fParticleSpeciesTrigger, kTRUE);
+ CleanUp(tmpList, mc, skipParticlesAbove);
+ TObjArray* tracksRecoMatchedAll = CloneAndReduceTrackList(tmpList);
+ delete tmpList;
+
// associated
TObjArray* tracksCorrelateRecoMatchedAll = tracksRecoMatchedAll;
if (fParticleSpeciesAssociated != fParticleSpeciesTrigger)
{
- tracksCorrelateRecoMatchedAll = fAnalyseUE->GetAcceptedParticles(inputEvent, mc, kFALSE, fParticleSpeciesAssociated, kTRUE);
- CleanUp(tracksCorrelateRecoMatchedAll, mc, skipParticlesAbove);
+ tmpList = fAnalyseUE->GetAcceptedParticles(inputEvent, mc, kFALSE, fParticleSpeciesAssociated, kTRUE);
+ CleanUp(tmpList, mc, skipParticlesAbove);
+ tracksCorrelateRecoMatchedAll = CloneAndReduceTrackList(tmpList);
+ delete tmpList;
}
// (RECO-matched (quantities from MC particle) all particles)
// Get RECO tracks
// triggers
- TObjArray* tracks = fAnalyseUE->GetAcceptedParticles(inputEvent, 0, kTRUE, fParticleSpeciesTrigger, kTRUE);
- CleanUp(tracks, mc, skipParticlesAbove);
+ tmpList = fAnalyseUE->GetAcceptedParticles(inputEvent, 0, kTRUE, fParticleSpeciesTrigger, kTRUE);
+ CleanUp(tmpList, mc, skipParticlesAbove);
+ TObjArray* tracks = CloneAndReduceTrackList(tmpList);
+ delete tmpList;
+
// associated
TObjArray* tracksCorrelate = tracks;
if (fParticleSpeciesAssociated != fParticleSpeciesTrigger)
{
- tracksCorrelate = fAnalyseUE->GetAcceptedParticles(inputEvent, 0, kTRUE, fParticleSpeciesAssociated, kTRUE);
- CleanUp(tracksCorrelate, mc, skipParticlesAbove);
+ tmpList = fAnalyseUE->GetAcceptedParticles(inputEvent, 0, kTRUE, fParticleSpeciesAssociated, kTRUE);
+ CleanUp(tmpList, mc, skipParticlesAbove);
+ tracksCorrelate = CloneAndReduceTrackList(tmpList);
+ delete tmpList;
}
// (RECO all tracks)
if (fFillMixed)
{
AliEventPool* pool2 = fPoolMgr->GetEventPool(centrality, zVtx + 100);
+ ((TH2F*) fListOfHistos->FindObject("mixedDist"))->Fill(centrality, pool2->NTracksInPool());
if (pool2->IsReady())
{
for (Int_t jMix=0; jMix<pool2->GetCurrentNEvents(); jMix++)
delete tracksMC;
}
+//____________________________________________________________________
+AliGenEventHeader* AliAnalysisTaskPhiCorrelations::GetFirstHeader()
+{
+ // get first MC header from either ESD/AOD (including cocktail header if available)
+
+ if (fMcEvent)
+ {
+ // ESD
+ AliHeader* header = (AliHeader*) fMcEvent->Header();
+ if (!header)
+ return 0;
+
+ AliGenCocktailEventHeader* cocktailHeader = dynamic_cast<AliGenCocktailEventHeader*> (header->GenEventHeader());
+ if (cocktailHeader)
+ return dynamic_cast<AliGenEventHeader*> (cocktailHeader->GetHeaders()->First());
+
+ return dynamic_cast<AliGenEventHeader*> (header->GenEventHeader());
+ }
+ else
+ {
+ // AOD
+ AliAODMCHeader* header = (AliAODMCHeader*) fAOD->GetList()->FindObject(AliAODMCHeader::StdBranchName());
+ if (!header)
+ return 0;
+
+ return header->GetCocktailHeader(0);
+ }
+}
+
//____________________________________________________________________
void AliAnalysisTaskPhiCorrelations::AnalyseDataMode()
{
// skip fast cluster events here if requested
if (fSkipFastCluster && (fInputHandler->IsEventSelected() & AliVEvent::kFastOnly))
return;
-
+
// Support for ESD and AOD based analysis
AliVEvent* inputEvent = fAOD;
if (!inputEvent)
// Printf("%d %f", tracks->GetEntriesFast(), centrality);
delete tracks;
}
+ else if (fCentralityMethod == "V0A_MANUAL")
+ {
+ // for pp
+
+ //Total multiplicity in the VZERO A detector
+ Float_t MV0A=inputEvent->GetVZEROData()->GetMTotV0A();
+ Float_t MV0AScaled=0.;
+ if (fMap){
+ TParameter<float>* sf=(TParameter<float>*)fMap->GetValue(Form("%d",inputEvent->GetRunNumber()));
+ if(sf)MV0AScaled=MV0A*sf->GetVal();
+ }
+
+ if (MV0AScaled > 0)
+ centrality = MV0AScaled;
+ else
+ centrality = -1;
+ }
else
{
if (fAOD)
// Trigger selection ************************************************
if (!fSkipTrigger && !fAnalyseUE->TriggerSelection(fInputHandler)) return;
-
+
// Fill the "event-counting-container", it is needed to get the number of events remaining after each event-selection cut
fHistos->FillEvent(centrality, AliUEHist::kCFStepTriggered);
+ // Pileup selection ************************************************
+ if (fAnalysisUtils && fAnalysisUtils->IsPileUpEvent(inputEvent))
+ {
+ // count the removed events
+ fHistos->FillEvent(centrality, AliUEHist::kCFStepAnaTopology);
+
+ return;
+ }
+
// Vertex selection *************************************************
if(!fAnalyseUE->VertexSelection(inputEvent, fnTracksVertex, fZVertex)) return;
// Fill the "event-counting-container", it is needed to get the number of events remaining after each event-selection cut
fHistos->FillEvent(centrality, AliUEHist::kCFStepVertex);
+ // fill V0 control histograms
+ if (fCentralityMethod == "V0A_MANUAL")
+ {
+ ((TH2F*) fListOfHistos->FindObject("V0AMult"))->Fill(inputEvent->GetVZEROData()->GetMTotV0A(), centrality);
+ if (fAOD)
+ ((TH2F*) fListOfHistos->FindObject("V0AMultCorrelation"))->Fill(inputEvent->GetVZEROData()->GetMTotV0A(), fAOD->GetTracklets()->GetNumberOfTracklets());
+ }
+
// optimization
- if (centrality < 0 && !fCompareCentralities)
+ if (centrality < 0)
return;
- TObjArray* tracks = fAnalyseUE->GetAcceptedParticles(inputEvent, 0, kTRUE, fParticleSpeciesTrigger, kTRUE);
+ TObjArray* tracks = 0;
+
+ if (fTriggersFromDetector == 0)
+ tracks = fAnalyseUE->GetAcceptedParticles(inputEvent, 0, kTRUE, fParticleSpeciesTrigger, kTRUE);
+ else if (fTriggersFromDetector == 1 || fTriggersFromDetector == 2)
+ {
+ tracks = new TObjArray;
+ tracks->SetOwner(kTRUE);
+
+ AliVVZERO* vZero = inputEvent->GetVZEROData();
+
+ const Int_t vZeroStart = (fTriggersFromDetector == 1) ? 32 : 0;
+
+ TH1F* singleCells = (TH1F*) fListOfHistos->FindObject("V0SingleCells");
+ for (Int_t i=vZeroStart; i<vZeroStart+32; i++)
+ {
+ Float_t weight = vZero->GetMultiplicity(i);
+ singleCells->Fill(weight);
+
+ // rough estimate of multiplicity
+ for (Int_t j=0; j<TMath::Nint(weight); j++)
+ {
+ AliDPhiBasicParticle* particle = new AliDPhiBasicParticle((AliVVZERO::GetVZEROEtaMax(i) + AliVVZERO::GetVZEROEtaMin(i)) / 2, AliVVZERO::GetVZEROAvgPhi(i), 1.1, 0); // fit pT = 1.1 and charge = 0
+ particle->SetUniqueID(-1); // not needed here
+
+ tracks->Add(particle);
+ }
+ }
+ }
+ else
+ AliFatal(Form("Invalid setting for fTriggersFromDetector: %d", fTriggersFromDetector));
+
//Printf("Accepted %d tracks", tracks->GetEntries());
// check for outlier in centrality vs number of tracks (rough constants extracted from correlation histgram)
return;
}
+ if (fRejectZeroTrackEvents && tracks->GetEntriesFast() == 0)
+ {
+ AliInfo(Form("Rejecting event because it has no tracks: %f %d", centrality, tracks->GetEntriesFast()));
+ fHistos->FillEvent(centrality, AliUEHist::kCFStepAnaTopology);
+ delete tracks;
+ return;
+ }
+
+ if (fCentralityWeights && !AcceptEventCentralityWeight(centrality))
+ {
+ AliInfo(Form("Rejecting event because of centrality weighting: %f", centrality));
+ fHistos->FillEvent(centrality, AliUEHist::kCFStepAnaTopology);
+ delete tracks;
+ return;
+ }
+
// correlate particles with...
TObjArray* tracksCorrelate = 0;
- if (fParticleSpeciesAssociated != fParticleSpeciesTrigger)
+ if (fParticleSpeciesAssociated != fParticleSpeciesTrigger || fTriggersFromDetector > 0)
tracksCorrelate = fAnalyseUE->GetAcceptedParticles(inputEvent, 0, kTRUE, fParticleSpeciesAssociated, kTRUE);
// reference multiplicity
if (fFillpT)
weight = -1;
- // fill two different centralities (syst study)
- // the zvtx axis is used to distinguish the results of both centralities: configured centrality in zvtx = 0, SPD in zvtx = 2
- if (fCompareCentralities)
- zVtx = 0;
-
// Fill containers at STEP 6 (reconstructed)
if (centrality >= 0)
{
fHistos->FillCorrelations(centrality, zVtx, AliUEHist::kCFStepBiasStudy, tracks, tracksCorrelate, weight, kTRUE, kTRUE, bSign, fTwoTrackEfficiencyCut, kTRUE);
}
- // fill second time with SPD centrality
- if (fCompareCentralities && centralityObj)
- {
- centrality = centralityObj->GetCentralityPercentile("CL1");
- if (centrality >= 0 && !fSkipStep6)
- fHistos->FillCorrelations(centrality, 2, AliUEHist::kCFStepReconstructed, tracks, tracksCorrelate, weight, kFALSE, kFALSE, 0, 0.02, kTRUE);
- }
-
// create a list of reduced objects. This speeds up processing and reduces memory consumption for the event pool
TObjArray* tracksClone = CloneAndReduceTrackList(tracks);
delete tracks;
for (Int_t i=0; i<tracks->GetEntriesFast(); i++)
{
- AliVParticle* particle = (AliVParticle*) tracks->At(i);
- tracksClone->Add(new AliDPhiBasicParticle(particle->Eta(), particle->Phi(), particle->Pt(), particle->Charge()));
+ AliVParticle* particle = (AliVParticle*) tracks->UncheckedAt(i);
+ AliDPhiBasicParticle* copy = new AliDPhiBasicParticle(particle->Eta(), particle->Phi(), particle->Pt(), particle->Charge());
+ copy->SetUniqueID(particle->GetUniqueID());
+ tracksClone->Add(copy);
}
return tracksClone;
fInputHandler = (AliInputEventHandler*)
((AliAnalysisManager::GetAnalysisManager())->GetInputEventHandler());
// MC handler
- fMcHandler = dynamic_cast<AliMCEventHandler*> (AliAnalysisManager::GetAnalysisManager()->GetMCtruthEventHandler());
+ fMcHandler = dynamic_cast<AliInputEventHandler*> (AliAnalysisManager::GetAnalysisManager()->GetMCtruthEventHandler());
}
//____________________________________________________________________
if (before > tracks->GetEntriesFast())
AliInfo(Form("Reduced from %d to %d", before, tracks->GetEntriesFast()));
}
+
+//____________________________________________________________________
+Bool_t AliAnalysisTaskPhiCorrelations::AcceptEventCentralityWeight(Double_t centrality)
+{
+ // rejects "randomly" events such that the centrality gets flat
+ // uses fCentralityWeights histogram
+
+ // TODO code taken and adapted from AliRDHFCuts; waiting for general class AliCentralityFlattening
+
+ Double_t weight = fCentralityWeights->GetBinContent(fCentralityWeights->FindBin(centrality));
+ Double_t centralityDigits = centrality*100. - (Int_t)(centrality*100.);
+
+ Bool_t result = kFALSE;
+ if (centralityDigits < weight)
+ result = kTRUE;
+
+ AliInfo(Form("Centrality: %f; Digits: %f; Weight: %f; Result: %d", centrality, centralityDigits, weight, result));
+
+ return result;
+}
+
+//____________________________________________________________________
+void AliAnalysisTaskPhiCorrelations::ShiftTracks(TObjArray* tracks, Double_t angle)
+{
+ // shifts the phi angle of all tracks by angle
+ // 0 <= angle <= 2pi
+
+ for (Int_t i=0; i<tracks->GetEntriesFast(); ++i)
+ {
+ AliDPhiBasicParticle* part = (AliDPhiBasicParticle*) tracks->At(i);
+ Double_t newAngle = part->Phi() + angle;
+ if (newAngle >= TMath::TwoPi())
+ newAngle -= TMath::TwoPi();
+
+ part->SetPhi(newAngle);
+ }
+}
+
\ No newline at end of file
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff