#include <CudaMatrixPrivate.cuh>
#include <math.h>
#include <thrust/transform.h>
#include <thrust/functional.h>
#include <thrust/execution_policy.h>
using namespace thrust::placeholders;

struct PowOperator: public thrust::unary_function<float, float>{
    float exponent;
    PowOperator(float v):exponent(v) {}
    void setExponent(float v){
        exponent = v;
    }
    
    __host__ __device__
    float operator()(const float& x)  {
        return powf(x, exponent);
    }
};

void unaryAdd(float* in1, float* in2, float* out, unsigned long  length)
{
    thrust::plus<float> op;
    thrust::transform(thrust::device,in1,in1+length,in2,out,op);
}

void unaryAdd(float* in1, const float& in2, float* out, unsigned long  length)
{
    thrust::transform(thrust::device,in1,in1+length,out,in2 + _1);
}

void unaryMul(float* in1, float* in2, float* out, unsigned long  length)
{
    thrust::multiplies<float> op;
    thrust::transform(thrust::device,in1,in1+length,in2,out,op);
}

void unaryMul(float* in1, const float& in2, float* out, unsigned long  length)
{
    thrust::transform(thrust::device,in1, in1+length, out, in2 * _1);
}

void unaryNeg(float* in1, float* out, unsigned long length){
    thrust::negate<float> op;
    thrust::transform(thrust::device,in1,in1+length,out,op);
}

void unarySub(float* in1, float* in2, float* out, unsigned long length){
    thrust::minus<float> op;
    thrust::transform(thrust::device,in1,in1+length,in2,out,op);
}

void unaryDiv(float* in1, float* in2, float* out, unsigned long length){
    thrust::divides<float> op;
    thrust::transform(thrust::device,in1,in1+length,in2,out,op);
}

void unarySub(const float& in1, float* in2, float* out, unsigned long length){
    thrust::transform(thrust::device,in2,in2+length,out,in1-_1);
}

void unaryDiv(const float& in1, float* in2, float* out, unsigned long length){
    thrust::transform(thrust::device,in2,in2+length,out,in1/_1);

}

void unarySub(float* in1, const float& in2, float* out, unsigned long length){
    thrust::transform(thrust::device,in1,in1+length,out,_1-in2);
}

void unaryDiv(float* in1, const float& in2, float* out, unsigned long length){
    thrust::transform(thrust::device,in1,in1+length,out,_1/in2);
}

void unaryPow(float* in1, float N,float* out, unsigned long length){
    if (N == 0.0f)
    {
            thrust::fill(thrust::device,out,out+length,1);
            return;
    }
    if (N == 1.0f)
    {
        thrust::copy(thrust::device,in1,in1+length,out);
        return;
    }
    if (N == 2.0f){
        thrust::square<float> op;
        thrust::transform(thrust::device,in1,in1+length,out,op);
        return;
    }
    thrust::transform(thrust::device,in1,in1+length,out,PowOperator(N));
}

void thrustFill(float* aBegin, float* aEnd, float aValue)
{
    thrust::fill(thrust::device, aBegin, aEnd, aValue);
}

void thrustFill(float* aBegin, float* aEnd, std::complex<float> aValue)
{
    thrust::fill(thrust::device, (std::complex<float>*)aBegin, (std::complex<float>*)aEnd, aValue);
}