RntStructs.h

Go to the documentation of this file.

#ifndef RecoTracker_MkFitCore_standalone_RntDumper_RntStructs_h
#define RecoTracker_MkFitCore_standalone_RntDumper_RntStructs_h

#include "RecoTracker/MkFitCore/interface/IdxChi2List.h"

#include "ROOT/REveVector.hxx"
#include "Math/Point3D.h"
#include "Math/Vector3D.h"

// From CMSSW data formats
typedef ROOT::Math::PositionVector3D<ROOT::Math::Cartesian3D<float> > XYZPointF;
typedef ROOT::Math::DisplacementVector3D<ROOT::Math::Cartesian3D<float> > XYZVectorF;
typedef ROOT::Math::DisplacementVector3D<ROOT::Math::CylindricalEta3D<float> > RhoEtaPhiVectorF;
// typedef ROOT::Math::DisplacementVector3D<ROOT::Math::Polar3D<float> > RThetaPhiVectorF;

using RVec = ROOT::Experimental::REveVector;

struct HeaderLayer {
  int event, iter_idx, iter_algo, eta_region, layer;
  float qb_min, qb_max;  // qbar layer limits, r for barrel, z for endcap
  bool is_barrel, is_pix, is_stereo;

  HeaderLayer() = default;
  HeaderLayer& operator=(const HeaderLayer&) = default;
};

struct State {
  RVec pos, mom;

  State() = default;
  State& operator=(const State&) = default;
};

struct PropState : public State {
  float dalpha;  // helix angle during propagation
  int fail_flag;

  PropState() = default;
  PropState& operator=(const PropState&) = default;
};

struct SimSeedInfo {
  State s_sim;
  State s_seed;
  int sim_lbl, seed_lbl, seed_idx;
  int n_hits, n_match;
  bool has_sim = false;

  float good_frac() const { return (float)n_match / n_hits; }

  SimSeedInfo() = default;
  SimSeedInfo& operator=(const SimSeedInfo&) = default;
};

struct BinSearch {
  float phi, dphi, q, dq;
  short unsigned int p1, p2, q1, q2;
  short int wsr;
  bool wsr_in_gap;
  bool has_nans = false;

  bool nan_check();

  BinSearch() = default;
  BinSearch& operator=(const BinSearch&) = default;
};

struct HitInfo {
  RVec hit_pos;
  float hit_q, hit_qhalflen, hit_qbar, hit_phi;
  int hit_lbl;

  HitInfo() = default;
  HitInfo& operator=(const HitInfo&) = default;
};

struct HitMatchInfo : public HitInfo {
  RVec trk_pos, trk_mom;
  float ddq, ddphi;
  float chi2_true;
  int hit_index;
  bool match;
  bool presel;
  bool prop_ok;
  bool has_ic2list{false};
  mkfit::IdxChi2List ic2list;

  bool accept() const { return presel && prop_ok; }

  HitMatchInfo() = default;
  HitMatchInfo& operator=(const HitMatchInfo&) = default;
};

struct CandInfo {
  SimSeedInfo ssi;
  State s_ctr;
  PropState ps_min, ps_max;
  BinSearch bso;
  BinSearch bsn;
  std::vector<HitMatchInfo> hmi;
  int n_all_hits = 0, n_hits_pass = 0, n_hits_match = 0, n_hits_pass_match = 0;
  int ord_first_match = -1;
  float dphi_first_match = -9999.0f, dq_first_match = -9999.0f;
  bool has_nans = false;

  CandInfo(const SimSeedInfo& s, const State& c) : ssi(s), s_ctr(c) {}

  void nan_check();
  void reset_hits_match() {
    n_all_hits = n_hits_pass = n_hits_match = n_hits_pass_match = 0;
    ord_first_match = -1;
    dphi_first_match = dq_first_match = -9999.0f;
  }

  bool assignIdxChi2List(const mkfit::IdxChi2List& ic2l) {
    for (auto& hm : hmi) {
      if (hm.hit_index == ic2l.hitIdx) {
        hm.has_ic2list = true;
        hm.ic2list = ic2l;
        return true;
      }
    }
    return false;
  }

  CandInfo() = default;
  CandInfo& operator=(const CandInfo&) = default;
};

struct FailedPropInfo {
  SimSeedInfo ssi;
  State s_prev;
  State s_final;
  bool has_nans = false;

  FailedPropInfo(const SimSeedInfo& s, const State& p, const State& f) : ssi(s), s_prev(p), s_final(f) {}

  void nan_check();

  FailedPropInfo() = default;
  FailedPropInfo& operator=(const FailedPropInfo&) = default;
};

#endif