CDCDedxCosineCor.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
 
#pragma once
 
#include <framework/logging/Logger.h>
 
#include <TObject.h>
#include <cmath>
 
namespace Belle2 {
  class CDCDedxCosineCor: public TObject {
 
  public:
 
    CDCDedxCosineCor(): m_cosgains() {};
 
    explicit CDCDedxCosineCor(const std::vector<double>& cosgains): m_cosgains(cosgains) {};
 
    ~CDCDedxCosineCor() {};
 
    CDCDedxCosineCor& operator*=(CDCDedxCosineCor const& rhs)
    {
      if (m_cosgains.size() % rhs.getSize() != 0) {
        B2WARNING("Cosine gain parameters do not match, cannot merge!");
        return *this;
      }
      std::vector<double> rhsgains = rhs.getCosCor();
      int scale = std::floor(m_cosgains.size() / rhs.getSize() + 0.001);
      for (unsigned int bin = 0; bin < m_cosgains.size(); ++bin) {
        m_cosgains[bin] *= rhsgains[std::floor(bin / scale + 0.001)];
      }
      return *this;
    }
 
    unsigned int getSize() const { return m_cosgains.size(); };
 
    const std::vector<double>& getCosCor() const {return m_cosgains; };
 
    void setCosCor(unsigned int bin, double value)
    {
      if (bin < m_cosgains.size()) m_cosgains[bin] = value;
      else B2ERROR("CDCDedxCosineCor: invalid bin number, value not set");
    }
 
    double getMean(unsigned int bin) const
    {
      if (bin < m_cosgains.size()) return m_cosgains[bin];
      return 1.0;
    }
 
    double getMean(double costh) const
    {
      if (std::abs(costh) > 1) return 0;
 
      // gains are stored at the center of the bins
      // find the bin center immediately preceding this value of costh
      double binsize = 2.0 / m_cosgains.size();
      int bin = std::floor((costh - 0.5 * binsize + 1.0) / binsize);
 
      // extrapolation
      // extrapolate backward for lowest half-bin and center positive half-bin
      // extrapolate forward for highest half-bin and center negative half-bin
      int thisbin = bin, nextbin = bin + 1;
      if ((costh + 1) < (binsize / 2) || (costh > 0 && std::fabs(costh) < (binsize / 2))) {
        thisbin = bin + 1; nextbin = bin + 2;
      } else {
        if ((costh - 1) > -1.0 * (binsize / 2) || (costh < 0 && std::fabs(costh) < (binsize / 2))) {
          thisbin = bin - 1; nextbin = bin;
        }
      }
      double frac = ((costh - 0.5 * binsize + 1.0) / binsize) - thisbin;
 
      if (thisbin < 0 || (unsigned)nextbin >= m_cosgains.size()) {
        B2WARNING("Problem with extrapolation of CDC dE/dx cosine correction");
        return 1.0;
      }
      return ((m_cosgains[nextbin] - m_cosgains[thisbin]) * frac + m_cosgains[thisbin]);
    };
 
  private:
    std::vector<double> m_cosgains; 
    ClassDef(CDCDedxCosineCor, 8); 
  };
} // end namespace Belle2