GoldenPattern.h

Go to the documentation of this file.

#ifndef OMTF_GoldenPattern_H
#define OMTF_GoldenPattern_H
 
#include <vector>
#include <ostream>
 
#include "L1Trigger/L1TMuonOverlap/interface/OMTFinput.h"
 
class OMTFConfigMaker;
class OMTFProcessor;
class OMTFConfiguration;
// Key
struct Key {
  Key(int iEta = 99, unsigned int iPt = 0, int iCharge = 0, unsigned int iNumber = 999)
      : theEtaCode(iEta), thePtCode(iPt), theCharge(iCharge), theNumber(iNumber) {}
 
  inline bool operator<(const Key &o) const { return (theNumber < o.theNumber); }
 
  bool operator==(const Key &o) const {
    return theEtaCode == o.theEtaCode && thePtCode == o.thePtCode && theCharge == o.theCharge &&
           theNumber == o.theNumber;
  }
 
  friend std::ostream &operator<<(std::ostream &out, const Key &o) {
    out << "Key_" << o.theNumber << ": (eta=" << o.theEtaCode << ", pt=" << o.thePtCode << ", charge=" << o.theCharge
        << ")";
    return out;
  }
 
  unsigned int number() const { return theNumber; }
 
  int theEtaCode;
  unsigned int thePtCode;
  int theCharge;
  unsigned int theNumber;
};
// Golden Pattern
class OMTFinput;
 
class GoldenPattern {
public:
  typedef std::vector<int> vector1D;
  typedef std::vector<vector1D> vector2D;
  typedef std::vector<vector2D> vector3D;
  typedef std::pair<int, bool> layerResult;
 
  //
  // GoldenPatterns methods
  //
  GoldenPattern(const Key &aKey, const OMTFConfiguration *omtfConfig) : theKey(aKey), myOmtfConfig(omtfConfig) {}
 
  Key key() const { return theKey; }
 
  void setMeanDistPhi(const vector2D &aMeanDistPhi) { meanDistPhi = aMeanDistPhi; }
 
  const vector2D &getMeanDistPhi() const { return meanDistPhi; }
 
  const vector3D &getPdf() const { return pdfAllRef; }
 
  void setPdf(const vector3D &aPdf) { pdfAllRef = aPdf; }
 
  int meanDistPhiValue(unsigned int iLayer, unsigned int iRefLayer) const { return meanDistPhi[iLayer][iRefLayer]; }
 
  int pdfValue(unsigned int iLayer, unsigned int iRefLayer, unsigned int iBin) const {
    return pdfAllRef[iLayer][iRefLayer][iBin];
  }
 
  GoldenPattern::layerResult process1Layer1RefLayer(unsigned int iRefLayer,
                                                    unsigned int iLayer,
                                                    const int refPhi,
                                                    const OMTFinput::vector1D &layerHits);
 
  friend std::ostream &operator<<(std::ostream &out, const GoldenPattern &aPattern);
 
  void addCount(unsigned int iRefLayer, unsigned int iLayer, const int refPhi, const OMTFinput::vector1D &layerHits);
 
  void reset();
 
  void normalise(unsigned int nPdfAddrBits);
 
  int propagateRefPhi(int phiRef, int etaRef, unsigned int iRefLayer);
 
  bool hasCounts();
 
private:
  Key theKey;
 
  vector3D pdfAllRef;
 
  vector2D meanDistPhi;
 
  vector2D meanDistPhiCounts;
 
  const OMTFConfiguration *myOmtfConfig;
};
#endif