/*
	expr_vector.c

	vector expressions

	Copyright (C) 2022 Bill Currie <bill@taniwha.org>

	Author: Bill Currie <bill@taniwha.org>
	Date: 2022/04/27

	This program is free software; you can redistribute it and/or
	modify it under the terms of the GNU General Public License
	as published by the Free Software Foundation; either version 2
	of the License, or (at your option) any later version.

	This program is distributed in the hope that it will be useful,
	but WITHOUT ANY WARRANTY; without even the implied warranty of
	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.

	See the GNU General Public License for more details.

	You should have received a copy of the GNU General Public License
	along with this program; if not, write to:

		Free Software Foundation, Inc.
		59 Temple Place - Suite 330
		Boston, MA  02111-1307, USA

*/
#ifdef HAVE_CONFIG_H
# include "config.h"
#endif

#include "tools/qfcc/include/diagnostic.h"
#include "tools/qfcc/include/expr.h"
#include "tools/qfcc/include/type.h"
#include "tools/qfcc/include/value.h"

const expr_t *
new_vector_value (const type_t *ele_type, int width, int count,
				  const expr_t **elements, bool implicit)
{
	const type_t *vec_type = vector_type (ele_type, width);
	pr_type_t   value[type_size (vec_type)] = {};

	for (int i = 0, offs = 0; i < count; i++) {
		auto src_type = get_type (elements[i]);
		value_store (value + offs, src_type, elements[i]);
		offs += type_size (src_type);
	}

	return new_value_expr (new_type_value (vec_type, value), implicit);
}

const expr_t *
new_matrix_value (const type_t *ele_type, int cols, int rows, int count,
				  const expr_t **elements, bool implicit)
{
	const type_t *mat_type = matrix_type (ele_type, cols, rows);
	pr_type_t   value[type_size (mat_type)] = {};

	for (int i = 0, offs = 0; i < count; i++) {
		auto src_type = get_type (elements[i]);
		value_store (value + offs, src_type, elements[i]);
		offs += type_size (src_type);
	}

	return new_value_expr (new_type_value (mat_type, value), implicit);
}

const expr_t *
new_vector_list_gather (const type_t *type, const expr_t **elements, int count)
{
	auto vec = new_expr ();
	vec->type = ex_vector;
	vec->vector.type = type;
	list_gather (&vec->vector.list, elements, count);
	return vec;
}

const expr_t *
new_vector_list (const expr_t *expr_list)
{
	const type_t *ele_type = type_default;
	int         count = list_count (&expr_list->list);
	const expr_t *elements[count + 1];
	list_scatter (&expr_list->list, elements);
	elements[count] = 0;

	int         width = 0;
	for (int i = 0; i < count; i++) {
		auto e = elements[i];
		auto t = get_type (e);
		if (!t) {
			return e;
		}
		if (!is_math (t)) {
			return error (e, "invalid type for vector element");
		}
		width += type_width (t);
		if (is_nonscalar (t)) {
			t = base_type (t);
		}
		if (type_promotes (t, ele_type)) {
			ele_type = t;
		}
	}
	if (width < 2) {
		return error (expr_list, "not a vector");
	}
	if (width > 4) {
		return error (expr_list, "resulting vector is too large: %d elements",
					  width);
	}

	int         all_constant = 1;
	int         all_implicit = 1;
	for (int i = 0; i < count; i++) {
		auto e = elements[i];
		int         cast_width = type_width (get_type (e));
		const type_t *cast_type = vector_type (ele_type, cast_width);
		all_implicit = all_implicit && e->implicit;
		elements[i] = cast_expr (cast_type, fold_constants (e));
		all_constant = all_constant && is_constant (elements[i]);
	}

	switch (count) {
		case 4:
			// all scalars (otherwise width would be too large)
			break;
		case 3:
			// shuffle any vectors to the beginning of the list (there should
			// be only one, but future...)
			for (int i = 1; i < count; i++) {
				if (is_nonscalar (get_type (elements[i]))) {
					auto t = elements[i];
					int  j = i;
					for (; j > 0 && is_scalar (get_type (elements[j])); j--) {
						elements[j] = elements[j - 1];
					}
					elements[j] = t;
				}
			}
			break;
		case 2:
			if (is_scalar (get_type (elements[0]))
				&& is_nonscalar (get_type (elements[1]))) {
				// swap s, v to be v, s (ie, vector always comes before scalar)
				auto t = elements[0];
				elements[0] = elements[1];
				elements[1] = t;
			}
			break;
		case 1:
			if (is_scalar (get_type (elements[0]))) {
				internal_error (expr_list, "confused about vectors");
			}
			// it's already a vector
			return elements[0];
	}

	if (all_constant) {
		return new_vector_value (ele_type, width, count, elements,
								 all_implicit);
	}

	auto type = vector_type (ele_type, width);
	return new_vector_list_gather (type, elements, count);
}

const expr_t *
vector_to_compound (const expr_t *vector)
{
	if (vector->type != ex_vector) {
		internal_error (vector, "not a vector expression");
	}
	int count = list_count (&vector->vector.list);
	const expr_t *elements[count];
	list_scatter (&vector->vector.list, elements);

	scoped_src_loc (vector);
	auto compound = new_compound_init ();
	compound->compound.type = vector->vector.type;
	for (int i = 0; i < count; i++) {
		auto ele = new_element (elements[i], nullptr);
		append_init_element (&compound->compound, ele);
	}
	return compound;
}