NuclearInteractionTrackImporter.cc

Go to the documentation of this file.

#include "TrackFromParentImporter.h"
#include "DataFormats/ParticleFlowReco/interface/PFDisplacedTrackerVertex.h"

namespace {
  class NuclAdaptor {
  public:
    static bool check_importable(const reco::PFDisplacedTrackerVertexCollection::value_type& t) {
      const auto& vtx = t.displacedVertexRef();
      return ( vtx->isNucl() && vtx->position().rho() > 2.7 );
    }
    static const reco::PFRecTrackRefVector&
    get_track_refs(const reco::PFDisplacedTrackerVertexCollection::value_type& t) {
      return t.pfRecTracks();
    }
    static void set_element_info(reco::PFBlockElement* elem,
                 const edm::Ref<reco::PFDisplacedTrackerVertexCollection>& parref) {
      const reco::PFBlockElementTrack *tkelem = 
    static_cast<const reco::PFBlockElementTrack*>(elem);
      const reco::PFRecTrackRef& reftrack = tkelem->trackRefPF();
      reco::PFBlockElement::TrackType tkType = reco::PFBlockElement::DEFAULT;     
      if(parref->isIncomingTrack(reftrack)) 
    tkType = reco::PFBlockElement::T_TO_DISP;
      else if (parref->isOutgoingTrack(reftrack)) 
    tkType = reco::PFBlockElement::T_FROM_DISP;   
      elem->setDisplacedVertexRef(parref,tkType);
    }
  };
}

typedef pflow::importers::TrackFromParentImporter<reco::PFDisplacedTrackerVertexCollection,NuclAdaptor> NuclearInteractionTrackImporter;

DEFINE_EDM_PLUGIN(BlockElementImporterFactory, 
          NuclearInteractionTrackImporter, 
          "NuclearInteractionTrackImporter");