CDCTriggerNeuroModule.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 CDCTRIGGERNEUROMODULE_H
#define CDCTRIGGERNEUROMODULE_H
 
#include <framework/core/Module.h>
#include <trg/cdc/NeuroTrigger.h>
#include <trg/cdc/NeuroTriggerParameters.h>
#include <framework/datastore/StoreArray.h>
#include <framework/database/DBObjPtr.h>
#include <trg/cdc/dbobjects/CDCTriggerNeuroConfig.h>
#include <trg/cdc/dataobjects/CDCTriggerTrack.h>
#include <trg/cdc/dataobjects/CDCTriggerSegmentHit.h>
#include <trg/cdc/dataobjects/CDCTriggerMLPInput.h>
 
namespace Belle2 {
  class CDCTriggerNeuroModule : public Module {
  public:
    CDCTriggerNeuroModule();
 
    virtual ~CDCTriggerNeuroModule() {}
 
    virtual void initialize() override;
 
    virtual void event() override;
    float hwInputIdShuffle(float tsid, int sl);
 
  protected:
    std::string m_filename;
    std::string m_arrayname;
    std::string m_hitCollectionName;
    std::string m_EventTimeName;
    std::string m_inputCollectionName;
    std::string m_realinputCollectionName;
    std::string m_outputCollectionName;
    NeuroTrigger m_NeuroTrigger;
    bool m_fixedPoint;
    std::vector<unsigned> m_precision;
    std::string m_et_option;
    bool m_writeMLPinput;
    bool m_hardwareCompatibilityMode;
    bool m_neuroTrackInputMode;
    bool m_min4axials;
    StoreArray<CDCTriggerTrack> m_tracks2D;
    StoreArray<CDCTriggerTrack> m_realtracks2D;
    StoreArray<CDCTriggerTrack> m_tracksNN;
    StoreArray<CDCTriggerSegmentHit> m_segmentHits;
    StoreArray<CDCTriggerMLPInput> m_mlpInput;
    DBObjPtr<CDCTriggerNeuroConfig> m_cdctriggerneuroconfig;
  };
}
#endif