AlignmentProducer.h

Go to the documentation of this file.

#ifndef Alignment_CommonAlignmentAlgorithm_TrackerAlignmentProducer_h
#define Alignment_CommonAlignmentAlgorithm_TrackerAlignmentProducer_h


#include "Alignment/CommonAlignmentProducer/interface/AlignmentProducerBase.h"
#include "FWCore/Framework/interface/ESProducerLooper.h"
#include "FWCore/Framework/interface/Run.h"

class AlignmentProducer : public edm::ESProducerLooper, public AlignmentProducerBase {
public:
  AlignmentProducer(const edm::ParameterSet&);

  ~AlignmentProducer() override = default;

  virtual std::shared_ptr<TrackerGeometry> produceTracker(const TrackerDigiGeometryRecord&);

  void beginOfJob(const edm::EventSetup&) override;

  void endOfJob() override;

  void startingNewLoop(unsigned int iLoop) override;

  Status endOfLoop(const edm::EventSetup&, unsigned int iLoop) override;

  void beginRun(const edm::Run&, const edm::EventSetup&) override;

  void endRun(const edm::Run&, const edm::EventSetup&) override;

  void beginLuminosityBlock(const edm::LuminosityBlock&, const edm::EventSetup&) override;

  void endLuminosityBlock(const edm::LuminosityBlock&, const edm::EventSetup&) override;

  Status duringLoop(const edm::Event&, const edm::EventSetup&) override;

private:
  bool getTrajTrackAssociationCollection(const edm::Event&, edm::Handle<TrajTrackAssociationCollection>&) override;
  bool getBeamSpot(const edm::Event&, edm::Handle<reco::BeamSpot>&) override;
  bool getTkFittedLasBeamCollection(const edm::Run&, edm::Handle<TkFittedLasBeamCollection>&) override;
  bool getTsosVectorCollection(const edm::Run&, edm::Handle<TsosVectorCollection>&) override;
  bool getAliClusterValueMap(const edm::Event&, edm::Handle<AliClusterValueMap>&) override;

  const unsigned int maxLoops_;  

  edm::EDGetTokenT<TrajTrackAssociationCollection> trajTrackAssociationCollectionToken_;
  edm::EDGetTokenT<reco::BeamSpot> bsToken_;
  edm::EDGetTokenT<TkFittedLasBeamCollection> tkFittedLasBeamCollectionToken_;
  edm::EDGetTokenT<TsosVectorCollection> tsosVectorCollectionToken_;
  edm::EDGetTokenT<AliClusterValueMap> aliClusterValueMapToken_;
};

//------------------------------------------------------------------------------
inline bool AlignmentProducer::getTrajTrackAssociationCollection(const edm::Event& event,
                                                                 edm::Handle<TrajTrackAssociationCollection>& result) {
  return event.getByToken(trajTrackAssociationCollectionToken_, result);
}

//------------------------------------------------------------------------------
inline bool AlignmentProducer::getBeamSpot(const edm::Event& event, edm::Handle<reco::BeamSpot>& result) {
  return event.getByToken(bsToken_, result);
}

//------------------------------------------------------------------------------
inline bool AlignmentProducer::getTkFittedLasBeamCollection(const edm::Run& run,
                                                            edm::Handle<TkFittedLasBeamCollection>& result) {
  return run.getByToken(tkFittedLasBeamCollectionToken_, result);
}

//------------------------------------------------------------------------------
inline bool AlignmentProducer::getTsosVectorCollection(const edm::Run& run, edm::Handle<TsosVectorCollection>& result) {
  return run.getByToken(tsosVectorCollectionToken_, result);
}

//------------------------------------------------------------------------------
inline bool AlignmentProducer::getAliClusterValueMap(const edm::Event& event, edm::Handle<AliClusterValueMap>& result) {
  return event.getByToken(aliClusterValueMapToken_, result);
}

#endif