EMCAL/AliEMCALRecPoint.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 /* $Id$ */
  16 //_________________________________________________________________________
  17 //  Base Class for EMCAL Reconstructed Points
  18 //  Why should I put meaningless comments
  19 //  just to satisfy
  20 //  the code checker
  21 //*-- Author: Gines Martinez (SUBATECH)
  22
  23 // --- ROOT system ---
  24 #include "TPad.h"
  25 #include "TClonesArray.h"
  26
  27 // --- Standard library ---
  28 #include <stdio.h>
  29
  30 // --- AliRoot header files ---
  31
  32 #include "AliEMCALGeometry.h"
  33 #include "AliEMCALDigit.h"
  34 #include "AliEMCALRecPoint.h"
  35 #include "AliEMCALGetter.h"
  36
  37 ClassImp(AliEMCALRecPoint)
  38
  39
  40 //____________________________________________________________________________
  41 AliEMCALRecPoint::AliEMCALRecPoint()
  42   : AliRecPoint()
  43 {
  44   // ctor
  45
  46   fMaxTrack = 0 ;
  47   fTheta = fPhi = 0. ;
  48   fEMCALArm = 0;
  49   fTower = kFALSE ;
  50 }
  51
  52 //____________________________________________________________________________
  53 AliEMCALRecPoint::AliEMCALRecPoint(const char * opt) : AliRecPoint(opt)
  54 {
  55   // ctor
  56
  57   fMaxTrack = 200 ;
  58   fTheta = fPhi = 0. ;
  59   fEMCALArm = 1;
  60
  61 }
  62
  63 //____________________________________________________________________________
  64 Int_t AliEMCALRecPoint::DistancetoPrimitive(Int_t px, Int_t py)
  65 {
  66   // Compute distance from point px,py to  a AliEMCALRecPoint considered as a Tmarker
  67   // Compute the closest distance of approach from point px,py to this marker.
  68   // The distance is computed in pixels units.
  69
  70   TVector3 pos(0.,0.,0.) ;
  71   GetLocalPosition( pos) ;
  72   Float_t x =  pos.X() ;
  73   Float_t y =  pos.Z() ;
  74   const Int_t kMaxDiff = 10;
  75   Int_t pxm  = gPad->XtoAbsPixel(x);
  76   Int_t pym  = gPad->YtoAbsPixel(y);
  77   Int_t dist = (px-pxm)*(px-pxm) + (py-pym)*(py-pym);
  78
  79   if (dist > kMaxDiff) return 9999;
  80   return dist;
  81 }
  82
  83 //___________________________________________________________________________
  84  void AliEMCALRecPoint::Draw(Option_t *option)
  85  {
  86    // Draw this AliEMCALRecPoint with its current attributes
  87
  88    AppendPad(option);
  89  }
  90
  91 //______________________________________________________________________________
  92 void AliEMCALRecPoint::ExecuteEvent(Int_t event, Int_t px, Int_t py)
  93 {
  94   // Execute action corresponding to one event
  95   // This member function is called when a AliEMCALRecPoint is clicked with the locator
  96   //
  97   // If Left button is clicked on AliEMCALRecPoint, the digits are switched on
  98   // and switched off when the mouse button is released.
  99
 100   //  static Int_t pxold, pyold;
 101
 102   static TGraph *  digitgraph = 0 ;
 103   static TPaveText* clustertext = 0 ;
 104
 105   if (!gPad->IsEditable()) return;
 106
 107   switch (event) {
 108
 109
 110   case kButton1Down:{
 111     AliEMCALDigit * digit ;
 112     AliEMCALGetter * gime = AliEMCALGetter::GetInstance() ;
 113     AliEMCALGeometry * emcalgeom =  const_cast<AliEMCALGeometry*>(gime->EMCALGeometry());
 114
 115     Int_t iDigit;
 116     Int_t relid[4] ;
 117
 118     const Int_t kMulDigit=AliEMCALRecPoint::GetDigitsMultiplicity() ;
 119     Float_t * xi = new Float_t [kMulDigit] ;
 120     Float_t * zi = new Float_t [kMulDigit] ;
 121
 122     for(iDigit = 0; iDigit < kMulDigit; iDigit++) {
 123       Fatal("AliEMCALRecPoint::ExecuteEvent", " -> Something wrong with the code");
 124       digit = 0 ; //dynamic_cast<AliEMCALDigit *>((fDigitsList)[iDigit]);
 125       emcalgeom->AbsToRelNumbering(digit->GetId(), relid) ;
 126       emcalgeom->PosInAlice(relid, xi[iDigit], zi[iDigit]) ;
 127     }
 128
 129     if (!digitgraph) {
 130       digitgraph = new TGraph(fMulDigit,xi,zi);
 131       digitgraph-> SetMarkerStyle(5) ;
 132       digitgraph-> SetMarkerSize(1.) ;
 133       digitgraph-> SetMarkerColor(1) ;
 134       digitgraph-> Draw("P") ;
 135     }
 136     if (!clustertext) {
 137
 138       TVector3 pos(0.,0.,0.) ;
 139       GetLocalPosition(pos) ;
 140       clustertext = new TPaveText(pos.X()-10,pos.Z()+10,pos.X()+50,pos.Z()+35,"") ;
 141       Text_t  line1[40] ;
 142       Text_t  line2[40] ;
 143       sprintf(line1,"Energy=%1.2f GeV",GetEnergy()) ;
 144       sprintf(line2,"%d Digits",GetDigitsMultiplicity()) ;
 145       clustertext ->AddText(line1) ;
 146       clustertext ->AddText(line2) ;
 147       clustertext ->Draw("");
 148     }
 149     gPad->Update() ;
 150     Print() ;
 151     delete[] xi ;
 152     delete[] zi ;
 153    }
 154
 155 break;
 156
 157   case kButton1Up:
 158     if (digitgraph) {
 159       delete digitgraph  ;
 160       digitgraph = 0 ;
 161     }
 162     if (clustertext) {
 163       delete clustertext ;
 164       clustertext = 0 ;
 165     }
 166
 167     break;
 168
 169   }
 170 }
 171 //____________________________________________________________________________
 172 void AliEMCALRecPoint::EvalAll(Float_t logWeight,TClonesArray * digits) {
 173   //evaluates (if necessary) all RecPoint data members
 174
 175   EvalPrimaries(digits) ;
 176 }
 177
 178 //____________________________________________________________________________
 179 void AliEMCALRecPoint::EvalEMCALArm(AliEMCALDigit * digit)
 180 {
 181   // Returns the EMCAL module in which the RecPoint is found
 182
 183
 184   if( fEMCALArm == 0){
 185   Int_t relid[4] ;
 186
 187   AliEMCALGetter * gime = AliEMCALGetter::GetInstance() ;
 188   AliEMCALGeometry * emcalgeom =  const_cast<AliEMCALGeometry*>(gime->EMCALGeometry());
 189
 190   emcalgeom->AbsToRelNumbering(digit->GetId(), relid) ;
 191   fEMCALArm = relid[0];
 192   }
 193 }
 194
 195 //______________________________________________________________________________
 196 void  AliEMCALRecPoint::EvalPrimaries(TClonesArray * digits)
 197 {
 198   // Constructs the list of primary particles (tracks) which have contributed to this RecPoint
 199
 200   AliEMCALDigit * digit ;
 201   Int_t * tempo    = new Int_t[fMaxTrack] ;
 202
 203   Int_t index ;
 204   for ( index = 0 ; index < GetDigitsMultiplicity() ; index++ ) { // all digits
 205     digit = dynamic_cast<AliEMCALDigit *>(digits->At( fDigitsList[index] )) ;
 206     Int_t nprimaries = digit->GetNprimary() ;
 207     Int_t * newprimaryarray = new Int_t[nprimaries] ;
 208     Int_t ii ;
 209     for ( ii = 0 ; ii < nprimaries ; ii++)
 210       newprimaryarray[ii] = digit->GetPrimary(ii+1) ;
 211
 212     Int_t jndex ;
 213     for ( jndex = 0 ; jndex < nprimaries ; jndex++ ) { // all primaries in digit
 214       if ( fMulTrack > fMaxTrack ) {
 215         fMulTrack = - 1 ;
 216         Error("GetNprimaries", "increase fMaxTrack ")  ;
 217         break ;
 218       }
 219       Int_t newprimary = newprimaryarray[jndex] ;
 220       Int_t kndex ;
 221       Bool_t already = kFALSE ;
 222       for ( kndex = 0 ; kndex < fMulTrack ; kndex++ ) { //check if not already stored
 223         if ( newprimary == tempo[kndex] ){
 224           already = kTRUE ;
 225           break ;
 226         }
 227       } // end of check
 228       if ( !already) { // store it
 229         tempo[fMulTrack] = newprimary ;
 230         fMulTrack++ ;
 231       } // store it
 232     } // all primaries in digit
 233     delete newprimaryarray ;
 234   } // all digits
 235
 236
 237   fTracksList = new Int_t[fMulTrack] ;
 238   for(index = 0; index < fMulTrack; index++)
 239    fTracksList[index] = tempo[index] ;
 240
 241   delete tempo ;
 242
 243 }
 244
 245 //____________________________________________________________________________
 246 void AliEMCALRecPoint::GetGlobalPosition(TVector3 & gpos) const
 247 {
 248   // returns the position of the cluster in the global reference system of ALICE
 249
 250   AliEMCALGeometry * emcalgeom = AliEMCALGetter::GetInstance()->EMCALGeometry();
 251   gpos.SetX(fPhi) ;
 252   if (IsTower() )
 253     gpos.SetY(emcalgeom->GetIP2Tower()) ;
 254   else
 255     gpos.SetY(emcalgeom->GetIP2PreShower()) ;
 256   gpos.SetZ(fTheta) ;
 257 }
 258
 259
 260 //______________________________________________________________________________
 261 void AliEMCALRecPoint::Paint(Option_t *)
 262 {
 263   // Paint this ALiRecPoint as a TMarker  with its current attributes
 264
 265   TVector3 pos(0.,0.,0.)  ;
 266   GetLocalPosition(pos)   ;
 267   Coord_t x = pos.X()     ;
 268   Coord_t y = pos.Z()     ;
 269   Color_t markercolor = 1 ;
 270   Size_t  markersize = 1. ;
 271   Style_t markerstyle = 5 ;
 272
 273   if (!gPad->IsBatch()) {
 274     gVirtualX->SetMarkerColor(markercolor) ;
 275     gVirtualX->SetMarkerSize (markersize)  ;
 276     gVirtualX->SetMarkerStyle(markerstyle) ;
 277   }
 278   gPad->SetAttMarkerPS(markercolor,markerstyle,markersize) ;
 279   gPad->PaintPolyMarker(1,&x,&y,"") ;
 280 }