a7a589f3065fc7c600eee8a977924c14db15647e
[u/mrichter/AliRoot.git] / MUON / AliMUONSDigitizerV2.cxx
1 /**************************************************************************
2 * Copyright(c) 1998-1999, 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 // $Id$
17
18 #include <TClonesArray.h>
19
20 #include "AliMUONSDigitizerV2.h"
21
22 #include "AliLog.h"
23 #include "AliMUON.h"
24 #include "AliMUONChamber.h"
25 #include "AliMUONVDigit.h"
26 #include "AliMUONHit.h"
27 #include "AliMpDEManager.h"
28 #include "AliMpCDB.h"
29 #include "AliLoader.h"
30 #include "AliRun.h"
31 #include "AliRunLoader.h"
32 #include "AliMUONVDigitStore.h"
33 #include "AliMUONVHitStore.h"
34
35 //-----------------------------------------------------------------------------
36 /// The sdigitizer performs the transformation from hits (energy deposits by
37 /// the transport code) to sdigits (equivalent of charges on pad).
38 ///
39 /// It does so by converting the energy deposit into a charge and then spreading
40 /// this charge over several pads, according to the response function (a 
41 /// Mathieson distribution, basically).
42 /// 
43 /// See also AliMUONResponseV0, which is doing the real job (in DisIntegrate
44 /// method), while this sdigitizer is just "steering" the process.
45 ///
46 /// Please note that we do *not* merge sdigits after creation, which means
47 /// that after sdigitization, a given pad can have several sdigits. This
48 /// merging is taken care of later on by the digitizer(V3).
49 //-----------------------------------------------------------------------------
50
51 ClassImp(AliMUONSDigitizerV2)
52
53 //_____________________________________________________________________________
54 AliMUONSDigitizerV2::AliMUONSDigitizerV2() 
55 : TTask("AliMUONSDigitizerV2","From Hits to SDigits for MUON")
56 {
57   ///
58   /// ctor.
59   ///
60   if ( ! AliMpCDB::LoadMpSegmentation() ) 
61   {
62     AliFatal("Could not access mapping from OCDB !");
63   }
64 }
65
66 //_____________________________________________________________________________
67 AliMUONSDigitizerV2::~AliMUONSDigitizerV2()
68 {
69   ///
70   /// dtor.
71   ///
72 }
73
74 //_____________________________________________________________________________
75 void
76 AliMUONSDigitizerV2::Exec(Option_t*)
77 {
78   ///
79   /// Go from hits to sdigits.
80   ///
81   /// In the code below, apart from the loop itself (which look complicated
82   /// but is really only a loop on each hit in the input file) the main
83   /// work is done in AliMUONResponse::DisIntegrate method, which converts
84   /// a single hit in (possibly) several sdigits.
85   ///
86   
87   AliDebug(1,"");
88   
89   AliRunLoader* runLoader = AliRunLoader::GetRunLoader();
90   AliLoader* loader = runLoader->GetDetectorLoader("MUON");
91
92   loader->LoadHits("READ");
93   
94   AliMUON* muon = static_cast<AliMUON*>(gAlice->GetModule("MUON"));
95     
96   Int_t nofEvents(runLoader->GetNumberOfEvents());
97   
98   TString classname = muon->DigitStoreClassName();
99   
100   AliMUONVDigitStore* sDigitStore = AliMUONVDigitStore::Create(classname.Data());
101   
102   if (!sDigitStore)
103   {
104     AliFatal(Form("Could not create digitstore of class %s",classname.Data()));
105   }
106   
107   AliInfo(Form("Will use digitStore of type %s",sDigitStore->ClassName()));
108           
109   for ( Int_t iEvent = 0; iEvent < nofEvents; ++iEvent ) 
110   {    
111     // Loop over events.
112     TObjArray tdlist;
113     tdlist.SetOwner(kTRUE);
114     
115     AliDebug(1,Form("iEvent=%d",iEvent));
116     runLoader->GetEvent(iEvent);
117   
118     loader->MakeSDigitsContainer();
119
120     TTree* treeS = loader->TreeS();
121
122     if ( !treeS )
123     {
124       AliFatal("");
125     }
126
127     sDigitStore->Connect(*treeS);
128     
129     TTree* treeH = loader->TreeH();
130
131     AliMUONVHitStore* hitStore = AliMUONVHitStore::Create(*treeH);
132     hitStore->Connect(*treeH);
133     
134     Long64_t nofTracks = treeH->GetEntries();
135     
136     for ( Long64_t iTrack = 0; iTrack < nofTracks; ++iTrack )
137     {
138       // Loop over the tracks of this event.
139       treeH->GetEvent(iTrack);
140
141       AliMUONHit* hit;
142       TIter next(hitStore->CreateIterator());
143       Int_t ihit(0);
144       
145       while ( ( hit = static_cast<AliMUONHit*>(next()) ) )       
146       {
147         Int_t chamberId = hit->Chamber()-1;
148         AliMUONChamber& chamber = muon->Chamber(chamberId);
149         AliMUONResponse* response = chamber.ResponseModel();
150         
151         // This is the heart of this method : the dis-integration
152         TList digits;        
153         response->DisIntegrate(*hit,digits);
154         
155         TIter next(&digits);
156         AliMUONVDigit* d;
157         while ( ( d = (AliMUONVDigit*)next() ) )
158         {
159           // Update some sdigit information that could not be known
160           // by the DisIntegrate method
161           d->SetHit(ihit);
162           d->AddTrack(iTrack,d->Charge());
163           tdlist.Add(d);
164         }
165         ++ihit;
166       }
167       hitStore->Clear();
168     } // end of loop on tracks within an event
169     
170     TIter next(&tdlist);
171     AliMUONVDigit* d;
172     
173     while ( ( d = static_cast<AliMUONVDigit*>(next()) ) )
174     {
175       if ( d->Charge() > 0 ) // that check would be better in the disintegrate
176         // method, but to compare with old sdigitizer, it has to be there.
177       {
178         AliMUONVDigit* added = sDigitStore->Add(*d,AliMUONVDigitStore::kMerge);
179         if (!added)
180         {
181           AliError("Could not add digit to digitStore");
182         }
183       }
184     }
185
186     treeS->Fill();
187     
188     loader->WriteSDigits("OVERWRITE");
189     
190     sDigitStore->Clear();
191     
192     loader->UnloadSDigits();
193     
194     delete hitStore;
195     
196   } // loop on events
197   
198   loader->UnloadHits();  
199   
200   delete sDigitStore;
201 }