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(), m_layerToGroup() {};

    CDCDedx1DCell(short version,
                  const std::vector<std::vector<double>>& groupGains,
                  const std::vector<unsigned int>& layerToGroup);

    ~CDCDedx1DCell() {};

    CDCDedx1DCell& operator*=(CDCDedx1DCell const& rhs);

    short getVersion() const { return m_version; };

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

    unsigned int getNLayers() const
    {
      if (!m_layerToGroup.empty()) return m_layerToGroup.size();
 
      if (m_onedgains.size() == 2) return 56;
 
      return 0;
    };

    unsigned int getGroup(unsigned int layer) const
    {
      if (layer >= getNLayers()) {
        B2ERROR("Layer out of range");
        return std::numeric_limits<unsigned int>::max();
      }
 
      if (!m_layerToGroup.empty()) {
        return m_layerToGroup[layer];
      }
 
      if (m_onedgains.size() == 2) {
        return (layer < 8) ? 0 : 1;
      }
 
      B2ERROR("Invalid 1DCell payload state");
      return std::numeric_limits<unsigned int>::max();
    };

    unsigned int getNBins(unsigned int layer) const
    {
      const unsigned int group = getGroup(layer);
 
      if (group >= m_onedgains.size()) return 0;
 
      return m_onedgains[group].size();
    };

    const std::vector<std::vector<double>>& getOneDCell() const
    {
      return m_onedgains;
    }

 
    const std::vector<unsigned int>& getLayerMap() const
    {
      return m_layerToGroup;
    }

    double getMean(unsigned int layer, unsigned int bin) const;
 

    void setMean(unsigned int layer, unsigned int bin, double value);
 

    double getMean(unsigned int layer, double enta) const;
 
 
  private:

    bool isValidGroupedPayload() const;

    short m_version; 
    std::vector<std::vector<double>> m_onedgains; 
    std::vector<unsigned int> m_layerToGroup;  
 
    ClassDef(CDCDedx1DCell, 6); 
  };

} // end namespace Belle2