+/**************************************************************************
+ * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
+ * *
+ * Author: The ALICE Off-line Project. *
+ * Contributors are mentioned in the code where appropriate. *
+ * *
+ * Permission to use, copy, modify and distribute this software and its *
+ * documentation strictly for non-commercial purposes is hereby granted *
+ * without fee, provided that the above copyright notice appears in all *
+ * copies and that both the copyright notice and this permission notice *
+ * appear in the supporting documentation. The authors make no claims *
+ * about the suitability of this software for any purpose. It is *
+ * provided "as is" without express or implied warranty. *
+ **************************************************************************/
+
+/* $Id$ */
+
+// Read background particles from a FLUKA boundary source file
+// This is a very special generator that works for background studies for the muon-spectrometer.
+// The input files come from FLUKA simulations.
+// Files can be chained.
+// Author: andreas.morsch@cern.ch
+
+#include <RVersion.h>
+#include "TPDGCode.h"
+#include "TDatabasePDG.h"
+#include <TVirtualMC.h>
+
#include "AliGenFLUKAsource.h"
-#include "AliGenMUONlib.h"
-#include "AliMC.h"
#include "AliRun.h"
-#include "AliPDG.h"
-#include <TDirectory.h>
+
#include <TFile.h>
#include <TTree.h>
+#include <TChain.h>
#include <stdlib.h>
- ClassImp(AliGenFLUKAsource)
- AliGenFLUKAsource::AliGenFLUKAsource()
+ClassImp(AliGenFLUKAsource)
+
+AliGenFLUKAsource::AliGenFLUKAsource()
:AliGenerator(-1)
{
- //
+ // Constructor
fName="FLUKA";
fTitle="FLUKA Boundary Source";
// Read in all particle types by default
// whole volume of the MUON Arm
fZshift=0;
// Set the default file
- fFileName="flukasource.root";
+ fFileName="";
fTreeFluka=0;
+ fTreeChain = new TChain("h1");
//
// Read all particles
fNpart=-1;
-
}
AliGenFLUKAsource::AliGenFLUKAsource(Int_t npart)
:AliGenerator(npart)
{
- //
- fName="FLUKA";
- fTitle="FLUKA Boundary Source";
+ // Constructor
+ fName = "FLUKA";
+ fTitle = "FLUKA Boundary Source";
// Read in all particle types by default
fIkine=6;
// Set maximum admitted age of particles to 1.e-05 by default
// whole volume of the MUON Arm
fZshift=0;
// Set the default file
- fFileName="flukasource.root";
+ fFileName="";
fTreeFluka=0;
+ fTreeChain = new TChain("h1");
+ fSourceId=-1;
+}
+AliGenFLUKAsource::AliGenFLUKAsource(const AliGenFLUKAsource & FLUKAsource):
+ AliGenerator(FLUKAsource)
+{
+// Copy constructor
+ FLUKAsource.Copy(*this);
}
+
//____________________________________________________________
AliGenFLUKAsource::~AliGenFLUKAsource()
{
- delete fTreeFluka;
+// Destructor
+ if (fTreeFluka) delete fTreeFluka;
+ if (fTreeChain) delete fTreeChain;
}
-//____________________________________________________________
-void AliGenFLUKAsource::FlukaInit()
+void AliGenFLUKAsource::AddFile(const Text_t *filename)
{
-//
-// reset the existing file environment and open a new root file if
-// the pointer to the Fluka tree is null
+// Add a file to the chain
+ fTreeChain->Add(filename);
- TFile *File=0;
- if (fTreeFluka==0) {
- if (!File) {
- File = new TFile(fFileName);
- File->cd();
- cout<<"I have opened "<<fFileName<<" file "<<endl;
- }
-// get the tree address in the Fluka boundary source file
- fTreeFluka = (TTree*)gDirectory->Get("h1");
- } else {
- File = fTreeFluka->GetCurrentFile();
- File->cd();
- }
+}
- TTree *h2=fTreeFluka;
+
+//____________________________________________________________
+void AliGenFLUKAsource::FlukaInit()
+{
+// Set branch addresses of data entries
+ TChain *h2=fTreeChain;
//Set branch addresses
- h2->SetBranchAddress("Ip",&Ip);
- h2->SetBranchAddress("Ipp",&Ipp);
- h2->SetBranchAddress("Xi",&Xi);
- h2->SetBranchAddress("Yi",&Yi);
- h2->SetBranchAddress("Zi",&Zi);
- h2->SetBranchAddress("Px",&Px);
- h2->SetBranchAddress("Py",&Py);
- h2->SetBranchAddress("Pz",&Pz);
- h2->SetBranchAddress("Ekin",&Ekin);
- h2->SetBranchAddress("Zv",&Zv);
- h2->SetBranchAddress("Rv",&Rv);
- h2->SetBranchAddress("Itra",&Itra);
- h2->SetBranchAddress("Igas",&Igas);
- h2->SetBranchAddress("Wgt",&Wgt);
- h2->SetBranchAddress("Etag",&Etag);
- h2->SetBranchAddress("Ptg",&Ptg);
- h2->SetBranchAddress("Age",&Age);
+ h2->SetBranchAddress("Ip",&fIp);
+ h2->SetBranchAddress("Ipp",&fIpp);
+ h2->SetBranchAddress("Xi",&fXi);
+ h2->SetBranchAddress("Yi",&fYi);
+ h2->SetBranchAddress("Zi",&fZi);
+ h2->SetBranchAddress("Px",&fPx);
+ h2->SetBranchAddress("Py",&fPy);
+ h2->SetBranchAddress("Pz",&fPz);
+ h2->SetBranchAddress("Ekin",&fEkin);
+ h2->SetBranchAddress("Zv",&fZv);
+ h2->SetBranchAddress("Rv",&fRv);
+ h2->SetBranchAddress("Itra",&fItra);
+ h2->SetBranchAddress("Igas",&fIgas);
+ h2->SetBranchAddress("Wgt",&fWgt);
+ h2->SetBranchAddress("Etag",&fEtag);
+ h2->SetBranchAddress("Ptg",&fPtg);
+ h2->SetBranchAddress("Age",&fAge);
}
//____________________________________________________________
void AliGenFLUKAsource::Generate()
{
+// Generate one event
- const Int_t ifluge[28]={kProton, kProtonBar, kElectron, kPositron,
+ const Int_t kIfluge[28]={kProton, kProtonBar, kElectron, kPositron,
kNuE, kNuEBar, kGamma, kNeutron, kNeutronBar,
kMuonPlus, kMuonMinus, kK0Long , kPiPlus, kPiMinus,
kKPlus, kKMinus, kLambda0, kLambda0Bar, kK0Short,
kSigmaMinus, kSigmaPlus, kSigma0, kPi0, kK0, kK0Bar,
0,kNuMu,kNuMuBar};
- Float_t polar[3]= {0,0,0};
- //
- Float_t origin[3];
- Float_t p[3];
- Float_t prwn;
- Float_t wgt, fwgt;
- Float_t phi;
- char name[100];
- Float_t amass, charge, tlife;
- Int_t itrtyp;
- Int_t iwgt;
- Int_t i, j, part, nt;
- static Int_t irwn;
+ Float_t polar[3]= {0,0,0};
//
- Float_t random[2];
+ Float_t origin[3];
+ Float_t p[3];
+ Float_t prwn;
+ Float_t wgt, fwgt;
+ Float_t phi;
+ Float_t amass;
+ Int_t iwgt;
+ Int_t i, j, part, nt;
+ static Int_t irwn=0;
+ //
+ Float_t random[2];
//
- FlukaInit();
- TTree *h2=fTreeFluka;
- Int_t nentries = (Int_t) h2->GetEntries();
- if (fNpart == -1) fNpart=Int_t(nentries*fFrac);
-
+ FlukaInit();
+ TChain *h2=fTreeChain;
+ Int_t nentries = (Int_t) h2->GetEntries();
+ if (fNpart == -1) fNpart=Int_t(nentries*fFrac);
+
+
// loop over number of particles
- Int_t nb=0;
- for (i=0; i<fNpart;i++) {
+ Int_t nb=0;
Int_t ev=gMC->CurrentEvent();
- Int_t entry=fNpart*(ev-1)+i;
- nb = (Int_t)h2->GetEvent(entry);
- if (irwn > nentries) {
- printf("No more entries in the FLUKA boundary source file\n");
- TFile *File=0;
- // Get AliRun object or create it
- if (!gAlice) {
- gAlice = (AliRun*)File->Get("gAlice");
- if (gAlice) printf("AliRun object found on file\n");
- if (!gAlice) gAlice = new AliRun("gAlice","Alice test program");
- }
- TTree *fAli=gAlice->TreeK();
- if (fAli) File =fAli->GetCurrentFile();
- File->cd();
- printf("Generate - I'm out \n");
- return;
- }
- if (Ip > 28 || Ip < 0) {
- irwn++;
- continue;
- }
-
- if ((Ip != fIkine && fIkine != 6 && fIkine != 9) || Age > fAgeMax){
- irwn++;
- continue;
- } else if (fIkine == 9) {
- if (Ip == 7 || Ip == 8 || Age > fAgeMax) {
- irwn++;
- continue;
- }
- }
-
+ for (i=0; i<fNpart;i++) {
+ Int_t entry=fNpart*(ev-1)+i;
+ nb = (Int_t)h2->GetEvent(entry);
+ if (irwn > nentries) {
+ printf("No more entries in the FLUKA boundary source file\n");
+ TFile *pFile=0;
+ // Get AliRun object or create it
+ if (!gAlice) {
+ gAlice = (AliRun*)pFile->Get("gAlice");
+ if (gAlice) printf("AliRun object found on file\n");
+ if (!gAlice) gAlice = new AliRun("gAlice","Alice test program");
+ }
+ TTree *fAli=gAlice->TreeK();
+ if (fAli) pFile =fAli->GetCurrentFile();
+ pFile->cd();
+ printf("Generate - I'm out \n");
+ return;
+ }
+
+ Int_t ifip = Int_t(fIp);
+
- irwn++;
- printf("\n Particle type: %f \n \n ", Ip);
+ if (fSourceId != -1 && fIgas !=fSourceId) {
+ irwn++;
+ continue;
+ }
+
+ if (ifip > 28 || ifip < 0) {
+ irwn++;
+ continue;
+ }
+
+ if ((ifip != fIkine && fIkine != kAll && fIkine != kCharged
+ && fIkine != 10) || fAge > fAgeMax){
+ irwn++;
+ continue;
+ } else if (fIkine == kCharged) {
+ if (ifip == 7 || ifip == 8 || fAge > fAgeMax) {
+ irwn++;
+ continue;
+ }
+ } else if (fIkine == kNoNeutron) {
+ if (ifip == 8 || fAge > fAgeMax) {
+ irwn++;
+ continue;
+ }
+ }
- if (Ip ==7){
- prwn=Ekin;
- part=1;
- } else if (Ip == 8) {
- prwn=sqrt(Ekin*Ekin + 2.*0.93956563);
- part=13;
- } else {
- part=ifluge[int(Ip)-1];
- gMC->Gfpart(part, name, itrtyp,
- amass, charge, tlife);
- prwn=sqrt(Ekin*Ekin + 2.*amass);
- }
- origin[0]=Xi;
- origin[1]=Yi;
- origin[2]=Zi;
-
- p[0]=Px*prwn;
- p[1]=Py*prwn;
- p[2]=Pz*prwn;
-
- //handle particle weight correctly
- wgt = (part == 13) ? Wgt*fAddWeight : Wgt;
- iwgt=Int_t(wgt);
- fwgt=wgt-Float_t(iwgt);
- gMC->Rndm(random,2);
- if (random[0] < fwgt) iwgt++;
- if (part==1 && iwgt>100) iwgt=100;
- Int_t nstack=0;
- for (j=0; j<iwgt; j++) {
- gAlice->SetTrack(1,-1,part,p,origin,polar,0,"Primary",nt);
- gMC->Rndm(random,2);
- phi=2*random[1]*TMath::Pi();
- Float_t pn1=p[0]*TMath::Sin(phi) - p[1]*TMath::Cos(phi);
- Float_t pn2=p[0]*TMath::Cos(phi) + p[1]*TMath::Sin(phi);
- p[0]=pn1;
- p[1]=pn2;
- Float_t on1=origin[0]*TMath::Sin(phi)-origin[1]*TMath::Cos(phi);
- Float_t on2=origin[0]*TMath::Cos(phi)+origin[1]*TMath::Sin(phi);
- origin[0]=on1;
- origin[1]=on2;
- nstack++;
+
+ irwn++;
+//
+// PDG code from FLUKA particle type (ifip)
+ part=kIfluge[int(ifip)-1];
+//
+// Calculate momentum from kinetic energy and mass of the particle
+#if ROOT_VERSION_CODE > 197895
+ amass = gMC->ParticleMass(part);
+#else
+ amass = (TDatabasePDG::Instance())->GetParticle(part)->Mass();
+#endif
+ prwn=fEkin*sqrt(1. + 2.*amass/fEkin);
+
+
+ origin[0]=fYi;
+ origin[1]=fXi;
+ origin[2]=fZi;
+
+ p[0]=fPy*prwn;
+ p[1]=fPx*prwn;
+ p[2]=fPz*prwn;
+
+ //handle particle weight correctly
+ wgt = (part == 13) ? fWgt*fAddWeight : fWgt;
+ iwgt=Int_t(wgt);
+ fwgt=wgt-Float_t(iwgt);
+ Rndm(random,2);
+ if (random[0] < fwgt) iwgt++;
+ if (part==1 && iwgt>100) iwgt=100;
+ Int_t nstack=0;
+ for (j=0; j<iwgt; j++) {
+ PushTrack(fTrackIt,-1,part,p,origin,polar,fAge,kPPrimary,nt);
+ Rndm(random,2);
+ phi=2*random[1]*TMath::Pi();
+ Float_t pn1=p[0]*TMath::Sin(phi) - p[1]*TMath::Cos(phi);
+ Float_t pn2=p[0]*TMath::Cos(phi) + p[1]*TMath::Sin(phi);
+ p[0]=pn1;
+ p[1]=pn2;
+ Float_t on1=origin[0]*TMath::Sin(phi)-origin[1]*TMath::Cos(phi);
+ Float_t on2=origin[0]*TMath::Cos(phi)+origin[1]*TMath::Sin(phi);
+ origin[0]=on1;
+ origin[1]=on2;
+ nstack++;
+ }
+ if (nstack == 0) continue;
}
- if (nstack == 0) continue;
- }
-
- TFile *File=0;
+
+ TFile *pFile=0;
// Get AliRun object or create it
if (!gAlice) {
- gAlice = (AliRun*)File->Get("gAlice");
+ gAlice = (AliRun*)pFile->Get("gAlice");
if (gAlice) printf("AliRun object found on file\n");
if (!gAlice) gAlice = new AliRun("gAlice","Alice test program");
}
TTree *fAli=gAlice->TreeK();
- if (fAli) File =fAli->GetCurrentFile();
- File->cd();
+ if (fAli) pFile =fAli->GetCurrentFile();
+ pFile->cd();
}
+AliGenFLUKAsource& AliGenFLUKAsource::operator=(const AliGenFLUKAsource& rhs)
+{
+// Assignment operator
+ rhs.Copy(*this);
+ return (*this);
+}
+void AliGenFLUKAsource::Copy(AliGenFLUKAsource &) const
+{
+ Fatal("Copy","Not implemented!\n");
+}
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff