Functions
	get_work_dir (run_number, tag=None)

	init ()

Variables
str	mcEngine = "TGeant4"

str	simEngine = "Pythia8"

int	runnr = 1000

int	nev = 500

bool	checkOverlap = False

bool	G4only = False

bool	storeOnlyMuons = False

bool	withEvtGen = True

int	boostDiMuon = 1.

int	boostFactor = 1.

bool	charm = False

bool	beauty = False

float	chicc = 1.7e-3

float	chibb = 1.6e-7

int	npot = 5E13

int	nStart = 0

str	charmInputFile = "root://eoslhcb.cern.ch//eos/ship/data/Charm/Cascade-parp16-MSTP82-1-MSEL4-76Mpot_1.root"

str	outputDir = "."

	theSeed = int(10000 * time.time() % 10000000)

str	work_dir = "./"

float	ecut = 0.5

int	dy = 10.

int	dv = 5

int	ds = 7

int	nud = 1

	logger = logging.getLogger(os.path.splitext(os.path.basename(os.sys.argv[0]))[0])

	args = init()

	ship_geo = ConfigRegistry.loadpy("$FAIRSHIP/geometry/charm-geometry_config.py", Setup = args.CharmdetSetup, cTarget = args.CharmTarget)

str	txt = 'pythia8_Geant4_'

str	outFile = outputDir+'/'+txt+str(runnr)+'_'+str(ecut)+'.root'

str	parFile = outputDir+'/ship.params.'+txt+str(runnr)+'_'+str(ecut)+'.root'

	timer = ROOT.TStopwatch()

	run = ROOT.FairRunSim()

	rtdb = run.GetRuntimeDb()

	modules = shipDet_conf.configure(run,ship_geo)

	primGen = ROOT.FairPrimaryGenerator()

	P8gen = ROOT.FixedTargetGenerator()

	gMC = ROOT.TVirtualMC.GetMC()

	fStack = gMC.GetStack()

	gProcessTable = ROOT.G4ProcessTable.GetProcessTable()

	procAnnihil = gProcessTable.FindProcess(ROOT.G4String('AnnihiToMuPair'),ROOT.G4String('e+'))

	procGMuPair = gProcessTable.FindProcess(ROOT.G4String('GammaToMuPair'),ROOT.G4String('gamma'))

	rtime = timer.RealTime()

	ctime = timer.CpuTime()

str	tmpFile = outFile+"tmp"

	fin = ROOT.gROOT.GetListOfFiles()[0]

	fHeader = fin.FileHeader

	poteq = P8gen.GetPotForCharm()

	t = fin.cbmsim

	fout = ROOT.TFile(tmpFile,'recreate' )

	sTree = t.CloneTree(0)

int	nEvents = 0

	rc = t.GetEvent(n)

	xcopy = x.Clone()

	ff = fin.FileHeader.Clone(fout.GetName())

	sGeo = ROOT.gGeoManager

Function Documentation

◆ get_work_dir()

run_MufluxfixedTarget.get_work_dir	(	run_number,
		tag = `None`
	)

Definition at line 52 of file run_MufluxfixedTarget.py.

def get_work_dir(run_number,tag=None):
  import socket
  host = socket.gethostname()
  job_base_name = os.path.splitext(os.path.basename(os.sys.argv[0]))[0]
  if tag:  out_dir = "{host}_{base}_{runnr}_{comment}".format(host=host, base=job_base_name, runnr=run_number, comment=tag)
  else:    out_dir = "{host}_{base}_{runnr}".format(host=host, base=job_base_name, runnr=run_number)
  return out_dir
 
 

◆ init()

run_MufluxfixedTarget.init ( )

Definition at line 61 of file run_MufluxfixedTarget.py.

