CDCDedxMomentumCor.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2015 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Jake Bennett                                             *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
 
#include <framework/logging/Logger.h>
 
#include <TObject.h>
#include <cmath>
 
namespace Belle2 {
  class CDCDedxMomentumCor: public TObject {
 
  public:
 
    CDCDedxMomentumCor(): m_momcor() {};
 
    explicit CDCDedxMomentumCor(const std::vector<double>& momcor): m_momcor(momcor) {};
 
    ~CDCDedxMomentumCor() {};
 
    CDCDedxMomentumCor& operator*=(CDCDedxMomentumCor const& rhs)
    {
      if (m_momcor.size() != rhs.getSize()) {
        B2WARNING("Momentum correction parameters do not match, cannot merge!");
        return *this;
      }
      std::vector<double> rhsgains = rhs.getMomCor();
      for (unsigned int bin = 0; bin < m_momcor.size(); ++bin) {
        m_momcor[bin] *= rhsgains[bin];
      }
      return *this;
    }
 
    unsigned int getSize() const { return m_momcor.size(); };
 
    std::vector<double> getMomCor() const {return m_momcor; };
 
    double getMean(unsigned int bin) const
    {
      if (bin > m_momcor.size()) return 1.0;
      else return m_momcor[bin];
    }
 
    double getMean(double mom) const
    {
      if (std::abs(mom) > 10.0) return 0;
 
      // gains are stored at the center of the bins
      // find the bin center immediately preceding this value of mom
      double binsize = 10.0 / m_momcor.size();
      int bin = std::floor(mom / binsize);
 
      return m_momcor[bin];
    };
 
  private:
 
    std::vector<double> m_momcor; 
    ClassDef(CDCDedxMomentumCor, 5); 
  };
} // end namespace Belle2