/*******************************************************

* Copyright (c) 2014, ArrayFire

* All rights reserved.

*

* This file is distributed under 3-clause BSD license.

* The complete license agreement can be obtained at:

* http://arrayfire.com/licenses/BSD-3-Clause

********************************************************/

#include <af/array.h>

#include <af/dim4.hpp>

#include <af/statistics.h>

#include "common.hpp"

#include "error.hpp"

namespace af {

#define INSTANTIATE_STDEV(T) \

template<> \

AFAPI T stdev(const array& in, const af_var_bias bias) { \

double ret_val; \

AF_THROW(af_stdev_all_v2(&ret_val, NULL, in.get(), bias)); \

return (T)ret_val; \

} \

template<> \

AFAPI T stdev(const array& in) { \

return stdev<T>(in, AF_VARIANCE_POPULATION); \

}

template<>

AFAPI af_cfloat stdev(const array& in, const af_var_bias bias) {

double real, imag;

AF_THROW(af_stdev_all_v2(&real, &imag, in.get(), bias));

return {static_cast<float>(real), static_cast<float>(imag)};

}

template<>

AFAPI af_cdouble stdev(const array& in, const af_var_bias bias) {

double real, imag;

AF_THROW(af_stdev_all_v2(&real, &imag, in.get(), bias));

return {real, imag};

}

template<>

AFAPI af_cfloat stdev(const array& in) {

return stdev<af_cfloat>(in, AF_VARIANCE_POPULATION);

}

template<>

AFAPI af_cdouble stdev(const array& in) {

return stdev<af_cdouble>(in, AF_VARIANCE_POPULATION);

}

INSTANTIATE_STDEV(float);

INSTANTIATE_STDEV(double);

INSTANTIATE_STDEV(int);

INSTANTIATE_STDEV(unsigned int);

INSTANTIATE_STDEV(long long);

INSTANTIATE_STDEV(unsigned long long);

INSTANTIATE_STDEV(short);

INSTANTIATE_STDEV(unsigned short);

INSTANTIATE_STDEV(char);

INSTANTIATE_STDEV(signed char);

INSTANTIATE_STDEV(unsigned char);

#undef INSTANTIATE_STDEV

array stdev(const array& in, const af_var_bias bias, const dim_t dim) {

af_array temp = 0;

AF_THROW(af_stdev_v2(&temp, in.get(), bias, getFNSD(dim, in.dims())));

return array(temp);

}

array stdev(const array& in, const dim_t dim) {

return stdev(in, AF_VARIANCE_POPULATION, dim);

}

} // namespace af