#include	"dda.h"

#include	<string.h>

#include	"timer.h"
#include	"serial.h"
#include	"sermsg.h"

#ifndef	ABS
#define	ABS(v)		(((v) >= 0)?(v):(-(v)))
#endif

#ifndef	ABSDELTA
#define	ABSDELTA(a, b)	(((a) >= (b))?((a) - (b)):((b) - (a)))
#endif

/*
	move queue
*/

uint8_t	mb_head = 0;
uint8_t	mb_tail = 0;
DDA movebuffer[MOVEBUFFER_SIZE];

/*
	position tracking
*/

TARGET startpoint = { 0, 0, 0, 0, FEEDRATE_SLOW_Z };
TARGET current_position = { 0, 0, 0, 0, FEEDRATE_SLOW_Z };

uint8_t queue_full() {
	if (mb_tail == 0)
		return mb_head == (MOVEBUFFER_SIZE - 1);
	else
		return mb_head == (mb_tail - 1);
}

uint8_t queue_empty() {
	return (mb_tail == mb_head) && !movebuffer[mb_tail].live;
}

void enqueue(TARGET *t) {
	while (queue_full())
		delay(WAITING_DELAY);

	uint8_t h = mb_head;
	h++;
	if (h == MOVEBUFFER_SIZE)
		h = 0;
	dda_create(t, &movebuffer[h]);
	mb_head = h;
}

void next_move() {
	if (queue_empty()) {
		// queue is empty
		disable_steppers();
		setTimer(DEFAULT_TICK);
// 		disableTimerInterrupt();
	}
	else {
		uint8_t t = mb_tail;
		t++;
		if (t == MOVEBUFFER_SIZE)
			t = 0;
		dda_start(&movebuffer[t]);
		mb_tail = t;
	}
}

/*
	utility functions
*/

// courtesy of http://www.oroboro.com/rafael/docserv.php/index/programming/article/distance
uint32_t approx_distance( uint32_t dx, uint32_t dy )
{
	uint32_t min, max, approx;

	if ( dx < dy )
	{
		min = dx;
		max = dy;
	} else {
		min = dy;
		max = dx;
	}

	approx = ( max * 1007 ) + ( min * 441 );
	if ( max < ( min << 4 ))
		approx -= ( max * 40 );

	// add 512 for proper rounding
	return (( approx + 512 ) >> 10 );
}

// courtesy of http://www.oroboro.com/rafael/docserv.php/index/programming/article/distance
uint32_t approx_distance_3( uint32_t dx, uint32_t dy, uint32_t dz )
{
	uint32_t min, med, max, approx;

	if ( dx < dy )
	{
		min = dy;
		med = dx;
	} else {
		min = dx;
		med = dy;
	}

	if ( dz < min )
	{
		max = med;
		med = min;
		min = dz;
	} else if ( dz < med ) {
		max = med;
		med = dz;
	} else {
		max = dz;
	}

	approx = ( max * 860 ) + ( med * 851 ) + ( min * 520 );
	if ( max < ( med << 1 )) approx -= ( max * 294 );
	if ( max < ( min << 2 )) approx -= ( max * 113 );
	if ( med < ( min << 2 )) approx -= ( med *  40 );

	// add 512 for proper rounding
	return (( approx + 512 ) >> 10 );
}

uint32_t abs32(int32_t v) {
	if (v < 0)
		return (uint32_t) (-v);
	return (uint32_t) (v);
}

/*
	CREATE
*/

void dda_create(TARGET *target, DDA *dda) {
	uint32_t	distance;

	serial_writestr_P(PSTR("\n{DDA_CREATE: ["));

	// we end at the passed target
	memcpy(&(dda->endpoint), target, sizeof(TARGET));

	dda->x_delta = abs32(dda->endpoint.X - startpoint.X);
	dda->y_delta = abs32(dda->endpoint.Y - startpoint.Y);
	dda->z_delta = abs32(dda->endpoint.Z - startpoint.Z);
	dda->e_delta = abs32(dda->endpoint.E - startpoint.E);
	dda->f_delta = abs32(dda->endpoint.F - startpoint.F);

	serwrite_uint32(dda->x_delta); serial_writechar(',');
	serwrite_uint32(dda->y_delta); serial_writechar(',');
	serwrite_uint32(dda->z_delta); serial_writechar(',');
	serwrite_uint32(dda->e_delta); serial_writechar(',');
	serwrite_uint32(dda->f_delta); serial_writestr_P(PSTR("] ["));

	dda->total_steps = dda->x_delta;
	if (dda->y_delta > dda->total_steps)
		dda->total_steps = dda->y_delta;
	if (dda->z_delta > dda->total_steps)
		dda->total_steps = dda->z_delta;

	if (dda->e_delta > dda->total_steps)
		dda->total_steps = dda->e_delta;

	if (dda->total_steps == 0)
		dda->nullmove = 1;

	serwrite_uint32(dda->total_steps); serial_writechar(',');

	if (dda->f_delta > dda->total_steps) {
		dda->f_scale = dda->f_delta / dda->total_steps;
		if (dda->f_scale > 3) {
			dda->f_delta = dda->total_steps;
		}
		else {
			dda->f_scale = 1;
			dda->total_steps = dda->f_delta;
		}
	}
	else {
		dda->f_scale = 1;
	}

	serwrite_uint32(dda->total_steps); serial_writechar(',');

	dda->x_direction = (dda->endpoint.X >= startpoint.X)?1:0;
	dda->y_direction = (dda->endpoint.Y >= startpoint.Y)?1:0;
	dda->z_direction = (dda->endpoint.Z >= startpoint.Z)?1:0;
	dda->e_direction = (dda->endpoint.E >= startpoint.E)?1:0;
	dda->f_direction = (dda->endpoint.F >= startpoint.F)?1:0;

	dda->x_counter = dda->y_counter = dda->z_counter = dda->e_counter = dda->f_counter =
		-(dda->total_steps >> 1);

	// since it's unusual to combine X, Y and Z changes in a single move on reprap, check if we can use simpler approximations before trying the full 3d approximation.
	if (dda->z_delta == 0)
		distance = approx_distance(dda->x_delta * 1000, dda->y_delta * 1000) / ((uint32_t) STEPS_PER_MM_X);
	else if (dda->x_delta == 0 && dda->y_delta == 0)
		distance = dda->z_delta * ((uint32_t) (1000 / STEPS_PER_MM_Z));
	else
		distance = approx_distance_3(dda->x_delta * ((uint32_t) (1000 * STEPS_PER_MM_Z / STEPS_PER_MM_X)), dda->y_delta * ((uint32_t) (1000 * STEPS_PER_MM_Z / STEPS_PER_MM_Y)), dda->z_delta * 1000) / ((uint32_t) STEPS_PER_MM_Z);

	if (distance < 2)
		distance = dda->e_delta * ((uint32_t) (1000 / STEPS_PER_MM_E));
	if (distance < 2)
		distance = dda->f_delta;

	// pre-calculate move speed in millimeter microseconds per step minute for less math in interrupt context
	// mm (distance) * 1000 us/ms * 60000000 us/min / step (total_steps) = mm.us per step.min
	// so in the interrupt we must simply calculate
	// mm.us per step.min / mm per min (F) = us per step
	if (dda->total_steps > 0)
		dda->move_duration = distance * 60000 / dda->total_steps;
	else
		dda->move_duration = 0;

	serwrite_uint32(dda->move_duration); serial_writestr_P(PSTR("] }\n"));

	// next dda starts where we finish
	memcpy(&startpoint, target, sizeof(TARGET));

	// not running yet, we fire up in dda_start()
	dda->live = 0;

	// fire up
	enableTimerInterrupt();
}

/*
	START
*/

void dda_start(DDA *dda) {
	// called from interrupt context: keep it simple!
	if (dda->nullmove) {
		// just change speed?
		current_position.F = dda->endpoint.F;
		return;
	}

	x_direction(dda->x_direction);
	y_direction(dda->y_direction);
	z_direction(dda->z_direction);
	e_direction(dda->e_direction);

	enable_steppers();

	dda->firstep = 1;

	dda->live = 1;
}

/*
	CAN STEP
*/

uint8_t	can_step(uint8_t min, uint8_t max, int32_t current, int32_t target, uint8_t dir) {
	if (current == target)
		return 0;

	if (dir) {
		// forwards/positive
		if (max)
			return 0;
		if (current > target)
			return 0;
	}
	else {
		// backwards/negative
		if (min)
			return 0;
		if (target > current)
			return 0;
	}

	return 255;
}

/*
	STEP
*/

