Adding a reminder for coders
[u/mrichter/AliRoot.git] / EMCAL / AliEMCALReconstructor.cxx
index 3ca7095..b3fba39 100644 (file)
 // Reconstruction class. Redesigned from the old AliReconstructionner class and 
 // derived from STEER/AliReconstructor. 
 // 
-//-- Aleksei Pavlinov : added staf for EMCAL jet trigger 9Apr 25, 2008)
-//                    : fgDigitsArr should read just once at event
+
 
 // --- ROOT system ---
-#include <TList.h>
 #include <TClonesArray.h>
-#include <TH2.h>
 #include "TGeoManager.h"
 #include "TGeoMatrix.h"
 
 #include "AliEMCALRawUtils.h"
 #include "AliEMCALDigit.h"
 #include "AliEMCALClusterizerv1.h"
+#include "AliEMCALClusterizerv2.h"
 #include "AliEMCALClusterizerNxN.h"
 #include "AliEMCALRecPoint.h"
 #include "AliEMCALPID.h"
-#include "AliEMCALTrigger.h"
 #include "AliRawReader.h"
 #include "AliCDBEntry.h"
 #include "AliCDBManager.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), fList(0), fGeom(0),fCalibData(0),fPedestalData(0),fTriggerData(0x0) 
+  : fGeom(0),fCalibData(0),fPedestalData(0),fTriggerData(0x0), fMatches(0x0)
 {
   // ctor
 
-  fgRawUtils = new AliEMCALRawUtils;
+  // AliDebug(2, "Mark.");  
 
+  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() && rl->GetAliRun()->GetDetector("EMCAL")) {
-    fGeom = dynamic_cast<AliEMCAL*>(rl->GetAliRun()->GetDetector("EMCAL"))->GetGeometry();
-  } else {
+  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();
+      AliCDBEntry *entry = (AliCDBEntry*) 
+       AliCDBManager::Instance()->Get("EMCAL/Calib/Data");
+      if (entry) fCalibData =  (AliEMCALCalibData*) entry->GetObject();
     }
-       
+  
   if(!fCalibData)
-               AliFatal("Calibration parameters not found in CDB!");
-       
+    AliFatal("Calibration parameters not found in CDB!");
+  
   //Get calibration parameters 
   if(!fPedestalData)
     {
@@ -117,23 +120,25 @@ AliEMCALReconstructor::AliEMCALReconstructor()
   if(!fPedestalData)
     AliFatal("Dead map not found in CDB!");
   
-  InitClusterizer();
-       
   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);
-       
+  
+  //Init temporary list of digits
+  fgDigitsArr     = new TClonesArray("AliEMCALDigit",1000);
+  fgClustersArr   = new TObjArray(1000);
+  fgTriggerDigits = new TClonesArray("AliEMCALTriggerRawDigit",1000);  
+
+  //Track matching
+  fMatches = new TList();
+  fMatches->SetOwner(kTRUE);
 } 
 
 //____________________________________________________________________________
@@ -141,9 +146,13 @@ AliEMCALReconstructor::~AliEMCALReconstructor()
 {
   // dtor
 
+  //AliDebug(2, "Mark.");
+
   if(fGeom)              delete fGeom;
-  if(fCalibData)         delete fCalibData;
-  if(fPedestalData)      delete fPedestalData;
+  
+  //No need to delete, recovered from OCDB
+  //if(fCalibData)         delete fCalibData;
+  //if(fPedestalData)      delete fPedestalData;
   
   if(fgDigitsArr){
     fgDigitsArr->Clear("C");
@@ -155,44 +164,74 @@ AliEMCALReconstructor::~AliEMCALReconstructor()
     delete fgClustersArr; 
   }
   
+  if(fgTriggerDigits){
+    fgTriggerDigits->Clear();
+    delete fgTriggerDigits; 
+  }
+  
   if(fgRawUtils)         delete fgRawUtils;
   if(fgClusterizer)      delete fgClusterizer;
   if(fgTriggerProcessor) delete fgTriggerProcessor;
   
+  if(fMatches) { fMatches->Delete(); delete fMatches; fMatches = 0;}
+  
   AliCodeTimer::Instance()->Print();
 } 
 
