]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - EMCAL/AliEMCALReconstructor.cxx
Coverity 16153
[u/mrichter/AliRoot.git] / EMCAL / AliEMCALReconstructor.cxx
index a20a6a16a4f6c47ffc0160d77fefdeda24bc8078..bd3ff34f422df411ac4cf1cd6e1fb6722df8817f 100644 (file)
 /* $Id$ */
 
 //_________________________________________________________________________
-//*--
-//*-- Yves Schutz (SUBATECH) 
+//--
+//-- Yves Schutz (SUBATECH) 
 // Reconstruction class. Redesigned from the old AliReconstructionner class and 
 // derived from STEER/AliReconstructor. 
 // 
+
+
 // --- ROOT system ---
+#include <TClonesArray.h>
+#include "TGeoManager.h"
+#include "TGeoMatrix.h"
 
 // --- Standard library ---
 
 // --- AliRoot header files ---
 #include "AliEMCALReconstructor.h"
 
-#include "AliRun.h"
-#include "AliESD.h"
-#include "AliRunLoader.h"
+#include "AliCodeTimer.h"
+#include "AliCaloCalibPedestal.h"
+#include "AliEMCALCalibData.h"
+#include "AliESDEvent.h"
+#include "AliESDCaloCluster.h"
+#include "AliESDCaloCells.h"
+#include "AliESDtrack.h"
 #include "AliEMCALLoader.h"
 #include "AliEMCALRawUtils.h"
+#include "AliEMCALDigit.h"
 #include "AliEMCALClusterizerv1.h"
+#include "AliEMCALClusterizerNxN.h"
 #include "AliEMCALRecPoint.h"
 #include "AliEMCALPID.h"
 #include "AliEMCALTrigger.h"
 #include "AliRawReader.h"
-
-
-ClassImp(AliEMCALReconstructor)
-
+#include "AliCDBEntry.h"
+#include "AliCDBManager.h"
+#include "AliEMCALGeometry.h"
+#include "AliEMCAL.h"
+#include "AliESDVZERO.h"
+#include "AliCDBManager.h"
+#include "AliRunLoader.h"
+#include "AliRun.h"
+#include "AliEMCALTriggerData.h"
+#include "AliEMCALTriggerElectronics.h"
+#include "AliEMCALTriggerDCSConfigDB.h"
+#include "AliEMCALTriggerDCSConfig.h"
+#include "AliEMCALTriggerData.h"
+#include "AliEMCALTriggerRawDigit.h"
+#include "AliEMCALTriggerPatch.h"
+#include "AliEMCALTriggerTypes.h"
+
+ClassImp(AliEMCALReconstructor) 
+  
+ const AliEMCALRecParam*     AliEMCALReconstructor::fgkRecParam        = 0;   // EMCAL rec. parameters
+AliEMCALRawUtils*           AliEMCALReconstructor::fgRawUtils         = 0;   // EMCAL raw utilities class
+AliEMCALClusterizer*        AliEMCALReconstructor::fgClusterizer      = 0;   // EMCAL clusterizer class
+TClonesArray*               AliEMCALReconstructor::fgDigitsArr        = 0;   // list of digits, to be used multiple times
+TObjArray*                  AliEMCALReconstructor::fgClustersArr      = 0;   // list of clusters, to be used multiple times
+TClonesArray*               AliEMCALReconstructor::fgTriggerDigits    = 0;   // list of trigger digits, to be used multiple times
+AliEMCALTriggerElectronics* AliEMCALReconstructor::fgTriggerProcessor = 0x0;
 //____________________________________________________________________________
 AliEMCALReconstructor::AliEMCALReconstructor() 
