Utilities/test/FU_scouting_2.py

0001 from __future__ import print_function
0002 import FWCore.ParameterSet.Config as cms
0003 import FWCore.ParameterSet.VarParsing as VarParsing
0004 import os
0005
0006 options = VarParsing.VarParsing ('analysis')
0007
0008 options.register ('runNumber',
0009 #                  325175, # default value
0010                   325172, # default value
0011                   VarParsing.VarParsing.multiplicity.singleton,
0012                   VarParsing.VarParsing.varType.int,          # string, int, or float
0013                   "Run Number")
0014
0015 options.register ('buBaseDir',
0016                   'ramdisk', # default value
0017                   VarParsing.VarParsing.multiplicity.singleton,
0018                   VarParsing.VarParsing.varType.string,          # string, int, or float
0019                   "BU base directory")
0020
0021 options.register ('fuBaseDir',
0022                   'data', # default value
0023                   VarParsing.VarParsing.multiplicity.singleton,
0024                   VarParsing.VarParsing.varType.string,          # string, int, or float
0025                   "BU base directory")
0026
0027 options.register ('fffBaseDir',
0028                   '.', # default value
0029                   VarParsing.VarParsing.multiplicity.singleton,
0030                   VarParsing.VarParsing.varType.string,          # string, int, or float
0031                   "FFF base directory")
0032
0033 options.register ('numThreads',
0034                   1, # default value
0035                   VarParsing.VarParsing.multiplicity.singleton,
0036                   VarParsing.VarParsing.varType.int,          # string, int, or float
0037                   "Number of CMSSW threads")
0038
0039 options.register ('numFwkStreams',
0040                   1, # default value
0041                   VarParsing.VarParsing.multiplicity.singleton,
0042                   VarParsing.VarParsing.varType.int,          # string, int, or float
0043                   "Number of CMSSW streams")
0044
0045
0046 options.parseArguments()
0047
0048 cmsswbase = os.path.expandvars("$CMSSW_BASE/")
0049
0050 process = cms.Process("TESTFU")
0051 process.maxEvents = cms.untracked.PSet(
0052     input = cms.untracked.int32(-1)
0053 )
0054
0055 process.options = cms.untracked.PSet(
0056     numberOfThreads = cms.untracked.uint32(options.numThreads),
0057     numberOfStreams = cms.untracked.uint32(options.numFwkStreams),
0058     numberOfConcurrentLuminosityBlocks = cms.untracked.uint32(1) # ShmStreamConsumer requires synchronization at LuminosityBlock boundaries
0059 )
0060 process.MessageLogger = cms.Service("MessageLogger",
0061     cout = cms.untracked.PSet(threshold = cms.untracked.string( "INFO" )),
0062     destinations = cms.untracked.vstring( 'cout' )
0063 )
0064
0065 process.FastMonitoringService = cms.Service("FastMonitoringService",
0066     sleepTime = cms.untracked.int32(1)
0067 )
0068
0069 process.EvFDaqDirector = cms.Service("EvFDaqDirector",
0070     useFileBroker = cms.untracked.bool(False),
0071     fileBrokerHostFromCfg = cms.untracked.bool(True),
0072     fileBrokerHost = cms.untracked.string("htcp40.cern.ch"),
0073     runNumber = cms.untracked.uint32(options.runNumber),
0074     baseDir = cms.untracked.string(options.fffBaseDir+"/"+options.fuBaseDir),
0075     buBaseDir = cms.untracked.string(options.fffBaseDir+"/"+options.buBaseDir),
0076     directorIsBU = cms.untracked.bool(False),
0077 )
0078
0079 try:
0080   os.makedirs(options.fffBaseDir+"/"+options.fuBaseDir+"/run"+str(options.runNumber).zfill(6))
0081 except Exception as ex:
0082   print(str(ex))
0083   pass
0084
0085 ram_dir_path=options.buBaseDir+"/run"+str(options.runNumber).zfill(6)+"/"
0086
0087 process.source = cms.Source("DAQSource",
0088 #    dataMode = cms.untracked.string("ScoutingRun2Muon"),
0089     dataMode = cms.untracked.string("ScoutingRun2Multi"),
0090     verifyChecksum = cms.untracked.bool(True),
0091     useL1EventID = cms.untracked.bool(False),
0092     eventChunkSize = cms.untracked.uint32(8),
0093     eventChunkBlock = cms.untracked.uint32(8),
0094     numBuffers = cms.untracked.uint32(2),
0095     maxBufferedFiles = cms.untracked.uint32(2),
0096     fileListMode = cms.untracked.bool(True),
0097     fileNames = cms.untracked.vstring(
0098 #        ram_dir_path+"run325175_ls0001_index000001.raw"
0099         ram_dir_path+"run325172_ls0010_index000000.raw",
0100         ram_dir_path+"run325172_ls0380_index000000.raw"
0101     )
0102
0103 )
0104
0105 process.output = cms.OutputModule("PoolOutputModule",
0106     fileName = cms.untracked.string('file:PoolOutputTest.root'),
0107     outputCommands = cms.untracked.vstring("drop *","keep *_rawDataCollector_*_*")
0108     )
0109
0110
0111 process.ep = cms.EndPath(
0112 #  process.streamA
0113 #  + process.streamB
0114 #  + process.streamC
0115 # + process.streamD
0116   process.output
0117 )