-// //____________________________________________________________________________
-// void AliEMCALReconstructor::Init()
-// {
-//   // Trigger hists - Oct 24, 2007
-//    fList = AliEMCALHistoUtilities::GetTriggersListOfHists(kTRUE);
-// }
-
-//____________________________________________________________________________
-void AliEMCALReconstructor::InitClusterizer() 
+//____________________________________________________________________________                                  
+void AliEMCALReconstructor::InitClusterizer() const
 {
-  //Init the clusterizer with geometry and calibration pointers, avoid doing it twice.
-  
-  AliEMCALRecParam *recParam = NULL;
-  AliCDBEntry *entry = (AliCDBEntry*) 
-    AliCDBManager::Instance()->Get("EMCAL/Calib/RecoParam");
-  //Get The reco param for the default event specie
-  if (entry) 
-    recParam = (AliEMCALRecParam*)((TObjArray *) entry->GetObject())->At(0);
+  //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(); 
+    }
+  }
   
-  if(!recParam)  
-    AliFatal("RecoParam not found in CDB!");
+  //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;
+      
+      else if(clusterizerType == AliEMCALRecParam::kClusterizerv2  && !strcmp(fgClusterizer->Version(),"clu-v2"))  return;
+      
+      //Need to create new clusterizer, the one set previously is not the correct one     
+      delete fgClusterizer;
+    }
+    else return;
+  }
   
-  if (recParam->GetClusterizerFlag() == AliEMCALRecParam::kClusterizerv1)
+  if      (clusterizerType  == AliEMCALRecParam::kClusterizerv1)
     {
-      fgClusterizer = new AliEMCALClusterizerv1(fGeom, fCalibData,fPedestalData); 
+      fgClusterizer = new AliEMCALClusterizerv1 (fGeom, fCalibData,fPedestalData);
     }
-  else
+  else if (clusterizerType  == AliEMCALRecParam::kClusterizerNxN)
     {
-      fgClusterizer = new AliEMCALClusterizerNxN(fGeom, fCalibData,fPedestalData); 
+      fgClusterizer = new AliEMCALClusterizerNxN(fGeom, fCalibData,fPedestalData);
     }
-  
+  else if (clusterizerType  == AliEMCALRecParam::kClusterizerv2)
+  {
+    fgClusterizer = new AliEMCALClusterizerv2   (fGeom, fCalibData,fPedestalData);
+  }
+  else 
+  {
+    AliFatal(Form("Unknown clusterizer %d ", clusterizerType));
+  }
 }
 
 //____________________________________________________________________________
@@ -203,82 +242,92 @@ void AliEMCALReconstructor::Reconstruct(TTree* digitsTree, TTree* clustersTree)
   // segment maker needs access to the AliESD object to retrieve the tracks reconstructed by 
   // the global tracking.
   // Works on the current event.
-
+  
   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);
     
-                 if(Debug())
-                         fgClusterizer->Digits2Clusters("deb all") ;
-                 else
-                         fgClusterizer->Digits2Clusters("");
-    
-                 fgClusterizer->Clear();
-
-         }//digits array exists and has somethind
+    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();        
+
+  // Deleting the recpoints at the end of the reconstruction call
+  fgClusterizer->DeleteRecPoints();
 }
 
 //____________________________________________________________________________
 void AliEMCALReconstructor::ConvertDigits(AliRawReader* rawReader, TTree* digitsTree) const
-
+  
 {
   // Conversion from raw data to
   // EMCAL digits.
   // Works on a single-event basis
-
+  
   rawReader->Reset() ; 
-
+  
   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->Raw2Digits(rawReader,fgDigitsArr,fPedestalData,digitsTrg,fTriggerData);
+    //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!");
+    AliDebug(1," Calibration Event, skip!");
   }
-       
+  
   digitsTree->Fill();
   digitsTrg->Delete();
   delete digitsTrg;
-
+  
 }
 
 
