VertexFinder.h

Go to the documentation of this file.

#ifndef __L1Trigger_VertexFinder_VertexFinder_h__
#define __L1Trigger_VertexFinder_VertexFinder_h__

#include "DataFormats/Common/interface/Ptr.h"
#include "DataFormats/Math/interface/deltaPhi.h"
#include "DataFormats/TrackerCommon/interface/TrackerTopology.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "L1Trigger/VertexFinder/interface/AlgoSettings.h"
#include "L1Trigger/VertexFinder/interface/RecoVertex.h"

#include <algorithm>
#include <iterator>
#include <vector>

namespace l1tVertexFinder {

  typedef std::vector<L1Track> FitTrackCollection;
  typedef std::vector<RecoVertex<>> RecoVertexCollection;

  class VertexFinder {
  public:
    VertexFinder(FitTrackCollection& fitTracks, const AlgoSettings& settings) {
      fitTracks_ = fitTracks;
      settings_ = &settings;
    }
    ~VertexFinder() {}

    struct SortTracksByZ0 {
      inline bool operator()(const L1Track track0, const L1Track track1) { return (track0.z0() < track1.z0()); }
    };

    struct SortTracksByPt {
      inline bool operator()(const L1Track track0, const L1Track track1) {
        return (std::abs(track0.pt()) > std::abs(track1.pt()));
      }
    };


    unsigned int IterationsPerTrack() const { return double(iterations_) / double(fitTracks_.size()); }
    unsigned int numInputTracks() const { return fitTracks_.size(); }
    unsigned int NumIterations() const { return iterations_; }
    unsigned int numVertices() const { return vertices_.size(); }
    RecoVertex<> primaryVertex() const {
      if (pv_index_ < vertices_.size())
        return vertices_[pv_index_];
      else {
        edm::LogWarning("VertexFinder") << "PrimaryVertex::No Primary Vertex has been found.";
        return RecoVertex<>();
      }
    }
    unsigned int primaryVertexId() const { return pv_index_; }
    const std::vector<RecoVertex<>>& vertices() const { return vertices_; }
    const FitTrackCollection& fitTracks() const { return fitTracks_; }

    void findPrimaryVertex();
    void associatePrimaryVertex(double trueZ0);
    void GapClustering();
    float maxDistance(RecoVertex<> cluster0, RecoVertex<> cluster1);
    float minDistance(RecoVertex<> cluster0, RecoVertex<> cluster1);
    float meanDistance(RecoVertex<> cluster0, RecoVertex<> cluster1);
    float centralDistance(RecoVertex<> cluster0, RecoVertex<> cluster1);
    void agglomerativeHierarchicalClustering();
    void adaptiveVertexReconstruction();
    void fastHistoLooseAssociation();
    void fastHisto(const TrackerTopology* tTopo);
    void SortVerticesInPt() {
      std::sort(vertices_.begin(), vertices_.end(), [](const RecoVertex<>& vertex0, const RecoVertex<>& vertex1) {
        return (vertex0.pt() > vertex1.pt());
      });
    }
    void SortVerticesInZ0() {
      std::sort(vertices_.begin(), vertices_.end(), [](const RecoVertex<>& vertex0, const RecoVertex<>& vertex1) {
        return (vertex0.z0() < vertex1.z0());
      });
    }
    unsigned int numIterations() const { return iterations_; }
    unsigned int iterationsPerTrack() const { return double(iterations_) / double(fitTracks_.size()); }

    template <typename ForwardIterator, typename T>
    void strided_iota(ForwardIterator first, ForwardIterator last, T value, T stride) {
      while (first != last) {
        *first++ = value;
        value += stride;
      }
    }


    void AdaptiveVertexReconstruction();
    void AgglomerativeHierarchicalClustering();
    float CentralDistance(RecoVertex<> cluster0, RecoVertex<> cluster1);
    void computeAndSetVertexParameters(RecoVertex<>& vertex,
                                       const std::vector<float>& bin_centers,
                                       const std::vector<unsigned int>& counts);
    void DBSCAN();
    void FastHistoLooseAssociation();
    void FastHisto(const TrackerTopology* tTopo);
    void HPV();
    void Kmeans();
    float MaxDistance(RecoVertex<> cluster0, RecoVertex<> cluster1);
    float MinDistance(RecoVertex<> cluster0, RecoVertex<> cluster1);
    float MeanDistance(RecoVertex<> cluster0, RecoVertex<> cluster1);
    void PVR();

  private:
    const AlgoSettings* settings_;
    std::vector<RecoVertex<>> vertices_;
    unsigned int numMatchedVertices_;
    FitTrackCollection fitTracks_;
    unsigned int pv_index_;
    unsigned int iterations_;
  };

}  // end namespace l1tVertexFinder

#endif