Configuration/DataProcessing/python/Impl/hltScouting.py - master - githubmirror/cmssw

Configuration/DataProcessing/python/Impl/hltScouting.py
103 строки3 KB
Shahzad Malik Muzaffar
[OPERATIONS] py2/3 compatibility:drop use of __future__
22 ноя 2024, 20:16
22 ноя 2024, 20:166a1a334
#!/usr/bin/env python3
"""
_hltScouting_

Scenario supporting proton collisions with input HLT scouting data

"""


import os
import sys

from Configuration.DataProcessing.Scenario import *
from Configuration.DataProcessing.Utils import stepSKIMPRODUCER, addMonitoring, dictIO, nanoFlavours, gtNameAndConnect
import FWCore.ParameterSet.Config as cms

import warnings

class hltScouting(Scenario):
    def __init__(self):
        Scenario.__init__(self)
        self.recoSeq = ''
        self.cbSc = 'pp'
        self.isRepacked = False
        self.promptCustoms = ['Configuration/DataProcessing/RecoTLR.customisePrompt']
        self.promptModifiers = cms.ModifierChain()
    """
    _hltScouting_

    Implement configuration building for data processing for proton
    collision data taking with input HLT scouting data
    """

    def promptReco(self, globalTag, **args):
        """
        _promptReco_

        Proton collision data taking prompt reco with input HLT scouting data

        """

        options = Options()
        options.__dict__.update(defaultOptions.__dict__)
        options.scenario = self.cbSc
        
        if 'nThreads' in args:
            options.nThreads = args['nThreads']

        PhysicsSkimStep = ''
        if 'PhysicsSkims' in args:
            PhysicsSkimStep = stepSKIMPRODUCER(args['PhysicsSkims'])

        miniAODStep = ''
        nanoAODStep = ''
        
        if 'outputs' in args:
            outputs = []
            for a in args['outputs']:
                if a['dataTier'] in ['NANOAOD', 'NANOEDMAOD']:
                    if 'nanoFlavours' in args:
                        for nanoFlavour in args['nanoFlavours']:
                            if nanoFlavour != '@Scout':
                                warnings.warn('nanoFlavour: ' + nanoFlavour + 'is currently not supported and will be removed from outputs. Only supported nanoFlavour is @Scout')
                        args['nanoFlavours'] = ['@Scout']
                        nanoAODStep = ',NANO' + nanoFlavours(args['nanoFlavours'])
                    else:
                        nanoAODStep = ',NANO:@Scout' # default to Scouting NANO
                    outputs.append(a)
                else:
                    warnings.warn('dataTier:' + str(a['dataTier']) + ' is currently not supported and will be removed from outputs')
            if {output['dataTier'] for output in outputs} != {a['dataTier'] for a in args['outputs']}:
                warnings.warn('The outputs will be changed from ' + str(args['outputs']) + ' to' + str(outputs))
                args['outputs'] = outputs

        if not 'customs' in args:
            args['customs'] = []

        for c in self.promptCustoms:
            args['customs'].append(c)
        options.customisation_file = args['customs']
        
        options.isRepacked = args.get('repacked', self.isRepacked)
        
        options.step = ''
        options.step += self.recoSeq + PhysicsSkimStep
        options.step += miniAODStep + nanoAODStep

        dictIO(options, args)
        options.conditions = gtNameAndConnect(globalTag, args)
        
        process = cms.Process('HLTSCOUT', cms.ModifierChain(self.eras, self.promptModifiers))
        cb = ConfigBuilder(options, process = process, with_output = True)

        # Input source
        process.source = cms.Source("PoolSource",
            fileNames = cms.untracked.vstring()
        )

        cb.prepare()

        addMonitoring(process)
        
        return process