//__author__ = "Konstantin Klementiev, Roman Chernikov"
//__date__ = "22 Jan 2016"

#define PRIVATELAYERS 1000
#ifdef cl_khr_fp64
#if __OPENCL_VERSION__<120
#pragma OPENCL EXTENSION cl_khr_fp64: enable
#endif
#endif

#include "xrt_complex.cl"

__constant float PI =    3.141592653589793238;
__constant float ch = 12398.4186;  // {5}   {c*h[eV*A]}
__constant float twoPi = 6.283185307179586476;
__constant float chbar = 1973.269606712496640;  // {c*hbar[eV*A]}
__constant float r0 = 2.817940285e-5;  // A
__constant float avogadro = 6.02214199e23;  // atoms/mol

float get_f0(float qOver4pi, int iN, __global float* f0cfs)
  {
    float res = f0cfs[iN*11+5];
    //printf("c=%g\n",f0cfs[Zi*11+5]);
    for (int i=0;i<5;i++)
    {
            //printf("a[%i]=%g\n",i,f0cfs[Zi*11+i]);
            //printf("b[%i]=%g\n",i,f0cfs[Zi*11+i+6]);
            res += f0cfs[iN*11+i] * exp(-f0cfs[iN*11+i+6] * qOver4pi * qOver4pi);
    }
    mem_fence(CLK_LOCAL_MEM_FENCE);
    return res;
  }


float2 get_f1f2(float E, int iN, int VectorMax,
                 __global float* E_vector,
                 __global float* f1_vector, __global float* f2_vector)
  {
    //printf("VectMax, E, iN  %i %f %i\n",VectorMax,E,iN);
    //printf("Estart, Eend: %f, %f\n",E_vector[iN*300], E_vector[iN*300+VectorMax]);
    int pn = floor((VectorMax-1)*log(E-E_vector[iN*300])/log(E_vector[iN*300+VectorMax]-E_vector[iN*300]));
    //printf("pn %i\n",pn);
    if (E_vector[iN*300+pn]>E)
      {
        while (E_vector[iN*300+pn] > E) pn--;
      }
    else
      {
        while (E_vector[iN*300+pn] < E) pn++;
        pn--;
      }
    mem_fence(CLK_LOCAL_MEM_FENCE);

    float dE = (E - E_vector[iN*300+pn]) / (E_vector[iN*300+pn+1]-E_vector[iN*300+pn]);

    float f1 = f1_vector[iN*300+pn] + dE *(f1_vector[iN*300+pn+1] - f1_vector[iN*300+pn]);
    float f2 = f2_vector[iN*300+pn] + dE *(f2_vector[iN*300+pn+1] - f2_vector[iN*300+pn]);
    mem_fence(CLK_LOCAL_MEM_FENCE);
    return (float2)(f1, f2);
  }

float8 get_structure_factor_fcc(float E, float factDW,
                                  int4 hkl_i, float sinThetaOverLambda,
                                  const int maxEl,  __global float* elements,
                                  __global float* f0cfs, __global float* E_vector,
                                  __global float* f1_vector, __global float* f2_vector)
  {
        float4 hkl;
        hkl.x = (float)hkl_i.x;
        hkl.y = (float)hkl_i.y;
        hkl.z = (float)hkl_i.z;
        hkl.w = 0;
        float2 anomalousPart = get_f1f2(E, 0, round(elements[7]), E_vector, f1_vector, f2_vector);
        float2 F0 = 4 * ((float2)(elements[5],0) + anomalousPart) * factDW;
        //printf("F0_gsf: %g + %gj\n", F0.x, F0.y);
        //printf("anomalous part: %g + %gj\n", anomalousPart.x, anomalousPart.y);
        float residue = dot(fabs(remainder(hkl,2.)),1.);
        //printf("residue: %e", residue);
        float Fcoef=0.;

        if (residue==0 || residue==3) Fcoef = 1.;
        //printf("f0: %g\n", get_f0(sinThetaOverLambda, Z, f0cfs));
        float2 Fhkl = 4 * Fcoef * factDW *
                         (r2cmp(get_f0(sinThetaOverLambda, 0, f0cfs)) + anomalousPart);
        //printf("Fhkl_gsf: %g + %gj\n", Fhkl.x, Fhkl.y);
        //mem_fence(CLK_LOCAL_MEM_FENCE);
        return (float8)(F0, Fhkl, Fhkl, cmp0);
  }

float Si_dl_l(float t)
  {
        float res;
        float t2 = t * t;
        float t3 = t2 * t;
        float t4 = t3 * t;
        if (t >= 0.0 && t < 30.0)
          {
            res = -2.154537e-004;
          }
        else if (t >= 30.0 && t < 130.0)
          {
            res = -2.303956e-014 * t4 + 7.834799e-011 * t3 -
                1.724143e-008 * t2 + 8.396104e-007 * t - 2.276144e-004;
          }
        else if (t >= 130.0 && t < 293.0)
          {
            res = -1.223001e-011 * t3 + 1.532991e-008 * t2 -
                3.263667e-006 * t - 5.217231e-005;
          }
        else if (t >= 293.0 && t <= 1000.0)
          {
            res = -1.161022e-012 * t3 + 3.311476e-009 * t2 +
                1.124129e-006 * t - 5.844535e-004;
          }
        else res = 1.0e+100;
        mem_fence(CLK_LOCAL_MEM_FENCE);
        return res;
  }

float2 get_distance(float8 lattice, float4 hkl)
    {
        float2 res;
        float4 basis = lattice.lo;
        float4 angles = lattice.hi;

        float4 cos_abg = cos(angles);
        float4 sin_abg = sin(angles);

        float V = basis.x * basis.y * basis.z *
            sqrt(1 - pown(cos_abg.x,2) - pown(cos_abg.y,2) - pown(cos_abg.z,2) + 2*cos_abg.x*cos_abg.y*cos_abg.z);
        //mass = 0.;
        //for (i=0;i<nmax;i++)
        //    mass += elements[i].s4 * elements[i].s6;

        //rho = mass / avogadro / V * 1e24;
        float d = V / (basis.x * basis.y * basis.z) /
            sqrt(pown(hkl.x*sin_abg.x/basis.x,2) + pown(hkl.y*sin_abg.y/basis.y,2) + pown(hkl.z*sin_abg.z/basis.z,2) +
                 2*hkl.x*hkl.y * (cos_abg.x*cos_abg.y - cos_abg.z) / (basis.x*basis.y) +
                 2*hkl.x*hkl.z * (cos_abg.x*cos_abg.z - cos_abg.y) / (basis.x*basis.z) +
                 2*hkl.y*hkl.z * (cos_abg.y*cos_abg.z - cos_abg.x) / (basis.y*basis.z));
        float chiToF = -r0 / PI / V;  // minus!
        //printf("V, d: %g %g\n", V, d);
        mem_fence(CLK_LOCAL_MEM_FENCE);
        res = (float2)(d,chiToF);
        return res;
    }

float2 get_distance_Si(float temperature, float4 dhkl)
    {
        //float2 res;
        float aSi = 5.419490 * (Si_dl_l(temperature) - Si_dl_l(273.15) + 1);
        mem_fence(CLK_LOCAL_MEM_FENCE);
        float d = aSi/sqrt(dot(dhkl,dhkl));
        float chiToF = -r0 / PI / pown(aSi,3);
        return (float2)(d,chiToF);
    }


float8 get_structure_factor_diamond(float E, float factDW,
                                  int4 hkl_i, float sinThetaOverLambda,
                                  const int maxEl, __global float* elements,
                                  __global float* f0cfs, __global float* E_vector,
                                  __global float* f1_vector, __global float* f2_vector)
  {
        float8 res;
        float4 hkl;
        hkl.x = (float)hkl_i.x;
        hkl.y = (float)hkl_i.y;
        hkl.z = (float)hkl_i.z;
        hkl.w = 0;
        //float2 im1 = (float2)(0,1);
        float2 diamondToFcc = cmp1 + exp_c(cmpi1 * 0.5 * PI * dot(hkl,1));
        //printf("hkl: %i, %i, %i\n", hkl_i.x,hkl_i.y,hkl_i.z);
        //printf("diamond2fcc %g + %gj\n",diamondToFcc.x, diamondToFcc.y);
        float8 Fd = get_structure_factor_fcc(E, factDW,
                                  hkl_i, sinThetaOverLambda,
                                  maxEl, elements,
                                  f0cfs, E_vector,
                                  f1_vector, f2_vector);
        mem_fence(CLK_LOCAL_MEM_FENCE);
        float2 F0 = (Fd.lo).lo;
        float2 Fhkl = (Fd.lo).hi;
        float2 Fhkl_ = (Fd.hi).lo;
        res =(float8)(F0 * 2,
                         prod_c(Fhkl,diamondToFcc),
                         prod_c(Fhkl_,conj_c(diamondToFcc)),
                         cmp0);
        mem_fence(CLK_LOCAL_MEM_FENCE);
        return res;
  }

