FitChannelInfo.h

Go to the documentation of this file.

/*
 * FitChannelInfo.h
 *
 *  Created on: 20 Apr 2020
 *      Author: Valerio Formato
 */

#ifndef HERD_FitChannelInfo_H_
#define HERD_FitChannelInfo_H_

#ifdef HS_USE_ROOT
#include "Rtypes.h"
#endif

#include <bitset>
#include <vector>

namespace Herd {
class FitChannelInfo {
public:
  virtual ~FitChannelInfo() = default;

  void SetNChannels(size_t nChannels) {
    m_nChannels = nChannels;
    m_pedestals.resize(m_nChannels);
    m_noises.resize(m_nChannels);
    m_flags.resize(m_nChannels);
  };

  size_t GetNChannels() { return m_nChannels; }

  float GetChannelPedestal(size_t channelID) const { return m_pedestals[channelID]; };

  float GetChannelNoise(size_t channelID) const { return m_noises[channelID]; };

  std::bitset<16> GetChannelFlags(size_t channelID) const { return m_flags[channelID]; };

  void SetChannelPedestal(size_t channelID, float value) { m_pedestals[channelID] = value; };

  void SetChannelNoise(size_t channelID, float value) { m_noises[channelID] = value; };

  void SetChannelFlags(size_t channelID, std::bitset<16> value) { m_flags[channelID] = value; };

private:
  size_t m_nChannels = 0;
  std::vector<float> m_pedestals;
  std::vector<float> m_noises;
  std::vector<std::bitset<16>> m_flags;

#ifdef HS_USE_ROOT
  ClassDef(FitChannelInfo, 1)
#endif
};
} // namespace Herd

#endif