CDCDedxWireGain.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>
 
namespace Belle2 {
  class CDCDedxWireGain: public TObject {
 
  public:
 
    CDCDedxWireGain(): m_wiregains() {};
 
    explicit CDCDedxWireGain(const std::vector<double>& wiregains): m_wiregains(wiregains) {};
 
    ~CDCDedxWireGain() {};
 
    CDCDedxWireGain& operator*=(CDCDedxWireGain const& rhs)
    {
      for (unsigned int bin = 0; bin < m_wiregains.size(); ++bin) {
        m_wiregains[bin] *= rhs.getWireGain(bin);
      }
      return *this;
    }
 
    void setWireGain(int wire, double gain)
    {
      if (wire < 0 || (unsigned)wire >= m_wiregains.size())
        B2WARNING("Asking for a CDC Wire that is not found!");
      else m_wiregains[wire] = gain;
    }
 
    float getWireGain(int wire) const
    {
      if (wire < 0 || (unsigned)wire >= m_wiregains.size()) {
        B2WARNING("Asking for a CDC Wire that is not found!");
        return 1.0;
      }
      double gain = m_wiregains[wire];
      return gain;
    };
 
  private:
    std::vector<double> m_wiregains; 
    ClassDef(CDCDedxWireGain, 5); 
  };
} // end namespace Belle2