PFBlockElementTrack.h

Go to the documentation of this file.

#ifndef __PFBlockElementTrack__
#define __PFBlockElementTrack__
 
#include <iostream>
 
#include "DataFormats/Math/interface/Point3D.h"
 
#include "DataFormats/ParticleFlowReco/interface/PFBlockElement.h"
#include "DataFormats/ParticleFlowReco/interface/PFRecTrackFwd.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"
 
namespace reco {
 
  class PFBlockElementTrack final : public PFBlockElement {
  public:
    PFBlockElementTrack() {}
 
    PFBlockElementTrack(const PFRecTrackRef& ref);
 
    PFBlockElement* clone() const override { return new PFBlockElementTrack(*this); }
 
    void Dump(std::ostream& out = std::cout, const char* tab = " ") const override;
 
    bool trackType(TrackType trType) const override { return (trackType_ >> trType) & 1; }
 
    void setTrackType(TrackType trType, bool value) override {
      if (value)
        trackType_ = trackType_ | (1 << trType);
      else
        trackType_ = trackType_ ^ (1 << trType);
    }
 
    void setPositionAtECALEntrance(float x, float y, float z) { positionAtECALEntrance_.SetCoordinates(x, y, z); }
 
    const math::XYZPointF& positionAtECALEntrance() const { return positionAtECALEntrance_; }
 
    const PFRecTrackRef& trackRefPF() const override { return trackRefPF_; }
 
    const reco::TrackRef& trackRef() const override { return trackRef_; }
 
    static constexpr unsigned int kPrimaryMask = 1 << T_TO_DISP;
 
    static constexpr unsigned int kSecondaryMask = (1 << T_FROM_DISP) | (1 << T_FROM_GAMMACONV) | (1 << T_FROM_V0);
 
    static constexpr unsigned int kLinkedToDisplacedVertexMask = kPrimaryMask | kSecondaryMask;
 
    bool isSecondary() const override { return trackType_ & kSecondaryMask; }
 
    bool isPrimary() const override { return trackType_ & kPrimaryMask; }
 
    bool isLinkedToDisplacedVertex() const override { return trackType_ & kLinkedToDisplacedVertexMask; }
 
    const PFDisplacedTrackerVertexRef& displacedVertexRef(TrackType trType) const override {
      if (trType == T_TO_DISP)
        return displacedVertexDaughterRef_;
      else if (trType == T_FROM_DISP)
        return displacedVertexMotherRef_;
      else
        return nullPFDispVertex_;
    }
 
    void setDisplacedVertexRef(const PFDisplacedTrackerVertexRef& niref, TrackType trType) override {
      if (trType == T_TO_DISP) {
        displacedVertexDaughterRef_ = niref;
        setTrackType(trType, true);
      } else if (trType == T_FROM_DISP) {
        displacedVertexMotherRef_ = niref;
        setTrackType(trType, true);
      }
    }
 
    const reco::MuonRef& muonRef() const override { return muonRef_; }
 
    void setMuonRef(const MuonRef& muref) override {
      muonRef_ = muref;
      setTrackType(MUON, true);
    }
 
    const ConversionRefVector& convRefs() const override { return convRefs_; }
 
    void setConversionRef(const ConversionRef& convRef, TrackType trType) override {
      convRefs_.push_back(convRef);
      setTrackType(trType, true);
    }
 
    const VertexCompositeCandidateRef& V0Ref() const override { return v0Ref_; }
 
    void setV0Ref(const VertexCompositeCandidateRef& V0Ref, TrackType trType) override {
      v0Ref_ = V0Ref;
      setTrackType(trType, true);
    }
 
  private:
    PFRecTrackRef trackRefPF_;
 
    reco::TrackRef trackRef_;
 
    unsigned int trackType_;
 
    math::XYZPointF positionAtECALEntrance_;
 
    PFDisplacedTrackerVertexRef displacedVertexMotherRef_;
 
    PFDisplacedTrackerVertexRef displacedVertexDaughterRef_;
 
    reco::MuonRef muonRef_;
 
    ConversionRefVector convRefs_;
 
    VertexCompositeCandidateRef v0Ref_;
  };
}  // namespace reco
 
#endif