Repack.py

Go to the documentation of this file.

#!/usr/bin/env python3
"""
_Repack_
 
Module that generates standard repack configurations
 
"""
 
import FWCore.ParameterSet.Config as cms
 
 
def repackProcess(**args):
    """
    _repackProcess_
 
    Creates and returns a repack process
 
    supported options:
 
    - outputs      : defines output modules
 
    """
    process = cms.Process("REPACK")
    process.load("FWCore.MessageLogger.MessageLogger_cfi")
 
    process.maxEvents = cms.untracked.PSet( input = cms.untracked.int32(-1) )
 
    process.configurationMetadata = cms.untracked.PSet(
        name = cms.untracked.string("repack-config"),
        version = cms.untracked.string("none"),
        annotation = cms.untracked.string("auto generated configuration")
        )
 
    process.options = cms.untracked.PSet(
        Rethrow = cms.untracked.vstring("ProductNotFound","TooManyProducts","TooFewProducts"),
        wantSummary = cms.untracked.bool(False)
        )
 
    process.source = cms.Source(
        "NewEventStreamFileReader",
        fileNames = cms.untracked.vstring()
        )
 
    outputs = args.get('outputs', [])
 
    if len(outputs) > 0:
        process.outputPath = cms.EndPath()
 
    for output in outputs:
 
        moduleLabel = output['moduleLabel']
        selectEvents = output.get('selectEvents', None)
        maxSize = output.get('maxSize', None)
 
        outputModule = cms.OutputModule(
            "PoolOutputModule",
            fileName = cms.untracked.string("%s.root" % moduleLabel)
            )
 
        outputModule.dataset = cms.untracked.PSet(dataTier = cms.untracked.string("RAW"))
 
        if maxSize != None:
            outputModule.maxSize = cms.untracked.int32(maxSize)
 
        if selectEvents != None:
            outputModule.SelectEvents = cms.untracked.PSet(
                SelectEvents = cms.vstring(selectEvents)
                )
 
        setattr(process, moduleLabel, outputModule)
 
        process.outputPath += outputModule
 
    return process