#pragma OPENCL EXTENSION cl_khr_byte_addressable_store : enable

__kernel void word_count(__global char* text_data,  __global int* size, __global int* out_data)
{

  //unsigned long long *head_tail_flag = o_data +2;

  uint gid = get_global_id(0);

  int length = *size;
  int word_flag = 0;
  int word_num = 0;
  int line_num = 0;
  int i = 0;
    
  int start = gid*length/2;
  int end = start + length/2;
  
  //head_tail_flag[0] = (i_data[0] != 0x20) && (i_data[0] != 0x0A);
  //word_num -= 1-head_tail_flag[0];

  for (i = 0; i < length; i++) {
  //for (i = start; i < end; i++) {
    if (text_data[i] == 0x20) {
      word_flag = 1;
    } else if (text_data[i] == 0x0A) {
      line_num += 1;
      word_flag = 1;
    } else {
      word_num += word_flag;
      word_flag = 0;
    }
  }

  word_num += word_flag;

  //head_tail_flag[1] = (i_data[i-1] != 0x20) && (i_data[i-1] != 0x0A);
  
  // s->printf("SPE word %d line %d\n",word_num,line_num);

  int index = gid*2;
  
  out_data[index] = word_num;
  out_data[index+1] = line_num;


  
  return 0;

}