#include <TClonesArray.h>
#include <TObjArray.h>
#include <TPDGCode.h>
+#include <TMCProcess.h>
#include <TParticle.h>
#include <TParticlePDG.h>
+#include <TDatabasePDG.h>
#include <TTree.h>
#include <TDirectory.h>
//_______________________________________________________________________
AliStack::AliStack():
- fParticles(),
+ fParticles("TParticle", 1000),
fParticleMap(),
fParticleFileMap(0),
fParticleBuffer(0),
//
//_____________________________________________________________________________
-void AliStack::PushTrack(Int_t done, Int_t parent, Int_t pdg, Float_t *pmom,
- Float_t *vpos, Float_t *polar, Float_t tof,
+void AliStack::PushTrack(Int_t done, Int_t parent, Int_t pdg, const Float_t *pmom,
+ const Float_t *vpos, const Float_t *polar, Float_t tof,
TMCProcess mech, Int_t &ntr, Float_t weight, Int_t is)
{
//
= new(fParticles[fLoadPoint++])
TParticle(pdg, is, parent, -1, kFirstDaughter, kLastDaughter,
px, py, pz, e, vx, vy, vz, tof);
-
+
particle->SetPolarisation(polx, poly, polz);
particle->SetWeight(weight);
particle->SetUniqueID(mech);
particle->SetBit(kDaughtersBit);
// Add the particle to the stack
-
fParticleMap.AddAtAndExpand(particle, fNtrack);//CHECK!!
if(parent>=0) {
- particle = dynamic_cast<TParticle*>(fParticleMap.At(parent));
+ particle = GetParticleMapEntry(parent);
if (particle) {
particle->SetLastDaughter(fNtrack);
if(particle->GetFirstDaughter()<0) particle->SetFirstDaughter(fNtrack);
TParticle* particle = Particle(i);
- if (!particle->TestBit(kDoneBit))
+ if (!particle->TestBit(kDoneBit)) {
+ fCurrentTrack = particle;
return particle;
+ }
else
return 0;
}
if(i<=fHgwmk) fTrackLabelMap[i]=i ;
else {
fTrackLabelMap[i] = -99;
- if((part=dynamic_cast<TParticle*>(fParticleMap.At(i)))) {
+ if((part=GetParticleMapEntry(i))) {
//
// Check of this track should be kept for physics reasons
if (KeepPhysics(part)) KeepTrack(i);
// Invalid daughter information for the parent of the first particle
// generated. This may or may not be the current primary according to
// whether decays have been recorded among the primaries
- part = dynamic_cast<TParticle*>(fParticleMap.At(fHgwmk+1));
+ part = GetParticleMapEntry(fHgwmk+1);
fParticleMap.At(part->GetFirstMother())->ResetBit(kDaughtersBit);
// Second pass, build map between old and new numbering
for(i=fHgwmk+1; i<fNtrack; i++) {
fTrackLabelMap[i]=nkeep;
// If old and new are different, have to move the pointer
if(i!=nkeep) fParticleMap[nkeep]=fParticleMap.At(i);
- part = dynamic_cast<TParticle*>(fParticleMap.At(nkeep));
+ part = GetParticleMapEntry(nkeep);
// as the parent is always *before*, it must be already
// in place. This is what we are checking anyway!
- if((parent=part->GetFirstMother())>fHgwmk)
+ if((parent=part->GetFirstMother())>fHgwmk) {
if(fTrackLabelMap[parent]==-99) Fatal("PurifyKine","fTrackLabelMap[%d] = -99!\n",parent);
- else part->SetFirstMother(fTrackLabelMap[parent]);
+ else part->SetFirstMother(fTrackLabelMap[parent]);}
nkeep++;
}
}
// Fix daughters information
for (i=fHgwmk+1; i<nkeep; i++) {
- part = dynamic_cast<TParticle*>(fParticleMap.At(i));
+ part = GetParticleMapEntry(i);
parent = part->GetFirstMother();
if(parent>=0) {
- father = dynamic_cast<TParticle*>(fParticleMap.At(parent));
+ father = GetParticleMapEntry(parent);
if(father->TestBit(kDaughtersBit)) {
if(i<father->GetFirstDaughter()) father->SetFirstDaughter(i);
// Now the output bit, from fHgwmk to nkeep we write everything and we erase
if(nkeep > fParticleFileMap.GetSize()) fParticleFileMap.Set(Int_t (nkeep*1.5));
for (i=fHgwmk+1; i<nkeep; ++i) {
- fParticleBuffer = dynamic_cast<TParticle*>(fParticleMap.At(i));
+ fParticleBuffer = GetParticleMapEntry(i);
fParticleFileMap[i]=static_cast<Int_t>(TreeK()->GetEntries());
TreeK()->Fill();
fParticleMap[i]=fParticleBuffer=0;
for (i = 0; i < nNew; i++) {
if (fParticleMap.At(fHgwmk + 1 + i)) {
- tmp[i] = (TParticle*) (fParticleMap.At(fHgwmk + 1 + i));
+ tmp[i] = GetParticleMapEntry(fHgwmk + 1 + i);
} else {
tmp[i] = 0x0;
}
TParticle* parP;
if (i == -1) {
ipa = tmp[0]->GetFirstMother();
- parP =dynamic_cast<TParticle*>(fParticleMap.At(ipa));
+ parP = GetParticleMapEntry(ipa);
} else {
ipa = (fHgwmk + 1 + i);
// Skip deleted particles
return kTRUE;
}
-Bool_t AliStack::KeepPhysics(TParticle* part)
+Bool_t AliStack::KeepPhysics(const TParticle* part)
{
//
// Some particles have to kept on the stack for reasons motivated
// by physics analysis. Decision is put here.
//
Bool_t keep = kFALSE;
+
+ Int_t parent = part->GetFirstMother();
+ if (parent >= 0 && parent <= fHgwmk) {
+ TParticle* father = GetParticleMapEntry(parent);
//
// Keep first-generation daughter from primaries with heavy flavor
//
- Int_t parent = part->GetFirstMother();
- if (parent >= 0 && parent <= fHgwmk) {
- TParticle* father = dynamic_cast<TParticle*>(Particles()->At(parent));
Int_t kf = father->GetPdgCode();
kf = TMath::Abs(kf);
Int_t kfl = kf;
// meson ?
if (kfl > 10) kfl/=100;
// baryon
- if (kfl > 10) kfl/=10;
- if (kfl > 10) kfl/=10;
+ if (kfl > 10) kfl/=10;
+ if (kfl > 10) kfl/=10;
if (kfl >= 4) {
keep = kTRUE;
}
+ //
+ // e+e- from pair production of primary gammas
+ //
+ if ((part->GetUniqueID()) == kPPair) keep = kTRUE;
}
return keep;
}
}
Bool_t allFilled = kFALSE;
- TObject *part;
- for(Int_t i=0; i<fHgwmk+1; ++i)
- if((part=fParticleMap.At(i))) {
- fParticleBuffer = dynamic_cast<TParticle*>(part);
+ TParticle *part;
+ for(Int_t i=0; i<fHgwmk+1; ++i) {
+ if((part=GetParticleMapEntry(i))) {
+ fParticleBuffer = part;
fParticleFileMap[i]= static_cast<Int_t>(TreeK()->GetEntries());
TreeK()->Fill();
fParticleBuffer=0;
if (allFilled) AliWarning(Form("Why != 0 part # %d?\n",i));
}
else
- {
+ {
// // printf("Why = 0 part # %d?\n",i); => We know.
// break;
// we don't break now in order to be sure there is no
// particle !=0 left.
// To be removed later and replaced with break.
if(!allFilled) allFilled = kTRUE;
- }
+ }
+ }
}
//_____________________________________________________________________________
Int_t curr=track;
while(1) {
- particle=dynamic_cast<TParticle*>(fParticleMap.At(curr));
+ particle = GetParticleMapEntry(curr);
// If the particle is flagged the three from here upward is saved already
if(particle->TestBit(kKeepBit)) return;
// Store a pointer in the TObjArray
fParticleMap.AddAt(fParticles[nentries],i);
}
- return dynamic_cast<TParticle*>(fParticleMap.At(i));
+ return GetParticleMapEntry(i);
}
//_____________________________________________________________________________
// Return number of the parent of the current track
//
- TParticle* current = (TParticle*)fParticleMap.At(fCurrent);
+ TParticle* current = GetParticleMapEntry(fCurrent);
if (current)
return current->GetFirstMother();
//
// Dumps particle i in the stack
//
- dynamic_cast<TParticle*>(fParticleMap.At(i))->Print();
+ GetParticleMapEntry(i)->Print();
}
//_____________________________________________________________________________
printf("\n=======================================================================\n\n");
// print particle file map
- printf("\nParticle file map: \n");
- for (i=0; i<fNtrack; i++)
- printf(" %d th entry: %d \n",i,fParticleFileMap[i]);
+ // printf("\nParticle file map: \n");
+ // for (i=0; i<fNtrack; i++)
+ // printf(" %d th entry: %d \n",i,fParticleFileMap[i]);
}
"\n\n=======================================================================\n");
for (Int_t i=0;i<fNtrack;i++)
{
- TParticle* particle = dynamic_cast<TParticle*>(fParticleMap[i]);
+ TParticle* particle = GetParticleMapEntry(i);
if (particle) {
printf("-> %d ",i); particle->Print();
printf("--------------------------------------------------------------\n");
"\n=======================================================================\n\n");
}
+//_____________________________________________________________________________
+void AliStack::SetCurrentTrack(Int_t track)
+{
+ fCurrent = track;
+ if (fCurrent < fNprimary) fCurrentTrack = Particle(track);
+}
+
+
+//_____________________________________________________________________________
//
// protected methods
//
TParticle *part;
int i;
for(i=0; i<fHgwmk+1; i++) {
- part = dynamic_cast<TParticle*>(fParticleMap.At(i));
+ part = GetParticleMapEntry(i);
if(part) if(!part->TestBit(kDaughtersBit)) {
part->SetFirstDaughter(-1);
part->SetLastDaughter(-1);
// search secondaries
//for(Int_t i=fNtrack-1; i>=0; i--) {
for(Int_t i=fNtrack-1; i>fHgwmk; i--) {
- particle = dynamic_cast<TParticle*>(fParticleMap.At(i));
+ particle = GetParticleMapEntry(i);
if ((particle) && (!particle->TestBit(kDoneBit))) {
fCurrent=i;
return particle;
// take next primary if all secondaries were done
while (fCurrentPrimary>=0) {
fCurrent = fCurrentPrimary;
- particle = dynamic_cast<TParticle*>(fParticleMap.At(fCurrentPrimary--));
+ particle = GetParticleMapEntry(fCurrentPrimary--);
if ((particle) && (!particle->TestBit(kDoneBit))) {
return particle;
}
}
//__________________________________________________________________________________________
-TTree* AliStack::TreeK()
-{
-//returns TreeK
- return fTreeK;
-}
-//__________________________________________________________________________________________
-
void AliStack::ConnectTree(TTree* tree)
{
//
Bool_t AliStack::IsStable(Int_t pdg) const
{
-//
-// Decide whether particle (pdg) is stable
-//
+ //
+ // Decide whether particle (pdg) is stable
+ //
+
+
+ // All ions/nucleons are considered as stable
+ // Nuclear code is 10LZZZAAAI
+ if(pdg>1000000000)return kTRUE;
- const Int_t kNstable = 14;
- Int_t i;
-
- Int_t pdgStable[kNstable] = {
- kGamma, // Photon
- kElectron, // Electron
- kMuonPlus, // Muon
- kPiPlus, // Pion
- kKPlus, // Kaon
- kProton, // Proton
- kNeutron, // Neutron
- kLambda0, // Lambda_0
- kSigmaMinus, // Sigma Minus
- kSigma0, // Sigma_0
- kSigmaPlus, // Sigma Plus
- 3312, // Xsi Minus
- 3322, // Xsi
- 3334, // Omega
- };
+ const Int_t kNstable = 18;
+ Int_t i;
+
+ Int_t pdgStable[kNstable] = {
+ kGamma, // Photon
+ kElectron, // Electron
+ kMuonPlus, // Muon
+ kPiPlus, // Pion
+ kKPlus, // Kaon
+ kK0Short, // K0s
+ kK0Long, // K0l
+ kProton, // Proton
+ kNeutron, // Neutron
+ kLambda0, // Lambda_0
+ kSigmaMinus, // Sigma Minus
+ kSigmaPlus, // Sigma Plus
+ 3312, // Xsi Minus
+ 3322, // Xsi
+ 3334, // Omega
+ kNuE, // Electron Neutrino
+ kNuMu, // Muon Neutrino
+ kNuTau // Tau Neutrino
+ };
- Bool_t isStable = kFALSE;
- for (i = 0; i < kNstable; i++) {
- if (pdg == TMath::Abs(pdgStable[i])) {
- isStable = kTRUE;
- break;
- }
+ Bool_t isStable = kFALSE;
+ for (i = 0; i < kNstable; i++) {
+ if (pdg == TMath::Abs(pdgStable[i])) {
+ isStable = kTRUE;
+ break;
}
-
- return isStable;
+ }
+
+ return isStable;
}
+//_____________________________________________________________________________
Bool_t AliStack::IsPhysicalPrimary(Int_t index)
{
//
//
// Initial state particle
- if (ist > 20) return kFALSE;
+ if (ist > 1) return kFALSE;
Int_t pdg = TMath::Abs(p->GetPdgCode());
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff