9fa326153c4f194ba94318c9c6afe298826db6c5
[u/mrichter/AliRoot.git] / RICH / AliRICHDigitizer.cxx
1 /**************************************************************************
2  * Copyright(c) 1998-2000, ALICE Experiment at CERN, All rights reserved. *
3  *                                                                        *
4  * Author: The ALICE Off-line Project.                                    *
5  * Contributors are mentioned in the code where appropriate.              *
6  *                                                                        *
7  * Permission to use, copy, modify and distribute this software and its   *
8  * documentation strictly for non-commercial purposes is hereby granted   *
9  * without fee, provided that the above copyright notice appears in all   *
10  * copies and that both the copyright notice and this permission notice   *
11  * appear in the supporting documentation. The authors make no claims     *
12  * about the suitability of this software for any purpose. It is          *
13  * provided "as is" without express or implied warranty.                  *
14  **************************************************************************/
15
16
17 #include <Riostream.h> 
18
19 #include <TTree.h> 
20 #include <TObjArray.h>
21 #include <TFile.h>
22 #include <TDirectory.h>
23 #include <TParticle.h>
24
25 #include <AliRunLoader.h>
26 #include <AliLoader.h>
27
28 #include "AliRICHDigitizer.h"
29 #include "AliRICH.h"
30 #include "AliRunDigitizer.h"
31
32 ClassImp(AliRICHDigitizer)
33
34 //__________________________________________________________________________________________________
35 Bool_t AliRICHDigitizer::Init()
36 {
37 //This methode is called from AliRunDigitizer after the corresponding file is open
38   if(GetDebug())Info("Init","Start.");
39   fRich=(AliRICH*)gAlice->GetDetector("RICH");
40   Rich()->Param()->GenSigmaThMap();
41   return kTRUE;
42 }//Init()
43 //__________________________________________________________________________________________________
44 void AliRICHDigitizer::Exec(Option_t*)
45 {
46   if(GetDebug())Info("Exec","Start with %i input(s) for event %i",fManager->GetNinputs(),fManager->GetOutputEventNr());
47   
48   AliRunLoader *pInAL=0, *pOutAL;//in and out Run loaders
49   AliLoader    *pInRL=0, *pOutRL;//in and out RICH loaders
50  
51   pOutAL = AliRunLoader::GetRunLoader(fManager->GetOutputFolderName());
52   pOutRL = pOutAL->GetLoader("RICHLoader");
53   pOutRL->MakeTree("D");   Rich()->MakeBranch("D"); //create TreeD with RICH branches in output stream
54   for(Int_t inFileN=0;inFileN<fManager->GetNinputs();inFileN++){//files loop
55     pInAL = AliRunLoader::GetRunLoader(fManager->GetInputFolderName(inFileN)); pInRL = pInAL->GetLoader("RICHLoader");
56     pInRL->LoadSDigits(); pInAL->GetEvent(fManager->GetOutputEventNr()); pInRL->TreeS()->GetEntry(0);
57     pInRL->UnloadSDigits();
58   }//files loop
59   
60   Rich()->SDigits()->Sort();                     //sort them according to Id() methode
61   Int_t combiPid=0,chamber=0,x=0,y=0,tid[3],id=0; Double_t q=0;
62   Int_t iNdigitsPerPad=0;//how many sdigits for a given pad
63   for(Int_t i=0;i<Rich()->SDigits()->GetEntries();i++){//sdigits loop (sorted)
64     AliRICHdigit *pSdig=(AliRICHdigit*)Rich()->SDigits()->At(i);
65     if(pSdig->Id()==id){//still the same pad
66       iNdigitsPerPad++;         q+=pSdig->Q();       combiPid+=pSdig->CombiPid();//sum up charge and cfm
67       if(iNdigitsPerPad<=3)        tid[iNdigitsPerPad-1]=pSdig->Tid(0);
68       else                         Warning("SDigits2Digits","More then 3 sdigits for the given pad");
69     }else{//new pad, add the pevious one
70         if(id!=kBad&&Rich()->Param()->IsOverTh(chamber,x,y,q)) Rich()->AddDigit(chamber,x,y,(Int_t)q,combiPid,tid); //add newly created dig
71         combiPid=pSdig->CombiPid(); chamber=pSdig->C(); id=pSdig->Id();  x=pSdig->X(); y=pSdig->Y(); q=pSdig->Q();  //init all values by current sdig
72         iNdigitsPerPad=1; tid[0]=pSdig->Tid(0); tid[1]=tid[2]=kBad;
73       }
74   }//sdigits loop (sorted)
75   if(Rich()->SDigits()->GetEntries()&&Rich()->Param()->IsOverTh(chamber,x,y,q)) Rich()->AddDigit(chamber,x,y,(Int_t)q,combiPid,tid);//add the last dig
76   pOutRL->TreeD()->Fill();              //fill the tree with the list of digits
77   pOutRL->WriteDigits("OVERWRITE");     //serialize them to file
78             
79   if(GetDebug())Info("Exec","Stop.");
80 }//Exec()
81 //__________________________________________________________________________________________________