#include <SDL.h>
#include "viewer.h"
#include "viewer_types.h"
#include "SceneGraph.h"
#include "scene_graph_pack.h"
#include "sys.h"
#include "Func.h"
#include "error.h"
#include "TaskManager.h"
#include "Keyboard.h"
#include "Joystick.h"

extern void post2runLoop(void *);
extern void post2runDraw(void *);

/* measure for FPS (Frame Per Second) */
int start_time;
int this_time;
int frames;

extern SceneGraphPtr scene_graph;
extern SceneGraphPtr scene_graph_viewer;

/* Data Pack */
SceneGraphPack *sgpack;
PolygonPack *ppack;
SpanPackPtr spackList;
SpanPackPtr *spackList_ptr;
int spackList_length;
int spackList_length_align;

/**
 * Joystick があればそれを使い、
 * 無ければキーボードを返す
 */
static Pad*
create_controller(void)
{
    if (SDL_NumJoysticks()) {
	SDL_Joystick *joy = SDL_JoystickOpen(0);
	if (!joy) {
	    fprintf(stderr, "%s: failed to open joystick", __FUNCTION__);
	    return new Keyboard;
	} else {
	    printf("Use Joystick\n");
	    return new Joystick(joy);
	}
    } else {
	printf("Use Keyboard\n");
	return new Keyboard;
    }
}


Viewer::Viewer(int b, int w, int h, int _num)
{
    bpp = b;
    width = w;
    height = h;
    spe_num = _num;
}

int
Viewer::get_ticks()
{
    int time;
    time = SDL_GetTicks();
    return time;
}

bool
Viewer::quit_check()
{
    SDL_Event event;

    while(SDL_PollEvent(&event)) {
	if (event.type==SDL_QUIT) {
	    return true;
	}
    }

    Uint8 *keys=SDL_GetKeyState(NULL);

    if (keys[SDLK_q] == SDL_PRESSED) {
	return true;
    }

    return false;
}

void
Viewer::quit()
{
    SDL_Quit();
}

void
Viewer::swap_buffers()
{
    SDL_GL_SwapBuffers();
}

extern void node_init(void);
extern void create_cube_split(int);
extern void create_snake_bg(int);
extern void universe_init(void);
extern void picture_init(void);

void
Viewer::run_init(char *xml, int sg_number)
{
    HTaskPtr task_next;
    HTaskPtr task_tex;

    start_time = get_ticks();
    this_time  = 0;
    frames     = 0;

    //scene_graph = SceneGraph::createFromXMLfile(xml);
    /**
     * これを実行すると、
     * scene_graph にオリジナルの SceneGraph Object のリストが、
     * scene_graph_viewer に描画用の SceneGraph が生成される
     */
    //SceneGraph::createFromXMLfile(xml);
    //polygon->viewer  = this;

    switch (sg_number) {
    case 0:
    case 1:
	create_cube_split(sg_number);
	break;
    case 2:
    case 3:
    case 4:
	create_snake_bg(sg_number);
	break;
    case 5:
	universe_init();
	break;
    case 6:
	node_init();
	break;
    case 7:
	picture_init();
	break;
    default:
	node_init();
	break;
    }

    scene_graph->controller = create_controller();

    sgpack = (SceneGraphPack*)manager->malloc(sizeof(SceneGraphPack));
    sgpack->init();
    ppack  = (PolygonPack*)manager->malloc(sizeof(PolygonPack));

    spackList_length = (this->height + split_screen_h - 1) / split_screen_h;
    spackList = (SpanPack*)manager->malloc(sizeof(SpanPack)*spackList_length);

    // SPU に送る address list は 16 倍数でないといけない。
    // spackList_length*sizeof(SpanPack*) が 16 倍数になるような
    // length_align を求めている。
    spackList_length_align = (spackList_length + 3)&(~3);

    /* 各 SPU が持つ、SpanPack の address list */
    spackList_ptr =
	(SpanPack**)manager->malloc(sizeof(SpanPack*)*spackList_length_align);

    for (int i = 0; i < spackList_length; i++) {
	spackList_ptr[i] = &spackList[i];
    }

    for (int i = 1; i <= spackList_length; i++) {
	spackList[i-1].init(i*split_screen_h);
    }

    task_next = manager->create_task(TASK_DUMMY);
    task_next->set_post(&post2runLoop, NULL);

#if 0
    // 諸事情で、今は SceneGraphPack を作らずに
    // そのまま SceneGraph でやっています
    HTaskPtr task_sgp;
    task_sgp = manager->create_task(TASK_CREATE_SGP);
    task_sgp->add_param((uint32)scene_graph);
    task_sgp->add_param((uint32)sgpack);
    task_next->wait_for(task_sgp);
    task_sgp->spawn();
#endif

    int tex_width = scene_graph->texture_image->w;
    int tex_height = scene_graph->texture_image->h;
    int tex_blocksize = tex_width*tex_height*4;

    for (int i = 0; i < spe_num; i++) {
	task_tex = manager->create_task(TASK_INIT_TEXTURE);
	task_tex->set_cpu((CPU_TYPE)((int)SPE_0 + i));
	task_next->wait_for(task_tex);
	task_tex->spawn();
    }

    task_next->spawn();
}

