DataSet.h

Go to the documentation of this file.

 
#ifndef KIDS_DataSet_H
#define KIDS_DataSet_H
 
#include <fstream>
#include <map>
#include <memory>
#include <tuple>
#include <type_traits>
#include <vector>
 
#include "kids/Exception.h"
#include "kids/Node.h"
#include "kids/Shape.h"
#include "kids/Tensor.h"
#include "kids/Types.h"
#include "kids/Variable.h"
#include "kids/concat.h"
 
namespace PROJECT_NS {
 
class DataSet final : public Node {
   private:
    DataSet& operator=(const DataSet&) = delete;
 
   public:
    using DataType = std::map<std::string, std::shared_ptr<Node>>;
    std::shared_ptr<DataType> _data;
 
    DataSet();
 
    kids_int* def(VARIABLE<kids_int>& var);
 
    kids_real* def(VARIABLE<kids_real>& var);
 
    kids_complex* def(VARIABLE<kids_complex>& var);
 
    kids_int* def_int(const std::string& key, Shape S = 1, const std::string& info = "");
 
    kids_int* def_int(const std::string& key, kids_int* arr_in, Shape S = 1, const std::string& info = "");
 
    kids_int* def_int(const std::string& key, const std::string& key_in, const std::string& info = "");
 
    DataSet& _def_int(const std::string& key, Shape S = 1, const std::string& info = "");
 
    kids_real* def_real(const std::string& key, Shape S = 1, const std::string& info = "");
 
    kids_real* def_real(const std::string& key, kids_real* arr_in, Shape S = 1, const std::string& info = "");
 
    kids_real* def_real(const std::string& key, const std::string& key_in, const std::string& info = "");
 
    DataSet& _def_real(const std::string& key, Shape S = 1, const std::string& info = "");
 
    kids_complex* def_complex(const std::string& key, Shape S = 1, const std::string& info = "");
 
    kids_complex* def_complex(const std::string& key, kids_complex* arr_in, Shape S = 1, const std::string& info = "");
 
    kids_complex* def_complex(const std::string& key, const std::string& key_in, const std::string& info = "");
 
    DataSet& _def_complex(const std::string& key, Shape S = 1, const std::string& info = "");
 
    DataSet& _def(const std::string& key, const std::string& key_in, const std::string& info = "");
 
    DataSet& _undef(const std::string& key);
 
    std::tuple<kids_dtype, void*, Shape*> obtain(const std::string& key);
 
    Node* node(const std::string& key);
 
    DataSet* at(const std::string& key);
 
    virtual std::string help(const std::string& name);
 
    virtual std::string repr();
 
    virtual void dump(std::ostream& os);
 
    virtual void load(std::istream& is);
 
   private:
    template <typename T>
    T* def(const std::string& key, Shape S = 1, const std::string& info = "");
 
    class DataSetKeyParser {
       public:
        std::vector<std::string> terms;
        DataSetKeyParser(const std::string& key, const std::string& delimiter = ".") {
            size_t start = 0, end;
            while ((end = key.find(delimiter, start)) != std::string::npos) {
                terms.emplace_back(key, start, end - start);
                start = end + delimiter.length();
            }
            terms.emplace_back(key, start);
        }
    };
};
 
};  // namespace PROJECT_NS
 
#endif  // KIDS_DataSet_H