float8 get_structure_factor_general(float E, float factDW,
                                    int4 hkl_i, float sinThetaOverLambda,
                                    const int maxEl, __global float* elements,
                                    __global float* f0cfs, __global float* E_vector,
                                    __global float* f1_vector, __global float* f2_vector)
    {
        int i;
        float4 hkl,xyz;
        float2 F0,Fhkl,Fhkl_,anomalousPart,fact,expiHr;
        //float2 zero2 = (float2)(0,0);
        hkl.x = (float)hkl_i.x;
        hkl.y = (float)hkl_i.y;
        hkl.z = (float)hkl_i.z;
        hkl.w = 0;
        F0 = cmp0; Fhkl = cmp0; Fhkl_ = cmp0;
        for (i=0;i<maxEl;i++)
            {
                anomalousPart = get_f1f2(E, i, round(elements[i*8+7]), E_vector, f1_vector, f2_vector);
                mem_fence(CLK_LOCAL_MEM_FENCE);
                F0 += elements[i*8+4] * (r2cmp(elements[i*8+5]) + anomalousPart) * factDW;
                fact = elements[i*8+4] * (r2cmp(get_f0(sinThetaOverLambda, i, f0cfs)) + anomalousPart) * factDW;
                mem_fence(CLK_LOCAL_MEM_FENCE);
                xyz = (float4)(elements[i*8],elements[i*8+1],elements[i*8+2],0);
                expiHr = exp_c((float2)(0,2 * PI * dot(xyz, hkl)));
                Fhkl += prod_c(fact, expiHr);
                Fhkl_ += div_c(fact, expiHr);
                mem_fence(CLK_LOCAL_MEM_FENCE);
            }

        return (float8)(F0, Fhkl, Fhkl_, cmp0);
  }

float8 get_structure_factor_general_E(float E, float factDW,
                                    int4 hkl_i, float sinThetaOverLambda,
                                    const int maxEl, __global float* elements,
                                    __global float* f0cfs, __global float2* f1f2)
    {
        int i;
        float4 hkl,xyz;
        float2 F0,Fhkl,Fhkl_,anomalousPart,fact,expiHr;
        //float2 zero2 = (float2)(0,0);
        hkl.x = (float)hkl_i.x;
        hkl.y = (float)hkl_i.y;
        hkl.z = (float)hkl_i.z;
        hkl.w = 0;
        F0 = cmp0; Fhkl = cmp0; Fhkl_ = cmp0;
        for (i=0;i<maxEl;i++)
            {
                anomalousPart = f1f2[i];
                mem_fence(CLK_LOCAL_MEM_FENCE);
                F0 += elements[i*8+4] * (r2cmp(elements[i*8+5]) + anomalousPart) * factDW;
                fact = elements[i*8+4] * (r2cmp(get_f0(sinThetaOverLambda, i, f0cfs)) + anomalousPart) * factDW;
                mem_fence(CLK_LOCAL_MEM_FENCE);
                xyz = (float4)(elements[i*8],elements[i*8+1],elements[i*8+2],0);
                expiHr = exp_c((float2)(0,2 * PI * dot(xyz, hkl)));
                Fhkl += prod_c(fact, expiHr);
                Fhkl_ += div_c(fact, expiHr);
                mem_fence(CLK_LOCAL_MEM_FENCE);
            }

        return (float8)(F0, Fhkl, Fhkl_, cmp0);
  }



float2 for_one_polarization(float polFactor,
                             float2 chih, float2 chih_, float2 chi0,
                             float k02, float k0s, float kHs,
                             float2 alpha, float b, float thickness, int2 geom)
  {
            float2 im1 = (float2)(0, 1);
            float2 ra, rb;
            float2 delta = sqrt_c(sqr_c(alpha) + polFactor * polFactor / b *
                                    prod_c(chih, chih_));

            float t = thickness * 1.e7;
            float2 L = t * delta * k02 / 2 / kHs;
            //printf("geom.lo, hi: %i, %i\n",geom.lo, geom.hi);
            if (geom.lo == 1) //Bragg
              {
                if (geom.hi == 1) //transmitted
                  {
                    //printf("Bragg transmitted, %i, %i\n", geom.lo, geom.hi);
                    ra = prod_c(rec_c(cos_c(L) - prod_c(prod_c(im1, alpha), div_c(sin_c(L),delta))),
                        exp_c(prod_c(im1,chi0 - alpha * b) * k02 * t / 2 / k0s));
                  }
                else // reflected
                  {
                    if (thickness == 0 || thickness > 0.51) // is None:  # thick Bragg
                      {
                        //printf("am I here?, t=%d\n",thickness);
                        ra = div_c(chih * polFactor, alpha + delta);
                        float2 ad = alpha - delta;
                        if (abs_c(ad) == 0) ad = (float2)(1e-100, 0);
                        rb = div_c(chih * polFactor, ad);

                        if (isnan(abs_c(ra)) || (abs_c(rb) < abs_c(ra))) ra = rb;
                        //return ra / sqrt(fabs(b));
                      }
                    else
                    {
                    //printf("Bragg reflected, %i, %i\n",geom.lo, geom.hi);
                    ra = polFactor * div_c(chih,(alpha + div_c(prod_c(im1,delta),tan_c(L))));
                    }
                  }
              }
            else  // Laue
              {
                if (geom.hi == 1) //transmitted
                  {
                    //printf("Laue transmitted, %i, %i\n",geom.lo, geom.hi);
                    ra = prod_c(cos_c(L) + prod_c(prod_c(im1, alpha), div_c(sin_c(L),delta)),
                          exp_c(prod_c(im1,chi0 - alpha * b) * k02 * t / 2 / k0s));
                  }
                else
                  {
                    //printf("Laue reflected, %i, %i\n",geom.lo, geom.hi);
                    //printf("delta, L, t, k0s, k02: %g+j%g, %g+j%g, %g, %g, %g\n",
                    //       delta.x, delta.y, L.x, L.y, t, k0s, k02);
                    ra = prod_c(div_c(prod_c(chih * polFactor,sin_c(L)), delta),
                        exp_c(prod_c(im1, chi0 - alpha * b) * k02 * t / 2 / k0s));
                  }
              }
            if (geom.hi == 0) ra /= sqrt(fabs(b));
            return ra;
  }


float get_Bragg_angle(float E, float d)
  {
        return asin(ch / (2 * d * E));
  }

float get_dtheta_symmetric_Bragg(float E, float d, int4 hkl,
                               float chiToF, float factDW,
                               const int maxEl, __global float* elements,
                               __global float* f0cfs, __global float* E_vector,
                               __global float* f1_vector, __global float* f2_vector)
  {
        float8 F = get_structure_factor_general(E, factDW, hkl, 0.5 / d,
                                             maxEl, elements, f0cfs, E_vector, f1_vector, f2_vector);
        mem_fence(CLK_LOCAL_MEM_FENCE);
        float2 chi0 = (F.lo).lo * chiToF * pown(ch / E,2);
        return (chi0 / sin(2 * get_Bragg_angle(E, d))).lo;
  }

float get_dtheta_symmetric_Bragg_E(float E, float d, int4 hkl,
                               float chiToF, float factDW,
                               const int maxEl, __global float* elements,
                               __global float* f0cfs, __global float2* f1f2)
  {
        float8 F = get_structure_factor_general_E(E, factDW, hkl, 0.5 / d,
                                             maxEl, elements, f0cfs, f1f2);
        mem_fence(CLK_LOCAL_MEM_FENCE);
        float2 chi0 = (F.lo).lo * chiToF * pown(ch / E,2);
        return (chi0 / sin(2 * get_Bragg_angle(E, d))).lo;
  }

