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