# "_Lr0_LwG", "_Lr1_LwG", "_Lr2_LwG", "_Lr3_LwG",
# "_Lr0_LwG", "_Lr1_MdG", "_Lr2_MdG", "_Lr3_MdG",
# "_Lr0_LwG", "_Lr1_HiG", "_Lr2_HiG", "_Lr3_HiG" )
#
#for cell in cells:
# ExtraContentPhotons.append("Photons."+cell)
# for layer in layers_gains:
# ExtraContentPhotons.append("Photons."+cell+layer)
#
#for cell in cells:
# ExtraContentElectrons.append("Electrons."+cell)
# for layer in layers_gains:
# ExtraContentElectrons.append("Electrons."+cell+layer)
from DerivationFrameworkCalo.DerivationFrameworkCaloFactories import GainDecorator, getGainDecorations
GainDecoratorTool = GainDecorator()
ExtraContentPhotons.extend(getGainDecorations(GainDecoratorTool))
ExtraContentElectrons.extend(getGainDecorations(GainDecoratorTool))
ExtraContentAll = ExtraContentElectrons + ExtraContentMuons + ExtraContentPhotons + ExtraContentGSFConversionVertices + ExtraContentHLTPhotons + ExtraContentPrimaryVertices
ExtraContentAllTruth = ExtraElectronsTruth + ExtraMuonsTruth + ExtraPhotonsTruth
ExtraContainersTruth = [
"TruthEvents", "TruthParticles", "TruthVertices", "AntiKt4TruthJets",
"egammaTruthParticles", "MuonTruthParticles"
#,"BTagging_AntiKt4TruthWZ"
#,"AntiKt4TruthWZJets"
]
ExtraContainersPhotons = [
"Photons",
"GSFTrackParticles",
HIGG1D1SlimmingHelper.SmartCollections = ["Electrons",
"Photons",
"Muons",
"MET_Reference_AntiKt4EMTopo",
"AntiKt4EMTopoJets",
"BTagging_AntiKt4EMTopo",
"InDetTrackParticles",
"PrimaryVertices" ]
HIGG1D1SlimmingHelper.AllVariables = ["Electrons","Photons","egammaClusters","GSFConversionVertices","TruthEvents", "TruthParticles", "TruthVertices", "AntiKt4TruthJets","AntiKt4TruthWZJets","PrimaryVertices","MET_Truth", "egammaTruthParticles","CaloCalTopoClusters"]
HIGG1D1SlimmingHelper.ExtraVariables = ["Muons.quality.EnergyLoss.energyLossType",
"GSFTrackParticles.parameterY.parameterZ.vx.vy",
"InDetTrackParticles.vx.vy",
"AntiKt4EMTopoJets.JetEMScaleMomentum_pt.JetEMScaleMomentum_eta.JetEMScaleMomentum_phi.JetEMScaleMomentum_m.Jvt.JvtJvfcorr.JvtRpt",
"CombinedMuonTrackParticles.z0.vz.definingParametersCovMatrix",
"BTagging_AntiKt4EMTopo.MV1_discriminant",
"ExtrapolatedMuonTrackParticles.z0.vz.definingParametersCovMatrix",
"MuonTruthParticles.truthOrigin.truthType",
"Electrons.maxEcell_time.maxEcell_energy.maxEcell_gain.maxEcell_onlId.maxEcell_x.maxEcell_y.maxEcell_z",
"Photons.maxEcell_time.maxEcell_energy.maxEcell_gain.maxEcell_onlId.maxEcell_x.maxEcell_y.maxEcell_z"]
HIGG1D1SlimmingHelper.ExtraVariables.extend( getGainDecorations(HIGG1D1_GainDecoratorTool) )
#addMETOutputs(HIGG1D1SlimmingHelper)
HIGG1D1SlimmingHelper.IncludeEGammaTriggerContent = True
HIGG1D1SlimmingHelper.AppendContentToStream(HIGG1D1Stream)