def init():
  global runnr, nev, ecut, G4only, tauOnly,JpsiMainly, work_dir,Debug,withEvtGen,boostDiMuon,\
         boostFactor,charm,beauty,charmInputFile,nStart,storeOnlyMuons,chicc,chibb,npot, nStart
  logger.info("SHiP proton-on-taget simulator (C) Thomas Ruf, 2017")
 
  ap = argparse.ArgumentParser(
      description='Run SHiP "pot" simulation')
  ap.add_argument('-d', '--debug', action='store_true', dest='debug')
  ap.add_argument('-f', '--force', action='store_true', help="force overwriting output directory")
  ap.add_argument('-cs', '--CharmdetSetup', type=int, dest='CharmdetSetup',help="setting detector setup", default=0)
  ap.add_argument('-ct', '--CharmTarget', type=int, dest='CharmTarget',help="choosing target configuration for charm exposure", default=3)
  ap.add_argument('-r', '--run-number', type=int, dest='runnr', default=runnr)
  ap.add_argument('-e', '--ecut', type=float, help="energy cut", dest='ecut', default=ecut)
  ap.add_argument('-n', '--num-events', type=int, help="number of events to generate", dest='nev', default=nev)
  ap.add_argument('-G', '--G4only', action='store_true', dest='G4only',     default=False, help="use Geant4 directly, no Pythia8")
  ap.add_argument('-P', '--PythiaDecay', action='store_true', dest='pythiaDecay', default=False,  help="use Pythia8 for decays")
  ap.add_argument('-V', '--EvtGen',      action='store_true', dest='withEvtGen',  default=withEvtGen,   help="use EvtGen for decays")
  ap.add_argument('-t', '--tau-only',     action='store_true', dest='tauOnly',     default=False)
  ap.add_argument('-J', '--Jpsi-mainly',  action='store_true', dest='JpsiMainly',  default=False)
  ap.add_argument('-b', '--boostDiMuon', type=float,   dest='boostDiMuon',  default=boostDiMuon, help="boost Di-muon branching ratios")
  ap.add_argument('-X', '--boostFactor', type=float,   dest='boostFactor',  default=boostFactor, help="boost Di-muon prod cross sections")
  ap.add_argument('-C', '--charm',      action='store_true',  dest='charm',  default=charm, help="generate charm decays")
  ap.add_argument('-B', '--beauty',     action='store_true',  dest='beauty', default=beauty, help="generate beauty decays")
  ap.add_argument('-M', '--storeOnlyMuons',  action='store_true',  dest='storeOnlyMuons',  default=storeOnlyMuons, help="store only muons, ignore neutrinos")
# for charm production       
  ap.add_argument('-cc','--chicc',action='store_true',  dest='chicc',  default=chicc, help="ccbar over mbias cross section")
  ap.add_argument('-bb','--chibb',action='store_true',  dest='chibb',  default=chibb, help="bbbar over mbias cross section")
  ap.add_argument('-p','--pot',action='store_true',  dest='npot',  default=npot, help="number of protons on target per spill to normalize on")
  ap.add_argument('-S','--nStart',action='store_true',  dest='nStart',  default=nStart, help="first event of input file to start")
  ap.add_argument('-I', '--InputFile', type=str, dest='charmInputFile',  default=charmInputFile, help="input file for charm/beauty decays")
  ap.add_argument('-o','--output'    , type=str, help="output directory", dest='work_dir', default=None)
  args = ap.parse_args()
  if args.debug:
      logger.setLevel(logging.DEBUG)
  runnr        = args.runnr
  nev          = args.nev
  ecut         = args.ecut
  tauOnly      = args.tauOnly
  JpsiMainly   = args.JpsiMainly
  G4only       = args.G4only
  boostFactor  = args.boostFactor
  boostDiMuon  = args.boostDiMuon
  storeOnlyMuons = args.storeOnlyMuons
  if G4only:
    args.charm  = False
    args.beauty = False
    args.withEvtGen = False
    args.pythiaDecay = False
  elif args.pythiaDecay:
    withEvtGen = False
    args.withEvtGen = False
    logger.info("use Pythia8 as primary decayer")
  else:
    withEvtGen = True
    logger.info("use EvtGen as primary decayer")
  withEvtGen     = args.withEvtGen
  charm  = args.charm
  beauty = args.beauty
  if charm and beauty: 
    logger.warn("charm and beauty decays are set! Beauty gets priority")
    charm = False
  charmInputFile = args.charmInputFile
  nStart = int(args.nStart)
  Debug  = args.debug
  if args.work_dir is None:
    if charm: args.work_dir = get_work_dir(runnr,"charm")
    if beauty: args.work_dir = get_work_dir(runnr,"beauty")
    else: args.work_dir = get_work_dir(runnr)
  work_dir = args.work_dir
  logger.debug("work_dir: %s" % work_dir)
  logger.debug("command line arguments: %s", args)
  if os.path.exists(work_dir):
    logger.warn("output directory '%s' already exists." % work_dir)
    if args.force:
      logger.warn("...cleaning")
      for root, dirs, files in os.walk(work_dir):
        for f in files:
          os.unlink(os.path.join(root, f))
        for d in dirs:
          shutil.rmtree(os.path.join(root, d))
    else:
      logger.warn("...use '-f' option to overwrite it")
  else:
    os.makedirs(work_dir)
  return args
 

Variable Documentation

◆ args

