CDCTriggerNeuroDataModule.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/core/Module.h>
#include <trg/cdc/NeuroTrigger.h>
#include <trg/cdc/NeuroTriggerParameters.h>
#include <trg/cdc/dataobjects/CDCTriggerMLPData.h>
#include <iostream>
#include "boost/iostreams/filter/gzip.hpp"
#include "boost/iostreams/filtering_streambuf.hpp"
#include "boost/iostreams/filtering_stream.hpp"
#include "boost/multi_array.hpp"
#define BOOST_MULTI_ARRAY_NO_GENERATORS
namespace Belle2 {
  class CDCTriggerNeuroDataModule : public Module {
  public:
    CDCTriggerNeuroDataModule();
 
    virtual ~CDCTriggerNeuroDataModule() {}
 
    virtual void initialize() override;
 
    virtual void event() override;
 
    virtual void terminate() override;
 
    bool loadIDHist(const std::string& filename, NeuroTrigger& neuro);
 
    void saveData_gzip(const std::string& filename, const std::vector<CDCTriggerMLPData>& dataset,
                       boost::iostreams::filtering_streambuf<boost::iostreams::output> outStream);
 
  protected:
    //module parameters
    std::string m_hitCollectionName;
    std::string m_inputCollectionName;
    bool m_trainOnRecoTracks;
    std::string m_targetCollectionName;
    std::string m_EventTimeName;
    std::string m_filename;
    std::string m_configFileName;
    std::string m_writeconfigFileName;
    NeuroTriggerParameters m_neuroParameters;
    bool m_neuroTrackInputMode;
    int m_trackcounter;
    //other data
    NeuroTrigger m_NeuroTrigger;
    std::vector<CDCTriggerMLPData> m_trainSet;
 
    StoreArray<CDCTriggerTrack> m_tracks;
    bool m_singleUse;
  };
}