"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