particle_single.cl 9.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353
  1. // #if defined(cl_amd_fp64) || defined(cl_khr_fp64)
  2. // #if defined(cl_amd_fp64)
  3. // #pragma OPENCL EXTENSION cl_amd_fp64 : enable
  4. // #elif defined(cl_khr_fp64)
  5. // #pragma OPENCL EXTENSION cl_khr_fp64 : enable
  6. // #endif
  7. #define SCALE_FACTOR 300
  8. /** added this function. was missing in original float version.
  9. * Takes in a float and returns an integer that approximates to that float
  10. * @return if the mantissa < .5 => return value < input value; else return value > input value
  11. */
  12. float dev_round_float(float value) {
  13. int newValue = (int) (value);
  14. if (value - newValue < .5f)
  15. return newValue;
  16. else
  17. return newValue++;
  18. }
  19. /********************************
  20. * CALC LIKELIHOOD SUM
  21. * DETERMINES THE LIKELIHOOD SUM BASED ON THE FORMULA: SUM( (IK[IND] - 100)^2 - (IK[IND] - 228)^2)/ 100
  22. * param 1 I 3D matrix
  23. * param 2 current ind array
  24. * param 3 length of ind array
  25. * returns a float representing the sum
  26. ********************************/
  27. float calcLikelihoodSum(__global unsigned char * I, __global int * ind, int numOnes, int index){
  28. float likelihoodSum = 0.0;
  29. int x;
  30. for(x = 0; x < numOnes; x++)
  31. likelihoodSum += (pow((float)(I[ind[index*numOnes + x]] - 100),2) - pow((float)(I[ind[index*numOnes + x]]-228),2))/50.0;
  32. return likelihoodSum;
  33. }
  34. /****************************
  35. CDF CALCULATE
  36. CALCULATES CDF
  37. param1 CDF
  38. param2 weights
  39. param3 Nparticles
  40. *****************************/
  41. void cdfCalc(__global float * CDF, __global float * weights, int Nparticles){
  42. int x;
  43. CDF[0] = weights[0];
  44. for(x = 1; x < Nparticles; x++){
  45. CDF[x] = weights[x] + CDF[x-1];
  46. }
  47. }
  48. /*****************************
  49. * RANDU
  50. * GENERATES A UNIFORM DISTRIBUTION
  51. * returns a float representing a randomily generated number from a uniform distribution with range [0, 1)
  52. ******************************/
  53. float d_randu(__global int * seed, int index)
  54. {
  55. int M = INT_MAX;
  56. int A = 1103515245;
  57. int C = 12345;
  58. int num = A*seed[index] + C;
  59. seed[index] = num % M;
  60. return fabs(seed[index] / ((float) M));
  61. }
  62. /**
  63. * Generates a normally distributed random number using the Box-Muller transformation
  64. * @note This function is thread-safe
  65. * @param seed The seed array
  66. * @param index The specific index of the seed to be advanced
  67. * @return a float representing random number generated using the Box-Muller algorithm
  68. * @see http://en.wikipedia.org/wiki/Normal_distribution, section computing value for normal random distribution
  69. */
  70. float d_randn(__global int * seed, int index){
  71. //Box-Muller algortihm
  72. float pi = 3.14159265358979323846;
  73. float u = d_randu(seed, index);
  74. float v = d_randu(seed, index);
  75. float cosine = cos(2*pi*v);
  76. float rt = -2*log(u);
  77. return sqrt(rt)*cosine;
  78. }
  79. /****************************
  80. UPDATE WEIGHTS
  81. UPDATES WEIGHTS
  82. param1 weights
  83. param2 likelihood
  84. param3 Nparticles
  85. ****************************/
  86. float updateWeights(__global float * weights, __global float * likelihood, int Nparticles){
  87. int x;
  88. float sum = 0;
  89. for(x = 0; x < Nparticles; x++){
  90. weights[x] = weights[x] * exp(likelihood[x]);
  91. sum += weights[x];
  92. }
  93. return sum;
  94. }
  95. int findIndexBin(__global float * CDF, int beginIndex, int endIndex, float value)
  96. {
  97. if(endIndex < beginIndex)
  98. return -1;
  99. int middleIndex;
  100. while(endIndex > beginIndex)
  101. {
  102. middleIndex = beginIndex + ((endIndex-beginIndex)/2);
  103. if(CDF[middleIndex] >= value)
  104. {
  105. if(middleIndex == 0)
  106. return middleIndex;
  107. else if(CDF[middleIndex-1] < value)
  108. return middleIndex;
  109. else if(CDF[middleIndex-1] == value)
  110. {
  111. while(CDF[middleIndex] == value && middleIndex >= 0)
  112. middleIndex--;
  113. middleIndex++;
  114. return middleIndex;
  115. }
  116. }
  117. if(CDF[middleIndex] > value)
  118. endIndex = middleIndex-1;
  119. else
  120. beginIndex = middleIndex+1;
  121. }
  122. return -1;
  123. }
  124. /*******************************************
  125. * OpenCL helper function to read a single element from a 2d image
  126. * param1: img
  127. * param2: index
  128. *******************************************/
  129. float tex1Dfetch(__read_only image2d_t img, int index){
  130. const sampler_t smp = CLK_NORMALIZED_COORDS_FALSE | //Natural coordinates
  131. CLK_ADDRESS_CLAMP | //Clamp to zeros
  132. CLK_FILTER_NEAREST; //Don't interpolate
  133. if (index < 0) return 0;
  134. //Divide desired position by 4 because there are 4 components per pixel
  135. int imgPos = index >> 2;
  136. int2 coords;
  137. coords.x = imgPos >> 13;
  138. coords.y = imgPos & 0x1fff; //Reads the float4
  139. float4 temp = read_imagef(img, smp, coords);
  140. //Computes the remainder of imgPos / 4 to check if function should return x,y,z or w component.
  141. imgPos = index & 0x0003;
  142. if (imgPos < 2){
  143. if (imgPos == 0) return temp.x;
  144. else return temp.y;
  145. }
  146. else{
  147. if (imgPos == 2) return temp.z;
  148. else return temp.w;
  149. }
  150. }
  151. /*****************************
  152. * CUDA Find Index Kernel Function to replace FindIndex
  153. * param1: arrayX
  154. * param2: arrayY
  155. * param3: CDF
  156. * param4: u
  157. * param5: xj
  158. * param6: yj
  159. * param7: weights
  160. * param8: Nparticles
  161. *****************************/
  162. __kernel void find_index_kernel(__global float * arrayX, __global float * arrayY,
  163. __global float * CDF, __global float * u, __global float * xj,
  164. __global float * yj, __global float * weights, int Nparticles
  165. ){
  166. int i = get_global_id(0);
  167. if(i < Nparticles){
  168. int index = -1;
  169. int x;
  170. for(x = 0; x < Nparticles; x++){
  171. if(CDF[x] >= u[i]){
  172. index = x;
  173. break;
  174. }
  175. }
  176. if(index == -1){
  177. index = Nparticles-1;
  178. }
  179. xj[i] = arrayX[index];
  180. yj[i] = arrayY[index];
  181. //weights[i] = 1 / ((float) (Nparticles)); //moved this code to the beginning of likelihood kernel
  182. }
  183. barrier(CLK_GLOBAL_MEM_FENCE);
  184. }
  185. __kernel void normalize_weights_kernel(__global float * weights, int Nparticles, __global float * partial_sums, __global float * CDF, __global float * u, __global int * seed)
  186. {
  187. int i = get_global_id(0);
  188. int local_id = get_local_id(0);
  189. __local float u1;
  190. __local float sumWeights;
  191. if(0 == local_id)
  192. sumWeights = partial_sums[0];
  193. barrier(CLK_LOCAL_MEM_FENCE);
  194. if(i < Nparticles) {
  195. weights[i] = weights[i]/sumWeights;
  196. }
  197. barrier(CLK_GLOBAL_MEM_FENCE);
  198. if(i == 0) {
  199. cdfCalc(CDF, weights, Nparticles);
  200. u[0] = (1/((float)(Nparticles))) * d_randu(seed, i); // do this to allow all threads in all blocks to use the same u1
  201. }
  202. barrier(CLK_GLOBAL_MEM_FENCE);
  203. if(0 == local_id)
  204. u1 = u[0];
  205. barrier(CLK_LOCAL_MEM_FENCE);
  206. if(i < Nparticles)
  207. {
  208. u[i] = u1 + i/((float)(Nparticles));
  209. }
  210. }
  211. __kernel void sum_kernel(__global float* partial_sums, int Nparticles)
  212. {
  213. int i = get_global_id(0);
  214. size_t THREADS_PER_BLOCK = get_local_size(0);
  215. if(i == 0)
  216. {
  217. int x;
  218. float sum = 0;
  219. int num_blocks = ceil((float) Nparticles / (float) THREADS_PER_BLOCK);
  220. for (x = 0; x < num_blocks; x++) {
  221. sum += partial_sums[x];
  222. }
  223. partial_sums[0] = sum;
  224. }
  225. }
  226. /*****************************
  227. * OpenCL Likelihood Kernel Function to replace FindIndex
  228. * param1: arrayX
  229. * param2: arrayY
  230. * param2.5: CDF
  231. * param3: ind
  232. * param4: objxy
  233. * param5: likelihood
  234. * param6: I
  235. * param6.5: u
  236. * param6.75: weights
  237. * param7: Nparticles
  238. * param8: countOnes
  239. * param9: max_size
  240. * param10: k
  241. * param11: IszY
  242. * param12: Nfr
  243. *****************************/
  244. __kernel void likelihood_kernel(__global float * arrayX, __global float * arrayY,__global float * xj, __global float * yj, __global float * CDF, __global int * ind, __global int * objxy, __global float * likelihood, __global unsigned char * I, __global float * u, __global float * weights, const int Nparticles, const int countOnes, const int max_size, int k, const int IszY, const int Nfr, __global int *seed, __global float * partial_sums, __local float* buffer){
  245. int block_id = get_group_id(0);
  246. int thread_id = get_local_id(0);
  247. int i = get_global_id(0);
  248. size_t THREADS_PER_BLOCK = get_local_size(0);
  249. int y;
  250. int indX, indY;
  251. if(i < Nparticles){
  252. arrayX[i] = xj[i];
  253. arrayY[i] = yj[i];
  254. weights[i] = 1 / ((float) (Nparticles)); //Donnie - moved this line from end of find_index_kernel to prevent all weights from being reset before calculating position on final iteration.
  255. arrayX[i] = arrayX[i] + 1.0 + 5.0*d_randn(seed, i);
  256. arrayY[i] = arrayY[i] - 2.0 + 2.0*d_randn(seed, i);
  257. }
  258. barrier(CLK_GLOBAL_MEM_FENCE);
  259. if(i < Nparticles)
  260. {
  261. for(y = 0; y < countOnes; y++){
  262. indX = dev_round_float(arrayX[i]) + objxy[y*2 + 1];
  263. indY = dev_round_float(arrayY[i]) + objxy[y*2];
  264. ind[i*countOnes + y] = abs(indX*IszY*Nfr + indY*Nfr + k);
  265. if(ind[i*countOnes + y] >= max_size)
  266. ind[i*countOnes + y] = 0;
  267. }
  268. likelihood[i] = calcLikelihoodSum(I, ind, countOnes, i);
  269. likelihood[i] = likelihood[i]/countOnes-SCALE_FACTOR;
  270. weights[i] = weights[i] * exp(likelihood[i]); //Donnie Newell - added the missing exponential function call
  271. }
  272. buffer[thread_id] = 0.0; // DEBUG!!!!!!!!!!!!!!!!!!!!!!!!
  273. //buffer[thread_id] = i;
  274. barrier(CLK_LOCAL_MEM_FENCE | CLK_GLOBAL_MEM_FENCE);
  275. if(i < Nparticles){
  276. buffer[thread_id] = weights[i];
  277. }
  278. barrier(CLK_LOCAL_MEM_FENCE);
  279. /* for some reason the get_local_size(0) call was not returning 512. */
  280. //for(unsigned int s=get_local_size(0)/2; s>0; s>>=1)
  281. for(unsigned int s=THREADS_PER_BLOCK/2; s>0; s>>=1)
  282. {
  283. if(thread_id < s)
  284. {
  285. buffer[thread_id] += buffer[thread_id + s];
  286. }
  287. barrier(CLK_LOCAL_MEM_FENCE);
  288. }
  289. if(thread_id == 0)
  290. {
  291. partial_sums[block_id] = buffer[0];
  292. }
  293. }//*/
  294. //#endif