Alignment/CommonAlignmentProducer/plugins/AlignmentTracksFromVertexCompositeCandidateSelectorModule.cc - master - githubmirror/cmssw

Alignment/CommonAlignmentProducer/plugins/AlignmentTracksFromVertexCompositeCandidateSelectorModule.cc

51 строка2 KB

mmusich

add AlignmentTracksFromVertexCompositeCandidateSelector and used it in ALCARECOTkAlV0s

19 янв 2024, 12:45

19 янв 2024, 12:45dc69109

#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "CommonTools/UtilAlgos/interface/ObjectSelector.h"

//the selectores used to select the tracks
#include "Alignment/CommonAlignmentProducer/interface/AlignmentTracksFromVertexCompositeCandidateSelector.h"

// the following include is necessary to clone all track branches
// including recoTrackExtras and TrackingRecHitsOwned.
// if you remove it the code will compile, but the cloned
// tracks have only the recoTracks branch!
#include "CommonTools/RecoAlgos/interface/TrackSelector.h"

struct TrackFromVertexCompositeCandidateConfigSelector {
  typedef std::vector<const reco::Track *> container;
  typedef container::const_iterator const_iterator;
  typedef reco::TrackCollection collection;

  TrackFromVertexCompositeCandidateConfigSelector(const edm::ParameterSet &cfg, edm::ConsumesCollector &&iC)
      : theBaseSelector(cfg, iC) {}

  const_iterator begin() const { return theSelectedTracks.begin(); }
  const_iterator end() const { return theSelectedTracks.end(); }
  size_t size() const { return theSelectedTracks.size(); }

  void select(const edm::Handle<reco::TrackCollection> &c, const edm::Event &evt, const edm::EventSetup &setup) {
    theSelectedTracks = theBaseSelector.select(c, evt, setup);
  }

private:
  container theSelectedTracks;

  AlignmentTrackFromVertexCompositeCandidateSelector theBaseSelector;
};

class AlignmentTrackFromVertexCompositeCandidateSelectorModule
    : public ObjectSelector<TrackFromVertexCompositeCandidateConfigSelector> {
public:
  AlignmentTrackFromVertexCompositeCandidateSelectorModule(const edm::ParameterSet &ps)
      : ObjectSelector<TrackFromVertexCompositeCandidateConfigSelector>(ps) {}
  static void fillDescriptions(edm::ConfigurationDescriptions &descriptions) {
    edm::ParameterSetDescription desc;
    desc.setComment("Alignment Tracks Selector from VertexCompositeCandidates");
    desc.add<edm::InputTag>("src", edm::InputTag("generalTracks"));
    desc.add<bool>("filter", false);
    desc.add<edm::InputTag>("vertexCompositeCandidates", edm::InputTag("generalV0Candidates:Kshort"));
    descriptions.addWithDefaultLabel(desc);
  }
};

DEFINE_FWK_MODULE(AlignmentTrackFromVertexCompositeCandidateSelectorModule);