Thor functors for MCTrackInfo

DaVinciExamples/python/DaVinciExamples/tupling/example-tupling-mctrack-hlt.py

+###############################################################################
+# (c) Copyright 2022 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.                                       #
+###############################################################################
+# Run this example in this diractory with:
+# ../../../../run davinci \
+# --inputfiledb test_hlt1_trigger_decisions ../../../../Phys/DaVinci/options/DaVinciDB-Example.yaml \
+# --joboptfile example-tupling-mctrack-hlt.yaml \
+# --user_algorithms example-tupling-mctrack-hlt:main
+
+from FunTuple import FunctorCollection
+import Functors as F
+from FunTuple import FunTuple_Particles as Funtuple
+from DaVinci import options
+from DaVinci.truth_matching import configured_MCTruthAndBkgCatAlg
+from DaVinci.algorithms import add_filter
+from PyConf.components import force_location
+
+from DaVinci.mc_track import MCP2MCTrackInfo
+
+
+def main():
+
+    # Input
+    B_data = force_location(
+        "/Event/HLT2/Hlt2B2OC_BdToDmPi_DmToPimPimKp_Line/Particles")
+
+    # Filter
+    filter_Dpi = add_filter(options, "HDRFilter",
+                            "HLT_PASS('Hlt2B2OC_BdToDmPi_DmToPimPimKp_Line')")
+
+    # Truth matching
+    mctruth = configured_MCTruthAndBkgCatAlg(
+        inputs=B_data,
+        filter_MCP=True,
+        # output_level=1,
+        root_in_tes="/Event/HLT2")
+    MC_TRUTH = lambda func: F.MAP_INPUT(Functor=func, Relations=mctruth.MCAssocTable)
+
+    # Tracks tools
+    MCP2MCTRACKINFO = MCP2MCTrackInfo(mctruth.TruthMatchedMCP, process='Hlt2')
+
+    # Variables
+    variables_all = FunctorCollection({
+        # REC
+        "P":
+        F.P,
+        "PT":
+        F.PT,
+        "PX":
+        F.PX,
+        "PY":
+        F.PY,
+        "PZ":
+        F.PZ,
+        "ENERGY":
+        F.ENERGY,
+        # TRUTH MATCHING
+        "TRUEP":
+        MC_TRUTH(F.P),
+        "TRUEPT":
+        MC_TRUTH(F.PT),
+        "TRUEPX":
+        MC_TRUTH(F.PX),
+        "TRUEPY":
+        MC_TRUTH(F.PY),
+        "TRUEPZ":
+        MC_TRUTH(F.PZ),
+        "TRUEENERGY":
+        MC_TRUTH(F.ENERGY),
+        # TRUE TRACK
+        "MC_HAST":
+        MC_TRUTH(MCP2MCTRACKINFO.HasT),
+        "MC_HASTT":
+        MC_TRUTH(MCP2MCTRACKINFO.HasTT),
+        "MC_HASVELO":
+        MC_TRUTH(MCP2MCTRACKINFO.HasVelo),
+        "MC_ACCT":
+        MC_TRUTH(MCP2MCTRACKINFO.AccT),
+        "MC_ACCTT":
+        MC_TRUTH(MCP2MCTRACKINFO.AccTT),
+        "MC_ACCVELO":
+        MC_TRUTH(MCP2MCTRACKINFO.AccVelo),
+        # RECONSTRUCTIBLE CATEGORY
+        "MC_RECONSTRUCTIBLE":
+        MC_TRUTH(MCP2MCTRACKINFO.Reconstructible)
+    })
+
+    variables = {'ALL': variables_all}
+
+    # Define fields
+    fields = {
+        'B0': '[[B0]CC -> (D- -> K+ pi- pi-) pi+]CC',
+        'D': '[[B0]CC -> ^(D- -> K+ pi- pi-) pi+]CC',
+        'pi': '[[B0]CC -> (D- -> K+ pi- pi-) ^pi+]CC',
+        'Kplus': '[[B0]CC -> (D- -> ^K+ pi- pi-) pi+]CC',
+        'piminus_1': '[[B0]CC -> (D- -> K+ ^pi- pi-) pi+]CC',
+        'piminus_2': '[[B0]CC -> (D- -> K+ pi- ^pi-) pi+]CC'
+    }
+
+    # Make tuple algorithm
+    tuple_Dpi = Funtuple(
+        name="Dpi",
+        tuple_name="DecayTree",
+        fields=fields,
+        variables=variables,
+        inputs=B_data)
+
+    # Run
+    return {"KstG": [filter_Dpi, tuple_Dpi]}, []

DaVinciExamples/python/DaVinciExamples/tupling/example-tupling-mctrack-hlt.yaml

+###############################################################################
+# (c) Copyright 2020-2021 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.                                       #
+###############################################################################
+
+evt_max: 50
+annsvc_config: 'root://eoslhcb.cern.ch//eos/lhcb/wg/dpa/wp3/tests/hlt2_integration_B0_100.tck.json'
+histo_file: 'DV_histo_example-tupling-mctrack-hlt.root'
+ntuple_file: 'DV_tuple_example-tupling-mctrack-hlt.root'
+lumi: false
+print_freq: 1
+process: 'Hlt2'

