CDCTriggerMLPData.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.                  *
 **************************************************************************/
#ifndef CDCTRIGGERMLPDATA_H
#define CDCTRIGGERMLPDATA_H
 
#include <TObject.h>
 
namespace Belle2 {
  class CDCTriggerMLPData : public TObject {
  public:
    CDCTriggerMLPData(): inputSamples(), targetSamples(), hitCounters(), trackCounter(0) { }
    ~CDCTriggerMLPData() { }
 
    void addCounters(unsigned nWires)
    {
      std::vector<unsigned short> counters;
      counters.assign(nWires, 0);
      hitCounters.push_back(counters);
    }
    void addHit(unsigned iSL, int iTS);
    void countTrack() { ++trackCounter; }
    void addSample(const std::vector<float>& input, const std::vector<float>& target)
    {
      inputSamples.push_back(input);
      targetSamples.push_back(target);
    }
 
    short getTrackCounter() const { return trackCounter; }
    unsigned short getHitCounter(unsigned iSL, int iTS) const;
    unsigned nSamples() const { return targetSamples.size(); }
    const std::vector<float>& getInput(unsigned i) const { return inputSamples[i]; }
    const std::vector<float>& getTarget(unsigned i) const { return targetSamples[i]; }
 
  private:
    std::vector<std::vector<float>> inputSamples;
    std::vector<std::vector<float>> targetSamples;
    std::vector<std::vector<unsigned short>> hitCounters;
    short trackCounter;
 
    ClassDef(CDCTriggerMLPData, 1);
  };
}
#endif