]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - EMCAL/AliEMCALReconstructor.cxx
Cluster with index 0 now considered for track matching
[u/mrichter/AliRoot.git] / EMCAL / AliEMCALReconstructor.cxx
index 08d0a9c3ef2c3c8862f156122b4b07dcc73a96d5..a9cff4de608d3a4ae217a9fffeaee3ef19d18fd4 100644 (file)
 // --- AliRoot header files ---
 #include "AliEMCALReconstructor.h"
 
+#include "AliRun.h"
 #include "AliESD.h"
+#include "AliESDtrack.h"
 #include "AliRunLoader.h"
 #include "AliEMCALLoader.h"
+#include "AliEMCALRawUtils.h"
 #include "AliEMCALClusterizerv1.h"
 #include "AliEMCALRecPoint.h"
 #include "AliEMCALPID.h"
+#include "AliEMCALTrigger.h"
 #include "AliRawReader.h"
 
 
@@ -72,58 +76,151 @@ void AliEMCALReconstructor::Reconstruct(AliRunLoader* runLoader) const
   TString branchName(runLoader->GetEventFolder()->GetName() ) ;  
   
   AliEMCALClusterizerv1 clu(headerFile, branchName);
-  clu.SetEventRange(0, -1) ; // do all the events
-  if ( Debug() ) 
-    clu.ExecuteTask("deb all") ; 
-  else 
-    clu.ExecuteTask("pseudo") ;  
+  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() ; 
 }
 
 //____________________________________________________________________________
-void AliEMCALReconstructor::Reconstruct(AliRunLoader* runLoader, AliRawReader* rawreader) const 
+void AliEMCALReconstructor::Reconstruct(AliRunLoader* runLoader, AliRawReader* rawReader) 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.
-  // Here we reconstruct from Raw Data
+  // 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.
   
-  rawreader->Reset() ; 
   TString headerFile(runLoader->GetFileName()) ; 
   TString branchName(runLoader->GetEventFolder()->GetName()) ;  
 
+  static AliEMCALRawUtils rawUtils;
   AliEMCALClusterizerv1 clu(headerFile, branchName);
-  clu.SetEventRange(0, -1) ; // do all the events
-  if ( Debug() ) 
-    clu.ExecuteTask("deb pseudo all") ; 
-  else 
-    clu.ExecuteTask("pseudo") ;  
+  AliEMCALLoader *emcalLoader = dynamic_cast<AliEMCALLoader*>(runLoader->GetDetectorLoader("EMCAL"));    
+  Int_t iEvent = 0;
+  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() ; 
+  }
 }
 
 //____________________________________________________________________________
 void AliEMCALReconstructor::FillESD(AliRunLoader* runLoader, AliESD* esd) const
 {
-  // Called by AliReconstruct after Reconstruct() and global tracking and vertxing 
+  // 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() ;
 
-  TString headerFile(runLoader->GetFileName()) ; 
-  TString branchName(runLoader->GetEventFolder()->GetName()) ;  
   // Creates AliESDCaloCluster from AliEMCALRecPoints 
-  AliRunLoader *rl = AliRunLoader::GetRunLoader();
-  AliEMCALLoader *emcalLoader = dynamic_cast<AliEMCALLoader*>(rl->GetDetectorLoader("EMCAL"));
-  rl->LoadRecPoints();
-  rl->LoadKinematics(); // To get the primary label
-  rl->LoadDigits();     // To get the primary label
-  rl->LoadHits();       // To get the primary label
-  rl->GetEvent(eventNumber);
+  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 
-  //  esd->SetNumberOfEMCALClusters(nClusters); // have to be change - Feb 25, 2007; some cluster may be discard
+
+  //######################################################
+  //#########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);
+  
+  //######################################################
+  //#######################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);
@@ -148,16 +245,20 @@ void AliEMCALReconstructor::FillESD(AliRunLoader* runLoader, AliESD* esd) const
 
     // 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;
+    Int_t newdigitMult = 0; 
     for (Int_t iDigit=0; iDigit<digitMult; iDigit++) {
-      if(timeFloat[iDigit] < 65536/1e9*100) {
+      if (amplFloat[iDigit] > 0) {
        amplList[newdigitMult] = (UShort_t)(amplFloat[iDigit]*500);
-        if(amplList[newdigitMult] > 0) { // accept digit if poztive amplitude
-         timeList[newdigitMult] = (UShort_t)(timeFloat[iDigit]*1e9*100); // Time in units of 100 ns = 0.1 ps
-         digiList[newdigitMult] = (UShort_t)(digitInts[iDigit]);
-          newdigitMult++;
-       }
+        // 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++;
       }
+      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
@@ -180,34 +281,48 @@ void AliEMCALReconstructor::FillESD(AliRunLoader* runLoader, AliESD* esd) const
         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]);       
+
+
       // fills the ESDCaloCluster
       AliESDCaloCluster * ec = new AliESDCaloCluster() ; 
+      ec->SetEMCAL(kTRUE);
       ec->SetClusterType(clust->GetClusterType());
       ec->SetGlobalPosition(xyz);
       ec->SetClusterEnergy(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
-        ec->SetPrimaryIndex(clust->GetPrimaryIndex());
+        ec->SetTrackMatchedIndex( matchedTrack[iClust]); //track matched 
       } 
-    // add the cluster to the esd object
+
+      // add the cluster to the esd object
       esd->AddCaloCluster(ec);
       delete ec;
+
     } else { // no new ESD cluster
         delete [] amplList;
         delete [] timeList;
         delete [] digiList;
     }
   } // cycle on clusters
-  //  printf(" %i : nClusters %i : RP %i PC %i \n", eventNumber, nClusters, nRP, nPC);
   esd->SetNumberOfEMCALClusters(nClustersNew);
   //if(nClustersNew != nClusters) 
   //printf(" ##### nClusters %i -> new %i ##### \n", nClusters, nClustersNew );
@@ -219,3 +334,4 @@ void AliEMCALReconstructor::FillESD(AliRunLoader* runLoader, AliESD* esd) const
   pid->RunPID(esd);
 }
 
+