Thread libraries filtered out
[u/mrichter/AliRoot.git] / TOF / AliTOFMerger.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 #include <TTree.h> 
17 #include <TVector.h>
18 #include <TObjArray.h>
19 #include <TFile.h>
20 #include <TDirectory.h>
21
22
23 #include "AliTOFMerger.h"
24 #include "AliTOF.h"
25 #include "AliTOFSDigitizer.h"
26 #include "AliTOFhit.h"
27 #include "AliTOFdigit.h"
28
29 #include "AliRun.h"
30 #include "AliPDG.h"
31
32 #include <stdlib.h>
33 #include <iostream.h>
34 #include <fstream.h>
35
36 ClassImp(AliTOFMerger)
37
38 //___________________________________________
39   AliTOFMerger::AliTOFMerger() 
40 {
41 // Default ctor    
42     fNDigits = 0;
43     fEvNrSig = 0;
44     fEvNrBgr = 0;
45     fMerge =kDigitize;
46     fDigits = 0;
47     fSDigits =0;
48     fFnBgr = 0;
49     fFnSig = 0;
50     fBgrFile = 0;
51 }
52
53 //------------------------------------------------------------------------
54 AliTOFMerger::~AliTOFMerger()
55 {
56 // Dtor
57   if(fSDigits)  {
58     fSDigits->Delete();
59     delete fSDigits ;
60     fSDigits = 0;
61   }
62   delete fBgrFile;
63   fBgrFile = 0;
64   if(fFnBgr) {
65     delete[] fFnBgr;
66     fFnBgr = 0;
67   }
68   if(fFnSig) {
69     delete[] fFnSig;
70     fFnSig = 0;
71   }
72 }
73
74
75 //------------------------------------------------------------------------
76 void AliTOFMerger::Init()
77 {
78 // Initialisation
79     if (fMerge) fBgrFile = InitBgr();
80     
81 }
82
83
84
85 //------------------------------------------------------------------------
86 TFile* AliTOFMerger::InitBgr()
87 {
88 // Initialise background event
89     TFile *file = new TFile(fFnBgr);
90 // add error checking later
91     printf("\n AliTOFMerger has opened %s file with background event \n", fFnBgr);
92     return file;
93 }
94
95 //------------------------------------------------------------------------
96 void AliTOFMerger::Digitise()
97 {
98 // as in FMD
99 // keep galice.root for signal and name differently the file for 
100 // background when add! otherwise the track info for signal will be lost !
101
102
103
104 #ifdef DEBUG
105   cout<<"ALiTOFMerger::>SDigits2Digits start...\n";
106 #endif
107
108   AliTOF* TOF = (AliTOF *) gAlice->GetDetector("TOF") ;
109
110
111   TFile *f1 =0;
112   TTree *TK = gAlice->TreeK();
113   if (TK) f1 = TK->GetCurrentFile();
114
115   gAlice->GetEvent(fEvNrSig) ;
116   
117   if(gAlice->TreeD() == 0)      
118     gAlice->MakeTree("D") ;
119   gAlice->TreeD()->Reset();
120
121   // read and write digits for signal
122    ReadWriteDigit(fEvNrSig);
123
124    if(fMerge){ 
125      // bgr file
126     fBgrFile->cd();
127     // gAlice->TreeS()->Reset();
128     gAlice = (AliRun*)fBgrFile->Get("gAlice");
129     ReadWriteDigit(fEvNrBgr);
130    } //if merge
131
132
133   f1->cd();
134    
135   //Make branch for digits
136   TOF->MakeBranch("D");
137
138   gAlice->TreeD()->Reset();
139   gAlice->TreeD()->Fill();
140   
141   fDigits   = TOF->Digits();
142   
143   gAlice->TreeD()->Write(0,TObject::kOverwrite) ;
144   
145   gAlice->ResetDigits();
146
147 }
148
149 //---------------------------------------------------------------------
150
151 void AliTOFMerger::ReadWriteDigit(Int_t iEvNum)
152 {
153 //
154 // Read Sdigits from the current file and write them in the TreeD
155 //
156   AliTOFdigit* tofsdigit;
157   
158   AliTOF * tofinfile = (AliTOF *) gAlice->GetDetector("TOF") ;
159   
160   gAlice->GetEvent(iEvNum) ;
161   if(gAlice->TreeS()==0) {
162     cout<<" TreeS==0 -> return"<<gAlice->TreeS()<<endl; 
163     return ;}
164   
165   Int_t ndig, k;
166   Int_t    tracks[3];    // track info
167   Int_t    vol[5];       // location for a digit
168   Float_t  digit[2];     // TOF digit variables
169
170   gAlice->ResetDigits();
171   gAlice->TreeS()->GetEvent(iEvNum);
172   TClonesArray * TOFdigits   = tofinfile->SDigits();
173   
174   ndig=TOFdigits->GetEntries();
175
176   for (k=0; k<ndig; k++) {
177     tofsdigit= (AliTOFdigit*) TOFdigits->UncheckedAt(k);
178
179         tracks[0] = tofsdigit->GetTrack(0);
180         tracks[1] = tofsdigit->GetTrack(1);
181         tracks[2] = tofsdigit->GetTrack(2);
182
183         vol[0] = tofsdigit->GetSector();
184         vol[1] = tofsdigit->GetPlate();
185         vol[2] = tofsdigit->GetStrip();
186         vol[3] = tofsdigit->GetPadx();
187         vol[4] = tofsdigit->GetPadz();
188
189         digit[0] = tofsdigit->GetTdc();
190         digit[1] = tofsdigit->GetAdc();
191
192         new ((*fDigits)[fNDigits++]) AliTOFdigit(tracks, vol, digit);
193   } // end loop on sdigits in the current file
194 }
195
196
197