#include <stdint.h>
#include <vx_intrinsics.h>
#include "common.h"
#include "texsw.h"

#define ENABLE_SW

struct tile_arg_t {
  	struct kernel_arg_t* state;	
  	uint32_t tile_width;
 	uint32_t tile_height;
  	float deltaX;
  	float deltaY;
};

void kernel_body(int task_id, void* arg) {
	struct tile_arg_t* _arg = (struct tile_arg_t*)(arg);
	struct kernel_arg_t* state = _arg->state;
	
	uint32_t xoffset = 0;
	uint32_t yoffset = task_id * _arg->tile_height;	
	uint8_t* dst_ptr = (uint8_t*)(state->dst_ptr + xoffset * state->dst_stride + yoffset * state->dst_pitch);

	float fv = yoffset * _arg->deltaY;
	for (uint32_t y = 0; y < _arg->tile_height; ++y) {
		uint32_t* dst_row = (uint32_t*)dst_ptr;
		float fu = xoffset * _arg->deltaX;
		for (uint32_t x = 0; x < _arg->tile_width; ++x) {
			int32_t u = (int32_t)(fu * (1<<20));
			int32_t v = (int32_t)(fv * (1<<20));
		#ifdef ENABLE_SW
			if (state->use_sw) {
				dst_row[x] = (state->filter == 2) ? tex3_sw(state, 0, u, v, state->lod) : tex_sw(state, 0, u, v, state->lod);
			} else {
		#endif
			dst_row[x] = (state->filter == 2) ? vx_tex3(0, u, v, state->lod) : vx_tex(0, u, v, state->lod);
		#ifdef ENABLE_SW
			}
		#endif
			fu += _arg->deltaX;
		}
		dst_ptr += state->dst_pitch;
		fv += _arg->deltaY;
	}
}

int main() {
	struct kernel_arg_t* arg = (struct kernel_arg_t*)KERNEL_ARG_DEV_MEM_ADDR;

	// configure texture unit
	vx_csr_write(CSR_TEX_ADDR(0),   arg->src_ptr);
	vx_csr_write(CSR_TEX_MIPOFF(0), 0);	
	vx_csr_write(CSR_TEX_WIDTH(0),  arg->src_logWidth);
	vx_csr_write(CSR_TEX_HEIGHT(0), arg->src_logHeight);
	vx_csr_write(CSR_TEX_FORMAT(0), arg->format);
	vx_csr_write(CSR_TEX_WRAP(0),   (arg->wrap << 2) | arg->wrap);
	vx_csr_write(CSR_TEX_FILTER(0), (arg->filter ? 1 : 0));

	struct tile_arg_t targ;
	targ.state       = arg;
	targ.tile_width  = arg->dst_width;
	targ.tile_height = (arg->dst_height + arg->num_tasks - 1) / arg->num_tasks;    
	targ.deltaX      = 1.0f / arg->dst_width;
	targ.deltaY      = 1.0f / arg->dst_height;
	
	vx_spawn_tasks(arg->num_tasks, kernel_body, &targ);
}