#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "TaskManager.h"
#include "Task.h"
#include "ListData.h"
#include "Func.h"

extern void task_init(void);

static int length = DATA_NUM;
static int task = 3;
static int count = 3;
static int data_count = 3;

const char *usr_help_str = "Usage: ./twice [-length data_length] [-count task_num]\n\
  -length  Number of data (default DATA_NUM (Func.h))\n\
  -count   Number of task (default 1)\n";


static void
print_data(int *data, int size, const char *title)
{
    printf("%s ---\n", title);
    for (int i = 0; i < size; i++) {
	printf("%2d ", data[i]);
    }
    printf("\n");
}

/**
 * タスク終了後の data1, data2 の確認
 */
static void
twice_result(SchedTask *s, void *a, void *b)
{
    int* data = (int*)a;
    int* task_buf = (int*)b;
    print_data(data, length*count, "after");
    free(data);
    free(task_buf);
}

static int
init(int argc, char **argv)
{
    for (int i = 1; argv[i]; ++i) {
        if (strcmp(argv[i], "-length") == 0) {
            length = atoi(argv[++i]);
        } else if (strcmp(argv[i], "-task") == 0) {
            task = atoi(argv[++i]);
        } else if (strcmp(argv[i], "-count") == 0) {
            count = atoi(argv[++i]);
        } else if (strcmp(argv[i], "-data_count") == 0) {
            data_count = atoi(argv[++i]);
        }
    }

    return 0;
}

#if 0
static void print_ListData(int size, ListElement *t) 
{
    for(int i = 0;i<size;i++) {
	printf("ListData %d %d %lx\n",i,t[i].size,(unsigned long)t[i].addr);
    }
}
#endif

void
twice_init(TaskManager *manager,int *data, int length)
{


    for (int i = 0; i < length*count; i++) {
	data[i] = i;
    }

    print_data(data, length*count, "before");

    /**
     * Create Task
     *   create_task(Task ID);
     */ 

    HTask *twice_main = manager->create_task_array(Twice,count,data_count+1,data_count,data_count);
    Task *t = twice_main->next_task_array(Twice, 0);
#if 0
    printf("allocate task size 0x%0x\n",t->size());
    printf("allocate task total size 0x%0x = 0x%0x * %d\n",twice_main->r_size,
		t->size(), count);
#endif
    void *task_buf = twice_main->rbuf;
    twice_main->set_post(twice_result, (void*)data, task_buf);

    t = 0;
    for(int i = 0;i<count;i++) {
	t = twice_main->next_task_array(Twice, t);
	int length2 = length/data_count;
	t->set_param(0, (memaddr)data_count);

	for(int j = 0;j<data_count;j++) {
	    /**
	     * Set 32bits parameter
	     *   add_param(32bit parameter);
	     */
	    t->set_param(j+1, (memaddr)length2);
	    /**
	     * Set of Input Data
	     *   add_inData(address of input data, size of input data);
	     */
	    t->set_inData(j,data, sizeof(int)*length2);
	    /**
	     * Set of Output area
	     *   add_outData(address of output area, size of output area);
	     */
	    t->set_outData(j,data, sizeof(int)*length2);
	    data += length2;
	}
#if 0
	print_ListData(data_count,t->inData(0));
	print_ListData(data_count,t->outData(0));
	printf("pos 0x%0lx size 0x%0x\n",(unsigned long)t, t->size());
#endif
    }
    twice_main->spawn_task_array(t->next());

    twice_main->set_cpu(SPE_ANY);
    /*
     * set_post() で ppe task を渡せるようにしたい
     */
    // add Active Queue
    twice_main->spawn();    
}

int
TMmain(TaskManager *manager,int argc, char *argv[])
{
    if (init(argc, argv) < 0) {
	return -1;
    }

    // Task Register
    //   ppe/task_init.cc
    task_init();

    for (int i = 0; i < task; ++i) {
	int *data = (int*)manager->allocate(sizeof(int)*length*count);
	twice_init(manager, data, length);
    }

    return 0;
}