CDCDedxWireGain.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>
 
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