def Config(self, stream, configKey): ### smart sliming from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper SlHelper = SlimmingHelper(configKey + "SlimmingHelper") SlHelper.SmartCollections = self.smSlContainer[configKey] useSmartSlimmingIfSupported(self.checkContainers.get(configKey, []), self.allVarContainer[configKey], self.Items[configKey], SlHelper.NamesAndTypes) SlHelper.AllVariables = self.allVarContainer[configKey] # SlHelper.StaticContent = self.Items[configKey] SlHelper.ExtraVariables = self.extraVariables.get(configKey, []) SlHelper.IncludeMuonTriggerContent = self.UseTriggerContent.get( configKey, True) SlHelper.AppendContentToStream(stream) ### other items for item in self.Items[configKey]: stream.AddItem(item)
"Photons", "Muons", "TauJets", "MET_Reference_AntiKt4EMTopo", "AntiKt4LCTopoJets", "AntiKt4EMTopoJets", "BTagging_AntiKt4LCTopo", "BTagging_AntiKt4EMTopo", "InDetTrackParticles", "PrimaryVertices" ] STDM5SlimmingHelper.IncludeEGammaTriggerContent = True STDM5SlimmingHelper.IncludeMuonTriggerContent = True STDM5SlimmingHelper.ExtraVariables = ExtraContentAll STDM5SlimmingHelper.AllVariables = ExtraContainersAll if globalflags.DataSource()=='geant4': STDM5SlimmingHelper.ExtraVariables += ExtraContentAllTruth STDM5SlimmingHelper.AllVariables += ExtraContainersTruth STDM5SlimmingHelper.AppendContentToStream(STDM5Stream) STDM5Stream.AddItem("xAOD::EventShape#*") STDM5Stream.AddItem("xAOD::EventShapeAuxInfo#*") #STDM5Stream.AddItem("xAOD::TrigNavigation#*") #STDM5Stream.AddItem("xAOD::TrigNavigationAuxInfo#*") if globalflags.DataSource()=='geant4': STDM5Stream.AddItem( "xAOD::TruthParticleContainer#TruthMuons" ) STDM5Stream.AddItem( "xAOD::TruthParticleAuxContainer#TruthMuonsAux." ) STDM5Stream.AddItem( "xAOD::TruthParticleContainer#TruthElectrons" )
#==================================================================== # Add the containers to the output stream - slimming done here #==================================================================== from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper #from DerivationFrameworkHiggs.HIGG5D2ContentList import HIGG5D2Content from DerivationFrameworkHiggs.HIGG5D2ExtraContent import * HIGG5D2SlimmingHelper = SlimmingHelper("HIGG5D2SlimmingHelper") HIGG5D2SlimmingHelper.SmartCollections = [ "Electrons", "Photons", "Muons", "TauJets", "MET_Reference_AntiKt4EMTopo", "MET_Reference_AntiKt4LCTopo", "AntiKt4EMTopoJets", "AntiKt4LCTopoJets", "BTagging_AntiKt4EMTopo", "BTagging_AntiKt4LCTopo", "InDetTrackParticles", "PrimaryVertices" ] HIGG5D2SlimmingHelper.ExtraVariables = ExtraContent HIGG5D2SlimmingHelper.AllVariables = ExtraContainers if globalflags.DataSource() == 'geant4': HIGG5D2SlimmingHelper.ExtraVariables += ExtraContentTruth HIGG5D2SlimmingHelper.AllVariables += ExtraContainersTruth # Add the jet containers to the stream addJetOutputs(HIGG5D2SlimmingHelper, ["HIGG5D2Jets"]) # Add the MET containers to the stream addMETOutputs(HIGG5D2SlimmingHelper, ["AntiKt4LCTopo", "Track"]) HIGG5D2SlimmingHelper.IncludeMuonTriggerContent = True HIGG5D2SlimmingHelper.IncludeEGammaTriggerContent = True #HIGG5D2SlimmingHelper.IncludeBPhysTriggerContent = True #HIGG5D2SlimmingHelper.IncludeJetTauEtMissTriggerContent = True HIGG5D2SlimmingHelper.IncludeEtMissTriggerContent = True
#======================================= # CREATE THE DERIVATION KERNEL ALGORITHM #======================================= from DerivationFrameworkCore.DerivationFrameworkCoreConf import DerivationFramework__DerivationKernel DerivationFrameworkJob += exot7Seq exot7Seq += CfgMgr.DerivationFramework__DerivationKernel( "EXOT7Kernel_skim", AugmentationTools=[EXOT7DeltaRTool], SkimmingTools=[EXOT7StringSkimmingTool]) exot7Seq += CfgMgr.DerivationFramework__DerivationKernel( "EXOT7Kernel", ThinningTools=thinningTools) #==================================================================== # Add the containers to the output stream - slimming done here #==================================================================== from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper from DerivationFrameworkExotics.EXOT7ContentList import * EXOT7SlimmingHelper = SlimmingHelper("EXOT7SlimmingHelper") EXOT7SlimmingHelper.StaticContent = EXOT7Content EXOT7SlimmingHelper.SmartCollections = EXOT7SmartCollections EXOT7SlimmingHelper.ExtraVariables = EXOT7ExtraVariables EXOT7SlimmingHelper.AllVariables = EXOT7AllVariables #EXOT7SlimmingHelper.IncludeJetTauEtMissTriggerContent = True EXOT7SlimmingHelper.IncludeJetTriggerContent = True EXOT7SlimmingHelper.IncludeBJetTriggerContent = True EXOT7SlimmingHelper.AppendContentToStream(EXOT7Stream)
#==================================================================== # Add the containers to the output stream - slimming done here #==================================================================== from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper JETM7SlimmingHelper = SlimmingHelper("JETM7SlimmingHelper") JETM7SlimmingHelper.SmartCollections = [ "Electrons", "Photons", "Muons", "TauJets", "InDetTrackParticles", "PrimaryVertices" ] JETM7SlimmingHelper.AllVariables = [ "BTagging_AntiKt4LCTopo", "BTagging_AntiKt4EMTopo", "CaloCalTopoClusters", "MuonTruthParticles", "egammaTruthParticles", "TruthParticles", "TruthEvents", "TruthVertices", "MuonSegments" ] JETM7SlimmingHelper.ExtraVariables = [ "Muons.energyLossType.EnergyLoss.ParamEnergyLoss.MeasEnergyLoss.EnergyLossSigma.MeasEnergyLossSigma.ParamEnergyLossSigmaPlus.ParamEnergyLossSigmaMinus" ] # Trigger content JETM7SlimmingHelper.IncludeMuonTriggerContent = True JETM7SlimmingHelper.IncludeEGammaTriggerContent = True # Add the jet containers to the stream addJetOutputs(JETM7SlimmingHelper, ["SmallR"]) # Add the MET containers to the stream addMETOutputs(JETM7SlimmingHelper, ["Diagnostic", "AntiKt4LCTopo", "AntiKt4EMPFlow", "Track"]) JETM7SlimmingHelper.AppendContentToStream(JETM7Stream)
"AntiKtVR30Rmax4Rmin02TrackJets_BTagging201903", "AntiKt4EMPFlowJets_BTagging201810", "AntiKt4EMPFlowJets_BTagging201903", "AntiKt4EMTopoJets_BTagging201810", "BTagging_AntiKtVR30Rmax4Rmin02Track_201810", "BTagging_AntiKtVR30Rmax4Rmin02Track_201903", "BTagging_AntiKt4EMPFlow_201810", "BTagging_AntiKt4EMPFlow_201903", "BTagging_AntiKt4EMTopo_201810", ] JETM6SlimmingHelper.AllVariables = [ "TruthEvents", "Kt4EMTopoOriginEventShape","Kt4EMPFlowEventShape", ] JETM6SlimmingHelper.ExtraVariables = ['CaloCalTopoClusters.calE.calEta.calM.calPhi.CENTER_MAG'] JETM6SlimmingHelper.ExtraVariables += ['Electrons.'+NewTrigVars["Electrons"],'Muons.'+NewTrigVars["Muons"],'Photons.'+NewTrigVars["Photons"]] JETM6SlimmingHelper.ExtraVariables += [ 'HLT_xAOD__JetContainer_a4tcemsubjesFS.ActiveArea.ActiveArea4vec_eta.ActiveArea4vec_m.ActiveArea4vec_phi.ActiveArea4vec_pt.AverageLArQF.BchCorrCell.CentroidR.DetectorEta.EMFrac.EnergyPerSampling.FracSamplingMax.FracSamplingMaxIndex.HECFrac.HECQuality.JetConstitScaleMomentum_eta.JetConstitScaleMomentum_m.JetConstitScaleMomentum_phi.JetConstitScaleMomentum_pt.JetEtaJESScaleMomentum_eta.JetEtaJESScaleMomentum_m.JetEtaJESScaleMomentum_phi.JetEtaJESScaleMomentum_pt.JetPileupScaleMomentum_eta.JetPileupScaleMomentum_m.JetPileupScaleMomentum_phi.JetPileupScaleMomentum_pt.LArQuality.N90Constituents.NegativeE.Timing.eta.kinematics.m.phi.pt', 'HLT_xAOD__JetContainer_a4tcemsubjesISFS.ActiveArea.ActiveArea4vec_eta.ActiveArea4vec_m.ActiveArea4vec_phi.ActiveArea4vec_pt.AverageLArQF.BchCorrCell.CentroidR.DetectorEta.EMFrac.EnergyPerSampling.FracSamplingMax.FracSamplingMaxIndex.HECFrac.HECQuality.JetConstitScaleMomentum_eta.JetConstitScaleMomentum_m.JetConstitScaleMomentum_phi.JetConstitScaleMomentum_pt.JetEtaJESScaleMomentum_eta.JetEtaJESScaleMomentum_m.JetEtaJESScaleMomentum_phi.JetEtaJESScaleMomentum_pt.JetPileupScaleMomentum_eta.JetPileupScaleMomentum_m.JetPileupScaleMomentum_phi.JetPileupScaleMomentum_pt.LArQuality.N90Constituents.NegativeE.Timing.eta.kinematics.m.phi.pt', 'HLT_xAOD__JetContainer_a10tclcwsubjesFS.ActiveArea.ActiveArea4vec_eta.ActiveArea4vec_m.ActiveArea4vec_phi.ActiveArea4vec_pt.AverageLArQF.BchCorrCell.CentroidR.DetectorEta.EMFrac.EnergyPerSampling.FracSamplingMax.FracSamplingMaxIndex.HECFrac.HECQuality.JetConstitScaleMomentum_eta.JetConstitScaleMomentum_m.JetConstitScaleMomentum_phi.JetConstitScaleMomentum_pt.JetEMScaleMomentum_eta.JetEMScaleMomentum_m.JetEMScaleMomentum_phi.JetEMScaleMomentum_pt.JetEtaJESScaleMomentum_eta.JetEtaJESScaleMomentum_m.JetEtaJESScaleMomentum_phi.JetEtaJESScaleMomentum_pt.JetPileupScaleMomentum_eta.JetPileupScaleMomentum_m.JetPileupScaleMomentum_phi.JetPileupScaleMomentum_pt.LArQuality.N90Constituents.NegativeE.Timing.eta.kinematics.m.phi.pt', ] for truthc in [ "TruthTopQuark", "TruthBosons", "TruthHF" ]: JETM6SlimmingHelper.StaticContent.append("xAOD::TruthParticleContainer#"+truthc+"WithDecayParticles") JETM6SlimmingHelper.StaticContent.append("xAOD::TruthParticleAuxContainer#"+truthc+"WithDecayParticlesAux.") JETM6SlimmingHelper.StaticContent.append("xAOD::TruthVertexContainer#"+truthc+"WithDecayVertices")
PHYSLITESlimmingHelper.SmartCollections = [ "InDetTrackParticles", "PrimaryVertices", ] # Variables to save, per object type PHYSLITESlimmingHelper.ExtraVariables = [ "AnalysisElectrons.trackParticleLinks.pt.eta.phi.m.charge.author.DFCommonElectronsLHVeryLoose.DFCommonElectronsLHLoose.DFCommonElectronsLHLooseBL.DFCommonElectronsLHMedium.DFCommonElectronsLHTight.DFCommonElectronsLHVeryLooseIsEMValue.DFCommonElectronsLHLooseIsEMValue.DFCommonElectronsLHLooseBLIsEMValue.DFCommonElectronsLHMediumIsEMValue.DFCommonElectronsLHTightIsEMValue.DFCommonElectronsECIDS.DFCommonElectronsECIDSResult.ptvarcone20.ptvarcone40.topoetcone20.topoetcone20ptCorrection.ptcone20_TightTTVA_pt500.ptcone20_TightTTVA_pt1000.ptvarcone20_TightTTVA_pt1000.ptvarcone30_TightTTVA_pt500.ptvarcone30_TightTTVA_pt1000.caloClusterLinks.ambiguityLink.truthParticleLink.truthOrigin.truthType.truthPdgId.firstEgMotherTruthType.firstEgMotherTruthOrigin.firstEgMotherTruthParticleLink.firstEgMotherPdgId.ambiguityType.OQ", "AnalysisPhotons.pt.eta.phi.m.author.OQ.DFCommonPhotonsIsEMLoose.DFCommonPhotonsIsEMTight.DFCommonPhotonsIsEMTightIsEMValue.DFCommonPhotonsIsEMTightPtIncl.DFCommonPhotonsIsEMTightPtInclIsEMValue.DFCommonPhotonsCleaning.DFCommonPhotonsCleaningNoTime.ptcone20.topoetcone20.topoetcone40.topoetcone20ptCorrection.topoetcone40ptCorrection.caloClusterLinks.vertexLinks.ambiguityLink.truthParticleLink.truthOrigin.truthType", "GSFTrackParticles.chiSquared.phi.d0.theta.qOverP.definingParametersCovMatrix.z0.vz.charge.vertexLink", "CaloCalTopoClusters.rawE.rawEta.rawPhi.rawM.calE.calEta.calPhi.calM.e_sampl", "egammaClusters.calE.calEta.calPhi.e_sampl.eta_sampl.ETACALOFRAME.PHICALOFRAME.ETA2CALOFRAME.PHI2CALOFRAME.constituentClusterLinks", "AnalysisMuons.pt.eta.phi.truthType.truthOrigin.author.muonType.quality.inDetTrackParticleLink.muonSpectrometerTrackParticleLink.combinedTrackParticleLink.InnerDetectorPt.MuonSpectrometerPt.DFCommonGoodMuon.ptcone20.ptcone30.ptcone40.ptvarcone20.ptvarcone30.ptvarcone40.topoetcone20.topoetcone30.topoetcone40.truthParticleLink.charge.extrapolatedMuonSpectrometerTrackParticleLink.allAuthors.ptcone20_TightTTVA_pt1000.ptcone20_TightTTVA_pt500.ptvarcone30_TightTTVA_pt1000.ptvarcone30_TightTTVA_pt500.numberOfPrecisionLayers.combinedTrackOutBoundsPrecisionHits.numberOfPrecisionLayers.numberOfPrecisionHoleLayers.numberOfGoodPrecisionLayers.innerSmallHits.innerLargeHits.middleSmallHits.middleLargeHits.outerSmallHits.outerLargeHits.extendedSmallHits.extendedLargeHits.extendedSmallHoles.isSmallGoodSectors.cscUnspoiledEtaHits.EnergyLoss.energyLossType.momentumBalanceSignificance.scatteringCurvatureSignificance.scatteringNeighbourSignificance", "CombinedMuonTrackParticles.qOverP.d0.z0.vz.phi.theta.truthOrigin.truthType.definingParametersCovMatrix.numberOfPixelDeadSensors.numberOfPixelHits.numberOfPixelHoles.numberOfSCTDeadSensors.numberOfSCTHits.numberOfSCTHoles.numberOfTRTHits.numberOfTRTOutliers.chiSquared.numberDoF", "ExtrapolatedMuonTrackParticles.d0.z0.vz.definingParametersCovMatrix.truthOrigin.truthType.qOverP.theta.phi", "MuonSpectrometerTrackParticles.phi.d0.z0.vz.definingParametersCovMatrix.vertexLink.theta.qOverP.truthParticleLink", "AnalysisTauJets.pt.eta.phi.m.tauTrackLinks.jetLink.charge.isTauFlags.BDTJetScore.BDTEleScore.ptFinalCalib.etaFinalCalib.phiFinalCalib.mFinalCalib.electronLink.EleMatchLikelihoodScore.pt_combined.eta_combined.phi_combined.m_combined.BDTJetScoreSigTrans.BDTEleScoreSigTrans.PanTau_DecayMode.RNNJetScore.RNNJetScoreSigTrans.IsTruthMatched.truthOrigin.truthType.truthParticleLink.truthJetLink", "AnalysisJets.pt.eta.phi.m.JetConstitScaleMomentum_pt.JetConstitScaleMomentum_eta.JetConstitScaleMomentum_phi.JetConstitScaleMomentum_m.NumTrkPt500.SumPtTrkPt500.DetectorEta.Jvt.JVFCorr.JvtRpt.NumTrkPt1000.TrackWidthPt1000.GhostMuonSegmentCount.PartonTruthLabelID.ConeTruthLabelID.HadronConeExclExtendedTruthLabelID.HadronConeExclTruthLabelID.TrueFlavor.DFCommonJets_jetClean_LooseBad.DFCommonJets_jetClean_TightBad.Timing.btagging.btaggingLink.GhostTrack.DFCommonJets_fJvt.DFCommonJets_QGTagger_NTracks.DFCommonJets_QGTagger_TracksWidth.DFCommonJets_QGTagger_TracksC1.PSFrac", "BTagging_AntiKt4EMPFlow_201903.DL1r_pu.DL1rmu_pu.DL1r_pb.DL1rmu_pb.DL1r_pc.DL1rmu_pc", "TruthPrimaryVertices.t.x.y.z", "MET_Core_AnalysisMET.name.mpx.mpy.sumet.source", "METAssoc_AnalysisMET.", ] if DerivationFrameworkIsMonteCarlo: from DerivationFrameworkMCTruth.MCTruthCommon import addTruth3ContentToSlimmerTool addTruth3ContentToSlimmerTool(PHYSLITESlimmingHelper) # Extra trigger collections trigmatching_helper_notau.add_to_slimming(PHYSLITESlimmingHelper)
outputGroup="EXOT12") if globalflags.DataSource() == 'geant4': replaceBuggyAntiKt4TruthWZJets(exot12Seq) #exot12Seq += CfgMgr.DerivationFramework__DerivationKernel("EXOT12Kernel_jet", # SkimmingTools = [EXOT12SkimmingTool_jet]) exot12Seq += CfgMgr.DerivationFramework__DerivationKernel( "EXOT12Kernel", ThinningTools=thinningTools) #==================================================================== # Add the containers to the output stream - slimming done here #==================================================================== from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper from DerivationFrameworkExotics.EXOT12ContentList import * EXOT12SlimmingHelper = SlimmingHelper("EXOT12SlimmingHelper") EXOT12SlimmingHelper.SmartCollections = EXOT12SmartContent EXOT12SlimmingHelper.ExtraVariables = EXOT12ExtraVariablesContent EXOT12SlimmingHelper.AllVariables = EXOT12AllVariablesContent EXOT12SlimmingHelper.StaticContent = EXOT12Content if globalflags.DataSource() == 'geant4': EXOT12SlimmingHelper.ExtraVariables += EXOT12ExtraVariablesContentTruth EXOT12SlimmingHelper.AllVariables += EXOT12AllVariablesContentTruth EXOT12SlimmingHelper.StaticContent += EXOT12ContentTruth EXOT12SlimmingHelper.IncludeEGammaTriggerContent = True EXOT12SlimmingHelper.IncludeMuonTriggerContent = True EXOT12SlimmingHelper.AppendContentToStream(EXOT12Stream)
"MET_Reference_AntiKt4EMPFlow", "AntiKt4EMTopoJets","AntiKt4LCTopoJets","AntiKt4EMPFlowJets", "AntiKt4EMPFlowJets_BTagging201810", "AntiKt4EMPFlowJets_BTagging201903", "AntiKt4EMTopoJets_BTagging201810", "BTagging_AntiKt4EMPFlow_201810", "BTagging_AntiKt4EMPFlow_201903", "BTagging_AntiKt4EMTopo_201810", ] JETM2SlimmingHelper.AllVariables = ["MuonTruthParticles", "egammaTruthParticles", "TruthParticles", "TruthEvents", "TruthVertices", "MuonSegments", "Kt4EMTopoOriginEventShape","Kt4LCTopoOriginEventShape","Kt4EMPFlowEventShape", ] JETM2SlimmingHelper.ExtraVariables = ["Electrons."+NewTrigVars["Electrons"], "Muons.energyLossType.EnergyLoss.ParamEnergyLoss.MeasEnergyLoss.EnergyLossSigma.MeasEnergyLossSigma.ParamEnergyLossSigmaPlus.ParamEnergyLossSigmaMinus."+NewTrigVars["Muons"], "TauJets.IsTruthMatched.truthParticleLink.truthJetLink"] for truthc in [ "TruthMuons", "TruthElectrons", "TruthPhotons", "TruthTaus", # "TruthNeutrinos" ]: JETM2SlimmingHelper.StaticContent.append("xAOD::TruthParticleContainer#"+truthc) JETM2SlimmingHelper.StaticContent.append("xAOD::TruthParticleAuxContainer#"+truthc+"Aux.") # Trigger content JETM2SlimmingHelper.IncludeMuonTriggerContent = True JETM2SlimmingHelper.IncludeEGammaTriggerContent = True
# CREATE THE DERIVATION KERNEL ALGORITHM #======================================= from DerivationFrameworkCore.DerivationFrameworkCoreConf import DerivationFramework__DerivationKernel DerivationFrameworkJob += CfgMgr.DerivationFramework__DerivationKernel( "HIGG3D2Kernel", ThinningTools=thinningTools) #==================================================================== # SET UP STREAM #==================================================================== streamName = derivationFlags.WriteDAOD_HIGG3D2Stream.StreamName fileName = buildFileName(derivationFlags.WriteDAOD_HIGG3D2Stream) HIGG3D2Stream = MSMgr.NewPoolRootStream(streamName, fileName) HIGG3D2Stream.AcceptAlgs(["HIGG3D2Kernel"]) augStream = MSMgr.GetStream(streamName) evtStream = augStream.GetEventStream() #==================================================================== # Add the containers to the output stream - slimming done here #==================================================================== from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper HIGG3D2SlimmingHelper = SlimmingHelper("HIGG3D2SlimmingHelper") from DerivationFrameworkHiggs.HIGG3D2ExtraContent import * HIGG3D2SlimmingHelper.ExtraVariables = HIGG3D2ExtraVariables if globalflags.DataSource() == 'geant4': HIGG3D2SlimmingHelper.AllVariables = HIGG3D2ExtraTruthContainers HIGG3D2SlimmingHelper.AppendContentToStream(HIGG3D2Stream)
EXOT5SlimmingHelper = SlimmingHelper('EXOT5SlimmingHelper') EXOT5SlimmingHelper.SmartCollections = [ 'Electrons', 'Muons', 'Photons', 'TauJets', 'InDetTrackParticles', 'PrimaryVertices', 'MET_Reference_AntiKt4EMTopo', 'BTagging_AntiKt4EMTopo', ] EXOT5SlimmingHelper.ExtraVariables = [ 'Photons.author.Loose.Tight', 'Electrons.author.Medium.Tight.Loose.charge', 'Muons.ptcone20.ptcone30.ptcone40.etcone20.etcone30.etcone40', 'CombinedMuonTrackParticles.d0.z0.vz.definingParametersCovMatrix', # SUSYTools 'ExtrapolatedMuonTrackParticles.d0.z0.vz.definingParametersCovMatrix', # SUSYTools 'TauJets.TruthCharge.TruthProng.IsTruthMatched.TruthPtVis.truthOrigin.truthType.truthParticleLink.truthJetLink', ] EXOT5SlimmingHelper.AllVariables = [ 'AntiKt4EMTopoJets', 'AntiKt4TruthJets', 'MET_Truth', 'MET_Track', 'MET_LocHadTopo', 'TruthEvents', 'TruthParticles', 'MuonSegments', ] if DerivationFrameworkIsMonteCarlo: EXOT5SlimmingHelper.StaticContent = [
svcMgr += createThinningSvc(svcName="STDM2ThinningSvc", outStreams=[evtStream]) #==================================================================== # Add the containers to the output stream - slimming done here #==================================================================== from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper from DerivationFrameworkSM.STDMExtraContent import * STDM2SlimmingHelper = SlimmingHelper("STDM2SlimmingHelper") STDM2SlimmingHelper.SmartCollections = [ "Electrons", "Photons", "AntiKt4EMTopoJets", "AntiKt4LCTopoJets", "BTagging_AntiKt4LCTopo", "BTagging_AntiKt4EMTopo", "InDetTrackParticles", "PrimaryVertices" ] STDM2SlimmingHelper.IncludeEGammaTriggerContent = True STDM2SlimmingHelper.ExtraVariables = ExtraContentElectrons + ExtraContentPhotons + ExtraContentBtag + ExtraContentVtx + ExtraContentTrk + ExtraContentJets + [ "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" ] STDM2SlimmingHelper.AllVariables = ExtraContainersJets #+ExtraContainers6Jets #do not exist for now if globalflags.DataSource() == 'geant4': STDM2SlimmingHelper.ExtraVariables += ExtraElectronsTruth + ExtraPhotonsTruth STDM2SlimmingHelper.AllVariables += ExtraContainersTruth + ExtraContainersTruthPhotons #["AntiKt6TruthJets", "BTagging_AntiKt6TruthWZ","AntiKt6TruthWZJets"] do not exist for now STDM2SlimmingHelper.AppendContentToStream(STDM2Stream) STDM2Stream.AddItem("xAOD::EventShape#*") STDM2Stream.AddItem("xAOD::EventShapeAuxInfo#*")
def setup(TOPQname, stream): DFisMC = (globalflags.DataSource() == 'geant4') TOPQSlimmingHelper = SlimmingHelper(TOPQname + "SlimmingHelper") #===================================================== # ADD BTagging_AntiKt4EMPFlow COLLECTION TO DICTIONARY #===================================================== TOPQSlimmingHelper.AppendToDictionary = { "BTagging_AntiKt4EMPFlow": "xAOD::BTaggingContainer", "BTagging_AntiKt4EMPFlowAux": "xAOD::BTaggingAuxContainer" } #================================ # SMART SLIMMING #================================ TOPQSlimmingHelper.SmartCollections = [] TOPQSlimmingHelper.SmartCollections += TOPQSmartSlimmingCollections print("TOPQSlimmingHelper.SmartCollections: ", TOPQSlimmingHelper.SmartCollections) #================================================================= # EXTRA VARIABLES FROM DerivationFrameworkTop.TOPQCommonExtraContent #================================================================= TOPQSlimmingHelper.ExtraVariables = [] TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesAntiKt4EMTopoJets TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesAntiKt4EMPFlowJets TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesBTagging_AntiKt4EMPFlow TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesBTagging_AntiKt4EMTopo TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesPhotons TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesElectrons TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesMuons TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesTaus TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesTrackJets TOPQSlimmingHelper.ExtraVariables += Config.GetExtraPromptVariablesForDxAOD( ) # TOPQSlimmingHelper.ExtraVariables += ["CaloCalTopoClusters.calE.calEta.calPhi.calM.rawM.rawE.rawEta.rawPhi.e_sampl.eta_sampl.etaCalo.phiCalo"] TOPQSlimmingHelper.ExtraVariables += [ "CaloCalTopoClusters.calPt.calEta.calPhi.calM.calE.CENTER_MAG" ] if DFisMC: TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesPhotonsTruth TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesElectronsTruth TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesMuonsTruth TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesTausTruth TOPQSlimmingHelper.ExtraVariables += TOPQExtraVarsBTag_HLT_Container TOPQSlimmingHelper.ExtraVariables += TOPQExtraVarsJet_EF_Container TOPQSlimmingHelper.ExtraVariables += TOPQExtraVarsJet_Split_Container print("TOPQSlimmingHelper.ExtraVariables: ", TOPQSlimmingHelper.ExtraVariables) #================================ # EXTRA COLLECTIONS - user added #================================ TOPQSlimmingHelper.AllVariables = [] TOPQSlimmingHelper.AllVariables += TOPQExtraContainersStandard TOPQSlimmingHelper.AllVariables += TOPQExtraContainersTrigger if DFisMC: TOPQSlimmingHelper.AllVariables += TOPQExtraContainersTruth print("TOPQSlimmingHelper.AllVariables: ", TOPQSlimmingHelper.AllVariables) #================================ # CREATED ON-THE-FLY COLLECTIONS #================================ TOPQSlimmingHelper.StaticContent = [] TOPQSlimmingHelper.StaticContent += TOPQStaticContent if DFisMC: TOPQSlimmingHelper.StaticContent += TOPQStaticContentTruth print("TOPQSlimmingHelper.StaticContent: ", TOPQSlimmingHelper.StaticContent) #================================ # TRIGGER CONTENT #================================ TOPQSlimmingHelper.IncludeMuonTriggerContent = True TOPQSlimmingHelper.IncludeEGammaTriggerContent = True TOPQSlimmingHelper.IncludeJetTriggerContent = True TOPQSlimmingHelper.IncludeTauTriggerContent = True TOPQSlimmingHelper.IncludeEtMissTriggerContent = True #TOPQSlimmingHelper.IncludeJetTauEtMissTriggerContent = True #TOPQSlimmingHelper.IncludeBJetTriggerContent = True #TOPQSlimmingHelper.IncludeBPhysTriggerContent = True #TOPQSlimmingHelper.IncludeMinBiasTriggerContent = True #==================================== # ADD AntiKT4EMPFlowJets MET OUTPUT #==================================== # needs to be after where AllVaraibles[] and StaticContent[] are created # adds METAssoc_XXXX, MET_Core_XXXX, and MET_Reference_XXXX # uses DerivationFrameworkJetEtMiss.METCommon.py addMETOutputs(TOPQSlimmingHelper, ["AntiKt4EMPFlow"]) #==================================== # ADD CUSTOM JET OUTPUT (new rel 21 method) #==================================== addJetOutputs( TOPQSlimmingHelper, ["SmallR", "LargeR", TOPQname], [], # smart list [ "AntiKt3PV0TrackJets", "AntiKt4PV0TrackJets", "AntiKt4LCTopoJets", "AntiKt10LCTopoJets", "AntiKt10TruthWZJets", ] # veto list ) #================================ # THIS NEEDS TO BE THE LAST LINE #================================ TOPQSlimmingHelper.AppendContentToStream(stream)
"AntiKt4TruthJets", "AntiKt4EMTopoJets", "AntiKt4LCTopoJets" ] #addDefaultTrimmedJets(jetm9Seq,"JETM9") if jetFlags.useTruth: replaceBuggyAntiKt4TruthWZJets(jetm9Seq, "JETM9") for radius in [0.2, 0.3, 0.5, 0.6, 0.7, 0.8]: if jetFlags.useTruth: addRscanJets("AntiKt", radius, "Truth", jetm9Seq, "JETM9") addRscanJets("AntiKt", radius, "TruthWZ", jetm9Seq, "JETM9") addRscanJets("AntiKt", radius, "LCTopo", jetm9Seq, "JETM9") #==================================================================== # Add the containers to the output stream - slimming done here #==================================================================== from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper JETM9SlimmingHelper = SlimmingHelper("JETM9SlimmingHelper") JETM9SlimmingHelper.SmartCollections = [ "AntiKt4EMTopoJets", "AntiKt4LCTopoJets", "PrimaryVertices" ] JETM9SlimmingHelper.AllVariables = ["TruthEvents", "MuonSegments"] JETM9SlimmingHelper.ExtraVariables = ["TruthVertices.z"] # Trigger content JETM9SlimmingHelper.IncludeJetTriggerContent = True # Add the jet containers to the stream addJetOutputs(JETM9SlimmingHelper, ["JETM9"]) JETM9SlimmingHelper.AppendContentToStream(JETM9Stream) JETM9Stream.RemoveItem("xAOD::TrigNavigation#*") JETM9Stream.RemoveItem("xAOD::TrigNavigationAuxInfo#*")
"EXOT3Kernel", # ThinningTools = [EXOT3TPThinningTool,EXOT3MuonTPThinningTool,EXOT3ElectronTPThinningTool, EXOT3PhotonTPThinningTool] ThinningTools=thinningTools) #==================================================================== # Add the containers to the output stream - slimming done here #==================================================================== from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper from DerivationFrameworkExotics.EXOT3ContentList import * EXOT3SlimmingHelper = SlimmingHelper("EXOT3SlimmingHelper") # Containers to be smart slimmed, see https://svnweb.cern.ch/trac/atlasoff/browser/PhysicsAnalysis # /DerivationFramework/DerivationFrameworkExamples/trunk/share/SlimmingExample.py#L38 EXOT3SlimmingHelper.SmartCollections = EXOT3SmartContent EXOT3SlimmingHelper.ExtraVariables = EXOT3ExtraVariables # Keep all variables for containers which we don't want to smart slim and were # not created in the derivation EXOT3SlimmingHelper.AllVariables = EXOT3AllVariablesContent # Add jet collections created by derivation job EXOT3SlimmingHelper.StaticContent = EXOT3StaticContent # (Dont) Add jet triger content #EXOT3SlimmingHelper.IncludeJetTauEtMissTriggerContent = True EXOT3SlimmingHelper.IncludeJetTriggerContent = True EXOT3SlimmingHelper.IncludeEGammaTriggerContent = True EXOT3SlimmingHelper.AppendContentToStream(EXOT3Stream)
# Add the containers to the output stream - slimming done here #==================================================================== from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper HIGG3D1SlimmingHelper = SlimmingHelper("HIGG3D1SlimmingHelper") from DerivationFrameworkHiggs.HIGG3D1ExtraContent import * HIGG3D1SlimmingHelper.SmartCollections = [ "Electrons", "Muons", "Photons", "MET_Reference_AntiKt4LCTopo", "MET_Reference_AntiKt4EMTopo", "AntiKt4EMTopoJets", "AntiKt4LCTopoJets", "BTagging_AntiKt4LCTopo", "BTagging_AntiKt4EMTopo", "BTagging_AntiKt4Track", "BTagging_AntiKt2Track", "InDetTrackParticles", "PrimaryVertices" ] HIGG3D1SlimmingHelper.ExtraVariables = HIGG3D1ExtraVariables HIGG3D1SlimmingHelper.AllVariables = HIGG3D1ExtraContainers if globalflags.DataSource() == 'geant4': HIGG3D1SlimmingHelper.AllVariables += HIGG3D1ExtraTruthContainers HIGG3D1SlimmingHelper.ExtraVariables += HIGG3D1ExtraTruthVariables # Add Trigger content HIGG3D1SlimmingHelper.IncludeMuonTriggerContent = True HIGG3D1SlimmingHelper.IncludeEGammaTriggerContent = True # Add MET to output stream #addMETOutputs(HIGG3D1SlimmingHelper) HIGG3D1SlimmingHelper.AppendContentToStream(HIGG3D1Stream)
#==================================================================== # CONTENT LIST #==================================================================== # This might be the kind of set-up one would have for a muon based analysis from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper SUSY4SlimmingHelper = SlimmingHelper("SUSY4SlimmingHelper") SUSY4SlimmingHelper.SmartCollections = ["Electrons", "Photons", "Muons", "TauJets", "AntiKt4EMTopoJets", "MET_Reference_AntiKt4EMTopo", "BTagging_AntiKt4EMTopo", "InDetTrackParticles", "PrimaryVertices", "AntiKt4EMPFlowJets"] SUSY4SlimmingHelper.AllVariables = ["TruthParticles", "TruthEvents", "TruthVertices", "MET_Truth", "InDetTrackParticles", "METAssoc_AntiKt4EMPFlow", "MET_Core_AntiKt4EMPFlow", "MET_Reference_AntiKt4EMPFlow"] SUSY4SlimmingHelper.ExtraVariables = ["BTagging_AntiKt4EMTopo.MV1_discriminant.MV1c_discriminant", "Electrons.Loose.Medium.Tight", "Muons.ptcone30.ptcone20.charge.quality.InnerDetectorPt.MuonSpectrometerPt.CaloLRLikelihood.CaloMuonIDTag", "Photons.author.Loose.Tight", "AntiKt4EMTopoJets.JvtJvfcorr.JvtRpt.Jvt.TruthLabelID.BchCorrCell.NumTrkPt1000.NumTrkPt500.SumPtTrkPt500.SumPtTrkPt1000.TrackWidthPt1000.TrackWidthPt500.DFCommonJets_Calib_pt.DFCommonJets_Calib_eta.DFCommonJets_Calib_phi", "AntiKt4EMPFlowJets.JvtJvfcorr.JvtRpt.Jvt.TruthLabelID.BchCorrCell.NumTrkPt1000.NumTrkPt500.SumPtTrkPt500.SumPtTrkPt1000.TrackWidthPt1000.TrackWidthPt500", "GSFTrackParticles.z0.d0.vz.definingParametersCovMatrix", "CombinedMuonTrackParticles.d0.z0.vz.definingParametersCovMatrix.truthOrigin.truthType", "ExtrapolatedMuonTrackParticles.d0.z0.vz.definingParametersCovMatrix.truthOrigin.truthType", "TauJets.TruthCharge.TruthProng.IsTruthMatched.TruthPtVis.truthOrigin.truthType.truthParticleLink.truthJetLink", "MuonTruthParticles.barcode.decayVtxLink.e.m.pdgId.prodVtxLink.px.py.pz.recoMuonLink.status.truthOrigin.truthType", "AntiKt4TruthJets.eta.m.phi.pt.TruthLabelID", "InDetTrackParticles.truthOrigin.truthType.truthMatchProbability.TrkIso*"] SUSY4SlimmingHelper.IncludeMuonTriggerContent = False SUSY4SlimmingHelper.IncludeEGammaTriggerContent = False SUSY4SlimmingHelper.IncludeBPhysTriggerContent = False SUSY4SlimmingHelper.IncludeJetTriggerContent = False SUSY4SlimmingHelper.IncludeTauTriggerContent = False SUSY4SlimmingHelper.IncludeEtMissTriggerContent = False SUSY4SlimmingHelper.IncludeBJetTriggerContent = False
SUSY5SlimmingHelper.SmartCollections = [ "PhotonCollection", "ElectronCollection", "Muons", "TauRecContainer", "AntiKt4LCTopoJets", "MET_RefFinal", "BTagging_AntiKt4LCTopo", "InDetTrackParticles", "PrimaryVertices" ] SUSY5SlimmingHelper.AllVariables = [ "AntiKt4TruthJets", "BTagging_AntiKt4Truth", "MuonTruthParticle", "TruthParticle", "TruthEvent", "TruthVertex", "MET_Truth", "BTagging_AntiKt4LCTopo" ] SUSY5SlimmingHelper.ExtraVariables = [ "BTagging_AntiKt4LCTopo.MV1_discriminant", "PhotonCollection.author.Loose.Tight", "ElectronCollection.author.Loose.Medium.Tight.truthOrigin.truthType.charge.ptcone20.etcone30", "Muons.ptcone30.ptcone20.etcone30.charge.quality.InnerDetectorPt.MuonSpectrometerPt.CaloLRLikelihood.CaloMuonIDTag", "AntiKt4LCTopoJets.ActiveArea4vec_pt.ActiveArea4vec_eta.ActiveArea4vec_phi.ActiveArea4vec_m.EnergyPerSampling.NumTrkPt1000.TrackWidthPt1000.GhostMuonSegment.NumTrkPt500", "GSFTrackParticles.z0.d0.vz.definingParametersCovMatrix", "CombinedMuonTrackParticles.d0.z0.vz.definingParametersCovMatrix.truthOrigin.truthType", "ExtrapolatedMuonTrackParticles.d0.z0.vz.definingParametersCovMatrix.truthOrigin.truthType", "TauRecContainer.TruthCharge.TruthProng.IsTruthMatched.TruthPtVis.truthOrigin.truthType" ] SUSY5SlimmingHelper.AppendContentToStream(SUSY5Stream) # Add the MET containers to the stream addMETOutputs(SUSY5Stream, ["SUSY"]) for item in OutputJets["LargeR"]: if "Trim" not in item: continue SUSY5Stream.AddItem("xAOD::JetContainer_v1#" + item) SUSY5Stream.AddItem("xAOD::JetAuxContainer_v1#" + item + "Aux.")
'SecVtx_ConvVtxContainer_Electrons' : 'xAOD::VertexContainer', 'SecVtx_ConvVtxContainer_ElectronsAux': 'xAOD::VertexAuxContainer', 'SecVtxContainer_Muons' : 'xAOD::VertexContainer', 'SecVtxContainer_MuonsAux' : 'xAOD::VertexAuxContainer', 'RefittedPriVtx' : 'xAOD::VertexContainer', 'RefittedPriVtxAux' : 'xAOD::VertexAuxContainer', } # Add extra variables to collections MUON5SlimmingHelper.ExtraVariables = ["Muons.clusterLink.allAuthors.charge.extrapolatedMuonSpectrometerTrackParticleLink." "scatteringCurvatureSignificance.scatteringNeighbourSignificance", "Electrons.author.charge.topoetcone30.ptvarcone30", "AntiKt4EMTopoJets.JetEMScaleMomentum_pt.JetEMScaleMomentum_eta.JetEMScaleMomentum_phi.JetEMScaleMomentum_m." "ConeTruthLabelID.PartonTruthLabelID.SumPtTrkPt1000.Jvt.JvtJvfcorr.JvtRpt.TileStatus", "GSFTrackParticles.z0.vz.definingParametersCovMatrix", "CombinedMuonTrackParticles.z0.vz.definingParametersCovMatrix", "ExtrapolatedMuonTrackParticles.z0.vz.definingParametersCovMatrix", "PrimaryVertices.x.y.z.trackParticleLinks.vertexType.neutralParticleLinks", "TauJets.pantau_CellBasedInput_isPanTauCandidate.pantau_CellBasedInput_DecayMode." "ptPanTauCellBased.etaPanTauCellBased.phiPanTauCellBased.mPanTauCellBased." "pantau_CellBasedInput_BDTValue_1p0n_vs_1p1n.pantau_CellBasedInput_BDTValue_1p1n_vs_1pXn." "pantau_CellBasedInput_BDTValue_3p0n_vs_3pXn.ele_match_lhscore.ele_olr_pass." "electronLink.ptDetectorAxis.etaDetectorAxis.phiDetectorAxis.mDetectorAxis", "TauNeutralParticleFlowObjects.pt.eta.phi.m.e.rapidity.bdtPi0Score", "TauChargedParticleFlowObjects.pt.eta.phi.m"] MUON5SlimmingHelper.ExtraVariables += LepTagConfig.GetExtraPromptVariablesForDxAOD(onlyBDT=False) MUON5SlimmingHelper.ExtraVariables += LepTagConfig.GetExtraImprovedPromptVariablesForDxAOD() MUON5SlimmingHelper.ExtraVariables += ElectronsCPDetailedContent MUON5SlimmingHelper.AllVariables = ["egammaClusters", "CaloCalTopoClusters", "MuonClusterCollection", "TopoClusterIsoCentralEventShape", "TopoClusterIsoForwardEventShape", "AntiKt4PV0TrackJets", "BTagging_AntiKt4Track", "InDetTrackParticles","GSFConversionVertices","GSFTrackParticles"] if is_MC: MUON5SlimmingHelper.AllVariables += ["TruthParticles", "TruthEvents", "TruthVertices", "AntiKt4TruthJets"]
SkimmingTools=skimmingTools, ThinningTools=thinningTools) #==================================================================== # Add the containers to the output stream - slimming done here #==================================================================== from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper HIGG3D3SlimmingHelper = SlimmingHelper("HIGG3D3SlimmingHelper") from DerivationFrameworkHiggs.HIGG3D3ExtraContent import * #HIGG3D3SlimmingHelper.UserContent = HIGG3D3Content HIGG3D3SlimmingHelper.SmartCollections = [ "Electrons", "Muons", "AntiKt4EMTopoJets", "InDetTrackParticles", "BTagging_AntiKt4EMTopo", "PrimaryVertices", "MET_Reference_AntiKt4EMTopo" ] HIGG3D3SlimmingHelper.ExtraVariables = HIGG3D3ExtraVariables HIGG3D3SlimmingHelper.AllVariables = HIGG3D3ExtraContainers if globalflags.DataSource() == 'geant4': HIGG3D3SlimmingHelper.AllVariables += HIGG3D3ExtraTruthContainers HIGG3D3SlimmingHelper.ExtraVariables += HIGG3D3ExtraTruthVariables # Add Trigger content HIGG3D3SlimmingHelper.IncludeMuonTriggerContent = True HIGG3D3SlimmingHelper.IncludeEGammaTriggerContent = True # Add MET to output stream #addMETOutputs(HIGG3D3SlimmingHelper) HIGG3D3SlimmingHelper.AppendContentToStream(HIGG3D3Stream)
"SUSY8GenFilt", SimBarcodeOffset=DerivationFrameworkSimBarcodeOffset) AugmentationTools.append(ToolSvc.SUSY8GenFilt) #============================================================================== # Augment after skim #============================================================================== SeqSUSY8 += CfgMgr.DerivationFramework__DerivationKernel( "SUSY8KernelAug", AugmentationTools=AugmentationTools, ThinningTools=thinningTools, ) #==================================================================== # CONTENT LIST #==================================================================== from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper from DerivationFrameworkSUSY.SUSY8ContentList import * SUSY8SlimmingHelper = SlimmingHelper("SUSY8SlimmingHelper") SUSY8SlimmingHelper.SmartCollections = SUSY8SmartContent SUSY8SlimmingHelper.ExtraVariables = SUSY8ExtraVariables SUSY8SlimmingHelper.AllVariables = SUSY8AllVariablesContent SUSY8SlimmingHelper.IncludeMuonTriggerContent = True SUSY8SlimmingHelper.IncludeJetTriggerContent = True SUSY8SlimmingHelper.IncludeEtMissTriggerContent = True SUSY8SlimmingHelper.AppendContentToStream(SUSY8Stream)
SUSY5SlimmingHelper.SmartCollections = [ "Electrons", "Photons", "Muons", "TauJets", "AntiKt4EMTopoJets", "MET_Reference_AntiKt4EMTopo", "BTagging_AntiKt4EMTopo", "InDetTrackParticles", "PrimaryVertices" ] SUSY5SlimmingHelper.AllVariables = [ "TruthParticles", "TruthEvents", "TruthVertices", "MET_Truth" ] SUSY5SlimmingHelper.ExtraVariables = [ "BTagging_AntiKt4EMTopo.MV1_discriminant.MV1c_discriminant", "Muons.ptcone30.ptcone20.charge.quality.InnerDetectorPt.MuonSpectrometerPt.CaloLRLikelihood.CaloMuonIDTag", "Photons.author.Loose.Tight", "AntiKt4EMTopoJets.NumTrkPt1000.TrackWidthPt1000.NumTrkPt500.DFCommonJets_Calib_pt.DFCommonJets_Calib_eta.DFCommonJets_Calib_phi", "GSFTrackParticles.z0.d0.vz.definingParametersCovMatrix", "CombinedMuonTrackParticles.d0.z0.vz.definingParametersCovMatrix.truthOrigin.truthType", "ExtrapolatedMuonTrackParticles.d0.z0.vz.definingParametersCovMatrix.truthOrigin.truthType", "TauJets.IsTruthMatched.TruthPtVis.truthOrigin.truthType.truthParticleLink.truthJetLink", "MuonTruthParticles.barcode.decayVtxLink.e.m.pdgId.prodVtxLink.px.py.pz.recoMuonLink.status.truthOrigin.truthType", "AntiKt4TruthJets.eta.m.phi.pt.TruthLabelDeltaR_B.TruthLabelDeltaR_C.TruthLabelDeltaR_T.TruthLabelID.ConeTruthLabelID.PartonTruthLabelID", "Electrons.bkgTruthType.bkgTruthOrigin", "MET_Track", "AntiKt3PV0TrackJets.eta.m.phi.pt.btagging.btaggingLink", "BTagging_AntiKt3Track.MV2c20_discriminant" ] SUSY5SlimmingHelper.IncludeMuonTriggerContent = True SUSY5SlimmingHelper.IncludeEGammaTriggerContent = True SUSY5SlimmingHelper.IncludeBPhysTriggerContent = False SUSY5SlimmingHelper.IncludeJetTriggerContent = True SUSY5SlimmingHelper.IncludeTauTriggerContent = True SUSY5SlimmingHelper.IncludeEtMissTriggerContent = True SUSY5SlimmingHelper.IncludeBJetTriggerContent = True
from DerivationFrameworkHiggs.HIGG2D2ExtraContent import * from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper HIGG2D2SlimmingHelper = SlimmingHelper("HIGG2D2SlimmingHelper") HIGG2D2SlimmingHelper.SmartCollections = ["Electrons", "Photons", "Muons", "TauJets", "MET_Reference_AntiKt4EMTopo", "MET_Reference_AntiKt4LCTopo", "AntiKt4EMTopoJets", "AntiKt4LCTopoJets", "BTagging_AntiKt4EMTopo", "BTagging_AntiKt4LCTopo", "InDetTrackParticles", "PrimaryVertices"] HIGG2D2SlimmingHelper.ExtraVariables = HIGG2D2ExtraContent HIGG2D2SlimmingHelper.AllVariables = HIGG2D2ExtraContainers if globalflags.DataSource()=='geant4': HIGG2D2SlimmingHelper.ExtraVariables += HIGG2D2ExtraContentTruth HIGG2D2SlimmingHelper.AllVariables += HIGG2D2ExtraContainersTruth # Add MET_RefFinalFix addMETOutputs(HIGG2D2SlimmingHelper,["AntiKt4LCTopo","Track"]) HIGG2D2SlimmingHelper.IncludeMuonTriggerContent = True HIGG2D2SlimmingHelper.IncludeEGammaTriggerContent = True HIGG2D2SlimmingHelper.AppendContentToStream(HIGG2D2Stream)
"BTagging_AntiKt4EMPFlow_201903", "BTagging_AntiKt4EMTopo_201810" ] JETM7SlimmingHelper.AllVariables = [ # "CaloCalTopoClusters", "MuonTruthParticles", "egammaTruthParticles", "TruthParticles", "TruthEvents", "TruthVertices", "MuonSegments", "Kt4EMTopoOriginEventShape", "Kt4LCTopoOriginEventShape", "Kt4EMPFlowEventShape", ] JETM7SlimmingHelper.ExtraVariables = [ "Electrons." + NewTrigVars["Electrons"], "Muons.energyLossType.EnergyLoss.ParamEnergyLoss.MeasEnergyLoss.EnergyLossSigma.MeasEnergyLossSigma.ParamEnergyLossSigmaPlus.ParamEnergyLossSigmaMinus." + NewTrigVars["Muons"] ] for truthc in [ "TruthMuons", "TruthElectrons", "TruthPhotons", "TruthTaus", # "TruthNeutrinos" ]: JETM7SlimmingHelper.StaticContent.append("xAOD::TruthParticleContainer#" + truthc) JETM7SlimmingHelper.StaticContent.append( "xAOD::TruthParticleAuxContainer#" + truthc + "Aux.") # Trigger content
def setup(TOPQname, stream): DFisMC = (globalflags.DataSource() == 'geant4') TOPQSlimmingHelper = SlimmingHelper(TOPQname + "SlimmingHelper") #===================================================== # ADD BTagging_AntiKt4EMPFlow COLLECTION TO DICTIONARY #===================================================== TOPQSlimmingHelper.AppendToDictionary = { "BTagging_AntiKt4EMPFlow": "xAOD::BTaggingContainer", "BTagging_AntiKt4EMPFlowAux": "xAOD::BTaggingAuxContainer" } #================================ # SMART SLIMMING #================================ TOPQSlimmingHelper.SmartCollections = [] TOPQSlimmingHelper.SmartCollections += TOPQSmartSlimmingCollections print "TOPQSlimmingHelper.SmartCollections: ", TOPQSlimmingHelper.SmartCollections #================================================================= # EXTRA VARIABLES FROM DerivationFrameworkTop.TOPQCommonExtraContent #================================================================= TOPQSlimmingHelper.ExtraVariables = [] TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesAntiKt4EMTopoJets TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesAntiKt4EMPFlowJets TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesBTagging_AntiKt4EMPFlow TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesBTagging_AntiKt4EMTopo TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesPhotons TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesElectrons TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesMuons TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesTaus if DFisMC: TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesPhotonsTruth TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesElectronsTruth TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesMuonsTruth TOPQSlimmingHelper.ExtraVariables += TOPQExtraVariablesTausTruth print "TOPQSlimmingHelper.ExtraVariables: ", TOPQSlimmingHelper.ExtraVariables #================================ # EXTRA COLLECTIONS - user added #================================ TOPQSlimmingHelper.AllVariables = [] TOPQSlimmingHelper.AllVariables += TOPQExtraContainersStandard TOPQSlimmingHelper.AllVariables += TOPQExtraContainersTrigger if DFisMC: TOPQSlimmingHelper.AllVariables += TOPQExtraContainersTruth print "TOPQSlimmingHelper.AllVariables: ", TOPQSlimmingHelper.AllVariables #================================ # CREATED ON-THE-FLY COLLECTIONS #================================ TOPQSlimmingHelper.StaticContent = [] TOPQSlimmingHelper.StaticContent += TOPQStaticContent if DFisMC: TOPQSlimmingHelper.StaticContent += TOPQStaticContentTruth print "TOPQSlimmingHelper.StaticContent: ", TOPQSlimmingHelper.StaticContent #================================ # TRIGGER CONTENT #================================ TOPQSlimmingHelper.IncludeMuonTriggerContent = True TOPQSlimmingHelper.IncludeEGammaTriggerContent = True TOPQSlimmingHelper.IncludeJetTriggerContent = True TOPQSlimmingHelper.IncludeTauTriggerContent = True TOPQSlimmingHelper.IncludeEtMissTriggerContent = True #TOPQSlimmingHelper.IncludeJetTauEtMissTriggerContent = True #TOPQSlimmingHelper.IncludeBJetTriggerContent = True #TOPQSlimmingHelper.IncludeBPhysTriggerContent = True #TOPQSlimmingHelper.IncludeMinBiasTriggerContent = True #==================================== # ADD AntiKT4EMPFlowJets MET OUTPUT #==================================== # needs to be after where AllVaraibles[] and StaticContent[] are created # adds METAssoc_XXXX, MET_Core_XXXX, and MET_Reference_XXXX # uses DerivationFrameworkJetEtMiss.METCommon.py addMETOutputs(TOPQSlimmingHelper, ["AntiKt4EMPFlow"]) #================================ # THIS NEEDS TO BE THE LAST LINE #================================ TOPQSlimmingHelper.AppendContentToStream(stream)
SUSY2SlimmingHelper = SlimmingHelper("SUSY2SlimmingHelper") SUSY2SlimmingHelper.SmartCollections = [ "Electrons", "Photons", "Muons", "TauJets", "MET_Reference_AntiKt4EMTopo", "AntiKt4EMTopoJets", "BTagging_AntiKt4EMTopo", "InDetTrackParticles", "PrimaryVertices" ] SUSY2SlimmingHelper.AllVariables = [ "TruthParticles", "TruthEvents", "TruthVertices", "MET_Truth" ] SUSY2SlimmingHelper.ExtraVariables = [ "BTagging_AntiKt4EMTopo.MV1_discriminant.MV1c_discriminant", "Muons.ptcone30.ptcone20.charge.quality.InnerDetectorPt.MuonSpectrometerPt.CaloLRLikelihood.CaloMuonIDTag", "AntiKt4EMTopoJets.NumTrkPt1000.TrackWidthPt1000.NumTrkPt500", "GSFTrackParticles.z0.d0.vz.definingParametersCovMatrix.truthOrigin.truthType", "InDetTrackParticles.truthOrigin.truthType", "CombinedMuonTrackParticles.d0.z0.vz.definingParametersCovMatrix.truthOrigin.truthType", "ExtrapolatedMuonTrackParticles.d0.z0.vz.definingParametersCovMatrix.truthOrigin.truthType", "TauJets.TruthCharge.TruthProng.IsTruthMatched.TruthPtVis.truthOrigin.truthType.truthParticleLink.truthJetLink", "MuonTruthParticles.barcode.decayVtxLink.e.m.pdgId.prodVtxLink.px.py.pz.recoMuonLink.status.truthOrigin.truthType", "AntiKt4TruthJets.eta.m.phi.pt.TruthLabelDeltaR_B.TruthLabelDeltaR_C.TruthLabelDeltaR_T.TruthLabelID.ConeTruthLabelID.PartonTruthLabelID", "Electrons.bkgMotherPdgId.bkgTruthOrigin" ] SUSY2SlimmingHelper.IncludeMuonTriggerContent = True SUSY2SlimmingHelper.IncludeEGammaTriggerContent = True #SUSY2SlimmingHelper.IncludeBPhysTriggerContent = True #SUSY2SlimmingHelper.IncludeJetTauEtMissTriggerContent = True if globalflags.DataSource() == 'geant4': SUSY2SlimmingHelper.StaticContent += [ "xAOD::TruthParticleContainer#TruthTaus",
"xAOD::TruthParticleContainer#TruthTaus", "xAOD::TruthParticleAuxContainer#TruthTausAux.", "xAOD::TruthParticleContainer#TruthPhotons", "xAOD::TruthParticleAuxContainer#TruthPhotonsAux.", "xAOD::TruthParticleContainer#TruthNeutrinos", "xAOD::TruthParticleAuxContainer#TruthNeutrinosAux." ] TAUP1SlimmingHelper.IncludeMuonTriggerContent = True TAUP1SlimmingHelper.IncludeTauTriggerContent = True TAUP1SlimmingHelper.IncludeEGammaTriggerContent = True TAUP1SlimmingHelper.IncludeEtMissTriggerContent = True TAUP1SlimmingHelper.IncludeJetTriggerContent = True TAUP1SlimmingHelper.IncludeBJetTriggerContent = True TAUP1SlimmingHelper.ExtraVariables = ExtraContentTAUP1 TAUP1SlimmingHelper.AllVariables = ExtraContainersTAUP1 if globalflags.DataSource() == "geant4": TAUP1SlimmingHelper.ExtraVariables += ExtraContentTruthTAUP1 TAUP1SlimmingHelper.AllVariables += ExtraContainersTruthTAUP1 TAUP1SlimmingHelper.AppendContentToStream(TAUP1Stream) TAUP1Stream.AddItem("xAOD::EventShape#*") TAUP1Stream.AddItem("xAOD::EventShapeAuxInfo#*") # Add MET_RefFinalFix # JRC TEMPORARILY COMMENTED # addMETOutputs(TAUP1Stream)
exot6Seq += CfgMgr.DerivationFramework__DerivationKernel( "EXOT6Kernel", ThinningTools=[ EXOT6TPThinningTool, EXOT6MuonTPThinningTool, EXOT6ElectronTPThinningTool, EXOT6PhotonTPThinningTool, EXOT6PhotonCCThinningTool ]) #==================================================================== # Add the containers to the output stream - slimming done here #==================================================================== from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper from DerivationFrameworkExotics.EXOT6ContentList import EXOT6SmartContent, EXOT6AllVariablesContent, EXOT6ExtraVariables, EXOT6UnslimmedContent EXOT6SlimmingHelper = SlimmingHelper("EXOT6SlimmingHelper") # Containers to be smart slimmed, see https://svnweb.cern.ch/trac/atlasoff/browser/PhysicsAnalysis # /DerivationFramework/DerivationFrameworkExamples/trunk/share/SlimmingExample.py#L38 EXOT6SlimmingHelper.SmartCollections = EXOT6SmartContent # Keep all variables for containers which we don't want to smart slim and were # not created in the derivation EXOT6SlimmingHelper.AllVariables = EXOT6AllVariablesContent # Adding extra missing variables from the smart slimming EXOT6SlimmingHelper.ExtraVariables = EXOT6ExtraVariables EXOT6SlimmingHelper.StaticContent = EXOT6UnslimmedContent EXOT6SlimmingHelper.AppendContentToStream(EXOT6Stream)
#======================================= # CREATE THE DERIVATION KERNEL ALGORITHM #======================================= from DerivationFrameworkCore.DerivationFrameworkCoreConf import DerivationFramework__DerivationKernel DerivationFrameworkJob += exot10Seq exot10Seq += CfgMgr.DerivationFramework__DerivationKernel( "EXOT10Kernel_skim", SkimmingTools=[EXOT10SkimmingTool]) exot10Seq += CfgMgr.DerivationFramework__DerivationKernel( "EXOT10Kernel", AugmentationTools=[EXOT10_MaxCellDecoratorTool], ThinningTools=[EXOT10PhotonCCThinningTool, EXOT10ElectronCCThinningTool]) #==================================================================== # Add the containers to the output stream - slimming done here #==================================================================== from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper from DerivationFrameworkExotics.EXOT10ContentList import * EXOT10SlimmingHelper = SlimmingHelper("EXOT10SlimmingHelper") EXOT10SlimmingHelper.StaticContent = EXOT10Content EXOT10SlimmingHelper.SmartCollections = EXOT10SmartContent EXOT10SlimmingHelper.AllVariables = EXOT10AllVariablesContent EXOT10SlimmingHelper.UserContent = EXOT10Content EXOT10SlimmingHelper.ExtraVariables = EXOT10ExtraVariables EXOT10SlimmingHelper.IncludeEGammaTriggerContent = True addMETOutputs(EXOT10SlimmingHelper) EXOT10SlimmingHelper.AppendContentToStream(EXOT10Stream)
EXOT8SlimmingHelper.SmartCollections = [ "AntiKt4EMTopoJets", "AntiKt4LCTopoJets", "BTagging_AntiKt4LCTopo", "BTagging_AntiKt4EMTopo", "BTagging_AntiKt2Track", "BTagging_AntiKt3Track", "BTagging_AntiKt4Track", "InDetTrackParticles", "PrimaryVertices", "Electrons", "Muons", "MET_Reference_AntiKt4LCTopo", "MET_Reference_AntiKt4EMTopo", ] EXOT8SlimmingHelper.ExtraVariables = ["Electrons.charge", "Muons.charge"] EXOT8SlimmingHelper.AllVariables = [ "TruthParticles", "TruthEvents", "TruthVertices", "AntiKt10LCTopoJets", "AntiKt2PV0TrackJets", "AntiKt3PV0TrackJets", "AntiKt4PV0TrackJets", #"BTagging_AntiKt4TruthWZ", #removed in new tags of MC15 "AntiKt4TruthJets", "AntiKt10TruthJets", "CombinedMuonTrackParticles", "ExtrapolatedMuonTrackParticles", ]
from DerivationFrameworkCore.DerivationFrameworkCoreConf import DerivationFramework__DerivationKernel DerivationFrameworkJob += CfgMgr.DerivationFramework__DerivationKernel("JETM2Kernel", SkimmingTools = [JETM2SkimmingTool], ThinningTools = thinningTools) #==================================================================== # Add the containers to the output stream - slimming done here #==================================================================== from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper JETM2SlimmingHelper = SlimmingHelper("JETM2SlimmingHelper") JETM2SlimmingHelper.SmartCollections = ["Electrons", "Photons", "Muons", "TauJets", "InDetTrackParticles", "PrimaryVertices"] JETM2SlimmingHelper.AllVariables = ["BTagging_AntiKt4LCTopo", "BTagging_AntiKt4EMTopo", "CaloCalTopoClusters", "MuonTruthParticles", "egammaTruthParticles", "TruthParticles", "TruthEvents", "TruthVertices", "MuonSegments" ] JETM2SlimmingHelper.ExtraVariables = ["Muons.energyLossType.EnergyLoss.ParamEnergyLoss.MeasEnergyLoss.EnergyLossSigma.MeasEnergyLossSigma.ParamEnergyLossSigmaPlus.ParamEnergyLossSigmaMinus"] # Trigger content JETM2SlimmingHelper.IncludeMuonTriggerContent = True JETM2SlimmingHelper.IncludeEGammaTriggerContent = True # Add the jet containers to the stream addJetOutputs(JETM2SlimmingHelper,["SmallR"]) # Add the MET containers to the stream addMETOutputs(JETM2SlimmingHelper,["Diagnostic","AntiKt4LCTopo","AntiKt4EMPFlow","Track"]) JETM2SlimmingHelper.AppendContentToStream(JETM2Stream)