spacepaste

Paste details

reply | raw

  1.  
  2. #pragma OPENCL EXTENSION cl_khr_fp64 : enable
  4. double2 sample(const double qy, const double qz,
  5. const double r, const double h, const int dwba) {
  6. return (double2)(qy,qz);
  7. }
  9. __kernel void dwba(const int ny, const int nz,
  10. const double r, const double h,
  11. const __global double* qy, const __global double *qz,
  12. __global double* res, const int dwba, const int vector) {
  14. if (vector == 0) {
  15. int iy = get_global_id(0);
  16. int iz = get_global_id(1);
  17. int ind = iz+nz*iy;
  19. if (qz[iz] < 0) {
  20. res[ind] = 0;
  21. return;
  22. }
  24. res[ind] = length(sample(qy[iy], qz[iz], r, h, dwba));
  25. } else {
  26. const double q0 = 2*M_PI/0.139;
  27. const double alpha_i = 0.2*M_PI/180.;
  28. double2 tmp;
  29. int i = get_global_id(0);
  30. res[2*i] = 0;
  31. res[2*i+1] = 1;
  32. if (i < ny) {
  33. if (dwba == 0) {
  34. tmp = sample(qy[i], 0, r, h, dwba);
  35. } else {
  36. tmp = sample(qy[i], -q0*sin(alpha_i), r, h, dwba);
  37. }
  38. res[2*i] = tmp.x;
  39. res[2*i+1] = tmp.y;
  40. }
  41. if (i < nz) {
  42. tmp = sample(0, qz[i], r, h, dwba);
  43. res[2*(i+ny)] = tmp.x;
  44. res[2*(i+ny)+1] = tmp.y;
  45. }
  46. }
  47. }
  48.  
Powered by spacepaste. Paste removal.