FairShip
Loading...
Searching...
No Matches
run_simScript Namespace Reference

Functions

None visualizeMagFields ()
 
None checkOverlapsWithGeant4 ()
 

Variables

bool DownScaleDiMuon = False
 
float theHNLMass = 1.0 * u.GeV
 
theDecayCouplings theProductionCouplings = None
 
float theDPmass = 0.2 * u.GeV
 
str mcEngine = "TGeant4"
 
str inclusive = "c"
 
bool MCTracksWithHitsOnly = False
 
bool MCTracksWithEnergyCutOnly = True
 
bool MCTracksWithHitsOrEnergyCut = False
 
bool charmonly = False
 
bool HNL = True
 
str inputFile = "$EOSSHIP/eos/experiment/ship/data/Charm/Cascade-parp16-MSTP82-1-MSEL4-978Bpot.root"
 
bool defaultInputFile = True
 
ArgumentParser parser = ArgumentParser()
 
 help
 
 action
 
 dest
 
ArgumentParser subparsers = parser.add_subparsers(dest="command", help="Which mode to run")
 
ArgumentParser pg_parser = subparsers.add_parser("PG", help="Use Particle Gun")
 
 default
 
 type
 
 int
 
 float
 
ArgumentParser genie_parser = subparsers.add_parser("Genie", help="Genie for reading and processing neutrino interactions")
 
 choices
 
 None
 
 const
 
 nargs
 
ArgumentParser options = parser.parse_args()
 
list available_snd_designs = [1, 2]
 
 SND_design
 
 else :
 
 try :
 
ArgumentParser motherMode = options.MM
 
int Opt_high = int(options.cosmics)
 
 theMass
 
list theCouplings = [float(c) for c in options.thecouplings.split(",")]
 
list theDecayCouplings = [float(c) for c in options.thedeccouplings.split(",")]
 
ArgumentParser seed = options.theSeed
 
geometry_config ship_geo
 
 pythia8
 
ArgumentParser run_identifier = options.output_tag if options.output_tag else str(uuid.uuid4())
 
outFile = f"{options.outputDir}/sim_{run_identifier}.root"
 
parFile = f"{options.outputDir}/params_{run_identifier}.root"
 
ROOT timer = ROOT.TStopwatch()
 
ROOT run = ROOT.FairRunSim()
 
ROOT rtdb = run.GetRuntimeDb()
 
shipDet_conf modules = shipDet_conf.configure(run, ship_geo)
 
ROOT primGen = ROOT.FairPrimaryGenerator()
 
ROOT P8gen = ROOT.HNLPythia8Generator()
 
pythia8darkphoton_conf passDPconf
 
ROOT test = ROOT.TPythia6()
 
ROOT P6gen = ROOT.tPythia6Generator()
 
ROOT EvtCalcGen = ROOT.EvtCalcGenerator()
 
 zTa
 
 z
 
 zDV
 
 nEvents = 0
 
ROOT myPgun = ROOT.FairBoxGenerator(options.pID, 1)
 
ROOT DISgen = ROOT.MuDISGenerator()
 
 mu_start
 
 mu_end
 
ROOT Geniegen = ROOT.GenieGenerator()
 
ROOT pdg = ROOT.TDatabasePDG.Instance()
 
ROOT Ntuplegen = ROOT.NtupleGenerator()
 
ut fileType = ut.checkFileExists(inputFile)
 
ROOT MuonBackgen = ROOT.MuonBackGenerator()
 
ROOT testf = ROOT.TFile.Open(inputFile[0])
 
 fastMuon
 
ROOT Cosmicsgen = ROOT.CosmicsGenerator()
 
 n_EVENTS
 
ROOT gMC = ROOT.TVirtualMC.GetMC()
 
ROOT fStack = gMC.GetStack()
 
float EnergyCut = 10.0 * u.MeV if options.mudis else 100.0 * u.MeV
 
ROOT trajFilter = ROOT.FairTrajFilter.Instance()
 
 fieldMap
 
geomGeant4 fieldMaker = geomGeant4.addVMCFields(ship_geo, verbose=True)
 
 onlyWithField
 
 True
 
 exclude
 
ROOT kParameterMerged = ROOT.kTRUE
 
ROOT parOut = ROOT.FairParRootFileIo(kParameterMerged)
 
geofile_name = f"{options.outputDir}/geo_{run_identifier}.root"
 
ROOT fGeo = ROOT.gGeoManager
 
ROOT rtime = timer.RealTime()
 
ROOT ctime = timer.CpuTime()
 
