VertexingUtils.h

Go to the documentation of this file.

#ifndef  VERTEXINGUTILS_ANALYZERS_H
#define  VERTEXINGUTILS_ANALYZERS_H
 
#include <cmath>
#include <vector>
 
#include "ROOT/RVec.hxx"
#include "edm4hep/ReconstructedParticleData.h"
#include "edm4hep/MCParticleData.h"
#include "edm4hep/TrackState.h"
 
#include "edm4hep/VertexData.h"
#include "edm4hep/Vertex.h"
 
#include "TLorentzVector.h"
#include "TVectorD.h"
#include "TVector3.h"
#include "TMatrixDSym.h"
 
#include "fastjet/JetDefinition.hh"
 
 
namespace FCCAnalyses{
 
namespace VertexingUtils{
 
  TVectorD XPtoPar(TVector3 x, TVector3 p, Double_t Q);
 
  TVector3 ParToP(TVectorD Par);
 
 
  struct FCCAnalysesVertex{
    edm4hep::VertexData vertex;
    int ntracks;
    int mc_ind; 
    ROOT::VecOps::RVec<int> reco_ind;      // indices of the tracks fitted to that vertex, in the collection of all tracks
    ROOT::VecOps::RVec<float> reco_chi2;
    ROOT::VecOps::RVec< TVector3 >  updated_track_momentum_at_vertex;
    ROOT::VecOps::RVec< TVectorD >  updated_track_parameters;
    ROOT::VecOps::RVec<float> final_track_phases;
  };
 
  struct FCCAnalysesV0{
    ROOT::VecOps::RVec<FCCAnalysesVertex> vtx; // vertex object
    ROOT::VecOps::RVec<int> pdgAbs;            // pdg ID from reconstructions
    ROOT::VecOps::RVec<double> invM;           // invariant mass
    ROOT::VecOps::RVec<int> nSV_jet;           // no of V0s per jet
  };
  
  struct FCCAnalysesVertexMC{
    TVector3 vertex;
    ROOT::VecOps::RVec<int> mc_ind;
    ROOT::VecOps::RVec<int> mc_indneutral;
    ROOT::VecOps::RVec<int> mother_ind;
    ROOT::VecOps::RVec<int> gmother_ind;
  };
 
  struct selTracks {
    selTracks( float arg_d0sig_min, float arg_d0sig_max, float arg_z0sig_min, float arg_z0sig_max)  ;
    float m_d0sig_min = 0;
    float m_d0sig_max = 3;
    float m_z0sig_min = 0;
    float m_z0sig_max = 3;
    ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData>  operator() ( ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData> recop,
                                     ROOT::VecOps::RVec<edm4hep::TrackState> tracks  ) ;
  };
 
  ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData> SelPrimaryTracks( ROOT::VecOps::RVec<int> recind,
                                       ROOT::VecOps::RVec<int> mcind,
                                       ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData> reco,
                                       ROOT::VecOps::RVec<edm4hep::MCParticleData> mc,
                                       TVector3 MC_EventPrimaryVertex) ;
 
  int get_Nvertex( ROOT::VecOps::RVec<FCCAnalysesVertex> TheVertexColl );
 
  FCCAnalysesVertex get_FCCAnalysesVertex(ROOT::VecOps::RVec<FCCAnalysesVertex> TheVertexColl, int index );
 
  edm4hep::VertexData get_VertexData( FCCAnalysesVertex TheVertex ) ;
 
  ROOT::VecOps::RVec<edm4hep::VertexData> get_VertexData( ROOT::VecOps::RVec<FCCAnalysesVertex> TheVertexColl ) ;
 
  edm4hep::VertexData get_VertexData( ROOT::VecOps::RVec<FCCAnalysesVertex> TheVertexColl, int index);
 
  int get_VertexNtrk( FCCAnalysesVertex TheVertex ) ;
 
  ROOT::VecOps::RVec<int> get_VertexNtrk( ROOT::VecOps::RVec<FCCAnalysesVertex> vertices ) ;
  
  ROOT::VecOps::RVec<int> get_VertexRecoInd( FCCAnalysesVertex TheVertex ) ;
 
