KernelPool.hpp

 
 
#pragma once
 
#include "AstGlobal.h"
#include <unordered_map>
#include <vector>
#include <string>
 
AST_NAMESPACE_BEGIN
 
union KernelData
{
public:
    KernelData(){
        new (&doubleData_) std::vector<double>();
    }
    KernelData(const std::vector<double>& doubleData)
    {
        new (&doubleData_) std::vector<double>(doubleData);
    }
    KernelData(const std::vector<int>& intData)
    {
        new (&intData_) std::vector<int>(intData);
    }
    KernelData(const std::vector<char>& charData)
    {
        new (&charData_) std::vector<char>(charData);
    }
    KernelData(const KernelData& other)
    {
        new (&charData_) std::vector<char>(other.charData_);
    }
    KernelData(KernelData&& other)
    {
        new (&charData_) std::vector<char>(std::move(other.charData_));
    }
    KernelData& operator=(const KernelData& other)
    {
        charData_ = other.charData_;
        return *this;
    }
    KernelData& operator=(KernelData&& other)
    {
        charData_ = std::move(other.charData_);
        return *this;
    }
    ~KernelData()
    {
        charData_.~vector<char>();
    }
    const std::vector<double>* getDoubleData() const { 
        if(charData_.size() == doubleData_.size() * sizeof(double))
            return &doubleData_; 
        return nullptr;
    }
    const std::vector<int>* getIntData() const {
        if(charData_.size() == intData_.size() * sizeof(int))
            return &intData_; 
        return nullptr;
    }
    const std::vector<char>* getCharData() const { 
        return &charData_; 
    }
 
    enum DataType
    {
        eDouble = 0,
        eInt = 1,
        eChar = 2,
    };
protected:
    // DataType type_;
    std::vector<double> doubleData_;
    std::vector<int>    intData_;
    std::vector<char>   charData_;
};
 
 
class AST_UTIL_API KernelPool
{
public:
    using DataMap = std::unordered_map<std::string, KernelData>;
    KernelPool() = default;
    ~KernelPool() = default;
 
    void setData(StringView name, const KernelData& data);
 
    void setDoubleData(StringView name, const std::vector<double>& data);
 
    void setIntData(StringView name, const std::vector<int>& data);
    
    void setCharData(StringView name, const std::vector<char>& data);
 
 
    const KernelData* getData(StringView name) const;
 
    const std::vector<double>* getDoubleData(StringView name) const;
    
    const std::vector<int>* getIntData(StringView name) const;
    
    const std::vector<char>* getCharData(StringView name) const;
 
    DataMap::iterator begin() { return dataMap_.begin(); }
    DataMap::iterator end() { return dataMap_.end(); }
    DataMap::const_iterator begin() const { return dataMap_.begin(); }
    DataMap::const_iterator end() const { return dataMap_.end(); }
    size_t size() const { return dataMap_.size(); }
protected:
    DataMap dataMap_;
};
 
AST_NAMESPACE_END