#include <TPDGCode.h>
#include <TParticle.h>
#include <TParticlePDG.h>
+#include <TDatabasePDG.h>
#include <TTree.h>
#include <TDirectory.h>
//_______________________________________________________________________
AliStack::AliStack():
- fParticles(0),
- fParticleMap(0),
+ fParticles("TParticle", 1000),
+ fParticleMap(),
fParticleFileMap(0),
fParticleBuffer(0),
fCurrentTrack(0),
//_______________________________________________________________________
AliStack::AliStack(Int_t size, const char* /*evfoldname*/):
- fParticles(new TClonesArray("TParticle",1000)),
- fParticleMap(new TObjArray(size)),
+ fParticles("TParticle",1000),
+ fParticleMap(size),
fParticleFileMap(0),
fParticleBuffer(0),
fCurrentTrack(0),
//_______________________________________________________________________
AliStack::AliStack(const AliStack& st):
TVirtualMCStack(st),
- fParticles(new TClonesArray("TParticle",1000)),
- fParticleMap(new TObjArray(*st.Particles())),
+ fParticles("TParticle",1000),
+ fParticleMap(*(st.Particles())),
fParticleFileMap(st.fParticleFileMap),
fParticleBuffer(0),
fCurrentTrack(0),
// Destructor
//
- if (fParticles) {
- fParticles->Delete();
- delete fParticles;
- }
- delete fParticleMap;
+ fParticles.Clear();
}
//
//
//_____________________________________________________________________________
-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)
{
//
const Int_t kLastDaughter=-1;
- TClonesArray &particles = *fParticles;
TParticle* particle
- = new(particles[fLoadPoint++])
+ = 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!!
+ 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;
}
// and renaming the particle id's in all the hits
//
- TObjArray &particles = *fParticleMap;
int nkeep = fHgwmk + 1, parent, i;
TParticle *part, *father;
- fTrackLabelMap.Set(particles.GetLast()+1);
+ fTrackLabelMap.Set(fParticleMap.GetLast()+1);
// Save in Header total number of tracks before compression
// If no tracks generated return now
if(i<=fHgwmk) fTrackLabelMap[i]=i ;
else {
fTrackLabelMap[i] = -99;
- if((part=dynamic_cast<TParticle*>(particles.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*>(particles.At(fHgwmk+1));
- particles.At(part->GetFirstMother())->ResetBit(kDaughtersBit);
+ 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++) {
- if(particles.At(i)->TestBit(kKeepBit)) {
+ if(fParticleMap.At(i)->TestBit(kKeepBit)) {
// This particle has to be kept
fTrackLabelMap[i]=nkeep;
// If old and new are different, have to move the pointer
- if(i!=nkeep) particles[nkeep]=particles.At(i);
- part = dynamic_cast<TParticle*>(particles.At(nkeep));
+ if(i!=nkeep) fParticleMap[nkeep]=fParticleMap.At(i);
+ 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*>(particles.At(i));
+ part = GetParticleMapEntry(i);
parent = part->GetFirstMother();
if(parent>=0) {
- father = dynamic_cast<TParticle*>(particles.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*>(particles.At(i));
+ fParticleBuffer = GetParticleMapEntry(i);
fParticleFileMap[i]=static_cast<Int_t>(TreeK()->GetEntries());
TreeK()->Fill();
- particles[i]=fParticleBuffer=0;
+ fParticleMap[i]=fParticleBuffer=0;
}
- for (i = nkeep; i < fNtrack; ++i) particles[i]=0;
+ for (i = nkeep; i < fNtrack; ++i) fParticleMap[i]=0;
Int_t toshrink = fNtrack-fHgwmk-1;
fLoadPoint-=toshrink;
- for(i=fLoadPoint; i<fLoadPoint+toshrink; ++i) fParticles->RemoveAt(i);
+ for(i=fLoadPoint; i<fLoadPoint+toshrink; ++i) fParticles.RemoveAt(i);
fNtrack=nkeep;
fHgwmk=nkeep-1;
return kTRUE;
if (nNew > 0) {
Int_t i, j;
- TObjArray &particles = *fParticleMap;
TArrayI map1(nNew);
//
// Copy pointers to temporary array
TParticle** tmp = new TParticle*[nNew];
for (i = 0; i < nNew; i++) {
- if (particles.At(fHgwmk + 1 + i)) {
- tmp[i] = (TParticle*) (particles.At(fHgwmk + 1 + i));
+ if (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*>(particles.At(ipa));
+ parP = GetParticleMapEntry(ipa);
} else {
ipa = (fHgwmk + 1 + i);
// Skip deleted particles
Int_t jpa = tmp[j]->GetFirstMother();
// Check if daughter of current parent
if (jpa == ipa) {
- particles[loadPoint] = tmp[j];
+ fParticleMap[loadPoint] = tmp[j];
// Re-establish daughter information
parP->SetLastDaughter(loadPoint);
if (parP->GetFirstDaughter() == -1) parP->SetFirstDaughter(loadPoint);
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
//
Int_t parent = part->GetFirstMother();
if (parent >= 0 && parent <= fHgwmk) {
- TParticle* father = dynamic_cast<TParticle*>(Particles()->At(parent));
+ TParticle* father = GetParticleMapEntry(parent);
Int_t kf = father->GetPdgCode();
kf = TMath::Abs(kf);
Int_t kfl = kf;
}
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;
- fParticleMap->AddAt(0,i);
+ fParticleMap.AddAt(0,i);
// When all primaries were filled no particle!=0
// should be left => to be removed later.
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;
// Flags a track to be kept
//
- fParticleMap->At(track)->SetBit(kKeepBit);
+ fParticleMap.At(track)->SetBit(kKeepBit);
}
//_____________________________________________________________________________
//
// Resets stack arrays
//
-
- if (fParticles)
- fParticles->Clear();
- else
- fParticles = new TClonesArray("TParticle",1000);
- if (fParticleMap) {
- fParticleMap->Clear();
- if (size>0) fParticleMap->Expand(size);}
- else
- fParticleMap = new TObjArray(size);
+ fParticles.Clear();
+ fParticleMap.Clear();
+ if (size>0) fParticleMap.Expand(size);
}
//_____________________________________________________________________________
//
// Return particle with specified ID
- if(!fParticleMap->At(i)) {
- Int_t nentries = fParticles->GetEntriesFast();
+ if(!fParticleMap.At(i)) {
+ Int_t nentries = fParticles.GetEntriesFast();
// algorithmic way of getting entry index
// (primary particles are filled after secondaries)
Int_t entry = TreeKEntry(i);
// Load particle at entry into fParticleBuffer
TreeK()->GetEntry(entry);
// Add to the TClonesarray
- new ((*fParticles)[nentries]) TParticle(*fParticleBuffer);
+ new (fParticles[nentries]) TParticle(*fParticleBuffer);
// Store a pointer in the TObjArray
- fParticleMap->AddAt((*fParticles)[nentries],i);
+ 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]);
}
// that are loaded in memory.
//
- TObjArray &particles = *fParticleMap;
printf(
"\n\n=======================================================================\n");
for (Int_t i=0;i<fNtrack;i++)
{
- TParticle* particle = dynamic_cast<TParticle*>(particles[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
//
// Set parent/daughter relations
//
- TObjArray &particles = *fParticleMap;
TParticle *part;
int i;
for(i=0; i<fHgwmk+1; i++) {
- part = dynamic_cast<TParticle*>(particles.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 = 15;
+ 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
+ };
- 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