#include <TClonesArray.h>
#include <TObjArray.h>
#include <TPDGCode.h>
+#include <TMCProcess.h>
#include <TParticle.h>
#include <TParticlePDG.h>
#include <TDatabasePDG.h>
= 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) {
// 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
// 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();
}
//_____________________________________________________________________________
"\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;
}
// Nuclear code is 10LZZZAAAI
if(pdg>1000000000)return kTRUE;
- const Int_t kNstable = 15;
+ const Int_t kNstable = 18;
Int_t i;
Int_t pdgStable[kNstable] = {
3312, // Xsi Minus
3322, // Xsi
3334, // Omega
+ kNuE, // Electron Neutrino
+ kNuMu, // Muon Neutrino
+ kNuTau // Tau Neutrino
};
Bool_t isStable = kFALSE;
return isStable;
}
+//_____________________________________________________________________________
Bool_t AliStack::IsPhysicalPrimary(Int_t index)
{
//
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff