Formatting changes.
[u/mrichter/AliRoot.git] / MUON / AliMUONClusterReconstructor.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 // -----------------------------------
19 // Class AliMUONClusterReconstructor
20 // ----------------------------------
21 // MUON cluster reconstructor for MUON
22 // Should implement a virtual class ClusterFinder to choose between VS and AZ method
23
24 #include "AliMUONClusterReconstructor.h"
25 #include "AliRun.h" // for gAlice
26 #include "AliRunLoader.h"
27 #include "AliLoader.h"
28
29 #include "AliMUON.h"
30 #include "AliMUONDigit.h"
31 #include "AliMUONConstants.h"
32 #include "AliMUONData.h"
33 #include "AliMUONClusterFinderVS.h"
34 #include "AliMUONClusterInput.h"
35 #include "AliMUONRawCluster.h"
36
37 #include "AliMpDEManager.h"
38
39 #include "AliLog.h"
40
41 /// \cond CLASSIMP
42 ClassImp(AliMUONClusterReconstructor) // Class implementation in ROOT context
43 /// \endcond
44  
45 //__________________________________________________________________________
46 AliMUONClusterReconstructor::AliMUONClusterReconstructor(AliMUONData* data)
47 : TObject(),
48   fMUONData(data),
49   fRecModel(new AliMUONClusterFinderVS()),
50   fDigitsCath0(new TClonesArray("AliMUONDigit",1000)),
51   fDigitsCath1(new TClonesArray("AliMUONDigit",1000))
52 {
53 /// Standard Constructor
54
55   fDigitsCath0->SetOwner(kTRUE); 
56   fDigitsCath1->SetOwner(kTRUE);
57 }
58
59 //__________________________________________________________________________
60 AliMUONClusterReconstructor::~AliMUONClusterReconstructor(void)
61 {
62 /// Destructor
63
64   delete fRecModel;
65   delete fDigitsCath0;
66   delete fDigitsCath1;
67 }
68
69 //______________________________________________________________________________
70 void
71 AliMUONClusterReconstructor::ClusterizeOneDE(Int_t detElemId)
72 {
73 /// Clusterize one detection element, and let fMUONData know about
74 /// the results.
75   
76   if ( fDigitsCath0->GetEntriesFast() || fDigitsCath1->GetEntriesFast() )
77   {
78     Int_t iChamber = AliMpDEManager::GetChamberId(detElemId);
79     AliMUONClusterInput::Instance()->SetDigits(iChamber, detElemId,
80                                                fDigitsCath0,fDigitsCath1);
81     AliDebug(3,Form("ClusterizeOneDE iChamber=%d DE=%d",iChamber,detElemId));
82     StdoutToAliDebug(3,cout << "DigitsCath0=" << endl;
83                      fDigitsCath0->Print();
84                      cout << "DigitsCath1=" << endl;
85                      fDigitsCath1->Print(););
86     fRecModel->FindRawClusters();
87     
88     // copy results into the output container
89     TClonesArray* tmp = fRecModel->GetRawClusters();
90     for (Int_t id = 0; id < tmp->GetEntriesFast(); ++id) 
91     {
92       AliMUONRawCluster* pClus = (AliMUONRawCluster*) tmp->At(id);
93       fMUONData->AddRawCluster(iChamber, *pClus);
94     }        
95     
96     // Reset the arrays
97     fDigitsCath0->Clear("C");
98     fDigitsCath1->Clear("C");
99   }
100 }
101
102 //____________________________________________________________________
103 void AliMUONClusterReconstructor::Digits2Clusters(Int_t chBeg)
104 {
105 /// Clusterize all the tracking chamber digits.
106 ///
107 /// For each chamber, we loop *once* on that chamber digits, and store them
108 /// in 2 temporary arrays (one pair of arrays per detection element, 
109 /// one array per cathode). Once a pair of arrays is full (i.e. all the digits
110 /// of that detection element have been stored), we clusterize this DE, and
111 /// move to the next one.
112   
113   if (!fRecModel)
114   {
115     AliWarning("No reco model defined. Nothing to do...");
116     return;
117   }
118   
119   Int_t iChamber(-1);
120   Int_t currentDE(-1);
121   
122   // Loop on chambers 
123   for ( iChamber = chBeg; iChamber < AliMUONConstants::NTrackingCh(); ++iChamber ) 
124   {
125     TClonesArray* muonDigits = fMUONData->Digits(iChamber); 
126     
127     Int_t ndig = muonDigits->GetEntriesFast();
128     if (!ndig) continue;
129     
130     muonDigits->Sort(); // the sort *must* be per DE (at least), otherwise
131                         // the following logic with currentDE will fail.
132     
133     currentDE = -1; // initialize the DE counter (that is used to track 
134                     // when we change of DE in the following loop over
135                     // all digits) to an invalid value.
136
137     for ( Int_t k = 0; k < ndig; ++k ) 
138     {
139       AliMUONDigit* digit = (AliMUONDigit*) muonDigits->UncheckedAt(k);
140       if ( ! digit->Signal() > 0 ) continue; // skip void digits.
141       
142       if ( digit->DetElemId() != currentDE )
143       {
144         AliDebug(3,Form("Switching DE from %d to %d",currentDE,digit->DetElemId()));
145         // we get to a new DE, so clusterize the previous one before
146         // moving on.
147         ClusterizeOneDE(currentDE);
148         currentDE = digit->DetElemId();
149       }
150       
151       // Add the digit to the array with the right cathode number.
152       if (digit->Cathode() == 0)
153       {
154         new((*fDigitsCath0)[fDigitsCath0->GetLast()+1]) AliMUONDigit(*digit);
155       }
156       else 
157       {
158         new((*fDigitsCath1)[fDigitsCath1->GetLast()+1]) AliMUONDigit(*digit);
159       }
160     } // end of loop on chamber digits
161     
162     // As the above logic is based on detecting a change in DE number,
163     // the last DE of each chamber has not been clusterized, so we do 
164     // it here.
165     ClusterizeOneDE(currentDE);
166   } // end of loop over chambers
167 }
168
169 //_______________________________________________________________________
170 void 
171 AliMUONClusterReconstructor::SetRecoModel(AliMUONClusterFinderVS* rec)
172
173 /// Set reconstruction model
174
175   delete fRecModel; 
176   fRecModel = rec;
177
178
179 //_______________________________________________________________________
180 void AliMUONClusterReconstructor::Trigger2Trigger() 
181 {
182 /// Copy trigger from TreeD to TreeR
183
184   fMUONData->SetTreeAddress("GLT");
185   fMUONData->GetTriggerD();
186 }