tens.hpp

Go to the documentation of this file.

#ifndef ZIO_TENS_H_SEEN
#define ZIO_TENS_H_SEEN

#include "zio/message.hpp"

namespace zio {

    namespace tens {

        // The default message "form", if none set.
        const char* form = "TENS";

        const char* type_name(const std::type_info& t);

        template<typename Type>
        const char* dtype() { return type_name(typeid(Type)); }

        void append(Message& msg, message_t&& data,
                    const std::vector<size_t>& shape,
                    size_t word_size, const char* tn);

        template<typename ElementType>
        void append(Message& msg, const ElementType* data, const std::vector<size_t>& shape) {
            size_t nbytes = sizeof(ElementType);
            size_t word = nbytes;
            for (auto s : shape) { nbytes *= s; }
            append(msg, zio::message_t((const void*)data, nbytes),
                   shape, word, type_name(typeid(ElementType)));
        }
                    
        const zio::message_t& at(const Message& msg, size_t index);

        template<typename ElementType>
        const ElementType* at(const Message& msg, size_t index) {
            const zio::message_t& ret = at(msg, index);
            if (ret.empty()) { return nullptr; }
            return (const ElementType*) ret.data();
        }

    }
}

#endif