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 : public PFBlockElement {
  public:

    PFBlockElementTrack() {} 

    PFBlockElementTrack(const PFRecTrackRef& ref);

    PFBlockElement* clone() const { return new PFBlockElementTrack(*this); }
    
    void Dump(std::ostream& out = std::cout, 
              const char* tab = " " ) const;

    virtual bool trackType(TrackType trType) const { 
      return (trackType_>>trType) & 1; 
    }

    virtual void setTrackType(TrackType trType, bool value) {
      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_;
    }
    
    PFRecTrackRef trackRefPF() const { return trackRefPF_; }
    
    reco::TrackRef trackRef() const { return trackRef_; }

    bool isSecondary() const { 
      return 
    trackType(T_FROM_DISP) || 
    trackType(T_FROM_GAMMACONV) || 
    trackType(T_FROM_V0); 
    }

    bool isPrimary() const{
      return trackType(T_TO_DISP); 
    }

    bool isLinkedToDisplacedVertex() const{
      return isSecondary() || isPrimary();
    }

    PFDisplacedTrackerVertexRef displacedVertexRef(TrackType trType) const {
      if (trType == T_TO_DISP)
    return displacedVertexDaughterRef_;
      else if (trType == T_FROM_DISP)
    return displacedVertexMotherRef_;
      else return PFDisplacedTrackerVertexRef();
    }

    void setDisplacedVertexRef(const PFDisplacedTrackerVertexRef& niref, TrackType trType) { 

      if (trType == T_TO_DISP) {
    displacedVertexDaughterRef_ = niref; setTrackType(trType,true);}
      else if (trType == T_FROM_DISP) {
        displacedVertexMotherRef_ = niref; setTrackType(trType,true);}
    } 
    
    reco::MuonRef muonRef() const { return muonRef_; }

    void setMuonRef(const MuonRef& muref) { 
      muonRef_=muref; setTrackType(MUON,true); 
    }

    ConversionRef convRef() const {return convRef_;} 

    void setConversionRef(const ConversionRef& convRef, TrackType trType) { 
      convRef_ = convRef; setTrackType(trType,true); 
    } 

    VertexCompositeCandidateRef V0Ref() const {return v0Ref_;} 

    void setV0Ref(const VertexCompositeCandidateRef& V0Ref, TrackType trType) { 
      v0Ref_ = V0Ref; setTrackType(trType,true); 
    } 


    
  private:

    PFRecTrackRef  trackRefPF_;

    reco::TrackRef trackRef_;

    unsigned int  trackType_;

    math::XYZPointF        positionAtECALEntrance_;
    
    PFDisplacedTrackerVertexRef  displacedVertexMotherRef_;

    PFDisplacedTrackerVertexRef  displacedVertexDaughterRef_;
                                 
    reco::MuonRef muonRef_;

    ConversionRef convRef_;      

    VertexCompositeCandidateRef v0Ref_;
  };
}

#endif