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)
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)
skimmingTools.append(EXOT1SkimmingTool) from DerivationFrameworkTools.DerivationFrameworkToolsConf import DerivationFramework__FilterCombinationOR EXOT1ORSkimmingTool = DerivationFramework__FilterCombinationOR( name="EXOT1ORSkimmingTool", FilterList=skimmingTools) ToolSvc += EXOT1ORSkimmingTool printfunc(EXOT1SkimmingTool) #======================================= # CREATE THE DERIVATION KERNEL ALGORITHM #======================================= from DerivationFrameworkCore.DerivationFrameworkCoreConf import DerivationFramework__DerivationKernel DerivationFrameworkJob += exot1Seq exot1Seq += CfgMgr.DerivationFramework__DerivationKernel( "EXOT1Kernel_skim", SkimmingTools=[EXOT1ORSkimmingTool]) exot1Seq += CfgMgr.DerivationFramework__DerivationKernel( "EXOT1Kernel", ThinningTools=thinningTools) #==================================================================== # Add the containers to the output stream - slimming done here #==================================================================== # Currently using same list as EXOT1 from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper from DerivationFrameworkExotics.EXOT1ContentList import * EXOT1SlimmingHelper = SlimmingHelper("EXOT1SlimmingHelper") EXOT1SlimmingHelper.StaticContent = EXOT1Content EXOT1SlimmingHelper.SmartCollections = EXOT1SmartContent EXOT1SlimmingHelper.AllVariables = EXOT1AllVariablesContent EXOT1SlimmingHelper.AppendContentToStream(EXOT1Stream)
#======================================= # 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)
name="JETM7Kernel", SkimmingTools=[JETM7SkimmingTool], ThinningTools=thinningTools) #==================================================================== # 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"])
"AntiKt10UFOCSSKSoftDropBeta100Zcut10Jets", "AntiKt10UFOCSSKBottomUpSoftDropBeta100Zcut5Jets", "AntiKt10UFOCSSKRecursiveSoftDropBeta100Zcut5NinfJets", "AntiKtVR30Rmax4Rmin02TrackJets_BTagging201810", "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" ]:
#==================================================================== # CONTENT LIST #==================================================================== # About SmartCollections and AllVariables: # If you want to use CP tools on a container, you MUST include that container as a SmartCollection. # You are then free to include it as AllVariables in addition, to pick up variables not needed by the CP tools but needed by your analysis. from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper SUSY1SlimmingHelper = SlimmingHelper("SUSY1SlimmingHelper") SUSY1SlimmingHelper.SmartCollections = [ "Electrons", "Photons", "MET_Reference_AntiKt4EMTopo", "Muons", "TauJets", "BTagging_AntiKt4EMTopo", "InDetTrackParticles", "PrimaryVertices", "BTagging_AntiKt4Track" ] SUSY1SlimmingHelper.AllVariables = [ "TruthParticles", "TruthEvents", "TruthVertices", "MET_Truth", "AntiKt4PV0TrackJets", "MET_Track" ] SUSY1SlimmingHelper.ExtraVariables = [ "BTagging_AntiKt4EMTopo.MV1_discriminant.MV1c_discriminant", "Muons.etcone30.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", # TODO: .DFCommonJets_Jvt", "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.TruthLabelDeltaR_B.TruthLabelDeltaR_C.TruthLabelDeltaR_T.TruthLabelID.ConeTruthLabelID.PartonTruthLabelID", ] if globalflags.DataSource() == 'geant4': SUSY1SlimmingHelper.StaticContent += [
EGAM2SlimmingHelper = SlimmingHelper("EGAM2SlimmingHelper") from DerivationFrameworkEGamma.EGAM2ExtraContent import * EGAM2SlimmingHelper.SmartCollections = [ "Electrons", "Photons", "Muons", "TauJets", "MET_Reference_AntiKt4EMTopo", "AntiKt4EMTopoJets", "BTagging_AntiKt4EMTopo", "InDetTrackParticles", "PrimaryVertices" ] # muons, tau, MET, b-tagging could be switched off if not needed and use too much space # Add egamma trigger objects EGAM2SlimmingHelper.IncludeEGammaTriggerContent = True # Extra variables EGAM2SlimmingHelper.ExtraVariables = ExtraContentAll EGAM2SlimmingHelper.AllVariables = ExtraContainersElectrons EGAM2SlimmingHelper.AllVariables += ExtraContainersTrigger if globalflags.DataSource() != 'geant4': EGAM2SlimmingHelper.AllVariables += ExtraContainersTriggerDataOnly if globalflags.DataSource() == 'geant4': EGAM2SlimmingHelper.ExtraVariables += ExtraContentAllTruth EGAM2SlimmingHelper.AllVariables += ExtraContainersTruth for tool in EGAM2_ClusterEnergyPerLayerDecorators: EGAM2SlimmingHelper.ExtraVariables.extend( getClusterEnergyPerLayerDecorations(tool)) # This line must come after we have finished configuring EGAM2SlimmingHelper EGAM2SlimmingHelper.AppendContentToStream(EGAM2Stream)
#if jetFlags.useTruth: # includePreTools=True # DerivationFrameworkJob += addTrimmedJets("AntiKt", 1.0, "Truth", rclus=0.2, ptfrac=0.05, includePreTools=True) #if includePreTools==False: # DerivationFrameworkJob += addTrimmedJets("AntiKt", 1.0, "LCTopo", rclus=0.2, ptfrac=0.05, includePreTools=True) #else: # DerivationFrameworkJob += addTrimmedJets("AntiKt", 1.0, "LCTopo", rclus=0.2, ptfrac=0.05) #==================================================================== # 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
#==================================================================== # Add the containers to the output stream - slimming done here #==================================================================== from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper JETM3SlimmingHelper = SlimmingHelper("JETM3SlimmingHelper") JETM3SlimmingHelper.SmartCollections = [ "Electrons", "Photons", "Muons", "TauJets", "InDetTrackParticles", "PrimaryVertices", "MET_Reference_AntiKt4EMTopo", "MET_Reference_AntiKt4LCTopo", "MET_Reference_AntiKt4EMPFlow", "AntiKt4EMTopoJets", "AntiKt4LCTopoJets", "AntiKt4EMPFlowJets" ] JETM3SlimmingHelper.AllVariables = [ "BTagging_AntiKt4LCTopo", "BTagging_AntiKt4EMTopo", "CaloCalTopoClusters", "MuonTruthParticles", "egammaTruthParticles", "TruthParticles", "TruthEvents", "TruthVertices", "MuonSegments", "JetETMissChargedParticleFlowObjects", "JetETMissNeutralParticleFlowObjects" ] JETM3SlimmingHelper.ExtraVariables = [ "Muons.energyLossType.EnergyLoss.ParamEnergyLoss.MeasEnergyLoss.EnergyLossSigma.MeasEnergyLossSigma.ParamEnergyLossSigmaPlus.ParamEnergyLossSigmaMinus" ] for truthc in [ "TruthMuons", "TruthElectrons", "TruthPhotons", "TruthTaus", # "TruthNeutrinos" ]: JETM3SlimmingHelper.StaticContent.append("xAOD::TruthParticleContainer#" + truthc)
#edmlist(set()) #==================================================================== # SET UP STREAM #==================================================================== # SKIMMING # The base name (DAOD_TRIG4 here) must match the string in # DerivationFrameworkProdFlags (in DerivationFrameworkCore) streamName = derivationFlags.WriteDAOD_TRIG4Stream.StreamName fileName = buildFileName( derivationFlags.WriteDAOD_TRIG4Stream ) TRIG4Stream = MSMgr.NewPoolRootStream( streamName, fileName ) # Only events that pass the filters listed below are written out. # Name must match that of the kernel above # AcceptAlgs = logical OR of filters # RequireAlgs = logical AND of filters TRIG4Stream.AcceptAlgs(["TRIG4Kernel"]) # Thinning from DerivationFrameworkTrigger.TrigThinningHelper import ThinningHelper TRIG4ThinningHelper = ThinningHelper("TRIG4ThinningHelper",edmlist,'slimming') TRIG4ThinningHelper.AppendToStream(TRIG4Stream) #==================================================================== # CONTENT LIST #==================================================================== from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper TRIG4SlimmingHelper = SlimmingHelper("TRIG4SlimmingHelper") ##Smart Slimming TRIG4SlimmingHelper.SmartCollections = [] TRIG4SlimmingHelper.AllVariables = edmlist TRIG4SlimmingHelper.AppendContentToStream(TRIG4Stream)
"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)
# container variables. Thus BTagging_AntiKt4EMTopo is needed in SmartCollections as well as AllVariables FTAG4SlimmingHelper.SmartCollections = ["Electrons","Muons", "MET_Reference_AntiKt4EMTopo", "AntiKt4EMTopoJets", "BTagging_AntiKt4EMTopo"] FTAG4SlimmingHelper.AllVariables = ["AntiKt3PV0TrackJets", "AntiKt2PV0TrackJets", "AntiKt4TruthJets", "BTagging_AntiKt4EMTopo", "BTagging_AntiKt2Track", "BTagging_AntiKt3Track", "BTagging_AntiKt4EMTopoJFVtx", "BTagging_AntiKt2TrackJFVtx", "BTagging_AntiKt3TrackJFVtx", "BTagging_AntiKt4EMTopoSecVtx", "BTagging_AntiKt2TrackSecVtx", "BTagging_AntiKt3TrackSecVtx", "TruthVertices", "TruthParticles", "TruthEvents", "MET_Truth", "MET_TruthRegions", "InDetTrackParticles", "PrimaryVertices" ] from DerivationFrameworkCore.AntiKt4EMTopoJetsCPContent import AntiKt4EMTopoJetsCPContent FTAG4SlimmingHelper.ExtraVariables.append(AntiKt4EMTopoJetsCPContent[1].replace("AntiKt4EMTopoJetsAux","AntiKt10LCTopoJets")) FTAG4SlimmingHelper.IncludeMuonTriggerContent = True
#======================================= from DerivationFrameworkCore.DerivationFrameworkCoreConf import DerivationFramework__DerivationKernel DerivationFrameworkJob += CfgMgr.DerivationFramework__DerivationKernel( name = "JETM4Kernel", SkimmingTools = [JETM4SkimmingTool], ThinningTools = thinningTools) #==================================================================== # Add the containers to the output stream - slimming done here #==================================================================== from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper JETM4SlimmingHelper = SlimmingHelper("JETM4SlimmingHelper") JETM4SlimmingHelper.SmartCollections = ["Electrons", "Photons", "Muons", "TauJets", "InDetTrackParticles", "PrimaryVertices"] JETM4SlimmingHelper.AllVariables = ["BTagging_AntiKt4LCTopo", "BTagging_AntiKt4EMTopo", "CaloCalTopoClusters", "MuonTruthParticles", "egammaTruthParticles", "TruthParticles", "TruthEvents", "TruthVertices", "MuonSegments" ] #JETM4SlimmingHelper.ExtraVariables = [] # Trigger content JETM4SlimmingHelper.IncludeEGammaTriggerContent = True # Add the jet containers to the stream addJetOutputs(JETM4SlimmingHelper,["SmallR"]) # Add the MET containers to the stream addMETOutputs(JETM4SlimmingHelper,["Diagnostic","AntiKt4LCTopo","AntiKt4EMPFlow","Track"]) JETM4SlimmingHelper.AppendContentToStream(JETM4Stream)
"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" ) STDM5Stream.AddItem( "xAOD::TruthParticleAuxContainer#TruthElectronsAux." )
#==================================================================== streamName = derivationFlags.WriteDAOD_JETM9Stream.StreamName fileName = buildFileName( derivationFlags.WriteDAOD_JETM9Stream ) JETM9Stream = MSMgr.NewPoolRootStream( streamName, fileName ) JETM9Stream.AcceptAlgs(["JETM9Kernel"]) # for thinning from AthenaServices.Configurables import ThinningSvc, createThinningSvc augStream = MSMgr.GetStream( streamName ) evtStream = augStream.GetEventStream() svcMgr += createThinningSvc( svcName="JETM9ThinningSvc", outStreams=[evtStream] ) #==================================================================== # 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", "TruthVertices", "MuonSegments" ] #JETM9SlimmingHelper.ExtraVariables = [] # 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#*")
"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"] MUON5SlimmingHelper.ExtraVariables += ["Electrons.truthOrigin.truthType.truthParticleLink.bkgTruthType.bkgTruthOrigin." "bkgTruthParticleLink.bkgMotherPdgId.deltaPhi1", "MuonTruthParticles.truthOrigin.truthType.truthParticleLink", "InDetTrackParticles.truthOrigin.truthType.truthParticleLink", "CombinedMuonTrackParticles.truthOrigin.truthType.truthParticleLink", "TauJets.IsTruthMatched.truthParticleLink.truthJetLink"] MUON5SlimmingHelper.StaticContent = ["xAOD::TruthParticleContainer#TruthMuons", "xAOD::TruthParticleAuxContainer#TruthMuonsAux.", "xAOD::TruthParticleContainer#TruthElectrons", "xAOD::TruthParticleAuxContainer#TruthElectronsAux.",
#==================================================================== # CONTENT LIST #==================================================================== # This might be the kind of set-up one would have for a muon based analysis from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper SUSY5SlimmingHelper = SlimmingHelper("SUSY5SlimmingHelper") 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)
"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)
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)
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)
# container variables. Thus BTagging_AntiKt4EMTopo is needed in SmartCollections as well as AllVariables FTAG3SlimmingHelper.SmartCollections = [ "Electrons", "Muons", "InDetTrackParticles", "PrimaryVertices", "AntiKt4EMTopoJets", "AntiKt4EMPFlowJets", "MET_Reference_AntiKt4EMTopo", ] FTAG3SlimmingHelper.AllVariables = [ "AntiKt10TruthWZJets", "BTagging_AntiKt2Track", "BTagging_AntiKt2TrackJFVtx", "BTagging_AntiKt10TruthWZ", "BTagging_AntiKt10TruthWZJFVtx", "TruthEvents", "TruthVertices", "TruthParticles", "TruthHFWithDecayParticles", "TruthHFWithDecayVertices", "TruthMuons", "MuonTruthParticles", "MET_Truth", "MET_TruthRegions", "HLT_xAOD__TrackParticleContainer_MuonEFInfo_CombTrackParticles", "HLT_xAOD__TrackParticleContainer_MuonEFInfo_CombTrackParticles_FullScan", "HLT_xAOD__MuonContainer_MuonEFInfo" ] FTAG3SlimmingHelper.ExtraVariables += [ "BTagging_AntiKt2TrackSecVtx.-vxTrackAtVertex", "BTagging_AntiKt10TruthSecVtx.-vxTrackAtVertex" ] if BTaggingFlags.Do2019Retraining: FTAG3SlimmingHelper.SmartCollections += \ ["AntiKt4EMTopoJets_BTagging201810"] FTAG3SlimmingHelper.AllVariables += \
# container variables. Thus BTagging_AntiKt4EMTopo is needed in SmartCollections as well as AllVariables FTAG4SlimmingHelper.SmartCollections = [ "Electrons", "Muons", "InDetTrackParticles", "AntiKt4EMTopoJets", "AntiKt4EMPFlowJets", "MET_Reference_AntiKt4EMPFlow", "MET_Reference_AntiKt4EMTopo" ] FTAG4SlimmingHelper.AllVariables = [ "BTagging_AntiKtVR30Rmax4Rmin02Track_201810", "BTagging_AntiKtVR30Rmax4Rmin02Track_201810JFVtx", "BTagging_AntiKtVR30Rmax4Rmin02Track_201903", "BTagging_AntiKtVR30Rmax4Rmin02Track_201903JFVtx", "BTagging_AntiKt10LCTopoTrimmedPtFrac5SmallR20ExCoM2Sub", "BTagging_AntiKt10LCTopoTrimmedPtFrac5SmallR20ExCoM2SubJFVtx", "BTagging_AntiKt2Track", "BTagging_AntiKt2TrackJFVtx", "TruthVertices", "TruthEvents", "TruthHFWithDecayParticles", "TruthHFWithDecayVertices", "MET_Truth", "MET_TruthRegions", "TruthParticles", ] FTAG4SlimmingHelper.ExtraVariables += [ AntiKt4EMTopoJetsCPContent[1].replace("AntiKt4EMTopoJetsAux", "AntiKt10LCTopoJets"), "InDetTrackParticles.truthMatchProbability.x.y.z.vx.vy.vz", "InDetTrackParticles.numberOfInnermostPixelLayerSplitHits.numberOfNextToInnermostPixelLayerSplitHits.numberOfNextToInnermostPixelLayerSharedHits", "InDetTrackParticles.numberOfPixelSplitHits.numberOfInnermostPixelLayerSharedHits.numberOfContribPixelLayers.hitPattern.radiusOfFirstHit",
JETM7SlimmingHelper = SlimmingHelper("JETM7SlimmingHelper") JETM7SlimmingHelper.SmartCollections = [ "Electrons", "Photons", "Muons", "TauJets", "InDetTrackParticles", "PrimaryVertices", "MET_Reference_AntiKt4EMTopo", "MET_Reference_AntiKt4LCTopo", "MET_Reference_AntiKt4EMPFlow", "AntiKt4EMTopoJets", "AntiKt4LCTopoJets", "AntiKt4EMPFlowJets", "AntiKt4EMPFlowJets_BTagging201810", "AntiKt4EMPFlowJets_BTagging201903", "AntiKt4EMTopoJets_BTagging201810", "BTagging_AntiKt4EMPFlow_201810", "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",
"BTagging_AntiKt4EMTopo"] FTAG5SlimmingHelper.AllVariables = ["AntiKt3PV0TrackJets", "AntiKt2PV0TrackJets", "AntiKt4PV0TrackJets", "AntiKt4TruthJets", "BTagging_AntiKt4EMTopo", "BTagging_AntiKt2Track", "BTagging_AntiKt3Track", "BTagging_AntiKt4EMTopoJFVtx", "BTagging_AntiKt2TrackJFVtx", "BTagging_AntiKt3TrackJFVtx", "BTagging_AntiKt4EMTopoSecVtx", "BTagging_AntiKt2TrackSecVtx", "BTagging_AntiKt3TrackSecVtx", "TruthVertices", "TruthParticles", "TruthEvents", "MET_Truth", "MET_TruthRegions", "InDetTrackParticles", "PrimaryVertices", "AntiKtVR50Rmax4Rmin0TrackJets", "BTagging_AntiKtVR50Rmax4Rmin0Track", "AntiKt10LCTopoJets", "AntiKt10LCTopoTrimmedPtFrac5SmallR20Jets", "AntiKt10TrackJets", "CaloCalTopoClusters", ] from DerivationFrameworkJetEtMiss.AntiKt4EMTopoJetsCPContent import AntiKt4EMTopoJetsCPContent
ThinningTools=thinningTools, ) #==================================================================== # CONTENT LIST #==================================================================== # This might be the kind of set-up one would have for a muon based analysis from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper 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
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)
# 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 HIGG5D2SlimmingHelper.IncludeJetTriggerContent = True
#======================================= from DerivationFrameworkCore.DerivationFrameworkCoreConf import DerivationFramework__DerivationKernel DerivationFrameworkJob += exot16Seq exot16Seq += CfgMgr.DerivationFramework__DerivationKernel( "EXOT16Kernel_skim", SkimmingTools=[EXOT16StringSkimmingTool]) exot16Seq += CfgMgr.DerivationFramework__DerivationKernel( "EXOT16Kernel", ThinningTools=thinningTools) #==================================================================== # Add the containers to the output stream - slimming done here #==================================================================== # Currently using same list as EXOT16 from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper from DerivationFrameworkExotics.EXOT16ContentList import * EXOT16SlimmingHelper = SlimmingHelper("EXOT16SlimmingHelper") EXOT16SlimmingHelper.AppendToDictionary = EXOT16Content EXOT16SlimmingHelper.SmartCollections = EXOT16SmartCollections EXOT16SlimmingHelper.AllVariables = EXOT16AllVariables EXOT16SlimmingHelper.ExtraVariables = EXOT16ExtraVariables addMETOutputs(EXOT16SlimmingHelper, ["MET_Track"], ["AntiKt4EMTopo"]) EXOT16SlimmingHelper.IncludeEGammaTriggerContent = True EXOT16SlimmingHelper.IncludeMuonTriggerContent = True EXOT16SlimmingHelper.AppendContentToStream(EXOT16Stream)
if is_MC: HIGG6D1SlimmingHelper.StaticContent = [ "xAOD::TruthParticleContainer#TruthMuons", "xAOD::TruthParticleAuxContainer#TruthMuonsAux.", "xAOD::TruthParticleContainer#TruthElectrons", "xAOD::TruthParticleAuxContainer#TruthElectronsAux.", "xAOD::TruthParticleContainer#TruthPhotons", "xAOD::TruthParticleAuxContainer#TruthPhotonsAux.", "xAOD::TruthParticleContainer#TruthNeutrinos", "xAOD::TruthParticleAuxContainer#TruthNeutrinosAux.", "xAOD::TruthParticleContainer#TruthTaus", "xAOD::TruthParticleAuxContainer#TruthTausAux." ] HIGG6D1SlimmingHelper.AllVariables = [ "TruthParticles", "TruthEvents", "MET_Truth", "METMap_Truth", "TruthVertices", "AntiKt4TruthJets" ] HIGG6D1SlimmingHelper.ExtraVariables += [ "AntiKt4LCTopoJets.PartonTruthLabelID.TruthLabelDeltaR_B.TruthLabelDeltaR_C.TruthLabelDeltaR_T" ] HIGG6D1SlimmingHelper.ExtraVariables += [ "AntiKt4EMTopoJets.PartonTruthLabelID.TruthLabelDeltaR_B.TruthLabelDeltaR_C.TruthLabelDeltaR_T" ] HIGG6D1SlimmingHelper.ExtraVariables += [ "Electrons.truthOrigin.truthType.truthParticleLink", "MuonTruthParticles.truthOrigin.truthType" ] HIGG6D1SlimmingHelper.ExtraVariables += [ "TauJets.IsTruthMatched.truthParticleLink" ]
"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)
"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", ] EXOT8SlimmingHelper.StaticContent = [ "xAOD::JetContainer#AntiKt10LCTopoTrimmedPtFrac5SmallR20Jets", "xAOD::JetAuxContainer#AntiKt10LCTopoTrimmedPtFrac5SmallR20JetsAux.", ] if globalflags.DataSource() == "geant4": EXOT8SlimmingHelper.StaticContent += [ "xAOD::JetContainer#AntiKt10TruthTrimmedPtFrac5SmallR20Jets", "xAOD::JetAuxContainer#AntiKt10TruthTrimmedPtFrac5SmallR20JetsAux.",
DerivationFrameworkJob += exot0Seq exot0Seq += CfgMgr.DerivationFramework__DerivationKernel("EXOT0Kernel_skim", SkimmingTools = [EXOT0SkimmingTool]) exot0Seq += CfgMgr.DerivationFramework__DerivationKernel("EXOT0Kernel", ThinningTools = thinningTools) #==================================================================== # SET UP STREAM #==================================================================== streamName = derivationFlags.WriteDAOD_EXOT0Stream.StreamName fileName = buildFileName( derivationFlags.WriteDAOD_EXOT0Stream ) EXOT0Stream = MSMgr.NewPoolRootStream( streamName, fileName ) EXOT0Stream.AcceptAlgs(["EXOT0Kernel"]) # Thinning from AthenaServices.Configurables import ThinningSvc, createThinningSvc augStream = MSMgr.GetStream( streamName ) evtStream = augStream.GetEventStream() svcMgr += createThinningSvc( svcName="EXOT0ThinningSvc", outStreams=[evtStream] ) #==================================================================== # Add the containers to the output stream - slimming done here #==================================================================== from DerivationFrameworkCore.SlimmingHelper import SlimmingHelper from DerivationFrameworkExotics.EXOT0ContentList import * EXOT0SlimmingHelper = SlimmingHelper("EXOT0SlimmingHelper") EXOT0SlimmingHelper.StaticContent = EXOT0Content EXOT0SlimmingHelper.AllVariables = EXOT0AllVariables EXOT0SlimmingHelper.SmartCollections = EXOT0SmartCollections EXOT0SlimmingHelper.IncludeEGammaTriggerContent = True EXOT0SlimmingHelper.IncludeMuonTriggerContent = True EXOT0SlimmingHelper.AppendContentToStream(EXOT0Stream)