CDCTriggerNeuroConfig.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2018 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Felix Meggendorfer                                       *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
#pragma once
 
#include <TObject.h>
#include <TFile.h>
#include <trg/cdc/dataobjects/CDCTriggerMLP.h>
#include <string>
 
namespace Belle2 {
  class CDCTriggerNeuroConfig: public TObject {
  public:
    CDCTriggerNeuroConfig() {}
 
    virtual ~CDCTriggerNeuroConfig() {};
 
    struct B2FormatLine {
      B2FormatLine() : start(0), end(0), offset(0), name("None") {}
      int start;
      int end;
      int offset;
      std::string name;
      std::string description;
    };
    void setB2Format(const std::vector<B2FormatLine>& format)
    {
      for (auto line : format) {
        addB2FormatLine(line);
      }
    }
    B2FormatLine getB2FormatLine(const std::string& name) const
    {
      B2FormatLine ret;
      for (auto line : m_B2Format) {
        if (name == line.name) {
          ret = line;
          continue;
        }
      }
      return ret;
    }
 
    void addB2FormatLine(const B2FormatLine& line)
    {
      m_B2Format.push_back(line);
    }
    void addB2FormatLine(int start, int end, int offset, const std::string& name, const std::string& description)
    {
      B2FormatLine b;
      b.start = start;
      b.end = end;
      b.offset = offset;
      b.name = name;
      b.description = description;
      addB2FormatLine(b);
    }
    void setNNName(const std::string& filename)
    {
      if (filename.size() <= 255) {
        m_NNName = filename;
      }
    }
    void loadMLPs(const std::string& filename, const std::string& arrayname)
    {
      TFile datafile(filename.c_str(), "READ");
      TObjArray* MLPs = (TObjArray*)datafile.Get(arrayname.c_str());
      if (!MLPs) {
        datafile.Close();
      }
      m_MLPs.clear();
      for (int isector = 0; isector < MLPs->GetEntriesFast(); ++isector) {
        CDCTriggerMLP* expert = dynamic_cast<CDCTriggerMLP*>(MLPs->At(isector));
        if (expert) m_MLPs.push_back(*expert);
      }
      MLPs->Clear();
      delete MLPs;
      datafile.Close();
    }
    void setNNNotes(const std::string& notes)
    {
      if (notes.size() <= 255) {
        m_NNNotes = notes;
      }
    }
    void setUseETF(bool b)
    {
      m_useETF = b;
    }
    void setPPNotes(const std::string& notes)
    {
      if (notes.size() <= 255) {
        m_PPNotes = notes;
      }
    }
    void setNNTFirmwareVersionID(const std::string& version)
    {
      if (version.size() <= 255) {
        m_NNTFirmwareVersionID = version;
      }
    }
    void setNNTFirmwareComment(const std::string& notes)
    {
      if (notes.size() <= 255) {
        m_NNTFirmwareComment = notes;
      }
    }
    std::vector<B2FormatLine> getB2Format()  const  {return m_B2Format;}
    std::string getNNName()                  const  {return m_NNName;}
    std::vector<CDCTriggerMLP> getMLPs()     const  {return m_MLPs;}
    std::string getNNNotes()                 const  {return m_NNNotes;}
    bool getUseETF()                         const  {return m_useETF;}
    std::string getPPNotes()                 const  {return m_PPNotes;}
    std::string getNNTFirmwareVersionID()    const  {return m_NNTFirmwareVersionID;}
    std::string getNNTFirmwareComment()      const  {return m_NNTFirmwareComment;}
 
  private:
    std::vector<B2FormatLine> m_B2Format = {};
 
    std::string m_NNName;
 
    std::vector<CDCTriggerMLP> m_MLPs;
 
    std::string m_NNNotes;
 
    bool m_useETF = false;
 
    std::string m_PPNotes;
 
    std::string m_NNTFirmwareVersionID;
 
    std::string m_NNTFirmwareComment;
 
    ClassDef(CDCTriggerNeuroConfig, 3); 
  };
} // end of namespace Belle2