#include <MahiFit.h>

Public Types
typedef BXVector::Index	Index

Public Member Functions
void	doFit (std::array< float, 4 > &correctedOutput, const int nbx) const

	MahiFit ()

void	phase1Apply (const HBHEChannelInfo &channelData, float &reconstructedEnergy, float &soiPlusOneEnergy, float &reconstructedTime, bool &useTriple, float &chi2) const

void	phase1Debug (const HBHEChannelInfo &channelData, MahiDebugInfo &mdi) const

void	setParameters (bool iDynamicPed, double iTS4Thresh, double chiSqSwitch, bool iApplyTimeSlew, HcalTimeSlew::BiasSetting slewFlavor, bool iCalculateArrivalTime, int iTimeAlgo, double iThEnergeticPulses, double iMeanTime, double iTimeSigmaHPD, double iTimeSigmaSiPM, const std::vector< int > &iActiveBXs, int iNMaxItersMin, int iNMaxItersNNLS, double iDeltaChiSqThresh, double iNnlsThresh)

void	setPulseShapeTemplate (int pulseShapeId, const HcalPulseShapes &ps, bool hasTimeInfo, const HcalTimeSlew *hcalTimeSlewDelay, unsigned int nSamples, const float gain)

	~MahiFit ()

Public Attributes
const HcalTimeSlew *	hcalTimeSlewDelay_ = nullptr

float	thEnergeticPulses_

float	thEnergeticPulsesFC_

Private Types
typedef std::pair< int, std::shared_ptr< FitterFuncs::PulseShapeFunctor > >	ShapeWithId

Private Member Functions
float	calculateArrivalTime (const unsigned int iBX) const

float	calculateChiSq () const

float	ccTime (const float itQ) const

const float	minimize () const

void	nnls () const

void	nnlsConstrainParameter (Index minratioidx) const

void	nnlsUnconstrainParameter (Index idxp) const

void	onePulseMinimize () const

void	resetPulseShapeTemplate (int pulseShapeId, const HcalPulseShapes &ps, unsigned int nSamples)

void	resetWorkspace () const

void	solveSubmatrix (PulseMatrix &mat, PulseVector &invec, PulseVector &outvec, unsigned nP) const

void	updateCov (const SampleMatrix &invCovMat) const

void	updatePulseShape (const float itQ, FullSampleVector &pulseShape, FullSampleVector &pulseDeriv, FullSampleMatrix &pulseCov) const

Private Attributes
std::vector< int >	activeBXs_

bool	applyTimeSlew_

int	bxOffsetConf_

unsigned int	bxSizeConf_

bool	calculateArrivalTime_

float	chiSqSwitch_

int	cntsetPulseShape_ = 0

int	currentPulseShapeId_ = INT_MIN

float	deltaChiSqThresh_

bool	dynamicPed_

std::vector< ShapeWithId >	knownPulseShapes_

float	meanTime_

int	nMaxItersMin_

int	nMaxItersNNLS_

float	nnlsThresh_

MahiNnlsWorkspace	nnlsWork_

float	norm_ = (1.f / std::sqrt(12))

FitterFuncs::PulseShapeFunctor *	psfPtr_ = nullptr

HcalTimeSlew::BiasSetting	slewFlavor_

int	timeAlgo_

float	timeSigmaHPD_

float	timeSigmaSiPM_

float	ts4Thresh_

float	tsDelay1GeV_ = 0.f

Static Private Attributes
static constexpr int	pedestalBX_ = 100

static constexpr float	timeLimit_ = 12.5f

Detailed Description

Definition at line 96 of file MahiFit.h.

Member Typedef Documentation

◆ Index

typedef BXVector::Index MahiFit::Index

Definition at line 136 of file MahiFit.h.

◆ ShapeWithId

typedef std::pair<int, std::shared_ptr<FitterFuncs::PulseShapeFunctor> > MahiFit::ShapeWithId

private

Definition at line 143 of file MahiFit.h.

Constructor & Destructor Documentation

◆ MahiFit()

MahiFit::MahiFit ( )

Definition at line 5 of file MahiFit.cc.

5 {}

◆ ~MahiFit()

MahiFit::~MahiFit ( )

inline

Definition at line 99 of file MahiFit.h.

99 {}

Member Function Documentation

◆ calculateArrivalTime()

float MahiFit::calculateArrivalTime ( const unsigned int iBX ) const

private

Definition at line 419 of file MahiFit.cc.

References MahiNnlsWorkspace::amplitudes, MahiNnlsWorkspace::ampVec, MahiNnlsWorkspace::bxOffset, MahiNnlsWorkspace::bxs, nnlsWork_, MahiNnlsWorkspace::nPulseTot, MahiNnlsWorkspace::pulseDerivMat, MahiNnlsWorkspace::pulseMat, submitPVValidationJobs::t, and timeLimit_.

Referenced by doFit().

                                                                     {
   if (nnlsWork_.nPulseTot > 1) {
     SamplePulseMatrix pulseDerivMatTMP = nnlsWork_.pulseDerivMat;
     for (unsigned int iBX = 0; iBX < nnlsWork_.nPulseTot; ++iBX) {
       nnlsWork_.pulseDerivMat.col(iBX) = pulseDerivMatTMP.col(nnlsWork_.bxs.coeff(iBX) + nnlsWork_.bxOffset);
     }
   }
 
   for (unsigned int iBX = 0; iBX < nnlsWork_.nPulseTot; ++iBX) {
     nnlsWork_.pulseDerivMat.col(iBX) *= nnlsWork_.ampVec.coeff(iBX);
   }
 
   //FIXME: avoid solve full equation for one element
   SampleVector residuals = nnlsWork_.pulseMat * nnlsWork_.ampVec - nnlsWork_.amplitudes;
   PulseVector solution = nnlsWork_.pulseDerivMat.colPivHouseholderQr().solve(residuals);
   float t = std::clamp((float)solution.coeff(itIndex), -timeLimit_, timeLimit_);
 
   return t;
 }

◆ calculateChiSq()

float MahiFit::calculateChiSq ( ) const

private

Definition at line 543 of file MahiFit.cc.

