/* ----------------------------------------------------------------------
   LAMMPS - Large-scale Atomic/Molecular Massively Parallel Simulator

   Original Version:
   http://lammps.sandia.gov, Sandia National Laboratories
   Steve Plimpton, sjplimp@sandia.gov

   See the README file in the top-level LAMMPS directory.

   -----------------------------------------------------------------------

   USER-CUDA Package and associated modifications:
   https://sourceforge.net/projects/lammpscuda/

   Christian Trott, christian.trott@tu-ilmenau.de
   Lars Winterfeld, lars.winterfeld@tu-ilmenau.de
   Theoretical Physics II, University of Technology Ilmenau, Germany

   See the README file in the USER-CUDA directory.

   This software is distributed under the GNU General Public License.
------------------------------------------------------------------------- */

#ifndef CUDA_PRECISION_H_
#define CUDA_PRECISION_H_
/* This File gives Type definitions for mixed precision calculation in the cuda part of LAMMPS-CUDA.
 * Predefined behaviour is given by global CUDA_PRECISION (can be overwritten during compilation).
 * ***_CFLOAT: type definition of given property
 * ***_F: constant extension in code (1.0 is interpreted as double while 1.0f is interpreted as float, now use: 1.0CUDA_F)
 */

#ifdef CUDA_USE_BINNING
#define CUDA_IF_BINNING(a) a
#else
#define CUDA_IF_BINNING(a)
#endif

//GLOBAL

#ifdef CUDA_PRECISION
#if CUDA_PRECISION == 1
#define CUDA_CFLOAT float
#define CUDA_F(x) x##f
#endif
#if CUDA_PRECISION == 2
#define CUDA_CFLOAT double
#define CUDA_F(x) x
#endif
#endif

#ifndef CUDA_PRECISION
#define CUDA_CFLOAT double
#define CUDA_F(x) x
#define CUDA_PRECISION 2
#endif
//--------------------------------
//-----------FFT-----------------
//--------------------------------

#ifdef FFT_PRECISION_CU
#if FFT_PRECISION_CU == 1
#define FFT_CFLOAT float
#define FFT_F(x) x##f
#endif
#if FFT_PRECISION_CU == 2
#define FFT_CFLOAT double
#define FFT_F(x) x
#endif
#endif

#ifndef FFT_PRECISION_CU
#define FFT_CFLOAT CUDA_CFLOAT
#define FFT_F(x) CUDA_F(x)
#define FFT_PRECISION_CU CUDA_PRECISION
#endif

//--------------------------------
//-----------PPPM-----------------
//--------------------------------

#ifndef PPPM_PRECISION
#define PPPM_PRECISION CUDA_PRECISION
#endif

#ifdef PPPM_PRECISION
#if PPPM_PRECISION == 1
#define PPPM_CFLOAT float
#ifdef float3
#define PPPM_CFLOAT3 float3
#else
struct PPPM_CFLOAT3 {
  PPPM_CFLOAT x;
  PPPM_CFLOAT y;
  PPPM_CFLOAT z;
};
#endif
#define PPPM_F(x) x##f
#endif
#if PPPM_PRECISION == 2
#define PPPM_CFLOAT double
struct PPPM_CFLOAT3 {
  PPPM_CFLOAT x;
  PPPM_CFLOAT y;
  PPPM_CFLOAT z;
};
#define PPPM_F(x) x
#endif
#endif


//--------------------------------
//-----------FORCE-----------------
//--------------------------------


#ifdef F_PRECISION
#if F_PRECISION == 1
#define F_CFLOAT float
#define F_F(x) x##f
#endif
#if F_PRECISION == 2
#define F_CFLOAT double
#define F_F(x) x
#endif
#endif

#ifndef F_PRECISION
#define F_CFLOAT CUDA_CFLOAT
#define F_F(x) CUDA_F(x)
#define F_PRECISION CUDA_PRECISION
#endif

#if F_PRECISION == 1
#define _SQRT_ sqrtf
#define _RSQRT_ rsqrtf
#define _EXP_ expf
#else
#define _SQRT_ sqrt
#define _RSQRT_ rsqrt
#define _EXP_ exp
#endif

#if F_PRECISION == 2
struct F_CFLOAT2 {
  F_CFLOAT x;
  F_CFLOAT y;
};
struct F_CFLOAT3 {
  F_CFLOAT x;
  F_CFLOAT y;
  F_CFLOAT z;
};
struct F_CFLOAT4 {
  F_CFLOAT x;
  F_CFLOAT y;
  F_CFLOAT z;
  F_CFLOAT w;
};
#else
#define F_CFLOAT2 float2
#define F_CFLOAT3 float3
#define F_CFLOAT4 float4
#endif

//--------------------------------
//-----------ENERGY-----------------
//--------------------------------

#ifndef ENERGY_PRECISION
#define ENERGY_CFLOAT CUDA_CFLOAT
#define ENERGY_F(x) CUDA_F(x)
#endif

#ifdef ENERGY_PRECISION
#if ENERGY_PRECISION == 1
#define ENERGY_CFLOAT float
#define ENERGY_F(x) x##f
#endif
#if ENERGY_PRECISION == 2
#define ENERGY_CFLOAT double
#define ENERGY_F(x) x
#endif
#endif

#ifndef ENERGY_PRECISION
#define ENERGY_CFLOAT CUDA_CFLOAT
#define ENERGY_F(x) CUDA_F(x)
#define ENERGY_PRECISION CUDA_PRECISION
#endif

//--------------------------------
//-----------POSITIONS------------
//--------------------------------

#ifdef X_PRECISION
#if X_PRECISION == 1
#define X_CFLOAT float
#define X_F(x) x##f
#endif
#if X_PRECISION == 2
#define X_CFLOAT double
#define X_F(x) x
#endif
#endif

#ifndef X_PRECISION
#define X_CFLOAT CUDA_CFLOAT
#define X_F(x) CUDA_F(x)
#define X_PRECISION CUDA_PRECISION
#endif

#if X_PRECISION == 2
struct X_CFLOAT2 {
  X_CFLOAT x;
  X_CFLOAT y;
};
struct X_CFLOAT3 {
  X_CFLOAT x;
  X_CFLOAT y;
  X_CFLOAT z;
};
struct X_CFLOAT4 {
  X_CFLOAT x;
  X_CFLOAT y;
  X_CFLOAT z;
  X_CFLOAT w;
};
#else
#define X_CFLOAT2 float2
#define X_CFLOAT3 float3
#define X_CFLOAT4 float4
#endif

//--------------------------------
//-----------velocities-----------
//--------------------------------

#ifdef V_PRECISION
#if V_PRECISION == 1
#define V_CFLOAT float
#define V_F(x) x##f
#endif
#if V_PRECISION == 2
#define V_CFLOAT double
#define V_F(x) x
#endif
#endif

#ifndef V_PRECISION
#define V_CFLOAT CUDA_CFLOAT
#define V_F(x) CUDA_F(x)
#define V_PRECISION CUDA_PRECISION
#endif

#if V_PRECISION == 2
struct V_CFLOAT4 {
  V_CFLOAT x;
  V_CFLOAT y;
  V_CFLOAT z;
  V_CFLOAT w;
};
#else
#define V_CFLOAT4 float4
#endif

#ifdef NO_PREC_TIMING
struct my_times {
  unsigned int tv_sec;
  unsigned int tv_nsec;
};

#define my_gettime(a,b)
#else
#define my_times timespec
#define my_gettime(a,b) clock_gettime(a,b)
#endif

#endif /*CUDA_PRECISION_H_*/