@@ -291,127 +340,122 @@ void AliEMCALReconstructor::FillESD(TTree* digitsTree, TTree* clustersTree,
   // Works on the current event
   // printf(" ## AliEMCALReconstructor::FillESD() is started ### \n ");
   //return;
-
-  //FIXME UNCOMMENT WHEN ESDTRIGGER AVAILABLE 
-//   // Trigger
-//   Int_t v0M[2] = {0, 0};
-       
-//   AliESDVZERO* esdV0 = esd->GetVZEROData();
-
-//   if (esdV0) 
-//   {
-//       for (Int_t i = 0; i < 32; i++)
-//       {
-//               v0M[0] += esdV0->GetAdcV0C(i);
-//               v0M[1] += esdV0->GetAdcV0A(i);
-//       }
-//   }
-//   else
-//   {
-//       AliWarning("Cannot retrieve V0 ESD! Run w/ null V0 charges");
-//   }
-
-//   TClonesArray *trgDigits = new TClonesArray("AliEMCALTriggerRawDigit",1000);
-       
-//   TBranch *branchtrg = digitsTree->GetBranch("EMTRG");
-       
-//   if (!branchtrg) 
-//   { 
-//       AliError("Can't get the branch with the EMCAL trigger digits!");
-//       return;
-//   }
-       
-//   branchtrg->SetAddress(&trgDigits);
-//   branchtrg->GetEntry(0);
   
-//   // Note: fgTriggerProcessor reset done at the end of this method
-//   fgTriggerProcessor->Digits2Trigger(trgDigits, v0M, fTriggerData);
-
-//   // Fill ESD
-//   AliESDCaloTrigger* trgESD = esd->GetCaloTrigger("EMCAL");
+  //########################################
+  // Trigger
+  //########################################
+  
+  static int saveOnce = 0;
   
-//   if (trgESD)
-//   {
-//       trgESD->Allocate(trgDigits->GetEntriesFast());
+  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);
          
-//       for (Int_t i = 0; i < trgDigits->GetEntriesFast(); i++)
-//       {       
-//               AliEMCALTriggerRawDigit* rdig = (AliEMCALTriggerRawDigit*)trgDigits->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);
-                         
-// //                    rdig->Print("");
-                         
-//                       trgESD->Add(px, py, a, t, times, rdig->GetNL0Times(), rdig->GetL1TimeSum());
-//               }
-//       }
+         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);
          
-// //    cout << "End of Adding................." << endl;
-
-//       trgESD->SetL1Threshold(0, fTriggerData->GetL1GammaThreshold());
+         esd->SetCaloTriggerType(type);
          
-//       trgESD->SetL1Threshold(1, fTriggerData->GetL1JetThreshold()  );
-
-//       for (Int_t i = 0; i < kTriggerTypeEnd; i++)
-//       {       
-//               for (Int_t j = 0; j < 2; j++)
-//               {
-//                       TClonesArray* patches = fTriggerData->GetPatches((TriggerType_t)i, j);
-                         
-//                       TIter NextPatch(patches);
-//                       while (AliEMCALTriggerPatch* p = (AliEMCALTriggerPatch*)NextPatch())
-//                       {
-//                               TVector2 pos; p->Position(pos);
-//                               trgESD->SetTriggerBits(pos.X(), pos.Y(), i, j);
-//                       }
-//               }
-//       }
-//   }
-
-//   // Resetting
-//   fTriggerData->Reset();
-// //  cout << "Reset trg data" << endl;
-  //FIXME UNCOMMENT WHEN ESDTRIGGER AVAILABLE 
+         saveOnce = 1;
+       }
+    }
+  
+  // Resetting
+  fTriggerData->Reset();
   
   //########################################
   //##############Fill CaloCells###############
   //########################################
+  
+  //Get input digits and put them in fgDigitsArr, clear the list before 
   ReadDigitsArrayFromTree(digitsTree);
-
-//   TClonesArray *digits = new TClonesArray("AliEMCALDigit",1000);
-//   TBranch *branchdig = digitsTree->GetBranch("EMCAL");
-//   if (!branchdig) { 
-//     AliError("can't get the branch with the EMCAL digits !");
-//     return;
-//   }
-//   branchdig->SetAddress(&digits);
-//   digitsTree->GetEvent(0);
+  
   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;
+  Float_t time   = 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++;
-         }
+    time   = dig->GetTime();      // Time already calibrated in clusterizer
+    energy = dig->GetAmplitude(); // energy calibrated in clusterizer
+    if(energy > 0 ){
+      fgClusterizer->Calibrate(energy,time,dig->GetId()); //Digits already calibrated in clusterizers
+      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, time);   
+        idignew++;
+      }
     }
   }
   emcCells.SetNumberOfCells(idignew);
   emcCells.Sort();
-
+  
   //------------------------------------------------------------
   //-----------------CLUSTERS-----------------------------
   //------------------------------------------------------------
