SmearObjects.h

Go to the documentation of this file.

#ifndef SMEAR_OBJECTS_ANALYZERS_H
#define SMEAR_OBJECTS_ANALYZERS_H
 
// Standard library
#include <cmath>
#include <vector>
// ROOT
#include "ROOT/RVec.hxx"
#include "TLorentzVector.h"
#include "TMath.h"
#include "TMatrixDSym.h"
#include "TRandom.h"
// EDM4hep
#include "edm4hep/MCParticleData.h"
#include "edm4hep/RecDqdxData.h"
// FCCAnalyses
#include "FCCAnalyses/ReconstructedParticle2Track.h"
#include "FCCAnalyses/TrackUtils.h"
 
namespace FCCAnalyses ::SmearObjects {

TVectorD TrackParamFromMC_DelphesConv(edm4hep::MCParticleData aMCParticle);

struct SmearedTracks {
  bool m_debug;
  TRandom m_random;
  float m_smear_parameters[5];
  SmearedTracks(float smear_d0, float smear_phi, float smear_omega,
                float smear_z0, float smear_tlambda, bool debug);
  ROOT::VecOps::RVec<edm4hep::TrackState>
  operator()(const ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData>
                 &allRecoParticles,
             const ROOT::VecOps::RVec<edm4hep::TrackState> &alltracks,
             const ROOT::VecOps::RVec<int> &RP2MC_indices,
             const ROOT::VecOps::RVec<edm4hep::MCParticleData> &mcParticles);
};

ROOT::VecOps::RVec<edm4hep::TrackState> mcTrackParameters(
    const ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData>
        &allRecoParticles,
    const ROOT::VecOps::RVec<edm4hep::TrackState> &alltracks,
    const ROOT::VecOps::RVec<int> &RP2MC_indices,
    const ROOT::VecOps::RVec<edm4hep::MCParticleData> &mcParticles);

TVectorD CovSmear(TVectorD x, TMatrixDSym C, TRandom *ran, bool debug);

struct SmearedTracksdNdx {
  bool m_debug; 
  TRandom m_random;
  float m_scale; 
 
  SmearedTracksdNdx(float m_scale, bool debug);

  ROOT::VecOps::RVec<edm4hep::RecDqdxData>
  operator()(const ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData>
                 &allRecoParticles,
             const TrackUtils::TrackDqdxHandler &dNdxHandler,
             const ROOT::VecOps::RVec<float> &length,
             const ROOT::VecOps::RVec<int> &RP2MC_indices,
             const ROOT::VecOps::RVec<edm4hep::MCParticleData> &mcParticles);
};

struct SmearedTracksTOF {
  bool m_debug;
  TRandom m_random;
  float m_scale;
  SmearedTracksTOF(float m_scale, bool debug);
  ROOT::VecOps::RVec<edm4hep::TrackerHit3DData>
  operator()(const ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData>
                 &allRecoParticles,
             const ROOT::VecOps::RVec<edm4hep::TrackData> &trackdata,
             const ROOT::VecOps::RVec<edm4hep::TrackerHit3DData> &trackerhits,
             const ROOT::VecOps::RVec<float> &length,
             const ROOT::VecOps::RVec<int> &RP2MC_indices,
             const ROOT::VecOps::RVec<edm4hep::MCParticleData> &mcParticles);
};

struct SmearedReconstructedParticle {
  bool m_debug;
  float m_scale;
  int m_type;
  int m_mode;
  SmearedReconstructedParticle(float scale, int type, int mode, bool debug);
 
  ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData>
  operator()(const ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData>
                 &allRecoParticles,
             const ROOT::VecOps::RVec<int> &RP2MC_indices,
             const ROOT::VecOps::RVec<edm4hep::MCParticleData> &mcParticles);
};
 
} // namespace FCCAnalyses::SmearObjects
 
#endif /* SMEAR_OBJECTS_ANALYZERS_H */