X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;f=ANALYSIS%2FAliAnalysisTaskSE.cxx;h=ef6bd2e4d5906629db359d3063a39cd80c8b44ab;hb=3aa6a1365c1972ad32a074f2799a140f355779d5;hp=8d90cc853afdb5a88a76a44c0a892d4a37a92720;hpb=8f6e7d1078fc857b9751f03ce74b3a45173dc2fc;p=u%2Fmrichter%2FAliRoot.git diff --git a/ANALYSIS/AliAnalysisTaskSE.cxx b/ANALYSIS/AliAnalysisTaskSE.cxx index 8d90cc853af..ef6bd2e4d59 100644 --- a/ANALYSIS/AliAnalysisTaskSE.cxx +++ b/ANALYSIS/AliAnalysisTaskSE.cxx @@ -26,6 +26,7 @@ #include "AliAnalysisManager.h" #include "AliAnalysisCuts.h" #include "AliAnalysisDataSlot.h" +#include "AliAnalysisDataContainer.h" #include "AliESDEvent.h" #include "AliESDfriend.h" @@ -34,12 +35,14 @@ #include "AliAODHeader.h" #include "AliAODTracklets.h" #include "AliAODCaloCells.h" +#include "AliAODCaloTrigger.h" #include "AliAODMCParticle.h" #include "AliVEvent.h" #include "AliAODHandler.h" #include "AliAODInputHandler.h" #include "AliMCEventHandler.h" #include "AliInputEventHandler.h" +#include "AliMultiInputEventHandler.h" #include "AliESDInputHandler.h" #include "AliMCEvent.h" #include "AliStack.h" @@ -62,6 +65,8 @@ TClonesArray* AliAnalysisTaskSE::fgAODMCParticles = NULL; AliAODTracklets* AliAnalysisTaskSE::fgAODTracklets = NULL; AliAODCaloCells* AliAnalysisTaskSE::fgAODEmcalCells = NULL; AliAODCaloCells* AliAnalysisTaskSE::fgAODPhosCells = NULL; +AliAODCaloTrigger* AliAnalysisTaskSE::fgAODEMCALTrigger = NULL; +AliAODCaloTrigger* AliAnalysisTaskSE::fgAODPHOSTrigger = NULL; TClonesArray* AliAnalysisTaskSE::fgAODDimuons = NULL; AliAnalysisTaskSE::AliAnalysisTaskSE(): @@ -76,7 +81,9 @@ AliAnalysisTaskSE::AliAnalysisTaskSE(): fTreeA(0x0), fCurrentRunNumber(-1), fHistosQA(0x0), - fSelectCollisions(0) + fOfflineTriggerMask(0), + fMultiInputHandler(0), + fMCEventHandler(0) { // Default constructor } @@ -93,7 +100,9 @@ AliAnalysisTaskSE::AliAnalysisTaskSE(const char* name): fTreeA(0x0), fCurrentRunNumber(-1), fHistosQA(0x0), - fSelectCollisions(0) + fOfflineTriggerMask(0), + fMultiInputHandler(0), + fMCEventHandler(0) { // Default constructor DefineInput (0, TChain::Class()); @@ -112,7 +121,9 @@ AliAnalysisTaskSE::AliAnalysisTaskSE(const AliAnalysisTaskSE& obj): fTreeA(0x0), fCurrentRunNumber(-1), fHistosQA(0x0), - fSelectCollisions(0) + fOfflineTriggerMask(0), + fMultiInputHandler(obj.fMultiInputHandler), + fMCEventHandler(obj.fMCEventHandler) { // Copy constructor fDebug = obj.fDebug; @@ -132,6 +143,9 @@ AliAnalysisTaskSE::AliAnalysisTaskSE(const AliAnalysisTaskSE& obj): AliAnalysisTaskSE& AliAnalysisTaskSE::operator=(const AliAnalysisTaskSE& other) { // Assignment + if(&other == this) return *this; + AliAnalysisTask::operator=(other); + AliAnalysisTask::operator=(other); fDebug = other.fDebug; fEntry = other.fEntry; @@ -143,28 +157,20 @@ AliAnalysisTaskSE& AliAnalysisTaskSE::operator=(const AliAnalysisTaskSE& other) fTreeA = other.fTreeA; fCurrentRunNumber = other.fCurrentRunNumber; fHistosQA = other.fHistosQA; - fSelectCollisions = other.fSelectCollisions; + fOfflineTriggerMask = other.fOfflineTriggerMask; + fMultiInputHandler = other.fMultiInputHandler; + fMCEventHandler = other.fMCEventHandler; return *this; } - +//______________________________________________________________________________ void AliAnalysisTaskSE::ConnectInputData(Option_t* /*option*/) { // Connect the input data if (fDebug > 1) printf("AnalysisTaskSE::ConnectInputData() \n"); -// -// ESD -// - fInputHandler = (AliInputEventHandler*) - ((AliAnalysisManager::GetAnalysisManager())->GetInputEventHandler()); -// -// Monte Carlo -// - AliMCEventHandler* mcH = 0; - mcH = (AliMCEventHandler*) ((AliAnalysisManager::GetAnalysisManager())->GetMCtruthEventHandler()); - if (mcH) { - fMCEvent = mcH->MCEvent(); - } + + // Connect input handlers (multi input handler is handled) + ConnectMultiHandler(); if (fInputHandler) { if ((fInputHandler->GetTree())->GetBranch("ESDfriend.")) @@ -177,6 +183,8 @@ void AliAnalysisTaskSE::ConnectInputData(Option_t* /*option*/) AliError("No Input Event Handler connected") ; return ; } + // Disconnect multi handler + DisconnectMultiHandler(); } void AliAnalysisTaskSE::CreateOutputObjects() @@ -186,7 +194,7 @@ void AliAnalysisTaskSE::CreateOutputObjects() // Default AOD if (fDebug > 1) printf("AnalysisTaskSE::CreateOutPutData() \n"); - AliAODHandler* handler = (AliAODHandler*) + AliAODHandler* handler = dynamic_cast ((AliAnalysisManager::GetAnalysisManager())->GetOutputEventHandler()); Bool_t merging = kFALSE; @@ -194,15 +202,13 @@ void AliAnalysisTaskSE::CreateOutputObjects() if (aodIH) { if (aodIH->GetMergeEvents()) merging = kTRUE; } - // Check if AOD replication has been required - if (handler) { fOutputAOD = handler->GetAOD(); fTreeA = handler->GetTree(); if (fOutputAOD && !(handler->IsStandard())) { - if ((handler->NeedsHeaderReplication()) && !(fgAODHeader)) + if ((handler->NeedsHeaderReplication() || merging) && !(fgAODHeader)) { if (fDebug > 1) AliInfo("Replicating header"); fgAODHeader = new AliAODHeader; @@ -263,13 +269,23 @@ void AliAnalysisTaskSE::CreateOutputObjects() fgAODCaloClusters->SetName("caloClusters"); handler->AddBranch("TClonesArray", &fgAODCaloClusters); - fgAODEmcalCells = new AliAODCaloCells(); - fgAODEmcalCells->SetName("emcalCells"); + fgAODEmcalCells = new AliAODCaloCells("emcalCells","emcalCells",AliVCaloCells::kEMCALCell); handler->AddBranch("AliAODCaloCells", &fgAODEmcalCells); - fgAODPhosCells = new AliAODCaloCells(); - fgAODPhosCells->SetName("phosCells"); + fgAODPhosCells = new AliAODCaloCells("phosCells","phosCells",AliVCaloCells::kPHOSCell); handler->AddBranch("AliAODCaloCells", &fgAODPhosCells); + } + if ((handler->NeedsCaloTriggerBranchReplication() || merging) && !(fgAODEMCALTrigger)) + { + if (fDebug > 1) AliInfo("Replicating EMCAL Calo Trigger branches\n"); + fgAODEMCALTrigger = new AliAODCaloTrigger("emcalTrigger","emcalTrigger"); + handler->AddBranch("AliAODCaloTrigger", &fgAODEMCALTrigger); + } + if ((handler->NeedsCaloTriggerBranchReplication() || merging) && !(fgAODPHOSTrigger)) + { + if (fDebug > 1) AliInfo("Replicating PHOS Calo Trigger branches\n"); + fgAODPHOSTrigger = new AliAODCaloTrigger("phosTrigger","phosTrigger"); + handler->AddBranch("AliAODCaloTrigger", &fgAODPHOSTrigger); } if ((handler->NeedsMCParticlesBranchReplication() || merging) && !(fgAODMCParticles)) { @@ -289,38 +305,48 @@ void AliAnalysisTaskSE::CreateOutputObjects() // cache the pointerd in the AODEvent fOutputAOD->GetStdContent(); } - } else { - AliWarning("No AOD Event Handler connected.") ; } + ConnectMultiHandler(); UserCreateOutputObjects(); + DisconnectMultiHandler(); } void AliAnalysisTaskSE::Exec(Option_t* option) { // // Exec analysis of one event - fDebug = 10; - printf("Task is active %5d\n", IsActive()); + + ConnectMultiHandler(); + + if ( fDebug >= 10) + printf("Task is active %5d\n", IsActive()); if (fDebug > 1) AliInfo("AliAnalysisTaskSE::Exec() \n"); // - AliAODHandler* handler = (AliAODHandler*) + AliAODHandler* handler = dynamic_cast ((AliAnalysisManager::GetAnalysisManager())->GetOutputEventHandler()); + AliAODInputHandler* aodH = dynamic_cast(fInputHandler); // -// Was event selected ? - Bool_t isSelected = kTRUE; - if( fInputHandler && fInputHandler->GetEventSelection() && fSelectCollisions) { - isSelected = fInputHandler->IsEventSelected(); +// Was event selected ? If no event selection mechanism, the event SHOULD be selected (AG) + UInt_t isSelected = AliVEvent::kAny; + if( fInputHandler && (fInputHandler->GetEventSelection() || aodH)) { + // Get the actual offline trigger mask for the event and AND it with the + // requested mask. If no mask requested select by default the event. + if (fOfflineTriggerMask) + isSelected = fOfflineTriggerMask & fInputHandler->IsEventSelected(); } +// Functionality below moved in the filter tasks (AG) +// if (handler) handler->SetFillAOD(isSelected); - if (handler) handler->SetFillAOD(isSelected); - - if( fInputHandler ) fEntry = fInputHandler->GetReadEntry(); - + if( fInputHandler ) { + fEntry = fInputHandler->GetReadEntry(); + fESDfriend = ((AliESDInputHandler*)fInputHandler)->GetESDfriend(); + } + // Notify the change of run number - if (InputEvent()->GetRunNumber() != fCurrentRunNumber) { + if (InputEvent() && (InputEvent()->GetRunNumber() != fCurrentRunNumber)) { fCurrentRunNumber = InputEvent()->GetRunNumber(); NotifyRun(); } @@ -336,67 +362,90 @@ void AliAnalysisTaskSE::Exec(Option_t* option) if (handler && aodH) { fMCEvent = aodH->MCEvent(); Bool_t merging = aodH->GetMergeEvents(); - - if (!(handler->IsStandard()) && !(handler->AODIsReplicated())) { - if ((handler->NeedsHeaderReplication()) && (fgAODHeader)) + + // Do not analyze merged events if last embedded file has less events than normal event, + // skip analysis after last embeded event + if(merging){ + if(aodH->GetReadEntry() + aodH->GetMergeOffset() >= aodH->GetTreeToMerge()->GetEntriesFast()){ + //printf("Skip Entry %lld, Offset %d, Tree Entries %d\n",aodH->GetReadEntry(),aodH->GetMergeOffset(), aodH->GetTreeToMerge()->GetEntries()); + + // Do I need to add the lines before the return? + // Added protection in case the derived task is not an AOD producer. + AliAnalysisDataSlot *out0 = GetOutputSlot(0); + if (out0 && out0->IsConnected()) PostData(0, fTreeA); + + DisconnectMultiHandler(); + + return; + } + //else printf("MERGE Entry %lld, Offset %d, Tree Entries %d\n",aodH->GetReadEntry(),aodH->GetMergeOffset(), aodH->GetTreeToMerge()->GetEntries()); + } + + AliAODEvent* aod = dynamic_cast(InputEvent()); + + if (aod && !(handler->IsStandard()) && !(handler->AODIsReplicated())) { + if ((handler->NeedsHeaderReplication() || merging) && (fgAODHeader)) { // copy the contents by assigment - *fgAODHeader = *(dynamic_cast(InputEvent()->GetHeader())); + *fgAODHeader = *(aod->GetHeader()); } - if ((handler->NeedsTracksBranchReplication() || merging) && (fgAODTracks)) + if ((handler->NeedsTracksBranchReplication() || (merging && aodH->GetMergeTracks())) && (fgAODTracks)) { - TClonesArray* tracks = (dynamic_cast(InputEvent()))->GetTracks(); + TClonesArray* tracks = aod->GetTracks(); new (fgAODTracks) TClonesArray(*tracks); } if ((handler->NeedsVerticesBranchReplication() || merging) && (fgAODVertices)) { - TClonesArray* vertices = (dynamic_cast(InputEvent()))->GetVertices(); + TClonesArray* vertices = aod->GetVertices(); new (fgAODVertices) TClonesArray(*vertices); } if ((handler->NeedsV0sBranchReplication()) && (fgAODV0s)) { - TClonesArray* v0s = (dynamic_cast(InputEvent()))->GetV0s(); + TClonesArray* v0s = aod->GetV0s(); new (fgAODV0s) TClonesArray(*v0s); } if ((handler->NeedsTrackletsBranchReplication()) && (fgAODTracklets)) { - *fgAODTracklets = *(dynamic_cast(InputEvent()))->GetTracklets(); + *fgAODTracklets = *aod->GetTracklets(); } if ((handler->NeedsPMDClustersBranchReplication()) && (fgAODPMDClusters)) { - TClonesArray* pmdClusters = (dynamic_cast(InputEvent()))->GetPmdClusters(); + TClonesArray* pmdClusters = aod->GetPmdClusters(); new (fgAODPMDClusters) TClonesArray(*pmdClusters); } - if ((handler->NeedsJetsBranchReplication() || merging) && (fgAODJets)) + if ((handler->NeedsJetsBranchReplication() || (merging &&aodH->GetMergeTracks())) && (fgAODJets)) { - TClonesArray* jets = (dynamic_cast(InputEvent()))->GetJets(); + TClonesArray* jets = aod->GetJets(); new (fgAODJets) TClonesArray(*jets); } if ((handler->NeedsFMDClustersBranchReplication()) && (fgAODFMDClusters)) { - TClonesArray* fmdClusters = (dynamic_cast(InputEvent()))->GetFmdClusters(); + TClonesArray* fmdClusters = aod->GetFmdClusters(); new (fgAODFMDClusters) TClonesArray(*fmdClusters); } - if ((handler->NeedsCaloClustersBranchReplication() || merging) && (fgAODCaloClusters)) + if ((handler->NeedsCaloClustersBranchReplication() || + (merging && (aodH->GetMergeEMCALClusters() || aodH->GetMergePHOSClusters()))) + && (fgAODCaloClusters)) { - TClonesArray* caloClusters = (dynamic_cast(InputEvent()))->GetCaloClusters(); + TClonesArray* caloClusters = aod->GetCaloClusters(); new (fgAODCaloClusters) TClonesArray(*caloClusters); } if ((handler->NeedsMCParticlesBranchReplication() || merging) && (fgAODMCParticles)) { - TClonesArray* mcParticles = (TClonesArray*) ((dynamic_cast(InputEvent()))->FindListObject("mcparticles")); - new (fgAODMCParticles) TClonesArray(*mcParticles); + TClonesArray* mcParticles = (TClonesArray*) (aod->FindListObject("mcparticles")); + if( mcParticles ) + new (fgAODMCParticles) TClonesArray(*mcParticles); } - if ((handler->NeedsDimuonsBranchReplication() || merging) && (fgAODDimuons)) + if ((handler->NeedsDimuonsBranchReplication() || (merging && aodH->GetMergeTracks())) && (fgAODDimuons)) { fgAODDimuons->Clear(); TClonesArray& dimuons = *fgAODDimuons; TClonesArray& tracksnew = *fgAODTracks; - Int_t nMuonTrack[10]; - for(Int_t imuon = 0; imuon < 10; imuon++) nMuonTrack[imuon] = 0; + Int_t nMuonTrack[100]; + for(Int_t imuon = 0; imuon < 100; imuon++) nMuonTrack[imuon] = 0; Int_t nMuons=0; for(Int_t ii=0; ii < fgAODTracks->GetEntries(); ii++){ AliAODTrack *track = (AliAODTrack*) fgAODTracks->At(ii); @@ -411,8 +460,6 @@ void AliAnalysisTaskSE::Exec(Option_t* option) Int_t index0 = nMuonTrack[i]; for(Int_t j = i+1; j < nMuons; j++){ Int_t index1 = nMuonTrack[j]; - AliAODTrack *t0 = (AliAODTrack*) fgAODTracks->At(index0); - AliAODTrack *t1 = (AliAODTrack*) fgAODTracks->At(index1); tracksnew.At(index0)->ResetBit(kIsReferenced); tracksnew.At(index0)->SetUniqueID(0); tracksnew.At(index1)->ResetBit(kIsReferenced); @@ -422,78 +469,201 @@ void AliAnalysisTaskSE::Exec(Option_t* option) } } } - // Additional merging if needed if (merging) { + Int_t nc; + // mcParticles TClonesArray* mcparticles = (TClonesArray*) ((aodH->GetEventToMerge())->FindListObject("mcparticles")); - Int_t npart = mcparticles->GetEntries(); - Int_t nc = fgAODMCParticles->GetEntries(); - Int_t nc0 = nc; - - for (Int_t i = 0; i < npart; i++) { + if( mcparticles ){ + Int_t npart = mcparticles->GetEntries(); + nc = fgAODMCParticles->GetEntries(); + + for (Int_t i = 0; i < npart; i++) { AliAODMCParticle* particle = (AliAODMCParticle*) mcparticles->At(i); new((*fgAODMCParticles)[nc++]) AliAODMCParticle(*particle); + } } - + // tracks TClonesArray* tracks = aodH->GetEventToMerge()->GetTracks(); - Int_t ntr = tracks->GetEntries(); - nc = fgAODTracks->GetEntries(); - for (Int_t i = 0; i < ntr; i++) { + if(tracks && aodH->GetMergeTracks()){ + Int_t ntr = tracks->GetEntries(); + nc = fgAODTracks->GetEntries(); + for (Int_t i = 0; i < ntr; i++) { AliAODTrack* track = (AliAODTrack*) tracks->At(i); AliAODTrack* newtrack = new((*fgAODTracks)[nc++]) AliAODTrack(*track); - newtrack->SetLabel(newtrack->GetLabel() + nc0); - } - - for (Int_t i = 0; i < nc; i++) - { - AliAODTrack* track = (AliAODTrack*) fgAODTracks->At(i); - track->ResetBit(kIsReferenced); - track->SetUniqueID(0); + newtrack->SetLabel(newtrack->GetLabel() + fgAODMCParticles->GetEntries()); + } + + for (Int_t i = 0; i < nc; i++) + { + AliAODTrack* track = (AliAODTrack*) fgAODTracks->At(i); + track->ResetBit(kIsReferenced); + track->SetUniqueID(0); + } } - // clusters TClonesArray* clusters = aodH->GetEventToMerge()->GetCaloClusters(); - Int_t ncl = clusters->GetEntries(); - nc = fgAODCaloClusters->GetEntries(); - for (Int_t i = 0; i < ncl; i++) { + if( clusters && (aodH->GetMergeEMCALClusters() || aodH->GetMergePHOSClusters())) { + Int_t ncl = clusters->GetEntries(); + nc = fgAODCaloClusters->GetEntries(); + for (Int_t i = 0; i < ncl; i++) { AliAODCaloCluster* cluster = (AliAODCaloCluster*) clusters->At(i); + if(cluster->IsEMCAL() && !aodH->GetMergeEMCALClusters() ) continue; + if(cluster->IsPHOS() && !aodH->GetMergePHOSClusters() ) continue; new((*fgAODCaloClusters)[nc++]) AliAODCaloCluster(*cluster); + } } - // cells - AliAODCaloCells* cellsA = aodH->GetEventToMerge()->GetEMCALCells(); - Int_t ncells = cellsA->GetNumberOfCells(); - nc = fgAODEmcalCells->GetNumberOfCells(); - - for (Int_t i = 0; i < ncells; i++) { - Int_t cn = cellsA->GetCellNumber(i); - Int_t pos = fgAODEmcalCells->GetCellPosition(cn); - if (pos >= 0) { - Double_t amp = cellsA->GetAmplitude(i) + fgAODEmcalCells->GetAmplitude(pos); - fgAODEmcalCells->SetCell(pos, cn, amp); - } else { - Double_t amp = cellsA->GetAmplitude(i); - fgAODEmcalCells->SetCell(nc++, cn, amp); + + // EMCAL cells + //*fgAODEmcalCells = *(aod->GetEMCALCells()); // This will be valid after 10.Mar.2011. + if(aodH->GetMergeEMCALCells()) { + AliAODCaloCells* copycells = aod->GetEMCALCells(); + fgAODEmcalCells->CreateContainer(copycells->GetNumberOfCells()); + nc = copycells->GetNumberOfCells(); + while( nc-- ){ fgAODEmcalCells->SetCell(nc,copycells->GetCellNumber(nc),copycells->GetAmplitude(nc)); } + + AliAODCaloCells* cellsA = aodH->GetEventToMerge()->GetEMCALCells(); + if( cellsA ){ + Int_t ncells = cellsA->GetNumberOfCells(); + nc = fgAODEmcalCells->GetNumberOfCells(); + for (Int_t i = 0; i < ncells; i++) { + Int_t cn = cellsA->GetCellNumber(i); + Int_t pos = fgAODEmcalCells->GetCellPosition(cn); + if (pos >= 0) { + Double_t amp = cellsA->GetAmplitude(i) + fgAODEmcalCells->GetAmplitude(pos); + fgAODEmcalCells->SetCell(pos, cn, amp); + } else { + AliAODCaloCells* copycells1 = new AliAODCaloCells(*fgAODEmcalCells); + fgAODEmcalCells->CreateContainer(nc+1); + Int_t nn = copycells1->GetNumberOfCells(); + while( nn-- ){ fgAODEmcalCells->SetCell(nn,copycells1->GetCellNumber(nn),copycells1->GetAmplitude(nn)); } + fgAODEmcalCells->SetCell(nc++,cn,cellsA->GetAmplitude(i)); + delete copycells1; + } + } fgAODEmcalCells->Sort(); } - } + } // merge emcal cells + + + // PHOS cells + //*fgAODPhosCells = *(aod->GetPHOSCells()); // This will be valid after 10.Mar.2011. + if(aodH->GetMergePHOSCells()) { + AliAODCaloCells* copycells = aod->GetPHOSCells(); + fgAODPhosCells->CreateContainer(copycells->GetNumberOfCells()); + nc = copycells->GetNumberOfCells(); + while( nc-- ){ fgAODPhosCells->SetCell(nc,copycells->GetCellNumber(nc),copycells->GetAmplitude(nc)); } + AliAODCaloCells* cellsP = aodH->GetEventToMerge()->GetPHOSCells(); + if( cellsP ){ + Int_t ncellsP = cellsP->GetNumberOfCells(); + nc = fgAODPhosCells->GetNumberOfCells(); + + for (Int_t i = 0; i < ncellsP; i++) { + Int_t cn = cellsP->GetCellNumber(i); + Int_t pos = fgAODPhosCells->GetCellPosition(cn); + if (pos >= 0) { + Double_t amp = cellsP->GetAmplitude(i) + fgAODPhosCells->GetAmplitude(pos); + fgAODPhosCells->SetCell(pos, cn, amp); + } else { + AliAODCaloCells* copycells1 = new AliAODCaloCells(*fgAODPhosCells); + fgAODPhosCells->CreateContainer(nc+1); + Int_t nn = copycells1->GetNumberOfCells(); + while( nn-- ){ fgAODPhosCells->SetCell(nn,copycells1->GetCellNumber(nn),copycells1->GetAmplitude(nn)); } + fgAODPhosCells->SetCell(nc++,cn,cellsP->GetAmplitude(i)); + delete copycells1; + } + } + fgAODPhosCells->Sort(); + } + } // Merge PHOS Cells + + if (aodH->GetMergeEMCALTrigger()) + { + Int_t EMCALts[48][64], px, py, ts; + Float_t EMCALfo[48][64], am; + + for (Int_t i = 0; i < 48; i++) for (Int_t j = 0; j < 64; j++) + { + EMCALts[i][j] = 0; + EMCALfo[i][j] = 0.; + } + + AliAODCaloTrigger& trg0 = *(aod->GetCaloTrigger("EMCAL")); + + trg0.Reset(); + while (trg0.Next()) + { + trg0.GetPosition(px, py); + + if (px > -1 && py > -1) + { + trg0.GetL1TimeSum(ts); + if (ts > -1) EMCALts[px][py] += ts; + + trg0.GetAmplitude(am); + if (am > -1) EMCALfo[px][py] += am; + } + } + + AliAODCaloTrigger& trg1 = *((aodH->GetEventToMerge())->GetCaloTrigger("EMCAL")); + + trg1.Reset(); + while (trg1.Next()) + { + trg1.GetPosition(px, py); + + if (px > -1 && py > -1) + { + trg1.GetL1TimeSum(ts); + if (ts > -1) EMCALts[px][py] += ts; + + trg1.GetAmplitude(am); + if (am > -1) EMCALfo[px][py] += am; + } + } + + int nEntries = 0; + for (Int_t i = 0; i < 48; i++) + for (Int_t j = 0; j < 64; j++) + if (EMCALts[i][j] || EMCALfo[i][j]) nEntries++; + fgAODEMCALTrigger->Allocate(nEntries); + Int_t L0times[10]; for (int i = 0; i < 10; i++) L0times[i] = -1; + for (Int_t i = 0; i < 48; i++) + for (Int_t j = 0; j < 64; j++) + if (EMCALts[i][j] || EMCALfo[i][j]) + fgAODEMCALTrigger->Add(i, j, EMCALfo[i][j], -1., L0times, 0, EMCALts[i][j], 0); + } + + if (aodH->GetMergePHOSTrigger()) + { + // To be implemented by PHOS + } } // merging handler->SetAODIsReplicated(); } } + // Call the user analysis - if (!fSelectCollisions || isSelected) - UserExec(option); + if (!fMCEventHandler) { + if (isSelected) + UserExec(option); + } else { + if (isSelected && (fMCEventHandler->InitOk())) + UserExec(option); + } // Added protection in case the derived task is not an AOD producer. AliAnalysisDataSlot *out0 = GetOutputSlot(0); if (out0 && out0->IsConnected()) PostData(0, fTreeA); + + DisconnectMultiHandler(); } const char* AliAnalysisTaskSE::CurrentFileName() @@ -509,7 +679,7 @@ const char* AliAnalysisTaskSE::CurrentFileName() void AliAnalysisTaskSE::AddAODBranch(const char* cname, void* addobj, const char *fname) { // Add a new branch to the aod tree - AliAODHandler* handler = (AliAODHandler*) + AliAODHandler* handler = dynamic_cast ((AliAnalysisManager::GetAnalysisManager())->GetOutputEventHandler()); if (handler) { handler->AddBranch(cname, addobj, fname); @@ -520,7 +690,7 @@ Bool_t AliAnalysisTaskSE::IsStandardAOD() const { // Check if the output AOD handler is configured for standard or delta AOD. // Users should first check that AODEvent() returns non-null. - AliAODHandler* handler = (AliAODHandler*) + AliAODHandler* handler = dynamic_cast ((AliAnalysisManager::GetAnalysisManager())->GetOutputEventHandler()); if (!handler) { Error("IsStandardAOD", "No AOD handler. Please use AODEvent() to check this first"); @@ -534,4 +704,58 @@ Bool_t AliAnalysisTaskSE::Notify() return (UserNotify()); } +const AliEventTag *AliAnalysisTaskSE::EventTag() const +{ +// Returns tag for the current event, if any. The return value should always be checked by the user. + if (!fInputHandler) { + Error("EventTag", "Input handler not yet available. Call this in UserExec"); + return NULL; + } + return fInputHandler->GetEventTag(); +} +void AliAnalysisTaskSE::LoadBranches() const +{ +// Load all branches declared in fBranchNames data member of the parent class. +// Should be called in UserExec. + if (!fInputHandler) { + Error("LoadBranches", "Input handler not available yet. Call this in UserExec"); + return; + } + AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager(); + if (mgr->GetAutoBranchLoading()) return; + TString taskbranches; + GetBranches(fInputHandler->GetDataType(), taskbranches); + if (taskbranches.IsNull()) return; + TObjArray *arr = taskbranches.Tokenize(","); + TIter next(arr); + TObject *obj; + while ((obj=next())) mgr->LoadBranch(obj->GetName()); +} + + +//_________________________________________________________________________________________________ +void AliAnalysisTaskSE::ConnectMultiHandler() +{ + // + // Connect MultiHandler + // + fInputHandler = (AliInputEventHandler *)((AliAnalysisManager::GetAnalysisManager())->GetInputEventHandler()); + fMultiInputHandler = dynamic_cast(fInputHandler); + if (fMultiInputHandler) { + fInputHandler = dynamic_cast(fMultiInputHandler->GetFirstInputEventHandler()); + fMCEventHandler = dynamic_cast(fMultiInputHandler->GetFirstMCEventHandler()); + } else { + fMCEventHandler = dynamic_cast((AliAnalysisManager::GetAnalysisManager())->GetMCtruthEventHandler()); + } + if (fMCEventHandler) fMCEvent = fMCEventHandler->MCEvent(); +} + +//_________________________________________________________________________________________________ +void AliAnalysisTaskSE::DisconnectMultiHandler() +{ + // + // Disconnect MultiHandler + // + if (fMultiInputHandler) fInputHandler = fMultiInputHandler; +}