run_MufluxfixedTarget.args = init()

Definition at line 147 of file run_MufluxfixedTarget.py.

◆ beauty

bool run_MufluxfixedTarget.beauty = False

Definition at line 18 of file run_MufluxfixedTarget.py.

◆ boostDiMuon

int run_MufluxfixedTarget.boostDiMuon = 1.

Definition at line 15 of file run_MufluxfixedTarget.py.

◆ boostFactor

int run_MufluxfixedTarget.boostFactor = 1.

Definition at line 16 of file run_MufluxfixedTarget.py.

◆ charm

bool run_MufluxfixedTarget.charm = False

Definition at line 17 of file run_MufluxfixedTarget.py.

◆ charmInputFile

str run_MufluxfixedTarget.charmInputFile = "root://eoslhcb.cern.ch//eos/ship/data/Charm/Cascade-parp16-MSTP82-1-MSEL4-76Mpot_1.root"

Definition at line 24 of file run_MufluxfixedTarget.py.

◆ checkOverlap

bool run_MufluxfixedTarget.checkOverlap = False

Definition at line 11 of file run_MufluxfixedTarget.py.

◆ chibb

float run_MufluxfixedTarget.chibb = 1.6e-7

Definition at line 20 of file run_MufluxfixedTarget.py.

◆ chicc

float run_MufluxfixedTarget.chicc = 1.7e-3

Definition at line 19 of file run_MufluxfixedTarget.py.

◆ ctime

run_MufluxfixedTarget.ctime = timer.CpuTime()

Definition at line 238 of file run_MufluxfixedTarget.py.

◆ ds

int run_MufluxfixedTarget.ds = 7

Definition at line 34 of file run_MufluxfixedTarget.py.

◆ dv

int run_MufluxfixedTarget.dv = 5

Definition at line 33 of file run_MufluxfixedTarget.py.

◆ dy

int run_MufluxfixedTarget.dy = 10.

Definition at line 32 of file run_MufluxfixedTarget.py.

◆ ecut

float run_MufluxfixedTarget.ecut = 0.5

Definition at line 30 of file run_MufluxfixedTarget.py.

◆ ff

run_MufluxfixedTarget.ff = fin.FileHeader.Clone(fout.GetName())

Definition at line 274 of file run_MufluxfixedTarget.py.

◆ fHeader

run_MufluxfixedTarget.fHeader = fin.FileHeader

Definition at line 249 of file run_MufluxfixedTarget.py.

◆ fin

run_MufluxfixedTarget.fin = ROOT.gROOT.GetListOfFiles()[0]

Definition at line 247 of file run_MufluxfixedTarget.py.

◆ fout

run_MufluxfixedTarget.fout = ROOT.TFile(tmpFile,'recreate' )

Definition at line 259 of file run_MufluxfixedTarget.py.

◆ fStack

run_MufluxfixedTarget.fStack = gMC.GetStack()

Definition at line 213 of file run_MufluxfixedTarget.py.

◆ G4only

bool run_MufluxfixedTarget.G4only = False

Definition at line 12 of file run_MufluxfixedTarget.py.

◆ gMC

run_MufluxfixedTarget.gMC = ROOT.TVirtualMC.GetMC()

Definition at line 212 of file run_MufluxfixedTarget.py.

◆ gProcessTable

run_MufluxfixedTarget.gProcessTable = ROOT.G4ProcessTable.GetProcessTable()

Definition at line 225 of file run_MufluxfixedTarget.py.

◆ logger

run_MufluxfixedTarget.logger = logging.getLogger(os.path.splitext(os.path.basename(os.sys.argv[0]))[0])

Definition at line 48 of file run_MufluxfixedTarget.py.

◆ mcEngine

str run_MufluxfixedTarget.mcEngine = "TGeant4"

Definition at line 7 of file run_MufluxfixedTarget.py.

◆ modules

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

Definition at line 175 of file run_MufluxfixedTarget.py.

◆ nev

int run_MufluxfixedTarget.nev = 500

Definition at line 10 of file run_MufluxfixedTarget.py.

◆ nEvents

int run_MufluxfixedTarget.nEvents = 0

Definition at line 261 of file run_MufluxfixedTarget.py.

◆ npot

int run_MufluxfixedTarget.npot = 5E13

Definition at line 21 of file run_MufluxfixedTarget.py.

◆ nStart

int run_MufluxfixedTarget.nStart = 0

Definition at line 22 of file run_MufluxfixedTarget.py.

◆ nud

int run_MufluxfixedTarget.nud = 1

Definition at line 35 of file run_MufluxfixedTarget.py.

