generic-library/opencv
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

fixed bug in ocl::equalizeHist

Browse Source
This commit is contained in:
Ilya Lavrenov 2013-10-24 13:59:25 +04:00
parent 2279c209c8
commit dab3000778
1 changed files with 32 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

54
modules/ocl/src/opencl/imgproc_histogram.cl
View File

@ -130,12 +130,10 @@ __kernel __attribute__((reqd_work_group_size(HISTOGRAM256_BIN_COUNT,1,1)))void c
globalHist[mad24(gx, hist_step, lid)] = bin1+bin2+bin3+bin4; globalHist[mad24(gx, hist_step, lid)] = bin1+bin2+bin3+bin4;
} }
__kernel void __attribute__((reqd_work_group_size(1,HISTOGRAM256_BIN_COUNT,1)))calc_sub_hist_border_D0( __kernel void __attribute__((reqd_work_group_size(1,HISTOGRAM256_BIN_COUNT,1)))
__global const uchar* src, calc_sub_hist_border_D0(__global const uchar* src, int src_step, int src_offset,
int src_step, int src_offset, __global int* globalHist, int left_col, int cols,
__global int* globalHist, int rows, int hist_step)
int left_col, int cols,
int rows, int hist_step)
{ {
int gidx = get_global_id(0); int gidx = get_global_id(0);
int gidy = get_global_id(1); int gidy = get_global_id(1);
@ -162,6 +160,7 @@ __kernel void __attribute__((reqd_work_group_size(1,HISTOGRAM256_BIN_COUNT,1)))c
globalHist[mad24(rowIndex, hist_step, lidy)] += subhist[lidy]; globalHist[mad24(rowIndex, hist_step, lidy)] += subhist[lidy];
} }
__kernel __attribute__((reqd_work_group_size(256,1,1)))void merge_hist(__global int* buf, __kernel __attribute__((reqd_work_group_size(256,1,1)))void merge_hist(__global int* buf,
__global int* hist, __global int* hist,
int src_step) int src_step)
@ -188,32 +187,43 @@ __kernel __attribute__((reqd_work_group_size(256,1,1)))void merge_hist(__global
hist[gx] = data[0]; hist[gx] = data[0];
} }
__kernel __attribute__((reqd_work_group_size(256,1,1)))void calLUT( __kernel __attribute__((reqd_work_group_size(256,1,1)))
__global uchar * dst, void calLUT(__global uchar * dst, __constant int * hist, int total)
__constant int * hist,
int total)
{ {
int lid = get_local_id(0); int lid = get_local_id(0);
__local int sumhist[HISTOGRAM256_BIN_COUNT+1]; __local int sumhist[HISTOGRAM256_BIN_COUNT];
__local float scale;
sumhist[lid]=hist[lid]; sumhist[lid] = hist[lid];
barrier(CLK_LOCAL_MEM_FENCE); barrier(CLK_LOCAL_MEM_FENCE);
if(lid==0) if (lid == 0)
{ {
int sum = 0; int sum = 0, i = 0;
int i = 0; while (!sumhist[i])
while (!sumhist[i]) ++i; ++i;
sumhist[HISTOGRAM256_BIN_COUNT] = sumhist[i];
for(sumhist[i++] = 0; i<HISTOGRAM256_BIN_COUNT; i++) if (total == sumhist[i])
{ {
sum+=sumhist[i]; scale = 1;
sumhist[i]=sum; for (int j = 0; j < HISTOGRAM256_BIN_COUNT; ++j)
sumhist[i] = i;
}
else
{
scale = 255.f/(total - sumhist[i]);
for (sumhist[i++] = 0; i < HISTOGRAM256_BIN_COUNT; i++)
{
sum += sumhist[i];
sumhist[i] = sum;
}
} }
} }
barrier(CLK_LOCAL_MEM_FENCE); barrier(CLK_LOCAL_MEM_FENCE);
float scale = 255.f/(total - sumhist[HISTOGRAM256_BIN_COUNT]); dst[lid]= convert_uchar_sat_rte(convert_float(sumhist[lid])*scale);
dst[lid]= lid == 0 ? 0 : convert_uchar_sat(convert_float(sumhist[lid])*scale);
} }
/* /*
///////////////////////////////equalizeHist////////////////////////////////////////////////// ///////////////////////////////equalizeHist//////////////////////////////////////////////////
__kernel __attribute__((reqd_work_group_size(256,1,1)))void equalizeHist( __kernel __attribute__((reqd_work_group_size(256,1,1)))void equalizeHist(