import FWCore.ParameterSet.Config as cms

process = cms.Process("MERGE")

process.load("FWCore.MessageService.MessageLogger_cfi")
process.MessageLogger.cerr.FwkReport.reportEvery = 1000
process.MessageLogger.cerr.threshold = 'ERROR'

import FWCore.Framework.test.cmsExceptionsFatalOption_cff
process.options = cms.untracked.PSet(
  fileMode = cms.untracked.string('FULLMERGE'),
  Rethrow = FWCore.Framework.test.cmsExceptionsFatalOption_cff.Rethrow
)

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

process.source = cms.Source("PoolSource",
    fileNames = cms.untracked.vstring(
        'file:testRunMerge0.root', 
        'file:testRunMerge1.root', 
        'file:testRunMerge2extra.root', 
        'file:testRunMerge3extra.root',
        'file:testRunMerge4.root',
        'file:testRunMerge5.root'
    ),
    inputCommands = cms.untracked.vstring(
        'keep *', 
        'drop *_C_*_*',
        'drop *_*_*_EXTRA',
        'drop edmtestThingWithMerge_makeThingToBeDropped1_*_*'
    )
    , duplicateCheckMode = cms.untracked.string('checkEachRealDataFile')
)

process.thingWithMergeProducer = cms.EDProducer("ThingWithMergeProducer")

process.test = cms.EDAnalyzer("TestMergeResults",
                            
    #   Check to see that the value we read matches what we know
    #   was written. Expected values listed below come in sets of three
    #      value expected in Thing
    #      value expected in ThingWithMerge
    #      value expected in ThingWithIsEqual
    #   Each set of 3 is tested at endRun for the expected
    #   run values or at endLuminosityBlock for the expected
    #   lumi values. And then the next set of three values
    #   is tested at the next endRun or endLuminosityBlock.
    #   When the sequence of parameter values is exhausted it stops checking
    #   0's are just placeholders, if the value is a "0" the check is not made.

    expectedBeginRunProd = cms.untracked.vint32(
        10001,   10002,  10003,   # end run 100
        10001,   10002,  10003,   # end run 1
        10001,   10002,  10003,   # end run 1, no merge of runs because ProcessHistoryID different
        10001,   10002,  10003,   # end run 2
        10001,   10002,  10004    # end run 1
    ),
                            
    expectedEndRunProd = cms.untracked.vint32(
        100001, 100002, 100003,   # end run 100
        100001, 100002, 100003,   # end run 1
        100001, 100002, 100003,   # end run 1, no merge of runs because ProcessHistoryID different
        100001, 100002, 100003,   # end run 2
        100001, 100002, 100004    # end run 1
    ),

    expectedBeginLumiProd = cms.untracked.vint32(
        101,       102,    103,   # end run 100 lumi 100
        101,       102,    103,   # end run 1 lumi 1
        101,       102,    103,   # end run 1 lumi 1, no merge of runs because ProcessHistoryID different
        101,       102,    103,   # end run 2 lumi 1
        101,       102,    104    # end run 1 lumi 1
    ),

    expectedEndLumiProd = cms.untracked.vint32(
        1001,     1002,   1003,   # end run 100 lumi 100
        1001,     1002,   1003,   # end run 1 lumi 1
        1001,     1002,   1003,   # end run 1 lumi 1, no merge of runs because ProcessHistoryID different
        1001,     1002,   1003,   # end run 2 lumi 1
        1001,     1002,   1004    # end run 1 lumi 1
    ),

    expectedBeginRunNew = cms.untracked.vint32(
        10001,   10002,  10003,   # end run 100
        10001,   10002,  10003,   # end run 1
        10001,   10002,  10003,   # end run 1
        10001,   10002,  10003,   # end run 2
        10001,   10002,  10003    # end run 1
    ),

    expectedEndRunNew = cms.untracked.vint32(
        100001, 100002, 100003,   # end run 100
        100001, 100002, 100003,   # end run 1
        100001, 100002, 100003,   # end run 1
        100001, 100002, 100003,   # end run 2
        100001, 100002, 100003    # end run 1
    ),

    expectedBeginLumiNew = cms.untracked.vint32(
        101,       102,    103,   # end run 100 lumi 100
        101,       102,    103,   # end run 1 lumi 1
        101,       102,    103,   # end run 1 lumi 1
        101,       102,    103,   # end run 2 lumi 1
        101,       102,    103    # end run 1 lumi 1
    ),

    expectedEndLumiNew = cms.untracked.vint32(
        1001,     1002,   1003,   # end run 100 lumi 100
        1001,     1002,   1003,   # end run 1 lumi 1
        1001,     1002,   1003,   # end run 1 lumi 1
        1001,     1002,   1003,   # end run 2 lumi 1
        1001,     1002,   1003    # end run 1 lumi 1
    ),

    expectedDroppedEvent1 = cms.untracked.vint32(13, 13, -1, -1, -1, 13),
    expectedDroppedEvent1NEvents = cms.untracked.vint32(1,10,10,10,1,99),

    verbose = cms.untracked.bool(False),
    testAlias = cms.untracked.bool(True)
)

process.out = cms.OutputModule("PoolOutputModule",
    fileName = cms.untracked.string('testRunMergeMERGE2.root'),
    outputCommands = cms.untracked.vstring(
        'keep *', 
        'drop *_makeThingToBeDropped_*_*',
        'drop *_aliasForThingToBeDropped2_*_*',
        'drop *_B_*_*',
        'drop *_G_*_*',
        'drop *_H_*_*',
        'drop *_I_*_*',
        'drop *_J_*_*'
    )
)

process.checker = cms.OutputModule("GetProductCheckerOutputModule")

process.path1 = cms.Path(process.thingWithMergeProducer + process.test)
process.e = cms.EndPath(process.out * process.checker)
