#include <stdio.h>
#include <string.h>
#include "Exec.h"
#include "Func.h"
#include "regexParser.h"
#include "threadedSearch.h"
#include "FileMapReduce.h"

/* これは必須 */
SchedDefineTask1(Exec,blockedGrep);

TSValue blockSearch(TransitionGeneratorPtr tg,Buffer buff) {
    TSValue tsv;
    tsv.buff = buff;
    tsv.tg = tg;
    tsv.current = tg->anyState->tState;
    tsv.result = NULL;
    tsv.resultEnd = &tsv.result;
    unsigned char *end = buff.buffend;
    buff.buffend = buff.buff+1;
    tSearch(tsv);
    tsv.blockBegin = tsv.current;
    buff.buffend = end;
    tSearch(tsv);
    tsv.blockEnd = tsv.current;
    if (tsv.blockEnd->state->bitState.bitContainer != 1) {
        ResultPtr r = NEW(Result);
        r->begin = tsv.buff.matchBegin;
        r->end = NULL;
        *tsv.resultEnd = r;
        r->next = NULL;
        tsv.resultEnd = &r->next;
    }
    return tsv;
}

static int
blockedGrep(SchedTask *s, void *rbuf, void *wbuf)
{
    long task_spwaned = (long)s->get_param(0);
    long division_size = (long)s->get_param(1);
    long length = (long)s->get_param(2);
    long out_size = (long)s->get_param(3);
    MapReduce *w = (MapReduce*)s->get_param(4);
    long allocation = task_spwaned + (long)s->x;
    unsigned char* i_data;
    unsigned long * o_data;
    if (division_size) {
        i_data = (unsigned char*)s->get_input(rbuf,0) + allocation*division_size;
        o_data = (unsigned long *)s->get_output(wbuf,1) + allocation*out_size;
    } else {
        i_data = (unsigned char*)s->get_input(0);
        o_data = (unsigned long *)s->get_output(0);
    }
    TransitionGeneratorPtr tg = (TransitionGeneratorPtr)w->global;
    Buffer buff;
    buff.buff = buff.buffptr = buff.matchBegin = i_data;
    buff.buffend = buff.buff + length;
    TSValue tsv = blockSearch(tg,buff);
    o_data[0] = (unsigned long)tsv.result;
    o_data[1] = (unsigned long)tsv.blockBegin->state;
    o_data[2] = (unsigned long)tsv.blockEnd->state;
    return 0;
}