MagneticField/Engine/test/regression.py - master - githubmirror/cmssw

MagneticField/Engine/test/regression.py
195 строк10 KB
Shahzad Malik Muzaffar
[RECONSTRUCTION] py2/3 compatibility:drop use of __future__
22 ноя 2024, 20:16
22 ноя 2024, 20:16e9abdc4

import FWCore.ParameterSet.Config as cms
import FWCore.ParameterSet.VarParsing as VarParsing
import sys

options = VarParsing.VarParsing()

options.register('producerType',
                 'static_DDD', #default value
                 VarParsing.VarParsing.multiplicity.singleton,
                 VarParsing.VarParsing.varType.string,
                 "MF producer to use. Valid values: 'static_DDD', 'static_DD4hep', 'fromDB', 'fromDB_DD4hep'")

options.register('era',
                 'RunII', #default value
                 VarParsing.VarParsing.multiplicity.singleton,
                 VarParsing.VarParsing.varType.string,
                 "'RunI'or 'RunII'")

options.register('current',
                 18000, #default value
                 VarParsing.VarParsing.multiplicity.singleton,
                 VarParsing.VarParsing.varType.float,
                 "Magnet current (nominal values: 18164=3.8T; 16730=3.5T; 14340=3T; 9500=2T; -1=loop in different IOVs, to test switching currents in the same job")

options.parseArguments()


process = cms.Process("MAGNETICFIELDTEST")

process.source = cms.Source("EmptySource")

process.maxEvents = cms.untracked.PSet(
    input = cms.untracked.int32(1)
)

REFERENCEFILE = 'none'

if options.current < 0 : # Test switching of maps in the same job, simulating different IOVs with different currents in runInfo
    # Note that this currently crashes with producerType='fromDB_DD4hep' when a different geometry has to be created in the switch, due a limitation of DD4hep.
    
    # FIXME: Only build the map and print the field at IP - no regression is actually run. To do that we would need to set up a mechanism to specify the
    # reference file file to be picked for each current at runtime.
    REFERENCEFILE = ''

    if options.producerType == 'static_DDD' or options.producerType == 'static_DD4hep' :
        sys.exit('Invalid configuration: current=-1 mode is not supported with static prouducers')
    process.source.numberEventsInLuminosityBlock =cms.untracked.uint32(1)

    if options.era=='RunI':
        process.source.firstLuminosityBlockForEachRun = cms.untracked.VLuminosityBlockID(
            cms.LuminosityBlockID(10,1),
            cms.LuminosityBlockID(20,2),
            cms.LuminosityBlockID(30,3),
            cms.LuminosityBlockID(40,4),
            cms.LuminosityBlockID(50,5),
            )

        process.riSource = cms.ESSource("EmptyESSource", recordName = cms.string("RunInfoRcd"),
                                        iovIsRunNotTime = cms.bool(True),
                                        firstValid = cms.vuint32(10,20,30,40,50))

        process.add_( cms.ESProducer("RunInfoTestESProducer",
                                     runInfos = cms.VPSet(cms.PSet(run = cms.int32(10), avg_current = cms.double(18000.)),
                                                          cms.PSet(run = cms.int32(20), avg_current = cms.double(16000.)),
                                                          cms.PSet(run = cms.int32(30), avg_current = cms.double(14000.)),
                                                          cms.PSet(run = cms.int32(40), avg_current = cms.double(10000.)),
                                                          cms.PSet(run = cms.int32(50), avg_current = cms.double(0.)),
                                                          ) ) )
    else :
        process.source.firstLuminosityBlockForEachRun = cms.untracked.VLuminosityBlockID(
            cms.LuminosityBlockID(300001,1),
            cms.LuminosityBlockID(300002,2),
            cms.LuminosityBlockID(300003,3),
            cms.LuminosityBlockID(300004,4),
            cms.LuminosityBlockID(300005,5),
            )
        process.add_( cms.ESProducer("RunInfoTestESProducer",
                                     runInfos = cms.VPSet(cms.PSet(run = cms.int32(300001), avg_current = cms.double(18000.)),
                                                          cms.PSet(run = cms.int32(300002), avg_current = cms.double(16000.)),
                                                          cms.PSet(run = cms.int32(300003), avg_current = cms.double(14000.)),
                                                          cms.PSet(run = cms.int32(300004), avg_current = cms.double(10000.)),
                                                          cms.PSet(run = cms.int32(300005), avg_current = cms.double(0.)),
                                                          ) ) )
        

    process.maxEvents.input = len(process.source.firstLuminosityBlockForEachRun)


if options.current > 18765 or (options.current <= 4779 and options.current>0) :
    sys.exit('ERROR: invalid current value: ' +  str(options.current))
elif options.current > 17543 :
    if options.era == 'RunII' :
        REFERENCEFILE = 'MagneticField/Engine/data/Regression/referenceField_160812_RII_3_8T.bin'
    elif options.era == 'RunI' :
        REFERENCEFILE = 'MagneticField/Engine/data/Regression/referenceField_160812_RI_3_8T.bin'
    else: 
        sys.exit('ERROR: Invalid era: ' + options.era)
