track counter implemented in ESD reader

[u/mrichter/AliRoot.git] / RICH / AliRICHDigitizer.cxx

diff --git a/RICH/AliRICHDigitizer.cxx b/RICH/AliRICHDigitizer.cxx
index 1224da6f8c72b2a8816c8825263dd364a5b83ead..3eaa2806e80798452c1d596c5f8a03b292dec9a8 100644 (file)
--- a/RICH/AliRICHDigitizer.cxx
+++ b/RICH/AliRICHDigitizer.cxx
@@ -14,10 +14,6 @@
  **************************************************************************/
 
 
-//Piotr.Skowronski@cern.ch :
-//Corrections applied in order to compile (only) with new I/O and folder structure
-//To be implemented correctly by responsible
-
 #include <Riostream.h> 
 
 #include <TTree.h> 
@@ -31,59 +27,74 @@
 
 #include "AliRICHDigitizer.h"
 #include "AliRICH.h"
-#include "AliRun.h"
-#include "AliPDG.h"
 #include "AliRunDigitizer.h"
 
 ClassImp(AliRICHDigitizer)
 
 //__________________________________________________________________________________________________
-AliRICHDigitizer::AliRICHDigitizer() 
-{//default constructor
-}//default ctor
-//__________________________________________________________________________________________________
-AliRICHDigitizer::AliRICHDigitizer(AliRunDigitizer *pManager) 
-                 :AliDigitizer(pManager)
-{//main ctor which should be used
-}//main ctor
-//__________________________________________________________________________________________________
-AliRICHDigitizer::~AliRICHDigitizer()
-{//dtor
-  if(fManager->GetDebug())Info("dtor","Start.");
-}//dtor
+Bool_t AliRICHDigitizer::Init()
+{
+//This methode is called from AliRunDigitizer after the corresponding file is open
+  if(GetDebug())Info("Init","Start.");
+  AliRunLoader *pOutAL = 
+    AliRunLoader::GetRunLoader(fManager->GetOutputFolderName());
+  if (!pOutAL->GetAliRun()) pOutAL->LoadgAlice();
+  fRich=(AliRICH*)pOutAL->GetAliRun()->GetDetector("RICH");
+  Rich()->P()->GenSigmaThMap();
+  return kTRUE;
+}//Init()
 //__________________________________________________________________________________________________
 void AliRICHDigitizer::Exec(Option_t*)
 {
-
-  AliRunLoader *pInAL, *pOutAL;//in and out Run loaders
-  AliLoader    *pInRL, *pOutRL;//in and out RICH loaders
+  if(GetDebug())Info("Exec","Start with %i input(s) for event %i",fManager->GetNinputs(),fManager->GetOutputEventNr());
+  
+  AliRunLoader *pInAL=0, *pOutAL;//in and out Run loaders
+  AliLoader    *pInRL=0, *pOutRL;//in and out RICH loaders
  
   pOutAL = AliRunLoader::GetRunLoader(fManager->GetOutputFolderName());
   pOutRL = pOutAL->GetLoader("RICHLoader");
-
+  pOutRL->MakeTree("D");   Rich()->MakeBranch("D"); //create TreeD with RICH branches in output stream
   
-     
-  AliRICH *pRICH = (AliRICH *) gAlice->GetDetector("RICH");
+  TClonesArray tmpCA("AliRICHdigit");//tmp storage for sdigits sum up from all input files
+  Int_t total=0;
+  for(Int_t inFileN=0;inFileN<fManager->GetNinputs();inFileN++){//files loop
+    pInAL = AliRunLoader::GetRunLoader(fManager->GetInputFolderName(inFileN)); 
+    pInRL = pInAL->GetLoader("RICHLoader"); if(pInRL==0) continue;//no RICH in this input, check the next input
+    if (!pInAL->GetAliRun()) pInAL->LoadgAlice();
+    AliRICH* rich=(AliRICH*)pInAL->GetAliRun()->GetDetector("RICH");
+    pInRL->LoadSDigits(); pInRL->TreeS()->GetEntry(0);
+    Info("Exec","input %i has %i sdigits",inFileN,rich->SDigits()->GetEntries());
+    for(Int_t i=0;i<rich->SDigits()->GetEntries();i++) {
+      new(tmpCA[total++]) AliRICHdigit(*(AliRICHdigit*)rich->SDigits()->At(i)); 
+      ((AliRICHdigit*)tmpCA[total-1])->AddTidOffset(fManager->GetMask(inFileN));
+    }
+    pInRL->UnloadSDigits();   rich->ResetSDigits();
+  }//files loop
   
-  if(!pOutRL->TreeD()) pOutRL->MakeTree("D");  pRICH->MakeBranch("D");
+  tmpCA.Sort();                     //sort them according to Id() methode
   
+  Int_t chFbMip=0,chamber=0,x=0,y=0,tid[3],id=0; Double_t q=0;
+  Int_t iNdigitsPerPad=0;//how many sdigits for a given pad
+  for(Int_t i=0;i<tmpCA.GetEntries();i++){//sdigits loop (sorted)
+    AliRICHdigit *pSdig=(AliRICHdigit*)tmpCA.At(i);//get new sdigit
+    if(pSdig->Id()==id){//still the same pad
+      iNdigitsPerPad++;         q+=pSdig->Q();       chFbMip+=pSdig->ChFbMi();//sum up charge and cfm
+      if(iNdigitsPerPad<=3)        tid[iNdigitsPerPad-1]=pSdig->Tid(0);
+      else                         if(GetDebug())Warning("Exec","More then 3 sdigits for the given pad");
+    }else{//new pad, add the pevious one
+        if(id!=kBad&&Rich()->P()->IsOverTh(chamber,x,y,q)) Rich()->AddDigit(chamber,x,y,(Int_t)q,chFbMip,tid); //add newly created dig
+        chFbMip=pSdig->ChFbMi(); chamber=pSdig->C(); id=pSdig->Id();  x=pSdig->X(); y=pSdig->Y(); q=pSdig->Q();  //init all values by current sdig
+        iNdigitsPerPad=1; tid[0]=pSdig->Tid(0); tid[1]=tid[2]=kBad;
+      }
+  }//sdigits loop (sorted)
+  if(tmpCA.GetEntries()&&Rich()->P()->IsOverTh(chamber,x,y,q)) Rich()->AddDigit(chamber,x,y,(Int_t)q,chFbMip,tid);//add the last dig
   
-  for(Int_t inputFile=0;inputFile<fManager->GetNinputs();inputFile++){//files loop
-    pInAL = AliRunLoader::GetRunLoader(fManager->GetInputFolderName(inputFile));
-    pInRL = pInAL->GetLoader("RICHLoader");
-    
-    
-  }//files loop
-
-      
-      
-  pOutRL->TreeD()->Fill();
-
-    
-  pRICH->ResetDigits(); /// ??? should it be here???
+  pOutRL->TreeD()->Fill();              //fill the tree with the list of digits
+  pOutRL->WriteDigits("OVERWRITE");     //serialize them to file
   
-  pOutRL->WriteDigits("OVERWRITE");
-
-  pOutRL->UnloadHits();
-  pOutRL->UnloadDigits();
+  tmpCA.Clear();
+  pOutRL->UnloadDigits();   Rich()->ResetDigits();
+            
+  if(GetDebug())Info("Exec","Stop.");
 }//Exec()
+//__________________________________________________________________________________________________