ITS/AliITSVertexer.cxx

   1 #include <AliESDVertex.h>
   2 #include "AliITSgeomTGeo.h"
   3 #include "AliITSVertexer.h"
   4 #include "AliRunLoader.h"
   5 #include "AliITSLoader.h"
   6 #include "AliMultiplicity.h"
   7 #include "AliITSMultReconstructor.h"
   8
   9 const Float_t AliITSVertexer::fgkPipeRadius = 3.0;
  10
  11 ClassImp(AliITSVertexer)
  12
  13 //////////////////////////////////////////////////////////////////////
  14 // Base class for primary vertex reconstruction                     //
  15 // AliESDVertexer is a class for full 3D primary vertex finding     //
  16 // derived classes: AliITSVertexerIons AliITSvertexer3D             //
  17 //                  AliITSVertexerCosmics                           //
  18 //////////////////////////////////////////////////////////////////////
  19
  20 //______________________________________________________________________
  21 AliITSVertexer::AliITSVertexer():AliVertexer(),
  22 fLadders(),
  23 fLadOnLay2(0)    {
  24   // Default Constructor
  25   SetLaddersOnLayer2();
  26 }
  27
  28 AliITSVertexer::AliITSVertexer(TString filename):AliVertexer(),
  29 fLadders(),
  30 fLadOnLay2(0)
  31 {
  32   // Standard constructor
  33   AliRunLoader *rl = AliRunLoader::GetRunLoader();
  34   if(!rl){
  35     Fatal("AliITSVertexer","Run Loader not found");
  36   }
  37   /*
  38   if(rl->LoadgAlice()){
  39     Fatal("AliITSVertexer","The AliRun object is not available - nothing done");
  40   }
  41   */
  42   fCurrentVertex  = 0;
  43   SetFirstEvent(0);
  44   SetLastEvent(0);
  45   //  rl->LoadHeader();
  46   AliITSLoader* itsLoader =  (AliITSLoader*) rl->GetLoader("ITSLoader");
  47   if(!filename.Contains("default"))itsLoader->SetVerticesFileName(filename);
  48   if(!filename.Contains("null"))itsLoader->LoadVertices("recreate");
  49
  50   //  Int_t lst;
  51   SetLastEvent(rl->GetNumberOfEvents()-1);
  52   /*
  53   if(rl->TreeE()){
  54     lst = static_cast<Int_t>(rl->TreeE()->GetEntries());
  55     SetLastEvent(lst-1);
  56   }
  57   */
  58   SetLaddersOnLayer2();
  59 }
  60
  61 //______________________________________________________________________
  62 AliITSVertexer::AliITSVertexer(const AliITSVertexer &vtxr) : AliVertexer(vtxr),
  63 fLadders(),
  64 fLadOnLay2(0)
  65 {
  66   // Copy constructor
  67   // Copies are not allowed. The method is protected to avoid misuse.
  68   Error("AliITSVertexer","Copy constructor not allowed\n");
  69 }
  70
  71 //______________________________________________________________________
  72 AliITSVertexer& AliITSVertexer::operator=(const AliITSVertexer& /* vtxr */){
  73   // Assignment operator
  74   // Assignment is not allowed. The method is protected to avoid misuse.
  75   Error("= operator","Assignment operator not allowed\n");
  76   return *this;
  77 }
  78
  79 //______________________________________________________________________
  80 AliITSVertexer::~AliITSVertexer() {
  81   // Destructor
  82  if(fLadders) delete [] fLadders;
  83 }
  84
  85 //______________________________________________________________________
  86 void AliITSVertexer::FindMultiplicity(Int_t evnumber){
  87   // Invokes AliITSMultReconstructor to determine the
  88   // charged multiplicity in the pixel layers
  89   if(fMult){delete fMult; fMult = 0;}
  90   Bool_t success=kTRUE;
  91   if(!fCurrentVertex)success=kFALSE;
  92   if(fCurrentVertex && fCurrentVertex->GetNContributors()<1)success=kFALSE;
  93   if(!success){
  94     AliWarning("Tracklets multiplicity not determined because the primary vertex was not found");
  95     return;
  96   }
  97   AliITSMultReconstructor* multReco = new AliITSMultReconstructor();
  98   AliRunLoader *rl =AliRunLoader::GetRunLoader();
  99   AliITSLoader* itsLoader = (AliITSLoader*)rl->GetLoader("ITSLoader");
 100   itsLoader->LoadRecPoints();
 101   rl->GetEvent(evnumber);
 102   TTree* itsClusterTree = itsLoader->TreeR();
 103   if (!itsClusterTree) {
 104     AliError(" Can't get the ITS cluster tree !\n");
 105     return;
 106   }
 107   Double_t vtx[3];
 108   fCurrentVertex->GetXYZ(vtx);
 109   Float_t vtxf[3];
 110   for(Int_t i=0;i<3;i++)vtxf[i]=vtx[i];
 111   multReco->SetHistOn(kFALSE);
 112   multReco->Reconstruct(itsClusterTree,vtxf,vtxf);
 113   Int_t notracks=multReco->GetNTracklets();
 114   Float_t *tht = new Float_t [notracks];
 115   Float_t *phi = new Float_t [notracks];
 116   Float_t *dphi = new Float_t [notracks];
 117   Int_t *labels = new Int_t[notracks];
 118   Int_t *labelsL2 = new Int_t[notracks];
 119   for(Int_t i=0;i<multReco->GetNTracklets();i++){
 120     tht[i] = multReco->GetTracklet(i)[0];
 121     phi[i] =  multReco->GetTracklet(i)[1];
 122     dphi[i] = multReco->GetTracklet(i)[2];
 123     labels[i] = static_cast<Int_t>(multReco->GetTracklet(i)[3]);
 124     labelsL2[i] = static_cast<Int_t>(multReco->GetTracklet(i)[4]);
 125   }
 126   Int_t nosingleclus=multReco->GetNSingleClusters();
 127   Float_t *ths = new Float_t [nosingleclus];
 128   Float_t *phs = new Float_t [nosingleclus];
 129   for(Int_t i=0;i<nosingleclus;i++){
 130     ths[i] = multReco->GetCluster(i)[0];
 131     phs[i] =  multReco->GetCluster(i)[1];
 132   }
 133   fMult = new AliMultiplicity(notracks,tht,phi,dphi,labels,labelsL2,nosingleclus,ths,phs);
 134   delete [] tht;
 135   delete [] phi;
 136   delete [] dphi;
 137   delete [] ths;
 138   delete [] phs;
 139   delete [] labels;
 140   delete [] labelsL2;
 141   itsLoader->UnloadRecPoints();
 142   delete multReco;
 143   return;
 144 }
 145
 146 //______________________________________________________________________
 147 void AliITSVertexer::SetLaddersOnLayer2(Int_t ladwid){
 148   // Calculates the array of ladders on layer 2 to be used with a
 149   // given ladder on layer 1
 150   fLadOnLay2=ladwid;
 151   //  AliRunLoader *rl =AliRunLoader::GetRunLoader();
 152   //  AliITSLoader* itsLoader = (AliITSLoader*)rl->GetLoader("ITSLoader");
 153   //  AliITSgeom* geom = itsLoader->GetITSgeom();
 154   Int_t ladtot1=AliITSgeomTGeo::GetNLadders(1);
 155   if(fLadders) delete [] fLadders;
 156   fLadders=new UShort_t[ladtot1];
 157
 158
 159   Double_t pos1[3],pos2[3];
 160   Int_t mod1=AliITSgeomTGeo::GetModuleIndex(2,1,1);
 161   AliITSgeomTGeo::GetTranslation(mod1,pos1);  // position of the module in the MRS
 162   Double_t phi0=TMath::ATan2(pos1[1],pos1[0]);
 163   if(phi0<0) phi0+=2*TMath::Pi();
 164   Int_t mod2=AliITSgeomTGeo::GetModuleIndex(2,2,1);
 165   AliITSgeomTGeo::GetTranslation(mod2,pos2);
 166   Double_t phi2=TMath::ATan2(pos2[1],pos2[0]);
 167   if(phi2<0) phi2+=2*TMath::Pi();
 168   Double_t deltaPhi= phi0-phi2; // phi width of a layer2 module
 169
 170   for(Int_t i= 0; i<ladtot1;i++){
 171     Int_t modlad= AliITSgeomTGeo::GetModuleIndex(1,i+1,1);
 172     Double_t posmod[3];
 173     AliITSgeomTGeo::GetTranslation(modlad,posmod);
 174     Double_t phimod=TMath::ATan2(posmod[1],posmod[0]);
 175     if(phimod<0) phimod+=2*TMath::Pi();
 176     Double_t phi1= phimod+deltaPhi*double(fLadOnLay2);
 177     if(phi1<0) phi1+=2*TMath::Pi();
 178     if(phi1>2*TMath::Pi()) phi1-=2*TMath::Pi();
 179     Double_t philad1=phi0-phi1;
 180     UShort_t lad1;
 181     Double_t ladder1=(philad1)/(deltaPhi) +1.;
 182     if(ladder1<1){ladder1=40+ladder1;}
 183     lad1=int(ladder1+0.5);
 184     fLadders[i]=lad1;
 185   }
 186 }
 187
 188
 189 //______________________________________________________________________
 190 void AliITSVertexer::WriteCurrentVertex(){
 191   // Write the current AliVertex object to file fOutFile
 192   AliRunLoader *rl = AliRunLoader::GetRunLoader();
 193   AliITSLoader* itsLoader =  (AliITSLoader*) rl->GetLoader("ITSLoader");
 194   fCurrentVertex->SetName("Vertex");
 195   //  const char * name = fCurrentVertex->GetName();
 196   //  itsLoader->SetVerticesContName(name);
 197   Int_t rc = itsLoader->PostVertex(fCurrentVertex);
 198   rc = itsLoader->WriteVertices();
 199 }
 200