NuclearInteraction.h

Go to the documentation of this file.

#ifndef DATAFORMAT_NUCLEARINTERACTION_
#define DATAFORMAT_NUCLEARINTERACTION_

#include "DataFormats/TrackReco/interface/TrackFwd.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/TrajectorySeed/interface/TrajectorySeedCollection.h"
#include "DataFormats/VertexReco/interface/Vertex.h"

namespace reco {

  class NuclearInteraction {
  public:
    typedef edm::RefVector<TrajectorySeedCollection> TrajectorySeedRefVector;
    typedef edm::Ref<TrajectorySeedCollection> TrajectorySeedRef;
    typedef reco::Vertex::trackRef_iterator trackRef_iterator;
    typedef TrajectorySeedRefVector::iterator seedRef_iterator;

    NuclearInteraction() {}

    NuclearInteraction(const TrajectorySeedRefVector& tseeds, const reco::Vertex& vtx, double lkh) {
      seeds_ = tseeds;
      vertex_ = vtx;
      likelihood_ = lkh;
    }

    const edm::RefToBase<reco::Track>& primaryTrack() const { return *(vertex_.tracks_begin()); }

    int secondaryTracksSize() const { return vertex_.tracksSize() - 1; }

    trackRef_iterator secondaryTracks_begin() const { return vertex_.tracks_begin() + 1; }

    trackRef_iterator secondaryTracks_end() const { return vertex_.tracks_end(); }

    int seedsSize() const { return seeds_.size(); }

    const TrajectorySeedRefVector& seeds() { return seeds_; }

    seedRef_iterator seeds_begin() const { return seeds_.begin(); }

    seedRef_iterator seeds_end() const { return seeds_.end(); }

    const reco::Vertex& vertex() const { return vertex_; }

    double likelihood() const { return likelihood_; }

  private:
    //reco::Track                 refittedPrimaryTrack_; // to be included in a futur version

    TrajectorySeedRefVector seeds_;

    reco::Vertex vertex_;

    double likelihood_;
  };
}  // namespace reco
#endif