// Base class for the clusterization algorithm (pure abstract)
//*--
//*-- Author: Yves Schutz SUBATECH
+//
+// Clusterization mother class. Contains common methods/data members of different
+// clusterizers. GCB 2010
//////////////////////////////////////////////////////////////////////////////
// --- ROOT system ---
-
-#include "TGeometry.h"
-#include "TDirectory.h"
-#include "TFile.h"
-#include "TTree.h"
+#include <TTree.h>
+#include <TFile.h>
+class TFolder;
+#include <TMath.h>
+#include <TMinuit.h>
+#include <TTree.h>
+class TSystem;
+#include <TBenchmark.h>
+#include <TBrowser.h>
+#include <TROOT.h>
// --- Standard library ---
-
-#include <iostream.h>
-#include <stdlib.h>
+#include <cassert>
// --- AliRoot header files ---
-
-#include "AliRun.h"
#include "AliEMCALClusterizer.h"
-#include "AliHeader.h"
-#include "AliEMCALGetter.h"
-#include "AliEMCALSDigitizer.h"
-#include "AliEMCALDigitizer.h"
+#include "AliEMCALReconstructor.h"
+#include "AliRunLoader.h"
+#include "AliRun.h"
+#include "AliLog.h"
+#include "AliEMCAL.h"
+#include "AliEMCALRecPoint.h"
+#include "AliEMCALRecParam.h"
+#include "AliEMCALGeometry.h"
+#include "AliEMCALRecParam.h"
+#include "AliCDBManager.h"
+#include "AliCaloCalibPedestal.h"
+#include "AliEMCALCalibData.h"
+class AliCDBStorage;
+#include "AliCDBEntry.h"
ClassImp(AliEMCALClusterizer)
//____________________________________________________________________________
- AliEMCALClusterizer::AliEMCALClusterizer():TTask("","")
+AliEMCALClusterizer::AliEMCALClusterizer():
+ fIsInputCalibrated(kFALSE),
+ fJustClusters(kFALSE),
+ fDigitsArr(NULL),
+ fTreeR(NULL),
+ fRecPoints(NULL),
+ fGeom(NULL),
+ fCalibData(NULL),
+ fCaloPed(NULL),
+ fADCchannelECA(0.),fADCpedestalECA(0.), fTimeECA(0.),
+ fTimeMin(-1.),fTimeMax(1.),fTimeCut(1.),
+ fDefaultInit(kFALSE),fToUnfold(kFALSE),
+ fNumberOfECAClusters(0), fECAClusteringThreshold(0.),
+ fECALocMaxCut(0.),fECAW0(0.),fMinECut(0.),fRejectBelowThreshold(0),
+ fClusterUnfolding(NULL)
{
// ctor
- fSplitFile = 0 ;
- fHitsFileName = "" ;
- fSDigitsFileName = "" ;
- fDigitsFileName = "" ;
+
+ Init();
+}
+//____________________________________________________________________________
+AliEMCALClusterizer::AliEMCALClusterizer(AliEMCALGeometry* geometry):
+ fIsInputCalibrated(kFALSE),
+ fJustClusters(kFALSE),
+ fDigitsArr(NULL),
+ fTreeR(NULL),
+ fRecPoints(NULL),
+ fGeom(geometry),
+ fCalibData(NULL),
+ fCaloPed(NULL),
+ fADCchannelECA(0.),fADCpedestalECA(0.), fTimeECA(0.),
+ fTimeMin(-1.),fTimeMax(1.),fTimeCut(1.),
+ fDefaultInit(kFALSE),fToUnfold(kFALSE),
+ fNumberOfECAClusters(0), fECAClusteringThreshold(0.),
+ fECALocMaxCut(0.),fECAW0(0.),fMinECut(0.),fRejectBelowThreshold(0),
+ fClusterUnfolding(NULL)
+{
+ // Ctor with the indication of the file where header Tree and digits Tree are stored.
+ // Use this contructor to avoid usage of Init() which uses runloader.
+ // Change needed by HLT - MP.
+ // Note for the future: the use on runloader should be avoided or optional at least.
+ // Another way is to make Init virtual and protected at least
+ // such that the deriving classes can overload Init();
+
+ if (!fGeom)
+ {
+ AliFatal("Geometry not initialized.");
+ }
+ Int_t i=0;
+ for (i = 0; i < 8; i++)
+ fSSPars[i] = 0.;
+ for (i = 0; i < 3; i++) {
+ fPar5[i] = 0.;
+ fPar6[i] = 0.;
+ }
}
//____________________________________________________________________________
-AliEMCALClusterizer::AliEMCALClusterizer(const char* headerFile, const char* name):
-TTask(name, headerFile)
+AliEMCALClusterizer::AliEMCALClusterizer(AliEMCALGeometry *geometry,
+ AliEMCALCalibData *calib,
+ AliCaloCalibPedestal *caloped):
+ fIsInputCalibrated(kFALSE),
+ fJustClusters(kFALSE),
+ fDigitsArr(NULL),
+ fTreeR(NULL),
+ fRecPoints(NULL),
+ fGeom(geometry),
+ fCalibData(calib),
+ fCaloPed(caloped),
+ fADCchannelECA(0.),fADCpedestalECA(0.), fTimeECA(0.),
+ fTimeMin(-1.),fTimeMax(1.),fTimeCut(1.),
+ fDefaultInit(kFALSE),fToUnfold(kFALSE),
+ fNumberOfECAClusters(0), fECAClusteringThreshold(0.),
+ fECALocMaxCut(0.),fECAW0(0.),fMinECut(0.),fRejectBelowThreshold(0),
+ fClusterUnfolding(NULL)
{
- // ctor
- fSplitFile = 0 ;
- fDigitsFileName = headerFile ;
- AliEMCALGetter * gime = AliEMCALGetter::GetInstance(headerFile, name) ;
- gime->Event(0,"D") ;
- fSDigitsFileName = gime->Digitizer()->GetTitle() ;
- gime = AliEMCALGetter::GetInstance(fSDigitsFileName, name) ;
- gime->Event(0,"S") ;
- fHitsFileName = gime->SDigitizer()->GetTitle() ;
+ // ctor, geometry and calibration are initialized elsewhere.
+
+ if (!fGeom)
+ AliFatal("Geometry not initialized.");
+
+ Int_t i=0;
+ for (i = 0; i < 8; i++)
+ fSSPars[i] = 0.;
+ for (i = 0; i < 3; i++) {
+ fPar5[i] = 0.;
+ fPar6[i] = 0.;
+ }
}
//____________________________________________________________________________
AliEMCALClusterizer::~AliEMCALClusterizer()
{
// dtor
+ //Already deleted in AliEMCALReconstructor.
+
+ if(fClusterUnfolding) delete fClusterUnfolding;
+
+ // make sure we delete the rec points array
+ DeleteRecPoints();
+
+ //Delete digits array
+ DeleteDigits();
+
+}
+
+//____________________________________________________________________________
+void AliEMCALClusterizer::DeleteRecPoints()
+{
+ // free the cluster array
+ if (fRecPoints)
+ {
+ AliDebug(2, "Deleting fRecPoints.");
+ fRecPoints->Delete();
+ delete fRecPoints;
+ fRecPoints = 0;
+ }
+}
+
+//____________________________________________________________________________
+void AliEMCALClusterizer::DeleteDigits()
+{
+ // free the digits array
+ if (fDigitsArr)
+ {
+ AliDebug(2, "Deleting fDigitsArr.");
+ fDigitsArr->Clear("C");
+ delete fDigitsArr;
+ fDigitsArr = 0;
+ }
+}
+
+//____________________________________________________________________________
+void AliEMCALClusterizer::Calibrate(Float_t & amp, Float_t & time, const Int_t absId)
+{
+ // Convert digitized amplitude into energy, calibrate time
+ // Calibration parameters are taken from OCDB : OCDB/EMCAL/Calib/Data
+
+ //Return energy with default parameters if calibration is not available
+ if (!fCalibData && !fCaloPed) {
+ if (fIsInputCalibrated == kTRUE)
+ {
+ AliDebug(10, Form("Input already calibrated!"));
+ return ;
+ }
+ else{
+ AliFatal("OCDB calibration and bad map parameters are not available");
+ return;
+ }
+ }
+
+ if (fGeom==0)
+ AliFatal("Did not get geometry from EMCALLoader") ;
+
+ Int_t iSupMod = -1;
+ Int_t nModule = -1;
+ Int_t nIphi = -1;
+ Int_t nIeta = -1;
+ Int_t iphi = -1;
+ Int_t ieta = -1;
+
+ Bool_t bCell = fGeom->GetCellIndex(absId, iSupMod, nModule, nIphi, nIeta) ;
+ if(!bCell) {
+ fGeom->PrintGeometry();
+ AliError(Form("Wrong cell id number : %i", absId));
+ //assert(0); // GCB: This aborts reconstruction of raw simulations
+ //where simulation had more SM than default geometry,
+ //change to return 0, to avoid aborting good generations.
+ amp = 0;
+ time = 0;
+ return ;
+ }
+
+ fGeom->GetCellPhiEtaIndexInSModule(iSupMod,nModule,nIphi, nIeta,iphi,ieta);
+
+ // Check if channel is bad (dead or hot), in this case return 0.
+ // Gustavo: 15-12-09 In case of RAW data this selection is already done, but not in simulation.
+ // for the moment keep it here but remember to do the selection at the sdigitizer level
+ // and remove it from here
+ if (fCaloPed) {
+ Int_t channelStatus = (Int_t)(fCaloPed->GetDeadMap(iSupMod))->GetBinContent(ieta,iphi);
+ if(channelStatus == AliCaloCalibPedestal::kHot || channelStatus == AliCaloCalibPedestal::kDead) {
+ AliDebug(2,Form("Tower from SM %d, ieta %d, iphi %d is BAD : status %d !!!",iSupMod,ieta,iphi, channelStatus));
+ amp = 0 ;
+ time = 0 ;
+ return ;
+ }
+ }
+
+ if (fIsInputCalibrated || !fCalibData)
+ {
+ AliDebug(10, Form("Input already calibrated!"));
+ return ;
+ }
+
+ Int_t bc = 0; // Get somehow the bunch crossing number
+
+ fADCchannelECA = fCalibData->GetADCchannel (iSupMod,ieta,iphi);
+ fADCpedestalECA = fCalibData->GetADCpedestal(iSupMod,ieta,iphi);
+ fTimeECA = fCalibData->GetTimeChannel(iSupMod,ieta,iphi, bc);
+
+ time -= fTimeECA ;
+ amp = amp * fADCchannelECA - fADCpedestalECA ;
- fSplitFile = 0 ;
-
}
//____________________________________________________________________________
-void AliEMCALClusterizer::SetSplitFile(const TString splitFileName)
+void AliEMCALClusterizer::GetCalibrationParameters()
{
- // Diverts the Digits in a file separate from the hits file
+ // Set calibration parameters:
+ // If calibration database exists, they are read from database,
+ // otherwise, they are taken from digitizer.
+ // It is a user responsilibity to open CDB before reconstruction,
+ // for example:
+ // AliCDBStorage* storage = AliCDBManager::Instance()->GetStorage("local://CalibDB");
+
+ if (fIsInputCalibrated)
+ return;
+
+ //Check if calibration is stored in data base
+ 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!");
+}
+
+//____________________________________________________________________________
+void AliEMCALClusterizer::GetCaloCalibPedestal()
+{
+ // Set calibration parameters:
+ // if calibration database exists, they are read from database,
+ // otherwise, they are taken from digitizer.
+ //
+ // It is a user responsilibity to open CDB before reconstruction,
+ // for example:
+ // AliCDBStorage* storage = AliCDBManager::Instance()->GetStorage("local://CalibDB");
+
+ if (fIsInputCalibrated)
+ return;
+
+ // Check if calibration is stored in data base
+ if(!fCaloPed)
+ {
+ AliCDBEntry *entry = (AliCDBEntry*)
+ AliCDBManager::Instance()->Get("EMCAL/Calib/Pedestals");
+ if (entry) fCaloPed = (AliCaloCalibPedestal*) entry->GetObject();
+ }
+
+ if(!fCaloPed)
+ AliFatal("Pedestal info not found in CDB!");
+}
+
+//____________________________________________________________________________
+void AliEMCALClusterizer::Init()
+{
+ // Make all memory allocations which can not be done in default constructor.
+ // Attach the Clusterizer task to the list of EMCAL tasks
+
+ AliRunLoader *rl = AliRunLoader::Instance();
+ if (rl->GetAliRun()){
+ AliEMCAL* emcal = dynamic_cast<AliEMCAL*>(rl->GetAliRun()->GetDetector("EMCAL"));
+ if(emcal)fGeom = emcal->GetGeometry();
+ }
+
+ if(!fGeom){
+ fGeom = AliEMCALGeometry::GetInstance(AliEMCALGeometry::GetDefaultGeometryName());
+ }
+
+ AliDebug(1,Form("geom %p",fGeom));
+ if(!gMinuit)
+ gMinuit = new TMinuit(100) ;
+
+ Int_t i=0;
+ for (i = 0; i < 8; i++)
+ fSSPars[i] = 0.;
+ for (i = 0; i < 3; i++) {
+ fPar5[i] = 0.;
+ fPar6[i] = 0.;
+ }
+}
+
+//____________________________________________________________________________
+void AliEMCALClusterizer::InitParameters()
+{
+ // Initializes the parameters for the Clusterizer from AliEMCALReconstructor::GetRecParam().
+
+ return InitParameters(AliEMCALReconstructor::GetRecParam());
+}
+
+//____________________________________________________________________________
+void AliEMCALClusterizer::InitParameters(const AliEMCALRecParam* recParam)
+{
+ // Initializes the parameters for the Clusterizer
+
+ fNumberOfECAClusters = 0 ;
+ fCalibData = 0 ;
+ fCaloPed = 0 ;
+
+ if(!recParam) {
+ AliFatal("Reconstruction parameters for EMCAL not set!");
+ }
- TDirectory * cwd = gDirectory ;
- fSplitFile = gAlice->InitTreeFile("R",splitFileName.Data());
- fSplitFile->cd() ;
- gAlice->Write(0, TObject::kOverwrite);
+ fECAClusteringThreshold = recParam->GetClusteringThreshold();
+ fECAW0 = recParam->GetW0();
+ fMinECut = recParam->GetMinECut();
+ fRejectBelowThreshold = recParam->GetRejectBelowThreshold();
+ fToUnfold = recParam->GetUnfold();
+ fECALocMaxCut = recParam->GetLocMaxCut();
+ fTimeCut = recParam->GetTimeCut();
+ fTimeMin = recParam->GetTimeMin();
+ fTimeMax = recParam->GetTimeMax();
- TTree *treeE = gAlice->TreeE();
- if (!treeE) {
- cerr << "ERROR: AliEMCALClusterizer::SetSplitFile -> No TreeE found "<<endl;
- abort() ;
- }
+ //For NxN
+ SetNRowDiff(recParam->GetNRowDiff());
+ SetNColDiff(recParam->GetNColDiff());
- // copy TreeE
- AliHeader *header = new AliHeader();
- treeE->SetBranchAddress("Header", &header);
- treeE->SetBranchStatus("*",1);
- TTree *treeENew = treeE->CloneTree();
- treeENew->Write(0, TObject::kOverwrite);
+ AliDebug(1,Form("Reconstruction parameters: fECAClusteringThreshold=%.3f GeV, fECAW=%.3f, fMinECut=%.3f GeV, "
+ "fToUnfold=%d, fECALocMaxCut=%.3f GeV, fTimeCut=%e s,fTimeMin=%e s,fTimeMax=%e s,fRejectBelowThreshold=%d",
+ fECAClusteringThreshold,fECAW0,fMinECut,fToUnfold,fECALocMaxCut,fTimeCut, fTimeMin, fTimeMax,
+ fRejectBelowThreshold));
+
+ if (fToUnfold) {
+ Int_t i=0;
+ for (i = 0; i < 8; i++) {
+ fSSPars[i] = recParam->GetSSPars(i);
+ } //end of loop over parameters
+ for (i = 0; i < 3; i++) {
+ fPar5[i] = recParam->GetPar5(i);
+ fPar6[i] = recParam->GetPar6(i);
+ } //end of loop over parameters
+
+ InitClusterUnfolding();
+
+ for (i = 0; i < 8; i++) {
+ AliDebug(1,Form("unfolding shower shape parameters: fSSPars=%f \n",fSSPars[i]));
+ }
+ for (i = 0; i < 3; i++) {
+ AliDebug(1,Form("unfolding parameter 5: fPar5=%f \n",fPar5[i]));
+ AliDebug(1,Form("unfolding parameter 6: fPar6=%f \n",fPar6[i]));
+ }
+ } // to unfold
+}
+
+//____________________________________________________________________________
+void AliEMCALClusterizer::Print(Option_t * /*option*/)const
+{
+ // Print clusterizer parameters
+
+ TString message("\n") ;
+
+ if (strcmp(GetName(),"") == 0) {
+ printf("AliEMCALClusterizer not initialized\n");
+ return;
+ }
+
+ // Print parameters
+ TString taskName(Version()) ;
+
+ printf("--------------- ");
+ printf("%s",taskName.Data()) ;
+ printf(" ");
+ printf("Clusterizing digits: ");
+ printf("\n ECA Local Maximum cut = %f", fECALocMaxCut);
+ printf("\n ECA Logarithmic weight = %f", fECAW0);
+ if (fToUnfold) {
+ printf("\nUnfolding on\n");
+ printf("Unfolding parameters: fSSpars: \n");
+ Int_t i=0;
+ for (i = 0; i < 8; i++) {
+ printf("fSSPars[%d] = %f \n", i, fSSPars[i]);
+ }
+ printf("Unfolding parameter 5 and 6: fPar5 and fPar6: \n");
+ for (i = 0; i < 3; i++) {
+ printf("fPar5[%d] = %f \n", i, fPar5[i]);
+ printf("fPar6[%d] = %f \n", i, fPar6[i]);
+ }
+ }
+ else
+ printf("\nUnfolding off\n");
+
+ printf("------------------------------------------------------------------");
+}
+
+//____________________________________________________________________________
+void AliEMCALClusterizer::PrintRecPoints(Option_t * option)
+{
+ // Prints list of RecPoints produced at the current pass of AliEMCALClusterizer
+
+ if (strstr(option,"deb")) {
+ printf("PrintRecPoints: Clusterization result:") ;
+ printf(" Found %d ECA Rec Points\n ",
+ fRecPoints->GetEntriesFast()) ;
+ }
- // copy AliceGeom
- TGeometry *AliceGeom = static_cast<TGeometry*>(cwd->Get("AliceGeom"));
- if (!AliceGeom) {
- cerr << "ERROR: AliEMCALClusterizer::SetSplitFile -> AliceGeom was not found in the input file "<<endl;
- abort() ;
+ if (strstr(option,"all")) {
+ if (strstr(option,"deb")) {
+ printf("\n-----------------------------------------------------------------------\n") ;
+ printf("Clusters in ECAL section\n") ;
+ printf("Index Ene(GeV) Multi Module GX GY GZ lX lY lZ Dispersion Lambda 1 Lambda 2 # of prim Primaries list\n") ;
+ }
+ Int_t index;
+ for (index = 0 ; index < fRecPoints->GetEntries() ; index++) {
+ AliEMCALRecPoint * rp = dynamic_cast<AliEMCALRecPoint * >(fRecPoints->At(index)) ;
+ if (!rp)
+ continue;
+
+ TVector3 globalpos;
+ //rp->GetGlobalPosition(globalpos);
+ TVector3 localpos;
+ rp->GetLocalPosition(localpos);
+ Float_t lambda[2];
+ rp->GetElipsAxis(lambda);
+
+ Int_t nprimaries=0;
+ Int_t * primaries = rp->GetPrimaries(nprimaries);
+ if(strstr(option,"deb"))
+ printf("\n%6d %8.4f %3d %4.1f %4.1f %4.1f %4.1f %4.1f %4.1f %4.1f %4f %4f %2d : ",
+ rp->GetIndexInList(), rp->GetEnergy(), rp->GetMultiplicity(),
+ globalpos.X(), globalpos.Y(), globalpos.Z(), localpos.X(), localpos.Y(), localpos.Z(),
+ rp->GetDispersion(), lambda[0], lambda[1], nprimaries) ;
+ if(strstr(option,"deb")){
+ for (Int_t iprimary=0; iprimary<nprimaries; iprimary++) {
+ printf("%d ", primaries[iprimary] ) ;
+ }
+ }
+ }
+
+ if(strstr(option,"deb"))
+ printf("\n-----------------------------------------------------------------------\n");
+ }
+}
+
+//___________________________________________________________________
+void AliEMCALClusterizer::PrintRecoInfo()
+{
+ // Print reco version
+
+ printf(" AliEMCALClusterizer::PrintRecoInfo() : version %s \n", Version() );
+}
+
+//____________________________________________________________________________
+void AliEMCALClusterizer::SetInput(TTree *digitsTree)
+{
+ // Read the digits from the input tree
+
+ TBranch *branch = digitsTree->GetBranch("EMCAL");
+ if (!branch)
+ {
+ AliError("can't get the branch with the EMCAL digits !");
+ return;
}
- AliceGeom->Write(0, TObject::kOverwrite);
- gAlice->MakeTree("R", fSplitFile);
- cwd->cd() ;
- cout << "INFO: AliEMCALClusterizer::SetSPlitMode -> RecPoints will be stored in " << splitFileName.Data() << endl ;
+ if (!fDigitsArr) fDigitsArr = new TClonesArray("AliEMCALDigit",100);
+ else fDigitsArr->Clear("C"); // avoid leak
+
+ branch->SetAddress(&fDigitsArr);
+ branch->GetEntry(0);
+}
+
+//____________________________________________________________________________
+void AliEMCALClusterizer::SetOutput(TTree *clustersTree)
+{
+ // Read the digits from the input tree
+
+ AliDebug(9, "Making array for EMCAL clusters");
+ fRecPoints = new TObjArray(1000);
+ if (clustersTree) {
+ fTreeR = clustersTree;
+ Int_t split = 0;
+ Int_t bufsize = 32000;
+ fTreeR->Branch("EMCALECARP", "TObjArray", &fRecPoints, bufsize, split);
+ }
+}
+
+//___________________________________________________________________
+void AliEMCALClusterizer::SetInputCalibrated(Bool_t val)
+{
+ // Flag to indicate that input is calibrated - the case when we run already on ESD
+
+ fIsInputCalibrated = val;
+}
+
+//___________________________________________________________________
+void AliEMCALClusterizer::SetJustClusters(Bool_t val)
+{
+ // Flag to indicate that we are running on ESDs, when calling
+ // rp->EvalAll(fECAW0,fDigitsArr,fJustClusters); in derived classes
+
+ fJustClusters = val;
}
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff