GraphStruct.hpp

Go to the documentation of this file.

#pragma once

#ifndef CHI_STRUCT_TYPE_HPP
#define CHI_STRUCT_TYPE_HPP

#include <string>
#include <vector>

#include "chi/DataType.hpp"
#include "chi/Fwd.hpp"

namespace chi {

struct GraphStruct {
    GraphStruct(GraphModule& mod, std::string name);

    Context& context() const { return *mContext; }

    GraphModule& module() const { return *mModule; }

    std::vector<NodeInstance*> setName(std::string newName, bool updateReferences = true);

    const std::string& name() const { return mName; }

    const std::vector<NamedDataType>& types() const { return mTypes; }

    void addType(DataType ty, std::string name, size_t addBefore, bool updateReferences = true);

    void modifyType(size_t id, DataType newTy, std::string newName, bool updateReferences = true);

    void removeType(size_t id, bool updateReferences = true);

    DataType dataType();

private:
    void updateNodeReferences();

    GraphModule* mModule;
    Context*     mContext;

    std::vector<NamedDataType> mTypes;

    std::string mName;

    DataType mDataType;
};
}  // namespace chi

#endif  // CHI_STRUCT_TYPE_HPP