DaVinciExamples/python/DaVinciExamples/tupling/example-tupling-mctrack-mctuple-hlt.py

+###############################################################################
+# (c) Copyright 2022 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.                                       #
+###############################################################################
+# Run this example in this diractory with:
+# ../../../../run davinci \
+# --inputfiledb test_hlt1_trigger_decisions ../../../../Phys/DaVinci/options/DaVinciDB-Example.yaml \
+# --joboptfile example-tupling-mctrack-hlt.yaml \
+# --user_algorithms example-tupling-mctrack-hlt:main
+
+from FunTuple import FunctorCollection
+import Functors as F
+from FunTuple import FunTuple_MCParticles as MCFuntuple
+from DaVinci import options
+from DaVinci.algorithms import add_filter
+from PyConf.components import force_location
+
+from DaVinci.mc_track import MCP2MCTrackInfo, MCP2Track
+
+
+def main():
+
+    # Input
+    MC_data = force_location("/Event/HLT2/MC/Particles")
+
+    # Filter
+    filter_Dpi = add_filter(options, "HDRFilter",
+                            "HLT_PASS('Hlt2B2OC_BdToDmPi_DmToPimPimKp_Line')")
+
+    # Tracks tools
+    MCP2MCTRACKINFO = MCP2MCTrackInfo(MC_data, process='Hlt2')
+    MCP2TRACK = MCP2Track(MC_data, process='Hlt2')
+
+    # Variables
+    variables_all = FunctorCollection({
+        # REC
+        "P":
+        F.P,
+        "PT":
+        F.PT,
+        "PX":
+        F.PX,
+        "PY":
+        F.PY,
+        "PZ":
+        F.PZ,
+        "ENERGY":
+        F.ENERGY,
+        # RECONSTRUCTED TRACK
+        "MC_RecoTrack_TYPE":
+        MCP2TRACK.Type,
+        "MC_RecoTrack_REF_X":
+        MCP2TRACK.ReferencePoint_X,
+        "MC_RecoTrack_REF_Y":
+        MCP2TRACK.ReferencePoint_Y,
+        "MC_RecoTrack_REF_Z":
+        MCP2TRACK.ReferencePoint_Z,
+        "MC_RecoTrack_NDOF":
+        MCP2TRACK.nDoF,
+        "MC_RecoTrack_GHOSTPROB":
+        MCP2TRACK.GhostProbability,
+        "MC_RecoTrack_FLAG":
+        MCP2TRACK.Flag,
+        "MC_RecoTrack_HAST":
+        MCP2TRACK.HasT,
+        "MC_RecoTrack_HASUT":
+        MCP2TRACK.HasUT,
+        "MC_RecoTrack_HASVELO":
+        MCP2TRACK.HasVelo,
+        "MC_RecoTrack_NHITS":
+        MCP2TRACK.nHits,
+        "MC_RecoTrack_NVPHITS":
+        MCP2TRACK.nVPHits,
+        "MC_RecoTrack_NUTHITS":
+        MCP2TRACK.nUTHits,
+        "MC_RecoTrack_NFTHITS":
+        MCP2TRACK.nFTHits,
+        # TRUE TRACK
+        "MC_HAST":
+        MCP2MCTRACKINFO.HasT,
+        "MC_HASTT":
+        MCP2MCTRACKINFO.HasTT,
+        "MC_HASVELO":
+        MCP2MCTRACKINFO.HasVelo,
+        "MC_ACCT":
+        MCP2MCTRACKINFO.AccT,
+        "MC_ACCTT":
+        MCP2MCTRACKINFO.AccTT,
+        "MC_ACCVELO":
+        MCP2MCTRACKINFO.AccVelo,
+        # RECONSTRUCTED CATEGORY AND RECONSTRUCTIBLE CATEGORY
+        "MC_RECONSTRUCTED":
+        MCP2TRACK.Reconstructed,
+        "MC_RECONSTRUCTIBLE":
+        MCP2MCTRACKINFO.Reconstructible,
+    })
+
+    variables = {'ALL': variables_all}
+
+    # Define fields
+    fields = {
+        'B0': '[[B0]CC -> (D- -> K+ pi- pi-) pi+]CC',
+        'D': '[[B0]CC -> ^(D- -> K+ pi- pi-) pi+]CC',
+        'pi': '[[B0]CC -> (D- -> K+ pi- pi-) ^pi+]CC',
+        'Kplus': '[[B0]CC -> (D- -> ^K+ pi- pi-) pi+]CC',
+        'piminus_1': '[[B0]CC -> (D- -> K+ ^pi- pi-) pi+]CC',
+        'piminus_2': '[[B0]CC -> (D- -> K+ pi- ^pi-) pi+]CC'
+    }
+
+    # Make tuple algorithm
+    tuple_Dpi = MCFuntuple(
+        name="DpiMC",
+        tuple_name="DecayTree",
+        fields=fields,
+        variables=variables,
+        inputs=MC_data)
+
+    # Run
+    return {"KstG": [filter_Dpi, tuple_Dpi]}, []

DaVinciExamples/python/DaVinciExamples/tupling/example-tupling-mctrack-mctuple-hlt.yaml

