New line for B -> Tau nu

Phys/StrippingSelections/python/StrippingSelections/StrippingRD/StrippingB2TauNu.py

+###############################################################################
+# (c) Copyright 2000-2019 CERN for the benefit of the LHCb Collaboration      #
+#                                                                             #
+# This software is distributed under the terms of the GNU General Public      #
+# Licence version 3 (GPL Version 3), copied verbatim in the file "COPYING".   #
+#                                                                             #
+# In applying this licence, CERN does not waive the privileges and immunities #
+# granted to it by virtue of its status as an Intergovernmental Organization  #
+# or submit itself to any jurisdiction.                                       #
+###############################################################################
+from StandardParticles import StdLooseAllPhotons
+from StandardParticles import StdLooseKaons, StdNoPIDsKaons
+from StandardParticles import StdNoPIDsPions, StdLoosePions
+from StrippingUtils.Utils import LineBuilder
+from StrippingConf.StrippingLine import StrippingLine
+from PhysSelPython.Wrappers import Selection, DataOnDemand, SimpleSelection, PassThroughSelection
+from GaudiConfUtils.ConfigurableGenerators import FilterDesktop, CombineParticles, OfflineVertexFitter, DaVinci__N3BodyDecays
+from GaudiConfUtils.ConfigurableGenerators import AddRelatedInfo
+from Configurables import RelInfoConeVariables, RelInfoVertexIsolation
+from Gaudi.Configuration import *
+__author__ = ['F. Blanc', "A. Venkateswaran"]
+__date__ = '13/07/2023'
+__version__ = '$Revision: 0.1$'
+
+# Stripping line for B->tau nu
+
+from GaudiKernel.SystemOfUnits import MeV
+from GaudiKernel.SystemOfUnits import mm
+"""
+  B->TauNu
+"""
+__all__ = ('B2TauNuConf', 'getRelInfoB2TauNu', 'default_config')
+
+default_config = {
+    'NAME': 'B2TauNu',
+    'BUILDERTYPE': 'B2TauNuConf',
+    'WGs': ['RD'],
+    'CONFIG': {
+        'SpdMult': '600',
+        #
+        'FDZ_MIN_B': 10,
+        'FDZ_MAX_B': 100,
+        'PT_MIN_B': 2500 * MeV,
+        'P_MIN_B': 10000 * MeV,
+        'FDCHI2_MIN_B': 16,
+        'VTXCHI2_MAX_B': 6,
+        'BPVDIRA_MIN_B': 0.9,
+        'MASS_MIN_pipipi': 400 * MeV,
+        'MASS_MAX_pipipi': 2100 * MeV,
+        'MVIS_MIN_B': 500 * MeV,
+        'MVIS_MAX_B': 1900 * MeV,
+        'CORRM_MIN_B': 1900 * MeV,
+        'CORRM_MAX_B': 10000 * MeV,
+        #
+        'COMB_PT': 800 * MeV,
+        'COMB_NUMPT': 1000 * MeV,
+        'COMB_MAXDOCA': 0.2 * mm,
+        #
+        'PT_Pion': 250 * MeV,
+        'P_Pion': 1000 * MeV,
+        'IPCHI2_Tr': 16,
+        'TRACKCHI2_Tr': 4,
+        'TRGHOPROB_Tr': 0.4,
+        'PROBNNPI_Pion': 0.55,
+        #
+        'B2TauNu_LineNoPrescale': 1,
+        'B2TauNu_LinePrescale': 0.1,
+        'B2TauNu_LinePostscale': 1,
+        #
+        'B2TauNu_HLTSelection': "HLT_PASS_RE('Hlt1(Two)?Track.*Decision')"
+    },
+    'STREAMS': ['Bhadron']
+}
+
+
+class B2TauNuConf(LineBuilder):
+    """
+      Builder for B->TauNu
+    """
+
+    __configuration_keys__ = default_config['CONFIG'].keys()
+
+    def __init__(self, name, config):
+
+        LineBuilder.__init__(self, name, config)
+
+        trackCuts = "(MIPCHI2DV(PRIMARY) > %(IPCHI2_Tr)s) & (TRCHI2DOF < %(TRACKCHI2_Tr)s) & (TRGHOSTPROB < %(TRGHOPROB_Tr)s)" % config
+        trackCuts += " & (PT > %(PT_Pion)s) & (P > %(P_Pion)s)" % config
+        PionCuts = trackCuts + " & (PROBNNpi > %(PROBNNPI_Pion)s)" % config
+
+        self.FilterSPD = {
+            'Code':
+            " ( recSummary(LHCb.RecSummary.nSPDhits,'Raw/Spd/Digits') < %(SpdMult)s )"
+            % config,
+            'Preambulo': [
+                "from LoKiNumbers.decorators import *",
+                "from LoKiCore.basic import LHCb"
+                ]
+            }
+
+        self.selPions = SimpleSelection(
+            "Pions" + name, FilterDesktop, [StdLoosePions], Code=PionCuts)
+        self.selTracks = SimpleSelection(
+            "Tracks" + name, FilterDesktop, [StdNoPIDsPions], Code=trackCuts)
+
+        self.selB2TauNu = self._makeB2TauNu(name, self.selPions, config)
+        self.selB2TauNuNoPID = self._makeB2TauNu(name + "NoPID", self.selTracks, config)
+        self.selB2TauNuSS = self._makeB2TauNu(name + "SS", self.selPions, config, SS=True)
+
+        # Finished making selections build and register lines
+
+        self.B2TauNu_Line = self._makeLine("B2TauNuLine",
+                                           self.selB2TauNu, config)
+        self.B2TauNuNoPID_Line = self._makeLine("B2TauNuNoPIDLine",
+                                                self.selB2TauNuNoPID, config, PreScale=True)
+        self.B2TauNuSS_Line = self._makeLine("B2TauNuSSLine",
+                                             self.selB2TauNuSS, config)
+
+# Make B candidate selection
+
+    def _makeB2TauNu(self, name, tracks, config, SS=False):
+
+        combcut = "(APT>%(COMB_PT)s) & (AMAXDOCA('')<%(COMB_MAXDOCA)s) & (ANUM(PT > %(COMB_NUMPT)s) >= 1)" % config
+        combcut += " & (in_range ( %(MASS_MIN_pipipi)s, AM, %(MASS_MAX_pipipi)s ) )" % config
+
+        mothercut = "( BPVVDCHI2 > %(FDCHI2_MIN_B)s ) & ( BPVVDZ > %(FDZ_MIN_B)s*mm ) & ( BPVVDZ < %(FDZ_MAX_B)s*mm )" % config
+        mothercut += " & (PT > %(PT_MIN_B)s) & (P > %(P_MIN_B)s) & (M > %(MVIS_MIN_B)s) & (M < %(MVIS_MAX_B)s) " % config
+        mothercut += " & (VFASPF(VCHI2) < %(VTXCHI2_MAX_B)s) & (BPVDIRA > %(BPVDIRA_MIN_B)s )" % config
+        mothercut += " & (BPVCORRM > %(CORRM_MIN_B)s ) & (BPVCORRM < %(CORRM_MAX_B)s ) " % config
+
+        descriptors = ["[B+ -> pi+ pi- pi+]cc"]
+        if SS:
+            descriptors = ["[B+ -> pi+ pi+ pi+]cc"]
+
+        Combine = DaVinci__N3BodyDecays(
+            DecayDescriptors=descriptors,
+            Combination12Cut="AM < 1800",
+            CombinationCut=combcut,
+            MotherCut=mothercut)
+
+        presel = Selection(name, Algorithm=Combine, RequiredSelections=[tracks])
+
+        # determine vertex isolation variables, and add them to the candidate for further selection
+        tool          = RelInfoVertexIsolation('TauIsoInfo')
+        algo          = AddRelatedInfo()
+        algo.Inputs   = [presel.outputLocation()]
+        algo.Location = 'Iso'
+        algo.Tool     = tool.getFullName()
+        tau_withinfo  = PassThroughSelection(name+'TauInfo', Algorithm=algo, RequiredSelection=presel)
+        loc  = presel.outputLocation().replace('/Particles', '/Iso')
+        # apply selection based on isolation variables
+        sel = SimpleSelection(name+"IsoFilter",
+                              FilterDesktop,
+                              [tau_withinfo],
+                              Code="RELINFO('"+loc+"', 'VTXISODCHI2ONETRACK', 100000.) >25")
+
+        return sel
+
+# Helpers to make lines
+
+    def _makeLine(self, name, sel, config, PreScale=False):
+
+        pscale = config['B2TauNu_LineNoPrescale']
+        if PreScale:
+            pscale = config['B2TauNu_LinePrescale']
+
+        line = StrippingLine(
+            name,
+            prescale=pscale,
+            postscale=config['B2TauNu_LinePostscale'],
+            # MDSTFlag = False,
+            FILTER=self.FilterSPD,
+            HLT1=config['B2TauNu_HLTSelection'],
+            RelatedInfoTools=getRelInfoB2TauNu(),
+            selection=sel,
+            MaxCandidates=50)
+        self.registerLine(line)
+        return line
+
+# Related Info
+def getRelInfoB2TauNu():
+
+    relInfo = []
+    for coneAngle in [0.5, 0.8, 1.0, 1.3, 1.5]:
+        conestr = str(coneAngle).replace('.', '')
+        relInfo += [{
+            "Type":
+            "RelInfoConeVariables",
+            "IgnoreUnmatchedDescriptors":
+            True,
+            "ConeAngle":
+            coneAngle,
+            "Location":
+            "VertexConeInfo",
+            "Variables": [
+                          'CONEANGLE', 'CONEMULT', 'CONEPASYM', 'CONEPTASYM',
+                          'CONEDELTAETA'
+                          ],
+            "DaughterLocations": {
+                                  "^[Beauty -> X+ X- X+]CC": 'P2ConeVar%s_B' % conestr
+                                  }
+            }]
+    relInfo += [{
+        "Type": "RelInfoVertexIsolation",
+        "Location": "VertexIsoInfo"
+        }]
+    return relInfo

Phys/StrippingSelections/python/StrippingSelections/StrippingRD/__init__.py

@@ -31,7 +31,8 @@ _selections = ('StrippingRareBaryonicMuMu',
                'StrippingB2KstTauTau', 
                'StrippingS2Hyperons',
                'StrippingB2XTauTau.py',
-               'StrippingB2STauTauInclusive.py'
+               'StrippingB2STauTauInclusive.py',
+               'StrippingB2TauNu.py'
         )
 
 for _sel in _selections :