/************************************************************************** * Copyright(c) 1998-1999, 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. * **************************************************************************/ //_________________________________________________________________________ // This analysis provides a new list of clusters to be used in other analysis // // Author: Gustavo Conesa Balbastre, // Adapted from analysis class from Deepa Thomas // // //_________________________________________________________________________ // --- Root --- #include "TString.h" #include "TRefArray.h" #include "TClonesArray.h" #include "TTree.h" #include "TGeoManager.h" #include "TROOT.h" #include "TInterpreter.h" #include "TFile.h" // --- AliRoot Analysis Steering #include "AliAnalysisTask.h" #include "AliAnalysisManager.h" #include "AliESDEvent.h" #include "AliGeomManager.h" #include "AliVCaloCells.h" #include "AliAODCaloCluster.h" #include "AliCDBManager.h" #include "AliCDBStorage.h" #include "AliCDBEntry.h" #include "AliLog.h" #include "AliVEventHandler.h" #include "AliAODInputHandler.h" // --- EMCAL #include "AliEMCALRecParam.h" #include "AliEMCALAfterBurnerUF.h" #include "AliEMCALGeometry.h" #include "AliEMCALClusterizerNxN.h" #include "AliEMCALClusterizerv1.h" #include "AliEMCALClusterizerv2.h" #include "AliEMCALRecPoint.h" #include "AliEMCALDigit.h" #include "AliCaloCalibPedestal.h" #include "AliEMCALCalibData.h" #include "AliEMCALRecoUtils.h" #include "AliAnalysisTaskEMCALClusterize.h" ClassImp(AliAnalysisTaskEMCALClusterize) //______________________________________________________________________________ AliAnalysisTaskEMCALClusterize::AliAnalysisTaskEMCALClusterize(const char *name) : AliAnalysisTaskSE(name) , fEvent(0) , fGeom(0), fGeomName("EMCAL_COMPLETEV1") , fGeomMatrixSet(kFALSE), fLoadGeomMatrices(kFALSE) , fCalibData(0), fPedestalData(0) , fOCDBpath("raw://"), fAccessOCDB(kFALSE) , fDigitsArr(0), fClusterArr(0), fCaloClusterArr(0) , fRecParam(0), fClusterizer(0) , fUnfolder(0), fJustUnfold(kFALSE) , fOutputAODBranch(0), fOutputAODBranchName("newEMCALClusters") , fFillAODFile(kTRUE), fFillAODHeader(0) , fFillAODCaloCells(0), fRun(-1) , fRecoUtils(0), fConfigName("") , fCellLabels(), fCellSecondLabels(), fCellTime() , fCellMatchdEta(), fCellMatchdPhi() , fMaxEvent(1000000000), fDoTrackMatching(kFALSE) , fSelectCell(kFALSE), fSelectCellMinE(0.005), fSelectCellMinFrac(0.001) , fRemoveLEDEvents(kFALSE), fRemoveExoticEvents(kFALSE) , fImportGeometryFromFile(kFALSE), fImportGeometryFilePath("") { //ctor for(Int_t i = 0; i < 12; i++) fGeomMatrix[i] = 0; for(Int_t j = 0; j < 24*48*11; j++) { fCellLabels[j] = -1; fCellSecondLabels[j] = -1; fCellTime[j] = 0.; fCellMatchdEta[j] = -999; fCellMatchdPhi[j] = -999; } fDigitsArr = new TClonesArray("AliEMCALDigit",12000); fClusterArr = new TObjArray(10000); fCaloClusterArr = new TObjArray(10000); fRecParam = new AliEMCALRecParam; fBranchNames = "ESD:AliESDHeader.,EMCALCells."; fRecoUtils = new AliEMCALRecoUtils(); } //______________________________________________________________ AliAnalysisTaskEMCALClusterize::AliAnalysisTaskEMCALClusterize() : AliAnalysisTaskSE("DefaultAnalysis_AliAnalysisTaskEMCALClusterize") , fEvent(0) , fGeom(0), fGeomName("EMCAL_COMPLETEV1") , fGeomMatrixSet(kFALSE), fLoadGeomMatrices(kFALSE) , fCalibData(0), fPedestalData(0) , fOCDBpath("raw://"), fAccessOCDB(kFALSE) , fDigitsArr(0), fClusterArr(0), fCaloClusterArr(0) , fRecParam(0), fClusterizer(0) , fUnfolder(0), fJustUnfold(kFALSE) , fOutputAODBranch(0), fOutputAODBranchName("newEMCALClusters") , fFillAODFile(kTRUE), fFillAODHeader(0) , fFillAODCaloCells(0), fRun(-1) , fRecoUtils(0), fConfigName("") , fCellLabels(), fCellSecondLabels(), fCellTime() , fCellMatchdEta(), fCellMatchdPhi() , fMaxEvent(1000000000), fDoTrackMatching(kFALSE) , fSelectCell(kFALSE), fSelectCellMinE(0.005), fSelectCellMinFrac(0.001) , fRemoveLEDEvents(kFALSE), fRemoveExoticEvents(kFALSE) , fImportGeometryFromFile(kFALSE), fImportGeometryFilePath("") { // Constructor for(Int_t i = 0; i < 12; i++) fGeomMatrix[i] = 0; for(Int_t j = 0; j < 24*48*11; j++) { fCellLabels[j] = -1; fCellSecondLabels[j] = -1; fCellTime[j] = 0.; fCellMatchdEta[j] = -999; fCellMatchdPhi[j] = -999; } fDigitsArr = new TClonesArray("AliEMCALDigit",12000); fClusterArr = new TObjArray(10000); fCaloClusterArr = new TObjArray(10000); fRecParam = new AliEMCALRecParam; fBranchNames = "ESD:AliESDHeader.,EMCALCells."; fRecoUtils = new AliEMCALRecoUtils(); } //_______________________________________________________________ AliAnalysisTaskEMCALClusterize::~AliAnalysisTaskEMCALClusterize() { //dtor if (fDigitsArr){ fDigitsArr->Clear("C"); delete fDigitsArr; } if (fClusterArr){ fClusterArr->Delete(); delete fClusterArr; } if (fCaloClusterArr){ fCaloClusterArr->Delete(); delete fCaloClusterArr; } if(fClusterizer) delete fClusterizer; if(fUnfolder) delete fUnfolder; if(fRecoUtils) delete fRecoUtils; } //_________________________________________________ Bool_t AliAnalysisTaskEMCALClusterize::AccessOCDB() { //Access to OCDB stuff fEvent = InputEvent(); if (!fEvent) { Warning("AccessOCDB","Event not available!!!"); return kFALSE; } if (fEvent->GetRunNumber()==fRun) return kTRUE; fRun = fEvent->GetRunNumber(); if(DebugLevel() > 1 ) printf("AliAnalysisTaksEMCALClusterize::AccessODCD() - Begin"); //fGeom = AliEMCALGeometry::GetInstance(fGeomName); AliCDBManager *cdb = AliCDBManager::Instance(); if (fOCDBpath.Length()){ cdb->SetDefaultStorage(fOCDBpath.Data()); printf("AliAnalysisTaksEMCALClusterize::AccessOCDB() - Default storage %s",fOCDBpath.Data()); } cdb->SetRun(fEvent->GetRunNumber()); // // EMCAL from RAW OCDB if (fOCDBpath.Contains("alien:")) { cdb->SetSpecificStorage("EMCAL/Calib/Data","alien://Folder=/alice/data/2010/OCDB"); cdb->SetSpecificStorage("EMCAL/Calib/Pedestals","alien://Folder=/alice/data/2010/OCDB"); } TString path = cdb->GetDefaultStorage()->GetBaseFolder(); // init parameters: //Get calibration parameters if(!fCalibData) { AliCDBEntry *entry = (AliCDBEntry*) AliCDBManager::Instance()->Get("EMCAL/Calib/Data"); if (entry) fCalibData = (AliEMCALCalibData*) entry->GetObject(); } if(!fCalibData) AliFatal("Calibration parameters not found in CDB!"); //Get calibration parameters if(!fPedestalData) { AliCDBEntry *entry = (AliCDBEntry*) AliCDBManager::Instance()->Get("EMCAL/Calib/Pedestals"); if (entry) fPedestalData = (AliCaloCalibPedestal*) entry->GetObject(); } if(!fPedestalData) AliFatal("Dead map not found in CDB!"); return kTRUE; } //_____________________________________________________ void AliAnalysisTaskEMCALClusterize::CheckAndGetEvent() { // Get the input event, it can depend in embedded events what you want to get // Also check if the quality of the event is good if not reject it fEvent = 0x0; AliAODInputHandler* aodIH = dynamic_cast((AliAnalysisManager::GetAnalysisManager())->GetInputEventHandler()); Int_t eventN = Entry(); if(aodIH) eventN = aodIH->GetReadEntry(); if (eventN > fMaxEvent) return ; //printf("Clusterizer --- Event %d-- \n",eventN); //Check if input event are embedded events //If so, take output event if (aodIH && aodIH->GetMergeEvents()) { fEvent = AODEvent(); if(!aodIH->GetMergeEMCALCells()) AliFatal("Events merged but not EMCAL cells, check analysis settings!"); if(DebugLevel() > 1){ printf("AliAnalysisTaksEMCALClusterize::UserExec() - Use embedded events\n"); printf("\t InputEvent N Clusters %d, N Cells %d\n",InputEvent()->GetNumberOfCaloClusters(), InputEvent()->GetEMCALCells()->GetNumberOfCells()); printf("\t MergedEvent N Clusters %d, N Cells %d\n",aodIH->GetEventToMerge()->GetNumberOfCaloClusters(), aodIH->GetEventToMerge()->GetEMCALCells()->GetNumberOfCells()); for (Int_t icl=0; icl < aodIH->GetEventToMerge()->GetNumberOfCaloClusters(); icl++) { AliAODCaloCluster *sigCluster = aodIH->GetEventToMerge()->GetCaloCluster(icl); if(sigCluster->IsEMCAL()) printf("\t \t Signal cluster: i %d, E %f\n",icl,sigCluster->E()); } printf("\t OutputEvent N Clusters %d, N Cells %d\n", AODEvent()->GetNumberOfCaloClusters(), AODEvent()->GetEMCALCells()->GetNumberOfCells()); } } else { fEvent = InputEvent(); if(fFillAODCaloCells) FillAODCaloCells(); if(fFillAODHeader) FillAODHeader(); } if (!fEvent) { Error("UserExec","Event not available"); return ; } //------------------------------------------------------------------------------------- // Reject events if LED was firing, use only for LHC11a data // Reject event if triggered by exotic cell and remove exotic cells if not triggered //------------------------------------------------------------------------------------- if(IsLEDEvent ()) { fEvent = 0x0 ; return ; } if(IsExoticEvent()) { fEvent = 0x0 ; return ; } //Magic line to write events to AOD filem put after event rejection AliAnalysisManager::GetAnalysisManager()->GetOutputEventHandler()->SetFillAOD(fFillAODFile); } //____________________________________________________ void AliAnalysisTaskEMCALClusterize::ClusterizeCells() { // Recluster calocells, transform them into digits, // feed the clusterizer with them and get new list of clusters //In case of MC, first loop on the clusters and fill MC label to array Int_t nClusters = fEvent->GetNumberOfCaloClusters(); Int_t nClustersOrg = 0; AliAODInputHandler* aodIH = dynamic_cast((AliAnalysisManager::GetAnalysisManager())->GetInputEventHandler()); if(aodIH && aodIH->GetEventToMerge()) //Embedding nClusters = aodIH->GetEventToMerge()->GetNumberOfCaloClusters(); //Get clusters directly from embedded signal for (Int_t i = 0; i < nClusters; i++) { AliVCluster *clus = 0; if(aodIH && aodIH->GetEventToMerge()) //Embedding clus = aodIH->GetEventToMerge()->GetCaloCluster(i); //Get clusters directly from embedded signal else clus = fEvent->GetCaloCluster(i); if(!clus) return; if(clus->IsEMCAL()){ Int_t label = clus->GetLabel(); Int_t label2 = -1 ; //printf("Org cluster E %f, Time %e, Id = ", clus->E(), clus->GetTOF() ); if (clus->GetNLabels()>=2) label2 = clus->GetLabelAt(1) ; UShort_t * index = clus->GetCellsAbsId() ; for(Int_t icell=0; icell < clus->GetNCells(); icell++ ){ fCellLabels[index[icell]] = label; fCellSecondLabels[index[icell]] = label2; fCellTime[index[icell]] = clus->GetTOF(); fCellMatchdEta[index[icell]] = clus->GetTrackDz(); fCellMatchdPhi[index[icell]] = clus->GetTrackDx(); //printf(" %d,", index[icell] ); } nClustersOrg++; } // printf("\n"); } // Transform CaloCells into Digits Int_t idigit = 0; Int_t id = -1; Float_t amp = -1; Double_t time = -1; AliVCaloCells *cells = fEvent->GetEMCALCells(); TTree *digitsTree = new TTree("digitstree","digitstree"); digitsTree->Branch("EMCAL","TClonesArray", &fDigitsArr, 32000); Int_t bc = InputEvent()->GetBunchCrossNumber(); for (Int_t icell = 0; icell < cells->GetNumberOfCells(); icell++) { // Get cell values, recalibrate and not include bad channels found in analysis, nor cells with too low energy, nor exotic cell id = cells->GetCellNumber(icell); Bool_t accept = fRecoUtils->AcceptCalibrateCell(id,bc,amp,time,cells); // Do not include cells with too low energy, nor exotic cell if(amp < fRecParam->GetMinECut() ) accept = kFALSE; // In case of AOD analysis cell time is 0, approximate replacing by time of the cluster the digit belongs. if (time*1e9 < 1.) { time = fCellTime[id]; //printf("cell %d time cluster %e\n",id, time); fRecoUtils->RecalibrateCellTime(id,bc,time); } if( accept && fRecoUtils->IsExoticCell(id,cells,bc)){ accept = kFALSE; } if( !accept ){ fCellLabels[id] =-1; //reset the entry in the array for next event fCellSecondLabels[id]=-1; //reset the entry in the array for next event fCellTime[id] = 0.; fCellMatchdEta[id] =-999; fCellMatchdPhi[id] =-999; if( DebugLevel() > 2 ) printf("AliAnalysisTaksEMCALClusterize::ClusterizeCells() - Remove channel absId %d, index %d of %d, amp %f, time %f\n", id,icell, cells->GetNumberOfCells(), amp, time*1.e9); continue; } //Create the digit, put a fake primary deposited energy to trick the clusterizer when checking the most likely primary new((*fDigitsArr)[idigit]) AliEMCALDigit( fCellLabels[id], fCellLabels[id],id, amp, time,AliEMCALDigit::kHG,idigit, 0, 0, 1); fCellLabels[id] =-1; //reset the entry in the array for next event idigit++; } //Fill the tree with digits digitsTree->Fill(); //------------------------------------------------------------------------------------- //Do the clusterization //------------------------------------------------------------------------------------- TTree *clustersTree = new TTree("clustertree","clustertree"); fClusterizer->SetInput(digitsTree); fClusterizer->SetOutput(clustersTree); fClusterizer->Digits2Clusters(""); //------------------------------------------------------------------------------------- //Transform the recpoints into AliVClusters //------------------------------------------------------------------------------------- clustersTree->SetBranchStatus("*",0); //disable all branches clustersTree->SetBranchStatus("EMCALECARP",1); //Enable only the branch we need TBranch *branch = clustersTree->GetBranch("EMCALECARP"); branch->SetAddress(&fClusterArr); branch->GetEntry(0); RecPoints2Clusters(fDigitsArr, fClusterArr, fCaloClusterArr); if(!fCaloClusterArr){ printf("AliAnalysisTaksEMCALClusterize::UserExec() - No array with CaloClusters, input RecPoints entries %d\n",fClusterArr->GetEntriesFast()); return; } if( DebugLevel() > 0 ){ printf("AliAnalysisTaksEMCALClusterize::ClusterizeCells() - N clusters: before recluster %d, after recluster %d\n",nClustersOrg, fCaloClusterArr->GetEntriesFast()); if(fCaloClusterArr->GetEntriesFast() != fClusterArr->GetEntriesFast()){ printf("\t Some RecRoints not transformed into CaloClusters (clusterizer %d, unfold %d): Input entries %d - Output entries %d - %d (not fast)\n", fRecParam->GetClusterizerFlag(),fRecParam->GetUnfold(), fClusterArr->GetEntriesFast(), fCaloClusterArr->GetEntriesFast(), fCaloClusterArr->GetEntries()); } } //Reset the array with second labels for this event memset(fCellSecondLabels, -1, sizeof(fCellSecondLabels)); //---CLEAN UP----- fClusterizer->Clear(); fDigitsArr ->Clear("C"); fClusterArr ->Delete(); // Do not Clear(), it leaks, why? clustersTree->Delete("all"); digitsTree ->Delete("all"); } //_____________________________________________________ void AliAnalysisTaskEMCALClusterize::ClusterUnfolding() { // Take the event clusters and unfold them AliVCaloCells *cells = fEvent->GetEMCALCells(); Double_t cellAmplitude = 0; Double_t cellTime = 0; Short_t cellNumber = 0; Int_t nClustersOrg = 0; // Fill the array with the EMCAL clusters, copy them for (Int_t i = 0; i < fEvent->GetNumberOfCaloClusters(); i++) { AliVCluster *clus = fEvent->GetCaloCluster(i); if(clus->IsEMCAL()){ //recalibrate/remove bad channels/etc if requested if(fRecoUtils->ClusterContainsBadChannel(fGeom,clus->GetCellsAbsId(), clus->GetNCells())){ continue; } if(fRecoUtils->IsRecalibrationOn()){ //Calibrate cluster fRecoUtils->RecalibrateClusterEnergy(fGeom, clus, cells); //CalibrateCells for (Int_t icell = 0; icell < cells->GetNumberOfCells(); icell++) { if (cells->GetCell(icell, cellNumber, cellAmplitude, cellTime) != kTRUE) break; Int_t imod = -1, iphi =-1, ieta=-1,iTower = -1, iIphi = -1, iIeta = -1; fGeom->GetCellIndex(cellNumber,imod,iTower,iIphi,iIeta); fGeom->GetCellPhiEtaIndexInSModule(imod,iTower,iIphi, iIeta,iphi,ieta); //Do not include bad channels found in analysis? if( fRecoUtils->IsBadChannelsRemovalSwitchedOn() && fRecoUtils->GetEMCALChannelStatus(imod, ieta, iphi)){ fCellLabels[cellNumber] =-1; //reset the entry in the array for next event fCellSecondLabels[cellNumber]=-1; //reset the entry in the array for next event fCellTime[cellNumber] = 0.; fCellMatchdEta[cellNumber] =-999; fCellMatchdPhi[cellNumber] =-999; continue; } cells->SetCell(icell, cellNumber, cellAmplitude*fRecoUtils->GetEMCALChannelRecalibrationFactor(imod,ieta,iphi),cellTime); }// cells loop }// recalibrate //Cast to ESD or AOD, needed to create the cluster array AliESDCaloCluster * esdCluster = dynamic_cast (clus); AliAODCaloCluster * aodCluster = dynamic_cast (clus); if (esdCluster){ fCaloClusterArr->Add( new AliESDCaloCluster(*esdCluster) ); }//ESD else if(aodCluster){ fCaloClusterArr->Add( new AliAODCaloCluster(*aodCluster) ); }//AOD else Warning("UserExec()"," - Wrong CaloCluster type?"); nClustersOrg++; } } //Do the unfolding fUnfolder->UnfoldClusters(fCaloClusterArr, cells); //CLEAN-UP fUnfolder->Clear(); } //_____________________________________________________ void AliAnalysisTaskEMCALClusterize::FillAODCaloCells() { // Put calo cells in standard branch AliVCaloCells &eventEMcells = *(fEvent->GetEMCALCells()); Int_t nEMcell = eventEMcells.GetNumberOfCells() ; AliAODCaloCells &aodEMcells = *(AODEvent()->GetEMCALCells()); aodEMcells.CreateContainer(nEMcell); aodEMcells.SetType(AliVCaloCells::kEMCALCell); Double_t calibFactor = 1.; for (Int_t iCell = 0; iCell < nEMcell; iCell++) { Int_t imod = -1, iphi =-1, ieta=-1,iTower = -1, iIphi = -1, iIeta = -1; fGeom->GetCellIndex(eventEMcells.GetCellNumber(iCell),imod,iTower,iIphi,iIeta); fGeom->GetCellPhiEtaIndexInSModule(imod,iTower,iIphi, iIeta,iphi,ieta); if(fRecoUtils->IsRecalibrationOn()){ calibFactor = fRecoUtils->GetEMCALChannelRecalibrationFactor(imod,ieta,iphi); } if(!fRecoUtils->GetEMCALChannelStatus(imod, ieta, iphi)){ //Channel is not declared as bad aodEMcells.SetCell(iCell,eventEMcells.GetCellNumber(iCell),eventEMcells.GetAmplitude(iCell)*calibFactor); } else { aodEMcells.SetCell(iCell,eventEMcells.GetCellNumber(iCell),0); } } aodEMcells.Sort(); } //__________________________________________________ void AliAnalysisTaskEMCALClusterize::FillAODHeader() { //Put event header information in standard AOD branch AliESDEvent* esdevent = dynamic_cast (fEvent); AliAODEvent* aodevent = dynamic_cast (fEvent); Double_t pos[3] ; Double_t covVtx[6]; for (Int_t i = 0; i < 6; i++) covVtx[i] = 0.; AliAODHeader* header = AODEvent()->GetHeader(); header->SetRunNumber(fEvent->GetRunNumber()); if(esdevent){ TTree* tree = fInputHandler->GetTree(); if (tree) { TFile* file = tree->GetCurrentFile(); if (file) header->SetESDFileName(file->GetName()); } } else if (aodevent) header->SetESDFileName(aodevent->GetHeader()->GetESDFileName()); header->SetBunchCrossNumber(fEvent->GetBunchCrossNumber()); header->SetOrbitNumber(fEvent->GetOrbitNumber()); header->SetPeriodNumber(fEvent->GetPeriodNumber()); header->SetEventType(fEvent->GetEventType()); //Centrality if(fEvent->GetCentrality()){ header->SetCentrality(new AliCentrality(*(fEvent->GetCentrality()))); } else{ header->SetCentrality(0); } //Trigger header->SetOfflineTrigger(fInputHandler->IsEventSelected()); // propagate the decision of the physics selection if (esdevent) header->SetFiredTriggerClasses(esdevent->GetFiredTriggerClasses()); else if (aodevent) header->SetFiredTriggerClasses(aodevent->GetFiredTriggerClasses()); header->SetTriggerMask(fEvent->GetTriggerMask()); header->SetTriggerCluster(fEvent->GetTriggerCluster()); if(esdevent){ header->SetL0TriggerInputs(esdevent->GetHeader()->GetL0TriggerInputs()); header->SetL1TriggerInputs(esdevent->GetHeader()->GetL1TriggerInputs()); header->SetL2TriggerInputs(esdevent->GetHeader()->GetL2TriggerInputs()); } else if (aodevent){ header->SetL0TriggerInputs(aodevent->GetHeader()->GetL0TriggerInputs()); header->SetL1TriggerInputs(aodevent->GetHeader()->GetL1TriggerInputs()); header->SetL2TriggerInputs(aodevent->GetHeader()->GetL2TriggerInputs()); } header->SetMagneticField(fEvent->GetMagneticField()); //header->SetMuonMagFieldScale(esdevent->GetCurrentDip()/6000.); header->SetZDCN1Energy(fEvent->GetZDCN1Energy()); header->SetZDCP1Energy(fEvent->GetZDCP1Energy()); header->SetZDCN2Energy(fEvent->GetZDCN2Energy()); header->SetZDCP2Energy(fEvent->GetZDCP2Energy()); header->SetZDCEMEnergy(fEvent->GetZDCEMEnergy(0),fEvent->GetZDCEMEnergy(1)); Float_t diamxy[2]={fEvent->GetDiamondX(),fEvent->GetDiamondY()}; Float_t diamcov[3]; fEvent->GetDiamondCovXY(diamcov); header->SetDiamond(diamxy,diamcov); if (esdevent) header->SetDiamondZ(esdevent->GetDiamondZ(),esdevent->GetSigma2DiamondZ()); else if (aodevent) header->SetDiamondZ(aodevent->GetDiamondZ(),aodevent->GetSigma2DiamondZ()); // // Int_t nVertices = 1 ;/* = prim. vtx*/; Int_t nCaloClus = fEvent->GetNumberOfCaloClusters(); AODEvent()->ResetStd(0, nVertices, 0, 0, 0, nCaloClus, 0, 0); // Access to the AOD container of vertices TClonesArray &vertices = *(AODEvent()->GetVertices()); Int_t jVertices=0; // Add primary vertex. The primary tracks will be defined // after the loops on the composite objects (V0, cascades, kinks) fEvent->GetPrimaryVertex()->GetXYZ(pos); Float_t chi = 0; if (esdevent){ esdevent->GetPrimaryVertex()->GetCovMatrix(covVtx); chi = esdevent->GetPrimaryVertex()->GetChi2toNDF(); } else if (aodevent){ aodevent->GetPrimaryVertex()->GetCovMatrix(covVtx); chi = aodevent->GetPrimaryVertex()->GetChi2perNDF();//Different from ESD? } AliAODVertex * primary = new(vertices[jVertices++]) AliAODVertex(pos, covVtx, chi, NULL, -1, AliAODVertex::kPrimary); primary->SetName(fEvent->GetPrimaryVertex()->GetName()); primary->SetTitle(fEvent->GetPrimaryVertex()->GetTitle()); } //_________________________________________ void AliAnalysisTaskEMCALClusterize::Init() { //Init analysis with configuration macro if available if(gROOT->LoadMacro(fConfigName) >=0){ printf("AliAnalysisTaksEMCALClusterize::Init() - Configure analysis with %s\n",fConfigName.Data()); AliAnalysisTaskEMCALClusterize *clus = (AliAnalysisTaskEMCALClusterize*)gInterpreter->ProcessLine("ConfigEMCALClusterize()"); fGeomName = clus->fGeomName; fLoadGeomMatrices = clus->fLoadGeomMatrices; fOCDBpath = clus->fOCDBpath; fAccessOCDB = clus->fAccessOCDB; fRecParam = clus->fRecParam; fJustUnfold = clus->fJustUnfold; fFillAODFile = clus->fFillAODFile; fRecoUtils = clus->fRecoUtils; fConfigName = clus->fConfigName; fMaxEvent = clus->fMaxEvent; fDoTrackMatching = clus->fDoTrackMatching; fOutputAODBranchName = clus->fOutputAODBranchName; for(Int_t i = 0; i < 12; i++) fGeomMatrix[i] = clus->fGeomMatrix[i] ; } // Init geometry, I do not like much to do it like this ... if(fImportGeometryFromFile && !gGeoManager) { printf("AliAnalysisTaskEMCALClusterize::Init() - Import geometry.root file\n"); TGeoManager::Import(Form("%s/geometry.root", fImportGeometryFilePath.Data())) ; // default need file "geometry.root" in local dir!!!! } } //_______________________________________________________ void AliAnalysisTaskEMCALClusterize::InitClusterization() { //Select clusterization/unfolding algorithm and set all the needed parameters if (fJustUnfold){ // init the unfolding afterburner delete fUnfolder; fUnfolder = new AliEMCALAfterBurnerUF(fRecParam->GetW0(),fRecParam->GetLocMaxCut(),fRecParam->GetMinECut()); return; } //First init the clusterizer delete fClusterizer; if (fRecParam->GetClusterizerFlag() == AliEMCALRecParam::kClusterizerv1) fClusterizer = new AliEMCALClusterizerv1 (fGeom, fCalibData, fPedestalData); else if(fRecParam->GetClusterizerFlag() == AliEMCALRecParam::kClusterizerv2) fClusterizer = new AliEMCALClusterizerv2(fGeom, fCalibData, fPedestalData); else if(fRecParam->GetClusterizerFlag() == AliEMCALRecParam::kClusterizerNxN){ fClusterizer = new AliEMCALClusterizerNxN(fGeom, fCalibData, fPedestalData); fClusterizer->SetNRowDiff(fRecParam->GetNRowDiff()); fClusterizer->SetNColDiff(fRecParam->GetNColDiff()); } else { AliFatal(Form("Clusterizer < %d > not available", fRecParam->GetClusterizerFlag())); } //Now set the parameters fClusterizer->SetECAClusteringThreshold( fRecParam->GetClusteringThreshold() ); fClusterizer->SetECALogWeight ( fRecParam->GetW0() ); fClusterizer->SetMinECut ( fRecParam->GetMinECut() ); fClusterizer->SetUnfolding ( fRecParam->GetUnfold() ); fClusterizer->SetECALocalMaxCut ( fRecParam->GetLocMaxCut() ); fClusterizer->SetTimeCut ( fRecParam->GetTimeCut() ); fClusterizer->SetTimeMin ( fRecParam->GetTimeMin() ); fClusterizer->SetTimeMax ( fRecParam->GetTimeMax() ); fClusterizer->SetInputCalibrated ( kTRUE ); fClusterizer->SetJustClusters ( kTRUE ); //In case of unfolding after clusterization is requested, set the corresponding parameters if(fRecParam->GetUnfold()){ Int_t i=0; for (i = 0; i < 8; i++) { fClusterizer->SetSSPars(i, fRecParam->GetSSPars(i)); }//end of loop over parameters for (i = 0; i < 3; i++) { fClusterizer->SetPar5 (i, fRecParam->GetPar5(i)); fClusterizer->SetPar6 (i, fRecParam->GetPar6(i)); }//end of loop over parameters fClusterizer->InitClusterUnfolding(); }// to unfold } //_________________________________________________ void AliAnalysisTaskEMCALClusterize::InitGeometry() { // Set the geometry matrix, for the first event, skip the rest // Also set once the run dependent calibrations if(!fGeomMatrixSet){ if(fLoadGeomMatrices){ for(Int_t mod=0; mod < (fGeom->GetEMCGeometry())->GetNumberOfSuperModules(); mod++){ if(fGeomMatrix[mod]){ if(DebugLevel() > 1) fGeomMatrix[mod]->Print(); fGeom->SetMisalMatrix(fGeomMatrix[mod],mod) ; } fGeomMatrixSet=kTRUE; }//SM loop }//Load matrices else if(!gGeoManager){ printf("AliAnalysisTaksEMCALClusterize::InitGeometry() - Get geo matrices from data"); //Still not implemented in AOD, just a workaround to be able to work at least with ESDs if(!strcmp(fEvent->GetName(),"AliAODEvent")) { if(DebugLevel() > 1) Warning("UserExec","Use ideal geometry, values geometry matrix not kept in AODs."); }//AOD else { if(DebugLevel() > 1) printf("AliAnalysisTaksEMCALClusterize::InitGeometry() - AliAnalysisTaskEMCALClusterize Load Misaligned matrices."); AliESDEvent* esd = dynamic_cast(fEvent) ; if(!esd) { Error("InitGeometry"," - This event does not contain ESDs?"); AliAnalysisManager::GetAnalysisManager()->GetOutputEventHandler()->SetFillAOD(kFALSE); return; } for(Int_t mod=0; mod < (fGeom->GetEMCGeometry())->GetNumberOfSuperModules(); mod++){ if(DebugLevel() > 1) esd->GetEMCALMatrix(mod)->Print(); if(esd->GetEMCALMatrix(mod)) fGeom->SetMisalMatrix(esd->GetEMCALMatrix(mod),mod) ; } fGeomMatrixSet=kTRUE; }//ESD }//Load matrices from Data //Recover time dependent corrections, put then in recalibration histograms. Do it once fRecoUtils->SetRunDependentCorrections(InputEvent()->GetRunNumber()); }//first event } //____________________________________________________ Bool_t AliAnalysisTaskEMCALClusterize::IsExoticEvent() { // Check if event is exotic, get an exotic cell and compare with triggered patch // If there is a match remove event ... to be completed, filled with something provisional if(!fRemoveExoticEvents) return kFALSE; // Loop on cells AliVCaloCells * cells = fEvent->GetEMCALCells(); Float_t totCellE = 0; Int_t bc = InputEvent()->GetBunchCrossNumber(); for(Int_t icell = 0; icell < cells->GetNumberOfCells(); icell++){ Float_t ecell = 0 ; Double_t tcell = 0 ; Int_t absID = cells->GetCellNumber(icell); Bool_t accept = fRecoUtils->AcceptCalibrateCell(absID,bc,ecell,tcell,cells); if(accept && !fRecoUtils->IsExoticCell(absID,cells,bc)) totCellE += ecell; } // TString triggerclasses = ""; // if(esdevent) triggerclasses = esdevent ->GetFiredTriggerClasses(); // else triggerclasses = ((AliAODEvent*)event)->GetFiredTriggerClasses(); // // // printf("AliAnalysisTaskEMCALClusterize - reject event %d with cluster - reject event with ncells in SM3 %d and SM4 %d\n",(Int_t)Entry(),ncellsSM3, ncellsSM4); // AliAnalysisManager::GetAnalysisManager()->GetOutputEventHandler()->SetFillAOD(kFALSE); // return; // //printf("TotE cell %f\n",totCellE); if(totCellE < 1) return kTRUE; return kFALSE; } //_________________________________________________ Bool_t AliAnalysisTaskEMCALClusterize::IsLEDEvent() { //Check if event is LED if(!fRemoveLEDEvents) return kFALSE; // Count number of cells with energy larger than 0.1 in SM3, cut on this number Int_t ncellsSM3 = 0; Int_t ncellsSM4 = 0; AliVCaloCells * cells = fEvent->GetEMCALCells(); for(Int_t icell = 0; icell < cells->GetNumberOfCells(); icell++){ if(cells->GetAmplitude(icell) > 0.1 && cells->GetCellNumber(icell)/(24*48)==3) ncellsSM3++; if(cells->GetAmplitude(icell) > 0.1 && cells->GetCellNumber(icell)/(24*48)==4) ncellsSM4++; } TString triggerclasses = ""; AliESDEvent *esdevent = dynamic_cast(fEvent); if(esdevent) triggerclasses = esdevent ->GetFiredTriggerClasses(); else triggerclasses = ((AliAODEvent*)fEvent)->GetFiredTriggerClasses(); Int_t ncellcut = 21; if(triggerclasses.Contains("EMC")) ncellcut = 35; if( ncellsSM3 >= ncellcut || ncellsSM4 >= 100 ){ printf("AliAnalysisTaksEMCALClusterize::IsLEDEvent() - reject event %d with ncells in SM3 %d and SM4 %d\n",(Int_t)Entry(),ncellsSM3, ncellsSM4); AliAnalysisManager::GetAnalysisManager()->GetOutputEventHandler()->SetFillAOD(kFALSE); return kTRUE; } return kFALSE; } //______________________________________________________________________________ void AliAnalysisTaskEMCALClusterize::RecPoints2Clusters(TClonesArray *digitsArr, TObjArray *recPoints, TObjArray *clusArray) { // Restore clusters from recPoints // Cluster energy, global position, cells and their amplitude fractions are restored Int_t j = 0; for(Int_t i = 0; i < recPoints->GetEntriesFast(); i++) { AliEMCALRecPoint *recPoint = (AliEMCALRecPoint*) recPoints->At(i); const Int_t ncells = recPoint->GetMultiplicity(); Int_t ncellsTrue = 0; if(recPoint->GetEnergy() < fRecParam->GetClusteringThreshold()) continue; // cells and their amplitude fractions UShort_t absIds[ncells]; Double32_t ratios[ncells]; //For later check embedding AliAODInputHandler* aodIH = dynamic_cast((AliAnalysisManager::GetAnalysisManager())->GetInputEventHandler()); Float_t clusterE = 0; for (Int_t c = 0; c < ncells; c++) { AliEMCALDigit *digit = (AliEMCALDigit*) digitsArr->At(recPoint->GetDigitsList()[c]); absIds[ncellsTrue] = digit->GetId(); ratios[ncellsTrue] = recPoint->GetEnergiesList()[c]/digit->GetAmplitude(); // In case of unfolding, remove digits with unfolded energy too low if(fSelectCell){ if (recPoint->GetEnergiesList()[c] < fSelectCellMinE || ratios[ncellsTrue] < fSelectCellMinFrac) { if(DebugLevel() > 1) { printf("AliAnalysisTaksEMCALClusterize::RecPoints2Clusters() - Too small energy in cell of cluster: cluster cell %f, digit %f\n", recPoint->GetEnergiesList()[c],digit->GetAmplitude()); } continue; } // if cuts }// Select cells //Recalculate cluster energy and number of cluster cells in case any of the cells was rejected ncellsTrue++; clusterE +=recPoint->GetEnergiesList()[c]; // In case of embedding, fill ratio with amount of signal, if (aodIH && aodIH->GetMergeEvents()) { //AliVCaloCells* inEMCALCells = InputEvent()->GetEMCALCells(); AliVCaloCells* meEMCALCells = aodIH->GetEventToMerge()->GetEMCALCells(); AliVCaloCells* ouEMCALCells = AODEvent()->GetEMCALCells(); Float_t sigAmplitude = meEMCALCells->GetCellAmplitude(absIds[ncellsTrue]); //Float_t bkgAmplitude = inEMCALCells->GetCellAmplitude(absIds[ncellsTrue]); Float_t sumAmplitude = ouEMCALCells->GetCellAmplitude(absIds[ncellsTrue]); //printf("\t AbsID %d, amplitude : bkg %f, sigAmplitude %f, summed %f - %f\n",absIds[ncellsTrue], bkgAmplitude, sigAmplitude, sumAmplitude, digit->GetAmplitude()); if(sumAmplitude > 0) ratios[ncellsTrue] = sigAmplitude/sumAmplitude; //printf("\t \t ratio %f\n",ratios[ncellsTrue]); }//Embedding }// cluster cell loop if (ncellsTrue < 1) { if (DebugLevel() > 1) printf("AliAnalysisTaskEMCALClusterize::RecPoints2Clusters() - Skipping cluster with no cells avobe threshold E = %f, ncells %d\n", recPoint->GetEnergy(), ncells); continue; } //if(ncellsTrue != ncells) printf("Old E %f, ncells %d; New E %f, ncells %d\n",recPoint->GetEnergy(),ncells,clusterE,ncellsTrue); if(clusterE < fRecParam->GetClusteringThreshold()) { if (DebugLevel()>1) printf("AliAnalysisTaskEMCALClusterize::RecPoints2Clusters() - Remove cluster with energy below seed threshold %f\n",clusterE); continue; } TVector3 gpos; Float_t g[3]; // calculate new cluster position recPoint->EvalGlobalPosition(fRecParam->GetW0(), digitsArr); recPoint->GetGlobalPosition(gpos); gpos.GetXYZ(g); // create a new cluster (*clusArray)[j] = new AliAODCaloCluster() ; AliAODCaloCluster *clus = dynamic_cast( clusArray->At(j) ) ; j++; clus->SetType(AliVCluster::kEMCALClusterv1); clus->SetE(clusterE); clus->SetPosition(g); clus->SetNCells(ncellsTrue); clus->SetCellsAbsId(absIds); clus->SetCellsAmplitudeFraction(ratios); clus->SetChi2(-1); //not yet implemented clus->SetTOF(recPoint->GetTime()) ; //time-of-flight clus->SetNExMax(recPoint->GetNExMax()); //number of local maxima clus->SetDistanceToBadChannel(recPoint->GetDistanceToBadTower()); if(ncells == ncellsTrue){ Float_t elipAxis[2]; recPoint->GetElipsAxis(elipAxis); clus->SetM02(elipAxis[0]*elipAxis[0]) ; clus->SetM20(elipAxis[1]*elipAxis[1]) ; clus->SetDispersion(recPoint->GetDispersion()); } else if(fSelectCell){ // In case some cells rejected, in unfolding case, recalculate // shower shape parameters and position AliVCaloCells* cells = 0x0; if (aodIH && aodIH->GetMergeEvents()) cells = AODEvent() ->GetEMCALCells(); else cells = InputEvent()->GetEMCALCells(); fRecoUtils->RecalculateClusterShowerShapeParameters(fGeom,cells,clus); fRecoUtils->RecalculateClusterPID(clus); fRecoUtils->RecalculateClusterPosition(fGeom,cells,clus); } //MC Int_t parentMult = 0; Int_t *parentList = recPoint->GetParents(parentMult); clus->SetLabel(parentList, parentMult); //Write the second major contributor to each MC cluster. Int_t iNewLabel ; for ( Int_t iLoopCell = 0 ; iLoopCell < clus->GetNCells() ; iLoopCell++ ){ Int_t idCell = clus->GetCellAbsId(iLoopCell) ; if(idCell>=0){ iNewLabel = 1 ; //iNewLabel makes sure we don't write twice the same label. for ( UInt_t iLoopLabels = 0 ; iLoopLabels < clus->GetNLabels() ; iLoopLabels++ ) { if ( fCellSecondLabels[idCell] == -1 ) iNewLabel = 0; // -1 is never a good second label. if ( fCellSecondLabels[idCell] == clus->GetLabelAt(iLoopLabels) ) iNewLabel = 0; } if (iNewLabel == 1) { Int_t * newLabelArray = new Int_t[clus->GetNLabels()+1] ; for ( UInt_t iLoopNewLabels = 0 ; iLoopNewLabels < clus->GetNLabels() ; iLoopNewLabels++ ){ newLabelArray[iLoopNewLabels] = clus->GetLabelAt(iLoopNewLabels) ; } newLabelArray[clus->GetNLabels()] = fCellSecondLabels[idCell] ; clus->SetLabel(newLabelArray,clus->GetNLabels()+1) ; delete [] newLabelArray; } }//positive cell number } } // recPoints loop } //____________________________________________________________ void AliAnalysisTaskEMCALClusterize::UserCreateOutputObjects() { // Init geometry, create list of output clusters fGeom = AliEMCALGeometry::GetInstance(fGeomName) ; if(fOutputAODBranchName.Length()!=0){ fOutputAODBranch = new TClonesArray("AliAODCaloCluster", 0); fOutputAODBranch->SetName(fOutputAODBranchName); //fOutputAODBranch->SetOwner(kFALSE); AddAODBranch("TClonesArray", &fOutputAODBranch); } else { AliFatal("fOutputAODBranchName not set\n"); } //PostData(0,fOutputAODBranch); } //_______________________________________________________ void AliAnalysisTaskEMCALClusterize::UserExec(Option_t *) { // Main loop // Called for each event //Remove the contents of output list set in the previous event fOutputAODBranch->Clear("C"); LoadBranches(); //Get the event, do some checks and settings CheckAndGetEvent() ; if (!fEvent) { if(DebugLevel() > 0 ) printf("AliAnalysisTaksEMCALClusterize::UserExec() - Skip Event %d", (Int_t) Entry()); return ; } //Init pointers, clusterizer, ocdb if(fAccessOCDB) AccessOCDB(); InitClusterization(); InitGeometry(); // only once // Make clusters if (fJustUnfold) ClusterUnfolding(); else ClusterizeCells() ; //Recalculate track-matching for the new clusters if(fDoTrackMatching) fRecoUtils->FindMatches(fEvent,fCaloClusterArr,fGeom); //Put the new clusters in the AOD list Int_t kNumberOfCaloClusters = fCaloClusterArr->GetEntriesFast(); for(Int_t i = 0; i < kNumberOfCaloClusters; i++){ AliAODCaloCluster *newCluster = (AliAODCaloCluster *) fCaloClusterArr->At(i); //Add matched track if(fDoTrackMatching){ Int_t trackIndex = fRecoUtils->GetMatchedTrackIndex(i); if(trackIndex >= 0){ newCluster->AddTrackMatched(fEvent->GetTrack(trackIndex)); if(DebugLevel() > 1) printf("AliAnalysisTaksEMCALClusterize::UserExec() - Matched Track index %d to new cluster %d \n",trackIndex,i); } Float_t dR = 999., dZ = 999.; fRecoUtils->GetMatchedResiduals(newCluster->GetID(),dR,dZ); newCluster->SetTrackDistance(dR,dZ); } else {// Assign previously assigned matched track in reco, very very rough Int_t absId0 = newCluster->GetCellsAbsId()[0]; // Assign match of first cell in cluster newCluster->SetTrackDistance(fCellMatchdPhi[absId0],fCellMatchdEta[absId0]); } //printf("New cluster E %f, Time %e, Id = ", newCluster->E(), newCluster->GetTOF() ); //for(Int_t icell=0; icell < newCluster->GetNCells(); icell++ ) printf(" %d,", newCluster->GetCellsAbsId() [icell] ); //printf("\n"); //In case of new bad channels, recalculate distance to bad channels if(fRecoUtils->IsBadChannelsRemovalSwitchedOn()) fRecoUtils->RecalculateClusterDistanceToBadChannel(fGeom, fEvent->GetEMCALCells(), newCluster); newCluster->SetID(i); new((*fOutputAODBranch)[i]) AliAODCaloCluster(*newCluster); if(DebugLevel() > 1 ) printf("AliAnalysisTaksEMCALClusterize::UserExec() - New cluster %d of %d, energy %f\n",newCluster->GetID(), kNumberOfCaloClusters, newCluster->E()); } // cluster loop fOutputAODBranch->Expand(kNumberOfCaloClusters); // resize TObjArray to 'remove' slots // Clean up fCaloClusterArr->Delete(); // Do not Clear(), it leaks, why? //PostData(0,fOutputAODBranch); }