void dda_step(DDA *dda) {
	uint8_t	step_option = 0;
#define	X_CAN_STEP	1
#define	Y_CAN_STEP	2
#define	Z_CAN_STEP	4
#define	E_CAN_STEP	8
#define	F_CAN_STEP	16
#define	REAL_MOVE		32
#define	F_REAL_STEP	64

	WRITE(SCK, 1);

	do {
// 		step_option |= can_step(x_min(), x_max(), current_position.X, dda->endpoint.X, dda->x_direction) & X_CAN_STEP;
// 		step_option |= can_step(y_min(), y_max(), current_position.Y, dda->endpoint.Y, dda->y_direction) & Y_CAN_STEP;
// 		step_option |= can_step(z_min(), z_max(), current_position.Z, dda->endpoint.Z, dda->z_direction) & Z_CAN_STEP;
		step_option |= can_step(0      , 0      , current_position.X, dda->endpoint.X, dda->x_direction) & X_CAN_STEP;
		step_option |= can_step(0      , 0      , current_position.Y, dda->endpoint.Y, dda->y_direction) & Y_CAN_STEP;
		step_option |= can_step(0      , 0      , current_position.Z, dda->endpoint.Z, dda->z_direction) & Z_CAN_STEP;
		step_option |= can_step(0      , 0      , current_position.E, dda->endpoint.E, dda->e_direction) & E_CAN_STEP;
		step_option |= can_step(0      , 0      , current_position.F, dda->endpoint.F, dda->f_direction) & F_CAN_STEP;

		if (step_option & X_CAN_STEP) {
			dda->x_counter -= dda->x_delta;
			if (dda->x_counter < 0) {
				step_option |= REAL_MOVE;

				x_step();
				if (dda->x_direction)
					current_position.X++;
				else
					current_position.X--;

				dda->x_counter += dda->total_steps;
			}
		}

		if (step_option & Y_CAN_STEP) {
			dda->y_counter -= dda->y_delta;
			if (dda->y_counter < 0) {
				step_option |= REAL_MOVE;

				y_step();
				if (dda->y_direction)
					current_position.Y++;
				else
					current_position.Y--;

				dda->y_counter += dda->total_steps;
			}
		}

		if (step_option & Z_CAN_STEP) {
			dda->z_counter -= dda->z_delta;
			if (dda->z_counter < 0) {
				step_option |= REAL_MOVE;

				z_step();
				if (dda->z_direction)
					current_position.Z++;
				else
					current_position.Z--;

				dda->z_counter += dda->total_steps;
			}
		}

		if (step_option & E_CAN_STEP) {
			dda->e_counter -= dda->e_delta;
			if (dda->e_counter < 0) {
				step_option |= REAL_MOVE;

				e_step();
				if (dda->e_direction)
					current_position.E++;
				else
					current_position.E--;

				dda->e_counter += dda->total_steps;
			}
		}

		if (step_option & F_CAN_STEP) {
			dda->f_counter -= dda->f_delta;
			if (dda->f_counter < 0) {

				dda->f_counter += dda->total_steps;

				if (dda->f_scale == 0)
					dda->f_scale = 1;

				if (dda->f_direction) {
					current_position.F += dda->f_scale;
					if (current_position.F > dda->endpoint.F)
						current_position.F = dda->endpoint.F;
				}
				else {
					current_position.F -= dda->f_scale;
					if (current_position.F < dda->endpoint.F)
						current_position.F = dda->endpoint.F;
				}

				step_option |= F_REAL_STEP;
			}
		}

		serial_writechar('[');
		serwrite_hex8(step_option);
		serial_writechar(':');
		serwrite_uint16(dda->f_scale);
		serial_writechar(',');
		serwrite_int32(current_position.F);
		serial_writechar('/');
		serwrite_int32(dda->endpoint.F);
		serial_writechar('#');
		serwrite_uint32(dda->move_duration);
		serial_writechar(']');
	} while (	((step_option & REAL_MOVE ) == 0)	&&
						((step_option & F_CAN_STEP) != 0)	);

	// turn off step outputs, hopefully they've been on long enough by now to register with the drivers
	unstep();

	// we have stepped in speed and now need to recalculate our delay
	if ((step_option & REAL_MOVE) && ((step_option & F_REAL_STEP) || (dda->firstep))) {
		setTimer(dda->move_duration / current_position.F);
		dda->firstep = 0;
	}

	// if we could step, we're still running
	dda->live = (step_option & (X_CAN_STEP | Y_CAN_STEP | Z_CAN_STEP | E_CAN_STEP | F_CAN_STEP))?1:0;

	WRITE(SCK, 0);
}