EMCAL/AliEMCALv1.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 //*-- Implementation version v1 of EMCAL Manager class
  20 //*-- An object of this class does not produce digits
  21 //*-- It is the one to use if you do want to produce outputs in TREEH
  22 //*--
  23 //*-- Author: Sahal Yacoob (LBL /UCT)
  24 //*--       : Jennifer Klay (LBL)
  25 // This Class not stores information on all particles prior to EMCAL entry - in order to facilitate analysis.
  26 // This is done by setting fIShunt =2, and flagging all parents of particles entering the EMCAL.
  27
  28 // 15/02/2002 .... Yves Schutz
  29 //  1. fSamplingFraction and fLayerToPreshowerRatio have been removed
  30 //  2. Timing signal is collected and added to hit
  31
  32 // --- ROOT system ---
  33 #include <TParticle.h>
  34 #include <TVirtualMC.h>
  35
  36 // --- Standard library ---
  37
  38 // --- AliRoot header files ---
  39 #include "AliEMCALv1.h"
  40 #include "AliEMCALHit.h"
  41 #include "AliEMCALGeometry.h"
  42 #include "AliRun.h"
  43 #include "AliMC.h"
  44
  45 ClassImp(AliEMCALv1)
  46
  47
  48 //______________________________________________________________________
  49 AliEMCALv1::AliEMCALv1():AliEMCALv0(), fCurPrimary(-1), fCurParent(-1), fCurTrack(-1){
  50   // ctor
  51 }
  52
  53 //______________________________________________________________________
  54 AliEMCALv1::AliEMCALv1(const char *name, const char *title):
  55     AliEMCALv0(name,title), fCurPrimary(-1), fCurParent(-1), fCurTrack(-1) {
  56     // Standard Creator.
  57
  58     fHits= new TClonesArray("AliEMCALHit",1000);
  59     //    gAlice->GetMCApp()->AddHitList(fHits); // 20-dec-04 - advice of Andreas
  60
  61     fNhits = 0;
  62     fIshunt     =  2; // All hits are associated with particles entering the calorimeter
  63     fTimeCut = 30e-09;
  64 }
  65
  66 //______________________________________________________________________
  67 AliEMCALv1::~AliEMCALv1(){
  68     // dtor
  69
  70     if ( fHits) {
  71         fHits->Delete();
  72         delete fHits;
  73         fHits = 0;
  74     }
  75 }
  76
  77 //______________________________________________________________________
  78 void AliEMCALv1::AddHit(Int_t shunt, Int_t primary, Int_t tracknumber, Int_t iparent, Float_t ienergy,
  79                         Int_t id, Float_t * hits,Float_t * p){
  80     // Add a hit to the hit list.
  81     // An EMCAL hit is the sum of all hits in a tower section
  82     //   originating from the same entering particle
  83     Int_t hitCounter;
  84
  85     AliEMCALHit *newHit;
  86     AliEMCALHit *curHit;
  87     Bool_t deja = kFALSE;
  88
  89     newHit = new AliEMCALHit(shunt, primary, tracknumber, iparent, ienergy, id, hits, p);
  90      for ( hitCounter = fNhits-1; hitCounter >= 0 && !deja; hitCounter-- ) {
  91         curHit = (AliEMCALHit*) (*fHits)[hitCounter];
  92         // We add hits with the same tracknumber, while GEANT treats
  93         // primaries succesively
  94         if(curHit->GetPrimary() != primary)
  95           break;
  96         if( *curHit == *newHit ) {
  97             *curHit = *curHit + *newHit;
  98             deja = kTRUE;
  99             } // end if
 100     } // end for hitCounter
 101
 102     if ( !deja ) {
 103         new((*fHits)[fNhits]) AliEMCALHit(*newHit);
 104         fNhits++;
 105     } // end if
 106
 107     delete newHit;
 108 }
 109 //______________________________________________________________________
 110 void AliEMCALv1::StepManager(void){
 111   // Accumulates hits as long as the track stays in a tower
 112
 113   Int_t          id[2];           // (phi, Eta) indices
 114   // position wrt MRS and energy deposited
 115   Float_t        xyzte[5]={0.,0.,0.,0.,0.};// position wrt MRS, time and energy deposited
 116   Float_t        pmom[4]={0.,0.,0.,0.};
 117   TLorentzVector pos; // Lorentz vector of the track current position.
 118   TLorentzVector mom; // Lorentz vector of the track current momentum.
 119   Int_t tracknumber =  gAlice->GetMCApp()->GetCurrentTrackNumber();
 120   static Float_t ienergy = 0;
 121   Int_t copy = 0;
 122
 123   AliEMCALGeometry * geom = GetGeometry() ;
 124
 125   static Int_t idXPHI = gMC->VolId("XPHI");
 126   if(gMC->CurrentVolID(copy) == idXPHI ) { // We are in a Scintillator Layer
 127     Float_t depositedEnergy ;
 128
 129     if( ((depositedEnergy = gMC->Edep()) > 0.)  && (gMC->TrackTime() < fTimeCut)){// Track is inside a scintillator and deposits some energy
 130        if (fCurPrimary==-1)
 131         fCurPrimary=gAlice->GetMCApp()->GetPrimary(tracknumber);
 132
 133       if (fCurParent==-1 || tracknumber != fCurTrack) {
 134         // Check parentage
 135         Int_t parent=tracknumber;
 136         if (fCurParent != -1) {
 137           while (parent != fCurParent && parent != -1) {
 138             TParticle *part=gAlice->GetMCApp()->Particle(parent);
 139             parent=part->GetFirstMother();
 140           }
 141         }
 142         if (fCurParent==-1 || parent==-1) {
 143           Int_t parent=tracknumber;
 144           TParticle *part=gAlice->GetMCApp()->Particle(parent);
 145           while (parent != -1 && geom->IsInEMCAL(part->Vx(),part->Vy(),part->Vz())) {
 146             parent=part->GetFirstMother();
 147             if (parent!=-1)
 148               part=gAlice->GetMCApp()->Particle(parent);
 149           }
 150           fCurParent=parent;
 151           if (fCurParent==-1)
 152             Error("StepManager","Cannot find parent");
 153           else {
 154             TParticle *part=gAlice->GetMCApp()->Particle(fCurParent);
 155             ienergy = part->Energy();
 156           }
 157           while (parent != -1) {
 158             part=gAlice->GetMCApp()->Particle(parent);
 159             part->SetBit(kKeepBit);
 160             parent=part->GetFirstMother();
 161           }
 162         }
 163         fCurTrack=tracknumber;
 164       }
 165       gMC->TrackPosition(pos);
 166       xyzte[0] = pos[0];
 167       xyzte[1] = pos[1];
 168       xyzte[2] = pos[2];
 169       xyzte[3] = gMC->TrackTime() ;
 170
 171       gMC->TrackMomentum(mom);
 172       pmom[0] = mom[0];
 173       pmom[1] = mom[1];
 174       pmom[2] = mom[2];
 175       pmom[3] = mom[3];
 176
 177       gMC->CurrentVolOffID(1, id[0]); // get the POLY copy number;
 178       gMC->CurrentVolID(id[1]); // get the phi number inside the layer
 179
 180       Int_t tower = (id[0]-1) % geom->GetNZ() + 1 + (id[1] - 1) * geom->GetNZ() ;
 181       Int_t layer = static_cast<Int_t>((id[0]-1)/(geom->GetNZ())) + 1 ;
 182       Int_t absid = tower;
 183       Int_t nlayers = geom->GetNECLayers();
 184       if ((layer > nlayers)||(layer<1))
 185         Fatal("StepManager", "Wrong calculation of layer number: layer = %d > %d\n", layer, nlayers) ;
 186
 187       Float_t lightYield =  depositedEnergy ;
 188       // Apply Birk's law (copied from G3BIRK)
 189
 190       if (gMC->TrackCharge()!=0) { // Check
 191           Float_t birkC1Mod = 0;
 192         if (fBirkC0==1){ // Apply correction for higher charge states
 193           if (TMath::Abs(gMC->TrackCharge())>=2)
 194             birkC1Mod=fBirkC1*7.2/12.6;
 195           else
 196             birkC1Mod=fBirkC1;
 197         }
 198         Float_t dedxcm;
 199         if (gMC->TrackStep()>0)
 200           dedxcm=1000.*gMC->Edep()/gMC->TrackStep();
 201         else
 202           dedxcm=0;
 203         lightYield=lightYield/(1.+birkC1Mod*dedxcm+fBirkC2*dedxcm*dedxcm);
 204       }
 205
 206       // use sampling fraction to get original energy --HG
 207       xyzte[4] = lightYield * geom->GetSampling();
 208
 209       if (gDebug == 2)
 210         printf("StepManager: id0 = %d, id1 = %d, absid = %d tower = %d layer = %d energy = %f\n", id[0], id[1], absid, tower, layer, xyzte[4]) ;
 211
 212       AddHit(fIshunt, fCurPrimary,tracknumber, fCurParent, ienergy, absid,  xyzte, pmom);
 213     } // there is deposited energy
 214   }
 215 }
 216
 217 void AliEMCALv1::RemapTrackHitIDs(Int_t *map) {
 218   // remap track index numbers for primary and parent indices
 219   // (Called by AliStack::PurifyKine)
 220   if (Hits()==0)
 221     return;
 222   TIter hitIter(Hits());
 223   Int_t iHit=0;
 224   while (AliEMCALHit *hit=dynamic_cast<AliEMCALHit*>(hitIter()) ) {
 225     if (map[hit->GetIparent()]==-99)
 226       cout << "Remapping, found -99 for parent id " << hit->GetIparent() << ", " << map[hit->GetIparent()] << ", iHit " << iHit << endl;
 227     hit->SetIparent(map[hit->GetIparent()]);
 228     if (map[hit->GetPrimary()]==-99)
 229       cout << "Remapping, found -99 for primary id " << hit->GetPrimary() << ", " << map[hit->GetPrimary()] << ", iHit " << iHit << endl;
 230     hit->SetPrimary(map[hit->GetPrimary()]);
 231     iHit++;
 232   }
 233 }
 234
 235 void AliEMCALv1::FinishPrimary() {
 236   // finish primary
 237   fCurPrimary=-1;
 238   fCurParent=-1;
 239   fCurTrack=-1;
 240 }