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