Jet Flavour Extension

.gitignore

@@ -36,3 +36,5 @@ benchmark/*
 /profile/*
 /statprof/*
 /debug/*
+ext/JetFlavourTagging/PROGRESS.md
+examples/flavour-tagging/data/events_080263084.root

Project.toml

@@ -18,10 +18,14 @@ StructArrays = "09ab397b-f2b6-538f-b94a-2f83cf4a842a"
 [weakdeps]
 EDM4hep = "eb32b910-dde9-4347-8fce-cd6be3498f0c"
 Makie = "ee78f7c6-11fb-53f2-987a-cfe4a2b5a57a"
+ONNXRunTime = "e034b28e-924e-41b2-b98f-d2bbeb830c6a"
+JSON = "682c06a0-de6a-54ab-a142-c8b1cf79cde6"
+PhysicalConstants = "5ad8b20f-a522-5ce9-bfc9-ddf1d5bda6ab"
 
 [extensions]
 EDM4hepJets = "EDM4hep"
 JetVisualisation = "Makie"
+JetFlavourTagging = ["EDM4hep", "ONNXRunTime", "JSON", "PhysicalConstants"]
 
 [compat]
 Accessors = "0.1.36"
@@ -38,9 +42,11 @@ Makie = "0.20, 0.21, 0.22, 0.23, 0.24"
 MuladdMacro = "0.2.4"
 Random = "1.9"
 Statistics = "1.9"
+ONNXRunTime = "1.3.1"
 StructArrays = "0.6.18, 0.7"
 Test = "1.9"
 julia = "1.9"
+PhysicalConstants = "0.2.4"
 
 [extras]
 Aqua = "4c88cf16-eb10-579e-8560-4a9242c79595"

examples/flavour-tagging/Project.toml

@@ -0,0 +1,20 @@
+[deps]
+Accessors = "7d9f7c33-5ae7-4f3b-8dc6-eff91059b697"
+ArgParse = "c7e460c6-2fb9-53a9-8c5b-16f535851c63"
+CodecZlib = "944b1d66-785c-5afd-91f1-9de20f533193"
+CodecZstd = "6b39b394-51ab-5f42-8807-6242bab2b4c2"
+EDM4hep = "eb32b910-dde9-4347-8fce-cd6be3498f0c"
+EnumX = "4e289a0a-7415-4d19-859d-a7e5c4648b56"
+IJulia = "7073ff75-c697-5162-941a-fcdaad2a7d2a"
+JSON = "682c06a0-de6a-54ab-a142-c8b1cf79cde6"
+JetReconstruction = "44e8cb2c-dfab-4825-9c70-d4808a591196"
+Logging = "56ddb016-857b-54e1-b83d-db4d58db5568"
+LorentzVectorHEP = "f612022c-142a-473f-8cfd-a09cf3793c6c"
+LorentzVectors = "3f54b04b-17fc-5cd4-9758-90c048d965e3"
+MuladdMacro = "46d2c3a1-f734-5fdb-9937-b9b9aeba4221"
+ONNXRunTime = "e034b28e-924e-41b2-b98f-d2bbeb830c6a"
+PhysicalConstants = "5ad8b20f-a522-5ce9-bfc9-ddf1d5bda6ab"
+StructArrays = "09ab397b-f2b6-538f-b94a-2f83cf4a842a"
+
+[sources]
+JetReconstruction = {path = "/Users/harrywanghc/Developer/2025/JuliaHEPForkToMain/JetReconstruction.jl"}

examples/flavour-tagging/README.md

@@ -0,0 +1,13 @@
+# Jet Reconstruction Flavour Tagging Module Examples
+
+## `simple-flavour-tagging.jl`
+
+```bash
+julia --project simple-flavour-tagging.jl
+```
+This script will perform a simple flavour tagging with one events only.
+It will use the `FlavourTagging` module to tag the jets in the event and print the results.
+
+## `simple-flavour-tagging.ipynb`
+
+The same as the above example, but using a Jupyter notebook.