TauImpactParameterInfo

TauImpactParameterTrackData

Macros

Line Code
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47
#ifndef DataFormats_BTauReco_TauImpactParameterInfo_h
#define DataFormats_BTauReco_TauImpactParameterInfo_h

#include "DataFormats/BTauReco/interface/RefMacros.h"
#include "DataFormats/Common/interface/AssociationMap.h"
#include "DataFormats/GeometryCommonDetAlgo/interface/Measurement1D.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"
#include "DataFormats/BTauReco/interface/IsolatedTauTagInfo.h"

namespace reco {

  struct TauImpactParameterTrackData {
    Measurement1D transverseIp;
    Measurement1D ip3D;
  };

  typedef edm::AssociationMap<edm::OneToValue<reco::TrackCollection, reco::TauImpactParameterTrackData> >
      TrackTauImpactParameterAssociationCollection;

  typedef TrackTauImpactParameterAssociationCollection::value_type TrackTauImpactParameterAssociation;

  class TauImpactParameterInfo {
  public:
    TauImpactParameterInfo() {}
    virtual ~TauImpactParameterInfo() {}

    virtual TauImpactParameterInfo *clone() const { return new TauImpactParameterInfo(*this); }

    float discriminator(double, double, double, bool, bool) const;
    float discriminator() const;

    const TauImpactParameterTrackData *getTrackData(const reco::TrackRef &) const;
    void storeTrackData(const reco::TrackRef &, const TauImpactParameterTrackData &);

    const IsolatedTauTagInfoRef &getIsolatedTauTag() const;
    void setIsolatedTauTag(const IsolatedTauTagInfoRef &);

  private:
    TrackTauImpactParameterAssociationCollection trackDataMap;
    IsolatedTauTagInfoRef isolatedTaus;
  };

  DECLARE_EDM_REFS(TauImpactParameterInfo)

}  // namespace reco

#endif  // DataFormats_BTauReco_TauImpactParameterInfo_h