GoldenPattern.cc

Go to the documentation of this file.

#include <iostream>
#include <iomanip>
#include <cmath>

#include "FWCore/MessageLogger/interface/MessageLogger.h"

#include "L1Trigger/L1TMuonOverlap/interface/GoldenPattern.h"
#include "L1Trigger/L1TMuonOverlap/interface/OMTFConfiguration.h"
#include "L1Trigger/L1TMuonOverlap/interface/OMTFinput.h"

GoldenPattern::layerResult GoldenPattern::process1Layer1RefLayer(unsigned int iRefLayer,
                                 unsigned int iLayer,
                                 const int phiRefHit,
                                 const OMTFinput::vector1D & layerHits){

  GoldenPattern::layerResult aResult;

  int phiMean = meanDistPhi[iLayer][iRefLayer];
  int phiDist = exp2(OMTFConfiguration::nPdfAddrBits);
  for(auto itHit: layerHits){
    if(itHit>=(int)OMTFConfiguration::nPhiBins) continue;
    if(abs(itHit-phiMean-phiRefHit)<abs(phiDist)) phiDist = itHit-phiMean-phiRefHit;
  }

  phiDist+=exp2(OMTFConfiguration::nPdfAddrBits-1);

  if(phiDist<0 ||
     phiDist>exp2(OMTFConfiguration::nPdfAddrBits)-1){
    return aResult;
  }
  
  int pdfVal = pdfAllRef[iLayer][iRefLayer][phiDist];
  return GoldenPattern::layerResult(pdfVal,pdfVal>0);
}
int GoldenPattern::propagateRefPhi(int phiRef, int etaRef, unsigned int iRefLayer){

  unsigned int iLayer = 2; //MB2 
  //if(etaRef>101) iLayer = 7;//RE2
  return phiRef + meanDistPhi[iLayer][iRefLayer];
  
}
void GoldenPattern::addCount(unsigned int iRefLayer,
                 unsigned int iLayer,
                 const int phiRefHit,
                 const OMTFinput::vector1D & layerHits){

  int nHitsInLayer = 0;
  int phiDist = exp2(OMTFConfiguration::nPdfAddrBits);

  for(auto itHit: layerHits){
    if(itHit>=(int)OMTFConfiguration::nPhiBins) continue;
    if(abs(itHit-phiRefHit)<phiDist) phiDist = itHit-phiRefHit;
    ++nHitsInLayer;
  }
  if(nHitsInLayer>1 || nHitsInLayer==0) return;

  if(phiDist>=(int)OMTFConfiguration::nPhiBins/2) phiDist-=(int)OMTFConfiguration::nPhiBins;
  if(phiDist<=-(int)OMTFConfiguration::nPhiBins/2) phiDist+=(int)OMTFConfiguration::nPhiBins;
  
  int phiDistShift=phiDist+exp2(OMTFConfiguration::nPdfAddrBits-1);
  
  if(phiDistShift<0 ||
     phiDistShift>exp2(OMTFConfiguration::nPdfAddrBits)-1){
    return;
  }  
  if((int)iLayer==OMTFConfiguration::refToLogicNumber[iRefLayer]) ++meanDistPhiCounts[iLayer][iRefLayer];
  ++pdfAllRef[iLayer][iRefLayer][phiDistShift];
}
std::ostream & operator << (std::ostream &out, const GoldenPattern & aPattern){

  out <<"GoldenPattern "<< aPattern.theKey <<std::endl;
  out <<"Number of reference layers: "<<aPattern.meanDistPhi[0].size()
      <<", number of measurement layers: "<<aPattern.pdfAllRef.size()
      <<std::endl;

  if(!aPattern.meanDistPhi.size()) return out;
  if(!aPattern.pdfAllRef.size()) return out;

  out<<"Mean dist phi per layer:"<<std::endl;
  for (unsigned int iRefLayer=0;iRefLayer<aPattern.meanDistPhi[0].size();++iRefLayer){
    out<<"Ref layer: "<<iRefLayer<<" (";
    for (unsigned int iLayer=0;iLayer<aPattern.meanDistPhi.size();++iLayer){   
      out<<std::setw(3)<<aPattern.meanDistPhi[iLayer][iRefLayer]<<"\t";
    }
    out<<")"<<std::endl;
  }

  if(aPattern.meanDistPhiCounts.size()){
    out<<"Counts number per layer:"<<std::endl;
    for (unsigned int iRefLayer=0;iRefLayer<aPattern.meanDistPhi[0].size();++iRefLayer){
      out<<"Ref layer: "<<iRefLayer<<" (";
      for (unsigned int iLayer=0;iLayer<aPattern.meanDistPhi.size();++iLayer){   
    out<<aPattern.meanDistPhiCounts[iLayer][iRefLayer]<<"\t";
      }
      out<<")"<<std::endl;
    }
  }
/*
  out<<"PDF per layer:"<<std::endl;
  for (unsigned int iRefLayer=0;iRefLayer<aPattern.pdfAllRef[0].size();++iRefLayer){
    out<<"Ref layer: "<<iRefLayer;
    for (unsigned int iLayer=0;iLayer<aPattern.pdfAllRef.size();++iLayer){   
      out<<", measurement layer: "<<iLayer<<std::endl;
      for (unsigned int iPdf=0;iPdf<exp2(OMTFConfiguration::nPdfAddrBits);++iPdf){   
    out<<std::setw(2)<<aPattern.pdfAllRef[iLayer][iRefLayer][iPdf]<<" ";
      }
      out<<std::endl;
    }
  }
*/
  return out;
}
void GoldenPattern::reset(){

  GoldenPattern::vector1D meanDistPhi1D(OMTFConfiguration::nRefLayers);
  GoldenPattern::vector2D meanDistPhi2D(OMTFConfiguration::nLayers);
  meanDistPhi2D.assign(OMTFConfiguration::nLayers, meanDistPhi1D);
  meanDistPhi = meanDistPhi2D;
  meanDistPhiCounts = meanDistPhi2D;

  GoldenPattern::vector1D pdf1D(exp2(OMTFConfiguration::nPdfAddrBits));
  GoldenPattern::vector2D pdf2D(OMTFConfiguration::nRefLayers);
  GoldenPattern::vector3D pdf3D(OMTFConfiguration::nLayers);

  pdf2D.assign(OMTFConfiguration::nRefLayers,pdf1D);
  pdfAllRef.assign(OMTFConfiguration::nLayers,pdf2D);

}
void GoldenPattern::normalise(){

  for (unsigned int iRefLayer=0;iRefLayer<pdfAllRef[0].size();++iRefLayer){
    for (unsigned int iLayer=0;iLayer<pdfAllRef.size();++iLayer){   
      for (unsigned int iPdf=0;iPdf<pdfAllRef[iLayer][iRefLayer].size();++iPdf){   
    float pVal = log((float)pdfAllRef[iLayer][iRefLayer][iPdf]/meanDistPhiCounts[OMTFConfiguration::refToLogicNumber[iRefLayer]][iRefLayer]);
    if(pVal<log(OMTFConfiguration::minPdfVal)) continue;
    meanDistPhi[iLayer][iRefLayer]+=(iPdf - exp2(OMTFConfiguration::nPdfAddrBits-1))*pdfAllRef[iLayer][iRefLayer][iPdf];
    if((int)iLayer!=OMTFConfiguration::refToLogicNumber[iRefLayer]) meanDistPhiCounts[iLayer][iRefLayer]+=pdfAllRef[iLayer][iRefLayer][iPdf];
      }
    }
  }

  for (unsigned int iRefLayer=0;iRefLayer<meanDistPhi[0].size();++iRefLayer){
    for (unsigned int iLayer=0;iLayer<meanDistPhi.size();++iLayer){   
      if(meanDistPhiCounts.size() && meanDistPhiCounts[iLayer][iRefLayer]){
    if (meanDistPhiCounts[iLayer][iRefLayer]<1000)  meanDistPhi[iLayer][iRefLayer] = 0;
    else meanDistPhi[iLayer][iRefLayer] = rint((float)meanDistPhi[iLayer][iRefLayer]/meanDistPhiCounts[iLayer][iRefLayer]);      
      }
    }
  }
  
  const float minPlog =  log(OMTFConfiguration::minPdfVal);
  const unsigned int nPdfValBits = OMTFConfiguration::nPdfValBits; 
  float pVal;
  int digitisedVal, truncatedValue;
  for (unsigned int iRefLayer=0;iRefLayer<pdfAllRef[0].size();++iRefLayer){
    for (unsigned int iLayer=0;iLayer<pdfAllRef.size();++iLayer){   
      for (unsigned int iPdf=0;iPdf<pdfAllRef[iLayer][iRefLayer].size();++iPdf){   
    if(!meanDistPhiCounts[OMTFConfiguration::refToLogicNumber[iRefLayer]][iRefLayer] ||
       !pdfAllRef[iLayer][iRefLayer][iPdf]) continue;   
    pVal = log((float)pdfAllRef[iLayer][iRefLayer][iPdf]/meanDistPhiCounts[OMTFConfiguration::refToLogicNumber[iRefLayer]][iRefLayer]);
    if(pVal<minPlog || meanDistPhiCounts[iLayer][iRefLayer]<1000){
      pdfAllRef[iLayer][iRefLayer][iPdf] = 0;
      continue;
    }
    digitisedVal = rint((std::pow(2,nPdfValBits)-1) - (pVal/minPlog)*(std::pow(2,nPdfValBits)-1));
    truncatedValue  = 0 | (digitisedVal  & ((int)pow(2,nPdfValBits)-1));
    pdfAllRef[iLayer][iRefLayer][iPdf] = truncatedValue;
      }
    }
  } 

       
  vector3D  pdfAllRefTmp = pdfAllRef;

  const unsigned int nPdfAddrBits = 7;
  for (unsigned int iRefLayer=0;iRefLayer<pdfAllRef[0].size();++iRefLayer){
    for (unsigned int iLayer=0;iLayer<pdfAllRef.size();++iLayer){   
      for (unsigned int iPdf=0;iPdf<pdfAllRef[iLayer][iRefLayer].size();++iPdf){
    pdfAllRef[iLayer][iRefLayer][iPdf] = 0;   
    int index = iPdf - exp2(OMTFConfiguration::nPdfAddrBits-1)  - meanDistPhi[iLayer][iRefLayer] + exp2(nPdfAddrBits-1);       
    if(index<0 || index>exp2(nPdfAddrBits)-1) continue;
    pdfAllRef[iLayer][iRefLayer][index] = pdfAllRefTmp[iLayer][iRefLayer][iPdf];
      }
    }
  }
}
bool GoldenPattern::hasCounts(){

 for (unsigned int iRefLayer=0;iRefLayer<meanDistPhi[0].size();++iRefLayer){
    for (unsigned int iLayer=0;iLayer<meanDistPhi.size();++iLayer){   
      if(meanDistPhiCounts.size() && meanDistPhiCounts[iLayer][iRefLayer]) return true;
  }
 }
 return false;
}