103 lines
3.5 KiB
C++
103 lines
3.5 KiB
C++
#ifndef TYPES_H
|
|
#define TYPES_H
|
|
|
|
#include "builder.h"
|
|
|
|
#include <string>
|
|
#include <memory>
|
|
|
|
namespace types {
|
|
enum class TypeKind {
|
|
Fundamental,
|
|
Function,
|
|
Pointer,
|
|
Array,
|
|
};
|
|
|
|
enum FundamentalTypeKind {
|
|
Int,
|
|
Bool,
|
|
Char,
|
|
Void,
|
|
/// @brief Mainly used for binop resolution
|
|
Any,
|
|
};
|
|
|
|
class Type {
|
|
public:
|
|
TypeKind m_kind;
|
|
Type(TypeKind kind) : m_kind{ kind } {}
|
|
virtual ~Type() = default;
|
|
virtual std::string formatted() = 0;
|
|
virtual llvm::Type* codegen(codegen::Builder& builder) = 0;
|
|
virtual std::pair<llvm::Value*, std::shared_ptr<Type>> load(codegen::Builder& builder, llvm::Value* ptr) = 0;
|
|
virtual std::optional<std::shared_ptr<Type>> return_type();
|
|
virtual bool is_signed();
|
|
virtual uint32_t size() = 0;
|
|
};
|
|
|
|
class FundamentalType : public Type {
|
|
public:
|
|
FundamentalTypeKind m_ty;
|
|
FundamentalType(FundamentalTypeKind kind) : Type(TypeKind::Fundamental), m_ty{ kind } {}
|
|
virtual ~FundamentalType() override = default;
|
|
virtual std::string formatted() override;
|
|
virtual llvm::Type* codegen(codegen::Builder& builder) override;
|
|
virtual std::pair<llvm::Value*, std::shared_ptr<Type>> load(codegen::Builder& builder, llvm::Value* ptr) override;
|
|
virtual bool is_signed() override;
|
|
virtual uint32_t size() override;
|
|
};
|
|
|
|
|
|
class FunctionType : public Type {
|
|
public:
|
|
std::shared_ptr<Type> m_ret_ty;
|
|
std::vector<std::shared_ptr<Type>> m_param_tys;
|
|
bool m_vararg;
|
|
FunctionType(std::shared_ptr<Type> ret_ty, std::vector<std::shared_ptr<Type>> param_tys, bool vararg)
|
|
: Type(TypeKind::Function)
|
|
, m_ret_ty{ std::move(ret_ty) }
|
|
, m_param_tys{ std::move(param_tys) }
|
|
, m_vararg{ vararg } {
|
|
}
|
|
virtual ~FunctionType() override = default;
|
|
virtual std::string formatted() override;
|
|
virtual llvm::Type* codegen(codegen::Builder& builder) override;
|
|
virtual std::pair<llvm::Value*, std::shared_ptr<Type>> load(codegen::Builder& builder, llvm::Value* ptr) override;
|
|
virtual std::optional<std::shared_ptr<Type>> return_type() override;
|
|
virtual uint32_t size() override;
|
|
};
|
|
|
|
class PointerType : public Type {
|
|
public:
|
|
std::shared_ptr<Type> m_inner;
|
|
|
|
PointerType(std::shared_ptr<Type> inner)
|
|
: Type(TypeKind::Pointer), m_inner{ std::move(inner) } {
|
|
}
|
|
virtual ~PointerType() override = default;
|
|
virtual std::string formatted() override;
|
|
virtual llvm::Type* codegen(codegen::Builder& builder) override;
|
|
virtual std::pair<llvm::Value*, std::shared_ptr<Type>> load(codegen::Builder& builder, llvm::Value* ptr) override;
|
|
virtual uint32_t size() override;
|
|
};
|
|
|
|
class ArrayType : public Type {
|
|
public:
|
|
std::shared_ptr<Type> m_inner;
|
|
uint32_t m_size;
|
|
|
|
ArrayType(std::shared_ptr<Type> inner, uint32_t size)
|
|
: Type(TypeKind::Array), m_inner{ std::move(inner) }, m_size{ size } {
|
|
}
|
|
virtual ~ArrayType() override = default;
|
|
virtual std::string formatted() override;
|
|
virtual llvm::Type* codegen(codegen::Builder& builder) override;
|
|
virtual std::pair<llvm::Value*, std::shared_ptr<Type>> load(codegen::Builder& builder, llvm::Value* ptr) override;
|
|
virtual uint32_t size() override;
|
|
};
|
|
|
|
bool types_equal(std::shared_ptr<types::Type> type1, std::shared_ptr<types::Type> type2);
|
|
}
|
|
|
|
#endif |