◆ outFile

str run_MufluxfixedTarget.outFile = outputDir+'/'+txt+str(runnr)+'_'+str(ecut)+'.root'

Definition at line 157 of file run_MufluxfixedTarget.py.

◆ outputDir

str run_MufluxfixedTarget.outputDir = "."

Definition at line 27 of file run_MufluxfixedTarget.py.

◆ P8gen

run_MufluxfixedTarget.P8gen = ROOT.FixedTargetGenerator()

Definition at line 179 of file run_MufluxfixedTarget.py.

◆ parFile

str run_MufluxfixedTarget.parFile = outputDir+'/ship.params.'+txt+str(runnr)+'_'+str(ecut)+'.root'

Definition at line 158 of file run_MufluxfixedTarget.py.

◆ poteq

run_MufluxfixedTarget.poteq = P8gen.GetPotForCharm()

Definition at line 253 of file run_MufluxfixedTarget.py.

◆ primGen

run_MufluxfixedTarget.primGen = ROOT.FairPrimaryGenerator()

Definition at line 178 of file run_MufluxfixedTarget.py.

◆ procAnnihil

run_MufluxfixedTarget.procAnnihil = gProcessTable.FindProcess(ROOT.G4String('AnnihiToMuPair'),ROOT.G4String('e+'))

Definition at line 226 of file run_MufluxfixedTarget.py.

◆ procGMuPair

run_MufluxfixedTarget.procGMuPair = gProcessTable.FindProcess(ROOT.G4String('GammaToMuPair'),ROOT.G4String('gamma'))

Definition at line 227 of file run_MufluxfixedTarget.py.

◆ rc

run_MufluxfixedTarget.rc = t.GetEvent(n)

Definition at line 263 of file run_MufluxfixedTarget.py.

◆ rtdb

run_MufluxfixedTarget.rtdb = run.GetRuntimeDb()

Definition at line 169 of file run_MufluxfixedTarget.py.

◆ rtime

run_MufluxfixedTarget.rtime = timer.RealTime()

Definition at line 237 of file run_MufluxfixedTarget.py.

◆ run

run_MufluxfixedTarget.run = ROOT.FairRunSim()

Definition at line 165 of file run_MufluxfixedTarget.py.

◆ runnr

int run_MufluxfixedTarget.runnr = 1000

Definition at line 9 of file run_MufluxfixedTarget.py.

◆ sGeo

run_MufluxfixedTarget.sGeo = ROOT.gGeoManager

Definition at line 286 of file run_MufluxfixedTarget.py.

◆ ship_geo

run_MufluxfixedTarget.ship_geo = ConfigRegistry.loadpy("$FAIRSHIP/geometry/charm-geometry_config.py", Setup = args.CharmdetSetup, cTarget = args.CharmTarget)

Definition at line 153 of file run_MufluxfixedTarget.py.

◆ simEngine

str run_MufluxfixedTarget.simEngine = "Pythia8"

Definition at line 8 of file run_MufluxfixedTarget.py.

◆ storeOnlyMuons

bool run_MufluxfixedTarget.storeOnlyMuons = False

Definition at line 13 of file run_MufluxfixedTarget.py.

◆ sTree

run_MufluxfixedTarget.sTree = t.CloneTree(0)

Definition at line 260 of file run_MufluxfixedTarget.py.

◆ t

run_MufluxfixedTarget.t = fin.cbmsim

Definition at line 258 of file run_MufluxfixedTarget.py.

◆ theSeed

run_MufluxfixedTarget.theSeed = int(10000 * time.time() % 10000000)

Definition at line 28 of file run_MufluxfixedTarget.py.

◆ timer

run_MufluxfixedTarget.timer = ROOT.TStopwatch()

Definition at line 161 of file run_MufluxfixedTarget.py.

◆ tmpFile

str run_MufluxfixedTarget.tmpFile = outFile+"tmp"

Definition at line 246 of file run_MufluxfixedTarget.py.

◆ txt

run_MufluxfixedTarget.txt = 'pythia8_Geant4_'

Definition at line 155 of file run_MufluxfixedTarget.py.

◆ withEvtGen

bool run_MufluxfixedTarget.withEvtGen = True

Definition at line 14 of file run_MufluxfixedTarget.py.

◆ work_dir

str run_MufluxfixedTarget.work_dir = "./"

Definition at line 29 of file run_MufluxfixedTarget.py.

◆ xcopy

run_MufluxfixedTarget.xcopy = x.Clone()

Definition at line 271 of file run_MufluxfixedTarget.py.

Functions

Variables