+###############################################################################
+# (c) Copyright 2020-2021 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.                                       #
+###############################################################################
+
+evt_max: 50
+annsvc_config: 'root://eoslhcb.cern.ch//eos/lhcb/wg/dpa/wp3/tests/hlt2_integration_B0_100.tck.json'
+histo_file: 'DV_histo_example-tupling-mctrack-mctuple-hlt.root'
+ntuple_file: 'DV_tuple_example-tupling-mctrack-mctuple-hlt.root'
+lumi: false
+print_freq: 1
+process: 'Hlt2'

DaVinciExamples/tests/qmtest/tupling.qms/test_davinci_tupling_mctrack_hlt.qmt

+<?xml version="1.0" ?>
+<!--
+###############################################################################
+# (c) Copyright 2021-2022 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.                                       #
+###############################################################################
+-->
+<!DOCTYPE extension  PUBLIC '-//QM/2.3/Extension//EN'  'http://www.codesourcery.com/qm/dtds/2.3/-//qm/2.3/extension//en.dtd'>
+<!--
+#######################################################
+# SUMMARY OF THIS TEST
+# ...................
+# Author: Jiahui Zhuo
+# Purpose: Test for MCTrack related functor in HLT2 process
+# Prerequisites: None
+# inputfiledb test_hlt1_trigger_decisions $DAVINCIROOT/options/DaVinciDB-Example.yaml
+# user_algorithms ../../python/DaVinciExamples/tupling/example-tupling-mctrack-hlt:main
+#######################################################
+-->
+<extension class="GaudiTest.GaudiExeTest" kind="test">
+  <argument name="program"><text>davinci</text></argument>
+  <argument name="args"><set>
+  <text>--inputfiledb</text>
+  <text>test_hlt1_trigger_decisions</text>
+  <text>$DAVINCIROOT/options/DaVinciDB-Example.yaml</text>
+  <text>--joboptfile</text>
+  <text>../../python/DaVinciExamples/tupling/example-tupling-mctrack-hlt.yaml</text>
+  <text>--user_algorithms</text>
+  <text>../../python/DaVinciExamples/tupling/example-tupling-mctrack-hlt:main</text>
+  </set></argument>
+  <argument name="validator"><text>
+findReferenceBlock("""
+RFileCnv                               INFO dumping contents of /NTUPLES/FILE1
+TFile: name=DV_tuple_example-tupling-mctrack-hlt.root, title=Gaudi Trees, option=CREATE
+******************************************************************************
+*Tree    :DecayTree : DecayTree                                              *
+""", stdout, result, causes, signature_offset = 0)
+
+import sys, os
+from ROOT import TFile
+
+B_vars_stored =['B0_ENERGY', 'B0_MC_ACCT', 'B0_MC_ACCTT', 'B0_MC_ACCVELO', 'B0_MC_HAST', 'B0_MC_HASTT', 'B0_MC_HASVELO', 'B0_MC_RECONSTRUCTIBLE', 'B0_P', 'B0_PT', 'B0_PX', 'B0_PY', 'B0_PZ', 'B0_TRUEENERGY', 'B0_TRUEP', 'B0_TRUEPT', 'B0_TRUEPX', 'B0_TRUEPY', 'B0_TRUEPZ', 'D_ENERGY', 'D_MC_ACCT', 'D_MC_ACCTT', 'D_MC_ACCVELO', 'D_MC_HAST', 'D_MC_HASTT', 'D_MC_HASVELO', 'D_MC_RECONSTRUCTIBLE', 'D_P', 'D_PT', 'D_PX', 'D_PY', 'D_PZ', 'D_TRUEENERGY', 'D_TRUEP', 'D_TRUEPT', 'D_TRUEPX', 'D_TRUEPY', 'D_TRUEPZ', 'Kplus_ENERGY', 'Kplus_MC_ACCT', 'Kplus_MC_ACCTT', 'Kplus_MC_ACCVELO', 'Kplus_MC_HAST', 'Kplus_MC_HASTT', 'Kplus_MC_HASVELO', 'Kplus_MC_RECONSTRUCTIBLE', 'Kplus_P', 'Kplus_PT', 'Kplus_PX', 'Kplus_PY', 'Kplus_PZ', 'Kplus_TRUEENERGY', 'Kplus_TRUEP', 'Kplus_TRUEPT', 'Kplus_TRUEPX', 'Kplus_TRUEPY', 'Kplus_TRUEPZ', 'pi_ENERGY', 'pi_MC_ACCT', 'pi_MC_ACCTT', 'pi_MC_ACCVELO', 'pi_MC_HAST', 'pi_MC_HASTT', 'pi_MC_HASVELO', 'pi_MC_RECONSTRUCTIBLE', 'pi_P', 'pi_PT', 'pi_PX', 'pi_PY', 'pi_PZ', 'pi_TRUEENERGY', 'pi_TRUEP', 'pi_TRUEPT', 'pi_TRUEPX', 'pi_TRUEPY', 'pi_TRUEPZ', 'piminus_1_ENERGY', 'piminus_1_MC_ACCT', 'piminus_1_MC_ACCTT', 'piminus_1_MC_ACCVELO', 'piminus_1_MC_HAST', 'piminus_1_MC_HASTT', 'piminus_1_MC_HASVELO', 'piminus_1_MC_RECONSTRUCTIBLE', 'piminus_1_P', 'piminus_1_PT', 'piminus_1_PX', 'piminus_1_PY', 'piminus_1_PZ', 'piminus_1_TRUEENERGY', 'piminus_1_TRUEP', 'piminus_1_TRUEPT', 'piminus_1_TRUEPX', 'piminus_1_TRUEPY', 'piminus_1_TRUEPZ', 'piminus_2_ENERGY', 'piminus_2_MC_ACCT', 'piminus_2_MC_ACCTT', 'piminus_2_MC_ACCVELO', 'piminus_2_MC_HAST', 'piminus_2_MC_HASTT', 'piminus_2_MC_HASVELO', 'piminus_2_MC_RECONSTRUCTIBLE', 'piminus_2_P', 'piminus_2_PT', 'piminus_2_PX', 'piminus_2_PY', 'piminus_2_PZ', 'piminus_2_TRUEENERGY', 'piminus_2_TRUEP', 'piminus_2_TRUEPT', 'piminus_2_TRUEPX', 'piminus_2_TRUEPY', 'piminus_2_TRUEPZ']
+
+#sort the expected vars
+B_vars_stored = sorted(B_vars_stored)
+
+#open the TFile and TTree
+ntuple = './DV_tuple_example-tupling-mctrack-hlt.root'
+if not os.path.isfile(ntuple): raise Exception(f"File: {ntuple} does not exist!")
+f      = TFile.Open(ntuple)
+t_B    = f.Get('Dpi/DecayTree')
+
+#sort the stores vars
+b_names = sorted([b.GetName() for b in t_B.GetListOfLeaves()])
+
+B_excluded_1 = set(B_vars_stored) - set(b_names)
+B_excluded_2 = set(b_names) - set(B_vars_stored)
+if len(B_excluded_1) != 0: raise Exception('Number of stored variables is less than what is expected. The extra variables expected are: ' , B_excluded_1)
+if len(B_excluded_2) != 0: raise Exception('Number of stored variables is greater than what is expected. The extra variables stored are: ', B_excluded_2)
+
+f.Close()
+print('Test successfully completed!')
+os.system(f"rm {ntuple}")
+countErrorLines({"FATAL":0, "ERROR":0})
+  </text></argument>
+</extension>

