Changes related to the extraction of the V0 finder into a separate class (A. Dainese...
[u/mrichter/AliRoot.git] / ITS / AliITSVertexerFast.cxx
1 /**************************************************************************
2  * Copyright(c) 1998-2003, 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 #include <Riostream.h>
16 #include <TArrayF.h>
17 #include <TRandom.h>
18 #include "AliESDVertex.h"
19 #include <AliITSVertexerFast.h>
20 #include "AliHeader.h"
21 #include "AliGenEventHeader.h"
22 #include "AliRun.h"
23 #include "AliITSLoader.h"
24 #include "AliRunLoader.h"
25
26 /////////////////////////////////////////////////////////////////////////
27 //                                                                     //
28 // Fast vertexer - True (i.e. generated) vertex coordinates            //
29 // are smeared with gaussians of given width                           //
30 // Origin: masera@to.infn.it     25/09/2003                            //
31 //                                                                     //
32 /////////////////////////////////////////////////////////////////////////
33 ClassImp(AliITSVertexerFast)
34
35
36
37 //______________________________________________________________________
38 AliITSVertexerFast::AliITSVertexerFast():AliITSVertexer(),
39 fSmear(0) 
40 {
41   // Default Constructor
42   fSmear = 0;
43   AliRunLoader *rl =AliRunLoader::Instance();
44   TTree *trK=(TTree*)rl->TreeK();
45   if(!trK)AliFatal("This class should be used only with simulated events!!");
46   rl->LoadHeader(); 
47 }
48
49 //______________________________________________________________________
50 AliITSVertexerFast::AliITSVertexerFast(Double_t *smear):AliITSVertexer(),
51 fSmear(0)
52 {
53   // Standard constructor
54   fSmear = new Double_t[3];
55   for(Int_t i=0;i<3;i++)fSmear[i]=smear[i];
56   AliInfo(Form("Gaussian smaring of the generated vertex. Sigmas (x,y,z) = %12.5f , %12.5f , %12.5f cm",fSmear[0],fSmear[1],fSmear[2]));
57   AliRunLoader *rl =AliRunLoader::Instance();
58   TTree *trK=(TTree*)rl->TreeK();
59   if(!trK)AliFatal("This class should be used only with simulated events!!");
60   rl->LoadHeader(); 
61
62 }
63
64
65 //______________________________________________________________________
66 AliITSVertexerFast::~AliITSVertexerFast(){
67   // Destructor
68   if(fSmear)delete [] fSmear;
69   fSmear = 0;
70 }
71
72 //______________________________________________________________________
73 AliESDVertex* AliITSVertexerFast::FindVertexForCurrentEvent(TTree *itsClusterTree){
74   // Defines the AliITSVertex for the current event
75   AliWarning(Form("This class should be used only with simulated events!! Input cluster tree (%p) will not be used!!",itsClusterTree));
76
77   fCurrentVertex = 0;
78   AliRunLoader *rl =AliRunLoader::Instance();
79   TArrayF primaryVertex(3);  // true vertex
80   AliHeader* header = rl->GetHeader();
81   AliGenEventHeader* genEventHeader = header->GenEventHeader();   
82   genEventHeader->PrimaryVertex(primaryVertex); 
83   
84   // Smearing
85   Double_t vrttrue[3],vrtx[3];
86   for(Int_t k=0; k<3;k++){
87     vrttrue[k] = static_cast<Double_t>(primaryVertex[k]);
88     vrtx[k] = gRandom->Gaus(vrttrue[k],fSmear[k]);
89   }
90   fCurrentVertex = new AliESDVertex(vrtx,fSmear,"Smeared Generated Vertex");
91   fCurrentVertex->SetTitle("vertexer: smearMC");
92   return fCurrentVertex;
93   
94 }
95
96 //________________________________________________________
97 void AliITSVertexerFast::PrintStatus() const {
98   // Print current status
99   cout <<"=======================================================\n";
100
101   cout<<"RMS for gaussian smearing: ";
102   for(Int_t k=0;k<3;k++)cout<<" "<<fSmear[k];
103   cout<<endl;
104 }
105