PHOS/AliPHOSFastRecParticle.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 //  A  Particle modified by PHOS response and produced by AliPHOSvFast
  20 //  To become a general class of AliRoot ?
  21 //
  22 //*-- Author: Yves Schutz (SUBATECH)
  23
  24 // --- ROOT system ---
  25
  26 // --- Standard library ---
  27
  28 #include <iostream.h>
  29
  30 // --- AliRoot header files ---
  31
  32 #include "AliPHOSFastRecParticle.h"
  33 #include "TPad.h"
  34 #include "TPaveText.h"
  35
  36 ClassImp(AliPHOSFastRecParticle) ;
  37
  38 //____________________________________________________________________________
  39 AliPHOSFastRecParticle::AliPHOSFastRecParticle() : TParticle()
  40 {
  41   // ctor
  42   fType = 0 ;
  43 }
  44
  45 //____________________________________________________________________________
  46  AliPHOSFastRecParticle::AliPHOSFastRecParticle(const AliPHOSFastRecParticle & rp)
  47 {
  48   // copy ctor
  49
  50   fType        = rp.fType ;
  51   fPdgCode     = rp.fPdgCode;
  52   fStatusCode  = rp.fStatusCode;
  53   fMother[0]   = rp.fMother[0];
  54   fMother[1]   = rp.fMother[1];
  55   fDaughter[0] = rp.fDaughter[0];
  56   fDaughter[1] = rp.fDaughter[1];
  57   fWeight      = rp.fWeight;
  58   fCalcMass    = rp.fCalcMass;
  59   fPx          = rp.fPx;
  60   fPy          = rp.fPy;
  61   fPz          = rp.fPz;
  62   fE           = rp.fE;
  63   fVx          = rp.fVx;
  64   fVy          = rp.fVy;
  65   fVz          = rp.fVz;
  66   fVt          = rp.fVt;
  67   fPolarTheta  = rp.fPolarTheta;
  68   fPolarPhi    = rp.fPolarPhi;
  69   fParticlePDG = rp.fParticlePDG;
  70 }
  71
  72 //____________________________________________________________________________
  73  AliPHOSFastRecParticle::AliPHOSFastRecParticle(const TParticle & pp)
  74 {
  75   // ctor from a TParticle (crummy?!)
  76
  77   TParticle & pnoconst = (TParticle &)(pp) ;
  78   AliPHOSFastRecParticle & p = (AliPHOSFastRecParticle &)(pnoconst) ;
  79   fType        = 0  ;
  80   fPdgCode     = p.fPdgCode;
  81   fStatusCode  = p.fStatusCode;
  82   fMother[0]   = p.fMother[0];
  83   fMother[1]   = p.fMother[1];
  84   fDaughter[0] = p.fDaughter[0];
  85   fDaughter[1] = p.fDaughter[1];
  86   fWeight      = p.fWeight;
  87   fCalcMass    = p.fCalcMass;
  88   fPx          = p.fPx;
  89   fPy          = p.fPy;
  90   fPz          = p.fPz;
  91   fE           = p.fE;
  92   fVx          = p.fVx;
  93   fVy          = p.fVy;
  94   fVz          = p.fVz;
  95   fVt          = p.fVt;
  96   fPolarTheta  = p.fPolarTheta;
  97   fPolarPhi    = p.fPolarPhi;
  98   fParticlePDG = p.fParticlePDG;
  99
 100 }
 101
 102 //____________________________________________________________________________
 103 Int_t AliPHOSFastRecParticle::DistancetoPrimitive(Int_t px, Int_t py)
 104 {
 105   //  Compute distance from point px,py to a AliPHOSFastRecParticle considered as a Tmarker
 106   //  Compute the closest distance of approach from point px,py to this marker.
 107   //  The distance is computed in pixels units.
 108
 109   Double_t kRADDEG = 180. / TMath::Pi() ;
 110   Coord_t x = Phi() * kRADDEG     ;
 111   Coord_t y = Theta() * kRADDEG     ;
 112   const Int_t kMaxDiff = 10;
 113   Int_t pxm  = gPad->XtoAbsPixel(x);
 114   Int_t pym  = gPad->YtoAbsPixel(y);
 115   Int_t dist = (px-pxm)*(px-pxm) + (py-pym)*(py-pym);
 116
 117   if (dist > kMaxDiff) return 9999;
 118   return dist;
 119 }
 120
 121 //___________________________________________________________________________
 122  void AliPHOSFastRecParticle::Draw(Option_t *option)
 123  {
 124    // Draw this AliPHOSFastRecParticle with its current attributes
 125
 126    AppendPad(option);
 127  }
 128
 129 //______________________________________________________________________________
 130 void AliPHOSFastRecParticle::ExecuteEvent(Int_t event, Int_t px, Int_t py)
 131 {
 132   //  Execute action corresponding to one event
 133   //  This member function is called when a AliPHOSFastRecParticle is clicked with the locator
 134
 135   if (!gPad->IsEditable())
 136     return ;
 137
 138   static TPaveText * clustertext = 0 ;
 139
 140   switch (event) {
 141
 142   case kButton1Down: {
 143     Double_t kRADDEG = 180. / TMath::Pi() ;
 144     Coord_t x = Phi() * kRADDEG     ;
 145     Coord_t y = Theta() * kRADDEG     ;
 146     clustertext = new TPaveText(x-1, y+1, x+5, y+3, "") ;
 147     Text_t  line1[40] ;
 148     Text_t  line2[40] ;
 149     sprintf( line1, "PID: %s ", (const char*)Name() ) ;
 150     sprintf( line2, "ENERGY: %f ", Energy() ) ;
 151     clustertext ->AddText(line1) ;
 152     clustertext ->AddText(line2) ;
 153     clustertext ->Draw("");
 154     gPad->Update() ;
 155     break ;
 156   }
 157
 158   case kButton1Up: {
 159     delete clustertext ;
 160     clustertext = 0 ;
 161     gPad->Update() ;
 162     break ;
 163   }
 164   }
 165
 166 }
 167
 168 //____________________________________________________________________________
 169 TString AliPHOSFastRecParticle::Name()
 170 {
 171   // Returns the name of the particle type (only valid if PIDv1 is employed)
 172
 173   TString  name ;
 174
 175   if(fType == 127)
 176     name = "PHOTON_LOPU_HIEF" ;    //PCA = 001 TOF = 111 CPV = 111
 177
 178   if(fType == 511)
 179     name = "PHOTON_HIPU_LOEF" ;    //PCA = 011 TOF = 111 CPV = 111
 180
 181   if(fType == 255)
 182         name = "PHOTON_MED_PU_EF" ;    //PCA = 111 TOF = 111 CPV = 111
 183
 184   if((fType == 383)||(fType == 447))
 185     name = "PHOTON_STRANGE" ;      //PCA = 101 or 110 TOF = 111 CPV = 111
 186
 187   if(fType == 63)
 188     name = "NEUTRAL_FAST_HADRON" ; //PCA = 000 TOF = 111 CPV = 111
 189
 190   if((fType == 504) || (fType == 505) ||(fType == 248)||(fType == 249)||(fType == 120)||(fType == 121))
 191     name = "CHARGED_FAST_EM" ;     //PCA = 111, 011 or 001 TOF =111 CPV = 000 or 001
 192
 193   if((fType == 56)||(fType == 57))
 194     name = "CHARGED_FAST_HADRON" ; //PCA = 000 TOF = 111 CPV = 000 or 001
 195
 196   if((fType < 8)&&(fType > 0))
 197     name = "NEUTRAL_SLOW_HADRON" ; //PCA = 000 TOF = 000 CPV = 001 or 011 or 111
 198
 199   if((fType == 0))
 200     name = "CHARGED_SLOW_HADRON" ; //PCA = 000 TOF = 000 CPV = 000
 201
 202   if((fType == 448) || (fType == 449) ||(fType == 192)||(fType == 193)||(fType == 64)||(fType == 64))
 203     name = "CHARGED_SLOW_EM" ;    //PCA = 111, 011 or 001 TOF =000 CPV = 000 or 001
 204
 205
 206   if ( name.Contains("PHOTON") )
 207     fPdgCode = 22 ; // Sets the pdg GetName() to gamma
 208   if ( name.Contains("NEUTRAL") )
 209     fPdgCode = 2112 ; // Sets the pdg GetName() to neutron
 210   if ( name.Contains("CHARGED_FAST_EM") )
 211     fPdgCode = 11 ; // Sets the pdg GetName() to electron
 212   if ( name.Contains("CHARGED_SLOW_EM") )
 213     fPdgCode = 13 ; // Sets the pdg GetName() to muon
 214   if ( name.Contains("CHARGED_FAST_HADRON") )
 215     fPdgCode = 211 ; // Sets the pdg GetName() to pion
 216   if ( name.Contains("CHARGED_SLOW_HADRON") )
 217     fPdgCode = 2212 ; // Sets the pdg GetName() to proton
 218
 219   return name ;
 220 }
 221
 222 //______________________________________________________________________________
 223 void AliPHOSFastRecParticle::Paint(Option_t *)
 224 {
 225   // Paint this ALiRecParticle in theta,phi coordinate as a TMarker  with its current attributes
 226
 227   Double_t kRADDEG = 180. / TMath::Pi() ;
 228   Coord_t x = Phi() * kRADDEG     ;
 229   Coord_t y = Theta() * kRADDEG     ;
 230   Color_t markercolor = 1 ;
 231   Size_t  markersize  = 1. ;
 232   Style_t markerstyle = 5 ;
 233
 234   if (!gPad->IsBatch()) {
 235     gVirtualX->SetMarkerColor(markercolor) ;
 236     gVirtualX->SetMarkerSize (markersize)  ;
 237     gVirtualX->SetMarkerStyle(markerstyle) ;
 238   }
 239   gPad->SetAttMarkerPS(markercolor,markerstyle,markersize) ;
 240   gPad->PaintPolyMarker(1,&x,&y,"") ;
 241 }
 242
 243 //____________________________________________________________________________
 244 void AliPHOSFastRecParticle::Print(const char * opt)
 245 {
 246   // Print the type, energy and momentum of the reconstructed particle
 247
 248   cout << "AliPHOSFastRecParticle > " << "type is  " << Name() << endl
 249        << "                     " << "Energy = " << fE << endl
 250        << "                     " << "Px     = " << fPx << endl
 251        << "                     " << "Py     = " << fPy << endl
 252        << "                     " << "Pz     = " << fPz << endl ;
 253 }