save.hpp

Go to the documentation of this file.

#include <list>
#include <unordered_set>

#include "pbm/data.hpp"

#ifndef PBM_SAVE_HPP
#define PBM_SAVE_HPP

namespace pbm
{

using PathedMapT = std::unordered_map<ade::TensptrT,StringsT>;

struct GraphSaver final : public ade::iTraveler
{
    GraphSaver (DataSaverT saver) : saver_(saver) {}

    void visit (ade::iLeaf* leaf) override
    {
        if (visited_.end() == visited_.find(leaf))
        {
            leaf->accept(stat);
            leaves_.push_back(leaf);
            visited_.emplace(leaf);
        }
    }

    void visit (ade::iFunctor* func) override
    {
        if (visited_.end() == visited_.find(func))
        {
            func->accept(stat);
            funcs_.push_back(func);
            visited_.emplace(func);

            ade::ArgsT children = func->get_children();
            for (auto& child : children)
            {
                child.get_tensor()->accept(*this);
            }
        }
    }

    void save (cortenn::Graph& out, PathedMapT labels = PathedMapT());

    std::list<ade::iLeaf*> leaves_;

    std::list<ade::iFunctor*> funcs_;

    std::unordered_set<ade::iTensor*> visited_;

    ade::GraphStat stat;

private:
    void save_coord (
        google::protobuf::RepeatedField<double>* coord,
        const ade::CoordptrT& mapper);

    void save_data (cortenn::Source& out, ade::iLeaf* in)
    {
        const ade::Shape& shape = in->shape();
        char* data = (char*) in->data();
        size_t nelems = shape.n_elems();
        size_t tcode = in->type_code();
        out.set_shape(std::string(shape.begin(), shape.end()));
        out.set_data(saver_(data, nelems, tcode));
        out.set_typecode(tcode);
    }

    DataSaverT saver_;
};

}

#endif // PBM_SAVE_HPP