#include "builder.h"

#include <mie/ctx.h>
#include <mie/ir/ptr.h>
#include <mie/select/graph.h>

static enum mie_status push_load(
	struct mie_select_builder *builder, struct mie_instr *instr)
{
	struct mie_select_graph *graph = mie_select_builder_get_graph(builder);
	struct mie_load *load = (struct mie_load *)instr;

	struct mie_select_value *chain
		= select_builder_get_mem_access(builder, load->l_src);
	if (!chain) {
		chain = &graph->g_entry;
	}

	struct mie_select_value *operands[] = {
		chain,
		select_builder_get_value(builder, load->l_src),
	};
	size_t nr_operands = sizeof operands / sizeof operands[0];

	struct mie_type *result[] = {
		load->l_type,
		mie_ctx_get_type(mie_select_builder_get_ctx(builder), MIE_TYPE_OTHER),
	};
	size_t nr_results = sizeof result / sizeof result[0];

	struct mie_select_value node = {0};
	enum mie_status status = mie_select_graph_get_node(
		graph, MIE_SELECT_OP_LOAD, operands, nr_operands, result,
		nr_results, &node);
	if (status != MIE_SUCCESS) {
		return status;
	}

	select_builder_set_mem_access(builder, load->l_src, &node);
	return select_builder_set_value(builder, MIE_VALUE(instr), &node);
}

static enum mie_status push_store(
	struct mie_select_builder *builder, struct mie_instr *instr)
{
	struct mie_store *store = (struct mie_store *)instr;
	struct mie_select_graph *graph = mie_select_builder_get_graph(builder);

	struct mie_select_value *chain
		= select_builder_get_mem_access(builder, store->s_dest);
	if (!chain) {
		chain = &graph->g_entry;
	}

	struct mie_select_value *operands[] = {
		chain,
		select_builder_get_value(builder, store->s_val),
		select_builder_get_value(builder, store->s_dest),
	};
	size_t nr_operands = sizeof operands / sizeof operands[0];

	struct mie_type *result[] = {
		mie_ctx_get_type(mie_select_builder_get_ctx(builder), MIE_TYPE_OTHER),
	};
	size_t nr_results = sizeof result / sizeof result[0];

	struct mie_select_value node = {0};
	enum mie_status status = mie_select_graph_get_node(
		graph, MIE_SELECT_OP_STORE, operands, nr_operands, result,
		nr_results, &node);
	if (status != MIE_SUCCESS) {
		return status;
	}

	select_builder_set_mem_access(builder, store->s_dest, &node);
	return select_builder_set_value(builder, MIE_VALUE(instr), &node);
}

struct select_instr_type select_load = {
	.i_push = push_load,
};

struct select_instr_type select_store = {
	.i_push = push_store,
};