WireCell::Sio::NumpyDepoSaver Class Reference

#include <NumpyDepoSaver.h>

Inheritance diagram for WireCell::Sio::NumpyDepoSaver:

Public Member Functions
	NumpyDepoSaver ()
virtual	~NumpyDepoSaver ()
virtual bool	operator() (const WireCell::IDepo::pointer &indepo, WireCell::IDepo::pointer &outdepo)
virtual WireCell::Configuration	default_configuration () const
	IConfigurable. More...
virtual void	configure (const WireCell::Configuration &config)
	Accept a configuration. More...
Public Member Functions inherited from WireCell::IDepoFilter
virtual	~IDepoFilter ()
virtual std::string	signature ()
	Set the signature for all subclasses. More...
Public Member Functions inherited from WireCell::IFunctionNode< IDepo, IDepo >
virtual	~IFunctionNode ()
virtual bool	operator() (const boost::any &anyin, boost::any &anyout)
	The calling signature: More...
virtual std::vector< std::string >	input_types ()
virtual std::vector< std::string >	output_types ()
Public Member Functions inherited from WireCell::IFunctionNodeBase
virtual	~IFunctionNodeBase ()
virtual NodeCategory	category ()
	Return the behavior category type. More...
virtual int	concurrency ()
	By default assume all subclasses are stateless. More...
Public Member Functions inherited from WireCell::INode
virtual	~INode ()
virtual void	reset ()
Public Member Functions inherited from WireCell::IComponent< INode >
virtual	~IComponent ()
Public Member Functions inherited from WireCell::Interface
virtual	~Interface ()
Public Member Functions inherited from WireCell::IConfigurable
virtual	~IConfigurable ()
Public Member Functions inherited from WireCell::IComponent< IConfigurable >
virtual	~IComponent ()

Private Attributes
Configuration	m_cfg
int	m_save_count
std::vector< WireCell::IDepo::pointer >	m_depos

Additional Inherited Members
Public Types inherited from WireCell::IDepoFilter
typedef std::shared_ptr< IDepoFilter >	pointer
Public Types inherited from WireCell::IFunctionNode< IDepo, IDepo >
typedef IDepo	input_type
typedef IDepo	output_type
typedef std::shared_ptr< const IDepo >	input_pointer
typedef std::shared_ptr< const IDepo >	output_pointer
typedef IFunctionNode< IDepo, IDepo >	signature_type
Public Types inherited from WireCell::IFunctionNodeBase
typedef std::shared_ptr< IFunctionNodeBase >	pointer
Public Types inherited from WireCell::INode
enum	NodeCategory {   unknown, sourceNode, sinkNode, functionNode,   queuedoutNode, joinNode, splitNode, faninNode,   fanoutNode, multioutNode, hydraNode }
Public Types inherited from WireCell::IComponent< INode >
typedef std::shared_ptr< INode >	pointer
	Access subclass facet by pointer. More...
typedef std::vector< pointer >	vector
	Vector of shared pointers. More...
Public Types inherited from WireCell::Interface
typedef std::shared_ptr< Interface >	pointer
Public Types inherited from WireCell::IComponent< IConfigurable >
typedef std::shared_ptr< IConfigurable >	pointer
	Access subclass facet by pointer. More...
typedef std::vector< pointer >	vector
	Vector of shared pointers. More...

Detailed Description

Definition at line 13 of file NumpyDepoSaver.h.

Constructor & Destructor Documentation

Sio::NumpyDepoSaver::NumpyDepoSaver

(

)

Definition at line 18 of file NumpyDepoSaver.cxx.

    : m_save_count(0)
{
}

Sio::NumpyDepoSaver::~NumpyDepoSaver ( )

virtual

Definition at line 23 of file NumpyDepoSaver.cxx.

{
}

Member Function Documentation

void Sio::NumpyDepoSaver::configure ( const WireCell::Configuration &  config )

virtual

Accept a configuration.

Implements WireCell::IConfigurable.

Definition at line 41 of file NumpyDepoSaver.cxx.

{
    m_cfg = config;
}

WireCell::Configuration Sio::NumpyDepoSaver::default_configuration ( ) const

virtual

IConfigurable.

Reimplemented from WireCell::IConfigurable.

Definition at line 28 of file NumpyDepoSaver.cxx.

{
    Configuration cfg;

    // The output file name to write.  Only compressed (zipped) Numpy
    // files are supported.  Writing is always in "append" mode.  It's
    // up to the user to delete a previous instance of the file if
    // it's old contents are not wanted.
    cfg["filename"] = "wct-frame.npz";
        
    return cfg;
}

bool Sio::NumpyDepoSaver::operator() ( const WireCell::IDepo::pointer &  indepo, WireCell::IDepo::pointer &  outdepo  )

virtual

IDepoFilter. This works by buffering depos and saving them at the same time a frame is saved.

Implements WireCell::IFunctionNode< IDepo, IDepo >.

Definition at line 69 of file NumpyDepoSaver.cxx.

{
    if (indepo) {
        outdepo = indepo;
        m_depos.push_back(indepo);
        return true;
    }
    outdepo = nullptr;

    const size_t ndepos = m_depos.size();
    if (!ndepos) {
        std::cerr << "NumpyDepoSaver: warning: EOS and no depos seen.\n";
        return true;
    }
        
    auto fdepos = flatten_depos(m_depos);
    const size_t nfdepos = fdepos.size();

    // time, charge, x, y, z, dlong, dtran
    const size_t ndata=7;
    Array::array_xxf data(nfdepos, ndata);
    // ID, pdg, gen, child
    const size_t ninfo = 4;
    Array::array_xxi info(nfdepos, ninfo);
    for (size_t idepo=0; idepo != nfdepos; ++idepo) {
        auto depogc = fdepos[idepo];
        auto depo = std::get<0>(depogc);
        auto gen = std::get<1>(depogc);
        auto child = std::get<2>(depogc);
        data(idepo, 0) = depo->time();
        data(idepo, 1) = depo->charge();
        data(idepo, 2) = depo->pos().x();
        data(idepo, 3) = depo->pos().y();
        data(idepo, 4) = depo->pos().z();
        data(idepo, 5) = depo->extent_long();
        data(idepo, 6) = depo->extent_tran();
        info(idepo, 0) = depo->id();
        info(idepo, 1) = depo->pdg();
        info(idepo, 2) = gen;
        info(idepo, 3) = child;
    }
    const std::string data_name = String::format("depo_data_%d", m_save_count);
    const std::string info_name = String::format("depo_info_%d", m_save_count);

    const std::string fname = m_cfg["filename"].asString();
    const std::string mode = "a";
    cnpy::npz_save(fname, data_name, data.data(), {ndata, ndepos}, mode);
    cnpy::npz_save(fname, info_name, info.data(), {ninfo, ndepos}, mode);
    m_depos.clear();

    ++m_save_count;
    return true;
}

Member Data Documentation

Configuration WireCell::Sio::NumpyDepoSaver::m_cfg

private

Definition at line 29 of file NumpyDepoSaver.h.

std::vector<WireCell::IDepo::pointer> WireCell::Sio::NumpyDepoSaver::m_depos

private

Definition at line 31 of file NumpyDepoSaver.h.

int WireCell::Sio::NumpyDepoSaver::m_save_count

private

Definition at line 30 of file NumpyDepoSaver.h.

---

The documentation for this class was generated from the following files:

• wire-cell-build/sio/inc/WireCellSio/NumpyDepoSaver.h
• wire-cell-build/sio/src/NumpyDepoSaver.cxx