float get_dtheta(float E, float d, int4 hkl,
                   float chiToF, float factDW,
                   const int maxEl, __global float* elements,
                   __global float* f0cfs, __global float* E_vector,
                   __global float* f1_vector, __global float* f2_vector,
                  float alpha, int2 geom)
  {
        float symm_dt = get_dtheta_symmetric_Bragg(E, d,
                                                    hkl, chiToF, factDW,
                                                    maxEl, elements,
                                                    f0cfs, E_vector,
                                                    f1_vector, f2_vector);
        float thetaB = get_Bragg_angle(E, d);
        float geom_factor = geom.lo == 1 ? -1. : 1.;
        float gamma0 = sin(thetaB + alpha);
        float gammah = geom_factor * sin(thetaB - alpha);
        float osqg0 = sqrt(1. - gamma0*gamma0);
        return -(geom_factor*gamma0 - geom_factor*sqrt(gamma0*gamma0 +
                 geom_factor*(gamma0 - gammah) * osqg0 * symm_dt)) / osqg0;
  }


float get_dtheta_E(float E, float d, int4 hkl,
                   float chiToF, float factDW,
                   const int maxEl, __global float* elements,
                   __global float* f0cfs, __global float2* f1f2,
                  float alpha, int2 geom)
  {
        float symm_dt = get_dtheta_symmetric_Bragg_E(E, d,
                                                      hkl, chiToF, factDW,
                                                      maxEl, elements,
                                                      f0cfs, f1f2);
        float thetaB = get_Bragg_angle(E, d);
        float geom_factor = geom.lo == 1 ? -1. : 1.;
        float gamma0 = sin(thetaB + alpha);
        float gammah = geom_factor * sin(thetaB - alpha);
        float osqg0 = sqrt(1. - gamma0*gamma0);
        return -(geom_factor*gamma0 - geom_factor*sqrt(gamma0*gamma0 +
                 geom_factor*(gamma0 - gammah) * osqg0 * symm_dt)) / osqg0;
  }

float4 get_amplitude_material(float E, int kind, float2 refrac_n, float beamInDotNormal,
                                float thickness, int fromVacuum)
{
    float2 n1, n2, n12s, rs, rp, p2, tf;
    float cosAlpha, sinAlpha;
    float2 n1cosAlpha, n2cosAlpha, n1cosBeta, n2cosBeta, cosBeta;
    if ((kind == 2) || (kind == 5))
        return (float4)(cmp1, cmp1);
    //n = self.get_refractive_index(E)
    if (fromVacuum > 0)
    {
        n1 = cmp1;
        n2 = refrac_n;
    }
    else
    {
        n1 = refrac_n;
        n2 = cmp1;
    }
    cosAlpha = fabs(beamInDotNormal);
    sinAlpha = sqrt(1. - beamInDotNormal * beamInDotNormal);
    if (isnan(sinAlpha)) sinAlpha = 0.;
    n1cosAlpha = n1 * cosAlpha;
    n2cosAlpha = n2 * cosAlpha;
    n12s = div_c(n1, n2) * sinAlpha;
    cosBeta = sqrt_c(cmp1 - sqr_c(n12s));
    n2cosBeta = prod_c(n2, cosBeta);
    n1cosBeta = prod_c(n1, cosBeta);
    if (kind < 3)  // reflectivity
    {
        rs = div_c((n1cosAlpha - n2cosBeta), (n1cosAlpha + n2cosBeta));
        rp = div_c((n2cosAlpha - n1cosBeta),
                   (n2cosAlpha + n1cosBeta));
        if (kind == 1)
        {
            p2 = exp_c(E * thickness * 1.e7 / chbar *
                        prod_c(2. * cmpi1, n2cosBeta));
            rs = prod_c(rs, div_c((cmp1 - p2),
                                  (cmp1 - prod_c(sqr_c(rs), p2))));
            rp = prod_c(rp, div_c((cmp1 - p2),
                                  (cmp1 - prod_c(sqr_c(rp), p2))));
        }
    }
    else if (kind > 2)  // transmittivity
    {
        tf = sqrt(
            (prod_c(n2cosBeta, conj_c(n1))).s0 / cosAlpha) / abs_c(n1);
        rs = div_c((2. * tf * n1cosAlpha), (n1cosAlpha + n2cosBeta));
        rp = div_c((2. * tf * n1cosAlpha), (n2cosAlpha + n1cosBeta));
    }
    return (float4)(rs, rp); //, abs(n.imag) * E / chbar * 2e8  # 1/cm
}

float4 get_amplitude_internal(float E, float d, int4 hkl,
                               float chiToF, float factDW,
                               float beamInDotNormal,
                               float beamOutDotNormal,
                               float beamInDotHNormal,
                               float thickness, int2 geom,
                               const int maxEl,
                               __global float* elements,
                               __global float* f0cfs,
                               __global float* E_vector,
                               __global float* f1_vector,
                               __global float* f2_vector)
{
    float waveLength = ch / E;
    float k = twoPi / waveLength;
    //printf("k %g\n", k);
    //printf("bIDN, bODN: %g, %g\n", beamInDotNormal, beamOutDotNormal);
    float k0s = -beamInDotNormal * k;
    float kHs = -beamOutDotNormal * k;

    float b = k0s / kHs;
    float k0H = fabs(beamInDotHNormal) * (twoPi / d) * k;
    float k02 = k * k;
    float H2 = pown(twoPi / d, 2);


    //float8 F = get_structure_factor_diamond(E, factDW, hkl, 0.5 / d,
    //                           maxEl, elements, f0cfs, E_vector, f1_vector, f2_vector);
    float8 F = get_structure_factor_general(E, factDW, hkl, 0.5 / d,
                               maxEl, elements, f0cfs, E_vector, f1_vector, f2_vector);
    mem_fence(CLK_LOCAL_MEM_FENCE);
    float2 F0 = (F.lo).lo;
    float2 Fhkl = (F.lo).hi;
    float2 Fhkl_ = (F.hi).lo;
    float lambdaSquare = pown(waveLength,2);
    float chiToFlambdaSquare = chiToF * lambdaSquare;
    float2 chi0 = conj_c(F0) * chiToFlambdaSquare;
    float2 chih = conj_c(Fhkl) * chiToFlambdaSquare;
    float2 chih_ = conj_c(Fhkl_) * chiToFlambdaSquare;
    //printf("chih, chih_: %g+j%g, %g+j%g\n", chih.x, chih.y, chih_.x, chih_.y);
    float2 alpha = r2cmp((0.5*H2 - k0H) / k02) + chi0 * 0.5 * (1 / b - 1);
    float2 curveS = for_one_polarization(1.,
                                chih, chih_, chi0,
                                k02, k0s, kHs,
                                alpha, b,
                                thickness, geom);  //# s polarization
    mem_fence(CLK_LOCAL_MEM_FENCE);
    float2 curveP = for_one_polarization(cos(2. * get_Bragg_angle(E, d)),
                                chih, chih_, chi0,
                                k02, k0s, kHs,
                                alpha, b,
                                thickness, geom);  //# p polarization
    mem_fence(CLK_LOCAL_MEM_FENCE);
    //printf("AS, AP: %g+j%g, %g+j%g\n", curveS.x, curveS.y, curveP.x, curveP.y);
    return (float4)(curveS, curveP);//  # , phi.real
}

