DataElements.hpp

浏览该文件的文档.

 
 
#pragma once
 
#include "AstGlobal.h"
#include "DataElement.hpp"
#include "AstUtil/Dimension.hpp"
#include "AstUtil/VariantVector.hpp"
#include <map>
#include <vector>
#include <string>
 
AST_NAMESPACE_BEGIN
 
class DataElements;
 
namespace detail
{
    template<typename T, typename MemberType, MemberType T::*Member>
    ElementExtractFunc makeExtractFunc()
    {
        return [](const VariantVector& data, VariantVector& element) -> errc_t
        {
            auto span = data.asSpan<const T>();
            if (!span.data()) return -1;
            using ElementType = typename std::remove_cv<typename std::remove_reference<MemberType>::type>::type;
            element.reset<ElementType>(span.size());
            auto* out = element.as<ElementType>();
            if (!out) return -1;
            for (size_t i = 0; i < span.size(); ++i)
                out[i] = span[i].*Member;
            return 0;
        };
    }
 
    template<typename T, typename Ret, Ret (T::*Getter)() const>
    ElementExtractFunc makeExtractFunc()
    {
        return [](const VariantVector& data, VariantVector& element) -> errc_t
        {
            auto span = data.asSpan<const T>();
            if (!span.data()) return -1;
            using ElementType = typename std::remove_cv<typename std::remove_reference<Ret>::type>::type;
            element.reset<ElementType>(span.size());
            auto* out = element.as<ElementType>();
            if (!out) return -1;
            for (size_t i = 0; i < span.size(); ++i)
                out[i] = (span[i].*Getter)();
            return 0;
        };
    }
} // namespace detail
 
 
class DataElements
{
public:
    DataElements() = default;
    ~DataElements() = default;
 
    const DataElement* getElement(StringView name) const;
    const DataElement* find(StringView name) const{return getElement(name);}
    
    const std::vector<DataElement>& getElements() const { return elements_; }
 
    void addElement(const DataElement& element);
 
    void addAlias(const char* alias, const char* name);
 
    template<typename T, typename MemberType, MemberType T::*Member>
    void addElement(const char* name, Dimension dimension = Dimension::Unit())
    {
        DataElement element;
        element.setName(name);
        element.setDimension(dimension);
        element.setExtractFunc(detail::makeExtractFunc<T, MemberType, Member>());
        addElement(element);
    }
 
    template<typename T, typename Ret, Ret (T::*Getter)() const>
    void addElement(const char* name, Dimension dimension = Dimension::Unit())
    {
        DataElement element;
        element.setName(name);
        element.setDimension(dimension);
        element.setExtractFunc(detail::makeExtractFunc<T, Ret, Getter>());
        addElement(element);
    }
    
private:
    std::vector<DataElement> elements_{};                            
    std::vector<std::pair<std::string, std::string>> aliasMap_{};    
};
 
const int i = sizeof(DataElements);
 
AST_NAMESPACE_END