#ifndef __SSEDEFS_H__
#define __SSEDEFS_H__

#include "cosmoType.h"

#if  CMK_USE_AVX
	#if !defined(__AVX__)
		#undef CMK_USE_AVX
		#define CMK_USE_AVX 0
	#else
		#warning "using AVX"
	#endif
#endif

#if CMK_USE_SSE2 && !defined(__SSE2__)
	#undef CMK_USE_SSE2
	#define CMK_USE_SSE2 0
#endif

#if CMK_USE_AVX || CMK_USE_SSE2
	#define CMK_SSE 1
#endif

#if CMK_USE_AVX
	#ifdef COSMO_FLOAT
		#error "single-precision AVX is not supported"
	#else
		#include "SSE-Double.h"
		#define SSE_VECTOR_WIDTH 4
		#define FORCE_INPUT_LIST_PAD 3
		typedef SSEDouble SSEcosmoType;
		#define SSELoad(where, arr, idx, field) where(arr[idx]field, arr[idx+1]field, arr[idx+2]field, arr[idx+3]field)
		#define SSEStore(what, arr, idx, field) { \
		  double p[4];  \
		  storeu(p, what); \
		  arr[idx]field = p[0]; \
		  arr[idx+1]field = p[1]; \
		  arr[idx+2]field = p[2]; \
		  arr[idx+3]field = p[3]; \
		}
		enum {cosmoMask=0xf};
	#endif
#elif CMK_USE_SSE2
	#ifdef COSMO_FLOAT
		#define SSE_COSMO_FLOAT
		#if defined(__SSE2__)
			#include "SSE-Float.h"
			#define SSE_VECTOR_WIDTH 4
			#define FORCE_INPUT_LIST_PAD 3
			typedef SSEFloat SSEcosmoType;
			#define SSELoad(where, arr, idx, field) where(arr[idx]field, arr[idx+1]field, arr[idx+2]field, arr[idx+3]field)
			#define SSEStore(what, arr, idx, field) { \
			  float p[4]; \
			  storeu(p, what); \
			  arr[idx]field = p[0]; \
			  arr[idx+1]field = p[1]; \
			  arr[idx+2]field = p[2]; \
			  arr[idx+3]field = p[3]; \
			}
			enum {cosmoMask=0xf};
                #else
                      #error("SSE not available");
                #endif
        #else
		#if defined(__SSE2__) && !defined(SSE_COSMO_FLOAT)
			#include "SSE-Double.h"
			#define SSE_VECTOR_WIDTH 2
			#define FORCE_INPUT_LIST_PAD 1
			typedef SSEDouble SSEcosmoType;
			#define SSELoad(where, arr, idx, field) where(arr[idx]field, arr[idx+1]field)
			#define SSEStore(what, arr, idx, field) { \
			  storel(&arr[idx]field, what); \
			  storeh(&arr[idx+1]field, what); \
			}
			enum {cosmoMask=0x3};
                #else
                      #error("SSE not available");
		#endif
	#endif
#endif

#endif