#include <TClonesArray.h>
#include <TParticle.h>
#include <TVirtualMC.h>
#include "AliEMCALv1.h"
#include "AliEMCALHit.h"
#include "AliEMCALGeometry.h"
#include "AliRun.h"
#include "AliMC.h"
#include "AliStack.h"
using std::cout;
using std::endl;
ClassImp(AliEMCALv1)
AliEMCALv1::AliEMCALv1()
: AliEMCALv0(),
fCurPrimary(-1),
fCurParent(-1),
fCurTrack(-1),
fTimeCut(30e-09)
{
}
AliEMCALv1::AliEMCALv1(const char *name, const char *title,
const Bool_t checkGeoAndRun)
: AliEMCALv0(name,title,checkGeoAndRun),
fCurPrimary(-1),
fCurParent(-1),
fCurTrack(-1),
fTimeCut(30e-09)
{
fHits= new TClonesArray("AliEMCALHit",1000);
fNhits = 0;
fIshunt = 2;
}
AliEMCALv1::~AliEMCALv1(){
if ( fHits ) {
fHits->Clear();
delete fHits;
fHits = 0;
}
}
void AliEMCALv1::AddHit(Int_t shunt, Int_t primary, Int_t tracknumber, Int_t iparent, Float_t ienergy,
Int_t id, Float_t * hits,Float_t * p){
Int_t hitCounter=0;
AliEMCALHit *newHit=0;
AliEMCALHit *curHit=0;
Bool_t deja = kFALSE;
newHit = new AliEMCALHit(shunt, primary, tracknumber, iparent, ienergy, id, hits, p);
for ( hitCounter = fNhits-1; hitCounter >= 0 && !deja; hitCounter-- ) {
curHit = (AliEMCALHit*) (*fHits)[hitCounter];
if(curHit->GetPrimary() != primary)
break;
if( *curHit == *newHit ) {
*curHit = *curHit + *newHit;
deja = kTRUE;
}
}
if ( !deja ) {
new((*fHits)[fNhits]) AliEMCALHit(*newHit);
fNhits++;
}
delete newHit;
}
void AliEMCALv1::StepManager(void){
Int_t id[2]={0,0};
Float_t xyzte[5]={0.,0.,0.,0.,0.};
Float_t pmom[4]={0.,0.,0.,0.};
TLorentzVector pos;
TLorentzVector mom;
Int_t tracknumber = gAlice->GetMCApp()->GetCurrentTrackNumber();
static Float_t ienergy = 0;
Int_t copy = 0;
AliEMCALGeometry * geom = GetGeometry() ;
TParticle *part=0;
Int_t parent=-1;
static Int_t idXPHI = TVirtualMC::GetMC()->VolId("XPHI");
if(TVirtualMC::GetMC()->CurrentVolID(copy) == idXPHI ) {
Float_t depositedEnergy ;
if( ((depositedEnergy = TVirtualMC::GetMC()->Edep()) > 0.) && (TVirtualMC::GetMC()->TrackTime() < fTimeCut)){
if (fCurPrimary==-1)
fCurPrimary=gAlice->GetMCApp()->GetPrimary(tracknumber);
if (fCurParent==-1 || tracknumber != fCurTrack) {
parent=tracknumber;
if (fCurParent != -1) {
while (parent != fCurParent && parent != -1) {
part=gAlice->GetMCApp()->Particle(parent);
parent=part->GetFirstMother();
}
}
if (fCurParent==-1 || parent==-1) {
parent=tracknumber;
part=gAlice->GetMCApp()->Particle(parent);
while (parent != -1 && geom->IsInEMCAL(part->Vx(),part->Vy(),part->Vz())) {
parent=part->GetFirstMother();
if (parent!=-1)
part=gAlice->GetMCApp()->Particle(parent);
}
fCurParent=parent;
if (fCurParent==-1)
Error("StepManager","Cannot find parent");
else {
part=gAlice->GetMCApp()->Particle(fCurParent);
ienergy = part->Energy();
}
while (parent != -1) {
part=gAlice->GetMCApp()->Particle(parent);
part->SetBit(kKeepBit);
parent=part->GetFirstMother();
}
}
fCurTrack=tracknumber;
}
TVirtualMC::GetMC()->TrackPosition(pos);
xyzte[0] = pos[0];
xyzte[1] = pos[1];
xyzte[2] = pos[2];
xyzte[3] = TVirtualMC::GetMC()->TrackTime() ;
TVirtualMC::GetMC()->TrackMomentum(mom);
pmom[0] = mom[0];
pmom[1] = mom[1];
pmom[2] = mom[2];
pmom[3] = mom[3];
TVirtualMC::GetMC()->CurrentVolOffID(1, id[0]);
TVirtualMC::GetMC()->CurrentVolID(id[1]);
Int_t tower = (id[0]-1) % geom->GetNZ() + 1 + (id[1] - 1) * geom->GetNZ() ;
Int_t layer = static_cast<Int_t>((id[0]-1)/(geom->GetNZ())) + 1 ;
Int_t absid = tower;
Int_t nlayers = geom->GetNECLayers();
if ((layer > nlayers)||(layer<1))
Fatal("StepManager", "Wrong calculation of layer number: layer = %d > %d\n", layer, nlayers) ;
Float_t lightYield = depositedEnergy ;
if (TVirtualMC::GetMC()->TrackCharge()!=0) {
Float_t birkC1Mod = 0;
if (fBirkC0==1){
if (TMath::Abs(TVirtualMC::GetMC()->TrackCharge())>=2)
birkC1Mod=fBirkC1*7.2/12.6;
else
birkC1Mod=fBirkC1;
}
Float_t dedxcm=0.;
if (TVirtualMC::GetMC()->TrackStep()>0)
dedxcm=1000.*TVirtualMC::GetMC()->Edep()/TVirtualMC::GetMC()->TrackStep();
else
dedxcm=0;
lightYield=lightYield/(1.+birkC1Mod*dedxcm+fBirkC2*dedxcm*dedxcm);
}
xyzte[4] = lightYield * geom->GetSampling();
if (gDebug == 2)
printf("StepManager: id0 = %d, id1 = %d, absid = %d tower = %d layer = %d energy = %f\n", id[0], id[1], absid, tower, layer, xyzte[4]) ;
AddHit(fIshunt, fCurPrimary,tracknumber, fCurParent, ienergy, absid, xyzte, pmom);
}
}
}
void AliEMCALv1::RemapTrackHitIDs(Int_t *map) {
if (Hits()==0)
return;
TIter hitIter(Hits());
Int_t iHit=0;
while (AliEMCALHit *hit=dynamic_cast<AliEMCALHit*>(hitIter()) ) {
if (map[hit->GetIparent()]==-99)
cout << "Remapping, found -99 for parent id " << hit->GetIparent() << ", " << map[hit->GetIparent()] << ", iHit " << iHit << endl;
hit->SetIparent(map[hit->GetIparent()]);
if (map[hit->GetPrimary()]==-99)
cout << "Remapping, found -99 for primary id " << hit->GetPrimary() << ", " << map[hit->GetPrimary()] << ", iHit " << iHit << endl;
hit->SetPrimary(map[hit->GetPrimary()]);
iHit++;
}
}
void AliEMCALv1::FinishPrimary() {
fCurPrimary=-1;
fCurParent=-1;
fCurTrack=-1;
}