示例#1
0
import os
from GaudiKernel import SystemOfUnits as units
from GaudiKernel import PhysicalConstants as constants
from Gaudi.Configuration import *

from FWCore.joboptions import parse_standard_job_options

args = parse_standard_job_options()

inputfile = "muons_for_seeding.root"
if args.inputfile:
    inputfile = args.inputfile

from Configurables import FCCDataSvc

podioevent = FCCDataSvc("EventDataSvc", input=inputfile)

from Configurables import PodioInput

podioinput = PodioInput(
    "PodioReader",
    collections=[
        "GenParticles",
        "GenVertices",
        "SimParticles",
        "SimVertices",
        "TrackerHits",
        "TrackerHitsPosition",
    ],
    OutputLevel=DEBUG,
)
from Gaudi.Configuration import *

###### Parser ################
import argparse
parser = argparse.ArgumentParser()
parser.add_argument('--input', type=str, default=None, help='specify an input File')
my_args, _ = parser.parse_known_args()

from FWCore.joboptions import parse_standard_job_options
args = parse_standard_job_options()


my_input = ""
outfilename = "pgun_output.root"
if args.nevents is not None:
    nEvents = args.nevents

if args.outputfile is not None:
    outfilename = args.outputfile

if my_args.input != None:
    my_input = my_args.input
################################


# Data service
from Configurables import FCCDataSvc
podioevent = FCCDataSvc("EventDataSvc")

# from Configurables import HepMCReader
from Configurables import HepEVTReader