CDCDedx1DCell.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 CDCDedx1DCell: public TObject {
 
  public:

    CDCDedx1DCell(): m_version(0), m_onedgains() {};

    CDCDedx1DCell(short version, const std::vector<std::vector<double>>& onedgains): m_version(version), m_onedgains(onedgains) {};

    ~CDCDedx1DCell() {};

    CDCDedx1DCell& operator*=(CDCDedx1DCell const& rhs)
    {
      if (m_version != rhs.getVersion()) {
        B2WARNING("1D cell gain parameters do not match, cannot merge!");
        return *this;
      }
      for (unsigned int layer = 0; layer < getSize(); ++layer) {
        double tempLayer = -1; //out of bound layer
        if (getSize() == 2)tempLayer = layer * 20; // 20 to make <8 and >=8 difference
        else if (getSize() == 56)tempLayer = layer;
        else B2ERROR("ERROR! Wrong # of constants vector array");
        for (unsigned int bin = 0; bin < getNBins(layer); ++bin) {
          m_onedgains[layer][bin] *= rhs.getMean(tempLayer, bin);
        }
      }
      return *this;
    }

    short getVersion() const { return m_version; };

    unsigned int getSize() const { return m_onedgains.size(); };

    unsigned int getNBins(unsigned int layer) const
    {
      if (m_onedgains.size() == 0) {
        B2ERROR("ERROR!");
        return 0;
      }
 
      if (m_onedgains.size() == 2) {
        if (layer == 0) return m_onedgains[0].size();
        else if (layer == 1) return m_onedgains[1].size();
        else {
          B2ERROR("ERROR! const vector not found");
          return 0;
        }
      } else if (m_onedgains.size() == 56) {
        if (layer < m_onedgains.size()) return m_onedgains[layer].size();
      } else {
        B2ERROR("ERROR! Wrong # of constants vector array: getNBins()");
        return 0;
      }
      return 0;
    };

    double getMean(unsigned int layer, unsigned int bin) const
    {
      unsigned int mylayer = 0;
      if (layer >= 8 && m_onedgains.size() == 2) mylayer = 1;
      else if (m_onedgains.size() == 56) mylayer = layer;
 
      if (mylayer < m_onedgains.size() and bin < m_onedgains[mylayer].size()) return m_onedgains[mylayer][bin];
 
      return 1.0;
    };

    void setMean(unsigned int layer, unsigned int bin, double value)
    {
      unsigned int mylayer = 0;
      if (layer >= 8 && m_onedgains.size() == 2) mylayer = 1;
      else if (m_onedgains.size() == 56) mylayer = layer;
 
      if (mylayer < m_onedgains.size() and bin < m_onedgains[mylayer].size()) m_onedgains[mylayer][bin] = value;
    };

    double getMean(unsigned int layer, double enta) const
    {
      if (layer > 56) {
        B2ERROR("No such layer!");
        return 0;
      }
 
      unsigned int mylayer = 0;
      if (layer >= 8 && m_onedgains.size() == 2) mylayer = 1;
      else if (m_onedgains.size() == 56) mylayer = layer;
 
      if (mylayer >= m_onedgains.size()) return 1.0;
 
      double piby2 =  M_PI / 2.0;
      // assume rotational symmetry
      if (enta < -piby2) enta += piby2;
      if (enta > piby2) enta -= piby2;
 
      double binsize = (m_onedgains[mylayer].size() != 0) ? 2.0 * piby2 / m_onedgains[mylayer].size() : 0.0;
      int bin = (binsize != 0.0) ? std::floor((enta + piby2) / binsize) : -1;
      if (bin < 0 || (unsigned)bin >= m_onedgains[mylayer].size()) {
        B2WARNING("Problem with CDC dE/dx 1D binning!");
        return 1.0;
      }
 
      return m_onedgains[mylayer][bin];
    };
 
  private:
    short m_version; 
    std::vector<std::vector<double>> m_onedgains; 
 
    ClassDef(CDCDedx1DCell, 5); 
  };

} // end namespace Belle2