VertexCompositeCandidate.h

Go to the documentation of this file.

#ifndef DataFormats_Candidate_VertexCompositeCandidate_H
#define DataFormats_Candidate_VertexCompositeCandidate_H

#include "DataFormats/Candidate/interface/VertexCompositeCandidateFwd.h"
#include "DataFormats/Candidate/interface/CompositeCandidate.h"

namespace reco {
  class VertexCompositeCandidate : public CompositeCandidate {
  public:
    VertexCompositeCandidate() : CompositeCandidate() {}
    VertexCompositeCandidate(
        Charge q, const LorentzVector &p4, const Point &vtx, int pdgId = 0, int status = 0, bool integerCharge = true)
        : CompositeCandidate(q, p4, vtx, pdgId, status, integerCharge), chi2_(0), ndof_(0) {}
    VertexCompositeCandidate(Charge q,
                             const LorentzVector &p4,
                             const Point &vtx,
                             const CovarianceMatrix &err,
                             double chi2,
                             double ndof,
                             int pdgId = 0,
                             int status = 0,
                             bool integerCharge = true);
    explicit VertexCompositeCandidate(const Candidate &p) : CompositeCandidate(p), chi2_(0), ndof_(0) {}
    explicit VertexCompositeCandidate(const CompositeCandidate &p) : CompositeCandidate(p), chi2_(0), ndof_(0) {}
    ~VertexCompositeCandidate() override;
    VertexCompositeCandidate *clone() const override;
    double vertexChi2() const override { return chi2_; }
    double vertexNdof() const override { return ndof_; }
    double vertexNormalizedChi2() const override { return chi2_ / ndof_; }
    double vertexCovariance(int i, int j) const override { return covariance_[idx(i, j)]; }
    using reco::LeafCandidate::vertexCovariance;  // avoid hiding the
    void fillVertexCovariance(CovarianceMatrix &v) const override;
    void setChi2AndNdof(double chi2, double ndof) {
      chi2_ = chi2;
      ndof_ = ndof;
    }
    void setCovariance(const CovarianceMatrix &m);

  private:
    Double32_t chi2_;
    Double32_t ndof_;
    Double32_t covariance_[size];
    index idx(index i, index j) const {
      int a = (i <= j ? i : j), b = (i <= j ? j : i);
      return b * (b + 1) / 2 + a;
    }
  };

}  // namespace reco

#endif