typetraits.h

#pragma once

#include "../iuhelpermath.h"
#include "iumathapi.h"

namespace iu {
template<typename PixelType>
struct IUMATH_DLLAPI type_trait {};

template<> struct IUMATH_DLLAPI type_trait<float2>
{
  static inline __host__ __device__ float2 make_complex(float x)
  {
    return make_float2(x, x);
  }
  static inline __host__ __device__ float2 make_complex(float x, float y)
  {
    return make_float2(x, y);
  }

  static inline __host__ __device__ float2 make(float x)
  {
    return make_float2(x, x);
  }

  static inline __host__ __device__  float abs(float2 x)
  {
    return length(x);
  }

  typedef float real_type;

  typedef float2 complex_type;

  struct is_complex { static const bool value = true; };

  static const char* name()
  {
      return "float2";
  }
};

template<> struct IUMATH_DLLAPI type_trait<double2>
{
  static inline __host__ __device__ double2 make_complex(double x)
  {
    return make_double2(x, x);
  }
  static inline __host__ __device__ double2 make_complex(double x, double y)
  {
    return make_double2(x, y);
  }

  static inline __host__ __device__ double2 make(double x)
  {
    return make_double2(x, x);
  }

  static inline __host__ __device__ double abs(double2 x)
  {
    return length(x);
  }

  typedef double real_type;

  typedef double2 complex_type;

  struct is_complex { static const bool value = true; };

  static inline const char* name()
  {
      return "double2";
  }
};

template<> struct IUMATH_DLLAPI type_trait<float>
{
  static inline __host__ __device__ float2 make_complex(float x)
  {
    return make_float2(x, x);
  }
  static inline __host__ __device__ float2 make_complex(float x, float y)
  {
    return make_float2(x, y);
  }

  static inline __host__ __device__ float make(float x)
  {
    return x;
  }

  static inline __host__ __device__ float abs(float x)
  {
    return fabs(x);
  }

  static inline __host__ __device__ float max(float x, float y)
  {
      return fmaxf(x, y);
  }

  typedef float real_type;

  typedef float2 complex_type;

  struct is_complex { static const bool value = false; };

  static const char* name()
  {
      return "float";
  }
};

template<> struct IUMATH_DLLAPI type_trait<double>
{
  static inline __host__ __device__ double2 make_complex(double x)
  {
    return make_double2(x, x);
  }

  static inline __host__ __device__ double2 make_complex(double x, double y)
  {
    return make_double2(x, y);
  }

  static inline __host__ __device__ double make(double x)
  {
    return x;
  }

  static inline __host__ __device__ double abs(double x)
  {
    return fabs(x);
  }

  static inline __host__ __device__ double max(double x, double y)
  {
      return fmax(x, y);
  }

  typedef double real_type;

  typedef double2 complex_type;

  struct is_complex { static const bool value = true; };

  static const char* name()
  {
      return "double";
  }
};

}

template<typename PixelType>
IUMATH_DLLAPI inline __host__ __device__ typename iu::type_trait<PixelType>::complex_type complex_multiply(
    const typename iu::type_trait<PixelType>::complex_type& src1,
    const typename iu::type_trait<PixelType>::complex_type& src2)
{
  typename iu::type_trait<PixelType>::complex_type dst;
  dst.x = src1.x * src2.x - src1.y * src2.y;
  dst.y = src1.x * src2.y + src1.y * src2.x;
  return dst;
}

template<typename PixelType>
IUMATH_DLLAPI inline __host__ __device__ typename iu::type_trait<PixelType>::complex_type complex_multiply_conjugate(
    const typename iu::type_trait<PixelType>::complex_type& src1,
    const typename iu::type_trait<PixelType>::complex_type& src2)
{
  typename iu::type_trait<PixelType>::complex_type dst;
  dst.x = src1.x * src2.x + src1.y * src2.y;
  dst.y = -src1.x * src2.y + src1.y * src2.x;
  return dst;
}

template<typename PixelType>
IUMATH_DLLAPI inline __host__ __device__ typename iu::type_trait<PixelType>::complex_type complex_conjugate(
    const typename iu::type_trait<PixelType>::complex_type& src)
{
  typename iu::type_trait<PixelType>::complex_type dst;
  dst.x = src.x;
  dst.y = -src.y;
  return dst;
}