@@ -422,36 +466,19 @@ void AliEMCALReconstructor::FillESD(TTree* digitsTree, TTree* clustersTree,
   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 = (const AliEMCALRecPoint*)fgClustersArr->At(iClust);
+    if(!clust) continue;
     //if(clust->GetClusterType()== AliVCluster::kEMCALClusterv1) nRP++; else nPC++;
-    if (Debug()) clust->Print();
+    // clust->Print(); //For debugging
     // Get information from EMCAL reconstruction points
     Float_t xyz[3];
     TVector3 gpos;
@@ -460,27 +487,30 @@ void AliEMCALReconstructor::FillESD(TTree* digitsTree, TTree* clustersTree,
       xyz[ixyz] = gpos[ixyz];
     Float_t elipAxis[2];
     clust->GetElipsAxis(elipAxis);
-       //Create digits lists
+    //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]);
-      //Uncomment when unfolding is done
-      //if(emcCells.GetCellAmplitude(digitInts[iCell])>0)
-      //fracList[newCellMult] = amplFloat[iCell]/(emcCells.GetCellAmplitude(digitInts[iCell])*calibration);//get cell calibration value 
-      //else
-      fracList[newCellMult] = 0; 
-      newCellMult++;
+        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++;
       }
     }
-
+    
     absIdList.Set(newCellMult);
     fracList.Set(newCellMult);
     
@@ -494,10 +524,10 @@ void AliEMCALReconstructor::FillESD(TTree* digitsTree, TTree* clustersTree,
       ec->SetType(AliVCluster::kEMCALClusterv1);
       ec->SetPosition(xyz);
       ec->SetE(clust->GetEnergy());
-               
+      
       //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];
@@ -514,92 +544,242 @@ void AliEMCALReconstructor::FillESD(TTree* digitsTree, TTree* clustersTree,
       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
+      //
+      //Track matching
+      //
+      fMatches->Clear();
+      Int_t nTracks = esd->GetNumberOfTracks();
+      for (Int_t itrack = 0; itrack < nTracks; itrack++)
+       {
+         AliESDtrack * track = esd->GetTrack(itrack) ; // retrieve track
+         if(track->GetEMCALcluster()==iClust)
+           {
+             Double_t dEta=-999, dPhi=-999;
+             Bool_t isMatch =  CalculateResidual(track, ec, dEta, dPhi);
+             if(!isMatch) 
+               {
+                 // AliDebug(10, "Not good");
+                 continue;
+               }
+             AliEMCALMatch *match = new AliEMCALMatch();
+             match->SetIndexT(itrack);
+             match->SetDistance(TMath::Sqrt(dEta*dEta+dPhi*dPhi));
+             match->SetdEta(dEta);
+             match->SetdPhi(dPhi);
+             fMatches->Add(match);
+           }
+       } 
+      fMatches->Sort(kSortAscending); //Sort matched tracks from closest to furthest
+      Int_t nMatch = fMatches->GetEntries();
+      TArrayI arrayTrackMatched(nMatch);
+      for(Int_t imatch=0; imatch<nMatch; imatch++)
+       {
+         AliEMCALMatch *match = (AliEMCALMatch*)fMatches->At(imatch);
+         arrayTrackMatched[imatch] = match->GetIndexT();
+         if(imatch==0)
+           {
+             ec->SetTrackDistance(match->GetdPhi(), match->GetdEta());
+           }
+       }
+      ec->AddTracksMatched(arrayTrackMatched);
+    
+      //add the cluster to the esd object
       esd->AddCaloCluster(ec);
+
       delete ec;
       delete [] newAbsIdList ;
       delete [] newFracList ;
-   }
- } // cycle on clusters
-
- 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) ;
+    }
+  } // cycle on clusters
 
+  //
+  //Reset the index of matched cluster for tracks
+  //to the one in CaloCluster array
+  Int_t ncls = esd->GetNumberOfCaloClusters();
+  for(Int_t icl=0; icl<ncls; icl++)
+    {
+      AliESDCaloCluster *cluster = esd->GetCaloCluster(icl);
+      if(!cluster || !cluster->IsEMCAL()) continue;
+      TArrayI *trackIndex = cluster->GetTracksMatched();
+      for(Int_t itr=0; itr<trackIndex->GetSize(); itr++)
+       {
+         AliESDtrack *track = esd->GetTrack(trackIndex->At(itr));
+         track->SetEMCALcluster(cluster->GetID());
+       }
+    }
+  
+  
+  //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
-       char path[255] ;
-       TGeoHMatrix * m = 0x0;
-       for(Int_t sm = 0; sm < fGeom->GetNumberOfSuperModules(); sm++){
-               sprintf(path,"/ALIC_1/XEN1_1/SMOD_%d",sm+1) ; //In Geometry modules numbered 1,2,.,5
-               if(sm >= 10) sprintf(path,"/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) ;
-               }
-       }
+  //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
 {
-  // See AliEMCALClusterizer::SetInput(TTree *digitsTree);
-  if(fgDigitsArr) {
-    // Clear previous digits 
+  // Read the digits from the input tree
+  // See AliEMCALClusterizer::SetInput(TTree *digitsTree);    
+  
+  // Clear previous digits in the list
+  if(fgDigitsArr){ 
     fgDigitsArr->Clear("C");
-    //delete fgDigitsArr;
   }
+  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;
-  }
-  fgDigitsArr = new TClonesArray("AliEMCALDigit",100);
+  }  
+  
   branch->SetAddress(&fgDigitsArr);
   branch->GetEntry(0);
 }
 