float4 get_amplitude_internal_E(float E, float d, int4 hkl,
                               float chiToF, float factDW,
                               float beamInDotNormal,
                               float beamOutDotNormal,
                               float beamInDotHNormal,
                               float thickness, int2 geom,
                               const int maxEl,
                               __global float* elements,
                               __global float* f0cfs,
                               __global float2* f1f2)
{
    float waveLength = ch / E;
    float k = twoPi / waveLength;
    //printf("k %g\n", k);
    //printf("bIDN, bODN: %g, %g\n", beamInDotNormal, beamOutDotNormal);
    float k0s = -beamInDotNormal * k;
    float kHs = -beamOutDotNormal * k;

    float b = k0s / kHs;
    float k0H = fabs(beamInDotHNormal) * (twoPi / d) * k;
    float k02 = k * k;
    float H2 = pown(twoPi / d, 2);


    //float8 F = get_structure_factor_diamond(E, factDW, hkl, 0.5 / d,
    //                           maxEl, elements, f0cfs, E_vector, f1_vector, f2_vector);
    //float8 F = get_structure_factor_general(E, factDW, hkl, 0.5 / d,
    //                           maxEl, elements, f0cfs, E_vector, f1_vector, f2_vector);
    float8 F = get_structure_factor_general_E(E, factDW, hkl, 0.5 / d,
                               maxEl, elements, f0cfs, f1f2);

    mem_fence(CLK_LOCAL_MEM_FENCE);
    float2 F0 = (F.lo).lo;
    float2 Fhkl = (F.lo).hi;
    float2 Fhkl_ = (F.hi).lo;
    float lambdaSquare = waveLength * waveLength;
    float chiToFlambdaSquare = chiToF * lambdaSquare;
    float2 chi0 = conj_c(F0) * chiToFlambdaSquare;
    float2 chih = conj_c(Fhkl) * chiToFlambdaSquare;
    float2 chih_ = conj_c(Fhkl_) * chiToFlambdaSquare;
    //printf("chih, chih_: %g+j%g, %g+j%g\n", chih.x, chih.y, chih_.x, chih_.y);
    float2 alpha = r2cmp((0.5*H2 - k0H) / k02) + chi0 * 0.5 * (1 / b - 1);
    float2 curveS = for_one_polarization(1.,
                                chih, chih_, chi0,
                                k02, k0s, kHs,
                                alpha, b,
                                thickness, geom);  //# s polarization
    mem_fence(CLK_LOCAL_MEM_FENCE);
    float2 curveP = for_one_polarization(cos(2. * get_Bragg_angle(E, d)),
                                chih, chih_, chi0,
                                k02, k0s, kHs,
                                alpha, b,
                                thickness, geom);  //# p polarization
    mem_fence(CLK_LOCAL_MEM_FENCE);
    //printf("AS, AP: %g+j%g, %g+j%g\n", curveS.x, curveS.y, curveP.x, curveP.y);
    return (float4)(curveS, curveP);//  # , phi.real
}


float4 get_amplitude_multilayer_internal(const int npairs,
                                            float2 rbs_si,
                                            float2 rbs_pi,
                                            float2 rtb_si,
                                            float2 rtb_pi,
                                            float2 rvt_si,
                                            float2 rvt_pi,
                                            float2 p2ti,
                                            float2 p2bi)
{
    int i;
    float2 rij_s, rij_p, p2i, rj_s, rj_p, rj2i, ri_si, ri_pi;
    //float2 rbt_s = -rtb_si;
    //float2 rbt_p = -rtb_pi;
    rj_s = rbs_si;
    rj_p = rbs_pi;
    int lsw = -1;

    for (i=2*npairs-1;i>0;i--)
    {
        lsw = -lsw;
        rij_s = lsw * rtb_si;
        rij_p = lsw * rtb_pi;
        p2i = (lsw + 1) * p2bi / 2. - (lsw - 1) * p2ti / 2.;

        rj2i = prod_c(rj_s, p2i);

        ri_si = div_c((rij_s + rj2i),
                (cmp1 + prod_c(rij_s, rj2i)));
        rj2i = prod_c(rj_p, p2i);
        ri_pi = div_c((rij_p + rj2i),
                (cmp1 + prod_c(rij_p, rj2i)));
        rj_s = ri_si;
        rj_p = ri_pi;
    }

    rij_s = rvt_si;
    rij_p = rvt_pi;

    p2i = p2ti;
    rj2i = prod_c(rj_s,p2i);
    ri_si = div_c((rij_s + rj2i),
                    (cmp1 + prod_c(rij_s, rj2i)));
    rj2i = prod_c(rj_p,p2i);
    ri_pi = div_c((rij_p + rj2i),
                    (cmp1 + prod_c(rij_p, rj2i)));
    //mem_fence(CLK_LOCAL_MEM_FENCE);
    return (float4)(ri_si, ri_pi);
}

float4 get_amplitude_graded_multilayer_internal(const int npairs,
                                            float2 rbs_si,
                                            float2 rbs_pi,
                                            float2 rtb_si,
                                            float2 rtb_pi,
                                            float2 rvt_si,
                                            float2 rvt_pi,
                                            float2 qti,
                                            float2 qbi,
										__global float* dti,
										__global float* dbi)
{
    int i;
    float2 rij_s, rij_p, p2i, rj_s, rj_p, rj2i, ri_si, ri_pi;
	float2 p2bi, p2ti;
    //float2 rbt_s = -rtb_si;
    //float2 rbt_p = -rtb_pi;
    rj_s = rbs_si;
    rj_p = rbs_pi;
    int lsw = -1;
    for (i=2*npairs-1;i>0;i--)
    {
        lsw = -lsw;
        rij_s = lsw * rtb_si;
        rij_p = lsw * rtb_pi;

		if (lsw>0) {
			//p2b
			p2bi = exp_c(prod_c(cmpi1, dbi[i/2] * qbi));
            p2i = p2bi;
		}
		else {//p2t
			p2ti = exp_c(prod_c(cmpi1, dti[i/2] * qti));
            p2i = p2ti;
		}
        //p2i = (lsw + 1) * p2bi / 2. - (lsw - 1) * p2ti / 2.;

        rj2i = prod_c(rj_s, p2i);

        ri_si = div_c((rij_s + rj2i),
                (cmp1 + prod_c(rij_s, rj2i)));
        rj2i = prod_c(rj_p, p2i);
        ri_pi = div_c((rij_p + rj2i),
                (cmp1 + prod_c(rij_p, rj2i)));
        rj_s = ri_si;
        rj_p = ri_pi;
    }

    rij_s = rvt_si;
    rij_p = rvt_pi;
    //printf("dti[0] %g, dbi[0] %g\n",di[0].x, di[0].y);
    p2i = exp_c(prod_c(cmpi1, dti[0] * qti));
    rj2i = prod_c(rj_s,p2i);
    ri_si = div_c((rij_s + rj2i),
                    (cmp1 + prod_c(rij_s, rj2i)));
    rj2i = prod_c(rj_p,p2i);
    ri_pi = div_c((rij_p + rj2i),
                    (cmp1 + prod_c(rij_p, rj2i)));
    mem_fence(CLK_LOCAL_MEM_FENCE);
    return (float4)(ri_si, ri_pi);
}


__kernel void get_amplitude(const int4 hkl,
                            const float factDW,
                            const float thickness,
                            const int geom_b,
                            const int maxEl,
                            __global float* BeamInDotNormal_gl,
                            __global float* BeamOutDotNormal_gl,
                            __global float* BeamInDotHNormal_gl,
                            __global float* Energy_gl,
                            __global float* IPDistance_gl,
                            __global float* ChiToF_gl,
                            __global float* elements,
                            __global float* f0cfs,
                            __global float* E_vector,
                            __global float* f1_vector,
                            __global float* f2_vector,
                            __global float2* refl_s,
                            __global float2* refl_p)
{
    unsigned int ii = get_global_id(0);
    float4 amplitudes;
    int2 geom;
    geom.lo = geom_b > 1 ? 1 : 0;
    geom.hi = (int)(fabs(remainder(geom_b,2.)));
    amplitudes = get_amplitude_internal(Energy_gl[ii], IPDistance_gl[ii], hkl,
                               ChiToF_gl[ii], factDW,
                               BeamInDotNormal_gl[ii], BeamOutDotNormal_gl[ii],
                               BeamInDotHNormal_gl[ii],
                               thickness, geom, maxEl, elements,
                               f0cfs, E_vector,
                               f1_vector, f2_vector);
    refl_s[ii] = amplitudes.lo;
    refl_p[ii] = amplitudes.hi;
    mem_fence(CLK_LOCAL_MEM_FENCE);
}

__kernel void get_amplitude_multilayer(const int npairs,
                            __global float2* rbs_s,
                            __global float2* rbs_p,
                            __global float2* rtb_s,
                            __global float2* rtb_p,
                            __global float2* rvt_s,
                            __global float2* rvt_p,
                            __global float2* p2t,
                            __global float2* p2b,
                            __global float2* ri_s,
                            __global float2* ri_p)
{
    unsigned int ii = get_global_id(0);
    float4 amplitudes;
    amplitudes = get_amplitude_multilayer_internal(npairs,
                    rbs_s[ii], rbs_p[ii], rtb_s[ii], rtb_p[ii],
                    rvt_s[ii], rvt_p[ii], p2t[ii], p2b[ii]);
    mem_fence(CLK_LOCAL_MEM_FENCE);
    ri_s[ii] = amplitudes.s01;
    ri_p[ii] = amplitudes.s23;
    mem_fence(CLK_LOCAL_MEM_FENCE);
}