DaVinciExamples/tests/qmtest/tupling.qms/test_davinci_tupling_mctrack_mctuple_hlt.qmt

+<?xml version="1.0" ?>
+<!--
+###############################################################################
+# (c) Copyright 2021-2022 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.                                       #
+###############################################################################
+-->
+<!DOCTYPE extension  PUBLIC '-//QM/2.3/Extension//EN'  'http://www.codesourcery.com/qm/dtds/2.3/-//qm/2.3/extension//en.dtd'>
+<!--
+#######################################################
+# SUMMARY OF THIS TEST
+# ...................
+# Author: Jiahui Zhuo
+# Purpose: Test for MCTrack related functor in HLT2 process and FunTuple_MCParticles
+# Prerequisites: None
+# inputfiledb test_hlt1_trigger_decisions $DAVINCIROOT/options/DaVinciDB-Example.yaml
+# user_algorithms ../../python/DaVinciExamples/tupling/example-tupling-mctrack-mctuple-hlt:main
+#######################################################
+-->
+<extension class="GaudiTest.GaudiExeTest" kind="test">
+  <argument name="program"><text>davinci</text></argument>
+  <argument name="args"><set>
+  <text>--inputfiledb</text>
+  <text>test_hlt1_trigger_decisions</text>
+  <text>$DAVINCIROOT/options/DaVinciDB-Example.yaml</text>
+  <text>--joboptfile</text>
+  <text>../../python/DaVinciExamples/tupling/example-tupling-mctrack-mctuple-hlt.yaml</text>
+  <text>--user_algorithms</text>
+  <text>../../python/DaVinciExamples/tupling/example-tupling-mctrack-mctuple-hlt:main</text>
+  </set></argument>
+  <argument name="validator"><text>
+findReferenceBlock("""
+RFileCnv                               INFO dumping contents of /NTUPLES/FILE1
+TFile: name=DV_tuple_example-tupling-mctrack-mctuple-hlt.root, title=Gaudi Trees, option=CREATE
+******************************************************************************
+*Tree    :DecayTree : DecayTree                                              *
+""", stdout, result, causes, signature_offset = 0)
+
+import sys, os
+from ROOT import TFile
+
+B_vars_stored =['B0_ENERGY', 'B0_MC_ACCT', 'B0_MC_ACCTT', 'B0_MC_ACCVELO', 'B0_MC_HAST', 'B0_MC_HASTT', 'B0_MC_HASVELO', 'B0_MC_RECONSTRUCTED', 'B0_MC_RECONSTRUCTIBLE', 'B0_MC_RecoTrack_FLAG', 'B0_MC_RecoTrack_GHOSTPROB', 'B0_MC_RecoTrack_HAST', 'B0_MC_RecoTrack_HASUT', 'B0_MC_RecoTrack_HASVELO', 'B0_MC_RecoTrack_NDOF', 'B0_MC_RecoTrack_NFTHITS', 'B0_MC_RecoTrack_NHITS', 'B0_MC_RecoTrack_NUTHITS', 'B0_MC_RecoTrack_NVPHITS', 'B0_MC_RecoTrack_REF_X', 'B0_MC_RecoTrack_REF_Y', 'B0_MC_RecoTrack_REF_Z', 'B0_MC_RecoTrack_TYPE', 'B0_P', 'B0_PT', 'B0_PX', 'B0_PY', 'B0_PZ', 'D_ENERGY', 'D_MC_ACCT', 'D_MC_ACCTT', 'D_MC_ACCVELO', 'D_MC_HAST', 'D_MC_HASTT', 'D_MC_HASVELO', 'D_MC_RECONSTRUCTED', 'D_MC_RECONSTRUCTIBLE', 'D_MC_RecoTrack_FLAG', 'D_MC_RecoTrack_GHOSTPROB', 'D_MC_RecoTrack_HAST', 'D_MC_RecoTrack_HASUT', 'D_MC_RecoTrack_HASVELO', 'D_MC_RecoTrack_NDOF', 'D_MC_RecoTrack_NFTHITS', 'D_MC_RecoTrack_NHITS', 'D_MC_RecoTrack_NUTHITS', 'D_MC_RecoTrack_NVPHITS', 'D_MC_RecoTrack_REF_X', 'D_MC_RecoTrack_REF_Y', 'D_MC_RecoTrack_REF_Z', 'D_MC_RecoTrack_TYPE', 'D_P', 'D_PT', 'D_PX', 'D_PY', 'D_PZ', 'Kplus_ENERGY', 'Kplus_MC_ACCT', 'Kplus_MC_ACCTT', 'Kplus_MC_ACCVELO', 'Kplus_MC_HAST', 'Kplus_MC_HASTT', 'Kplus_MC_HASVELO', 'Kplus_MC_RECONSTRUCTED', 'Kplus_MC_RECONSTRUCTIBLE', 'Kplus_MC_RecoTrack_FLAG', 'Kplus_MC_RecoTrack_GHOSTPROB', 'Kplus_MC_RecoTrack_HAST', 'Kplus_MC_RecoTrack_HASUT', 'Kplus_MC_RecoTrack_HASVELO', 'Kplus_MC_RecoTrack_NDOF', 'Kplus_MC_RecoTrack_NFTHITS', 'Kplus_MC_RecoTrack_NHITS', 'Kplus_MC_RecoTrack_NUTHITS', 'Kplus_MC_RecoTrack_NVPHITS', 'Kplus_MC_RecoTrack_REF_X', 'Kplus_MC_RecoTrack_REF_Y', 'Kplus_MC_RecoTrack_REF_Z', 'Kplus_MC_RecoTrack_TYPE', 'Kplus_P', 'Kplus_PT', 'Kplus_PX', 'Kplus_PY', 'Kplus_PZ', 'pi_ENERGY', 'pi_MC_ACCT', 'pi_MC_ACCTT', 'pi_MC_ACCVELO', 'pi_MC_HAST', 'pi_MC_HASTT', 'pi_MC_HASVELO', 'pi_MC_RECONSTRUCTED', 'pi_MC_RECONSTRUCTIBLE', 'pi_MC_RecoTrack_FLAG', 'pi_MC_RecoTrack_GHOSTPROB', 'pi_MC_RecoTrack_HAST', 'pi_MC_RecoTrack_HASUT', 'pi_MC_RecoTrack_HASVELO', 'pi_MC_RecoTrack_NDOF', 'pi_MC_RecoTrack_NFTHITS', 'pi_MC_RecoTrack_NHITS', 'pi_MC_RecoTrack_NUTHITS', 'pi_MC_RecoTrack_NVPHITS', 'pi_MC_RecoTrack_REF_X', 'pi_MC_RecoTrack_REF_Y', 'pi_MC_RecoTrack_REF_Z', 'pi_MC_RecoTrack_TYPE', 'pi_P', 'pi_PT', 'pi_PX', 'pi_PY', 'pi_PZ', 'piminus_1_ENERGY', 'piminus_1_MC_ACCT', 'piminus_1_MC_ACCTT', 'piminus_1_MC_ACCVELO', 'piminus_1_MC_HAST', 'piminus_1_MC_HASTT', 'piminus_1_MC_HASVELO', 'piminus_1_MC_RECONSTRUCTED', 'piminus_1_MC_RECONSTRUCTIBLE', 'piminus_1_MC_RecoTrack_FLAG', 'piminus_1_MC_RecoTrack_GHOSTPROB', 'piminus_1_MC_RecoTrack_HAST', 'piminus_1_MC_RecoTrack_HASUT', 'piminus_1_MC_RecoTrack_HASVELO', 'piminus_1_MC_RecoTrack_NDOF', 'piminus_1_MC_RecoTrack_NFTHITS', 'piminus_1_MC_RecoTrack_NHITS', 'piminus_1_MC_RecoTrack_NUTHITS', 'piminus_1_MC_RecoTrack_NVPHITS', 'piminus_1_MC_RecoTrack_REF_X', 'piminus_1_MC_RecoTrack_REF_Y', 'piminus_1_MC_RecoTrack_REF_Z', 'piminus_1_MC_RecoTrack_TYPE', 'piminus_1_P', 'piminus_1_PT', 'piminus_1_PX', 'piminus_1_PY', 'piminus_1_PZ', 'piminus_2_ENERGY', 'piminus_2_MC_ACCT', 'piminus_2_MC_ACCTT', 'piminus_2_MC_ACCVELO', 'piminus_2_MC_HAST', 'piminus_2_MC_HASTT', 'piminus_2_MC_HASVELO', 'piminus_2_MC_RECONSTRUCTED', 'piminus_2_MC_RECONSTRUCTIBLE', 'piminus_2_MC_RecoTrack_FLAG', 'piminus_2_MC_RecoTrack_GHOSTPROB', 'piminus_2_MC_RecoTrack_HAST', 'piminus_2_MC_RecoTrack_HASUT', 'piminus_2_MC_RecoTrack_HASVELO', 'piminus_2_MC_RecoTrack_NDOF', 'piminus_2_MC_RecoTrack_NFTHITS', 'piminus_2_MC_RecoTrack_NHITS', 'piminus_2_MC_RecoTrack_NUTHITS', 'piminus_2_MC_RecoTrack_NVPHITS', 'piminus_2_MC_RecoTrack_REF_X', 'piminus_2_MC_RecoTrack_REF_Y', 'piminus_2_MC_RecoTrack_REF_Z', 'piminus_2_MC_RecoTrack_TYPE', 'piminus_2_P', 'piminus_2_PT', 'piminus_2_PX', 'piminus_2_PY', 'piminus_2_PZ']
+
+#sort the expected vars
+B_vars_stored = sorted(B_vars_stored)
+
+#open the TFile and TTree
+ntuple = './DV_tuple_example-tupling-mctrack-mctuple-hlt.root'
+if not os.path.isfile(ntuple): raise Exception(f"File: {ntuple} does not exist!")
+f      = TFile.Open(ntuple)
+t_B    = f.Get('DpiMC/DecayTree')
+
+#sort the stores vars
+b_names = sorted([b.GetName() for b in t_B.GetListOfLeaves()])
+
+B_excluded_1 = set(B_vars_stored) - set(b_names)
+B_excluded_2 = set(b_names) - set(B_vars_stored)
+if len(B_excluded_1) != 0: raise Exception('Number of stored variables is less than what is expected. The extra variables expected are: ' , B_excluded_1)
+if len(B_excluded_2) != 0: raise Exception('Number of stored variables is greater than what is expected. The extra variables stored are: ', B_excluded_2)
+
+f.Close()
+print('Test successfully completed!')
+os.system(f"rm {ntuple}")
+countErrorLines({"FATAL":0, "ERROR":0})
+  </text></argument>
+</extension>

