PFBlockElement.h

Go to the documentation of this file.

#ifndef __PFBlockElement__
#define __PFBlockElement__

#include "DataFormats/ParticleFlowReco/interface/PFRecTrackFwd.h"
#include "DataFormats/ParticleFlowReco/interface/PFDisplacedTrackerVertex.h" 
#include "DataFormats/ParticleFlowReco/interface/PFClusterFwd.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"
#include "DataFormats/MuonReco/interface/MuonFwd.h"
#include "DataFormats/EgammaCandidates/interface/Conversion.h"
#include "DataFormats/EgammaCandidates/interface/ConversionFwd.h"
#include "DataFormats/Candidate/interface/VertexCompositeCandidate.h"
#include "DataFormats/Candidate/interface/VertexCompositeCandidateFwd.h"

#include <iostream>


namespace reco {
  class PFBlockElementCluster;
  class PFBlockElementTrack;
  
  
  class PFBlockElement {
  public:
    
    enum Type { 
      NONE=0,
      TRACK=1, 
      PS1=2, 
      PS2=3, 
      ECAL=4, 
      HCAL=5,
      GSF=6,
      BREM=7,
      HFEM=8,
      HFHAD=9,
      SC=10
    };

    enum TrackType {
      DEFAULT=0,
      T_FROM_DISP,
      T_TO_DISP,
      T_FROM_GAMMACONV,
      MUON,
      T_FROM_V0
    };

    PFBlockElement(Type type=NONE) :  
      type_(type), 
      locked_(false),
      index_( static_cast<unsigned>(-1) ) {
    }


    virtual ~PFBlockElement() {}
  
    virtual void Dump(std::ostream& out=std::cout, 
                      const char* tab=" " ) const;
    
    virtual PFBlockElement* clone() const = 0;
      
    void lock() {locked_ = true;}

    void unLock() {locked_ = false;}

    Type type() const { return type_; }

    virtual bool trackType(TrackType trType) const { return false; }

    virtual void setTrackType(TrackType trType, bool value) { 
             std::cout << "Error in PFBlockElement::setTrackType : this base class method is not implemented" << std::endl;}

    bool    locked() const {return locked_;}
    
    void     setIndex(unsigned index) { index_ = index; }

    unsigned index() const {return index_;} 

    virtual reco::TrackRef trackRef()  const {return reco::TrackRef(); }
    virtual PFRecTrackRef trackRefPF()  const {return PFRecTrackRef(); }
    virtual PFClusterRef clusterRef() const {return PFClusterRef(); }
    virtual PFDisplacedTrackerVertexRef displacedVertexRef(TrackType trType) const { return PFDisplacedTrackerVertexRef(); }
    virtual ConversionRef    convRef() const { return ConversionRef();}
    virtual MuonRef muonRef() const { return MuonRef(); }
    virtual VertexCompositeCandidateRef V0Ref()  const { return VertexCompositeCandidateRef(); }
    virtual void setDisplacedVertexRef(const PFDisplacedTrackerVertexRef& niref, TrackType trType) { 
      std::cout << "Error in PFBlockElement::setDisplacedVertexRef : this base class method is not implemented" << std::endl;}
    virtual void setConversionRef(const ConversionRef& convRef, TrackType trType) { 
      std::cout << "Error in PFBlockElement::setConversionRef : this base class method is not implemented" << std::endl;}
    virtual void setMuonRef(const MuonRef& muref) { 
      std::cout << "Error in PFBlockElement::setMuonRef : this base class method is not implemented" << std::endl;}
    virtual void setV0Ref(const VertexCompositeCandidateRef& v0ref,TrackType trType) { 
      
      std::cout << "Error in PFBlockElement::setV0Ref : this base class method is not implemented" << std::endl;
    }


    virtual bool isSecondary() const { return false; }
    virtual bool isPrimary() const { return false; }
    virtual bool isLinkedToDisplacedVertex() const {return false;}

    friend std::ostream& operator<<( std::ostream& out, 
                                     const PFBlockElement& element );
    
  protected:  
  
    Type     type_;

    bool       locked_;
    
    unsigned   index_;


  };
}
#endif