PhaseII.py

Go to the documentation of this file.

# Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
from Campaigns.Utils import Campaign
 
def PhaseIIPileUpBase(flags, collisions=200):
    """Phase-II Upgrade / Run 4 flags for MC with pile-up"""
    flags.Beam.NumberOfCollisions = collisions
    flags.Input.MCCampaign = Campaign.PhaseII
 
    flags.GeoModel.Align.Dynamic = False  # no dynamic alignment for now
 
    from LArConfiguration.LArConfigRun3 import LArConfigRun3PileUp
    LArConfigRun3PileUp(flags)
    flags.LAr.ROD.NumberOfCollisions = collisions  # Run 4 default
 
    flags.Digitization.DoInnerDetectorNoise = False  # disable noise for now
 
 
def PhaseIIPileUp1(flags):
    """Phase-II Upgrade / Run 4 flags for MC with pile-up, mu=1, conditions like mu=60"""
    PhaseIIPileUpBase(flags, collisions=60)
 
    flags.Digitization.PU.NumberOfLowPtMinBias = 1.996
    flags.Digitization.PU.NumberOfHighPtMinBias = 0.004
    flags.Digitization.PU.BunchStructureConfig = 'RunDependentSimData.BunchStructure_2017'
    flags.Digitization.PU.CustomProfile={
        'run': 242006,
        'startmu': 1.0,
        'endmu': 2.0,
        'stepmu': 1.0,
        'startlb': 1,
        'timestamp': 1412006000
    }
 
 
def PhaseIIPileUp60(flags):
    """Phase-II Upgrade / Run 4 flags for MC with pile-up, mu=60"""
    PhaseIIPileUpBase(flags, collisions=60)
 
    
    flags.Digitization.PU.NumberOfLowPtMinBias = 69.863
    flags.Digitization.PU.NumberOfHighPtMinBias = 0.137
    flags.Digitization.PU.BunchStructureConfig = 'RunDependentSimData.BunchStructure_2017'
    flags.Digitization.PU.CustomProfile={
        'run': 242006,
        'startmu': 50.0,
        'endmu': 70.0,
        'stepmu': 1.0,
        'startlb': 1,
        'timestamp': 1412006000
    }
 
 
def PhaseIIPileUp140(flags):
    """Phase-II Upgrade / Run 4 flags for MC with pile-up, mu=140"""
    PhaseIIPileUpBase(flags, collisions=140)
 
    
    flags.Digitization.PU.NumberOfLowPtMinBias = 149.707
    flags.Digitization.PU.NumberOfHighPtMinBias = 0.293
    flags.Digitization.PU.BunchStructureConfig = 'RunDependentSimData.BunchStructure_2017'
    flags.Digitization.PU.CustomProfile={
        'run': 242014,
        'startmu': 130.0,
        'endmu': 150.0,
        'stepmu': 1.0,
        'startlb': 1,
        'timestamp': 1412014000
    }
 
 
def PhaseIIPileUp200(flags):
    """Phase-II Upgrade / Run 4 flags for MC with pile-up, mu=200"""
    PhaseIIPileUpBase(flags, collisions=200)
 
    
    flags.Digitization.PU.NumberOfLowPtMinBias = 209.590
    flags.Digitization.PU.NumberOfHighPtMinBias = 0.410
    flags.Digitization.PU.BunchStructureConfig = 'RunDependentSimData.BunchStructure_2017'
    flags.Digitization.PU.CustomProfile={
        'run': 242020,
        'startmu': 190.0,
        'endmu': 210.0,
        'stepmu': 1.0,
        'startlb': 1,
        'timestamp': 1412020000
    }
 
 
def PhaseIIPileUpMC21a(flags):
    """Phase-II Upgrade / Run 4 flags for MC with pile-up, MC21a pile-up profile"""
    PhaseIIPileUpBase(flags, collisions=60)
 
    flags.Digitization.PU.NumberOfLowPtMinBias = 84.335
    flags.Digitization.PU.NumberOfHighPtMinBias = 0.165
    flags.Digitization.PU.BunchStructureConfig = 'RunDependentSimData.BunchStructure_Fill7314_BCMSPattern_Flat'
    flags.Digitization.PU.ProfileConfig = 'RunDependentSimData.PileUpProfile_run242006_MC21a_SingleBeamspot'
 
 
def PhaseIINoPileUp(flags):
    """Phase-II Upgrade / Run 4 flags for MC without pile-up"""
    flags.Beam.NumberOfCollisions = 0.
    flags.Input.MCCampaign = Campaign.PhaseII
 
    flags.GeoModel.Align.Dynamic = False  # no dynamic alignment for now
 
    from LArConfiguration.LArConfigRun3 import LArConfigRun3NoPileUp
    LArConfigRun3NoPileUp(flags)
 
    flags.Digitization.DoInnerDetectorNoise = False  # disable noise for now
 
 
def PhaseIISimulationNoIoV(flags):
    """Phase-II Upgrade / Run 4 flags for simulation"""
    flags.Input.MCCampaign = Campaign.PhaseII
 
    from SimulationConfig.SimEnums import TruthStrategy
    flags.Sim.PhysicsList = 'FTFP_BERT_ATL'
    flags.Sim.TruthStrategy = TruthStrategy.MC15aPlus
    flags.Sim.TightMuonStepping = True
 
    from SimulationConfig.G4Optimizations import enableBeamPipeKill, enableFrozenShowersFCalOnly
    enableBeamPipeKill(flags)
    enableFrozenShowersFCalOnly(flags)
    from SimulationConfig.G4Optimizations import enableG4Optimizations
    enableG4Optimizations(flags)
 
 
