TauImpactParameterInfo.h

Go to the documentation of this file.

#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 )

}

#endif // DataFormats_BTauReco_TauImpactParameterInfo_h