+//==================================================================================
+Bool_t AliEMCALReconstructor::CalculateResidual(AliESDtrack *track, AliESDCaloCluster *cluster, Double_t &dEta, Double_t &dPhi)const
+{
+  //
+  // calculate the residual between track and cluster
+  //
+
+  // If the esdFriend is available, use the TPCOuter point as the starting point of extrapolation
+  // Otherwise use the TPCInner point
+  AliExternalTrackParam *trkParam = 0;
+  const AliESDfriendTrack*  friendTrack = track->GetFriendTrack();
+  if(friendTrack && friendTrack->GetTPCOut())
+    trkParam = const_cast<AliExternalTrackParam*>(friendTrack->GetTPCOut());
+  else
+    trkParam = const_cast<AliExternalTrackParam*>(track->GetInnerParam());
+  if(!trkParam) return kFALSE;
+
+  //Perform extrapolation
+  Double_t trkPos[3];
+  Float_t  clsPos[3];
+
+  AliExternalTrackParam trkParamTmp (*trkParam);
+  cluster->GetPosition(clsPos);
+  TVector3 vec(clsPos[0],clsPos[1],clsPos[2]);
+  Double_t alpha =  ((int)(vec.Phi()*TMath::RadToDeg()/20)+0.5)*20*TMath::DegToRad();
+  //Rotate to proper local coordinate
+  vec.RotateZ(-alpha); 
+  trkParamTmp.Rotate(alpha); 
+  //extrapolation is done here
+  if(!AliTrackerBase::PropagateTrackToBxByBz(&trkParamTmp, vec.X(), track->GetMass(), GetRecParam()->GetExtrapolateStep(), kFALSE, 0.8, -1)) 
+    return kFALSE; 
+
+  //Calculate the residuals
+  trkParamTmp.GetXYZ(trkPos); 
+   
+  TVector3 clsPosVec(clsPos[0],clsPos[1],clsPos[2]);
+  TVector3 trkPosVec(trkPos[0],trkPos[1],trkPos[2]);
+      
+  Double_t clsPhi = clsPosVec.Phi();
+  if(clsPhi<0) clsPhi+=2*TMath::Pi();
+  Double_t trkPhi = trkPosVec.Phi();
+  if(trkPhi<0) trkPhi+=2*TMath::Pi();
+
+  dPhi = clsPhi-trkPhi;
+  dEta = clsPosVec.Eta()-trkPosVec.Eta();
+
+  return kTRUE;
+}
+
+//
+//==================================================================================
+//
+AliEMCALReconstructor::AliEMCALMatch::AliEMCALMatch() 
+  : TObject(),  
+    fIndexT(-1), 
+    fDistance(-999.),
+    fdEta(-999.),
+    fdPhi(-999.)
+{
+  //default constructor
 
+}
+
+//
+//==================================================================================
+//
+AliEMCALReconstructor::AliEMCALMatch::AliEMCALMatch(const AliEMCALMatch& copy)
+  : TObject(),
+    fIndexT(copy.fIndexT),
+    fDistance(copy.fDistance),
+    fdEta(copy.fdEta),
+    fdPhi(copy.fdPhi)
+{
+  //copy ctor
+}
+
+//
+//==================================================================================
+//
+Int_t AliEMCALReconstructor::AliEMCALMatch::Compare(const TObject *obj) const 
+{
+  //
+  // Compare wrt the residual
+  //
+       
+  AliEMCALReconstructor::AliEMCALMatch *that = (AliEMCALReconstructor::AliEMCALMatch*)obj;
+       
+  Double_t thisDist = fDistance;//fDistance;
+  Double_t thatDist = that->fDistance;//that->GetDistance();
+       
+  if (thisDist > thatDist) return 1;
+  else if (thisDist < thatDist) return -1;
+  return 0;
+}