void
Viewer::run_loop(void)
{
    HTaskPtr task_create_pp  = NULL;
    HTaskPtr task_create_sp  = NULL;
    HTaskPtr task_next;
    bool quit_flg;

    quit_flg = quit_check();

    if (quit_flg == true) {
	this_time = get_ticks();
	run_finish();
	return;
    }

    clean_pixels();

    for (int i = 1; i <= spackList_length; i++) {
	spackList[i-1].reinit(i*split_screen_h);
    }

    task_next = manager->create_task(TASK_DUMMY);
    task_next->set_post(post2runDraw, NULL);

#if 0
    // SceneGraphPack の update
    HTaskPtr task_update_sgp = NULL;
    task_update_sgp = manager->create_task(TASK_UPDATE_SGP);
    task_update_sgp->add_inData(sgpack, sizeof(SceneGraphPack));
    task_update_sgp->add_outData(sgpack, sizeof(SceneGraphPack));
    task_update_sgp->add_param(width);
    task_update_sgp->add_param(height);
    task_next->wait_for(task_update_sgp);
    task_update_sgp->spawn();
#else
    scene_graph->controller->check();
    scene_graph_viewer->all_execute(width, height);
#endif

#if 0
    // SceneGraphPack(配列) -> PolygonPack
    task_create_pp = manager->create_task(TASK_CREATE_PP);
    task_create_pp->add_inData(sgpack, sizeof(SceneGraphPack));
    task_create_pp->add_param((uint32)ppack);
#else
    // SceneGraph(木構造) -> PolygonPack
    task_create_pp = manager->create_task(TASK_CREATE_PP2);
    task_create_pp->add_param((uint32)scene_graph_viewer);
    task_create_pp->add_param((uint32)ppack);
#endif
    task_next->wait_for(task_create_pp);

    int range_base = spe_num;
    // 切り上げのつもり
    int range = (spackList_length + range_base - 1) / range_base;

    for (int i = 0; i < range_base; i++) {
	int index_start = range*i;
	int index_end = (index_start + range >= spackList_length)
	    ? spackList_length : index_start + range;

	task_create_sp = manager->create_task(TASK_CREATE_SPAN);
	task_create_sp->add_inData(ppack, sizeof(PolygonPack));
	task_create_sp->add_inData(spackList_ptr,
				   sizeof(SpanPack*)*spackList_length_align);
	task_create_sp->add_inData(&spackList[index_start], sizeof(SpanPack));

	task_create_sp->add_param(index_start);

	/**
	 * ex. screen_height が 480, spenum が 6 の場合、各SPEのy担当範囲
	 *   [  1.. 80] [ 81..160] [161..240]
	 *   [241..320] [321..400] [401..480]
	 *
	 * ex. screen_height が 1080, spenum が 5 の場合、
	 *   [  1..216] [217..432] [433..648]
	 *   [649..864] [865..1080]
	 */
	task_create_sp->add_param(index_start*split_screen_h + 1);
	task_create_sp->add_param(index_end*split_screen_h);

	task_next->wait_for(task_create_sp);
	task_create_sp->wait_for(task_create_pp);

	task_create_sp->set_cpu(SPE_ANY);
	task_create_sp->spawn();
    }

    task_create_pp->spawn();
    task_next->spawn();
}

void
Viewer::run_draw(void)
{
    HTaskPtr task_next;
    HTaskPtr task_draw;
    
    task_next = manager->create_task(TASK_DUMMY);
    task_next->set_post(post2runLoop, NULL);
    
    ppack->clear();

    for (int i = 0; i < spackList_length; i++) {
	SpanPack *spack = &spackList[i];
	int startx = 1;
	int endx = split_screen_w;

	int start_y = spack->info.y_top - split_screen_h + 1;
	//int end_y = spack->info.y_top;
	int rangey = (start_y + split_screen_h - 1 > this->height)
	    ? this->height - start_y + 1 : split_screen_h;

	while (startx < this->width) {
	    if (spack->info.size > 0) {
		// Draw SpanPack
		task_draw = manager->create_task(TASK_DRAW_SPAN);
		task_draw->add_inData(spack, sizeof(SpanPack));
	    } else {
		// Draw Background (現在は塗りつぶし)
		task_draw = manager->create_task(TASK_DRAW_BACK);
		task_draw->add_param(0xffffff);
	    }

	    for (int k = 0; k < rangey; k++) {
		task_draw->add_outData(
		    &pixels[(startx-1)+this->width*(k+start_y-1)],
		    (endx - startx + 1)*sizeof(int));
	    }

	    task_draw->add_param(startx);
	    task_draw->add_param(endx);
	    task_draw->add_param(rangey);
	    task_draw->set_cpu(SPE_ANY);
	    task_next->wait_for(task_draw);
	    task_draw->spawn();

	    startx += split_screen_w;
	    endx += split_screen_w;

	    if (endx > this->width) {
		endx = this->width;
	    }
	}
    }

    task_next->spawn();

    frames++;
}

void
Viewer::run_finish(void)
{
    if (this_time != start_time) {
	printf("%f FPS\n", (((float)frames)/(this_time-start_time))*1000.0);
    }

    scene_graph->delete_data();
    scene_graph->controller->close();
    delete scene_graph;

    quit();
}