def PhaseIISimulationSingleIoV(flags):
    """Phase-II Upgrade / Run 4 flags for simulation"""
    PhaseIISimulationNoIoV(flags)
 
    flags.Input.RunNumbers = [242000]
    flags.Input.OverrideRunNumber = True
    flags.Input.LumiBlockNumbers = [1] # dummy value
 
 
def PhaseIISimulation(flags):
    """Phase-II Upgrade / Run 4 flags for simulation (alias)"""
    PhaseIISimulationSingleIoV(flags)
 
 
def MC23PhaseIIPileUp1(flags):
    """Phase-II Upgrade / Run 4 flags for MC23 and above with pile-up, mu=1, conditions like mu=60"""
    PhaseIIPileUpBase(flags, collisions=60)
 
    flags.Digitization.PU.NumberOfLowPtMinBias = 1.996
    flags.Digitization.PU.NumberOfHighPtMinBias = 0.004
    flags.Digitization.PU.BunchStructureConfig = 'RunDependentSimData.BunchStructure_2017'
    flags.Digitization.PU.CustomProfile={
        'run': 350060,
        'startmu': 1.0,
        'endmu': 2.0,
        'stepmu': 1.0,
        'startlb': 1,
        'timestamp': 1625110000
    }
 
 
def MC23PhaseIIPileUp60(flags):
    """Phase-II Upgrade / Run 4 flags for MC23 and above and above with pile-up, mu=60"""
    PhaseIIPileUpBase(flags, collisions=60)
 
    
    flags.Digitization.PU.NumberOfLowPtMinBias = 69.863
    flags.Digitization.PU.NumberOfHighPtMinBias = 0.137
    flags.Digitization.PU.BunchStructureConfig = 'RunDependentSimData.BunchStructure_2017'
    flags.Digitization.PU.CustomProfile={
        'run': 350060,
        'startmu': 50.0,
        'endmu': 70.0,
        'stepmu': 1.0,
        'startlb': 1,
        'timestamp': 1625110000
    }
 
 
def MC23PhaseIIPileUp140(flags):
    """Phase-II Upgrade / Run 4 flags for MC23 and above with pile-up, mu=140"""
    PhaseIIPileUpBase(flags, collisions=140)
 
    
    flags.Digitization.PU.NumberOfLowPtMinBias = 149.707
    flags.Digitization.PU.NumberOfHighPtMinBias = 0.293
    flags.Digitization.PU.BunchStructureConfig = 'RunDependentSimData.BunchStructure_2017'
    flags.Digitization.PU.CustomProfile={
        'run': 350140,
        'startmu': 130.0,
        'endmu': 150.0,
        'stepmu': 1.0,
        'startlb': 1,
        'timestamp': 1625120000
    }
 
 
def MC23PhaseIIPileUp200(flags):
    """Phase-II Upgrade / Run 4 flags for MC23 and above with pile-up, mu=200"""
    PhaseIIPileUpBase(flags, collisions=200)
 
    
    flags.Digitization.PU.NumberOfLowPtMinBias = 209.590
    flags.Digitization.PU.NumberOfHighPtMinBias = 0.410
    flags.Digitization.PU.BunchStructureConfig = 'RunDependentSimData.BunchStructure_2017'
    flags.Digitization.PU.CustomProfile={
        'run': 350200,
        'startmu': 190.0,
        'endmu': 210.0,
        'stepmu': 1.0,
        'startlb': 1,
        'timestamp': 1625130000
    }
 
def MC23PhaseIIPileUpMC21a(flags):
    """Phase-II Upgrade / Run 4 flags for MC23 and above with pile-up, MC21a pile-up profile"""
    PhaseIIPileUpBase(flags, collisions=60)
 
    flags.Digitization.PU.NumberOfLowPtMinBias = 84.335
    flags.Digitization.PU.NumberOfHighPtMinBias = 0.165
    flags.Digitization.PU.BunchStructureConfig = 'RunDependentSimData.BunchStructure_Fill7314_BCMSPattern_Flat'
    flags.Digitization.PU.ProfileConfig = 'RunDependentSimData.PileUpProfile_run242006_MC21a_SingleBeamspot'
 
 
def MC23PhaseIINoPileUp(flags):
    """Phase-II Upgrade / Run 4 flags for MC23 and above without pile-up"""
    PhaseIINoPileUp(flags)
    
    
def MC23PhaseIISimulationNoIoV(flags):
    """Phase-II Upgrade / Run 4 flags for simulation"""
    from SimulationConfig.SimEnums import TruthStrategy
    flags.Sim.PhysicsList = 'FTFP_BERT_ATL'
    flags.Sim.TruthStrategy = TruthStrategy.MC15aPlus
 
    flags.Sim.TightMuonStepping = True
 
    from SimulationConfig.G4Optimizations import enableBeamPipeKill, enableFrozenShowersFCalOnly
    enableBeamPipeKill(flags)
    enableFrozenShowersFCalOnly(flags)
    from SimulationConfig.G4Optimizations import enableG4Optimizations
    enableG4Optimizations(flags)
 
 
def MC23PhaseIISimulationSingleIoV(flags):
    """Phase-II Upgrade / Run 4 flags for simulation"""
    MC23PhaseIISimulationNoIoV(flags)
 
    flags.Input.RunNumbers = [350000]
    flags.Input.OverrideRunNumber = True
    flags.Input.LumiBlockNumbers = [1] # dummy value
 
 
def MC23PhaseIISimulation(flags):
    """Phase-II Upgrade / Run 4 flags for simulation (alias)"""
    MC23PhaseIISimulationSingleIoV(flags)