test
CMS 3D CMS Logo

All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Pages
Merge.py
Go to the documentation of this file.
1 #!/usr/bin/env python
2 """
3 _Merge_
4 
5 Module that generates standard merge job configurations for use in any
6 standard processing
7 
8 """
9 
10 
11 from FWCore.ParameterSet.Config import Process, EndPath
12 from FWCore.ParameterSet.Modules import OutputModule, Source, Service
13 import FWCore.ParameterSet.Types as CfgTypes
14 
15 
16 def mergeProcess(*inputFiles, **options):
17  """
18  _mergeProcess_
19 
20  Creates and returns a merge process that will merge the provided
21  filenames
22 
23  supported options:
24 
25  - process_name : name of the process, defaults to Merge
26  - outputmod_label : label of the output module, defaults to Merged
27  - newDQMIO : specifies if the new DQM format should be used to merge the files
28  - output_file : sets the output file name
29  - output_lfn : sets the output LFN
30 
31  """
32  # //
33  # // process supported options
34  #//
35  processName = options.get("process_name", "Merge")
36  outputModLabel = options.get("outputmod_label", "Merged")
37  outputFilename = options.get("output_file", "Merged.root")
38  outputLFN = options.get("output_lfn", None)
39  dropDQM = options.get("drop_dqm", False)
40  newDQMIO = options.get("newDQMIO", False)
41 
42  # //
43  # // build process
44  #//
45  process = Process(processName)
46 
47  # //
48  # // input source
49  #//
50  if newDQMIO:
51  process.source = Source("DQMRootSource")
52  process.add_(Service("DQMStore"))
53  else:
54  process.source = Source("PoolSource")
55  if dropDQM:
56  process.source.inputCommands = CfgTypes.untracked.vstring('keep *','drop *_EDMtoMEConverter_*_*')
57  process.source.fileNames = CfgTypes.untracked(CfgTypes.vstring())
58  for entry in inputFiles:
59  process.source.fileNames.append(str(entry))
60 
61  # //
62  # // output module
63  #//
64  if newDQMIO:
65  outMod = OutputModule("DQMRootOutputModule")
66  else:
67  outMod = OutputModule("PoolOutputModule")
68  outMod.fileName = CfgTypes.untracked.string(outputFilename)
69  if outputLFN != None:
70  outMod.logicalFileName = CfgTypes.untracked.string(outputLFN)
71  setattr(process, outputModLabel, outMod)
72 
73  process.outputPath = EndPath(outMod)
74 
75  return process
def mergeProcess
Definition: Merge.py:16