  ROOT::VecOps::RVec<int> get_VertexRecoParticlesInd( FCCAnalysesVertex TheVertex, 
                              const ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData>& reco );
 
  int get_nTracks(ROOT::VecOps::RVec<edm4hep::TrackState> tracks);
 
  bool compare_Tracks( const edm4hep::TrackState& tr1, const edm4hep::TrackState& tr2 ) ;
 
 
  ROOT::VecOps::RVec<FCCAnalysesVertex> get_all_vertices( FCCAnalysesVertex PV,
                              ROOT::VecOps::RVec<FCCAnalysesVertex> SV ); 
 
  ROOT::VecOps::RVec<FCCAnalysesVertex> get_all_vertices( FCCAnalysesVertex PV,
                              ROOT::VecOps::RVec<ROOT::VecOps::RVec<FCCAnalysesVertex>> SV ); 
 
  double get_invM_pairs( FCCAnalysesVertex vertex,
             double m1 = 0.13957039,
             double m2 = 0.13957039) ;
 
  ROOT::VecOps::RVec<double> get_invM_pairs( ROOT::VecOps::RVec<FCCAnalysesVertex> vertices,
                         double m1 = 0.13957039,
                         double m2 = 0.13957039 ) ;  
 
  double get_invM( FCCAnalysesVertex vertex ) ;
 
  ROOT::VecOps::RVec<double> get_invM( ROOT::VecOps::RVec<FCCAnalysesVertex> vertices ) ;
 
  double get_PV2V0angle( FCCAnalysesVertex V0,
             FCCAnalysesVertex PV) ;
 
  double get_PV2vtx_angle( ROOT::VecOps::RVec<edm4hep::TrackState> tracks,
               FCCAnalysesVertex vtx,
               FCCAnalysesVertex PV ) ;
 
  double get_trackE( edm4hep::TrackState track ) ;
 
  int get_n_SV( FCCAnalysesV0 SV );
 
  ROOT::VecOps::RVec<TVector3> get_position_SV( FCCAnalysesV0 SV );
 
  ROOT::VecOps::RVec<int> get_pdg_V0( FCCAnalysesV0 V0 );
 
  ROOT::VecOps::RVec<double> get_invM_V0( FCCAnalysesV0 V0 );
 
  ROOT::VecOps::RVec<TVector3> get_p_SV( FCCAnalysesV0 SV );
 
  ROOT::VecOps::RVec<double> get_chi2_SV( FCCAnalysesV0 SV );
 
 
  int get_n_SV( ROOT::VecOps::RVec<FCCAnalysesVertex> vertices );
 
  ROOT::VecOps::RVec<TVector3> get_p_SV( ROOT::VecOps::RVec<FCCAnalysesVertex> vertices );
 
  ROOT::VecOps::RVec<TVector3> get_position_SV( ROOT::VecOps::RVec<FCCAnalysesVertex> vertices );
 
  ROOT::VecOps::RVec<double> get_pMag_SV( ROOT::VecOps::RVec<FCCAnalysesVertex> vertices );
 
  ROOT::VecOps::RVec<double> get_chi2_SV( ROOT::VecOps::RVec<FCCAnalysesVertex> vertices );
 
  ROOT::VecOps::RVec<double> get_norm_chi2_SV( ROOT::VecOps::RVec<FCCAnalysesVertex> vertices );
 
  ROOT::VecOps::RVec<int> get_nDOF_SV( ROOT::VecOps::RVec<FCCAnalysesVertex> vertices );
 
  ROOT::VecOps::RVec<double> get_theta_SV( ROOT::VecOps::RVec<FCCAnalysesVertex> vertices );
 
  ROOT::VecOps::RVec<double> get_phi_SV( ROOT::VecOps::RVec<FCCAnalysesVertex> vertices );
 
  ROOT::VecOps::RVec<double> get_relTheta_SV( ROOT::VecOps::RVec<FCCAnalysesVertex> vertices,
                          ROOT::VecOps::RVec<int> nSV_jet,
                          ROOT::VecOps::RVec<fastjet::PseudoJet> jets );
 
