ConvBremSeedProducer.h

Go to the documentation of this file.

#include <memory>

// user include files
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/EDProducer.h"

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/MakerMacros.h"

#include "FWCore/ParameterSet/interface/ParameterSet.h"

//COLLECTION
#include "DataFormats/TrajectorySeed/interface/TrajectorySeed.h"
#include "DataFormats/TrackerRecHit2D/interface/SiPixelRecHitCollection.h"
#include "DataFormats/TrackerRecHit2D/interface/SiStripMatchedRecHit2DCollection.h"
#include "DataFormats/TrackerRecHit2D/interface/SiStripRecHit2DCollection.h"
#include "DataFormats/ParticleFlowReco/interface/PFClusterFwd.h"
#include "DataFormats/TrackReco/interface/Track.h"

#include "RecoTracker/TkDetLayers/interface/GeometricSearchTracker.h"

#include "DataFormats/TrackingRecHit/interface/TrackingRecHitFwd.h"
#include "CommonTools/BaseParticlePropagator/interface/BaseParticlePropagator.h"

class DetLayer;
class TrajectoryStateOnSurface;
class ParticlePropagator;
class TrackerLayer;
class MagneticField;
class TrackerInteractionGeometry;
class TrackerGeometry;
class MagneticFieldMap;
class PropagatorWithMaterial;
class KFUpdator;
class TransientTrackingRecHitBuilder;

class ConvBremSeedProducer : public edm::EDProducer {
  typedef SiStripRecHit2DCollection::const_iterator StDetMatch;
  typedef SiPixelRecHitCollection::const_iterator PiDetMatch;
  typedef SiStripMatchedRecHit2DCollection::const_iterator MatDetMatch;
  typedef SiStripRecHit2DCollection::DetSet StDetSet;
  typedef SiPixelRecHitCollection::DetSet PiDetSet;
  typedef SiStripMatchedRecHit2DCollection::DetSet MatDetSet;
  typedef GeometricSearchDet::DetWithState DetWithState;

public:
  explicit ConvBremSeedProducer(const edm::ParameterSet&);
  ~ConvBremSeedProducer() override;

private:
  void beginRun(const edm::Run&, const edm::EventSetup&) override;
  void produce(edm::Event&, const edm::EventSetup&) override;
  void endRun(const edm::Run&, const edm::EventSetup&) override;
  void initializeLayerMap();
  std::vector<const DetLayer*> theLayerMap;
  TrajectoryStateOnSurface makeTrajectoryState(const DetLayer* layer,
                                               const ParticlePropagator& pp,
                                               const MagneticField* field) const;
  const DetLayer* detLayer(const TrackerLayer& layer, float zpos) const;

  bool isGsfTrack(const reco::Track&, const TrackingRecHit*);

  int GoodCluster(const BaseParticlePropagator& bpg,
                  const reco::PFClusterCollection& pfc,
                  float minep,
                  bool sec = false);

  std::vector<bool> sharedHits(const std::vector<std::pair<TrajectorySeed, std::pair<GlobalVector, float> > >&);

  edm::ParameterSet conf_;
  const GeometricSearchTracker* geomSearchTracker_;
  const TrackerInteractionGeometry* geometry_;
  const TrackerGeometry* tracker_;
  const MagneticField* magfield_;
  const MagneticFieldMap* fieldMap_;
  const PropagatorWithMaterial* propagator_;
  const KFUpdator* kfUpdator_;
  const TransientTrackingRecHitBuilder* hitBuilder_;
  std::vector<const DetLayer*> layerMap_;
  int negLayerOffset_;
  math::XYZVector B_;
};