Phys/DaVinci/python/DaVinci/mc_track.py

+###############################################################################
+# (c) Copyright 2022 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 Gaudi.Configuration import INFO
+from Functors.grammar import BoundFunctor
+from DaVinci.data_from_file import make_mc_track_info
+from PyConf.components import force_location
+from PyConf.Algorithms import MC2MCTrackInfoAlg, MC2TrackAlg
+import Functors as F
+
+
+class MCP2MCTrackInfo:
+    """
+    MCP2MCTrackInfo helper class, will create all possible True TrackInfo related functors and can be
+    accessed as the standard python class member.
+
+    Args:
+        mcparticles_input : MCParticle data handle (TES)
+        mc_track_info (optional) : MCTrackInfo handle (TES), default = make_mc_track_info()
+        process       (optional) : Process type (Hlt2 or Spruce), default = 'Hlt2'
+        root_in_tes   (optional) : Standard Gaudi Algorithm RootInTES location
+        output_level  (optional) : Standard Gaudi Algorithm OutputLevel
+
+    Example:
+        MC_TRINFO = MC_TrackInfo(MCParticles)
+        allvariables['HAST'] = MC_TRINFO.HasT 
+
+    """
+
+    # Copied definition from /LHCb/Event/MCEvent/include/Event/MCTrackInfo.h
+    class _Flag:
+        VeloR = 0x00000001
+        VeloPhi = 0x00000002
+        TT1 = 0x00000004
+        TT2 = 0x00000008
+        T1X = 0x00000010
+        T1S = 0x00000020
+        T2X = 0x00000040
+        T2S = 0x00000080
+        T3X = 0x00000100
+        T3S = 0x00000200
+        HasVelo = VeloR | VeloPhi
+        HasTT = TT1 | TT2
+        HasT1 = T1X | T1S
+        HasT2 = T2X | T2S
+        HasT3 = T3X | T3S
+        HasT = HasT1 | HasT2 | HasT3
+        HasVeloAndT = HasVelo | HasT
+        AccVeloR = 0x00000400
+        AccVeloPhi = 0x00000800
+        AccTT1 = 0x00001000
+        AccTT2 = 0x00002000
+        AccT1X = 0x00004000
+        AccT1S = 0x00008000
+        AccT2X = 0x00010000
+        AccT2S = 0x00020000
+        AccT3X = 0x00040000
+        AccT3S = 0x00080000
+        AccVelo = AccVeloR | AccVeloPhi
+        AccTT = AccTT1 | AccTT2
+        AccT1 = AccT1X | AccT1S
+        AccT2 = AccT2X | AccT2S
+        AccT3 = AccT3X | AccT3S
+        AccT = AccT1 | AccT2 | AccT3
+        AccVeloAndT = AccVelo | AccT
+        multVeloR = 20
+        multVeloPhi = 25
+        MultVeloR = 0x01F00000
+        MultVeloPhi = 0x3E000000
+
+    def get_info(self, mc_property: int):
+        return F.MAP_INPUT(
+            Functor=F.MC_TRACKINFO(mc_property), Relations=self.MC2MCTrackInfo)
+
+    def __init__(self,
+                 mcparticles_input,
+                 mc_track_info=make_mc_track_info(),
+                 process='Spruce',
+                 output_level=INFO):
+
+        if process in ['Spruce', 'Turbo']:
+            raise ValueError(
+                "MCTrackInfo is not persisted in Spruce/Turbo at the moment, this will be fixed in the future, see: "
+                "https://gitlab.cern.ch/lhcb/Moore/-/issues/440")
+        elif process != 'Hlt2':
+            raise ValueError(
+                f"The specified 'process' {process} not recognised. Can only be 'Hlt2' or 'Spruce' or 'Turbo'."
+            )
+
+        # Load Algorithm
+        self.Algorithm = MC2MCTrackInfoAlg(
+            Input=mcparticles_input,
+            MCTrackInfo=mc_track_info,
+            OutputLevel=output_level)
+
+        # Alias output of algorithm
+        self.MC2MCTrackInfo = self.Algorithm.MC2MCTrackInfo
+        self.MC2Reconstructible = self.Algorithm.MC2Reconstructible
+
+        # Create functors for MC_TRACKINFO
+        self.VeloR = self.get_info(self._Flag.VeloR)
+        self.VeloPhi = self.get_info(self._Flag.VeloPhi)
+        self.TT1 = self.get_info(self._Flag.TT1)
+        self.TT2 = self.get_info(self._Flag.TT2)
+        self.T1X = self.get_info(self._Flag.T1X)
+        self.T1S = self.get_info(self._Flag.T1S)
+        self.T2X = self.get_info(self._Flag.T2X)
+        self.T2S = self.get_info(self._Flag.T2S)
+        self.T3X = self.get_info(self._Flag.T3X)
+        self.T3S = self.get_info(self._Flag.T3S)
+        self.HasVelo = self.get_info(self._Flag.HasVelo)
+        self.HasTT = self.get_info(self._Flag.HasTT)
+        self.HasT1 = self.get_info(self._Flag.HasT1)
+        self.HasT2 = self.get_info(self._Flag.HasT2)
+        self.HasT3 = self.get_info(self._Flag.HasT3)
+        self.HasT = self.get_info(self._Flag.HasT)
+        self.HasVeloAndT = self.get_info(self._Flag.HasVeloAndT)
+        self.AccVeloR = self.get_info(self._Flag.AccVeloR)
+        self.AccVeloPhi = self.get_info(self._Flag.AccVeloPhi)
+        self.AccTT1 = self.get_info(self._Flag.AccTT1)
+        self.AccTT2 = self.get_info(self._Flag.AccTT2)
+        self.AccT1X = self.get_info(self._Flag.AccT1X)
+        self.AccT1S = self.get_info(self._Flag.AccT1S)
+        self.AccT2X = self.get_info(self._Flag.AccT2X)
+        self.AccT2S = self.get_info(self._Flag.AccT2S)
+        self.AccT3X = self.get_info(self._Flag.AccT3X)
+        self.AccT3S = self.get_info(self._Flag.AccT3S)
+        self.AccVelo = self.get_info(self._Flag.AccVelo)
+        self.AccTT = self.get_info(self._Flag.AccTT)
+        self.AccT1 = self.get_info(self._Flag.AccT1)
+        self.AccT2 = self.get_info(self._Flag.AccT2)
+        self.AccT3 = self.get_info(self._Flag.AccT3)
+        self.AccT = self.get_info(self._Flag.AccT)
+        self.AccVeloAndT = self.get_info(self._Flag.AccVeloAndT)
+        self.multVeloR = self.get_info(self._Flag.multVeloR)
+        self.multVeloPhi = self.get_info(self._Flag.multVeloPhi)
+        self.MultVeloR = self.get_info(self._Flag.MultVeloR)
+        self.MultVeloPhi = self.get_info(self._Flag.MultVeloPhi)
+
+        # UT alias
+        self.UT1 = self.TT1
+        self.UT2 = self.TT2
+        self.HasUT = self.HasTT
+        self.AccUT1 = self.AccTT1
+        self.AccUT2 = self.AccTT2
+        self.AccUT = self.AccTT
+
+        # Create functors for MC_PROPERTY
+        self.Property = F.MAP_INPUT(
+            Functor=F.MC_PROPERTY, Relations=self.MC2MCTrackInfo)
+
+        # Reconstructible
+        self.Reconstructible = F.MAP_INPUT(
+            Functor=F.MC_RECONSTRUCTIBLE, Relations=self.MC2Reconstructible)
+
+
+class MCP2Track:
+    """
+    MCP2Track helper class, will create all possible Reconstructed Track related functors and can be
+    accessed as the standard python class member, the member function Get(F) is a flexible
+    interface for future update.
+
+    Args:
+        mcparticles_input : MCParticle data handle (TES)
+        relations_locs: TES locations to the pre-existing relations for charged and neutral particles.
+            default = ["Relations/ChargedPP2MCP", "Relations/NeutralPP2MCP"]
+        process     (optional) : Process type (Hlt2 or Spruce), default = 'Hlt2'
+        root_in_tes (optional) : Standard Gaudi Algorithm RootInTES location
+        output_level (optional) : Standard Gaudi Algorithm OutputLevel
+
+    Example:
+        # Create the helper
+        MC_TR = MC_Track(MCParticles)
+
+        # Use predefined functor
+        allvariables['HAST'] = MC_TR.HasT 
+
+        # or use general interface, equivalent as above
+        allvariables['HAST'] = MC_TR.Get(F.TRACKHAST)
+
+    """
+
+    # Flexible interface for possible update
+    def get_info(self, Functor: BoundFunctor):
+        return F.MAP_INPUT(Functor=Functor, Relations=self.MC2Track)
+
+    def __init__(self,
+                 mcparticles_input,
+                 relations_locs=[
+                     "Relations/ChargedPP2MCP", "Relations/NeutralPP2MCP"
+                 ],
+                 process='Spruce',
+                 root_in_tes=None,
+                 output_level=INFO):
+
+        if not root_in_tes:
+            if process == 'Spruce':
+                root_in_tes = '/Event/Spruce/HLT2'
+            elif process == 'Hlt2' or process == 'Turbo':
+                root_in_tes = '/Event/HLT2'
+            else:
+                raise ValueError(
+                    f"The specified 'process' {process} not recognised. Can only be 'Hlt2' or 'Spruce' or 'Turbo'. Please check!"
+                )
+
+        if root_in_tes:
+            relations_locs = [
+                f'{root_in_tes}/{table_loc}' for table_loc in relations_locs
+            ]
+        ChargedPP2MCP = force_location(relations_locs[0])
+        NeutralPP2MCP = force_location(relations_locs[1])
+
+        # Load Algorithm
+        self.Algorithm = MC2TrackAlg(
+            Input=mcparticles_input,
+            ChargedPP2MCP=ChargedPP2MCP,
+            NeutralPP2MCP=NeutralPP2MCP,
+            OutputLevel=output_level)
+
+        # Alias the outputs
+        self.MC2Reconstructed = self.Algorithm.MC2Reconstructed
+        self.MC2Track = self.Algorithm.MC2Track
+
+        # Reconstructed
+        self.Reconstructed = F.MAP_INPUT(
+            Functor=F.MC_RECONSTRUCTED, Relations=self.MC2Reconstructed)
+
+        # Predefined track related functor
+        self.ReferencePoint = self.get_info(F.REFERENCEPOINT)
+        self.ReferencePoint_X = self.get_info(F.REFERENCEPOINT_X)
+        self.ReferencePoint_Y = self.get_info(F.REFERENCEPOINT_Y)
+        self.ReferencePoint_Z = self.get_info(F.REFERENCEPOINT_Z)
+        self.nDoF = self.get_info(F.NDOF)
+        self.GhostProbability = self.get_info(F.GHOSTPROB)
+        self.Flag = self.get_info(F.TRACKFLAG)
+        self.Type = self.get_info(F.TRACKTYPE)
+        self.HasT = self.get_info(F.TRACKHAST)
+        self.HasUT = self.get_info(F.TRACKHASUT)
+        self.HasVelo = self.get_info(F.TRACKHASVELO)
+        self.nHits = self.get_info(F.NHITS)
+        self.nVPHits = self.get_info(F.NVPHITS)
+        self.nUTHits = self.get_info(F.NUTHITS)
+        self.nFTHits = self.get_info(F.NFTHITS)
+        self.History = self.get_info(F.TRACKHISTORY)