-  : fDebug(kFALSE
+  : fGeom(0),fCalibData(0),fPedestalData(0),fTriggerData(0x0
 {
   // ctor
+  
+  fgRawUtils = new AliEMCALRawUtils;
+  
+  //To make sure we match with the geometry in a simulation file,
+  //let's try to get it first.  If not, take the default geometry
+  AliRunLoader *rl = AliRunLoader::Instance();
+  if (rl->GetAliRun()){
+    AliEMCAL * emcal = dynamic_cast<AliEMCAL*>(rl->GetAliRun()->GetDetector("EMCAL"));
+    if(emcal) fGeom = emcal->GetGeometry();
+  }
+  
+  if(!fGeom) {
+    AliInfo(Form("Using default geometry in reconstruction"));
+    fGeom =  AliEMCALGeometry::GetInstance(AliEMCALGeometry::GetDefaultGeometryName());
+  }
+  
+  //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!");
+  
+  if(!fGeom) AliFatal(Form("Could not get geometry!"));
+  
+  AliEMCALTriggerDCSConfigDB* dcsConfigDB = AliEMCALTriggerDCSConfigDB::Instance();
+  
+  const AliEMCALTriggerDCSConfig* dcsConfig = dcsConfigDB->GetTriggerDCSConfig();
+  
+  if (!dcsConfig) AliFatal("No Trigger DCS Configuration from OCDB!");
+  fgTriggerProcessor = new AliEMCALTriggerElectronics( dcsConfig );
+  
+  fTriggerData = new AliEMCALTriggerData();
+  
+  //Init temporary list of digits
+  fgDigitsArr     = new TClonesArray("AliEMCALDigit",1000);
+  fgClustersArr   = new TObjArray(1000);
+  fgTriggerDigits = new TClonesArray("AliEMCALTriggerRawDigit",1000);  
 } 
 
-//____________________________________________________________________________
-AliEMCALReconstructor::AliEMCALReconstructor(const AliEMCALReconstructor & rec)
-  : AliReconstructor(rec),
-    fDebug(rec.fDebug)
-{
-  //copy ctor
-}
-
 //____________________________________________________________________________
 AliEMCALReconstructor::~AliEMCALReconstructor()
 {
   // dtor
+  
+  if(fGeom)              delete fGeom;
+  
+  //No need to delete, recovered from OCDB
+  //if(fCalibData)         delete fCalibData;
+  //if(fPedestalData)      delete fPedestalData;
+  
+  if(fgDigitsArr){
+    fgDigitsArr->Clear("C");
+    delete fgDigitsArr; 
+  }
+  
+  if(fgClustersArr){
+    fgClustersArr->Clear();
+    delete fgClustersArr; 
+  }
+  
+  if(fgTriggerDigits){
+    fgTriggerDigits->Clear();
+    delete fgTriggerDigits; 
+  }
+  
+  if(fgRawUtils)         delete fgRawUtils;
+  if(fgClusterizer)      delete fgClusterizer;
+  if(fgTriggerProcessor) delete fgTriggerProcessor;
+  
+  AliCodeTimer::Instance()->Print();
 } 
 
+//____________________________________________________________________________                                  
+void AliEMCALReconstructor::InitClusterizer() const
+{
+  //Init the clusterizer with geometry and calibration pointers, avoid doing it twice.                          
+  Int_t clusterizerType = -1;
+  Int_t eventType = -1;
+  if(GetRecParam()) {
+    clusterizerType = GetRecParam()->GetClusterizerFlag();
+    eventType       = GetRecParam()->GetEventSpecie();
+  }
+  else{
+    AliCDBEntry *entry = (AliCDBEntry*)
+      AliCDBManager::Instance()->Get("EMCAL/Calib/RecoParam");
+    //Get The reco param for the default event specie                                                           
+    if (entry) {
+      AliEMCALRecParam *recParam  = (AliEMCALRecParam*)((TObjArray *) entry->GetObject())->At(0);
+      if(recParam) clusterizerType = recParam->GetClusterizerFlag(); 
+    }
+  }
+  
+  //Check if clusterizer previously set corresponds to what is needed for this event type                       
+  if(fgClusterizer){
+    if(eventType!=AliRecoParam::kCalib){
+      //printf("ReCreate clusterizer? Clusterizer set <%d>, Clusterizer in use <%s>\n",
+      //     clusterizerType, fgClusterizer->Version());
+      
+      if     (clusterizerType == AliEMCALRecParam::kClusterizerv1 && !strcmp(fgClusterizer->Version(),"clu-v1")) return;
+      
+      else if(clusterizerType == AliEMCALRecParam::kClusterizerNxN && !strcmp(fgClusterizer->Version(),"clu-NxN")) return;
+      
+      //Need to create new clusterizer, the one set previously is not the correct one     
+      delete fgClusterizer;
+    }
+    else return;
+  }
+  
+  if (clusterizerType  == AliEMCALRecParam::kClusterizerv1)
+    {
+      fgClusterizer = new AliEMCALClusterizerv1(fGeom, fCalibData,fPedestalData);
+    }
+  else
+    {
+      fgClusterizer = new AliEMCALClusterizerNxN(fGeom, fCalibData,fPedestalData);
+    }
+}
+
 //____________________________________________________________________________
-void AliEMCALReconstructor::Reconstruct(AliRunLoader* runLoader) const 
+void AliEMCALReconstructor::Reconstruct(TTree* digitsTree, TTree* clustersTree) const
 {
   // method called by AliReconstruction; 
   // Only the clusterization is performed,; the rest of the reconstruction is done in FillESD because the track
   // segment maker needs access to the AliESD object to retrieve the tracks reconstructed by 
   // the global tracking.
-  TString headerFile(runLoader->GetFileName()) ; 
-  TString branchName(runLoader->GetEventFolder()->GetName() ) ;  
+  // Works on the current event.
   
-  AliEMCALClusterizerv1 clu(headerFile, branchName);
-  AliEMCALLoader *emcalLoader = dynamic_cast<AliEMCALLoader*>(runLoader->GetDetectorLoader("EMCAL"));    
-
-  Int_t nEvents   = runLoader->GetNumberOfEvents();
-  runLoader->LoadDigits("EMCAL");
-  for (Int_t ievent = 0; ievent < nEvents; ievent++) {
-    runLoader->GetEvent(ievent);
-    if ( Debug() ) 
-      clu.ExecuteTask("deb all") ; 
-    else 
-      clu.ExecuteTask("pseudo") ;  
-  }   
-  // Unload the Digits and RecPoints
-  emcalLoader->UnloadDigits() ; 
-  emcalLoader->UnloadRecPoints() ; 
+  AliCodeTimerAuto("",0)
+    
+  //Get input digits and put them in fgDigitsArr, clear the list before 
+  ReadDigitsArrayFromTree(digitsTree);
+  
+  InitClusterizer();
+  
+  fgClusterizer->InitParameters();
+  fgClusterizer->SetOutput(clustersTree);
+  
+  //Skip clusterization of LED events
+  if (GetRecParam()->GetEventSpecie()!=AliRecoParam::kCalib){
+    
+    if(fgDigitsArr && fgDigitsArr->GetEntries()) {
+      
+      fgClusterizer->SetInput(digitsTree);
+      
+      //fgClusterizer->Digits2Clusters("deb all") ; //For debugging
+      fgClusterizer->Digits2Clusters("");
+      
+      fgClusterizer->Clear();
+      
+    }//digits array exists and has somethind
+  }//not a LED event
+  
+  clustersTree->Fill();        
 }
 
 //____________________________________________________________________________
-void AliEMCALReconstructor::Reconstruct(AliRunLoader* runLoader, AliRawReader* rawReader) const 
-
+void AliEMCALReconstructor::ConvertDigits(AliRawReader* rawReader, TTree* digitsTree) const
+  
 {
-  // Reconstruction loop for Raw Data processing
-  //
-  // Only the clusterization is performed
-  // Track-cluster matching is done in FillESD because the track
-  // segment maker needs access to the AliESD object to retrieve 
-  // the tracks reconstructed by the global tracking.
-  
-  TString headerFile(runLoader->GetFileName()) ; 
-  TString branchName(runLoader->GetEventFolder()->GetName()) ;  
-
-  static AliEMCALRawUtils rawUtils;
-  AliEMCALClusterizerv1 clu(headerFile, branchName);
-  AliEMCALLoader *emcalLoader = dynamic_cast<AliEMCALLoader*>(runLoader->GetDetectorLoader("EMCAL"));    
-  Int_t iEvent = 0;
+  // Conversion from raw data to
+  // EMCAL digits.
+  // Works on a single-event basis
+  
   rawReader->Reset() ; 
-  while (rawReader->NextEvent()) {  
-    runLoader->GetEvent(iEvent++);
-
-    rawUtils.Raw2Digits(rawReader);
-
-    if ( Debug() ) 
-      clu.ExecuteTask("deb pseudo all") ; 
-    else 
-      clu.ExecuteTask("pseudo") ;  
-    // Unload the RecPoints
-    emcalLoader->UnloadRecPoints() ; 
+  
+  fTriggerData->SetMode(1);    
+  
+  if(fgDigitsArr) fgDigitsArr->Clear("C");
+  
+  TClonesArray *digitsTrg = new TClonesArray("AliEMCALTriggerRawDigit", 32 * 96);
+  
+  Int_t bufsize = 32000;
+  digitsTree->Branch("EMCAL", &fgDigitsArr, bufsize);
+  digitsTree->Branch("EMTRG", &digitsTrg, bufsize);
+  
+  //Skip calibration events do the rest
+  Bool_t doFit = kTRUE;
+  if ( !(GetRecParam()->FitLEDEvents()) && GetRecParam()->GetEventSpecie()==AliRecoParam::kCalib) doFit = kFALSE;
+  if (doFit){
+    //must be done here because, in constructor, option is not yet known
+    fgRawUtils->SetOption(GetOption());
+    
+    //  fgRawUtils->SetRawFormatHighLowGainFactor(GetRecParam()->GetHighLowGainFactor());
+    
+    //   fgRawUtils->SetRawFormatOrder(GetRecParam()->GetOrderParameter());
+    //    fgRawUtils->SetRawFormatTau(GetRecParam()->GetTau());
+    fgRawUtils->SetNoiseThreshold(GetRecParam()->GetNoiseThreshold());
+    fgRawUtils->SetNPedSamples(GetRecParam()->GetNPedSamples());
+    fgRawUtils->SetRemoveBadChannels(GetRecParam()->GetRemoveBadChannels());
+    fgRawUtils->SetFittingAlgorithm(GetRecParam()->GetFittingAlgorithm());
+    fgRawUtils->SetFALTROUsage(GetRecParam()->UseFALTRO());
+    
+    //fgRawUtils->SetTimeMin(GetRecParam()->GetTimeMin());
+    //fgRawUtils->SetTimeMax(GetRecParam()->GetTimeMax());
+    
+    //  fgRawUtils->SetTimeMin(-99999 );
+    //  fgRawUtils->SetTimeMax( 99999 );
+    
+    fgRawUtils->Raw2Digits(rawReader,fgDigitsArr,fPedestalData,digitsTrg,fTriggerData);
+    
+  }//skip calibration event
+  else{
+    AliDebug(1," Calibration Event, skip!");
   }
+  
+  digitsTree->Fill();
+  digitsTrg->Delete();
+  delete digitsTrg;
+  
 }
 
+
 //____________________________________________________________________________
-void AliEMCALReconstructor::FillESD(AliRunLoader* runLoader, AliESD* esd) const
+void AliEMCALReconstructor::FillESD(TTree* digitsTree, TTree* clustersTree, 
+                                   AliESDEvent* esd) const
 {
   // Called by AliReconstruct after Reconstruct() and global tracking and vertexing 
-  const double timeScale = 1.e+11; // transition constant from sec to 0.01 ns 
-
-  Int_t eventNumber = runLoader->GetEventNumber() ;
-
-  // Creates AliESDCaloCluster from AliEMCALRecPoints 
-  AliEMCALLoader *emcalLoader = dynamic_cast<AliEMCALLoader*>(runLoader->GetDetectorLoader("EMCAL"));
-  runLoader->LoadRecPoints("EMCAL");
-  runLoader->GetEvent(eventNumber);
-  TObjArray *clusters = emcalLoader->RecPoints();
-  Int_t nClusters = clusters->GetEntries(), nClustersNew=0;
-  AliDebug(1,Form("Event %d: %d clusters",eventNumber,nClusters));
-  //  Int_t nRP=0, nPC=0; // in input
-  esd->SetFirstEMCALCluster(esd->GetNumberOfCaloClusters()); // Put after Phos clusters 
-
-  //#########Calculate trigger and set trigger info###########
-  AliEMCALTrigger tr ;
-  //   tr.SetPatchSize(1);//create 4x4 patches
-  tr.Trigger();
-  
-  Float_t maxAmp2x2  = tr.Get2x2MaxAmplitude();
-  Float_t maxAmpnxn  = tr.GetnxnMaxAmplitude();
-  Float_t ampOutOfPatch2x2  = tr.Get2x2AmpOutOfPatch() ;
-  Float_t ampOutOfPatchnxn  = tr.GetnxnAmpOutOfPatch() ;
-
-  AliEMCALGeometry * geom = 0;
-  if (runLoader->GetAliRun() && runLoader->GetAliRun()->GetDetector("EMCAL"))
-    geom = dynamic_cast<AliEMCAL*>(runLoader->GetAliRun()->GetDetector("EMCAL"))->GetGeometry();
-  if (geom == 0) 
-    geom = AliEMCALGeometry::GetInstance(AliEMCALGeometry::GetDefaulGeometryName());
-
-  Int_t iSM2x2      = tr.Get2x2SuperModule();
-  Int_t iSMnxn      = tr.GetnxnSuperModule();
-  Int_t iCellPhi2x2 = tr.Get2x2CellPhi();
-  Int_t iCellPhinxn = tr.GetnxnCellPhi();
-  Int_t iCellEta2x2 = tr.Get2x2CellEta();
-  Int_t iCellEtanxn = tr.GetnxnCellEta();
-
-  AliDebug(2, Form("Trigger 2x2 max amp %f, out amp %f, SM %d, iphi %d ieta %d",  maxAmp2x2, ampOutOfPatch2x2, iSM2x2,iCellPhi2x2, iCellEta2x2));
-  AliDebug(2, Form("Trigger 4x4 max amp %f , out amp %f, SM %d, iphi %d, ieta %d",  maxAmpnxn, ampOutOfPatchnxn, iSMnxn,iCellPhinxn, iCellEtanxn));
-
-  TVector3    pos2x2(-1,-1,-1);
-  TVector3    posnxn(-1,-1,-1);
-
-  Int_t iAbsId2x2 = geom->GetAbsCellIdFromCellIndexes( iSM2x2, iCellPhi2x2, iCellEta2x2) ;
-  Int_t iAbsIdnxn = geom->GetAbsCellIdFromCellIndexes( iSMnxn, iCellPhinxn, iCellEtanxn) ;
-  geom->GetGlobal(iAbsId2x2, pos2x2);
-  geom->GetGlobal(iAbsIdnxn, posnxn);
-  
-  TArrayF triggerPosition(6);
-  triggerPosition[0] = pos2x2(0) ;   
-  triggerPosition[1] = pos2x2(1) ;   
-  triggerPosition[2] = pos2x2(2) ;  
-  triggerPosition[3] = posnxn(0) ;   
-  triggerPosition[4] = posnxn(1) ;   
-  triggerPosition[5] = posnxn(2) ;  
-
-  TArrayF triggerAmplitudes(4);
-  triggerAmplitudes[0] = maxAmp2x2 ;   
-  triggerAmplitudes[1] = ampOutOfPatch2x2 ;    
-  triggerAmplitudes[2] = maxAmpnxn ;   
-  triggerAmplitudes[3] = ampOutOfPatchnxn ;   
-
-  esd->AddEMCALTriggerPosition(triggerPosition);
-  esd->AddEMCALTriggerAmplitudes(triggerAmplitudes);
+  // and V0 
+  // Works on the current event
+  // printf(" ## AliEMCALReconstructor::FillESD() is started ### \n ");
+  //return;
   
-  //######################################
-
-  //Fill CaloClusters 
-
+  //########################################
+  // Trigger
+  //########################################
+  
+  static int saveOnce = 0;
+  
+  Int_t v0M[2] = {0, 0};
+  
+  AliESDVZERO* esdV0 = esd->GetVZEROData();
+  
+  if (esdV0) 
+    {
+      for (Int_t i = 0; i < 32; i++)
+       {
+         v0M[0] += (Int_t)esdV0->GetAdcV0C(i);
+         v0M[1] += (Int_t)esdV0->GetAdcV0A(i);
+       }
+    }
+  else
+    {
+      AliWarning("Cannot retrieve V0 ESD! Run w/ null V0 charges");
+    }
+  
+  if (fgTriggerDigits) fgTriggerDigits->Clear();
+  
+  TBranch *branchtrg = digitsTree->GetBranch("EMTRG");
+  
+  if (!branchtrg) 
+    { 
+      AliError("Can't get the branch with the EMCAL trigger digits!");
+      return;
+    }
+  
+  branchtrg->SetAddress(&fgTriggerDigits);
+  branchtrg->GetEntry(0);
+  
+  // Note: fgTriggerProcessor reset done at the end of this method
+  fgTriggerProcessor->Digits2Trigger(fgTriggerDigits, v0M, fTriggerData);
+  
+  // Fill ESD
+  AliESDCaloTrigger* trgESD = esd->GetCaloTrigger("EMCAL");
+  
+  if (trgESD)
+    {
+      trgESD->Allocate(fgTriggerDigits->GetEntriesFast());
+      
+      for (Int_t i = 0; i < fgTriggerDigits->GetEntriesFast(); i++)
+       {         
+         AliEMCALTriggerRawDigit* rdig = (AliEMCALTriggerRawDigit*)fgTriggerDigits->At(i);
+         
+         Int_t px, py;
+         if (fGeom->GetPositionInEMCALFromAbsFastORIndex(rdig->GetId(), px, py))
+           {
+             Int_t a = -1, t = -1, times[10]; 
+             
+             rdig->GetMaximum(a, t);
+             rdig->GetL0Times(times);
+             
+             trgESD->Add(px, py, a, t, times, rdig->GetNL0Times(), rdig->GetL1TimeSum(), rdig->GetTriggerBits());
+           }
+       }
+      
+      trgESD->SetL1Threshold(0, fTriggerData->GetL1GammaThreshold());
+      
+      trgESD->SetL1Threshold(1, fTriggerData->GetL1JetThreshold()  );
+      
+      Int_t v0[2];
+      fTriggerData->GetL1V0(v0);
+      
+      trgESD->SetL1V0(v0);     
+      trgESD->SetL1FrameMask(fTriggerData->GetL1FrameMask());            
+      
+      if (!saveOnce && fTriggerData->GetL1DataDecoded()) 
+       {
+         int type[8] = {0};
+         fTriggerData->GetL1TriggerType(type);
+         
+         esd->SetCaloTriggerType(type);
+         
+         saveOnce = 1;
+       }
+    }
+  
+  // Resetting
+  fTriggerData->Reset();
+  
+  //########################################
+  //##############Fill CaloCells###############
+  //########################################
+  
+  //Get input digits and put them in fgDigitsArr, clear the list before 
+  ReadDigitsArrayFromTree(digitsTree);
+  
+  Int_t nDigits = fgDigitsArr->GetEntries(), idignew = 0 ;
+  AliDebug(1,Form("%d digits",nDigits));
+  
+  AliESDCaloCells &emcCells = *(esd->GetEMCALCells());
+  emcCells.CreateContainer(nDigits);
+  emcCells.SetType(AliVCaloCells::kEMCALCell);
+  Float_t energy = 0;
+  for (Int_t idig = 0 ; idig < nDigits ; idig++) {
+    const AliEMCALDigit * dig = (const AliEMCALDigit*)fgDigitsArr->At(idig);
+    if(dig->GetAmplitude() > 0 ){
+      energy = fgClusterizer->Calibrate(dig->GetAmplitude(),dig->GetTime(),dig->GetId()); //TimeR or Time?
+      if(energy > 0){ //Digits tagged as bad (dead, hot, not alive) are set to 0 in calibrate, remove them     
+        emcCells.SetCell(idignew,dig->GetId(),energy, dig->GetTime());   
+        idignew++;
+      }
+    }
+  }
+  emcCells.SetNumberOfCells(idignew);
+  emcCells.Sort();
+  
+  //------------------------------------------------------------
+  //-----------------CLUSTERS-----------------------------
+  //------------------------------------------------------------
+  clustersTree->SetBranchStatus("*",0); //disable all branches
+  clustersTree->SetBranchStatus("EMCALECARP",1); //Enable only the branch we need
+  if(fgClustersArr) fgClustersArr->Clear();
+  TBranch *branch = clustersTree->GetBranch("EMCALECARP");
+  branch->SetAddress(&fgClustersArr);
+  branch->GetEntry(0);
+  //clustersTree->GetEvent(0);
+  
+  Int_t nClusters = fgClustersArr->GetEntries(),  nClustersNew=0;
+  AliDebug(1,Form("%d clusters",nClusters));
+  
+  //######################################################
+  //#######################TRACK MATCHING###############
+  //######################################################
+  //Fill list of integers, each one is index of track to which the cluster belongs.
+  
+  // step 1 - initialize array of matched track indexes
+  Int_t *matchedTrack = new Int_t[nClusters];
+  for (Int_t iclus = 0; iclus < nClusters; iclus++)
+    matchedTrack[iclus] = -1;  // neg. index --> no matched track
+  
+  // step 2, change the flag for all matched clusters found in tracks
+  Int_t iemcalMatch = -1;
+  Int_t endtpc = esd->GetNumberOfTracks();
+  for (Int_t itrack = 0; itrack < endtpc; itrack++) {
+    AliESDtrack * track = esd->GetTrack(itrack) ; // retrieve track
+    iemcalMatch = track->GetEMCALcluster();
+    if(iemcalMatch >= 0) matchedTrack[iemcalMatch] = itrack;
+  } 
+  
+  //########################################
+  //##############Fill CaloClusters#############
+  //########################################
   for (Int_t iClust = 0 ; iClust < nClusters ; iClust++) {
-    const AliEMCALRecPoint * clust = emcalLoader->RecPoint(iClust);
-    //if(clust->GetClusterType()== AliESDCaloCluster::kClusterv1) nRP++; else nPC++;
-    if (Debug()) clust->Print();
+    const AliEMCALRecPoint * clust = (const AliEMCALRecPoint*)fgClustersArr->At(iClust);
+    //if(clust->GetClusterType()== AliVCluster::kEMCALClusterv1) nRP++; else nPC++;
+    // clust->Print(); //For debugging
     // Get information from EMCAL reconstruction points
     Float_t xyz[3];
     TVector3 gpos;
     clust->GetGlobalPosition(gpos);
-    for (Int_t ixyz=0; ixyz<3; ixyz++) 
+    for (Int_t ixyz=0; ixyz<3; ixyz++)
       xyz[ixyz] = gpos[ixyz];
-
-    Int_t digitMult = clust->GetMultiplicity();
-    UShort_t *amplList = new UShort_t[digitMult];
-    UShort_t *timeList = new UShort_t[digitMult];
-    UShort_t *digiList = new UShort_t[digitMult];
-    Float_t *amplFloat = clust->GetEnergiesList();
-    Float_t *timeFloat = clust->GetTimeList();
-    Int_t   *digitInts = clust->GetAbsId();
     Float_t elipAxis[2];
     clust->GetElipsAxis(elipAxis);
-
-    // Convert Float_t* and Int_t* to UShort_t* to save memory
-    // Problem : we should recalculate a cluster characteristics when discard digit(s)
-    Int_t newdigitMult = 0; 
-    for (Int_t iDigit=0; iDigit<digitMult; iDigit++) {
-      if (amplFloat[iDigit] > 0) {
-       amplList[newdigitMult] = (UShort_t)(amplFloat[iDigit]*500);
-        // Time in units of 0.01 ns = 10 ps
-        if(timeFloat[iDigit] < 65536./timeScale) 
-         timeList[newdigitMult] = (UShort_t)(timeFloat[iDigit]*timeScale);
-        else
-          timeList[newdigitMult] = 65535;
-       digiList[newdigitMult] = (UShort_t)(digitInts[iDigit]);
-        newdigitMult++;
+    //Create digits lists
+    Int_t cellMult = clust->GetMultiplicity();
+    //TArrayS digiList(digitMult);
+    Float_t *amplFloat = clust->GetEnergiesList();
+    Int_t   *digitInts = clust->GetAbsId();
+    TArrayS absIdList(cellMult);
+    TArrayD fracList(cellMult);
+    
+    Int_t newCellMult = 0;
+    for (Int_t iCell=0; iCell<cellMult; iCell++) {
+      if (amplFloat[iCell] > 0) {
+        absIdList[newCellMult] = (UShort_t)(digitInts[iCell]);
+        //Calculate Fraction
+        if(emcCells.GetCellAmplitude(digitInts[iCell])>0 && GetRecParam()->GetUnfold()){
+          fracList[newCellMult] = amplFloat[iCell]/(emcCells.GetCellAmplitude(digitInts[iCell]));//get cell calibration value 
+          
+        }
+        else{
+          fracList[newCellMult] = 0; 
+        }
+        newCellMult++;
       }
-      else if (clust->GetClusterType() != AliESDCaloCluster::kPseudoCluster)
-        Warning("FillESD()","Negative or 0 digit amplitude in cluster");
     }
-
-    if(newdigitMult > 0) { // accept cluster if it has some digit
+    
+    absIdList.Set(newCellMult);
+    fracList.Set(newCellMult);
+    
+    if(newCellMult > 0) { // accept cluster if it has some digit
       nClustersNew++;
-      if(newdigitMult != digitMult) { // some digits were deleted
-        UShort_t *amplListNew = new UShort_t[newdigitMult];
-        UShort_t *timeListNew = new UShort_t[newdigitMult];
-        UShort_t *digiListNew = new UShort_t[newdigitMult];
-        for (Int_t iDigit=0; iDigit<newdigitMult; iDigit++) {
-          amplListNew[iDigit] = amplList[iDigit];
-          timeListNew[iDigit] = timeList[iDigit];
-          digiListNew[iDigit] = digiList[iDigit];
-        }
-
-        delete [] amplList;
-        delete [] timeList;
-        delete [] digiList;
-
-        amplList = amplListNew;
-        timeList = timeListNew;
-        digiList = digiListNew;
-      }
-
       //Primaries
-      Int_t  primMult  = 0;
-      Int_t *primInts =  clust->GetPrimaries(primMult);
-      UShort_t *primList = new UShort_t[primMult];
-      for (Int_t ipr=0; ipr<primMult; ipr++) 
-       primList[ipr] = (UShort_t)(primInts[ipr]);       
-
-
+      Int_t  parentMult  = 0;
+      Int_t *parentList =  clust->GetParents(parentMult);
       // fills the ESDCaloCluster
-      AliESDCaloCluster * ec = new AliESDCaloCluster() ; 
-      ec->SetEMCAL(kTRUE);
-      ec->SetClusterType(clust->GetClusterType());
-      ec->SetGlobalPosition(xyz);
-      ec->SetClusterEnergy(clust->GetEnergy());
+      AliESDCaloCluster * ec = new AliESDCaloCluster() ;
+      ec->SetType(AliVCluster::kEMCALClusterv1);
+      ec->SetPosition(xyz);
+      ec->SetE(clust->GetEnergy());
       
-      ec->SetNumberOfDigits(newdigitMult);
-      ec->SetDigitAmplitude(amplList); //energies
-      ec->SetDigitTime(timeList);      //times
-      ec->SetDigitIndex(digiList);     //indices
-      if(clust->GetClusterType()== AliESDCaloCluster::kClusterv1){
-        ec->SetPrimaryIndex(clust->GetPrimaryIndex());
-        ec->SetNumberOfPrimaries(primMult);           //primary multiplicity
-        ec->SetListOfPrimaries(primList);                  //primary List for a cluster  
-        ec->SetClusterDisp(clust->GetDispersion());
-        ec->SetClusterChi2(-1); //not yet implemented
-        ec->SetM02(elipAxis[0]*elipAxis[0]) ;
-        ec->SetM20(elipAxis[1]*elipAxis[1]) ;
-        ec->SetM11(-1) ;        //not yet implemented
-      } 
-    // add the cluster to the esd object
+      //Distance to the nearest bad crystal
+      ec->SetDistanceToBadChannel(clust->GetDistanceToBadTower()); 
+      
+      ec->SetNCells(newCellMult);
+      //Change type of list from short to ushort
+      UShort_t *newAbsIdList  = new UShort_t[newCellMult];
+      Double_t *newFracList   = new Double_t[newCellMult];
+      for(Int_t i = 0; i < newCellMult ; i++) {
+        newAbsIdList[i]=absIdList[i];
+        newFracList[i] =fracList[i];
+      }
+      ec->SetCellsAbsId(newAbsIdList);
+      ec->SetCellsAmplitudeFraction(newFracList);
+      ec->SetDispersion(clust->GetDispersion());
+      ec->SetChi2(-1); //not yet implemented
+      ec->SetM02(elipAxis[0]*elipAxis[0]) ;
+      ec->SetM20(elipAxis[1]*elipAxis[1]) ;
+      ec->SetTOF(clust->GetTime()) ; //time-of-fligh
+      ec->SetNExMax(clust->GetNExMax());          //number of local maxima
+      TArrayI arrayTrackMatched(1);// Only one track, temporal solution.
+      arrayTrackMatched[0]= matchedTrack[iClust];
+      ec->AddTracksMatched(arrayTrackMatched);
+      
+      TArrayI arrayParents(parentMult,parentList);
+      ec->AddLabels(arrayParents);
+      
+      // add the cluster to the esd object
       esd->AddCaloCluster(ec);
       delete ec;
-    } else { // no new ESD cluster
-        delete [] amplList;
-        delete [] timeList;
-        delete [] digiList;
+      delete [] newAbsIdList ;
+      delete [] newFracList ;
     }
   } // cycle on clusters
-  esd->SetNumberOfEMCALClusters(nClustersNew);
-  //if(nClustersNew != nClusters) 
-  //printf(" ##### nClusters %i -> new %i ##### \n", nClusters, nClustersNew );
-
+  
+  delete [] matchedTrack;
+  
   //Fill ESDCaloCluster with PID weights
   AliEMCALPID *pid = new AliEMCALPID;
   //pid->SetPrintInfo(kTRUE);
   pid->SetReconstructor(kTRUE);
   pid->RunPID(esd);
+  delete pid;
+  
+  //Store EMCAL misalignment matrixes
+  FillMisalMatrixes(esd) ;
+  
+}
+
+//==================================================================================
+void AliEMCALReconstructor::FillMisalMatrixes(AliESDEvent* esd)const{
+  //Store EMCAL matrixes in ESD Header
+  
+  //Check, if matrixes was already stored
+  for(Int_t sm = 0 ; sm < fGeom->GetNumberOfSuperModules(); sm++){
+    if(esd->GetEMCALMatrix(sm)!=0)
+      return ;
+  }
+  
+  //Create and store matrixes
+  if(!gGeoManager){
+    AliError("Can not store misal. matrixes: no gGeoManager! \n") ;
+    return ;
+  }
+  //Note, that owner of copied marixes will be header
+  const Int_t bufsize = 255;
+  char path[bufsize] ;
+  TGeoHMatrix * m = 0x0;
+  for(Int_t sm = 0; sm < fGeom->GetNumberOfSuperModules(); sm++){
+    snprintf(path,bufsize,"/ALIC_1/XEN1_1/SMOD_%d",sm+1) ; //In Geometry modules numbered 1,2,.,5
+    if(sm >= 10) snprintf(path,bufsize,"/ALIC_1/XEN1_1/SM10_%d",sm-10+1) ;
+    
+    if (gGeoManager->CheckPath(path)){
+      gGeoManager->cd(path);
+      m = gGeoManager->GetCurrentMatrix() ;
+      //                       printf("================================================= \n");
+      //                       printf("AliEMCALReconstructor::FixMisalMatrixes(), sm %d, \n",sm);
+      //                       m->Print("");
+      esd->SetEMCALMatrix(new TGeoHMatrix(*m),sm) ;
+      //                       printf("================================================= \n");
+    }
+    else{
+      esd->SetEMCALMatrix(NULL,sm) ;
+    }
+  }
+}
+
+//__________________________________________________________________________
+void AliEMCALReconstructor::ReadDigitsArrayFromTree(TTree *digitsTree) const
+{
+  // Read the digits from the input tree
+  // See AliEMCALClusterizer::SetInput(TTree *digitsTree);    
+  
+  // Clear previous digits in the list
+  if(fgDigitsArr){ 
+    fgDigitsArr->Clear("C");
+  }
+  else{
+    // It should not happen, but just in case ...
+    fgDigitsArr = new TClonesArray("AliEMCALDigit",100); 
+  }
+  
+  // 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;
+  }  
+  
+  branch->SetAddress(&fgDigitsArr);
+  branch->GetEntry(0);
 }