DataFormats/TrackerRecHit2D/interface/SiTrackerMultiRecHit.h - master - githubmirror/cmssw

DataFormats/TrackerRecHit2D/interface/SiTrackerMultiRecHit.h
75 строк3 KB
Matti Kortelainen
Declare SiTrackerMultiRecHit::cloneSH() to override
15 янв 2020, 20:37
15 янв 2020, 20:375b59a35
#ifndef SiTrackerMultiRecHit_H
#define SiTrackerMultiRecHit_H

#include "DataFormats/TrackerRecHit2D/interface/BaseTrackerRecHit.h"
#include "DataFormats/TrackingRecHit/interface/TrackingRecHitFwd.h"
#include <vector>
#include <map>
/*
A rechit type suitable for tracking algorithm that use soft hit-to-track assignement, 
such as the Deterministic Annealing Filter (DAF) or the Multi Track Filter (MTF).
it contains an OwnVector with the component rechits and a vector of weights	
*/
class SiTrackerMultiRecHit : public BaseTrackerRecHit {
public:
  typedef BaseTrackerRecHit Base;
  SiTrackerMultiRecHit() : theHits(), theWeights(), annealing_(0) {}
  ~SiTrackerMultiRecHit() override {}

  SiTrackerMultiRecHit(const LocalPoint&,
                       const LocalError&,
                       GeomDet const& idet,
                       const std::vector<std::pair<const TrackingRecHit*, float> >&,
                       double);

  SiTrackerMultiRecHit* clone() const override { return new SiTrackerMultiRecHit(*this); }
#ifdef NO_DICT
  RecHitPointer cloneSH() const override { return std::make_shared<SiTrackerMultiRecHit>(*this); }
#endif

  //  virtual int dimension() const {return 2;}
  int dimension() const override;
  void getKfComponents(KfComponentsHolder& holder) const override;

  // at the momement nobody care of MultiHit!!!
  // used by trackMerger (to be improved)
  OmniClusterRef const& firstClusterRef() const override {
    return static_cast<BaseTrackerRecHit const&>(theHits.front()).firstClusterRef();
  }

  /// Access to component RecHits (if any)
  std::vector<const TrackingRecHit*> recHits() const override;

  /// Non-const access to component RecHits (if any)
  std::vector<TrackingRecHit*> recHits() override;

  //vector of weights
  std::vector<float> const& weights() const { return theWeights; }
  std::vector<float>& weights() { return theWeights; }

  //returns the weight for the i component
  using TrackingRecHit::weight;
  float weight(unsigned int i) const { return theWeights[i]; }
  float& weight(unsigned int i) { return theWeights[i]; }

  //get the annealing
  virtual double getAnnealingFactor() const { return annealing_; }

  bool sharesInput(const TrackingRecHit* other, SharedInputType what) const override;

private:
  edm::OwnVector<TrackingRecHit> theHits;
  std::vector<float> theWeights;
  double annealing_;
};

// Comparison operators
inline bool operator<(const SiTrackerMultiRecHit& one, const SiTrackerMultiRecHit& other) {
  if (one.geographicalId() < other.geographicalId()) {
    return true;
  } else {
    return false;
  }
}

#endif