'功能完善'

admin

2022-06-30 a17738e1545ff7dbef6398b8ec1eab93ab59c9a1

 ConsoleApplication/kernel.cl

@@ -1,10 +1,10 @@

__kernel void recognition_numbers_1(__global const unsigned char* a_in,
   __global const unsigned char* b_in, int width,int num_width,int num_height,int num_count,
   __global const unsigned char* b_in, int width, int num_width, int num_height, int num_count,
   __global unsigned char* result) {

   int p = get_global_id(0);
   int startIndex = p / width * width * num_width * num_height + p % (num_count*10) * num_width;
   int startIndex = p / width * width * num_width * num_height + p % (num_count * 10) * num_width;
   unsigned char t = 0;
   for (int r = 0;r < num_height;r++)
      for (int c = 0;c < num_width;c++) {
@@ -26,7 +26,7 @@
   int endIndex = (index + 1) * 10;

   //获取最小值
   int min =255;
   int min = 255;
   int minIndex = 11;

   for (int i = startIndex;i < endIndex;i++)
@@ -50,7 +50,7 @@
   return v >= 64 ? 1 : 0;
}

__kernel void split_num(__global const unsigned char* img_in, __global const int* pos_in, __global const unsigned char* zero, int width,int num_width,int num_height,int num_count,
__kernel void split_num(__global const unsigned char* img_in, __global const int* pos_in, __global const unsigned char* zero, int width, int num_width, int num_height, int num_count,
   __global unsigned char* result) {

   int index = get_global_id(0);
@@ -350,7 +350,7 @@
   //开始填充数据
   for (i = 0;i < maxNumberCount;i++) {

      unsigned char numData[100*100];
      unsigned char numData[100 * 100];

      if (fresult[i * 2] == -1) {
         //填充0
@@ -358,7 +358,7 @@
            for (int c = 0;c < num_width;c++) {
               unsigned char value = get_binary_value(zero[r * num_width + c]);
               //设置输出坐标的值 
               numData[r* num_width+c] = value;
               numData[r * num_width + c] = value;
            }
         }

@@ -373,10 +373,10 @@
                  //填充空白0   
                  value = get_binary_value(img_in[get_one_level_position(width, _startx + c, r)]);
                  //设置输出坐标的值 
                  numData[(r - starty)* num_width+c] = value;
                  numData[(r - starty) * num_width + c] = value;
               }
               else {
                  numData[(r - starty)* num_width+c] = 0;
                  numData[(r - starty) * num_width + c] = 0;
               }
            }
         }
@@ -404,7 +404,7 @@
            for (int c = 0;c < num_width;c++) {
               int findex = index_0 + index_1 + index_2 + index_3 + c;
               //printf("index:%d-findex:%d \n",index, findex);
               result[findex] = numData[r* num_width+c];
               result[findex] = numData[r * num_width + c];
            }
         }
      }
@@ -417,7 +417,7 @@



__kernel void createTemplateNumbers(__global unsigned char* numbers,int num_width,int num_height, int countPerLine,
__kernel void createTemplateNumbers(__global unsigned char* numbers, int num_width, int num_height, int countPerLine,
   __global unsigned char* result) {

   int gid = get_global_id(0);
@@ -587,16 +587,163 @@
__kernel void rgb2GrayImg(__global unsigned char* imgs, int width,
   __global unsigned char* result) {

   int rowIndex= get_global_id(0);
   int rowIndex = get_global_id(0);

   for (int c = 0;c < width;c++) {
      int index = rowIndex * width + c;
      int start = index * 3;
      unsigned char R = imgs[start];
      unsigned  G = imgs[start + 1];
      unsigned  B = imgs[start + 2];
      unsigned char G = imgs[start + 1];
      unsigned char B = imgs[start + 2];
      result[index] = (76 * R + 150 * G + 30 * B) >> 8;
   }
}

__kernel void rgba2GrayImg(__global unsigned char* imgs, int width,
   __global unsigned char* result) {

   int rowIndex = get_global_id(0);

   for (int c = 0;c < width;c++) {
      int index = rowIndex * width + c;
      int start = index * 4;
      unsigned char R = imgs[start];
      unsigned char G = imgs[start + 1];
      unsigned char B = imgs[start + 2];
      result[index] = (76 * R + 150 * G + 30 * B) >> 8;
   }
}


//L2非数字识别
__kernel void recognition_not_num(__global unsigned char* imgs, __global int* rowIndexs, int width,
   __global int* result) {

   int index = get_global_id(0);
   int row = index / 2;
   int baseIndex = row * 4 * 7;
   if (index % 2 == 0) {
      //涨跌停价
      //获取数据坐标
      int startx = baseIndex + 3 * 4;
      int starty = startx + 1;
      int endx = startx + 2;
      int endy = startx + 3;
      if (rowIndexs[startx] <= 0 && rowIndexs[starty] <= 0 && rowIndexs[endx] <= 0 && rowIndexs[endy] <= 0) {
         result[row * 2] = 0;
      }
      else {
         startx = rowIndexs[startx];
         starty = rowIndexs[starty];
         endx = rowIndexs[endx];
         endy = rowIndexs[endy];

         result[row * 2] = 1;
         //去除上下的空白
         for (int r = starty;r <= endy;r++)
         {
            bool empty = true;
            for (int c = startx;c <= endx;c++) {
               unsigned char value = imgs[get_one_level_position(width, c, r)];
               if (get_binary_value(value)) {
                  empty = 0;
                  break;
               }
            }
            if (!empty) {
               starty = r;
               break;
            }
         }
         for (int r = endy;r >= starty;r--)
         {
            bool empty = true;
            for (int c = startx;c <= endx;c++) {
               unsigned char value = imgs[get_one_level_position(width, c, r)];
               if (get_binary_value(value)) {
                  empty = 0;
                  break;
               }
            }
            if (!empty) {
               endy = r;
               break;
            }
         }
         int my = (starty + endy) / 2;
         //计算上半部分的值
         int topValue = 0;
         for (int r = starty;r <= my;r++)
         {
            for (int c = startx;c <= endx;c++) {
               unsigned char value = imgs[get_one_level_position(width, c, r)];
               topValue += get_binary_value(value);

            }
         }
         //计算下半部分的值
         int bottomValue = 0;
         for (int r = my;r <= endy;r++)
         {
            for (int c = startx;c <= endx;c++) {
               unsigned char value = imgs[get_one_level_position(width, c, r)];
               bottomValue += get_binary_value(value);

            }
         }

         if (topValue > bottomValue) {
            //涨停
            result[row * 2] = 1;
         }
         else {
            //跌停
            result[row * 2] = 2;
         }
      }
   }
   else {
      //操作类型
         //计算值
      int startx = baseIndex + 6 * 4;
      int starty = startx + 1;
      int endx = startx + 2;
      int endy = startx + 3;
      startx = rowIndexs[startx];
      starty = rowIndexs[starty];
      endx = rowIndexs[endx];
      endy = rowIndexs[endy];

      //printf("%d: %d %d %d %d\n",row, startx, starty, endx, endy);

      unsigned int pixelCount = 0;
      for (int r = starty;r <= endy;r++)
      {
         for (int c = startx;c <= endx;c++) {
            unsigned char val = imgs[get_one_level_position(width, c, r)];
            pixelCount+=get_binary_value(val);
         }
      }

      //初始化错误复制
      int value = 4;
      if (abs(pixelCount - 39) < 5) {
         //买
         value = 0;
      }
      else if (abs(pixelCount - 51) < 5) {
         //卖
         value = 2;
      }
      else if (abs(pixelCount - 105) < 5) {
         //买撤
         value = 1;
      }
      else if (abs(pixelCount - 117) < 5) {
         //卖撤
         value = 3;
      }
      result[row * 2 + 1] = value;
   }

}