tmpFile = outFile + "tmp"
 
xxx = outFile.split("/")
 
check = xxx[len(xxx) - 1]
 
bool fin = False
 
ff nm = ff.GetName().split("/")
 
bool t = fin["cbmsim"]
 
ROOT fout = ROOT.TFile(tmpFile, "recreate")
 
ROOT fSink = ROOT.FairRootFileSink(fout)
 
bool sTree = t.CloneTree(0)
 
list pointContainers = []
 
branch name = branch.GetName()
 
bool empty = True
 
getattr container = getattr(sTree, containerName)
 
ROOT branches = ROOT.TList()
 
os rc1 = os.system("rm " + outFile)
 
os rc2 = os.system("mv " + tmpFile + " " + outFile)
 
temp_filename = outFile.replace(".root", "_tmp.root")
 
f_outputfile output_tree = f_outputfile["cbmsim"]
 
ROOT muondis_tree = ROOT.TChain("DIS")
 
f_outputfile new_tree = output_tree.CloneTree(0)
 
array cross_section = array("f", [0.0])
 
f_outputfile cross_section_leaf = new_tree.Branch("CrossSection", cross_section, "CrossSection/F")
 
muondis_event mu = muondis_event.InMuon[0]
 
ROOT f_input = ROOT.TFile.Open(inputFile[0], "READ")
 
ROOT gst = f_input.gst
 
str selection_string = "(Entry$ >= " + str(options.firstEvent) + ")"
 
ROOT f_output = ROOT.TFile.Open(outFile, "UPDATE")
 
ROOT gst_copy = gst.CopyTree(selection_string)
 

Function Documentation

◆ checkOverlapsWithGeant4()

None run_simScript.checkOverlapsWithGeant4 ( )

Definition at line 897 of file run_simScript.py.

897def checkOverlapsWithGeant4() -> None:
898 # after /run/initialize, but prints warning messages, problems with TGeo volume
899 mygMC = ROOT.TGeant4.GetMC()
900 mygMC.ProcessGeantCommand("/geometry/test/recursion_start 0")
901 mygMC.ProcessGeantCommand("/geometry/test/recursion_depth 2")
902 mygMC.ProcessGeantCommand("/geometry/test/run")

◆ visualizeMagFields()

None run_simScript.visualizeMagFields ( )

Definition at line 893 of file run_simScript.py.

893def visualizeMagFields() -> None:
895
896

Variable Documentation

◆ action

run_simScript.action

Definition at line 46 of file run_simScript.py.

◆ available_snd_designs

list run_simScript.available_snd_designs = [1, 2]

Definition at line 270 of file run_simScript.py.

◆ branches

ROOT run_simScript.branches = ROOT.TList()

Definition at line 815 of file run_simScript.py.

◆ charmonly

bool run_simScript.charmonly = False

Definition at line 38 of file run_simScript.py.

◆ check

f run_simScript.check = xxx[len(xxx) - 1]

Definition at line 781 of file run_simScript.py.

◆ choices

run_simScript.choices

Definition at line 198 of file run_simScript.py.

◆ const

run_simScript.const

Definition at line 236 of file run_simScript.py.

◆ container

getattr run_simScript.container = getattr(sTree, containerName)

Definition at line 806 of file run_simScript.py.

◆ Cosmicsgen

ROOT run_simScript.Cosmicsgen = ROOT.CosmicsGenerator()

Definition at line 645 of file run_simScript.py.

◆ cross_section

array run_simScript.cross_section = array("f", [0.0])

Definition at line 855 of file run_simScript.py.

◆ cross_section_leaf

f_outputfile run_simScript.cross_section_leaf = new_tree.Branch("CrossSection", cross_section, "CrossSection/F")

Definition at line 856 of file run_simScript.py.

◆ ctime

ROOT run_simScript.ctime = timer.CpuTime()

Definition at line 762 of file run_simScript.py.

◆ default

run_simScript.default

Definition at line 62 of file run_simScript.py.

◆ defaultInputFile

bool run_simScript.defaultInputFile = True

Definition at line 42 of file run_simScript.py.

◆ dest

run_simScript.dest

Definition at line 47 of file run_simScript.py.

◆ DISgen

ROOT run_simScript.DISgen = ROOT.MuDISGenerator()

Definition at line 544 of file run_simScript.py.

◆ DownScaleDiMuon

bool run_simScript.DownScaleDiMuon = False

Definition at line 18 of file run_simScript.py.

◆ else

run_simScript.else :

Definition at line 273 of file run_simScript.py.

◆ empty

bool run_simScript.empty = True

Definition at line 804 of file run_simScript.py.

◆ EnergyCut

float run_simScript.EnergyCut = 10.0 * u.MeV if options.mudis else 100.0 * u.MeV

Definition at line 682 of file run_simScript.py.

◆ EvtCalcGen

ROOT run_simScript.EvtCalcGen = ROOT.EvtCalcGenerator()

Definition at line 512 of file run_simScript.py.

◆ exclude

run_simScript.exclude

Definition at line 724 of file run_simScript.py.

◆ f_input

ROOT run_simScript.f_input = ROOT.TFile.Open(inputFile[0], "READ")

Definition at line 871 of file run_simScript.py.

◆ f_output

ROOT run_simScript.f_output = ROOT.TFile.Open(outFile, "UPDATE")

Definition at line 880 of file run_simScript.py.

◆ fastMuon

run_simScript.fastMuon

Definition at line 635 of file run_simScript.py.

◆ fGeo

ROOT run_simScript.fGeo = ROOT.gGeoManager

Definition at line 751 of file run_simScript.py.

◆ fieldMaker

geomGeant4 run_simScript.fieldMaker = geomGeant4.addVMCFields(ship_geo, verbose=True)

Definition at line 718 of file run_simScript.py.

◆ fieldMap

run_simScript.fieldMap

Definition at line 717 of file run_simScript.py.

◆ fileType

ut run_simScript.fileType = ut.checkFileExists(inputFile)

Definition at line 598 of file run_simScript.py.

◆ fin

ROOT run_simScript.fin = False

Definition at line 782 of file run_simScript.py.

◆ float

run_simScript.float

Definition at line 64 of file run_simScript.py.

◆ fout

ROOT run_simScript.fout = ROOT.TFile(tmpFile, "recreate")

Definition at line 790 of file run_simScript.py.

◆ fSink

ROOT run_simScript.fSink = ROOT.FairRootFileSink(fout)

Definition at line 791 of file run_simScript.py.

◆ fStack

ROOT run_simScript.fStack = gMC.GetStack()

Definition at line 678 of file run_simScript.py.

◆ genie_parser

ArgumentParser run_simScript.genie_parser = subparsers.add_parser("Genie", help="Genie for reading and processing neutrino interactions")

Definition at line 78 of file run_simScript.py.

◆ Geniegen

ROOT run_simScript.Geniegen = ROOT.GenieGenerator()

Definition at line 561 of file run_simScript.py.

◆ geofile_name

f run_simScript.geofile_name = f"{options.outputDir}/geo_{run_identifier}.root"

Definition at line 741 of file run_simScript.py.

◆ gMC

ROOT run_simScript.gMC = ROOT.TVirtualMC.GetMC()

Definition at line 677 of file run_simScript.py.

◆ gst

ROOT run_simScript.gst = f_input.gst

Definition at line 873 of file run_simScript.py.

◆ gst_copy

ROOT run_simScript.gst_copy = gst.CopyTree(selection_string)

Definition at line 883 of file run_simScript.py.

◆ help

run_simScript.help

Definition at line 46 of file run_simScript.py.

◆ HNL

bool run_simScript.HNL = True

Definition at line 39 of file run_simScript.py.

◆ inclusive

bool run_simScript.inclusive = "c"

Definition at line 32 of file run_simScript.py.

◆ inputFile

str run_simScript.inputFile = "$EOSSHIP/eos/experiment/ship/data/Charm/Cascade-parp16-MSTP82-1-MSEL4-978Bpot.root"

Definition at line 41 of file run_simScript.py.

◆ int

run_simScript.int

Definition at line 62 of file run_simScript.py.

◆ kParameterMerged

ROOT run_simScript.kParameterMerged = ROOT.kTRUE

Definition at line 733 of file run_simScript.py.

◆ mcEngine

str run_simScript.mcEngine = "TGeant4"

Definition at line 30 of file run_simScript.py.

◆ MCTracksWithEnergyCutOnly

bool run_simScript.MCTracksWithEnergyCutOnly = True

Definition at line 35 of file run_simScript.py.

◆ MCTracksWithHitsOnly

bool run_simScript.MCTracksWithHitsOnly = False

Definition at line 34 of file run_simScript.py.

◆ MCTracksWithHitsOrEnergyCut

bool run_simScript.MCTracksWithHitsOrEnergyCut = False

Definition at line 36 of file run_simScript.py.

◆ modules

