armorpaint/paint/sources/nodes_brush/vector_math_node.c

#include "../global.h"

typedef struct vector_math_node {
	struct logic_node *base;
	char              *operation;
	vec4_t             v;
} vector_math_node_t;

logic_node_value_t *vector_math_node_get(vector_math_node_t *self, i32 from) {
	vec4_t v1 = logic_node_input_get(self->base->inputs->buffer[0])->_vec4;
	vec4_t v2 = logic_node_input_get(self->base->inputs->buffer[1])->_vec4;
	self->v   = v1;
	f32   f   = 0.0;
	char *op  = self->operation;
	if (string_equals(op, "Add")) {
		self->v = vec4_add(self->v, v2);
	}
	else if (string_equals(op, "Subtract")) {
		self->v = vec4_sub(self->v, v2);
	}
	else if (string_equals(op, "Average")) {
		self->v = vec4_add(self->v, v2);
		self->v.x *= 0.5;
		self->v.y *= 0.5;
		self->v.z *= 0.5;
	}
	else if (string_equals(op, "Dot Product")) {
		f       = vec4_dot(self->v, v2);
		self->v = (vec4_t){f, f, f, 1.0};
	}
	else if (string_equals(op, "Cross Product")) {
		self->v = vec4_cross(self->v, v2);
	}
	else if (string_equals(op, "Normalize")) {
		self->v = vec4_norm(self->v);
	}
	else if (string_equals(op, "Multiply")) {
		self->v.x *= v2.x;
		self->v.y *= v2.y;
		self->v.z *= v2.z;
	}
	else if (string_equals(op, "Divide")) {
		self->v.x /= v2.x == 0.0 ? 0.000001 : v2.x;
		self->v.y /= v2.y == 0.0 ? 0.000001 : v2.y;
		self->v.z /= v2.z == 0.0 ? 0.000001 : v2.z;
	}
	else if (string_equals(op, "Length")) {
		f       = vec4_len(self->v);
		self->v = (vec4_t){f, f, f, 1.0};
	}
	else if (string_equals(op, "Distance")) {
		f       = vec4_dist(self->v, v2);
		self->v = (vec4_t){f, f, f, 1.0};
	}
	else if (string_equals(op, "Project")) {
		self->v = v2;
		self->v = vec4_mult(self->v, vec4_dot(v1, v2) / (float)vec4_dot(v2, v2));
	}
	else if (string_equals(op, "Reflect")) {
		vec4_t tmp = (vec4_t){0.0, 0.0, 0.0, 1.0};
		tmp        = v2;
		tmp        = vec4_norm(tmp);
		self->v    = vec4_reflect(self->v, tmp);
	}
	else if (string_equals(op, "Scale")) {
		self->v.x *= v2.x;
		self->v.y *= v2.x;
		self->v.z *= v2.x;
	}
	else if (string_equals(op, "Absolute")) {
		self->v.x = math_abs(self->v.x);
		self->v.y = math_abs(self->v.y);
		self->v.z = math_abs(self->v.z);
	}
	else if (string_equals(op, "Minimum")) {
		self->v.x = math_min(v1.x, v2.x);
		self->v.y = math_min(v1.y, v2.y);
		self->v.z = math_min(v1.z, v2.z);
	}
	else if (string_equals(op, "Maximum")) {
		self->v.x = math_max(v1.x, v2.x);
		self->v.y = math_max(v1.y, v2.y);
		self->v.z = math_max(v1.z, v2.z);
	}
	else if (string_equals(op, "Floor")) {
		self->v.x = math_floor(v1.x);
		self->v.y = math_floor(v1.y);
		self->v.z = math_floor(v1.z);
	}
	else if (string_equals(op, "Ceil")) {
		self->v.x = math_ceil(v1.x);
		self->v.y = math_ceil(v1.y);
		self->v.z = math_ceil(v1.z);
	}
	else if (string_equals(op, "Fraction")) {
		self->v.x = v1.x - math_floor(v1.x);
		self->v.y = v1.y - math_floor(v1.y);
		self->v.z = v1.z - math_floor(v1.z);
	}
	else if (string_equals(op, "Modulo")) {
		self->v.x = math_fmod(v1.x, v2.x);
		self->v.y = math_fmod(v1.y, v2.y);
		self->v.z = math_fmod(v1.z, v2.z);
	}
	else if (string_equals(op, "Snap")) {
		self->v.x = math_floor(v1.x / (float)v2.x) * v2.x;
		self->v.y = math_floor(v1.y / (float)v2.y) * v2.y;
		self->v.z = math_floor(v1.z / (float)v2.z) * v2.z;
	}
	else if (string_equals(op, "Sine")) {
		self->v.x = math_sin(v1.x);
		self->v.y = math_sin(v1.y);
		self->v.z = math_sin(v1.z);
	}
	else if (string_equals(op, "Cosine")) {
		self->v.x = math_cos(v1.x);
		self->v.y = math_cos(v1.y);
		self->v.z = math_cos(v1.z);
	}
	else if (string_equals(op, "Tangent")) {
		self->v.x = math_tan(v1.x);
		self->v.y = math_tan(v1.y);
		self->v.z = math_tan(v1.z);
	}

	if (from == 0) {
		logic_node_value_t *v = GC_ALLOC_INIT(logic_node_value_t, {._vec4 = self->v});
		return v;
	}
	else {
		logic_node_value_t *v = GC_ALLOC_INIT(logic_node_value_t, {._f32 = f});
		return v;
	}
}

void *vector_math_node_create(ui_node_t *raw, f32_array_t *args) {
	vector_math_node_t *n = GC_ALLOC_INIT(vector_math_node_t, {0});
	n->base               = logic_node_create(n);
	n->base->get          = vector_math_node_get;
	n->v                  = (vec4_t){0.0, 0.0, 0.0, 1.0};
	return n;
}