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): 
  theEtaCode(iEta), thePtCode(iPt), theCharge(iCharge){}
    
    inline bool operator< (const Key & o) const {
    if (thePtCode > o.thePtCode) return true;
    else if (thePtCode==o.thePtCode && theCharge < o.theCharge) return true;
    else if (theCharge*thePtCode==o.theCharge*o.thePtCode && theEtaCode<o.theEtaCode) return true;
    else return false;
  }
   
  bool operator==(const Key& o) const {
    return theEtaCode==o.theEtaCode && thePtCode==o.thePtCode && theCharge==o.theCharge;
  }
  
  friend std::ostream & operator << (std::ostream &out, const Key & o) {
    out << "Key: (eta="<<o.theEtaCode<<", pt="<<o.thePtCode<<", charge="<<o.theCharge<<")";
    return out;
  }

  int theEtaCode;
  unsigned int thePtCode; 
  int          theCharge;

 };
// 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();

  int propagateRefPhi(int phiRef, int etaRef, unsigned int iRefLayer);

  bool hasCounts();

 private:

  Key theKey;
  
  vector3D pdfAllRef;

  vector2D meanDistPhi;

  vector2D meanDistPhiCounts;

  const OMTFConfiguration  * myOmtfConfig;

};
#endif