GenericTruncatedAverageDeDxEstimator.h

Go to the documentation of this file.

#ifndef RecoTrackerDeDx_GenericTruncatedAverageDeDxEstimator_h
#define RecoTrackerDeDx_GenericTruncatedAverageDeDxEstimator_h

#include "RecoTracker/DeDx/interface/BaseDeDxEstimator.h"
#include "RecoTracker/DeDx/interface/DeDxTools.h"
#include "DataFormats/TrackReco/interface/DeDxHit.h"
#include <numeric>

class GenericTruncatedAverageDeDxEstimator: public BaseDeDxEstimator
{
public: 
 GenericTruncatedAverageDeDxEstimator(const edm::ParameterSet& iConfig){
    m_fraction = iConfig.getParameter<double>("fraction");
    m_expo = iConfig.getParameter<double>("exponent");
 }

 std::pair<float,float> dedx(const reco::DeDxHitCollection& Hits) override{
    int first = 0, last = Hits.size();
    if (m_fraction > 0) { // truncate high charge ones
       last -= int(Hits.size()*m_fraction); 
    } else {
       first += int(Hits.size()*(-m_fraction)); 
    }
    double sumdedx = 0;
    for(int i = first; i < last; i++){
       sumdedx+=pow(Hits[i].charge(),m_expo);
    } 
   double avrdedx = (last-first) ? pow(sumdedx/(last-first),1.0/m_expo) :0.0;
   return  std::make_pair(avrdedx,-1);
 } 

private:
 float m_fraction, m_expo;

};

#endif