elif options.current > 15617 :
    REFERENCEFILE = 'MagneticField/Engine/data/Regression/referenceField_160812_3_5T.bin'
elif options.current > 11987 :
    REFERENCEFILE = 'MagneticField/Engine/data/Regression/referenceField_160812_3T.bin'
elif options.current > 4779 :
    REFERENCEFILE = 'MagneticField/Engine/data/Regression/referenceField_71212_2T.bin'


if options.producerType == 'static_DDD':
    if options.current > 17543 :
        process.load("MagneticField.Engine.volumeBasedMagneticField_160812_cfi") # 3.8T, RII
        if options.era == 'RunI' :
            process.VolumeBasedMagneticFieldESProducer.version = cms.string('grid_160812_3_8t_Run1') # 3.8T, RII
    elif options.current > 15617 :
        process.load("MagneticField.Engine.volumeBasedMagneticField_160812_cfi")
        process.VolumeBasedMagneticFieldESProducer.version = cms.string('grid_160812_3_5t') # 3.5T
        process.ParametrizedMagneticFieldProducer.parameters.BValue = cms.string('3_5T')
    elif options.current > 11987 :
        process.load("MagneticField.Engine.volumeBasedMagneticField_160812_cfi")
        process.VolumeBasedMagneticFieldESProducer.version = cms.string('grid_160812_3t')
        process.ParametrizedMagneticFieldProducer.parameters.BValue = cms.string('3_0T')
    elif options.current > 4779 :
        process.load("MagneticField.Engine.volumeBasedMagneticField_71212_cfi") #2.0T


elif options.producerType == 'static_DD4hep' :
    process.load("MagneticField.Engine.volumeBasedMagneticField_dd4hep_160812_cfi") 
    if options.current > 17543 :
        if options.era == 'RunI' :
            process.VolumeBasedMagneticFieldESProducer.version = cms.string('grid_160812_3_8t_Run1') # 3.8T, RII
    elif options.current > 15617 :
        process.VolumeBasedMagneticFieldESProducer.version = cms.string('grid_160812_3_5t') # 3.5T
        process.ParametrizedMagneticFieldProducer.parameters.BValue = cms.string('3_5T')
    elif options.current > 11987 :
        process.VolumeBasedMagneticFieldESProducer.version = cms.string('grid_160812_3t')
        process.ParametrizedMagneticFieldProducer.parameters.BValue = cms.string('3_0T')
    elif options.current > 4779 :
        sys.exit('ERROR: Unsupported current for static_DD4hep: ' + str(options.current))


elif options.producerType == 'fromDB' or options.producerType == 'fromDB_DD4hep':
    if options.producerType == 'fromDB':
        process.load("Configuration.StandardSequences.MagneticField_cff")
    elif options.producerType == 'fromDB_DD4hep':
        process.load("MagneticField.Engine.volumeBasedMagneticFieldFromDB_dd4hep_cfi")

    process.load("Configuration.StandardSequences.FrontierConditions_GlobalTag_cff")
    from Configuration.AlCa.GlobalTag import GlobalTag
    process.GlobalTag = GlobalTag(process.GlobalTag, 'auto:run2_data', '') # Note: testing the proper set-up of data and MC iovs is beyond the scope of this script.

    if options.era == 'RunII' :
        process.source.firstRun = cms.untracked.uint32(300000)
    elif options.era == 'RunI' :
        process.source.firstRun = cms.untracked.uint32(100)

    # Note that this tests the map and the valueOverride method, but not the mechanism using the actual RunInfo, one mor set of tests could be added for this purpose.
    if options.current > 17543:
        process.VolumeBasedMagneticFieldESProducer.valueOverride = -1 #3.8T is the default
    elif options.current > 15617 :
        process.VolumeBasedMagneticFieldESProducer.valueOverride = 17000 #3.5 T
    elif options.current > 11987 :
        process.VolumeBasedMagneticFieldESProducer.valueOverride = 14000 #3 T
    elif options.current > 4779 :
        process.VolumeBasedMagneticFieldESProducer.valueOverride = 10000 #2 T



else :
    print('\nERROR: invalid producerType', producerType,'\n')


print('\nRegression for MF built with', options.producerType, 'era:', options.era, 'current:', options.current,'\n')


process.testMagneticField = cms.EDAnalyzer("testMagneticField",

## Uncomment to write down the reference file
#	outputTable = cms.untracked.string("newtable.bin"),

## Use the specified reference file to compare with
	inputTable = cms.untracked.string(REFERENCEFILE),

## Valid input file types: "xyz", "rpz_m", "xyz_m", "TOSCA" 
	inputTableType = cms.untracked.string("xyz"),

## Resolution used for validation, number of points
	resolution     = cms.untracked.double(0.0001),
        numberOfPoints = cms.untracked.int32(1000000),

## Size of testing volume (cm):
	InnerRadius = cms.untracked.double(0),    #  default: 0 
	OuterRadius = cms.untracked.double(900),  #  default: 900 
        minZ  = cms.untracked.double(-2400),      #  default: -2400 
        maxZ  = cms.untracked.double(2400)        #  default: 2400 

)

process.p1 = cms.Path(process.testMagneticField)