  ROOT::VecOps::RVec<double> get_relPhi_SV( ROOT::VecOps::RVec<FCCAnalysesVertex> vertices,
                        ROOT::VecOps::RVec<int> nSV_jet,
                        ROOT::VecOps::RVec<fastjet::PseudoJet> jets );
  
  ROOT::VecOps::RVec<double> get_pointingangle_SV( ROOT::VecOps::RVec<FCCAnalysesVertex> vertices,
                           FCCAnalysesVertex PV );
 
  ROOT::VecOps::RVec<double> get_dxy_SV( ROOT::VecOps::RVec<FCCAnalysesVertex> vertices,
                     FCCAnalysesVertex PV );
 
  ROOT::VecOps::RVec<double> get_d3d_SV( ROOT::VecOps::RVec<FCCAnalysesVertex> vertices,
                     FCCAnalysesVertex PV );
 
  ROOT::VecOps::RVec<double> get_d3d_SV_obj( ROOT::VecOps::RVec<FCCAnalysesVertex> vertices,
                         TVector3 location );
 
  ROOT::VecOps::RVec<double> get_d3d_SV_obj( ROOT::VecOps::RVec<FCCAnalysesVertex> vertices,
                         edm4hep::Vector3d location );
 
  ROOT::VecOps::RVec<double> get_dR_SV_obj( ROOT::VecOps::RVec<FCCAnalysesVertex> vertices,
                        TVector3 location );
 
  ROOT::VecOps::RVec<double> get_dR_SV_obj( ROOT::VecOps::RVec<FCCAnalysesVertex> vertices,
                        edm4hep::Vector3d location );
 
 
  
  ROOT::VecOps::RVec<FCCAnalysesVertex> get_all_SVs( ROOT::VecOps::RVec<ROOT::VecOps::RVec<FCCAnalysesVertex>> vertices );
 
  int get_n_SV( ROOT::VecOps::RVec<ROOT::VecOps::RVec<FCCAnalysesVertex>> vertices );
 
  ROOT::VecOps::RVec<int> get_n_SV_jets( ROOT::VecOps::RVec<ROOT::VecOps::RVec<FCCAnalysesVertex>> vertices );
 
  std::vector<std::vector<edm4hep::TrackState>> get_tracksInJets( ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData> recoparticles,
                                  ROOT::VecOps::RVec<edm4hep::TrackState> thetracks,
                                  ROOT::VecOps::RVec<fastjet::PseudoJet> jets,
                                  std::vector<std::vector<int>> jet_consti );
 
  ROOT::VecOps::RVec<ROOT::VecOps::RVec<FCCAnalysesVertex>> get_svInJets( ROOT::VecOps::RVec<FCCAnalysesVertex> vertices,
                                      ROOT::VecOps::RVec<int> nSV_jet );
 
