GsfPFRecTrack.h

Go to the documentation of this file.

#ifndef DataFormats_ParticleFlowReco_GsfPFRecTrack_h
#define DataFormats_ParticleFlowReco_GsfPFRecTrack_h

#include "DataFormats/ParticleFlowReco/interface/PFRecTrack.h"
#include "DataFormats/ParticleFlowReco/interface/PFRecTrackFwd.h"
/* #include "DataFormats/Common/interface/RefToBase.h" */
#include "DataFormats/GsfTrackReco/interface/GsfTrackFwd.h"
#include "DataFormats/ParticleFlowReco/interface/GsfPFRecTrackFwd.h"
#include "DataFormats/ParticleFlowReco/interface/PFBrem.h"
#include <iostream>

namespace reco {

  class GsfPFRecTrack : public PFRecTrack {
  public:
    GsfPFRecTrack(){};
    GsfPFRecTrack(double charge,
                  AlgoType_t algoType,
                  int trackId,
                  const reco::GsfTrackRef& gtrackref,
                  const edm::Ref<std::vector<PFRecTrack> >& kfpfrectrackref);

    const reco::GsfTrackRef& gsfTrackRef() const { return gsfTrackRef_; }

    const edm::Ref<std::vector<PFRecTrack> >& kfPFRecTrackRef() const { return kfPFRecTrackRef_; }
    void addBrem(const reco::PFBrem& brem);

    void calculateBremPositionREP();

    const std::vector<reco::PFBrem>& PFRecBrem() const { return pfBremVec_; }

    int trackId() const { return trackId_; }

    void addConvBremPFRecTrackRef(const reco::PFRecTrackRef& pfrectracksref);

    const std::vector<reco::PFRecTrackRef>& convBremPFRecTrackRef() const { return assoPFRecTrack_; }

    void addConvBremGsfPFRecTrackRef(const reco::GsfPFRecTrackRef& gsfpfrectracksref);

    const std::vector<reco::GsfPFRecTrackRef>& convBremGsfPFRecTrackRef() const { return assoGsfPFRecTrack_; }

  private:
    reco::GsfTrackRef gsfTrackRef_;

    reco::PFRecTrackRef kfPFRecTrackRef_;

    std::vector<reco::PFBrem> pfBremVec_;

    std::vector<reco::PFRecTrackRef> assoPFRecTrack_;

    std::vector<reco::GsfPFRecTrackRef> assoGsfPFRecTrack_;

    int trackId_;
  };

}  // namespace reco

#endif