73 lines
3.5 KiB
C++
73 lines
3.5 KiB
C++
#pragma once
|
|
#include <memory>
|
|
#include <stdexcept>
|
|
#include "IR/IR_basic.h"
|
|
#include "naivebackend.h"
|
|
#include "tools.h"
|
|
|
|
namespace NaiveBackend {
|
|
inline void ScanForVar(FuncLayout &layout, std::shared_ptr<ActionItem> action,
|
|
const std::unordered_map<std::string, IRClassInfo> &low_level_class_info) {
|
|
if (std::dynamic_pointer_cast<JMPActionItem>(action)) {
|
|
throw std::runtime_error("JMPActionItem should not be in the layout");
|
|
} else if (auto binary_act = std::dynamic_pointer_cast<BinaryOperationAction>(action)) {
|
|
if (binary_act->result_full == "") {
|
|
throw std::runtime_error("BinaryOperationAction should have a result_full");
|
|
}
|
|
layout.AllocateItem(binary_act->result_full, CalcSize(binary_act->type));
|
|
} else if (auto alloca_act = std::dynamic_pointer_cast<AllocaAction>(action)) {
|
|
if (alloca_act->name_full == "") {
|
|
throw std::runtime_error("AllocaAction should have a name_full");
|
|
}
|
|
layout.AllocateItem(alloca_act->name_full, CalcSize(alloca_act->type), alloca_act->num);
|
|
} else if (auto load_act = std::dynamic_pointer_cast<LoadAction>(action)) {
|
|
if (load_act->result_full == "") {
|
|
throw std::runtime_error("LoadAction should have a result_full");
|
|
}
|
|
layout.AllocateItem(load_act->result_full, CalcSize(load_act->ty));
|
|
} else if (auto store_act = std::dynamic_pointer_cast<StoreAction>(action)) {
|
|
// just do nothing
|
|
} else if (auto get_element_act = std::dynamic_pointer_cast<GetElementPtrAction>(action)) {
|
|
if (get_element_act->result_full == "") {
|
|
throw std::runtime_error("GetElementPtrAction should have a result_full");
|
|
}
|
|
if (get_element_act->indices.size() == 1) {
|
|
layout.AllocateItem(get_element_act->result_full, CalcSize(get_element_act->ty));
|
|
} else if (get_element_act->indices.size() == 2) {
|
|
if (get_element_act->indices[0] != "0")
|
|
throw std::runtime_error("GetElementPtrAction with non-zero base index is not supported");
|
|
size_t element_idx = std::stoi(get_element_act->indices[1]);
|
|
auto class_ty = std::get<LLVMIRCLASSTYPE>(get_element_act->ty);
|
|
const IRClassInfo &class_info = low_level_class_info.at(class_ty.class_name_full);
|
|
layout.AllocateItem(get_element_act->result_full, 4);
|
|
} else {
|
|
throw std::runtime_error("GetElementPtrAction with more than 2 indices is not supported");
|
|
}
|
|
} else if (auto icmp_act = std::dynamic_pointer_cast<ICMPAction>(action)) {
|
|
if (icmp_act->result_full == "") {
|
|
throw std::runtime_error("ICMPAction should have a result_full");
|
|
}
|
|
layout.AllocateItem(icmp_act->result_full, 1);
|
|
} else if (auto call_act = std::dynamic_pointer_cast<CallItem>(action)) {
|
|
if (call_act->result_full == "") {
|
|
if (!std::holds_alternative<LLVMVOIDType>(call_act->return_type)) {
|
|
throw std::runtime_error("CallItem should have a result_full");
|
|
}
|
|
return;
|
|
}
|
|
layout.AllocateItem(call_act->result_full, CalcSize(call_act->return_type));
|
|
} else if (auto phi_act = std::dynamic_pointer_cast<PhiItem>(action)) {
|
|
if (phi_act->result_full == "") {
|
|
throw std::runtime_error("PhiItem should have a result_full");
|
|
}
|
|
layout.AllocateItem(phi_act->result_full, CalcSize(phi_act->ty));
|
|
} else if (auto select_act = std::dynamic_pointer_cast<SelectItem>(action)) {
|
|
if (select_act->result_full == "") {
|
|
throw std::runtime_error("SelectItem should have a result_full");
|
|
}
|
|
layout.AllocateItem(select_act->result_full, CalcSize(select_act->ty));
|
|
} else {
|
|
throw std::runtime_error("Unknown action type");
|
|
}
|
|
}
|
|
} // namespace NaiveBackend
|