ReconstructedParticle2Track.h

Go to the documentation of this file.

 
#ifndef  RECONSTRUCTEDPARTICLE2TRACK_ANALYZERS_H
#define  RECONSTRUCTEDPARTICLE2TRACK_ANALYZERS_H
 
#include <cmath>
#include <vector>
 
#include "ROOT/RVec.hxx"
#include "edm4hep/Quantity.h"
#include "edm4hep/ReconstructedParticleData.h"
#include "edm4hep/TrackData.h"
#include "edm4hep/TrackState.h"
#if __has_include("edm4hep/TrackerHit3DData.h")
#include "edm4hep/TrackerHit3DData.h"
#else
#include "edm4hep/TrackerHitData.h"
namespace edm4hep {
  using TrackerHit3DData = edm4hep::TrackerHitData;
}
#endif
#include <TVectorD.h>
#include <TVector3.h>
#include <TLorentzVector.h>
 
#include <TMath.h>
#include <iostream>
 
namespace FCCAnalyses{
 
namespace ReconstructedParticle2Track{
 
  ROOT::VecOps::RVec<float> getRP2TRK_mom (ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData> in, 
                       ROOT::VecOps::RVec<edm4hep::TrackState> tracks);
 
  ROOT::VecOps::RVec<float> getRP2TRK_charge(ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData> in,  
                         ROOT::VecOps::RVec<edm4hep::TrackState> tracks);
 
  //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<float> XPtoPar_dxy(const ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData>& in,
                    const ROOT::VecOps::RVec<edm4hep::TrackState>& tracks,
                    const TLorentzVector& V, // primary vertex
                    const float& Bz);
 
  ROOT::VecOps::RVec<float> XPtoPar_dz(const ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData>& in,
                                        const ROOT::VecOps::RVec<edm4hep::TrackState>& tracks,
                                        const TLorentzVector& V, // primary vertex
                                        const float& Bz);
 
  ROOT::VecOps::RVec<float> XPtoPar_phi(const ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData>& in,
                    const ROOT::VecOps::RVec<edm4hep::TrackState>& tracks,
                                        const TLorentzVector& V, // primary vertex
                                        const float& Bz);
 
  ROOT::VecOps::RVec<float> XPtoPar_C(const ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData>& in,
                    const ROOT::VecOps::RVec<edm4hep::TrackState>& tracks,
                                        const float& Bz);
 
  ROOT::VecOps::RVec<float> XPtoPar_ct(const ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData>& in,
                    const ROOT::VecOps::RVec<edm4hep::TrackState>& tracks,
                                        const float& Bz);
 
  ROOT::VecOps::RVec<float> getRP2TRK_D0 (ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData> in,
                      ROOT::VecOps::RVec<edm4hep::TrackState> tracks);
 
  ROOT::VecOps::RVec<float> getRP2TRK_Z0 (ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData> in,
                      ROOT::VecOps::RVec<edm4hep::TrackState> tracks);
 
  ROOT::VecOps::RVec<float> getRP2TRK_phi (ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData> in,
                       ROOT::VecOps::RVec<edm4hep::TrackState> tracks);
 
  ROOT::VecOps::RVec<float> getRP2TRK_omega (ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData> in,
                         ROOT::VecOps::RVec<edm4hep::TrackState> tracks);
 
  ROOT::VecOps::RVec<float> getRP2TRK_tanLambda (ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData> in,
                         ROOT::VecOps::RVec<edm4hep::TrackState> tracks);
 
  ROOT::VecOps::RVec<float> getRP2TRK_D0_sig (ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData> in,
                          ROOT::VecOps::RVec<edm4hep::TrackState> tracks);
 
  ROOT::VecOps::RVec<float> getRP2TRK_Z0_sig (ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData> in,
                          ROOT::VecOps::RVec<edm4hep::TrackState> tracks);
 
 
  ROOT::VecOps::RVec<float> getRP2TRK_D0_cov (ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData> in,
                          ROOT::VecOps::RVec<edm4hep::TrackState> tracks);
 
  ROOT::VecOps::RVec<float> getRP2TRK_Z0_cov (ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData> in,
                          ROOT::VecOps::RVec<edm4hep::TrackState> tracks);
 
  ROOT::VecOps::RVec<float> getRP2TRK_phi_cov (ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData> in,
                           ROOT::VecOps::RVec<edm4hep::TrackState> tracks);
 
  ROOT::VecOps::RVec<float> getRP2TRK_omega_cov (ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData> in,
                         ROOT::VecOps::RVec<edm4hep::TrackState> tracks);
 
  ROOT::VecOps::RVec<float> getRP2TRK_tanLambda_cov (ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData> in,
                             ROOT::VecOps::RVec<edm4hep::TrackState> tracks);
 
  ROOT::VecOps::RVec<float> getRP2TRK_d0_phi0_cov(ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData> in,
                          ROOT::VecOps::RVec<edm4hep::TrackState> tracks);
 
  ROOT::VecOps::RVec<float> getRP2TRK_d0_omega_cov(ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData> in,
                           ROOT::VecOps::RVec<edm4hep::TrackState> tracks);
 
  ROOT::VecOps::RVec<float> getRP2TRK_d0_z0_cov(ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData> in,
                        ROOT::VecOps::RVec<edm4hep::TrackState> tracks);
 
  ROOT::VecOps::RVec<float> getRP2TRK_d0_tanlambda_cov(ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData> in,
                               ROOT::VecOps::RVec<edm4hep::TrackState> tracks);
 
  ROOT::VecOps::RVec<float> getRP2TRK_phi0_omega_cov(ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData> in,
                             ROOT::VecOps::RVec<edm4hep::TrackState> tracks);
 
  ROOT::VecOps::RVec<float> getRP2TRK_phi0_z0_cov(ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData> in,
                          ROOT::VecOps::RVec<edm4hep::TrackState> tracks);
 
  ROOT::VecOps::RVec<float> getRP2TRK_phi0_tanlambda_cov(ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData> in,
                             ROOT::VecOps::RVec<edm4hep::TrackState> tracks) ;
 
  ROOT::VecOps::RVec<float> getRP2TRK_omega_z0_cov(ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData> in,
                           ROOT::VecOps::RVec<edm4hep::TrackState> tracks) ;
 
  ROOT::VecOps::RVec<float> getRP2TRK_omega_tanlambda_cov(ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData> in,
                              ROOT::VecOps::RVec<edm4hep::TrackState> tracks) ;
 
  ROOT::VecOps::RVec<float> getRP2TRK_z0_tanlambda_cov(ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData> in,
                               ROOT::VecOps::RVec<edm4hep::TrackState> tracks);
 
 
  ROOT::VecOps::RVec<edm4hep::TrackState> getRP2TRK( ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData> in,
                             ROOT::VecOps::RVec<edm4hep::TrackState> tracks ) ;
 
  ROOT::VecOps::RVec<int> get_recoindTRK( ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData> in, 
                      ROOT::VecOps::RVec<edm4hep::TrackState> tracks ) ;
  
  int getTK_n(ROOT::VecOps::RVec<edm4hep::TrackState> x) ;
 
  ROOT::VecOps::RVec<bool> hasTRK( ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData> in ) ;
 
}//end NS ReconstructedParticle2Track
 
}//end NS FCCAnalyses
#endif