shipDet_conf run_simScript.modules = shipDet_conf.configure(run, ship_geo)

Definition at line 406 of file run_simScript.py.

◆ motherMode

ArgumentParser run_simScript.motherMode = options.MM

Definition at line 290 of file run_simScript.py.

◆ mu

muondis_event run_simScript.mu = muondis_event.InMuon[0]

Definition at line 859 of file run_simScript.py.

◆ mu_end

run_simScript.mu_end

Definition at line 549 of file run_simScript.py.

◆ mu_start

run_simScript.mu_start

Definition at line 549 of file run_simScript.py.

◆ MuonBackgen

ROOT run_simScript.MuonBackgen = ROOT.MuonBackGenerator()

Definition at line 605 of file run_simScript.py.

◆ muondis_tree

ROOT run_simScript.muondis_tree = ROOT.TChain("DIS")

Definition at line 849 of file run_simScript.py.

◆ myPgun

ROOT run_simScript.myPgun = ROOT.FairBoxGenerator(options.pID, 1)

Definition at line 523 of file run_simScript.py.

◆ n_EVENTS

run_simScript.n_EVENTS

Definition at line 652 of file run_simScript.py.

◆ name

branch run_simScript.name = branch.GetName()

Definition at line 798 of file run_simScript.py.

◆ nargs

run_simScript.nargs

Definition at line 252 of file run_simScript.py.

◆ nEvents

int run_simScript.nEvents = 0

Definition at line 516 of file run_simScript.py.

◆ new_tree

f_outputfile run_simScript.new_tree = output_tree.CloneTree(0)

Definition at line 853 of file run_simScript.py.

◆ nm

ff run_simScript.nm = ff.GetName().split("/")

Definition at line 784 of file run_simScript.py.

◆ None

run_simScript.None

Definition at line 227 of file run_simScript.py.

◆ Ntuplegen

ROOT run_simScript.Ntuplegen = ROOT.NtupleGenerator()

Definition at line 590 of file run_simScript.py.

◆ onlyWithField

run_simScript.onlyWithField

Definition at line 724 of file run_simScript.py.

◆ Opt_high

int run_simScript.Opt_high = int(options.cosmics)

Definition at line 292 of file run_simScript.py.

◆ options

ArgumentParser run_simScript.options = parser.parse_args()

Definition at line 268 of file run_simScript.py.

◆ outFile

f run_simScript.outFile = f"{options.outputDir}/sim_{run_identifier}.root"

Definition at line 383 of file run_simScript.py.

◆ output_tree

f_outputfile run_simScript.output_tree = f_outputfile["cbmsim"]

Definition at line 848 of file run_simScript.py.

◆ P6gen

ROOT run_simScript.P6gen = ROOT.tPythia6Generator()

Definition at line 502 of file run_simScript.py.

◆ P8gen

ROOT run_simScript.P8gen = ROOT.HNLPythia8Generator()

Definition at line 413 of file run_simScript.py.

◆ parFile

f run_simScript.parFile = f"{options.outputDir}/params_{run_identifier}.root"

Definition at line 386 of file run_simScript.py.

◆ parOut

ROOT run_simScript.parOut = ROOT.FairParRootFileIo(kParameterMerged)

Definition at line 734 of file run_simScript.py.

◆ parser

ArgumentParser run_simScript.parser = ArgumentParser()

Definition at line 44 of file run_simScript.py.

◆ passDPconf

pythia8darkphoton_conf run_simScript.passDPconf
Initial value:
2 P8gen, options.theMass, options.theDPepsilon, inclusive, motherMode, options.deepCopy
3 )
def configure(P8gen, mass, epsilon, inclusive, motherMode, deepCopy=False, debug=True)

Definition at line 455 of file run_simScript.py.

◆ pdg

ROOT run_simScript.pdg = ROOT.TDatabasePDG.Instance()

Definition at line 579 of file run_simScript.py.

◆ pg_parser

ArgumentParser run_simScript.pg_parser = subparsers.add_parser("PG", help="Use Particle Gun")

Definition at line 57 of file run_simScript.py.

◆ pointContainers

list run_simScript.pointContainers = []

Definition at line 796 of file run_simScript.py.

◆ primGen

ROOT run_simScript.primGen = ROOT.FairPrimaryGenerator()

Definition at line 408 of file run_simScript.py.

◆ pythia8

run_simScript.pythia8

Definition at line 376 of file run_simScript.py.

◆ rc1

os run_simScript.rc1 = os.system("rm " + outFile)

Definition at line 837 of file run_simScript.py.

