X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;f=ANALYSIS%2FAliAnalysisTaskSE.cxx;h=1ab4532ba6e6687d86989e6dd5ac34ca5991c29e;hb=6121856b6a45285d416d090d516068fb8d83d3ca;hp=7f20b3b1b627d2b80e7a6f25ae86c2c9373e9b04;hpb=3cee83d70aa12527cb9007be343c8dc643d36335;p=u%2Fmrichter%2FAliRoot.git diff --git a/ANALYSIS/AliAnalysisTaskSE.cxx b/ANALYSIS/AliAnalysisTaskSE.cxx index 7f20b3b1b62..1ab4532ba6e 100644 --- a/ANALYSIS/AliAnalysisTaskSE.cxx +++ b/ANALYSIS/AliAnalysisTaskSE.cxx @@ -24,45 +24,63 @@ #include "AliAnalysisTaskSE.h" #include "AliAnalysisManager.h" +#include "AliAnalysisCuts.h" #include "AliAnalysisDataSlot.h" +#include "AliAnalysisDataContainer.h" + #include "AliESDEvent.h" +#include "AliESDfriend.h" #include "AliESD.h" #include "AliAODEvent.h" #include "AliAODHeader.h" #include "AliAODTracklets.h" +#include "AliAODCaloCells.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" #include "AliLog.h" +#include "AliAODDimuon.h" ClassImp(AliAnalysisTaskSE) //////////////////////////////////////////////////////////////////////// -AliAODHeader* AliAnalysisTaskSE::fgAODHeader = NULL; -TClonesArray* AliAnalysisTaskSE::fgAODTracks = NULL; -TClonesArray* AliAnalysisTaskSE::fgAODVertices = NULL; -TClonesArray* AliAnalysisTaskSE::fgAODV0s = NULL; -TClonesArray* AliAnalysisTaskSE::fgAODPMDClusters = NULL; -TClonesArray* AliAnalysisTaskSE::fgAODJets = NULL; -TClonesArray* AliAnalysisTaskSE::fgAODFMDClusters = NULL; -TClonesArray* AliAnalysisTaskSE::fgAODCaloClusters = NULL; -AliAODTracklets* AliAnalysisTaskSE::fgAODTracklets = NULL; - +AliAODHeader* AliAnalysisTaskSE::fgAODHeader = NULL; +TClonesArray* AliAnalysisTaskSE::fgAODTracks = NULL; +TClonesArray* AliAnalysisTaskSE::fgAODVertices = NULL; +TClonesArray* AliAnalysisTaskSE::fgAODV0s = NULL; +TClonesArray* AliAnalysisTaskSE::fgAODPMDClusters = NULL; +TClonesArray* AliAnalysisTaskSE::fgAODJets = NULL; +TClonesArray* AliAnalysisTaskSE::fgAODFMDClusters = NULL; +TClonesArray* AliAnalysisTaskSE::fgAODCaloClusters = NULL; +TClonesArray* AliAnalysisTaskSE::fgAODMCParticles = NULL; +AliAODTracklets* AliAnalysisTaskSE::fgAODTracklets = NULL; +AliAODCaloCells* AliAnalysisTaskSE::fgAODEmcalCells = NULL; +AliAODCaloCells* AliAnalysisTaskSE::fgAODPhosCells = NULL; +TClonesArray* AliAnalysisTaskSE::fgAODDimuons = NULL; AliAnalysisTaskSE::AliAnalysisTaskSE(): AliAnalysisTask(), fDebug(0), fEntry(0), fInputEvent(0x0), + fESDfriend(0x0), fInputHandler(0x0), fOutputAOD(0x0), fMCEvent(0x0), - fTreeA(0x0) + fTreeA(0x0), + fCurrentRunNumber(-1), + fHistosQA(0x0), + fOfflineTriggerMask(0), + fMultiInputHandler(0), + fMCEventHandler(0) { // Default constructor } @@ -72,10 +90,16 @@ AliAnalysisTaskSE::AliAnalysisTaskSE(const char* name): fDebug(0), fEntry(0), fInputEvent(0x0), + fESDfriend(0x0), fInputHandler(0x0), fOutputAOD(0x0), fMCEvent(0x0), - fTreeA(0x0) + fTreeA(0x0), + fCurrentRunNumber(-1), + fHistosQA(0x0), + fOfflineTriggerMask(0), + fMultiInputHandler(0), + fMCEventHandler(0) { // Default constructor DefineInput (0, TChain::Class()); @@ -87,20 +111,29 @@ AliAnalysisTaskSE::AliAnalysisTaskSE(const AliAnalysisTaskSE& obj): fDebug(0), fEntry(0), fInputEvent(0x0), + fESDfriend(0x0), fInputHandler(0x0), fOutputAOD(0x0), fMCEvent(0x0), - fTreeA(0x0) + fTreeA(0x0), + fCurrentRunNumber(-1), + fHistosQA(0x0), + fOfflineTriggerMask(0), + fMultiInputHandler(obj.fMultiInputHandler), + fMCEventHandler(obj.fMCEventHandler) { // Copy constructor - fDebug = obj.fDebug; - fEntry = obj.fEntry; - fInputEvent = obj.fInputEvent; - fInputHandler = obj.fInputHandler; - fOutputAOD = obj.fOutputAOD; - fMCEvent = obj.fMCEvent; - fTreeA = obj.fTreeA; - printf("Constructor (3) \n"); + fDebug = obj.fDebug; + fEntry = obj.fEntry; + fInputEvent = obj.fInputEvent; + fESDfriend = obj.fESDfriend; + fInputHandler = obj.fInputHandler; + fOutputAOD = obj.fOutputAOD; + fMCEvent = obj.fMCEvent; + fTreeA = obj.fTreeA; + fCurrentRunNumber = obj.fCurrentRunNumber; + fHistosQA = obj.fHistosQA; + } @@ -108,42 +141,59 @@ AliAnalysisTaskSE& AliAnalysisTaskSE::operator=(const AliAnalysisTaskSE& other) { // Assignment AliAnalysisTask::operator=(other); - fDebug = other.fDebug; - fEntry = other.fEntry; - fInputEvent = other.fInputEvent; - fInputHandler = other.fInputHandler; - fOutputAOD = other.fOutputAOD; - fMCEvent = other.fMCEvent; - fTreeA = other.fTreeA; + fDebug = other.fDebug; + fEntry = other.fEntry; + fInputEvent = other.fInputEvent; + fESDfriend = other.fESDfriend; + fInputHandler = other.fInputHandler; + fOutputAOD = other.fOutputAOD; + fMCEvent = other.fMCEvent; + fTreeA = other.fTreeA; + fCurrentRunNumber = other.fCurrentRunNumber; + fHistosQA = other.fHistosQA; + fOfflineTriggerMask = other.fOfflineTriggerMask; + fMultiInputHandler = other.fMultiInputHandler; + fMCEventHandler = other.fMCEventHandler; return *this; } +//______________________________________________________________________________ +Bool_t AliAnalysisTaskSE::CheckPostData() const +{ +// Checks if data was posted to all outputs defined by the task. If task does +// not have output slots this returns always kTRUE. + AliAnalysisDataContainer *coutput; + AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager(); + for (Int_t islot=1; islotGetContainer(); + if (!mgr->GetOutputs()->FindObject(coutput)) continue; + if (!coutput->GetData()) return kFALSE; + } + CheckOwnership(); + return kTRUE; +} 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) { - fInputEvent = fInputHandler->GetEvent(); + if ((fInputHandler->GetTree())->GetBranch("ESDfriend.")) + fESDfriend = ((AliESDInputHandler*)fInputHandler)->GetESDfriend(); + + fInputEvent = fInputHandler->GetEvent(); } else if( fMCEvent ) { AliWarning("No Input Event Handler connected, only MC Truth Event Handler") ; } else { AliError("No Input Event Handler connected") ; return ; } + // Disconnect multi handler + DisconnectMultiHandler(); } void AliAnalysisTaskSE::CreateOutputObjects() @@ -153,29 +203,34 @@ 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; + AliAODInputHandler* aodIH = static_cast((AliAnalysisManager::GetAnalysisManager())->GetInputEventHandler()); + if (aodIH) { + if (aodIH->GetMergeEvents()) merging = kTRUE; + } + // Check if AOD replication has been required - if (handler) { fOutputAOD = handler->GetAOD(); fTreeA = handler->GetTree(); - if (!(handler->IsStandard())) { + if (fOutputAOD && !(handler->IsStandard())) { if ((handler->NeedsHeaderReplication()) && !(fgAODHeader)) { if (fDebug > 1) AliInfo("Replicating header"); fgAODHeader = new AliAODHeader; handler->AddBranch("AliAODHeader", &fgAODHeader); } - if ((handler->NeedsTracksBranchReplication()) && !(fgAODTracks)) + if ((handler->NeedsTracksBranchReplication() || merging) && !(fgAODTracks)) { if (fDebug > 1) AliInfo("Replicating track branch\n"); fgAODTracks = new TClonesArray("AliAODTrack",500); fgAODTracks->SetName("tracks"); handler->AddBranch("TClonesArray", &fgAODTracks); } - if ((handler->NeedsVerticesBranchReplication()) && !(fgAODVertices)) + if ((handler->NeedsVerticesBranchReplication() || merging) && !(fgAODVertices)) { if (fDebug > 1) AliInfo("Replicating vertices branch\n"); fgAODVertices = new TClonesArray("AliAODVertex",500); @@ -202,7 +257,7 @@ void AliAnalysisTaskSE::CreateOutputObjects() fgAODPMDClusters->SetName("pmdClusters"); handler->AddBranch("TClonesArray", &fgAODPMDClusters); } - if ((handler->NeedsJetsBranchReplication()) && !(fgAODJets)) + if ((handler->NeedsJetsBranchReplication() || merging) && !(fgAODJets)) { if (fDebug > 1) AliInfo("Replicating Jets branch\n"); fgAODJets = new TClonesArray("AliAODJet",500); @@ -216,93 +271,309 @@ void AliAnalysisTaskSE::CreateOutputObjects() fgAODFMDClusters->SetName("fmdClusters"); handler->AddBranch("TClonesArray", &fgAODFMDClusters); } - if ((handler->NeedsCaloClustersBranchReplication()) && !(fgAODCaloClusters)) + if ((handler->NeedsCaloClustersBranchReplication() || merging) && !(fgAODCaloClusters)) { if (fDebug > 1) AliInfo("Replicating CaloClusters branch\n"); fgAODCaloClusters = new TClonesArray("AliAODCaloCluster",500); fgAODCaloClusters->SetName("caloClusters"); handler->AddBranch("TClonesArray", &fgAODCaloClusters); + + fgAODEmcalCells = new AliAODCaloCells("emcalCells","emcalCells",AliVCaloCells::kEMCALCell); + handler->AddBranch("AliAODCaloCells", &fgAODEmcalCells); + + fgAODPhosCells = new AliAODCaloCells("phosCells","phosCells",AliVCaloCells::kPHOSCell); + handler->AddBranch("AliAODCaloCells", &fgAODPhosCells); + + } + if ((handler->NeedsMCParticlesBranchReplication() || merging) && !(fgAODMCParticles)) + { + if (fDebug > 1) AliInfo("Replicating MCParticles branch\n"); + fgAODMCParticles = new TClonesArray("AliAODMCParticle",500); + fgAODMCParticles->SetName("mcparticles"); + handler->AddBranch("TClonesArray", &fgAODMCParticles); } + if ((handler->NeedsDimuonsBranchReplication() || merging) && !(fgAODDimuons)) + { + if (fDebug > 1) AliInfo("Replicating dimuon branch\n"); + fgAODDimuons = new TClonesArray("AliAODDimuon",0); + fgAODDimuons->SetName("dimuons"); + handler->AddBranch("TClonesArray", &fgAODDimuons); + } + + // 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 + + ConnectMultiHandler(); + + if ( fDebug >= 10) + printf("Task is active %5d\n", IsActive()); + if (fDebug > 1) AliInfo("AliAnalysisTaskSE::Exec() \n"); - if( fInputHandler ) - fEntry = fInputHandler->GetReadEntry(); +// + AliAODHandler* handler = dynamic_cast + ((AliAnalysisManager::GetAnalysisManager())->GetOutputEventHandler()); + + AliAODInputHandler* aodH = dynamic_cast(fInputHandler); +// +// Was event selected ? If no event selection mechanism, the event SHOULD be selected (AG) + UInt_t isSelected = AliVEvent::kAny; + if( fInputHandler && fInputHandler->GetEventSelection()) { + // 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( fInputHandler ) { + fEntry = fInputHandler->GetReadEntry(); + fESDfriend = ((AliESDInputHandler*)fInputHandler)->GetESDfriend(); + } + + +// Notify the change of run number + if (InputEvent() && (InputEvent()->GetRunNumber() != fCurrentRunNumber)) { + fCurrentRunNumber = InputEvent()->GetRunNumber(); + NotifyRun(); + } + else if( fMCEvent ) fEntry = fMCEvent->Header()->GetEvent(); if ( !((Entry()-1)%100) && fDebug > 0) AliInfo(Form("%s ----> Processing event # %lld", CurrentFileName(), Entry())); - AliAODHandler* handler = (AliAODHandler*) - ((AliAnalysisManager::GetAnalysisManager())->GetOutputEventHandler()); - AliAODInputHandler* aodH = dynamic_cast(fInputHandler); + + if (handler && aodH) { - if (!(handler->IsStandard()) && !(handler->AODIsReplicated())) { + fMCEvent = aodH->MCEvent(); + Bool_t merging = aodH->GetMergeEvents(); + AliAODEvent* aod = dynamic_cast(InputEvent()); + + if (aod && !(handler->IsStandard()) && !(handler->AODIsReplicated())) { if ((handler->NeedsHeaderReplication()) && (fgAODHeader)) { // copy the contents by assigment - *fgAODHeader = *(dynamic_cast(InputEvent()->GetHeader())); + *fgAODHeader = *(aod->GetHeader()); } - if ((handler->NeedsTracksBranchReplication()) && (fgAODTracks)) + if ((handler->NeedsTracksBranchReplication() || merging) && (fgAODTracks)) { - TClonesArray* tracks = (dynamic_cast(InputEvent()))->GetTracks(); + TClonesArray* tracks = aod->GetTracks(); new (fgAODTracks) TClonesArray(*tracks); } - if ((handler->NeedsVerticesBranchReplication()) && (fgAODVertices)) + 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(); - new (fgAODV0s) TClonesArray(*V0s); + 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(); - new (fgAODPMDClusters) TClonesArray(*PMDClusters); + TClonesArray* pmdClusters = aod->GetPmdClusters(); + new (fgAODPMDClusters) TClonesArray(*pmdClusters); } - if ((handler->NeedsJetsBranchReplication()) && (fgAODJets)) + if ((handler->NeedsJetsBranchReplication() || merging) && (fgAODJets)) { - TClonesArray* Jets = (dynamic_cast(InputEvent()))->GetJets(); - new (fgAODJets) TClonesArray(*Jets); + TClonesArray* jets = aod->GetJets(); + new (fgAODJets) TClonesArray(*jets); } if ((handler->NeedsFMDClustersBranchReplication()) && (fgAODFMDClusters)) { - TClonesArray* FMDClusters = (dynamic_cast(InputEvent()))->GetFmdClusters(); - new (fgAODFMDClusters) TClonesArray(*FMDClusters); + TClonesArray* fmdClusters = aod->GetFmdClusters(); + new (fgAODFMDClusters) TClonesArray(*fmdClusters); } - if ((handler->NeedsCaloClustersBranchReplication()) && (fgAODCaloClusters)) + if ((handler->NeedsCaloClustersBranchReplication() || merging) && (fgAODCaloClusters)) { - TClonesArray* CaloClusters = (dynamic_cast(InputEvent()))->GetCaloClusters(); - new (fgAODCaloClusters) TClonesArray(*CaloClusters); + TClonesArray* caloClusters = aod->GetCaloClusters(); + new (fgAODCaloClusters) TClonesArray(*caloClusters); + } + + if ((handler->NeedsMCParticlesBranchReplication() || merging) && (fgAODMCParticles)) + { + TClonesArray* mcParticles = (TClonesArray*) (aod->FindListObject("mcparticles")); + if( mcParticles ) + new (fgAODMCParticles) TClonesArray(*mcParticles); } - // - handler->SetAODIsReplicated(); + if ((handler->NeedsDimuonsBranchReplication() || merging) && (fgAODDimuons)) + { + fgAODDimuons->Clear(); + TClonesArray& dimuons = *fgAODDimuons; + TClonesArray& tracksnew = *fgAODTracks; + + 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); + if(track->IsMuonTrack()) { + nMuonTrack[nMuons]= ii; + nMuons++; + } + } + Int_t jDimuons=0; + if(nMuons >= 2){ + for(Int_t i = 0; i < nMuons; i++){ + Int_t index0 = nMuonTrack[i]; + for(Int_t j = i+1; j < nMuons; j++){ + Int_t index1 = nMuonTrack[j]; + tracksnew.At(index0)->ResetBit(kIsReferenced); + tracksnew.At(index0)->SetUniqueID(0); + tracksnew.At(index1)->ResetBit(kIsReferenced); + tracksnew.At(index1)->SetUniqueID(0); + new(dimuons[jDimuons++]) AliAODDimuon(tracksnew.At(index0),tracksnew.At(index1)); + } + } + } + } + // Additional merging if needed + if (merging) { + Int_t nc; + + // mcParticles + TClonesArray* mcparticles = (TClonesArray*) ((aodH->GetEventToMerge())->FindListObject("mcparticles")); + 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(); + if(tracks){ + 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() + 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(); + if( clusters ){ + Int_t ncl = clusters->GetEntries(); + nc = fgAODCaloClusters->GetEntries(); + for (Int_t i = 0; i < ncl; i++) { + AliAODCaloCluster* cluster = (AliAODCaloCluster*) clusters->At(i); + new((*fgAODCaloClusters)[nc++]) AliAODCaloCluster(*cluster); + } + } + + // EMCAL cells + //*fgAODEmcalCells = *(aod->GetEMCALCells()); // This will be valid after 10.Mar.2011. + { + 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* copycells = new AliAODCaloCells(*fgAODEmcalCells); + fgAODEmcalCells->CreateContainer(nc+1); + Int_t nn = copycells->GetNumberOfCells(); + while( nn-- ){ fgAODEmcalCells->SetCell(nn,copycells->GetCellNumber(nn),copycells->GetAmplitude(nn)); } + fgAODEmcalCells->SetCell(nc++,cn,cellsA->GetAmplitude(i)); + delete copycells; + } + } + fgAODEmcalCells->Sort(); + } + + // PHOS cells + //*fgAODPhosCells = *(aod->GetPHOSCells()); // This will be valid after 10.Mar.2011. + { + 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* copycells = new AliAODCaloCells(*fgAODPhosCells); + fgAODPhosCells->CreateContainer(nc+1); + Int_t nn = copycells->GetNumberOfCells(); + while( nn-- ){ fgAODPhosCells->SetCell(nn,copycells->GetCellNumber(nn),copycells->GetAmplitude(nn)); } + fgAODPhosCells->SetCell(nc++,cn,cellsP->GetAmplitude(i)); + delete copycells; + } + } + fgAODPhosCells->Sort(); + } + + } // merging + + handler->SetAODIsReplicated(); } } + // Call the user analysis - UserExec(option); - // Added protection in case the derived task is not an AOD producer. + 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() @@ -318,7 +589,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); @@ -329,7 +600,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"); @@ -337,3 +608,64 @@ Bool_t AliAnalysisTaskSE::IsStandardAOD() const } return handler->IsStandard(); } + +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; +}