__kernel void get_amplitude_graded_multilayer(const int npairs,
                            __global float2* rbs_s,
                            __global float2* rbs_p,
                            __global float2* rtb_s,
                            __global float2* rtb_p,
                            __global float2* rvt_s,
                            __global float2* rvt_p,
                            __global float2* qt_glo,
                            __global float2* qb_glo,
                            __global float* dti,
                            __global float* dbi,
                            __global float2* ri_s,
                            __global float2* ri_p)
{
    unsigned int ii = get_global_id(0);
    float4 amplitudes;
    amplitudes = get_amplitude_graded_multilayer_internal(npairs,
                    rbs_s[ii], rbs_p[ii], rtb_s[ii], rtb_p[ii],
                    rvt_s[ii], rvt_p[ii], qt_glo[ii], qb_glo[ii], dti, dbi);
    mem_fence(CLK_LOCAL_MEM_FENCE);
    ri_s[ii] = amplitudes.s01;
    ri_p[ii] = amplitudes.s23;
    mem_fence(CLK_LOCAL_MEM_FENCE);
}

__kernel void tt_laue_spherical_single(
                  const int nLayers,
                  const double2 Wgrad,
                  __global double2 *A,
                  __global double2 *B,
                  __global double2 *W,
                  __global double2 *V,
                  __global double2 *D0t,
                  __global double2 *Dht,
                  __global double2 *D0,
                  __global double2 *Dh
                  )
{
      double2 C1, C2, C3, C4;
      double8 D0_line, Dh_line, Mvec;
      unsigned int ii = get_global_id(0);
      unsigned int iloc = ii*(nLayers+3);
      int i, j;
      C1 = (double2)(1,0) - W[ii];
      C2 = (double2)(1,0) - V[ii];
      C3 = (double2)(1,0) + W[ii];
      C4 = (double2)(1,0) + V[ii];

      double2 AB = prod_c(A[ii], B[ii]);

      D0_line = (double8)(prod_c(C4, C1),
                         prod_c(A[ii], C1),
                         AB,
                         prod_c(A[ii], C3));
      Dh_line = (double8)(prod_c(B[ii], C4),
                         AB,
                         prod_c(B[ii], C2),
                         prod_c(C3, C2));

      D0t[iloc] = (double2)(0.0, 0.0);
      Dht[iloc] = (double2)(0.0, 0.0);
      D0t[iloc+1] = (double2)(1.0, 0.0);
      Dht[iloc+1] = (double2)(0.0, 0.0);

      double2 EEr = rec_c(prod_c(C1, C2) - AB);

      for (i=0; i<nLayers-1; i++)
        {
          // i -- ordinary number of the layer
          D0t[iloc+i+3] = (double2)(0.0, 0.0);
          Dht[iloc+i+3] = (double2)(0.0, 0.0);

          for (j=0; j<i+2; j++)
            {
              // j -- ordinary number of the element in the layer
              Mvec = (double8)(D0t[iloc+j+1],
                               Dht[iloc+j+1],
                               D0t[iloc+j],
                               Dht[iloc+j]);
              D0t[iloc+j+1] = prod_c(EEr, dot_c4(D0_line, Mvec));
              Dht[iloc+j+1] = prod_c(EEr, dot_c4(Dh_line, Mvec));
              //printf("Ray %i, Layer %i, Element %i\n", ii, i, j);
            }
//          D0[ii*(nLayers+1)+i] = D0t[iloc+i+1];
//          Dh[ii*(nLayers+1)+i] = Dht[iloc+i+1];
        }
      //if (ii==0) printf("Calculation completed\n");
      mem_fence(CLK_LOCAL_MEM_FENCE);
      for (i=0; i<(nLayers+1); i++)
        {
          D0[ii*(nLayers+1)+i] = D0t[iloc+i+1];
          Dh[ii*(nLayers+1)+i] = Dht[iloc+i+1];
        }
//    D0[ii] = D0t[iloc+nLayers-1];
//    Dh[ii] = Dht[iloc+nLayers-1];
}

__kernel void tt_laue_plain(
                  const int nLayers,
                  const double2 Wgrad,
                  __global double2 *A,
                  __global double2 *B,
                  __global double2 *W,
                  __global double2 *V,
                  __global double2 *D0,
                  __global double2 *Dh
                  )
{
      double2 C1, C2, C3, C4, D0t, Dht;
      double8 D0_line, Dh_line, Mvec;
      unsigned int ii = get_global_id(0);

      int i;
      C1 = (double2)(1.0, 0.0) - W[ii];
      C2 = (double2)(1.0, 0.0) - V[ii];
      C3 = (double2)(1.0, 0.0) + W[ii];
      C4 = (double2)(1.0, 0.0) + V[ii];

      double2 AB = prod_c(A[ii], B[ii]);

      D0_line = (double8)(prod_c(C4, C1),
                          prod_c(A[ii], C1),
                          AB,
                          prod_c(A[ii], C3));
      Dh_line = (double8)(prod_c(B[ii], C4),
                          AB,
                          prod_c(B[ii], C2),
                          prod_c(C3, C2));

      D0t = (double2)(1.0, 0.0);
      Dht = (double2)(0.0, 0.0);

      double2 EEr = rec_c(prod_c(C1, C2) - AB);
      for (i=0; i<nLayers; i++) {
          Mvec = (double8)(D0t,
                           Dht,
                           D0t,
                           Dht);
          D0t = prod_c(EEr, dot_c4(D0_line, Mvec));
          Dht = prod_c(EEr, dot_c4(Dh_line, Mvec));
        }

      mem_fence(CLK_LOCAL_MEM_FENCE);
      D0[ii] = D0t;
      Dh[ii] = Dht;
}

__kernel void tt_laue_plain_bent(
                  const int nOfLayers,
//                  const double2 Wgrad,
                  __global double2 *Wgrad_global,
                  __global double2 *A,
                  __global double2 *B,
                  __global double2 *W,
                  __global double2 *V,
                  __global double2 *D0,
                  __global double2 *Dh
                  )
{

    double2 C1, C2, C3, C4, D0t, Dht, Aloc, Bloc, Wloc, Vloc, Wi, Wgrad;
    double8 D0_line, Dh_line, Mvec;
    double2 cOne = (double2)(1.0, 0.0);
    unsigned int ii = get_global_id(0);
    int i;
    
    Aloc = A[ii]; Bloc = B[ii]; Wloc = W[ii];
    Vloc = V[ii];
    Wgrad = Wgrad_global[ii];
    double2 AB = prod_c(Aloc, Bloc);
    C2 = cOne - Vloc;
    C4 = cOne + Vloc;

    double2 BC2 = prod_c(Bloc, C2);
    double2 BC4 = prod_c(Bloc, C4);

    D0t = cOne;
    Dht = (double2)(0.0, 0.0);

      for (i=0; i<nOfLayers; i++) {
          Wi = (i == 0) ? Wloc : Wloc + Wgrad*((double)i - 0.5);
          C1 = cOne - Wi;
          C3 = cOne + Wi;

          D0_line = (double8)(prod_c(C4, C1),
                              prod_c(Aloc, C1),
                              AB,
                              prod_c(Aloc, C3));
          Dh_line = (double8)(BC4,
                              AB,
                              BC2,
                              prod_c(C3, C2));

          double2 EEr = rec_c(prod_c(C1, C2) - AB);
          Mvec = (double8)(D0t,
                           Dht,
                           D0t,
                           Dht);
          D0t = prod_c(EEr, dot_c4(D0_line, Mvec));
          Dht = prod_c(EEr, dot_c4(Dh_line, Mvec));
        }

      mem_fence(CLK_LOCAL_MEM_FENCE);
      D0[ii] = D0t;
      Dh[ii] = Dht;
}