References MahiNnlsWorkspace::amplitudes, MahiNnlsWorkspace::ampVec, MahiNnlsWorkspace::covDecomp, nnlsWork_, and MahiNnlsWorkspace::pulseMat.

Referenced by minimize().

                                     {
   return (nnlsWork_.covDecomp.matrixL().solve(nnlsWork_.pulseMat * nnlsWork_.ampVec - nnlsWork_.amplitudes))
       .squaredNorm();
 }

◆ ccTime()

float MahiFit::ccTime ( const float itQ ) const

private

Definition at line 354 of file MahiFit.cc.

References MahiNnlsWorkspace::amplitudes, applyTimeSlew_, HcalSpecialTimes::DEFAULT_ccTIME, HcalTimeSlew::delay(), dumpMFGeometry_cfg::delta, HLT_2024v14_cff::distance, f, FitterFuncs::PulseShapeFunctor::getPulseShape(), hcalTimeSlewDelay_, meanTime_, nnlsWork_, bTagMiniDQMTaggers::numerator, psfPtr_, FitterFuncs::PulseShapeFunctor::singlePulseShapeFuncMahi(), slewFlavor_, mathSSE::sqrt(), FrontierCondition_GT_autoExpress_cfi::t0, thEnergeticPulsesFC_, tsDelay1GeV_, MahiNnlsWorkspace::tsOffset, MahiNnlsWorkspace::tsSize, and geometryCSVtoXML::xx.

Referenced by doFit().

                                            {
   // those conditions are now on data time slices, can be done on the fitted pulse i.e. using nlsWork_.ampVec.coeff(itIndex);
 
   // Selecting energetic hits - Fitted Energy > 20 GeV
   if (itQ < thEnergeticPulsesFC_)
     return HcalSpecialTimes::DEFAULT_ccTIME;
 
   // Rejecting late hits  Energy in TS[3] > (Energy in TS[4] and TS[5])
   // With small OOTPU (Energy in TS[0] ,TS[1] and TS[2]) < 5 GeV
   // To speed up check around the fitted time (?)
 
   // distanze as in formula of page 6
   // https://indico.cern.ch/event/1142347/contributions/4793749/attachments/2412936/4129323/HCAL%20timing%20update.pdf
 
   float t0 = meanTime_;
 
   if (applyTimeSlew_) {
     if (itQ <= 1.f)
       t0 += tsDelay1GeV_;
     else
       t0 += hcalTimeSlewDelay_->delay(float(itQ), slewFlavor_);
   }
 
   float ccTime = 0.f;
   float distance_delta_max = 0.f;
 
   std::array<double, hcal::constants::maxSamples> pulseN;
 
   // making 8 TS out of the template i.e. 200 points
   int TS_SOIandAfter = 25 * (nnlsWork_.tsSize - nnlsWork_.tsOffset);
   int TS_beforeSOI = -25 * nnlsWork_.tsOffset;
 
   for (int deltaNS = TS_beforeSOI; deltaNS < TS_SOIandAfter; ++deltaNS) {  // from -75ns and + 125ns
     const float xx = t0 + deltaNS;
 
     psfPtr_->singlePulseShapeFuncMahi(&xx);
     psfPtr_->getPulseShape(pulseN);
 
     float pulse2 = 0;
     float norm2 = 0;
     float numerator = 0;
     //
     int delta = 4 - nnlsWork_.tsOffset;  // like in updatePulseShape
 
     // rm TS0 and TS7, to speed up and reduce noise
     for (unsigned int iTS = 1; iTS < (nnlsWork_.tsSize - 1); ++iTS) {
       //pulseN[iTS] is the area of the template
       float norm = nnlsWork_.amplitudes.coeffRef(iTS);
 
       //  Finding the distance after each iteration.
       numerator += norm * pulseN[iTS + delta];
       pulse2 += pulseN[iTS + delta] * pulseN[iTS + delta];
       norm2 += norm * norm;
     }
 
     float distance = numerator / sqrt(pulse2 * norm2);
     if (distance > distance_delta_max) {
       distance_delta_max = distance;
       ccTime = deltaNS;
     }
   }
 
   return ccTime;
 }

◆ doFit()

void MahiFit::doFit	(	std::array< float, 4 > &	correctedOutput,
		const int	nbx
	)		const

Definition at line 135 of file MahiFit.cc.

Referenced by phase1Apply().

                                                                       {
   unsigned int bxSize = 1;
 
   if (nbx == 1) {
     nnlsWork_.bxOffset = 0;
   } else {
     bxSize = bxSizeConf_;
     nnlsWork_.bxOffset = static_cast<int>(nnlsWork_.tsOffset) >= bxOffsetConf_ ? bxOffsetConf_ : nnlsWork_.tsOffset;
   }
 
   nnlsWork_.nPulseTot = bxSize;
 
   if (dynamicPed_)
     nnlsWork_.nPulseTot++;
   nnlsWork_.bxs.setZero(nnlsWork_.nPulseTot);
 
   if (nbx == 1) {
     nnlsWork_.bxs.coeffRef(0) = 0;
   } else {
     for (unsigned int iBX = 0; iBX < bxSize; ++iBX) {
       nnlsWork_.bxs.coeffRef(iBX) =
           activeBXs_[iBX] -
           ((static_cast<int>(nnlsWork_.tsOffset) + activeBXs_[0]) >= 0 ? 0 : (nnlsWork_.tsOffset + activeBXs_[0]));
     }
   }
 
   nnlsWork_.maxoffset = nnlsWork_.bxs.coeff(bxSize - 1);
   if (dynamicPed_)
     nnlsWork_.bxs[nnlsWork_.nPulseTot - 1] = pedestalBX_;
 
   nnlsWork_.pulseMat.setZero(nnlsWork_.tsSize, nnlsWork_.nPulseTot);
   nnlsWork_.pulseDerivMat.setZero(nnlsWork_.tsSize, nnlsWork_.nPulseTot);
 
   FullSampleVector pulseShapeArray;
   FullSampleVector pulseDerivArray;
   FullSampleMatrix pulseCov;
 
   int offset = 0;
   for (unsigned int iBX = 0; iBX < nnlsWork_.nPulseTot; ++iBX) {
     offset = nnlsWork_.bxs.coeff(iBX);
 
     if (offset == pedestalBX_) {
       nnlsWork_.pulseMat.col(iBX) = SampleVector::Ones(nnlsWork_.tsSize);
       nnlsWork_.pulseDerivMat.col(iBX) = SampleVector::Zero(nnlsWork_.tsSize);
     } else {
       pulseShapeArray.setZero(nnlsWork_.tsSize + nnlsWork_.maxoffset + nnlsWork_.bxOffset);
       if (calculateArrivalTime_)
         pulseDerivArray.setZero(nnlsWork_.tsSize + nnlsWork_.maxoffset + nnlsWork_.bxOffset);
       pulseCov.setZero(nnlsWork_.tsSize + nnlsWork_.maxoffset + nnlsWork_.bxOffset,
                        nnlsWork_.tsSize + nnlsWork_.maxoffset + nnlsWork_.bxOffset);
       nnlsWork_.pulseCovArray[iBX].setZero(nnlsWork_.tsSize, nnlsWork_.tsSize);
 
       updatePulseShape(
           nnlsWork_.amplitudes.coeff(nnlsWork_.tsOffset + offset), pulseShapeArray, pulseDerivArray, pulseCov);
 
       nnlsWork_.pulseMat.col(iBX) = pulseShapeArray.segment(nnlsWork_.maxoffset - offset, nnlsWork_.tsSize);
       if (calculateArrivalTime_)
         nnlsWork_.pulseDerivMat.col(iBX) = pulseDerivArray.segment(nnlsWork_.maxoffset - offset, nnlsWork_.tsSize);
       nnlsWork_.pulseCovArray[iBX] = pulseCov.block(
           nnlsWork_.maxoffset - offset, nnlsWork_.maxoffset - offset, nnlsWork_.tsSize, nnlsWork_.tsSize);
     }
   }
 
   const float chiSq = minimize();
 
   bool foundintime = false;
   unsigned int ipulseintime = 0;
 
   for (unsigned int iBX = 0; iBX < nnlsWork_.nPulseTot; ++iBX) {
     if (nnlsWork_.bxs.coeff(iBX) == 0) {
       ipulseintime = iBX;
       foundintime = true;
       break;
     }
   }
 
   if (foundintime) {
     correctedOutput.at(0) = nnlsWork_.ampVec.coeff(ipulseintime);       //charge
     correctedOutput.at(3) = nnlsWork_.ampVec.coeff(ipulseintime + 1U);  //charge for SOI+1
     if (correctedOutput.at(0) != 0) {
       // fixME store the timeslew
       float arrivalTime = 0.f;
       if (calculateArrivalTime_ && timeAlgo_ == 1)
         arrivalTime = calculateArrivalTime(ipulseintime);
       else if (calculateArrivalTime_ && timeAlgo_ == 2)
         arrivalTime = ccTime(nnlsWork_.ampVec.coeff(ipulseintime));
       correctedOutput.at(1) = arrivalTime;  //time
     } else
       correctedOutput.at(1) = -9999.f;  //time
 
     correctedOutput.at(2) = chiSq;  //chi2
   }
 }

◆ minimize()

const float MahiFit::minimize ( ) const

private

Definition at line 229 of file MahiFit.cc.

References funct::abs(), MahiNnlsWorkspace::ampVec, calculateChiSq(), deltaChiSqThresh_, mps_fire::i, MahiNnlsWorkspace::invcovp, nMaxItersMin_, nnls(), nnlsWork_, MahiNnlsWorkspace::noisecorr, MahiNnlsWorkspace::noiseTerms, MahiNnlsWorkspace::nPulseTot, onePulseMinimize(), MahiNnlsWorkspace::pedVal, MahiNnlsWorkspace::pedVals, MahiNnlsWorkspace::tsSize, and updateCov().

Referenced by doFit().

                                     {
   nnlsWork_.invcovp.setZero(nnlsWork_.tsSize, nnlsWork_.nPulseTot);
   nnlsWork_.ampVec.setZero(nnlsWork_.nPulseTot);
 
   SampleMatrix invCovMat;
   invCovMat.setConstant(nnlsWork_.tsSize, nnlsWork_.tsSize, nnlsWork_.pedVal);
   invCovMat += nnlsWork_.noiseTerms.asDiagonal();
 
   //Add off-Diagonal components up to first order
   if (nnlsWork_.noisecorr != 0) {
     for (unsigned int i = 1; i < nnlsWork_.tsSize; ++i) {
       auto const noiseCorrTerm = nnlsWork_.noisecorr * nnlsWork_.pedVals.coeff(i - 1) * nnlsWork_.pedVals.coeff(i);
       invCovMat(i - 1, i) += noiseCorrTerm;
       invCovMat(i, i - 1) += noiseCorrTerm;
     }
   }
 
   float oldChiSq = 9999;
   float chiSq = oldChiSq;
 
   for (int iter = 1; iter < nMaxItersMin_; ++iter) {
     updateCov(invCovMat);
 
     if (nnlsWork_.nPulseTot > 1) {
       nnls();
     } else {
       onePulseMinimize();
     }
 
     const float newChiSq = calculateChiSq();
     const float deltaChiSq = newChiSq - chiSq;
 
     if (newChiSq == oldChiSq && newChiSq < chiSq) {
       break;
     }
     oldChiSq = chiSq;
     chiSq = newChiSq;
 
     if (std::abs(deltaChiSq) < deltaChiSqThresh_)
       break;
   }
 
   return chiSq;
 }

◆ nnls()

void MahiFit::nnls ( ) const

private

Definition at line 439 of file MahiFit.cc.

References MahiNnlsWorkspace::amplitudes, MahiNnlsWorkspace::ampVec, MahiNnlsWorkspace::aTaMat, MahiNnlsWorkspace::aTbVec, MahiNnlsWorkspace::covDecomp, MahiNnlsWorkspace::invcovp, WZElectronSkims53X_cff::max, SiStripPI::min, nMaxItersNNLS_, nnlsConstrainParameter(), nnlsThresh_, nnlsUnconstrainParameter(), nnlsWork_, MahiNnlsWorkspace::nP, MahiNnlsWorkspace::nPulseTot, MahiNnlsWorkspace::pulseMat, solveSubmatrix(), DiMuonV_cfg::threshold, MahiNnlsWorkspace::tsSize, and cms::cuda::wmax.

Referenced by minimize().

                          {
   const unsigned int npulse = nnlsWork_.nPulseTot;
   const unsigned int nsamples = nnlsWork_.tsSize;
 
   PulseVector updateWork;
   PulseVector ampvecpermtest;
 
   nnlsWork_.invcovp = nnlsWork_.covDecomp.matrixL().solve(nnlsWork_.pulseMat);
   nnlsWork_.aTaMat.noalias() = nnlsWork_.invcovp.transpose().lazyProduct(nnlsWork_.invcovp);
   nnlsWork_.aTbVec.noalias() =
       nnlsWork_.invcovp.transpose().lazyProduct(nnlsWork_.covDecomp.matrixL().solve(nnlsWork_.amplitudes));
 
   int iter = 0;
   Index idxwmax = 0;
   float wmax = 0.0f;
   float threshold = nnlsThresh_;
 
   while (true) {
     if (iter > 0 || nnlsWork_.nP == 0) {
       if (nnlsWork_.nP == std::min(npulse, nsamples))
         break;
 
       const unsigned int nActive = npulse - nnlsWork_.nP;
       // exit if there are no more pulses to constrain
       if (nActive == 0)
         break;
 
       updateWork.noalias() = nnlsWork_.aTbVec - nnlsWork_.aTaMat.lazyProduct(nnlsWork_.ampVec);
 
       Index idxwmaxprev = idxwmax;
       float wmaxprev = wmax;
       wmax = updateWork.tail(nActive).maxCoeff(&idxwmax);
 
       if (wmax < threshold || (idxwmax == idxwmaxprev && wmax == wmaxprev)) {
         break;
       }
 
       if (iter >= nMaxItersNNLS_) {
         break;
       }
 
       //unconstrain parameter
       idxwmax += nnlsWork_.nP;
       nnlsUnconstrainParameter(idxwmax);
     }
 
     while (true) {
       if (nnlsWork_.nP == 0)
         break;
 
       ampvecpermtest.noalias() = nnlsWork_.ampVec.head(nnlsWork_.nP);
 
       solveSubmatrix(nnlsWork_.aTaMat, nnlsWork_.aTbVec, ampvecpermtest, nnlsWork_.nP);
 
       //check solution
       if (ampvecpermtest.head(nnlsWork_.nP).minCoeff() > 0.f) {
         nnlsWork_.ampVec.head(nnlsWork_.nP) = ampvecpermtest.head(nnlsWork_.nP);
         break;
       }
 
       //update parameter vector
       Index minratioidx = 0;
 
       // no realizable optimization here (because it autovectorizes!)
       float minratio = std::numeric_limits<float>::max();
       for (unsigned int ipulse = 0; ipulse < nnlsWork_.nP; ++ipulse) {
         if (ampvecpermtest.coeff(ipulse) <= 0.f) {
           const float c_ampvec = nnlsWork_.ampVec.coeff(ipulse);
           const float ratio = c_ampvec / (c_ampvec - ampvecpermtest.coeff(ipulse));
           if (ratio < minratio) {
             minratio = ratio;
             minratioidx = ipulse;
           }
         }
       }
       nnlsWork_.ampVec.head(nnlsWork_.nP) +=
           minratio * (ampvecpermtest.head(nnlsWork_.nP) - nnlsWork_.ampVec.head(nnlsWork_.nP));
 
       //avoid numerical problems with later ==0. check
       nnlsWork_.ampVec.coeffRef(minratioidx) = 0.f;
 
       nnlsConstrainParameter(minratioidx);
     }
 
     ++iter;
 
     //adaptive convergence threshold to avoid infinite loops but still
     //ensure best value is used
     if (iter % 10 == 0) {
       threshold *= 10.;
     }
   }
 }

◆ nnlsConstrainParameter()

void MahiFit::nnlsConstrainParameter ( Index minratioidx ) const

private

Definition at line 615 of file MahiFit.cc.

References MahiNnlsWorkspace::ampVec, MahiNnlsWorkspace::aTaMat, MahiNnlsWorkspace::aTbVec, MahiNnlsWorkspace::bxs, nnlsWork_, MahiNnlsWorkspace::nP, MahiNnlsWorkspace::pulseMat, and edm::swap().

Referenced by nnls().

                                                             {
   if (minratioidx != (nnlsWork_.nP - 1)) {
     nnlsWork_.aTaMat.col(nnlsWork_.nP - 1).swap(nnlsWork_.aTaMat.col(minratioidx));
     nnlsWork_.aTaMat.row(nnlsWork_.nP - 1).swap(nnlsWork_.aTaMat.row(minratioidx));
     nnlsWork_.pulseMat.col(nnlsWork_.nP - 1).swap(nnlsWork_.pulseMat.col(minratioidx));
     Eigen::numext::swap(nnlsWork_.aTbVec.coeffRef(nnlsWork_.nP - 1), nnlsWork_.aTbVec.coeffRef(minratioidx));
     Eigen::numext::swap(nnlsWork_.ampVec.coeffRef(nnlsWork_.nP - 1), nnlsWork_.ampVec.coeffRef(minratioidx));
     Eigen::numext::swap(nnlsWork_.bxs.coeffRef(nnlsWork_.nP - 1), nnlsWork_.bxs.coeffRef(minratioidx));
   }
   --nnlsWork_.nP;
 }

◆ nnlsUnconstrainParameter()

void MahiFit::nnlsUnconstrainParameter ( Index idxp ) const

private

Definition at line 603 of file MahiFit.cc.

References MahiNnlsWorkspace::ampVec, MahiNnlsWorkspace::aTaMat, MahiNnlsWorkspace::aTbVec, MahiNnlsWorkspace::bxs, nnlsWork_, MahiNnlsWorkspace::nP, MahiNnlsWorkspace::pulseMat, and edm::swap().

Referenced by nnls().

                                                        {
   if (idxp != nnlsWork_.nP) {
     nnlsWork_.aTaMat.col(nnlsWork_.nP).swap(nnlsWork_.aTaMat.col(idxp));
     nnlsWork_.aTaMat.row(nnlsWork_.nP).swap(nnlsWork_.aTaMat.row(idxp));
     nnlsWork_.pulseMat.col(nnlsWork_.nP).swap(nnlsWork_.pulseMat.col(idxp));
     Eigen::numext::swap(nnlsWork_.aTbVec.coeffRef(nnlsWork_.nP), nnlsWork_.aTbVec.coeffRef(idxp));
     Eigen::numext::swap(nnlsWork_.ampVec.coeffRef(nnlsWork_.nP), nnlsWork_.ampVec.coeffRef(idxp));
     Eigen::numext::swap(nnlsWork_.bxs.coeffRef(nnlsWork_.nP), nnlsWork_.bxs.coeffRef(idxp));
   }
   ++nnlsWork_.nP;
 }

◆ onePulseMinimize()

void MahiFit::onePulseMinimize ( ) const

private

Definition at line 533 of file MahiFit.cc.

References MahiNnlsWorkspace::amplitudes, MahiNnlsWorkspace::ampVec, MahiNnlsWorkspace::covDecomp, f, MahiNnlsWorkspace::invcovp, WZElectronSkims53X_cff::max, nnlsWork_, and MahiNnlsWorkspace::pulseMat.

Referenced by minimize().

                                      {
   nnlsWork_.invcovp = nnlsWork_.covDecomp.matrixL().solve(nnlsWork_.pulseMat);
 
   float aTaCoeff = (nnlsWork_.invcovp.transpose() * nnlsWork_.invcovp).coeff(0);
   float aTbCoeff =
       (nnlsWork_.invcovp.transpose() * (nnlsWork_.covDecomp.matrixL().solve(nnlsWork_.amplitudes))).coeff(0);
 
   nnlsWork_.ampVec.coeffRef(0) = std::max(0.f, aTbCoeff / aTaCoeff);
 }

◆ phase1Apply()

void MahiFit::phase1Apply	(	const HBHEChannelInfo &	channelData,
		float &	reconstructedEnergy,
		float &	soiPlusOneEnergy,
		float &	reconstructedTime,
		bool &	useTriple,
		float &	chi2
	)		const

Definition at line 50 of file MahiFit.cc.

References CustomPhysics_cfi::amplitude, MahiNnlsWorkspace::amplitudes, cms::cuda::assert(), isoTrack_cff::chi2, chiSqSwitch_, doFit(), f, HBHEChannelInfo::fcByPE(), nnlsWork_, MahiNnlsWorkspace::noisecorr, HBHEChannelInfo::noisecorr(), MahiNnlsWorkspace::noiseTerms, norm_, PresampleTask_cfi::nSamples, HBHEChannelInfo::nSamples(), MahiNnlsWorkspace::pedVal, MahiNnlsWorkspace::pedVals, resetWorkspace(), HBHEChannelInfo::soi(), ts4Thresh_, HBHEChannelInfo::tsDFcPerADC(), HBHEChannelInfo::tsGain(), MahiNnlsWorkspace::tsOffset, HBHEChannelInfo::tsPedestal(), HBHEChannelInfo::tsPedestalWidth(), HBHEChannelInfo::tsRawCharge(), MahiNnlsWorkspace::tsSize, and mitigatedMETSequence_cff::U.

Referenced by phase1Debug(), and SimpleHBHEPhase1Algo::reconstruct().

                                              {
   const unsigned nSamples = channelData.nSamples();
   const unsigned soi = channelData.soi();
 
   // Check some of the assumptions used by the subsequent code
   assert(nSamples == 8 || nSamples == 10);
   assert(nnlsWork_.tsSize <= nSamples);
   assert(soi + 1U < nnlsWork_.tsSize);
 
   resetWorkspace();
 
   nnlsWork_.tsOffset = soi;
 
   // Packing energy, time, chiSq, soiPlus1Energy, in that order
   std::array<float, 4> reconstructedVals{{0.f, -9999.f, -9999.f, 0.f}};
 
   double tsTOT = 0, tstrig = 0;  // in GeV
   for (unsigned int iTS = 0; iTS < nnlsWork_.tsSize; ++iTS) {
     auto const amplitude = channelData.tsRawCharge(iTS) - channelData.tsPedestal(iTS);
 
     nnlsWork_.amplitudes.coeffRef(iTS) = amplitude;
 
     //ADC granularity
     auto const noiseADC = norm_ * channelData.tsDFcPerADC(iTS);
 
     //Electronic pedestal
     auto const pedWidth = channelData.tsPedestalWidth(iTS);
 
     //Photostatistics
     auto const noisePhotoSq = (amplitude > pedWidth) ? (amplitude * channelData.fcByPE()) : 0.f;
 
     //Total uncertainty from all sources
     nnlsWork_.noiseTerms.coeffRef(iTS) = noiseADC * noiseADC + noisePhotoSq + pedWidth * pedWidth;
 
     nnlsWork_.pedVals.coeffRef(iTS) = pedWidth;
 
     tsTOT += amplitude;
     if (iTS == soi)
       tstrig += amplitude;
   }
 
   // This preserves the original Mahi approach but will have to
   // change in case we eventually calibrate gains per capId
   const float gain0 = channelData.tsGain(0);
   tsTOT *= gain0;
   tstrig *= gain0;
 
   useTriple = false;
   if (tstrig > ts4Thresh_ && tsTOT > 0) {
     nnlsWork_.noisecorr = channelData.noisecorr();
 
     if (nnlsWork_.noisecorr != 0) {
       nnlsWork_.pedVal = 0.f;
     } else {
       //Average pedestal width (for covariance matrix constraint)
       nnlsWork_.pedVal = 0.25f * (channelData.tsPedestalWidth(0) * channelData.tsPedestalWidth(0) +
                                   channelData.tsPedestalWidth(1) * channelData.tsPedestalWidth(1) +
                                   channelData.tsPedestalWidth(2) * channelData.tsPedestalWidth(2) +
                                   channelData.tsPedestalWidth(3) * channelData.tsPedestalWidth(3));
     }
 
     // only do pre-fit with 1 pulse if chiSq threshold is positive
     if (chiSqSwitch_ > 0) {
       doFit(reconstructedVals, 1);
       if (reconstructedVals[2] > chiSqSwitch_) {
         doFit(reconstructedVals, 0);  //nbx=0 means use configured BXs
         useTriple = true;
       }
     } else {
       doFit(reconstructedVals, 0);
       useTriple = true;
     }
   }
 
   reconstructedEnergy = reconstructedVals[0] * gain0;
   soiPlusOneEnergy = reconstructedVals[3] * gain0;
   reconstructedTime = reconstructedVals[1];
   chi2 = reconstructedVals[2];
 }

◆ phase1Debug()

void MahiFit::phase1Debug	(	const HBHEChannelInfo &	channelData,
		MahiDebugInfo &	mdi
	)		const

Definition at line 627 of file MahiFit.cc.

                                                                                       {
   float recoEnergy, soiPlus1Energy, recoTime, chi2;
   bool use3;
   phase1Apply(channelData, recoEnergy, soiPlus1Energy, recoTime, use3, chi2);
 
   mdi.nSamples = channelData.nSamples();
   mdi.soi = channelData.soi();
 
   mdi.use3 = use3;
 
   mdi.inTimeConst = nnlsWork_.dt;
   mdi.inPedAvg = 0.25 * (channelData.tsPedestalWidth(0) * channelData.tsPedestalWidth(0) +
                          channelData.tsPedestalWidth(1) * channelData.tsPedestalWidth(1) +
                          channelData.tsPedestalWidth(2) * channelData.tsPedestalWidth(2) +
                          channelData.tsPedestalWidth(3) * channelData.tsPedestalWidth(3));
   mdi.inGain = channelData.tsGain(0);
 
   for (unsigned int iTS = 0; iTS < channelData.nSamples(); ++iTS) {
     double charge = channelData.tsRawCharge(iTS);
     double ped = channelData.tsPedestal(iTS);
 
     mdi.inNoiseADC[iTS] = norm_ * channelData.tsDFcPerADC(iTS);
 
     if ((charge - ped) > channelData.tsPedestalWidth(iTS)) {
       mdi.inNoisePhoto[iTS] = sqrt((charge - ped) * channelData.fcByPE());
     } else {
       mdi.inNoisePhoto[iTS] = 0;
     }
 
     mdi.inPedestal[iTS] = channelData.tsPedestalWidth(iTS);
     mdi.totalUCNoise[iTS] = nnlsWork_.noiseTerms.coeffRef(iTS);
 
     if (channelData.hasTimeInfo()) {
       mdi.inputTDC[iTS] = channelData.tsRiseTime(iTS);
     } else {
       mdi.inputTDC[iTS] = -1;
     }
   }
 
   mdi.arrivalTime = recoTime;
   mdi.chiSq = chi2;
 
   for (unsigned int iBX = 0; iBX < nnlsWork_.nPulseTot; ++iBX) {
     if (nnlsWork_.bxs.coeff(iBX) == 0) {
       mdi.mahiEnergy = nnlsWork_.ampVec.coeff(iBX);
       for (unsigned int iTS = 0; iTS < nnlsWork_.tsSize; ++iTS) {
         mdi.count[iTS] = iTS;
         mdi.inputTS[iTS] = nnlsWork_.amplitudes.coeff(iTS);
         mdi.itPulse[iTS] = nnlsWork_.pulseMat.col(iBX).coeff(iTS);
       }
     } else if (nnlsWork_.bxs.coeff(iBX) == pedestalBX_) {
       mdi.pedEnergy = nnlsWork_.ampVec.coeff(iBX);
     } else if (nnlsWork_.bxs.coeff(iBX) >= -3 && nnlsWork_.bxs.coeff(iBX) <= 4) {
       int ootIndex = nnlsWork_.bxs.coeff(iBX);
       if (ootIndex > 0)
         ootIndex += 2;
       else
         ootIndex += 3;
       mdi.ootEnergy[ootIndex] = nnlsWork_.ampVec.coeff(iBX);
       for (unsigned int iTS = 0; iTS < nnlsWork_.tsSize; ++iTS) {
         mdi.ootPulse[ootIndex][iTS] = nnlsWork_.pulseMat.col(iBX).coeff(iTS);
       }
     }
   }
 }

◆ resetPulseShapeTemplate()

void MahiFit::resetPulseShapeTemplate	(	int	pulseShapeId,
		const HcalPulseShapes &	ps,
		unsigned int	nSamples
	)

private

Definition at line 578 of file MahiFit.cc.

References cntsetPulseShape_, currentPulseShapeId_, HcalPulseShapes::getShape(), knownPulseShapes_, hcal::constants::maxSamples, AlCaHLTBitMon_ParallelJobs::p, and psfPtr_.

Referenced by setPulseShapeTemplate().

                                                             {
   ++cntsetPulseShape_;
 
   psfPtr_ = nullptr;
   for (auto& elem : knownPulseShapes_) {
     if (elem.first == pulseShapeId) {
       psfPtr_ = &*elem.second;
       break;
     }
   }
 
   if (!psfPtr_) {
     // only the pulse shape itself from PulseShapeFunctor is used for Mahi
     // the uncertainty terms calculated inside PulseShapeFunctor are used for Method 2 only
     auto p = std::make_shared<FitterFuncs::PulseShapeFunctor>(
         ps.getShape(pulseShapeId), false, false, false, 1, 0, 0, hcal::constants::maxSamples);
     knownPulseShapes_.emplace_back(pulseShapeId, p);
     psfPtr_ = &*p;
   }
 
   currentPulseShapeId_ = pulseShapeId;
 }

◆ resetWorkspace()

void MahiFit::resetWorkspace ( ) const

private

Definition at line 742 of file MahiFit.cc.

References MahiNnlsWorkspace::amplitudes, MahiNnlsWorkspace::bxOffset, MahiNnlsWorkspace::maxoffset, nnlsWork_, MahiNnlsWorkspace::noiseTerms, MahiNnlsWorkspace::nP, MahiNnlsWorkspace::nPulseTot, MahiNnlsWorkspace::pedVals, and MahiNnlsWorkspace::tsOffset.

Referenced by phase1Apply().

                                    {
   nnlsWork_.nPulseTot = 0;
   nnlsWork_.tsOffset = 0;
   nnlsWork_.bxOffset = 0;
   nnlsWork_.maxoffset = 0;
   nnlsWork_.nP = 0;
 
   // NOT SURE THIS IS NEEDED
   nnlsWork_.amplitudes.setZero();
   nnlsWork_.noiseTerms.setZero();
   nnlsWork_.pedVals.setZero();
 }

◆ setParameters()

void MahiFit::setParameters	(	bool	iDynamicPed,
		double	iTS4Thresh,
		double	chiSqSwitch,
		bool	iApplyTimeSlew,
		HcalTimeSlew::BiasSetting	slewFlavor,
		bool	iCalculateArrivalTime,
		int	iTimeAlgo,
		double	iThEnergeticPulses,
		double	iMeanTime,
		double	iTimeSigmaHPD,
		double	iTimeSigmaSiPM,
		const std::vector< int > &	iActiveBXs,
		int	iNMaxItersMin,
		int	iNMaxItersNNLS,
		double	iDeltaChiSqThresh,
		double	iNnlsThresh
	)

Definition at line 7 of file MahiFit.cc.

References activeBXs_, applyTimeSlew_, bxOffsetConf_, bxSizeConf_, calculateArrivalTime_, hltHbhereco_cfi::chiSqSwitch, chiSqSwitch_, deltaChiSqThresh_, dynamicPed_, meanTime_, nMaxItersMin_, nMaxItersNNLS_, nnlsThresh_, slewFlavor_, thEnergeticPulses_, HBHEMahiParameters_cfi::timeAlgo, timeAlgo_, timeSigmaHPD_, timeSigmaSiPM_, and ts4Thresh_.

                                                 {
   dynamicPed_ = iDynamicPed;
 
   ts4Thresh_ = iTS4Thresh;
   chiSqSwitch_ = chiSqSwitch;
 
   applyTimeSlew_ = iApplyTimeSlew;
   slewFlavor_ = slewFlavor;
 
   calculateArrivalTime_ = iCalculateArrivalTime;
   timeAlgo_ = timeAlgo;
   thEnergeticPulses_ = iThEnergeticPulses;
   meanTime_ = iMeanTime;
   timeSigmaHPD_ = iTimeSigmaHPD;
   timeSigmaSiPM_ = iTimeSigmaSiPM;
 
   activeBXs_ = iActiveBXs;
 
   nMaxItersMin_ = iNMaxItersMin;
   nMaxItersNNLS_ = iNMaxItersNNLS;
 
   deltaChiSqThresh_ = iDeltaChiSqThresh;
   nnlsThresh_ = iNnlsThresh;
 
   bxOffsetConf_ = -(*std::min_element(activeBXs_.begin(), activeBXs_.end()));
   bxSizeConf_ = activeBXs_.size();
 }

◆ setPulseShapeTemplate()

void MahiFit::setPulseShapeTemplate	(	int	pulseShapeId,
		const HcalPulseShapes &	ps,
		bool	hasTimeInfo,
		const HcalTimeSlew *	hcalTimeSlewDelay,
		unsigned int	nSamples,
		const float	gain
	)

Definition at line 548 of file MahiFit.cc.

References MahiNnlsWorkspace::amplitudes, cms::cuda::assert(), currentPulseShapeId_, HcalTimeSlew::delay(), MahiNnlsWorkspace::dt, hcalTimeSlewDelay_, nnlsWork_, MahiNnlsWorkspace::noiseTerms, PresampleTask_cfi::nSamples, MahiNnlsWorkspace::pedVals, resetPulseShapeTemplate(), slewFlavor_, thEnergeticPulses_, thEnergeticPulsesFC_, timeSigmaHPD_, timeSigmaSiPM_, tsDelay1GeV_, and MahiNnlsWorkspace::tsSize.

                                                        {
   if (hcalTimeSlewDelay != hcalTimeSlewDelay_) {
     assert(hcalTimeSlewDelay);
     hcalTimeSlewDelay_ = hcalTimeSlewDelay;
     tsDelay1GeV_ = hcalTimeSlewDelay->delay(1.0, slewFlavor_);
   }
 
   if (pulseShapeId != currentPulseShapeId_) {
     resetPulseShapeTemplate(pulseShapeId, ps, nSamples);
   }
 
   // 1 sigma time constraint
   nnlsWork_.dt = hasTimeInfo ? timeSigmaSiPM_ : timeSigmaHPD_;
 
   if (nnlsWork_.tsSize != nSamples) {
     nnlsWork_.tsSize = nSamples;
     nnlsWork_.amplitudes.resize(nSamples);
     nnlsWork_.noiseTerms.resize(nSamples);
     nnlsWork_.pedVals.resize(nSamples);
   }
 
   // threshold in GeV for ccTime
   thEnergeticPulsesFC_ = thEnergeticPulses_ / gain0;
 }

◆ solveSubmatrix()

void MahiFit::solveSubmatrix	(	PulseMatrix &	mat,
		PulseVector &	invec,
		PulseVector &	outvec,
		unsigned	nP
	)		const

private

Definition at line 693 of file MahiFit.cc.

References Exception, and groupFilesInBlocks::temp.

Referenced by nnls().

                                                                                                          {
   using namespace Eigen;
   switch (nP) {  // pulse matrix is always square.
     case 10: {
       Matrix<float, 10, 10> temp = mat;
       outvec.head<10>() = temp.ldlt().solve(invec.head<10>());
     } break;
     case 9: {
       Matrix<float, 9, 9> temp = mat.topLeftCorner<9, 9>();
       outvec.head<9>() = temp.ldlt().solve(invec.head<9>());
     } break;
     case 8: {
       Matrix<float, 8, 8> temp = mat.topLeftCorner<8, 8>();
       outvec.head<8>() = temp.ldlt().solve(invec.head<8>());
     } break;
     case 7: {
       Matrix<float, 7, 7> temp = mat.topLeftCorner<7, 7>();
       outvec.head<7>() = temp.ldlt().solve(invec.head<7>());
     } break;
     case 6: {
       Matrix<float, 6, 6> temp = mat.topLeftCorner<6, 6>();
       outvec.head<6>() = temp.ldlt().solve(invec.head<6>());
     } break;
     case 5: {
       Matrix<float, 5, 5> temp = mat.topLeftCorner<5, 5>();
       outvec.head<5>() = temp.ldlt().solve(invec.head<5>());
     } break;
     case 4: {
       Matrix<float, 4, 4> temp = mat.topLeftCorner<4, 4>();
       outvec.head<4>() = temp.ldlt().solve(invec.head<4>());
     } break;
     case 3: {
       Matrix<float, 3, 3> temp = mat.topLeftCorner<3, 3>();
       outvec.head<3>() = temp.ldlt().solve(invec.head<3>());
     } break;
     case 2: {
       Matrix<float, 2, 2> temp = mat.topLeftCorner<2, 2>();
       outvec.head<2>() = temp.ldlt().solve(invec.head<2>());
     } break;
     case 1: {
       Matrix<float, 1, 1> temp = mat.topLeftCorner<1, 1>();
       outvec.head<1>() = temp.ldlt().solve(invec.head<1>());
     } break;
     default:
       throw cms::Exception("HcalMahiWeirdState")
           << "Weird number of pulses encountered in Mahi, module is configured incorrectly!";
   }
 }

◆ updateCov()

void MahiFit::updateCov ( const SampleMatrix & invCovMat ) const

private

Definition at line 334 of file MahiFit.cc.

References MahiNnlsWorkspace::ampVec, MahiNnlsWorkspace::bxOffset, MahiNnlsWorkspace::bxs, MahiNnlsWorkspace::covDecomp, nnlsWork_, MahiNnlsWorkspace::nPulseTot, HLT_IsoTrack_cff::offset, pedestalBX_, and MahiNnlsWorkspace::pulseCovArray.

Referenced by minimize().

                                                            {
   SampleMatrix invCovMat = samplecov;
 
   for (unsigned int iBX = 0; iBX < nnlsWork_.nPulseTot; ++iBX) {
     auto const amp = nnlsWork_.ampVec.coeff(iBX);
     if (amp == 0)
       continue;
 
     int offset = nnlsWork_.bxs.coeff(iBX);
 
     if (offset == pedestalBX_)
       continue;
     else {
       invCovMat.noalias() += amp * amp * nnlsWork_.pulseCovArray[offset + nnlsWork_.bxOffset];
     }
   }
 
   nnlsWork_.covDecomp.compute(invCovMat);
 }

◆ updatePulseShape()

void MahiFit::updatePulseShape	(	const float	itQ,
		FullSampleVector &	pulseShape,
		FullSampleVector &	pulseDeriv,
		FullSampleMatrix &	pulseCov
	)		const

private

Definition at line 274 of file MahiFit.cc.

References applyTimeSlew_, calculateArrivalTime_, HcalTimeSlew::delay(), dumpMFGeometry_cfg::delta, MahiNnlsWorkspace::dt, f, FitterFuncs::PulseShapeFunctor::getPulseShape(), hcalTimeSlewDelay_, MahiNnlsWorkspace::maxoffset, meanTime_, nnlsWork_, psfPtr_, FitterFuncs::PulseShapeFunctor::singlePulseShapeFuncMahi(), slewFlavor_, FrontierCondition_GT_autoExpress_cfi::t0, createJobs::tmp, tsDelay1GeV_, MahiNnlsWorkspace::tsOffset, MahiNnlsWorkspace::tsSize, and geometryCSVtoXML::xx.

Referenced by doFit().

                                                                  {
   // set a null pulse shape for negative / or null TS
   if (itQ <= 0.f)
     return;
 
   float t0 = meanTime_;
 
   if (applyTimeSlew_) {
     if (itQ <= 1.f)
       t0 += tsDelay1GeV_;
     else
       t0 += hcalTimeSlewDelay_->delay(float(itQ), slewFlavor_);
   }
 
   std::array<double, hcal::constants::maxSamples> pulseN;
   std::array<double, hcal::constants::maxSamples> pulseM;
   std::array<double, hcal::constants::maxSamples> pulseP;
 
   const float xx = t0;
   const float xxm = -nnlsWork_.dt + t0;
   const float xxp = nnlsWork_.dt + t0;
 
   psfPtr_->singlePulseShapeFuncMahi(&xx);
   psfPtr_->getPulseShape(pulseN);
 
   psfPtr_->singlePulseShapeFuncMahi(&xxm);
   psfPtr_->getPulseShape(pulseM);
 
   psfPtr_->singlePulseShapeFuncMahi(&xxp);
   psfPtr_->getPulseShape(pulseP);
 
   //in the 2018+ case where the sample of interest (SOI) is in TS3, add an extra offset to align
   //with previous SOI=TS4 case assumed by psfPtr_->getPulseShape()
   int delta = 4 - nnlsWork_.tsOffset;
 
   auto invDt = 0.5 / nnlsWork_.dt;
 
   for (unsigned int iTS = 0; iTS < nnlsWork_.tsSize; ++iTS) {
     pulseShape(iTS + nnlsWork_.maxoffset) = pulseN[iTS + delta];
     if (calculateArrivalTime_)
       pulseDeriv(iTS + nnlsWork_.maxoffset) = (pulseM[iTS + delta] - pulseP[iTS + delta]) * invDt;
 
     pulseM[iTS + delta] -= pulseN[iTS + delta];
     pulseP[iTS + delta] -= pulseN[iTS + delta];
   }
 
   for (unsigned int iTS = 0; iTS < nnlsWork_.tsSize; ++iTS) {
     for (unsigned int jTS = 0; jTS < iTS + 1; ++jTS) {
       auto const tmp = 0.5 * (pulseP[iTS + delta] * pulseP[jTS + delta] + pulseM[iTS + delta] * pulseM[jTS + delta]);
 
       pulseCov(jTS + nnlsWork_.maxoffset, iTS + nnlsWork_.maxoffset) = tmp;
       if (iTS != jTS)
         pulseCov(iTS + nnlsWork_.maxoffset, jTS + nnlsWork_.maxoffset) = tmp;
     }
   }
 }