#!/usr/bin/env python3
# -*- coding: utf-8 -*-
##########################################################################
# 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 EWP inclusive skims:
B->Xgamma, B->Xll, B->Xll (LFV modes)
"""
import basf2 as b2
import modularAnalysis as ma
from skim import BaseSkim, fancy_skim_header
from stdCharged import stdE, stdK, stdMu, stdPi
from stdPhotons import stdPhotons
from variables import variables as vm
__liaison__ = "Trevor Shillington <trshillington@hep.physics.mcgill.ca>"
_VALIDATION_SAMPLE = "mdst14.root"
[docs]@fancy_skim_header
class BtoXgamma(BaseSkim):
"""
Reconstructed decay modes:
* :math:`B^+ \\to X\\gamma` inclusive
Event-level cuts:
* :math:`\\text{foxWolframR2} < 0.5` constructed using tracks with
:math:`p_T>0.1\\,\\text{GeV}` and clusters with :math:`E>0.1\\,\\text{GeV}`.
* :math:`n_{\\text{tracks}} \\geq 3`
Cuts on photons:
* :math:`\\text{clusterE9E21}>0.9`
* :math:`1.4\\,\\text{GeV}<\\E_{\\gamma}<3.4\\,\\text{GeV}` in CMS frame
"""
__authors__ = ["Trevor Shillington"]
__description__ = ":math:`B\\to X\\gamma` inclusive skim."
__contact__ = __liaison__
__category__ = "physics, electroweak penguins, radiative decays"
validation_sample = _VALIDATION_SAMPLE
[docs] def load_standard_lists(self, path):
stdPi("all", path=path)
stdPhotons("loose", path=path, loadPhotonBeamBackgroundMVA=False)
[docs] def build_lists(self, path):
"""Build the skim list for :math:`B \\to X\\gamma` decays."""
# event level cuts: R2 and require a minimum number of tracks + decent photons
ma.fillParticleList(decayString='pi+:BtoXgamma_eventshape', cut='pt > 0.1', path=path)
ma.fillParticleList(decayString='gamma:BtoXgamma_eventshape', cut='E > 0.1', path=path,
loadPhotonBeamBackgroundMVA=False)
ma.buildEventShape(inputListNames=['pi+:BtoXgamma_eventshape', 'gamma:BtoXgamma_eventshape'],
allMoments=False,
foxWolfram=True,
harmonicMoments=False,
cleoCones=False,
thrust=False,
collisionAxis=False,
jets=False,
sphericity=False,
checkForDuplicates=False,
path=path)
# Apply event cuts R2 < 0.5 and nTracks >= 3
path = self.skim_event_cuts('foxWolframR2 < 0.5 and nTracks >= 3', path=path)
# Apply gamma cuts clusterE9E21 > 0.9 and 1.4 < E_gamma < 3.4 GeV (in CMS frame)
ma.cutAndCopyList('gamma:ewp', 'gamma:loose', 'clusterE9E21 > 0.9 and 1.4 < useCMSFrame(E) < 3.4', path=path)
ma.reconstructDecay('B+:gamma -> gamma:ewp', '', path=path, allowChargeViolation=True)
return ['B+:gamma']
[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
histogram_filename = f'{self}_Validation.root'
stdK('all', path=path)
stdPhotons('cdc', path=path, loadPhotonBeamBackgroundMVA=False)
ma.cutAndCopyList('gamma:sig', 'gamma:cdc', 'clusterNHits > 1.5 and E > 1.5', True, path)
ma.reconstructDecay('K*0:sig -> K+:all pi-:all', '0.6 < M < 1.6', path=path)
ma.reconstructDecay('B0:sig -> K*0:sig gamma:sig', '5.22 < Mbc < 5.3 and abs(deltaE)< .5', path=path)
# the variables that are printed out are: Mbc and deltaE
create_validation_histograms(
rootfile=histogram_filename,
particlelist='B0:sig',
variables_1d=[
('Mbc', 100, 5.2, 5.3, 'Signal B0 Mbc', __liaison__,
'Mbc of the signal B0', '', 'Mbc [GeV/c^2]', 'Candidates'),
('deltaE', 100, -1, 1, 'Signal B0 deltaE', __liaison__,
'deltaE of the signal B0', '', 'deltaE [GeV]', 'Candidates')
],
variables_2d=[],
path=path)
[docs]@fancy_skim_header
class BtoXll(BaseSkim):
"""
Reconstructed decay modes:
* :math:`B^+ \\to X e^+ e^-`
* :math:`B^+ \\to X e^+ e^+`
* :math:`B^+ \\to X \\mu^+ \\mu^-`
* :math:`B^+ \\to X \\mu^+ \\mu^+`
Event-level cuts:
* :math:`\\text{foxWolframR2} < 0.5` constructed using tracks with
:math:`p_T>0.1\\,\\text{GeV}` and clusters with :math:`E>0.1\\,\\text{GeV}`.
* :math:`n_{\\text{tracks}} \\geq 3`
Cuts on electrons:
* :math:`\\text{electronID} > 0.1`
* :math:`p > 0.395\\,\\text{GeV}` in lab frame
* :math:`dr<0.5 and abs(dz)<2`
Cuts on muons:
* :math:`\\text{muonID} > 0.5`
* :math:`p > 0.395\\,\\text{GeV}` in lab frame
* :math:`dr<0.5 and abs(dz)<2`
Cut on dilepton energy:
* :math:`E_{\\ell\\ell}>1.5\\,\\text{GeV}` in CMS frame.
"""
__authors__ = ["Trevor Shillington"]
__description__ = ":math:`B\\to X\\ell\\ell` (no LFV modes) inclusive skim."
__contact__ = __liaison__
__category__ = "physics, electroweak penguins, radiative decays"
validation_sample = _VALIDATION_SAMPLE
[docs] def load_standard_lists(self, path):
stdE("all", path=path)
stdMu("all", path=path)
stdPi("all", path=path)
stdPhotons("all", path=path, loadPhotonBeamBackgroundMVA=False)
[docs] def build_lists(self, path):
"""Build the skim list for :math:`B \\to X\\ell\\ell` non-LFV decays."""
# event level cuts: R2 and require a minimum number of tracks
ma.fillParticleList(decayString='pi+:BtoXll_eventshape', cut='pt > 0.1', path=path)
ma.fillParticleList(decayString='gamma:BtoXll_eventshape', cut='E > 0.1', path=path)
ma.buildEventShape(inputListNames=['pi+:BtoXll_eventshape', 'gamma:BtoXll_eventshape'],
allMoments=False,
foxWolfram=True,
harmonicMoments=False,
cleoCones=False,
thrust=False,
collisionAxis=False,
jets=False,
sphericity=False,
checkForDuplicates=False,
path=path)
# Apply event cuts R2 < 0.5 and nTracks >= 3
path = self.skim_event_cuts('foxWolframR2 < 0.5 and nTracks >= 3', path=path)
# Apply electron cut p > 0.395 GeV, electronID > 0.1 + fairTrack
# Apply muon cuts p > 0.395 GeV, muonID > 0.5 + fairTrack
fairTrack = 'dr < 0.5 and abs(dz) < 2'
ma.cutAndCopyList('e+:ewp', 'e+:all', 'p > 0.395 and electronID > 0.1 and ' + fairTrack, path=path)
ma.cutAndCopyList('mu+:ewp', 'mu+:all', 'p > 0.395 and muonID > 0.5 and ' + fairTrack, path=path)
# Apply dilepton cut E_ll > 1.5 GeV (in CMS frame)
E_dilep_cut = 'formula(daughter(0, useCMSFrame(E))+daughter(1, useCMSFrame(E))) > 1.5'
# B+ reconstruction:
# oppositely charged leptons
ma.reconstructDecay('B+:ch1 -> e+:ewp e-:ewp', E_dilep_cut, dmID=1, path=path, allowChargeViolation=True)
ma.reconstructDecay('B+:ch2 -> mu+:ewp mu-:ewp', E_dilep_cut, dmID=2, path=path, allowChargeViolation=True)
# same charge leptons
ma.reconstructDecay('B+:ch3 -> e+:ewp e+:ewp', E_dilep_cut, dmID=3, path=path, allowChargeViolation=True)
ma.reconstructDecay('B+:ch4 -> mu+:ewp mu+:ewp', E_dilep_cut, dmID=4, path=path, allowChargeViolation=True)
ma.copyLists('B+:xll', ['B+:ch1', 'B+:ch2', 'B+:ch3', 'B+:ch4'], path=path)
return ['B+:xll']
[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
histogram_filename = f'{self}_Validation.root'
stdK(listtype='good', path=path)
stdMu(listtype='good', path=path)
ma.reconstructDecay("B+:signal -> K+:good mu+:good mu-:good", "Mbc > 5.2 and deltaE < 0.5 and deltaE > -0.5", path=path)
create_validation_histograms(
rootfile=histogram_filename,
particlelist='B+:signal',
variables_1d=[
('deltaE', 100, -0.5, 0.5, 'Signal B deltaE', __liaison__,
'deltaE of the Signal B', '', 'deltaE [GeV]', 'Candidates'),
('Mbc', 100, 5.2, 5.3, 'Signal B Mbc', __liaison__,
'Mbc of the signal B', '', 'Mbc [GeV/c^2]', 'Candidates')],
variables_2d=[],
path=path)
[docs]@fancy_skim_header
class BtoXll_LFV(BaseSkim):
"""
Reconstructed decay modes:
* :math:`B^+ \\to X e^+ \\mu^-`
* :math:`B^+ \\to X \\mu^+ e^-`
* :math:`B^+ \\to X e^+ \\mu^+`
Event-level cuts:
* :math:`\\text{foxWolframR2} < 0.5` constructed using tracks with
:math:`p_T>0.1\\,\\text{GeV}` and clusters with :math:`E>0.1\\,\\text{GeV}`.
* :math:`n_{\\text{tracks}} \\geq 3`
Cuts on electrons:
* :math:`\\text{electronID} > 0.1`
* :math:`p > 0.395\\,\\text{GeV}` in lab frame
* :math:`dr<0.5 and abs(dz)<2`
Cuts on muons:
* :math:`\\text{muonID} > 0.5`
* :math:`p > 0.395\\,\\text{GeV}` in lab frame
* :math:`dr<0.5 and abs(dz)<2`
Cut on dilepton energy:
* :math:`E_{\\ell\\ell}>1.5\\,\\text{GeV}` in CMS frame.
"""
__authors__ = ["Trevor Shillington"]
__description__ = ":math:`B\\to X\\ell\\ell` (LFV modes only) inclusive skim."
__contact__ = __liaison__
__category__ = "physics, electroweak penguins, radiative decays"
[docs] def load_standard_lists(self, path):
stdE("all", path=path)
stdMu("all", path=path)
stdPi("all", path=path)
stdPhotons("all", path=path, loadPhotonBeamBackgroundMVA=False)
[docs] def build_lists(self, path):
"""Build the skim list for :math:`B \\to X\\ell\\ell` LFV decays."""
# Create lists for buildEventShape (basically all tracks and clusters)
ma.cutAndCopyList('pi+:BtoXllLFV_eventshape', 'pi+:all', 'pt> 0.1', path=path)
ma.cutAndCopyList('gamma:BtoXllLFV_eventshape', 'gamma:all', 'E > 0.1', path=path)
# buildEventShape to access R2
ma.buildEventShape(inputListNames=['pi+:BtoXllLFV_eventshape', 'gamma:BtoXllLFV_eventshape'],
allMoments=False,
foxWolfram=True,
harmonicMoments=False,
cleoCones=False,
thrust=False,
collisionAxis=False,
jets=False,
sphericity=False,
checkForDuplicates=False,
path=path)
# Apply event cuts R2 < 0.5 and nTracks >= 3
path = self.skim_event_cuts('foxWolframR2 < 0.5 and nTracks >= 3', path=path)
# Apply electron cut p > 0.395 GeV, electronID > 0.1 + fairTrack
# Apply muon cuts p > 0.395 GeV, muonID > 0.5 + fairTrack
fairTrack = 'dr < 0.5 and abs(dz) < 2'
ma.cutAndCopyList('e+:ewp', 'e+:all', 'p > 0.395 and electronID > 0.1 and ' + fairTrack, path=path)
ma.cutAndCopyList('mu+:ewp', 'mu+:all', 'p > 0.395 and muonID > 0.5 and ' + fairTrack, path=path)
# Apply dilepton cut E_ll > 1.5 GeV (in CMS frame)
E_dilep_cut = 'formula(daughter(0, useCMSFrame(E))+daughter(1, useCMSFrame(E))) > 1.5'
# B+ reconstruction:
# oppositely charged leptons
ma.reconstructDecay('B+:lfvch1 -> e+:ewp mu-:ewp', E_dilep_cut, dmID=1, path=path, allowChargeViolation=True)
ma.reconstructDecay('B+:lfvch2 -> mu+:ewp e-:ewp', E_dilep_cut, dmID=2, path=path, allowChargeViolation=True)
# same charge leptons
ma.reconstructDecay('B+:lfvch3 -> e+:ewp mu+:ewp', E_dilep_cut, dmID=3, path=path, allowChargeViolation=True)
ma.copyLists('B+:lfv', ['B+:lfvch1', 'B+:lfvch2', 'B+:lfvch3'], path=path)
return ['B+:lfv']
[docs]class inclusiveBplusToKplusNuNu(BaseSkim):
"""
Reconstructed decay modes:
* :math:`B^+ \\to K\\nu\\nu` inclusive
Track cleanup:
* :math:`p_t > 0.1`
* :math:`thetaInCDCAcceptance`
* :math:`dr<0.5 and abs(dz)<3.0`
Event cleanup:
* :math:`3 < nCleanedTracks < 11`
Kaon cuts:
* :math:`track cleanup + event cleanup + nPXDHits > 0`
* :math:`p_t rank=1`
* :math:`kaonID>0.01`
MVA info and cuts:
* mva_identifier: MVAFastBDT_InclusiveBplusToKplusNuNu_Skim
* Global Tag: mva_inclusiveBplusToKplusNuNu
* :math:`mva\\_identifier > 0.5`
"""
__authors__ = ["Cyrille Praz"]
__description__ = "Inclusive skim for :math:`B\\to K\\nu\\nu` analysis"
__contact__ = __liaison__
__category__ = "physics, electroweak penguins, radiative decays"
NoisyModules = ["ParticleCombiner"]
validation_sample = _VALIDATION_SAMPLE
[docs] def build_lists(self, path):
# Default cleanup also used in and ma.buildEventShape
track_cleanup = 'pt > 0.1'
track_cleanup += ' and thetaInCDCAcceptance'
track_cleanup += ' and abs(dz) < 3.0'
track_cleanup += ' and dr < 0.5'
# Min 4 tracks and Max 10 tracks per event.
event_cleanup = 'nCleanedTracks({}) > 3'.format(track_cleanup)
event_cleanup += ' and nCleanedTracks({}) < 11'.format(track_cleanup)
# Define the signal
total_cleanup = track_cleanup + ' and ' + event_cleanup + ' and ' + 'nPXDHits>0'
ma.fillParticleList('K+:inclusiveBplusToKplusNuNu', cut=total_cleanup, path=path)
ma.rankByHighest('K+:inclusiveBplusToKplusNuNu', 'pt', path=path)
ma.applyCuts('K+:inclusiveBplusToKplusNuNu', 'extraInfo(pt_rank)==1', path=path)
ma.applyCuts('K+:inclusiveBplusToKplusNuNu', 'kaonID>1e-2', path=path)
ma.reconstructDecay(decayString='B+:inclusiveBplusToKplusNuNu -> K+:inclusiveBplusToKplusNuNu', cut='', path=path)
# Build the event-based variables that we need
ma.buildEventShape(inputListNames=[],
default_cleanup=True,
allMoments=False,
cleoCones=True,
collisionAxis=False,
foxWolfram=True,
harmonicMoments=True,
jets=False,
sphericity=True,
thrust=True,
checkForDuplicates=False,
path=path)
# Apply a MVA by reading from the DB
mva_identifier = 'MVAFastBDT_InclusiveBplusToKplusNuNu_Skim'
b2.conditions.append_globaltag('mva_inclusiveBplusToKplusNuNu')
path.add_module('MVAExpert', listNames=['B+:inclusiveBplusToKplusNuNu'],
extraInfoName=mva_identifier, identifier=mva_identifier)
vm.addAlias(mva_identifier, f'extraInfo({mva_identifier})')
ma.applyCuts('B+:inclusiveBplusToKplusNuNu', mva_identifier + '>0.5', path=path)
return ['B+:inclusiveBplusToKplusNuNu']
[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
histogram_filename = f'{self}_Validation.root'
# Default cleanup also used in and ma.buildEventShape
track_cleanup = 'pt > 0.1'
track_cleanup += ' and thetaInCDCAcceptance'
track_cleanup += ' and abs(dz) < 3.0'
track_cleanup += ' and abs(dr) < 0.5'
# Define a couple of aliases
vm.addAlias('kaon_pt', 'daughter(0,pt)')
vm.addAlias('nCleanedTracks_simple_cleanup', 'nCleanedTracks({})'.format(track_cleanup))
# Output validation histograms
create_validation_histograms(
rootfile=histogram_filename,
particlelist='B+:inclusiveBplusToKplusNuNu',
variables_1d=[
('kaon_pt',
10,
0,
5,
'Kaon pt',
__liaison__,
'Transverse momentum of the kaon candidate',
'Maximum between 1.5 and 2 GeV/c',
'Kaon pt [GeV/c]',
'Candidates'),
('nCleanedTracks_simple_cleanup',
12,
0,
12,
'Number of cleaned tracks',
__liaison__,
'Number of cleaned tracks in the event',
'Should be between 4 and 10, with two local maxima at 4 and 6',
'Number of cleaned tracks',
'Events'),
('sphericity',
10,
0,
1,
'Event Sphericity',
__liaison__,
'Sphericity computed by ma.buildEventShape',
'Maximum around 0.3',
'Event Sphericity',
'Events')],
variables_2d=[],
path=path)