L1MuBMTrackFinder.h

Go to the documentation of this file.

//-------------------------------------------------
//
//
//--------------------------------------------------
#ifndef L1MUBM_TRACK_FINDER_H
#define L1MUBM_TRACK_FINDER_H

//---------------
// C++ Headers --
//---------------

#include <vector>
#include <iostream>
#include <memory>

//----------------------
// Base Class Headers --
//----------------------

#include <map>

//------------------------------------
// Collaborating Class Declarations --
//------------------------------------

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "FWCore/Utilities/interface/EDGetToken.h"
#include "FWCore/Utilities/interface/ESGetToken.h"
#include "DataFormats/L1DTTrackFinder/interface/L1MuDTChambPhContainer.h"
#include "DataFormats/L1TMuon/interface/RegionalMuonCand.h"
#include "DataFormats/L1TMuon/interface/L1MuBMTrack.h"
#include "DataFormats/L1TMuon/interface/L1MuBMTrackSegEta.h"
#include "DataFormats/L1TMuon/interface/L1MuBMTrackSegPhi.h"
#include "CondFormats/L1TObjects/interface/L1TMuonBarrelParams.h"
#include "CondFormats/DataRecord/interface/L1TMuonBarrelParamsRcd.h"

class L1MuBMTFConfig;
class L1MuBMSecProcMap;
class L1MuBMSecProcId;
class L1MuBMSectorProcessor;
class L1MuBMEtaProcessor;
class L1MuBMWedgeSorter;
class L1MuBMMuonSorter;
class L1MuRegionalCand;
class L1MuDTTrack;
class L1MuDTTrackSegPhi;
class L1MuDTTrackSegEta;

//              ---------------------
//              -- Class Interface --
//              ---------------------

class L1MuBMTrackFinder {
public:
  typedef l1t::RegionalMuonCandBxCollection::const_iterator TFtracks_const_iter;
  typedef l1t::RegionalMuonCandBxCollection::iterator TFtracks_iter;

  L1MuBMTrackFinder(const edm::ParameterSet& ps, edm::ConsumesCollector&& iC);

  virtual ~L1MuBMTrackFinder();

  void setup(edm::ConsumesCollector&&);

  void run(const edm::Event& e, const edm::EventSetup& c);

  void reset();

  inline int setAdd(int ust, int rel_add);

  const L1MuBMSectorProcessor* sp(const L1MuBMSecProcId&) const;

  inline const L1MuBMEtaProcessor* ep(int id) const { return m_epvec[id]; }

  inline const L1MuBMWedgeSorter* ws(int id) const { return m_wsvec[id]; }

  inline const L1MuBMMuonSorter* ms() const { return m_ms; }

  int numberOfTracks();

  TFtracks_const_iter begin(int bx);

  TFtracks_const_iter end(int bx);

  void clear();

  int numberOfTracks(int bx);

  static const L1MuBMTFConfig* config() { return m_config.get(); }

  l1t::RegionalMuonCandBxCollection& getcache() { return _cache; }
  l1t::RegionalMuonCandBxCollection& getcache0() { return _cache0; }
  L1MuBMTrackCollection& getcache1() { return _cache1; }
  L1MuBMTrackSegPhiCollection& getcache2() { return _cache2; }
  L1MuBMTrackSegEtaCollection& getcache3() { return _cache3; }

private:
  virtual void reconstruct(const edm::Event& e, const edm::EventSetup& c) {
    reset();
    run(e, c);
  }

private:
  l1t::RegionalMuonCandBxCollection _cache0;
  l1t::RegionalMuonCandBxCollection _cache;
  L1MuBMTrackCollection _cache1;
  L1MuBMTrackSegPhiCollection _cache2;
  L1MuBMTrackSegEtaCollection _cache3;

  L1MuBMSecProcMap* m_spmap;                 
  std::vector<L1MuBMEtaProcessor*> m_epvec;  
  std::vector<L1MuBMWedgeSorter*> m_wsvec;   
  L1MuBMMuonSorter* m_ms;                    

  static std::shared_ptr<L1MuBMTFConfig> m_config;  

  edm::EDGetTokenT<L1MuDTChambPhContainer> m_DTDigiToken;
  edm::ESGetToken<L1TMuonBarrelParams, L1TMuonBarrelParamsRcd> m_mbParamsToken;
};

#endif