NumpyDepoSaver.h

Go to the documentation of this file.

/** Save depos to a Numpy file. */

#ifndef WIRECELLSIO_NUMPYDEPOSAVER
#define WIRECELLSIO_NUMPYDEPOSAVER

#include "WireCellIface/IDepoFilter.h"
#include "WireCellIface/IConfigurable.h"

namespace WireCell {
    namespace Sio {

        // This saver will buffer depos in memory until EOS is received.
        class NumpyDepoSaver : public WireCell::IDepoFilter,
                               public WireCell::IConfigurable {
        public:
            NumpyDepoSaver();
            virtual ~NumpyDepoSaver();

            /// IDepoFilter.  This works by buffering depos and saving
            /// them at the same time a frame is saved.
            virtual bool operator()(const WireCell::IDepo::pointer& indepo,
                                    WireCell::IDepo::pointer& outdepo);

            /// IConfigurable
            virtual WireCell::Configuration default_configuration() const;
            virtual void configure(const WireCell::Configuration& config);
        private:

            Configuration m_cfg;
            int m_save_count;   // count frames saved
            std::vector<WireCell::IDepo::pointer> m_depos;
      };

    }
}
#endif