/*
__kernel void tt_bragg_plain_single(
                  const int nLayers,
                  const double2 Wgrad,
                  __global double2 *A,
                  __global double2 *B,
                  __global double2 *W,
                  __global double2 *V,
                  __global double2 *D0,
                  __global double2 *Dh
                  )
{
      double2 C1, C2, C3, C4, D0t, Dht;
      double8 D0_line, Dh_line, Mvec;
      unsigned int ii = get_global_id(0);

      int i;
      C1 = (double2)(1.0, 0.0) - W[ii];
      C2 = (double2)(1.0, 0.0) - V[ii];
      C3 = (double2)(1.0, 0.0) + W[ii];
      C4 = (double2)(1.0, 0.0) + V[ii];

      double2 AB = prod_c(A[ii], B[ii]);

      D0_line = (double8)(prod_c(C4, C1),
                          prod_c(A[ii], C1),
                          AB,
                          prod_c(A[ii], C3));
      Dh_line = (double8)(prod_c(B[ii], C4),
                          AB,
                          prod_c(B[ii], C2),
                          prod_c(C3, C2));

      D0t = (double2)(1.0, 0.0);
      Dht = (double2)(0.0, 0.0);

      double2 EEr = rec_c(prod_c(C1, C2) - AB);
      for (i=0; i<nLayers; i++) {
          Mvec = (double8)(D0t,
                           Dht,
                           D0t,
                           Dht);
          D0t = prod_c(EEr, dot_c4(D0_line, Mvec));
          Dht = prod_c(EEr, dot_c4(Dh_line, Mvec));
        }

      mem_fence(CLK_LOCAL_MEM_FENCE);
      D0[ii] = D0t;
      Dh[ii] = Dht;
}
*/
__kernel void tt_bragg_plain(
                  const int nLayers,
                  const double2 Wgrad,
                  __global double2 *A,
                  __global double2 *B,
                  __global double2 *W,
                  __global double2 *V,
                  __global double2 *D0t,
                  __global double2 *Dht,
                  __global double2 *D0,
                  __global double2 *Dh
//                  __global double2 *Dh_pic
                  )
{
    int arrSize = PRIVATELAYERS;
    __private double2 D0Arr[PRIVATELAYERS+1];
    __private double2 DhArr[PRIVATELAYERS+1];
    double2 C1,C2,C3,C4;
    double8 D0_line, Dh_line, Mvec;
    int diff, stripeWidth, stripeHeight;

    unsigned int ii = get_global_id(0);
    int i, j, n;
    unsigned int cnt_loc = ii * (nLayers+1);
    C1 = (double2)(1,0) - W[ii];
    C2 = (double2)(1,0) - V[ii];
    C3 = (double2)(1,0) + W[ii];
    C4 = (double2)(1,0) + V[ii];

    double2 AB = prod_c(A[ii], B[ii]);
    D0_line = (double8)(prod_c(C4, C1),
                        prod_c(A[ii], C1),
                        AB,
                        prod_c(A[ii], C3));
    Dh_line = (double8)(prod_c(B[ii], C4),
                        AB,
                        prod_c(B[ii], C2),
                        prod_c(C3, C2));
    double2 EEr = rec_c(prod_c(C1, C2) - AB);

    D0t[cnt_loc] = (double2)(0.0, 0.0);
    Dht[cnt_loc] = (double2)(0.0, 0.0);

    int nStripes = ceil((float)nLayers/(float)arrSize);

    for (n=0; n<nStripes; n++) {
        diff = nLayers - n*arrSize;
        stripeWidth = min(arrSize, diff);
        stripeHeight = n*arrSize;
        for (j=1; j<stripeWidth+1; j++) {
            D0Arr[j] = (double2)(0.0, 0.0);
            DhArr[j] = (double2)(0.0, 0.0);
            }
        //Parallel propagation
        for (i=0; i<stripeHeight; i++) {

            D0Arr[0] = D0t[cnt_loc+i];
            DhArr[0] = Dht[cnt_loc+i];
            for (j=0; j<stripeWidth; j++) {
                Mvec = (double8)(D0Arr[j],
                                 DhArr[j],
                                 D0Arr[j+1],
                                 DhArr[j+1]);
                D0Arr[j+1] = prod_c(EEr, dot_c4(D0_line, Mvec));
                DhArr[j+1] = prod_c(EEr, dot_c4(Dh_line, Mvec));
                }
            D0t[cnt_loc+i] = D0Arr[stripeWidth];
            Dht[cnt_loc+i] = DhArr[stripeWidth];
            }
        //Triangular propagation
        for (i=0; i<stripeWidth; i++) {
            D0Arr[i] = (double2)(1.0, 0.0);
            DhArr[i] = (double2)(0.0, 0.0);
            for (j=i; j<stripeWidth; j++) {
                Mvec = (double8)(D0Arr[j],
                                 DhArr[j],
                                 D0Arr[j+1],
                                 DhArr[j+1]);
                D0Arr[j+1] = prod_c(EEr, dot_c4(D0_line, Mvec));
                DhArr[j+1] = prod_c(EEr, dot_c4(Dh_line, Mvec));
                }
            D0t[cnt_loc+stripeHeight+i] = D0Arr[stripeWidth];
            Dht[cnt_loc+stripeHeight+i] = DhArr[stripeWidth];
            }
    }
    mem_fence(CLK_LOCAL_MEM_FENCE);
//    if (ii==0) printf("Finally writing Dh=%v2g from Dht[%i]\n", Dht[cnt_loc+nLayers-1], cnt_loc+nLayers-1);
    Dh[ii] = Dht[cnt_loc+nLayers-1];
    D0[ii] = D0t[cnt_loc+nLayers-1];
}

__kernel void tt_bragg_plain_bent(
                  const int nLayers,
                  __global double2 *Wgrad_global,
                  __global double2 *A,
                  __global double2 *B,
                  __global double2 *W,
                  __global double2 *V,
                  __global double2 *D0t,
                  __global double2 *Dht,
                  __global double2 *D0,
                  __global double2 *Dh
                  )
{
    int arrSize = PRIVATELAYERS;
    __private double2 D0Arr[PRIVATELAYERS+1];
    __private double2 DhArr[PRIVATELAYERS+1];
    double2 C1, C2, C3, C4, Aloc, Bloc, Wloc, Vloc, Wi, Wgrad;
    double8 D0_line, Dh_line, Mvec;
    double2 cOne = (double2)(1.0, 0.0);
    int diff, stripeWidth, stripeHeight;

    unsigned int ii = get_global_id(0);
    int i, j, n, iLayer;
    unsigned int cnt_loc = ii * (nLayers+1);

    Aloc = A[ii]; Bloc = B[ii]; Wloc = W[ii];
    Vloc = V[ii];
    Wgrad = Wgrad_global[ii];
    double2 AB = prod_c(Aloc, Bloc);
    C2 = cOne - Vloc;
    C4 = cOne + Vloc;

    double2 BC2 = prod_c(Bloc, C2);
    double2 BC4 = prod_c(Bloc, C4);

    D0t[cnt_loc] = (double2)(0.0, 0.0);
    Dht[cnt_loc] = (double2)(0.0, 0.0);

    int nStripes = ceil((float)nLayers/(float)arrSize);

    for (n=0; n<nStripes; n++) {
        diff = nLayers - n*arrSize;
        stripeWidth = min(arrSize, diff);
        stripeHeight = n*arrSize;
        for (j=1; j<stripeWidth+1; j++) {
            D0Arr[j] = (double2)(0.0, 0.0);
            DhArr[j] = (double2)(0.0, 0.0);
            }
        //Parallel propagation
        for (i=0; i<stripeHeight; i++) {

            D0Arr[0] = D0t[cnt_loc+i];
            DhArr[0] = Dht[cnt_loc+i];
            for (j=0; j<stripeWidth; j++) {

                iLayer = n*arrSize + i + j + 2;
//                if (ii==0) printf("Parallel: n, i, j, iLayer: %i, %i, %i, %i\n", n, i, j, iLayer);
                Wi = (iLayer == 0) ? Wloc : Wloc + Wgrad * fabs((double)(iLayer)  - 0.5) * 0.5;
                C1 = cOne - Wi;
                C3 = cOne + Wi;

                D0_line = (double8)(prod_c(C4, C1),
                                    prod_c(Aloc, C1),
                                    AB,
                                    prod_c(Aloc, C3));
                Dh_line = (double8)(BC4,
                                    AB,
                                    BC2,
                                    prod_c(C3, C2));

                double2 EEr = rec_c(prod_c(C1, C2) - AB);

                Mvec = (double8)(D0Arr[j],
                                 DhArr[j],
                                 D0Arr[j+1],
                                 DhArr[j+1]);
                D0Arr[j+1] = prod_c(EEr, dot_c4(D0_line, Mvec));
                DhArr[j+1] = prod_c(EEr, dot_c4(Dh_line, Mvec));
                }
            D0t[cnt_loc+i] = D0Arr[stripeWidth];
            Dht[cnt_loc+i] = DhArr[stripeWidth];
            }
        //Triangular propagation
        for (i=0; i<stripeWidth; i++) {

            D0Arr[i] = (double2)(1.0, 0.0);
            DhArr[i] = (double2)(0.0, 0.0);
            for (j=i; j<stripeWidth; j++) {

                iLayer = 2*n*(arrSize) + i + j + 2;
//                if (ii==0) printf("Triangular: n, i, j, iLayer: %i, %i, %i, %i\n", n, i, j, iLayer);
                Wi = (iLayer == 0) ? Wloc : Wloc + Wgrad * fabs((double)(iLayer)  - 0.5) * 0.5;
                C1 = cOne - Wi;
                C3 = cOne + Wi;

                D0_line = (double8)(prod_c(C4, C1),
                                    prod_c(Aloc, C1),
                                    AB,
                                    prod_c(Aloc, C3));
                Dh_line = (double8)(BC4,
                                    AB,
                                    BC2,
                                    prod_c(C3, C2));

                double2 EEr = rec_c(prod_c(C1, C2) - AB);

                Mvec = (double8)(D0Arr[j],
                                 DhArr[j],
                                 D0Arr[j+1],
                                 DhArr[j+1]);
                D0Arr[j+1] = prod_c(EEr, dot_c4(D0_line, Mvec));
                DhArr[j+1] = prod_c(EEr, dot_c4(Dh_line, Mvec));
                }
            D0t[cnt_loc+stripeHeight+i] = D0Arr[stripeWidth];
            Dht[cnt_loc+stripeHeight+i] = DhArr[stripeWidth];
            }
    }
    mem_fence(CLK_LOCAL_MEM_FENCE);
    Dh[ii] = Dht[cnt_loc+nLayers-1];
    D0[ii] = D0t[cnt_loc+nLayers-1];
}


