"PFMustache_RegPs1": ["scPreshowerEnergy/scRawEnergy"],
    "GedGsfElectron_RegPs2": [
        "scPreshowerEnergyPlane1/scRawEnergy",
        "scPreshowerEnergyPlane2/scRawEnergy"
    ],
    "PFMustache_RegPs2": [
        "scPreshowerEnergyPlane1/scRawEnergy",
        "scPreshowerEnergyPlane2/scRawEnergy"
    ],
}

clusterings = [
    ("PFMustache_RegPs1", "mustacheSCTree/SuperClusterTree"),
    ("GedGsfElectron_RegPs1", "gedGsfElectronTree/RegressionTree"),
    ("PFMustache_RegPs2", "mustacheSCTree/SuperClusterTree"),
    ("GedGsfElectron_RegPs2", "gedGsfElectronTree/RegressionTree"),
]

for tagClustering, treeClustering in clusterings:
    # define regressions
    batch.addRegression(tagClustering)
    batch.setInputTree(tagClustering, treeClustering)
    #batch.addVariablesEB(tagClustering, specificVariables[tagClustering], float)
    batch.addVariablesEE(tagClustering, specificVariablesEE[tagClustering])
    batch.addCuts(tagClustering, specificCuts[tagClustering])
    batch.doCombination(tagClustering, doCombine[tagClustering])

#batch.simulate = True

batch.info = info
specificVariablesEE = {
    "GedGsfElectron_RegPs1":["scPreshowerEnergy/scRawEnergy"],
    "PFMustache_RegPs1":["scPreshowerEnergy/scRawEnergy"],
    "GedGsfElectron_RegPs2":["scPreshowerEnergyPlane1/scRawEnergy","scPreshowerEnergyPlane2/scRawEnergy"],
    "PFMustache_RegPs2":["scPreshowerEnergyPlane1/scRawEnergy","scPreshowerEnergyPlane2/scRawEnergy"],
}



clusterings = [
    ("PFMustache_RegPs1","mustacheSCTree/SuperClusterTree"),
    ("GedGsfElectron_RegPs1","gedGsfElectronTree/RegressionTree"),
    ("PFMustache_RegPs2","mustacheSCTree/SuperClusterTree"),
    ("GedGsfElectron_RegPs2","gedGsfElectronTree/RegressionTree"),
]

for tagClustering,treeClustering in clusterings:
    # define regressions
    batch.addRegression(tagClustering)
    batch.setInputTree(tagClustering, treeClustering)
    #batch.addVariablesEB(tagClustering, specificVariables[tagClustering], float)
    batch.addVariablesEE(tagClustering, specificVariablesEE[tagClustering])
    batch.addCuts(tagClustering, specificCuts[tagClustering])
    batch.doCombination(tagClustering, doCombine[tagClustering])

#batch.simulate = True

batch.info = info