Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Jet Flavour data pre-processing and inference #224

Merged
merged 8 commits into from
Oct 31, 2022
Merged

Jet Flavour data pre-processing and inference #224

Show file tree
Hide file tree
Changes from 6 commits
Commits
Show all changes
8 commits
Select commit Hold shift + click to select a range
f79bd0d
added methods for computing tagging observables
selvaggi Oct 30, 2022
72a3d41
add stage1/2 macros for producing jet ntuples
selvaggi Oct 30, 2022
aad7c9f
first commit README
selvaggi Oct 30, 2022
a080b50
added all stage script
selvaggi Oct 30, 2022
04ea1db
commit inference example
selvaggi Oct 30, 2022
4475744
polished readme
selvaggi Oct 30, 2022
e87f8dc
Remove commented code
vvolkl Oct 31, 2022
f7ab67f
Apply suggestions from code review
vvolkl Oct 31, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
293 changes: 293 additions & 0 deletions analyzers/dataframe/FCCAnalyses/JetConstituentsUtils.h
View file
Open in desktop

Large diffs are not rendered by default.

49 changes: 49 additions & 0 deletions analyzers/dataframe/FCCAnalyses/ReconstructedParticle2Track.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -8,11 +8,60 @@
#include "ROOT/RVec.hxx"
#include "edm4hep/ReconstructedParticleData.h"
#include "edm4hep/TrackState.h"
#include <TVectorD.h>
#include <TVector3.h>
#include <TMath.h>
#include <iostream>

namespace FCCAnalyses{

namespace ReconstructedParticle2Track{

//compute the magnetic field Bz
ROOT::VecOps::RVec<float> getRP2TRK_Bz(const ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData>& rps,
const ROOT::VecOps::RVec<edm4hep::TrackState>& tracks); //here computed for all particles passed

float Bz(const ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData>& rps,
const ROOT::VecOps::RVec<edm4hep::TrackState>& tracks); //here only computed for the first charged particle encountered

/*
ROOT::VecOps::RVec<ROOT::VecOps::RVec<float> > XPtoPar(const ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData>& in,
const ROOT::VecOps::RVec<edm4hep::TrackState>& tracks,
const TVector3& x,
const float& Bz);
*/
vvolkl marked this conversation as resolved.
Show resolved Hide resolved

ROOT::VecOps::RVec<float> XPtoPar_dxy(const ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData>& in,
const ROOT::VecOps::RVec<edm4hep::TrackState>& tracks,
const TVector3& x,
const float& Bz);

ROOT::VecOps::RVec<float> XPtoPar_dz(const ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData>& in,
const ROOT::VecOps::RVec<edm4hep::TrackState>& tracks,
const TVector3& V,
const float& Bz);

ROOT::VecOps::RVec<float> XPtoPar_phi(const ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData>& in,
const ROOT::VecOps::RVec<edm4hep::TrackState>& tracks,
const TVector3& V,
const float& Bz);

ROOT::VecOps::RVec<float> XPtoPar_C(const ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData>& in,
const ROOT::VecOps::RVec<edm4hep::TrackState>& tracks,
const TVector3& V,
const float& Bz);

ROOT::VecOps::RVec<float> XPtoPar_ct(const ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData>& in,
const ROOT::VecOps::RVec<edm4hep::TrackState>& tracks,
const TVector3& V,
const float& Bz);

//ROOT::VecOps::RVec<float> XPtoPar_dxy(const ROOT::VecOps::RVec<TVectorD>& in);
//ROOT::VecOps::RVec<float> XPtoPar_dz(const ROOT::VecOps::RVec<TVectorD>& in);
//ROOT::VecOps::RVec<float> XPtoPar_phi0(const ROOT::VecOps::RVec<TVectorD>& in);
//ROOT::VecOps::RVec<float> XPtoPar_C(const ROOT::VecOps::RVec<TVectorD>& in);
//ROOT::VecOps::RVec<float> XPtoPar_ct(const ROOT::VecOps::RVec<TVectorD>& in);

vvolkl marked this conversation as resolved.
Show resolved Hide resolved
/// Return the D0 of a track to a reconstructed particle
ROOT::VecOps::RVec<float> getRP2TRK_D0 (ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData> in,
ROOT::VecOps::RVec<edm4hep::TrackState> tracks);
Expand Down