__kernel void tt_bragg_plain_thickness_bent(
                  const int nLayers,
                  __global double2 *Wgrad_global,
                  __global double2 *A,
                  __global double2 *B,
                  __global double2 *W,
                  __global double2 *V,
                  __global double2 *D0t,
                  __global double2 *Dht,
                  __global double2 *D0,
                  __global double2 *Dh
                  )
{
    int arrSize = PRIVATELAYERS;
    __private double2 D0Arr[PRIVATELAYERS+1];
    __private double2 DhArr[PRIVATELAYERS+1];
    double2 C1, C2, C3, C4, Aloc, Bloc, Wloc, Vloc, Wi, Wgrad;
    double8 D0_line, Dh_line, Mvec;
    double2 cOne = (double2)(1.0, 0.0);
    int diff, stripeWidth, stripeHeight;

    unsigned int ii = get_global_id(0);
    int i, j, n, iLayer;
    unsigned int cnt_loc = ii * (nLayers+1);

    Aloc = A[ii]; Bloc = B[ii]; Wloc = W[ii];
    Vloc = V[ii];
    Wgrad = Wgrad_global[ii];
    double2 AB = prod_c(Aloc, Bloc);
    C2 = cOne - Vloc;
    C4 = cOne + Vloc;

    double2 BC2 = prod_c(Bloc, C2);
    double2 BC4 = prod_c(Bloc, C4);

    D0t[cnt_loc] = (double2)(0.0, 0.0);
    Dht[cnt_loc] = (double2)(0.0, 0.0);

    int nStripes = ceil((float)nLayers/(float)arrSize);

    for (n=0; n<nStripes; n++) {
        diff = nLayers - n*arrSize;
        stripeWidth = min(arrSize, diff);
        stripeHeight = n*arrSize;
        for (j=1; j<stripeWidth+1; j++) {
            D0Arr[j] = (double2)(0.0, 0.0);
            DhArr[j] = (double2)(0.0, 0.0);
            }
        //Parallel propagation
        for (i=0; i<stripeHeight; i++) {

            D0Arr[0] = D0t[cnt_loc+i];
            DhArr[0] = Dht[cnt_loc+i];
            for (j=0; j<stripeWidth; j++) {

                iLayer = n*arrSize + i + j + 2;
//                if (ii==0) printf("Parallel: n, i, j, iLayer: %i, %i, %i, %i\n", n, i, j, iLayer);
                Wi = (iLayer == 0) ? Wloc : Wloc + Wgrad * fabs((double)(iLayer)  - 0.5) * 0.5;
                C1 = cOne - Wi;
                C3 = cOne + Wi;

                D0_line = (double8)(prod_c(C4, C1),
                                    prod_c(Aloc, C1),
                                    AB,
                                    prod_c(Aloc, C3));
                Dh_line = (double8)(BC4,
                                    AB,
                                    BC2,
                                    prod_c(C3, C2));

                double2 EEr = rec_c(prod_c(C1, C2) - AB);

                Mvec = (double8)(D0Arr[j],
                                 DhArr[j],
                                 D0Arr[j+1],
                                 DhArr[j+1]);
                D0Arr[j+1] = prod_c(EEr, dot_c4(D0_line, Mvec));
                DhArr[j+1] = prod_c(EEr, dot_c4(Dh_line, Mvec));
                }
            D0t[cnt_loc+i] = D0Arr[stripeWidth];
            Dht[cnt_loc+i] = DhArr[stripeWidth];
            }
        //Triangular propagation
        for (i=0; i<stripeWidth; i++) {

            D0Arr[i] = (double2)(1.0, 0.0);
            DhArr[i] = (double2)(0.0, 0.0);
            for (j=i; j<stripeWidth; j++) {

                iLayer = 2*n*(arrSize) + i + j + 2;
//                if (ii==0) printf("Triangular: n, i, j, iLayer: %i, %i, %i, %i\n", n, i, j, iLayer);
                Wi = (iLayer == 0) ? Wloc : Wloc + Wgrad * fabs((double)(iLayer)  - 0.5) * 0.5;
                C1 = cOne - Wi;
                C3 = cOne + Wi;

                D0_line = (double8)(prod_c(C4, C1),
                                    prod_c(Aloc, C1),
                                    AB,
                                    prod_c(Aloc, C3));
                Dh_line = (double8)(BC4,
                                    AB,
                                    BC2,
                                    prod_c(C3, C2));

                double2 EEr = rec_c(prod_c(C1, C2) - AB);

                Mvec = (double8)(D0Arr[j],
                                 DhArr[j],
                                 D0Arr[j+1],
                                 DhArr[j+1]);
                D0Arr[j+1] = prod_c(EEr, dot_c4(D0_line, Mvec));
                DhArr[j+1] = prod_c(EEr, dot_c4(Dh_line, Mvec));
                }
            D0t[cnt_loc+stripeHeight+i] = D0Arr[stripeWidth];
            Dht[cnt_loc+stripeHeight+i] = DhArr[stripeWidth];
            }
    }
    mem_fence(CLK_LOCAL_MEM_FENCE);
    Dh[ii] = Dht[cnt_loc+nLayers-1];
    D0[ii] = D0t[cnt_loc+nLayers-1];
}