  // --- for get_SV_jets --- //
  ROOT::VecOps::RVec<ROOT::VecOps::RVec<double>> get_invM( ROOT::VecOps::RVec<ROOT::VecOps::RVec<FCCAnalysesVertex>> vertices );
  ROOT::VecOps::RVec<ROOT::VecOps::RVec<TVector3>> get_p_SV( ROOT::VecOps::RVec<ROOT::VecOps::RVec<FCCAnalysesVertex>> vertices );
  ROOT::VecOps::RVec<ROOT::VecOps::RVec<double>> get_pMag_SV( ROOT::VecOps::RVec<ROOT::VecOps::RVec<FCCAnalysesVertex>> vertices );
  ROOT::VecOps::RVec<ROOT::VecOps::RVec<int>> get_VertexNtrk( ROOT::VecOps::RVec<ROOT::VecOps::RVec<FCCAnalysesVertex>> vertices );
  ROOT::VecOps::RVec<ROOT::VecOps::RVec<double>> get_chi2_SV( ROOT::VecOps::RVec<ROOT::VecOps::RVec<FCCAnalysesVertex>> vertices );
  ROOT::VecOps::RVec<ROOT::VecOps::RVec<double>> get_norm_chi2_SV( ROOT::VecOps::RVec<ROOT::VecOps::RVec<FCCAnalysesVertex>> vertices );
  ROOT::VecOps::RVec<ROOT::VecOps::RVec<int>> get_nDOF_SV( ROOT::VecOps::RVec<ROOT::VecOps::RVec<FCCAnalysesVertex>> vertices );
  ROOT::VecOps::RVec<ROOT::VecOps::RVec<double>> get_theta_SV( ROOT::VecOps::RVec<ROOT::VecOps::RVec<FCCAnalysesVertex>> vertices );
  ROOT::VecOps::RVec<ROOT::VecOps::RVec<double>> get_phi_SV( ROOT::VecOps::RVec<ROOT::VecOps::RVec<FCCAnalysesVertex>> vertices );
  ROOT::VecOps::RVec<ROOT::VecOps::RVec<double>> get_relTheta_SV( ROOT::VecOps::RVec<ROOT::VecOps::RVec<FCCAnalysesVertex>> vertices, ROOT::VecOps::RVec<fastjet::PseudoJet> jets );
  ROOT::VecOps::RVec<ROOT::VecOps::RVec<double>> get_relPhi_SV( ROOT::VecOps::RVec<ROOT::VecOps::RVec<FCCAnalysesVertex>> vertices, ROOT::VecOps::RVec<fastjet::PseudoJet> jets );
  ROOT::VecOps::RVec<ROOT::VecOps::RVec<double>> get_pointingangle_SV( ROOT::VecOps::RVec<ROOT::VecOps::RVec<FCCAnalysesVertex>> vertices, FCCAnalysesVertex PV );
  ROOT::VecOps::RVec<ROOT::VecOps::RVec<double>> get_dxy_SV( ROOT::VecOps::RVec<ROOT::VecOps::RVec<FCCAnalysesVertex>> vertices, FCCAnalysesVertex PV );
  ROOT::VecOps::RVec<ROOT::VecOps::RVec<double>> get_d3d_SV( ROOT::VecOps::RVec<ROOT::VecOps::RVec<FCCAnalysesVertex>> vertices, FCCAnalysesVertex PV );
  ROOT::VecOps::RVec<ROOT::VecOps::RVec<int>> get_pdg_V0( ROOT::VecOps::RVec<int> pdg, ROOT::VecOps::RVec<int> nSV_jet );
  ROOT::VecOps::RVec<ROOT::VecOps::RVec<double>> get_invM_V0( ROOT::VecOps::RVec<double> invM, ROOT::VecOps::RVec<int> nSV_jet );
  ROOT::VecOps::RVec<ROOT::VecOps::RVec<TVector3>> get_position_SV( ROOT::VecOps::RVec<ROOT::VecOps::RVec<FCCAnalysesVertex>> vertices ); 
  // --- for get_SV_jets --- //
 
  float get_trackMom( edm4hep::TrackState & atrack );
 
 
// --- Conversion methods between the Delphes and edm4hep conventions
 
  TVectorD Edm4hep2Delphes_TrackParam( const TVectorD& param, bool Units_mm );
  TVectorD Delphes2Edm4hep_TrackParam( const TVectorD& param, bool Units_mm );
  TMatrixDSym  Edm4hep2Delphes_TrackCovMatrix( const std::array<float, 21>&  covMatrix, bool Units_mm );
#if __has_include("edm4hep/CovMatrix6f.h")
  TMatrixDSym  Edm4hep2Delphes_TrackCovMatrix( const edm4hep::CovMatrix6f&  covMatrix, bool Units_mm );
#endif
  std::array<float, 21> Delphes2Edm4hep_TrackCovMatrix( const TMatrixDSym& cov, bool Units_mm ) ;
 
 
  TVectorD get_trackParam( edm4hep::TrackState & atrack, bool Units_mm = false) ;
  TMatrixDSym get_trackCov( const edm4hep::TrackState &  atrack, bool Units_mm = false) ;
 
  TVectorD ParToACTS(TVectorD Par);
  TMatrixDSym CovToACTS(TMatrixDSym Cov,TVectorD Par);
 
 
 
}//end NS VertexingUtils
 
}//end NS FCCAnalyses
#endif