TTUTrackingAlg.h

Go to the documentation of this file.

#ifndef TTUTRACKINGALG_H
#define TTUTRACKINGALG_H 1

// Include files
#include "L1Trigger/RPCTechnicalTrigger/interface/TTULogic.h"
#include "L1Trigger/RPCTechnicalTrigger/interface/TTUInput.h"

#include <iostream>
#include <vector>

class TTUTrackingAlg : public TTULogic {
public:
  TTUTrackingAlg();

  ~TTUTrackingAlg() override;  

  //... from TTULogic interface:

  bool process(const TTUInput&) override;

  void setBoardSpecs(const TTUBoardSpecs::TTUBoardConfig&) override;

  //...

  class Seed {
  public:
    Seed() {
      m_sectorId = -9;
      m_stationId = -1;
      m_tkLength = 0;
    };
    Seed(int _seId, int _stId, int _tl) {
      m_sectorId = _seId;
      m_stationId = _stId;
      m_tkLength = _tl;
    };

    bool operator==(const Seed& rhs) { return (m_sectorId == rhs.m_sectorId) && (m_stationId == rhs.m_stationId); };

    int m_sectorId;
    int m_stationId;
    int m_tkLength;
  };

  class Track {
  public:
    Track() { m_tracklength = 0; };

    Track(const Track&) = delete;
    Track(Track&&) = delete;
    Track& operator=(Track const&) = delete;
    Track& operator=(Track&&) = delete;

    void add(Seed* sd) {
      m_seeds.push_back(sd);
      ++m_tracklength;
    };

    void addnone() {
      if (not m_none) {
        m_none = std::make_unique<Seed>(0, 0, 0);
      }
      m_seeds.push_back(m_none.get());
      m_tracklength = -1;
    };

    void updateTrkLength() { m_tracklength = m_seeds.size(); };

    bool operator<(const Track& rhs) { return m_tracklength < rhs.m_tracklength; };

    int length() { return m_tracklength; };

    std::vector<Seed*> m_seeds;

  private:
    std::unique_ptr<Seed> m_none;
    int m_tracklength;
  };

  void setMinTrkLength(int val) { m_mintrklength = val; };

  template <class T>
  struct CompareMechanism {
    bool operator()(T const& a, T const& b) { return (*a) < (*b); }
  };

protected:
private:
  void runSeedBuster(const TTUInput&);

  void findNeighbors(Seed*, std::vector<Seed*>&);

  int executeTracker(Track*, std::vector<Seed*>&);

  void filter(Track*, std::vector<Seed*>&);

  void ghostBuster(Track*);

  void alignTracks();

  void cleanUp();

  int m_STscanorder[6];

  int m_SEscanorder[12];

  int m_mintrklength;

  std::vector<std::unique_ptr<Track>> m_tracks;

  std::vector<std::unique_ptr<Seed>> m_initialseeds;

  struct CompareSeeds {
    bool operator()(const Seed* a, const Seed* b) {
      //std::cout << (*a).m_sectorId << " " << (*b).m_sectorId << " "
      //<< (*a).m_stationId << " " << (*b).m_stationId << std::endl;
      return ((*a).m_sectorId == (*b).m_sectorId) && ((*a).m_stationId == (*b).m_stationId);
    }
  };

  struct SortBySector {
    bool operator()(const Seed* a, const Seed* b) { return ((*a).m_sectorId <= (*b).m_sectorId); }
  };

  struct SortByLayer {
    bool operator()(const Seed* a, const Seed* b) { return ((*a).m_stationId <= (*b).m_stationId); }
  };

  inline void print(const std::vector<Seed*>& seeds) {
    std::vector<Seed*>::const_iterator itr;
    for (itr = seeds.begin(); itr != seeds.end(); ++itr)
      std::cout << (*itr) << '\t';
    std::cout << '\n';
  };

  bool m_debug;
};
#endif  // TTUTRACKINGALG_H