__kernel void tt_laue_spherical(
                  const int nLayers,
                  const double2 Wgrad,
                  __global double2 *A,
                  __global double2 *B,
                  __global double2 *W,
                  __global double2 *V,
                  __global double2 *D0t,
                  __global double2 *Dht,
                  __global double2 *D0,
                  __global double2 *Dh
                  )
{
    int arrSize = PRIVATELAYERS;
    __private double2 D0Arr[PRIVATELAYERS+1];
    __private double2 DhArr[PRIVATELAYERS+1];
    double2 C1,C2,C3,C4;
    double8 D0_line, Dh_line, Mvec;
    int diff, stripeWidth, stripeHeight;

    unsigned int ii = get_global_id(0);
    int i, j, n;
    unsigned int cnt_loc = ii * (nLayers+1);
    C1 = (double2)(1,0) - W[ii];
    C2 = (double2)(1,0) - V[ii];
    C3 = (double2)(1,0) + W[ii];
    C4 = (double2)(1,0) + V[ii];

    double2 AB = prod_c(A[ii], B[ii]);
    D0_line = (double8)(prod_c(C4, C1),
                        prod_c(A[ii], C1),
                        AB,
                        prod_c(A[ii], C3));
    Dh_line = (double8)(prod_c(B[ii], C4),
                        AB,
                        prod_c(B[ii], C2),
                        prod_c(C3, C2));
    double2 EEr = rec_c(prod_c(C1, C2) - AB);

    D0t[cnt_loc] = (double2)(0.0, 0.0);
    Dht[cnt_loc] = (double2)(0.0, 0.0);

    int nStripes = ceil((float)nLayers/(float)(arrSize-1));

    for (n=0; n<nStripes; n++) {
        diff = nLayers - n*(arrSize-1) + 2;
        stripeWidth = min(arrSize, diff);
        stripeHeight = max(nLayers + 1 - (n+1)*(arrSize-1), 0);

        for (j=0; j<stripeWidth; j++) {
            D0Arr[j] = (double2)(0.0, 0.0);
            DhArr[j] = (double2)(0.0, 0.0);
        }
        //Parallel propagation
        for (i=0; i<stripeHeight; i++) {
            D0Arr[0] = D0t[cnt_loc+i];
            DhArr[0] = Dht[cnt_loc+i];
            for (j=1; j<stripeWidth; j++) {
                if ((n==0)&&(i==0)) {
                    D0Arr[1] = (double2)(1.0, 0.0);
                    DhArr[1] = (double2)(0.0, 0.0);
                } else {
                    Mvec = (double8)(D0Arr[j],
                                     DhArr[j],
                                     D0Arr[j-1],
                                     DhArr[j-1]);
                    D0Arr[j] = prod_c(EEr, dot_c4(D0_line, Mvec));
                    DhArr[j] = prod_c(EEr, dot_c4(Dh_line, Mvec));
                }
            }
            D0t[cnt_loc+i] = D0Arr[stripeWidth-1];
            Dht[cnt_loc+i] = DhArr[stripeWidth-1];
        }
        //Triangular propagation
       for (i=0; i<stripeWidth-1; i++) {
            D0Arr[0] = D0t[cnt_loc+i+stripeHeight];
            DhArr[0] = Dht[cnt_loc+i+stripeHeight];
            for (j=1; j<stripeWidth-i; j++) {
                Mvec = (double8)(D0Arr[j],
                                 DhArr[j],
                                 D0Arr[j-1],
                                 DhArr[j-1]);
                D0Arr[j] = prod_c(EEr, dot_c4(D0_line, Mvec));
                DhArr[j] = prod_c(EEr, dot_c4(Dh_line, Mvec));
            }
            D0[ii*(nLayers+1) + (nLayers-stripeHeight) - i] = D0Arr[stripeWidth-i-1];
            Dh[ii*(nLayers+1) + (nLayers-stripeHeight) - i] = DhArr[stripeWidth-i-1];
        }
    }
    mem_fence(CLK_LOCAL_MEM_FENCE);
}

__kernel void tt_laue_spherical_bent(
                  const int nLayers,
//                  const double2 Wgrad,
                  __global double2 *Wgrad_global,
                  __global double2 *A,
                  __global double2 *B,
                  __global double2 *W,
                  __global double2 *V,
                  __global double2 *D0t,
                  __global double2 *Dht,
                  __global double2 *D0,
                  __global double2 *Dh
                  )
{
    int arrSize = PRIVATELAYERS;
    __private double2 D0Arr[PRIVATELAYERS+1];
    __private double2 DhArr[PRIVATELAYERS+1];
    double2 C1, C2, C3, C4, Aloc, Bloc, Wloc, Vloc, Wi, Wgrad;
    double8 D0_line, Dh_line, Mvec;
    double2 cOne = (double2)(1.0, 0.0);
    int diff, stripeWidth, stripeHeight;

    unsigned int ii = get_global_id(0);
    int i, j, n, iLayer;
    unsigned int cnt_loc = ii * (nLayers+1);

    Aloc = A[ii]; Bloc = B[ii]; Wloc = W[ii];
    Vloc = V[ii];
    Wgrad = Wgrad_global[ii];
    double2 AB = prod_c(Aloc, Bloc);
    C2 = cOne - Vloc;
    C4 = cOne + Vloc;

    double2 BC2 = prod_c(Bloc, C2);
    double2 BC4 = prod_c(Bloc, C4);

    D0t[cnt_loc] = (double2)(0.0, 0.0);
    Dht[cnt_loc] = (double2)(0.0, 0.0);

    int nStripes = ceil((float)nLayers/(float)(arrSize-1));

    for (n=0; n<nStripes; n++) {
        diff = nLayers - n*(arrSize-1) + 2;
        stripeWidth = min(arrSize, diff);
        stripeHeight = max(nLayers + 1 - (n+1)*(arrSize-1), 0);

        for (j=0; j<stripeWidth; j++) {
            D0Arr[j] = (double2)(0.0, 0.0);
            DhArr[j] = (double2)(0.0, 0.0);
        }
        //Parallel propagation
        for (i=0; i<stripeHeight; i++) {
            D0Arr[0] = D0t[cnt_loc+i];
            DhArr[0] = Dht[cnt_loc+i];
            for (j=1; j<stripeWidth; j++) {
                iLayer = n*(arrSize-1) + i + j - 1;
                Wi = (iLayer == 0) ? Wloc : Wloc + Wgrad*((double)iLayer - 0.5);
                C1 = cOne - Wi;
                C3 = cOne + Wi;

                D0_line = (double8)(prod_c(C4, C1),
                                    prod_c(Aloc, C1),
                                    AB,
                                    prod_c(Aloc, C3));
                Dh_line = (double8)(BC4,
                                    AB,
                                    BC2,
                                    prod_c(C3, C2));

                double2 EEr = rec_c(prod_c(C1, C2) - AB);

                if ((n==0)&&(i==0)) {
                    D0Arr[1] = (double2)(1.0, 0.0);
                    DhArr[1] = (double2)(0.0, 0.0);
                } else {
                    Mvec = (double8)(D0Arr[j],
                                     DhArr[j],
                                     D0Arr[j-1],
                                     DhArr[j-1]);
                    D0Arr[j] = prod_c(EEr, dot_c4(D0_line, Mvec));
                    DhArr[j] = prod_c(EEr, dot_c4(Dh_line, Mvec));
                }
            }
            D0t[cnt_loc+i] = D0Arr[stripeWidth-1];
            Dht[cnt_loc+i] = DhArr[stripeWidth-1];
        }
        //Triangular propagation

       for (i=0; i<stripeWidth-1; i++) {
            D0Arr[0] = D0t[cnt_loc+i+stripeHeight];
            DhArr[0] = Dht[cnt_loc+i+stripeHeight];
            for (j=1; j<stripeWidth-i; j++) {
                iLayer = n*(arrSize-1) + i + j + stripeHeight - 1;
                Wi = (iLayer == 0) ? Wloc : Wloc + Wgrad*((double)iLayer - 0.5);
                C1 = cOne - Wi;
                C3 = cOne + Wi;

                D0_line = (double8)(prod_c(C4, C1),
                                    prod_c(Aloc, C1),
                                    AB,
                                    prod_c(Aloc, C3));
                Dh_line = (double8)(BC4,
                                    AB,
                                    BC2,
                                    prod_c(C3, C2));

                double2 EEr = rec_c(prod_c(C1, C2) - AB);
                Mvec = (double8)(D0Arr[j],
                                 DhArr[j],
                                 D0Arr[j-1],
                                 DhArr[j-1]);
                D0Arr[j] = prod_c(EEr, dot_c4(D0_line, Mvec));
                DhArr[j] = prod_c(EEr, dot_c4(Dh_line, Mvec));
            }
            D0[ii*(nLayers+1) + (nLayers-stripeHeight) - i] = D0Arr[stripeWidth-i-1];
            Dh[ii*(nLayers+1) + (nLayers-stripeHeight) - i] = DhArr[stripeWidth-i-1];
        }
    }
    mem_fence(CLK_LOCAL_MEM_FENCE);
}