◆ rc2

os run_simScript.rc2 = os.system("mv " + tmpFile + " " + outFile)

Definition at line 838 of file run_simScript.py.

◆ rtdb

ROOT run_simScript.rtdb = run.GetRuntimeDb()

Definition at line 400 of file run_simScript.py.

◆ rtime

ROOT run_simScript.rtime = timer.RealTime()

Definition at line 761 of file run_simScript.py.

◆ run

ROOT run_simScript.run = ROOT.FairRunSim()

Definition at line 396 of file run_simScript.py.

◆ run_identifier

ArgumentParser run_simScript.run_identifier = options.output_tag if options.output_tag else str(uuid.uuid4())

Definition at line 380 of file run_simScript.py.

◆ seed

ArgumentParser run_simScript.seed = options.theSeed

Definition at line 340 of file run_simScript.py.

◆ selection_string

str run_simScript.selection_string = "(Entry$ >= " + str(options.firstEvent) + ")"

Definition at line 875 of file run_simScript.py.

◆ ship_geo

geometry_config run_simScript.ship_geo
Initial value:
2 Yheight=options.dy,
3 strawDesign=options.strawDesign,
4 muShieldGeo=options.geofile,
5 shieldName=options.shieldName,
6 DecayVolumeMedium=options.decayVolMed,
7 SND=options.SND,
8 SND_design=options.SND_design,
9 TARGET_YAML=options.target_yaml,
10)
def create_config(str DecayVolumeMedium="helium", float Yheight=6.0, int strawDesign=10, muShieldGeo=None, str shieldName="New_HA_Design", int nuTargetPassive=1, bool SND=True, SND_design=None, TARGET_YAML=None)

Definition at line 360 of file run_simScript.py.

◆ SND_design

run_simScript.SND_design

Definition at line 272 of file run_simScript.py.

◆ sTree

bool run_simScript.sTree = t.CloneTree(0)

Definition at line 793 of file run_simScript.py.

◆ subparsers

ArgumentParser run_simScript.subparsers = parser.add_subparsers(dest="command", help="Which mode to run")

Definition at line 55 of file run_simScript.py.

◆ t

bool run_simScript.t = fin["cbmsim"]

Definition at line 789 of file run_simScript.py.

◆ temp_filename

f run_simScript.temp_filename = outFile.replace(".root", "_tmp.root")

Definition at line 842 of file run_simScript.py.

◆ test

ROOT run_simScript.test = ROOT.TPythia6()

Definition at line 501 of file run_simScript.py.

◆ testf

ROOT run_simScript.testf = ROOT.TFile.Open(inputFile[0])

Definition at line 612 of file run_simScript.py.

◆ theCouplings

list run_simScript.theCouplings = [float(c) for c in options.thecouplings.split(",")]

Definition at line 313 of file run_simScript.py.

◆ theDecayCouplings

list run_simScript.theDecayCouplings = [float(c) for c in options.thedeccouplings.split(",")]

Definition at line 317 of file run_simScript.py.

◆ theDPmass

float run_simScript.theDPmass = 0.2 * u.GeV

Definition at line 25 of file run_simScript.py.

◆ theHNLMass

float run_simScript.theHNLMass = 1.0 * u.GeV

Definition at line 21 of file run_simScript.py.

◆ theMass

run_simScript.theMass

Definition at line 309 of file run_simScript.py.

◆ theProductionCouplings

list run_simScript.theProductionCouplings = None

Definition at line 22 of file run_simScript.py.

◆ timer

ROOT run_simScript.timer = ROOT.TStopwatch()

Definition at line 392 of file run_simScript.py.

◆ tmpFile

f run_simScript.tmpFile = outFile + "tmp"

Definition at line 779 of file run_simScript.py.

◆ trajFilter

ROOT run_simScript.trajFilter = ROOT.FairTrajFilter.Instance()

Definition at line 699 of file run_simScript.py.

◆ True

run_simScript.True

Definition at line 724 of file run_simScript.py.

◆ try

run_simScript.try :

Definition at line 274 of file run_simScript.py.

◆ type

run_simScript.type

Definition at line 62 of file run_simScript.py.

◆ xxx

f run_simScript.xxx = outFile.split("/")

Definition at line 780 of file run_simScript.py.

◆ z

run_simScript.z

Definition at line 514 of file run_simScript.py.

◆ zDV

run_simScript.zDV

Definition at line 514 of file run_simScript.py.

◆ zTa

run_simScript.zTa

Definition at line 514 of file run_simScript.py.