Source code for skim.WGs.systematics

#!/usr/bin/env python3

##########################################################################
# basf2 (Belle II Analysis Software Framework)                           #
# Author: The Belle II Collaboration                                     #
#                                                                        #
# See git log for contributors and copyright holders.                    #
# This file is licensed under LGPL-3.0, see LICENSE.md.                  #
##########################################################################

""" Skim list building functions for systematics studies """

__authors__ = [
    "Sam Cunliffe",
    "Torben Ferber",
    "Ilya Komarov",
    "Yuji Kato"
]

import basf2 as b2
import modularAnalysis as ma
import vertex
from skim import BaseSkim, CombinedSkim, fancy_skim_header
from stdCharged import stdE, stdK, stdMu, stdPi, stdPr
from stdPhotons import stdPhotons
from stdPi0s import stdPi0s
from stdV0s import stdKshorts, stdLambdas
from variables import variables as vm

# TODO: Add liaison name and email address
__liaison__ = ""
__liaison_leptonID__ = "Marcel Hohmann"
_VALIDATION_SAMPLE = "mdst16.root"


[docs] @fancy_skim_header class SystematicsDstar(BaseSkim): """ Primarily used for hadron and lepton ID studies. Lists in this skim are those defined in `PiKFromDstarList`. """ __authors__ = ["Sam Cunliffe", "Torben Ferber", "Ilya Komarov", "Yuji Kato", "Racha Cheaib"] __description__ = "" __contact__ = __liaison__ __category__ = "systematics" ApplyHLTHadronCut = True
[docs] def load_standard_lists(self, path): stdK("all", path=path) stdPi("all", path=path)
TestSampleProcess = "ccbar"
[docs] def build_lists(self, path): return self.PiKFromDstarList(path)
[docs] def PiKFromDstarList(self, path): """Build PiKFromDstarList lists for systematics skims.""" D0Cuts = "1.75 < M < 2.0" DstarCuts = "massDifference(0)<0.16 and useCMSFrame(p) > 1.5" ma.cutAndCopyList("K-:syst", "K-:all", "dr<2 and abs(dz)<4", path=path) ma.cutAndCopyList("pi+:syst", "pi+:all", "dr<2 and abs(dz)<4", path=path) D0Channel = ["K-:syst pi+:syst"] D0List = [] for chID, channel in enumerate(D0Channel): ma.reconstructDecay(f"D0:syst{chID} -> {channel}", D0Cuts, chID, path=path) D0List.append(f"D0:syst{chID}") DstarChannel = [] for channel in D0List: DstarChannel.append(f"{channel} pi+:syst") DstarList = [] for chID, channel in enumerate(DstarChannel): ma.reconstructDecay(f"D*+:syst{chID} -> {channel}", DstarCuts, chID, path=path) DstarList.append(f"D*+:syst{chID}") return DstarList
[docs] @fancy_skim_header class SystematicsTracking(BaseSkim): """ Lists in this skim are those defined in `BtoDStarPiList` and `DstarToD0PiPartList`. """ __authors__ = ["Sam Cunliffe", "Torben Ferber", "Ilya Komarov", "Yuji Kato"] __description__ = "" __contact__ = __liaison__ __category__ = "systematics" ApplyHLTHadronCut = False
[docs] def load_standard_lists(self, path): stdK("loose", path=path) stdPi("loose", path=path) stdPi0s("eff40_May2020", path=path)
[docs] def build_lists(self, path): return self.BtoDStarPiList(path) + self.DstarToD0PiPartList(path)
[docs] def BtoDStarPiList(self, path): """Build BtoDStarPiList lists for systematics skims.""" D0Cuts = "1.835 < M < 1.895" DstarCuts = "massDifference(0)<0.16" B0Cuts = "Mbc > 5.2 and abs(deltaE) < 0.3" # D0 D0Channel = ["K+:loose pi-:loose", "K+:loose pi-:loose pi-:loose pi+:loose", "K+:loose pi-:loose pi0:eff40_May2020"] D0List = [] for chID, channel in enumerate(D0Channel): resonanceName = "anti-D0:loose" + str(chID) ma.reconstructDecay(resonanceName + " -> " + channel, D0Cuts, chID, path=path) # vertex.raveFit(resonanceName, 0.0, path=path) ma.copyLists("anti-D0:loose", ["anti-D0:loose0", "anti-D0:loose1", "anti-D0:loose2"], path=path) D0List.append("anti-D0:loose") # Dstar DstarChannel = [] for channel in D0List: DstarChannel.append(channel + " pi-:loose") DstarList = [] for chID, channel in enumerate(DstarChannel): resonanceName = "D*-:loose" + str(chID) ma.reconstructDecay(resonanceName + " -> " + channel, DstarCuts, chID, path=path) # vertex.raveFit(resonanceName, 0.0) DstarList.append(resonanceName) # B0 B0Channel = [] for channel in DstarList: B0Channel.append(channel + " pi+:loose") B0List = [] for chID, channel in enumerate(B0Channel): resonanceName = "B0:sys" + str(chID) ma.reconstructDecay(resonanceName + " -> " + channel, B0Cuts, chID, path=path) B0List.append(resonanceName) # vertex.raveFit(resonanceName, 0.0) return B0List
[docs] def DstarToD0PiPartList(self, path): """Build DstarToD0PiPartList lists for systematics skims.""" ma.fillParticleList("pi+:fromks", "chiProb > 0.001 and pionID > 0.1 and dr > 0.1", path=path) # D- DminusCuts = "1.0 < M < 1.75" DminusChannel = ["pi-:fromks pi+:loose pi-:loose"] for chID, channel in enumerate(DminusChannel): resonanceName = "D-:loose" + str(chID) ma.reconstructDecay(resonanceName + " -> " + channel, DminusCuts, chID, path=path) # Dstar DstarCuts = "massDifference(0)<0.2 and useCMSFrame(p) > 2.0" DstarChannel = [] DstarChannel.append("D-:loose0" + " pi+:loose") DstarList = [] for chID, channel in enumerate(DstarChannel): resonanceName = "D*0:loose" + str(chID) ma.reconstructDecay(resonanceName + " -> " + channel, DstarCuts, chID, path=path) DstarList.append(resonanceName) return DstarList
[docs] @fancy_skim_header class Resonance(BaseSkim): """ Lists in this skim are those defined in `getDsList`, `getDstarList`, `getSigmacList`, `getmumugList`, `getBZeroList`, and `getBPlusList`. """ __authors__ = ["Sam Cunliffe", "Torben Ferber", "Ilya Komarov", "Yuji Kato"] __description__ = "" __contact__ = __liaison__ __category__ = "systematics" ApplyHLTHadronCut = False
[docs] def load_standard_lists(self, path): stdK("loose", path=path) stdMu("loose", path=path) stdPi("loose", path=path) stdPr("loose", path=path) stdPi0s("eff40_May2020Fit", path=path)
[docs] def build_lists(self, path): return ( self.getDsList(path) + self.getDstarList(path) + self.getSigmacList(path) + self.getmumugList(path) + self.getBZeroList(path) + self.getBPlusList(path) )
[docs] def getDsList(self, path): """Build Ds list for systematics skims.""" DsCuts = "1.90 < M < 2.04" ma.reconstructDecay("phi:res -> K+:loose K-:loose", "1.01 < M < 1.03", path=path) ma.reconstructDecay("K*0:res -> K+:loose pi-:loose", "0.7 < M < 1.1", path=path) DsChannel = ["phi:res pi+:loose"] DsList = [] for chID, channel in enumerate(DsChannel): particlename = f"D_s+:Resonance{int(chID)}" ma.reconstructDecay(particlename + " -> " + channel, DsCuts, chID, path=path) DsList.append(particlename) return DsList
[docs] def getDstarList(self, path): """Build Dstar list for systematics skims.""" DplusCuts = "1.8 < M < 1.93" DstarCuts = "massDifference(0)<0.16 and useCMSFrame(p)>2.0" DplusChannel = ["K-:loose pi+:loose pi+:loose"] DplusList = [] for chID, channel in enumerate(DplusChannel): ma.reconstructDecay("D+:resonance" + str(chID) + " -> " + channel, DplusCuts, chID, path=path) vertex.raveFit("D+:resonance" + str(chID), 0.0, path=path) DplusList.append("D+:resonance" + str(chID)) DstarChannel = [] for channel in DplusList: DstarChannel.append(channel + " pi0:eff40_May2020") DstarList = [] for chID, channel in enumerate(DstarChannel): ma.reconstructDecay("D*+:resonance" + str(chID) + " -> " + channel, DstarCuts, chID, path=path) DstarList.append("D*+:resonance" + str(chID)) return DstarList
[docs] def getSigmacList(self, path): """Build Sigmac list for systematics skims.""" LambdacCuts = "2.24 < M < 2.33" SigmacCuts = "massDifference(0)<0.28 and useCMSFrame(p) > 2.5" LambdacChannel = ["p+:loose K-:loose pi+:loose"] LambdacList = [] for chID, channel in enumerate(LambdacChannel): ma.reconstructDecay("Lambda_c+:resonance" + str(chID) + " -> " + channel, LambdacCuts, chID, path=path) vertex.raveFit("Lambda_c+:resonance" + str(chID), 0.0, path=path) LambdacList.append("Lambda_c+:resonance" + str(chID)) SigmacList = [] SigmacPlusChannel = [] # Sigma_c++ for channel in LambdacList: SigmacPlusChannel.append(channel + " pi+:loose") for chID, channel in enumerate(SigmacPlusChannel): ma.reconstructDecay("Sigma_c++:resonance" + str(chID) + " -> " + channel, SigmacCuts, chID, path=path) SigmacList.append("Sigma_c++:resonance" + str(chID)) # Sigma_c0 Sigmac0Channel = [] for channel in LambdacList: Sigmac0Channel.append(channel + " pi-:loose") Sigmac0List = [] for chID, channel in enumerate(Sigmac0Channel): ma.reconstructDecay("Sigma_c0:resonance" + str(chID) + " -> " + channel, SigmacCuts, chID, path=path) Sigmac0List.append("Sigma_c0:resonance" + str(chID)) return SigmacList
[docs] def getmumugList(self, path): """Build mumug list for systematics skims.""" vphoChannel = ["mu+:loose mu-:loose"] vphocuts = "" vphoList = [] for chID, channel in enumerate(vphoChannel): resonanceName = "vpho:resonance" + str(chID) ma.reconstructDecay("vpho:resonance" + str(chID) + " -> " + channel, vphocuts, chID, path=path) ma.applyCuts(resonanceName, "nTracks == 2 and M < formula(Ecms*0.9877)", path=path) vertex.raveFit(resonanceName, 0.0, path=path) ma.applyCuts(resonanceName, "M < formula(Ecms*0.9877)", path=path) vphoList.append(resonanceName) return vphoList
[docs] def getBZeroList(self, path): """Build BZero list for systematics skims.""" BZeroCuts = "Mbc > 5.2 and abs(deltaE) < 0.3" BZeroChannel = ["D-:resonance0 pi+:loose"] BZeroList = [] for chID, channel in enumerate(BZeroChannel): resonanceName = "B0:resonance" + str(chID) ma.reconstructDecay(resonanceName + " -> " + channel, BZeroCuts, chID, path=path) BZeroList.append(resonanceName) return BZeroList
[docs] def getBPlusList(self, path): """Build Bplus list for systematics skims.""" antiDZeroCut = "1.82 < M < 1.90" antiDZeroChannel = ["K+:loose pi-:loose"] antiDZeroList = [] for chID, channel in enumerate(antiDZeroChannel): resonanceName = "anti-D0:resonance" + str(chID) ma.reconstructDecay(resonanceName + " -> " + channel, antiDZeroCut, chID, path=path) vertex.raveFit(resonanceName, 0.0, path=path) antiDZeroList.append(resonanceName) BPlusChannel = [] for channel in antiDZeroList: BPlusChannel.append(channel + " pi+:loose") BPlusCuts = "Mbc > 5.2 and abs(deltaE) < 0.3" BPlusList = [] for chID, channel in enumerate(BPlusChannel): ma.reconstructDecay("B+:resonance" + str(chID) + " -> " + channel, BPlusCuts, chID, path=path) BPlusList.append("B+:resonance" + str(chID)) return BPlusList
[docs] @fancy_skim_header class SystematicsRadMuMu(BaseSkim): """ We require one cluster-matched electron (the other is not required to match a cluster). No selection on the photon as the sample must be unbiased. """ __authors__ = ["Torben Ferber"] __description__ = ( "Skim of radiative muon pairs (:math:`ee\\to\\mu\\mu(\\gamma)`) " "for photon systematics." ) __contact__ = __liaison__ __category__ = "systematics, photon calibration" ApplyHLTHadronCut = False
[docs] def load_standard_lists(self, path): stdMu("all", path=path)
[docs] def build_lists(self, path): # the tight selection starts with all muons, but they must be cluster-matched and not be an electron MuonTightSelection = ("abs(dz) < 2.0 and abs(dr) < 0.5 and nCDCHits > 0 and " "clusterE > 0.0 and clusterE < 1.0") ma.cutAndCopyList("mu+:skimtight", "mu+:all", MuonTightSelection, path=path) # for the loose selection starts with all muons, but we accept tracks that # are not matched to a cluster, but if they are, they must not be an # electron MuonLooseSelection = "abs(dz) < 2.0 and abs(dr) < 0.5 and nCDCHits > 0 and clusterE < 1.0" ma.cutAndCopyList("mu+:skimloose", "mu+:all", MuonLooseSelection, path=path) # create a list of possible selections radmumulist = [] # selection ID0: # the radiative muon pair must be selected without looking at the photon. # exclude events with more than two good tracks RadMuMuSelection = "pRecoil > 0.075 and pRecoilTheta > 0.296706 and pRecoilTheta < 2.61799" RadMuMuPairChannel = "mu+:skimtight mu-:skimloose" chID = 0 ma.reconstructDecay("vpho:radmumu" + str(chID) + " -> " + RadMuMuPairChannel, RadMuMuSelection, chID, path=path) eventCuts = "nCleanedTracks(abs(dz) < 2.0 and abs(dr) < 0.5) == 2" ma.applyCuts("vpho:radmumu" + str(chID), eventCuts, path=path) radmumulist.append("vpho:radmumu" + str(chID)) # selection Id1: # todo: include pair conversions? return radmumulist
[docs] @fancy_skim_header class SystematicsEELL(BaseSkim): __authors__ = ["Ilya Komarov"] __description__ = "Systematics skim of :math:`ee\\to ee\\ell\\ell`" __contact__ = __liaison__ __category__ = "systematics, lepton ID" ApplyHLTHadronCut = False
[docs] def load_standard_lists(self, path): stdE("all", path=path)
[docs] def build_lists(self, path): # At skim level we avoid any PID-like requirements and just select events # with two good tracks coming from the interavtion region. eLooseSelection = "abs(dz) < 2.0 and abs(dr) < 0.5 and p > 0.3" ma.cutAndCopyList("e+:skimloose", "e+:all", eLooseSelection, path=path) # create a list of possible selections eelllist = [] # Lepon pair tracks are back-to-back-like EELLSelection = "useCMSFrame(pt)<0.3" eventCuts = "nCleanedTracks(abs(dz) < 2.0 and abs(dr) < 0.5) < 4" ma.reconstructDecay("gamma:eell -> e+:skimloose e-:skimloose", EELLSelection + " and " + eventCuts, path=path) eelllist.append("gamma:eell") return eelllist
[docs] @fancy_skim_header class SystematicsRadEE(BaseSkim): """ Constructed skim list contains radiative electron pairs for photon systematics. In particular this is for the endcaps where we have no track triggers, we require one cluster-matched electron (the other is not required to match a cluster). No selection on the photon as the sample must be unbiased. As this retains a lot of bhabha events (by construction) we allow for prescaling (and prefer prescaled rather than a biased sampe by requiring any selection on the photon or too much of a cut on the recoil momentum). Prescales are given in standard trigger terms (reciprocal), so prescale of 100 is 1% of events kept, *etc*. """ __authors__ = ["Sam Cunliffe"] __description__ = "Radiative electron pairs for photon systematics" __contact__ = __liaison__ __category__ = "systematics, photon calibration" ApplyHLTHadronCut = False
[docs] def load_standard_lists(self, path): stdE("all", path=path)
def __init__(self, prescale_all=1, prescale_fwd_electron=1, **kwargs): """ Parameters: prescale_all (int): the global prescale for this skim prescale_fwd_electron (int): the prescale electrons (e-) in the forward endcap **kwargs: Passed to constructor of `BaseSkim`. """ # Redefine __init__ to allow for additional optional arguments super().__init__(**kwargs) self.prescale_all = prescale_all self.prescale_fwd_electron = prescale_fwd_electron
[docs] def build_lists(self, path): # convert prescales from trigger convention prescale_all = str(float(1.0 / self.prescale_all)) prescale_fwd_electron = str(float(1.0 / self.prescale_fwd_electron)) # require a pair of good electrons one of which must be cluster-matched # with 3 GeV of energy goodtrack = "abs(dz) < 2.0 and abs(dr) < 0.5 and nCDCHits > 0" goodtrackwithcluster = f"{goodtrack} and clusterE > 3.0" ma.cutAndCopyList("e+:skimtight", "e+:all", goodtrackwithcluster, path=path) ma.cutAndCopyList("e+:skimloose", "e+:all", goodtrack, path=path) # a minimum momentum of 75 MeV/c recoiling against the pair, # and require that the recoil is within the CDC acceptance recoil = "pRecoil > 0.075 and 0.296706 < pRecoilTheta < 2.61799" # GeV/c, rad ma.reconstructDecay("vpho:radee -> e+:skimtight e-:skimloose", recoil, path=path) # apply event cuts (exactly two clean tracks in the event, and prescale # the whole event regardless of where the electron went) event_cuts = "[nCleanedTracks(abs(dz) < 2.0 and abs(dr) < 0.5) == 2]" # cm, cm event_cuts += f" and [eventRandom <= {prescale_all}]" # now prescale the *electron* (e-) in the forward endcap (for bhabhas) # note this is all done with cut strings to circumnavigate BII-3607 fwd_encap_border = "0.5480334" # rad (31.4 deg) electron_is_first = "daughter(0, charge) < 0" first_in_fwd_endcap = f"daughter(0, theta) < {fwd_encap_border}" first_not_in_fwd_endcap = f"daughter(0, theta) > {fwd_encap_border}" electron_is_second = "daughter(1, charge) < 0" second_in_fwd_endcap = f"daughter(1, theta) < {fwd_encap_border}" second_not_in_fwd_endcap = f"daughter(1, theta) > {fwd_encap_border}" passes_prescale = f"eventRandom <= {prescale_fwd_electron}" # # four possible scenarios: # 1) electron first in the decaystring and in fwd endcap: prescale these prescale_logic = f"[{electron_is_first} and {first_in_fwd_endcap} and {passes_prescale}]" # 2) electron second in string and in fwd endcap: prescale these prescale_logic += f" or [{electron_is_second} and {second_in_fwd_endcap} and {passes_prescale}]" # 3) electron first in string and not in fwd endcap (no prescale) prescale_logic += f" or [{electron_is_first} and {first_not_in_fwd_endcap}]" # 4) electron second in string and not in fwd endcap (no prescale) prescale_logic += f" or [{electron_is_second} and {second_not_in_fwd_endcap}]" # final candidate building with cuts and prescales prescale_logic = f"[{prescale_logic}]" ma.applyCuts("vpho:radee", event_cuts + " and " + prescale_logic, path=path) return ["vpho:radee"]
[docs] @fancy_skim_header class SystematicsLambda(BaseSkim): __authors__ = ["Sam Cunliffe", "Torben Ferber", "Ilya Komarov", "Yuji Kato", "Jake Bennett"] __description__ = "" __contact__ = __liaison__ __category__ = "systematics" ApplyHLTHadronCut = True
[docs] def load_standard_lists(self, path): stdLambdas(path=path)
[docs] def build_lists(self, path): vm.addAlias("fsig", "formula(flightDistance/flightDistanceErr)") vm.addAlias("pMom", "daughter(0,p)") vm.addAlias("piMom", "daughter(1,p)") vm.addAlias("daughtersPAsym", "formula((pMom-piMom)/(pMom+piMom))") LambdaList = [] ma.cutAndCopyList("Lambda0:syst0", "Lambda0:merged", "fsig>10 and daughtersPAsym>0.41", path=path) LambdaList.append("Lambda0:syst0") return LambdaList
[docs] @fancy_skim_header class SystematicsPhiGamma(BaseSkim): """ Uses the ``gamma:loose`` list and a cut on the number of tracks. Cuts applied: * :math:`E_{\\gamma}> 3\\,\\text{GeV}` AND * :math:`E_{\\gamma}< 8\\,\\text{GeV}` * :math:`n_{\\text{tracks}} \\geq 2` AND :math:`n_{\\text{tracks}} \\leq 4` * at least 1 candidate in the K_S0:merged or in the phi->K+:all K-:all lists """ __authors__ = ["Giuseppe Finocchiaro", "Benjamin Oberhof"] __description__ = ( "Skim for ISR - phi gamma analyses, " ":math:`e^+ e^- \\to \\phi \\gamma ` and " ":math:`\\phi` decays into two charged tracks " "(:math:`K^+K^-` or :math:`K_S K_L` with :math:`K_S\\to \\pi^+\\pi^-`)" ) __contact__ = "Giuseppe Finocchiaro <giuseppe.finocchiaro@lnf.infn.it>" __category__ = "systematics" ApplyHLTHadronCut = False TestSampleProcess = "ccbar" validation_sample = _VALIDATION_SAMPLE
[docs] def load_standard_lists(self, path): stdPhotons("loose", path=path) stdK("all", path=path) stdKshorts(path=path)
[docs] def build_lists(self, path): EventCuts = [ "[nTracks>=2] and [nTracks<=6]", "[nParticlesInList(gamma:PhiSystematics) > 0]", "[[nParticlesInList(phi:charged) > 0] or [nParticlesInList(K_S0:PhiSystematics) > 0]]" ] ma.cutAndCopyList("gamma:PhiSystematics", "gamma:loose", "3 < E < 8", path=path) ma.reconstructDecay('phi:charged -> K+:all K-:all', '0.9 < M < 1.2', path=path) ma.copyList('K_S0:PhiSystematics', 'K_S0:merged', path=path) path = self.skim_event_cuts(" and ".join(EventCuts), path=path) return ["gamma:PhiSystematics"]
[docs] def validation_histograms(self, path): # NOTE: the validation package is not part of the light releases, so this import # must be made here rather than at the top of the file. from validation_tools.metadata import create_validation_histograms vm.addAlias("gamma_E_CMS", "useCMSFrame(E)") vm.addAlias("gamma_E", "E") vm.addAlias("K_S0_mass", "M") vm.addAlias("phi_mass", "M") histoRootFile = f'{self}_Validation.root' variableshisto = [('gamma_E', 120, 2.5, 8.5, 'gamma_E', self.__contact__, 'Photon energy', ''), ('gamma_E_CMS', 100, 2.0, 7.0, 'gamma_E_CMS', self.__contact__, 'Photon energy in CMS', ''), ('nTracks', 15, 0, 15, 'nTracks', self.__contact__, 'Number of tracks', ''), ] variableshistoKS = [('K_S0_mass', 200, 0.4, 0.6, 'K_S0_mass', self.__contact__, 'Invariant KS0 mass', ''), ] variableshistoPhi = [('phi_mass', 200, 0.8, 1.2, 'phi_mass', self.__contact__, 'Invariant phi mass', ''), ] create_validation_histograms(path, histoRootFile, 'gamma:PhiSystematics', variableshisto) create_validation_histograms(path, histoRootFile, 'K_S0:merged', variableshistoKS) create_validation_histograms(path, histoRootFile, 'phi:charged', variableshistoPhi)
[docs] @fancy_skim_header class Random(BaseSkim): __authors__ = "Phil Grace" __contact__ = "Phil Grace <philip.grace@adelaide.edu.au>" __description__ = "Random skim to select a fixed fraction of events." __category__ = "systematics, random" ApplyHLTHadronCut = False def __init__(self, KeepPercentage=10, seed=None, **kwargs): """ Parameters: KeepPercentage (float): Percentage of events to be kept. seed (int): Set random seed to given number. If this argument is not given, this skim will not alter the random seed. **kwargs: Passed to constructor of `BaseSkim`. """ super().__init__(**kwargs) self.KeepPercentage = KeepPercentage self.seed = seed
[docs] def additional_setup(self, path): if self.seed is not None: b2.set_random_seed(int(self.seed))
[docs] def load_standard_lists(self, path): stdPi("all", path=path) stdPhotons("all", path=path)
[docs] def build_lists(self, path): # Select one photon/track per event with no other cuts, so that all events are # captured in the skim list if KeepPercentage=100. label = "RandomSkim" ma.copyList(f"pi+:{label}", "pi+:all", path=path) ma.copyList(f"gamma:{label}", "gamma:all", path=path) ma.applyRandomCandidateSelection(f"pi+:{label}", path=path) ma.applyRandomCandidateSelection(f"gamma:{label}", path=path) # Select fraction of events path = self.skim_event_cuts( f"eventRandom <= {self.KeepPercentage/100}", path=path ) return [f"pi+:{label}", f"gamma:{label}"]
[docs] @fancy_skim_header class SystematicsFourLeptonFromHLTFlag(BaseSkim): __authors__ = "Marcel Hohmann" __contact__ = __liaison_leptonID__ __description__ = "Skim to select all events that pass the HLT Four Lepton skim for lepton ID studies" __category__ = "systematics, leptonID" ApplyHLTHadronCut = False
[docs] def load_standard_lists(self, path): stdPi("all", path=path)
[docs] def build_lists(self, path): label = "FourLeptonHLT" ma.copyList(f"pi+:{label}", "pi+:all", path=path) ma.rankByLowest(f"pi+:{label}", "random", 1, "systematicsFourLeptonHLT_randomRank", path=path) path = self.skim_event_cuts( "SoftwareTriggerResult(software_trigger_cut&skim&accept_fourlep) == 1", path=path ) return [f"pi+:{label}"]
[docs] @fancy_skim_header class SystematicsRadMuMuFromHLTFlag(BaseSkim): __authors__ = "Marcel Hohmann" __contact__ = __liaison_leptonID__ __description__ = "Skim to select all events that pass the HLT RadMuMu skim for lepton ID studies" __category__ = "systematics, leptonID" ApplyHLTHadronCut = False
[docs] def load_standard_lists(self, path): stdPi("all", path=path)
[docs] def build_lists(self, path): label = "RadMuMuLeptonID" ma.copyList(f"pi+:{label}", "pi+:all", path=path) ma.rankByLowest(f"pi+:{label}", "random", 1, "systematicsRadMuMuLeptonID_randomRank", path=path) path = self.skim_event_cuts( "SoftwareTriggerResult(software_trigger_cut&skim&accept_radmumu) == 1", path=path ) return [f"pi+:{label}"]
[docs] @fancy_skim_header class SystematicsJpsi(BaseSkim): """ J/psi skim for lepton ID systematics studies. Lists in this skim are those defined in `JpsimumuTagProbe`, `JpsieeTagProbe`. """ __authors__ = ["Sam Cunliffe", "Torben Ferber", "Ilya Komarov", "Yuji Kato", "Racha Cheaib", "Marcel Hohmann"] __description__ = "" __contact__ = __liaison_leptonID__ __category__ = "systematics, leptonID"
[docs] def load_standard_lists(self, path): stdMu("all", path=path) stdE("all", path=path) stdPhotons("all", path=path)
TestSampleProcess = "ccbar" ApplyHLTHadronCut = True
[docs] def build_lists(self, path): return [ self.JpsimumuTagProbe(path), self.JpsieeTagProbe(path), ]
[docs] def JpsimumuTagProbe(self, path): """Build JpsimumuTagProbe lists for systematics skims.""" Cuts = "2.7 < M < 3.4" ma.reconstructDecay( "J/psi:systematics_mumu -> mu+:all mu-:all", f'{Cuts} and [daughter(0,muonID)>0.1 or daughter(1,muonID)>0.1]', path=path) return "J/psi:systematics_mumu"
[docs] def JpsieeTagProbe(self, path): """Build JpsieeTagProbe lists for systematics skims.""" Cuts = "2.7 < M < 3.4" ma.cutAndCopyList('gamma:brems', 'gamma:all', 'E<1', path=path) ma.correctBrems('e+:brems_corrected', 'e+:all', 'gamma:brems', path=path) ma.reconstructDecay( "J/psi:systematics_ee -> e+:brems_corrected e-:brems_corrected", f'{Cuts} and [daughter(0,electronID)>0.1 or daughter(1,electronID)>0.1]', path=path) return "J/psi:systematics_ee"
[docs] @fancy_skim_header class SystematicsKshort(BaseSkim): """ K-short skim for hadron and lepton ID systematics studies. As K-short candidates are abundant this skim has a high retention. To meet the retention criteria a prescale is added. The prescale is given in standard trigger terms (reciprocal). A prescale of 50 will keep 2% of events, etc. """ __authors__ = ["Marcel Hohmann"] __description__ = "Skim for K-short events for performance studies" __contact__ = __liaison_leptonID__ __category__ = "performance, leptonID" ApplyHLTHadronCut = True def __init__(self, prescale=4, **kwargs): """ Parameters: prescale (int): the global prescale for this skim. **kwargs: Passed to constructor of `BaseSkim`. """ self.prescale = prescale super().__init__(**kwargs)
[docs] def load_standard_lists(self, path): stdPi("all", path=path)
[docs] def build_lists(self, path): ma.reconstructDecay( 'K_S0:reco -> pi+:all pi-:all', '[0.30 < M < 0.70]', path=path) vertex.treeFit('K_S0:reco', 0.0, path=path) ma.applyCuts('K_S0:reco', '0.4 < M < 0.6', path=path) ma.fillParticleList('K_S0:v0 -> pi+ pi-', '[0.30 < M < 0.70]', True, path=path) vertex.treeFit('K_S0:v0', 0.0, path=path) ma.applyCuts('K_S0:v0', '0.4 < M < 0.6', path=path) ma.mergeListsWithBestDuplicate('K_S0:merged', ['K_S0:v0', 'K_S0:reco'], variable='particleSource', preferLowest=True, path=path) KS_cut = '[[cosAngleBetweenMomentumAndVertexVector>0.998] or '\ ' [formula(flightDistance/flightDistanceErr)>11] or '\ ' [flightTime>0.007]]' # and '\ # '[useAlternativeDaughterHypothesis(M, 0:p+) > 1.13068 and useAlternativeDaughterHypothesis(M, 0:pi-, 1:p+) > 1.13068]' ma.cutAndCopyList("K_S0:skim", "K_S0:merged", KS_cut, path=path) path = self.skim_event_cuts(f'eventRandom < {(1/self.prescale):.6f}', path=path) return ['K_S0:skim']
[docs] @fancy_skim_header class SystematicsBhabha(BaseSkim): """ Skim for selecting Bhabha events for leptonID studies. In case the retention exceeds 10% a prescale can be added. The prescale is given in standard trigger terms (reciprocal). """ __authors__ = ["Justin Skorupa"] __description__ = "Skim for Bhabha events for lepton ID study" __contact__ = __liaison_leptonID__ __category__ = "performance, leptonID" ApplyHLTHadronCut = False def __init__(self, prescale=1, **kwargs): """ Parameters: prescale (int): the global prescale for this skim. **kwargs: Passed to constructor of `BaseSkim`. """ self.prescale = prescale super().__init__(**kwargs)
[docs] def load_standard_lists(self, path): stdE("all", path=path)
[docs] def build_lists(self, path): goodtrack = "abs(dz) < 5 and abs(dr) < 2" goodtrackwithPID = f"{goodtrack} and electronID > 0.95 and clusterTheta > 0.59"\ " and clusterTheta < 2.15 and useCMSFrame(clusterE) > 2" ma.cutAndCopyList("e+:tight", "e+:all", goodtrackwithPID, path=path) ma.cutAndCopyList("e+:loose", "e+:all", goodtrack, path=path) ma.reconstructDecay( "vpho:bhabha -> e+:tight e-:loose", "", path=path) event_cuts = "[nCleanedTracks(abs(dz) < 5 and abs(dr) < 2) == 2]"\ f" and eventRandom < {(1/self.prescale):.6f}" ma.applyCuts("vpho:bhabha", event_cuts, path=path) return ["vpho:bhabha"]
[docs] @fancy_skim_header class SystematicsCombinedHadronic(CombinedSkim): """ Combined systematics skim for the four hadronic channels: SystematicsKshort, SystematicsJpsi, SystematicsDstar, SystemmaticsLambda. This is required for technical (data production) reasons, as it keeps the number of files low. See the definitions of the individual skims for the details. """ __authors__ = ["Marcel Hohmann"] __description__ = "Combined Skim of the systematic hadronic skims: Kshort, Jpsi, Dstar, Lambda." __contact__ = __liaison_leptonID__ __category__ = "performance, leptonID" __name__ = "SystematicsCombinedHadronic" produces_mdst_by_default = True def __init__(self, prescale_kshort=4, mdstOutput=True, **kwargs): """ Initialiser. Args: prescale_kshort (Optional[int]): offline prescale factor for KS skim. **kwargs: key-worded arguments. See CombinedSkim.__init__() """ kwargs.update(mdstOutput=mdstOutput, CombinedSkimName=self.__name__) kwargs.setdefault('udstOutput', False) skims_list = [SystematicsKshort(prescale=prescale_kshort), SystematicsDstar(), SystematicsLambda(), SystematicsJpsi()] super().__init__(*skims_list, **kwargs)
[docs] @fancy_skim_header class SystematicsCombinedLowMulti(CombinedSkim): """ Combined systematics skim for the four low multi channels: SystematicsFourLeptonFromHLTFlag, SystematicsRadmumuFromHLTFlag, SystematicsBhabha, TauThrust. This is required for technical (data production) reasons, as it keeps the number of files low. See the definitions of the individual skims for the details. """ __authors__ = ["Marcel Hohmann"] __description__ = "Combined Skim of the systematic low multi skims: FourLepton, Radmumu, Bhabha, TauThrust." __contact__ = __liaison_leptonID__ __category__ = "performance, leptonID" __name__ = "SystematicsCombinedLowMulti" produces_mdst_by_default = True def __init__(self, mdstOutput=True, **kwargs): """ Initialiser. Args: **kwargs: key-worded arguments. See CombinedSkim.__init__() """ kwargs.update(mdstOutput=mdstOutput, CombinedSkimName=self.__name__) kwargs.setdefault('udstOutput', False) from skim.WGs.taupair import TauThrust skims_list = [SystematicsFourLeptonFromHLTFlag(), SystematicsRadMuMuFromHLTFlag(), SystematicsBhabha(), TauThrust()] super().__init__(*skims_list, **kwargs)