Phys/DaVinci/python/DaVinci/truth_matching.py

@@ -23,6 +23,7 @@ def configured_MCTruthAndBkgCatAlg(
         process='Spruce',
         root_in_tes=None,
         redo_neutral_assoc=False,
+        filter_MCP=True,
         output_level=INFO):
     """
     Function to help configure the tools instantiated by the `MCTruthAndBkgCatAlg` algorithm.
@@ -58,11 +59,11 @@ def configured_MCTruthAndBkgCatAlg(
     if not root_in_tes:
         if process == 'Spruce':
             root_in_tes = '/Event/Spruce/HLT2'
-        elif process == 'Hlt2':
+        elif process == 'Hlt2' or process == 'Turbo':
             root_in_tes = '/Event/HLT2'
         else:
             raise ValueError(
-                f"The specified 'process' {process} not recognised. Can only be 'Hlt2' or 'Spruce'. Please check!"
+                f"The specified 'process' {process} not recognised. Can only be 'Hlt2' or 'Spruce' or 'Turbo'. Please check!"
             )
 
     # Tool used by DaVinciSmartAssociator
@@ -95,6 +96,7 @@ def configured_MCTruthAndBkgCatAlg(
 
     mctruth = MCTruthAndBkgCatAlg(
         Input=inputs,
+        filter_MCP=filter_MCP,
         DaVinciSmartAssociator=dv_assc,
         MCMatchObjP2MCRelator=mcrel_assc,
         BackgroundCategory=bkg_cat,