/* compiled with -msse2, etc*/

//#include <mmintrin.h>  /* MMX */
//#include <xmmintrin.h> /* SSE */
#include <emmintrin.h> /* SSE2 */
//#include <pmmintrin.h> /* SSE3 */
//#include <tmmintrin.h> /* SSSE3 */
//#include <smmintrin.h> /* SSE4.2 SSE4.1 */

#include<sys/time.h>
#include<time.h>
#include<stdio.h>

#include <stdint.h>

typedef uint64_t ticks;

static __inline__ ticks ticks_start(void)
{
	unsigned a, d;
	// Serialize + read TSC
	asm volatile(
		             "cpuid\n\t"
			     "rdtsc\n\t"
			     : "=a"(a), "=d"(d)
			     : "a"(0)
			     : "rbx", "rcx", "memory"
		     );
	return ((ticks)d << 32) | a;
}

static __inline__ ticks ticks_end(void)
{
	unsigned a, d;
	// Read TSC + serialize (rdtscp is partially serializing; cpuid completes)
	asm volatile(
		             "rdtscp\n\t"
			     : "=a"(a), "=d"(d)
			     :
			     : "rcx", "memory"
		     );
	asm volatile(
		             "cpuid\n\t"
			     :
			     : "a"(0)
			     : "rbx", "rcx", "rdx", "memory"
		     );
	return ((ticks)d << 32) | a;
}

/* plain version x = a*x */
void sapxy(int n, float a, float *x)
{
  int i;
  for (i=0; i<n; i++) x[i] = x[i]*a;
}

/* sse version */

void sse_sapxy(int n, float a, float *x)
{
  float aa[4];
  __m128 v_a;
  __m128 *v_x;
  int i;

  aa[0] = aa[1] = aa[2] = aa[3] = a;
  v_a = _mm_load_ps(aa);
  v_x = (__m128 *)x;

  printf("aaa\n"); fflush(0);
  
  for (i=0; i<n-3; i+=4) {
    *v_x = _mm_mul_ps(*v_x, v_a);
    v_x++;
  }
    printf("aaa1\n"); fflush(0);

  for (;i<n; i++)
    x[i] = x[i] * a;
}

float a[1000000];
float b[1000000];

int main()
{
  unsigned long long int t1, t2, t3;


  int i, j;

  for (i=0; i<1000000; i++) {
    a[i] = i*1.0;
    b[i] = i*1.0;
  }

  t1 = ticks_start();
  
  sapxy(999999, 3.14159, a+1);

  t2 = ticks_end();
  printf("sapxy time = %lld\n", t2-t1);

  t1 = ticks_start();
  sse_sapxy(9, 3.14159, b+1);
  t2 = ticks_end();

  printf("sse_sapxy time = %lld\n", t2-t1);
  for (i=0; i<999999; i++) {
    if (a[i] != b[i]) {printf("Wrong.\n"); exit(0);}
    //printf("%d %f %f\n", i, z3[i], z4[i]);
  }
  printf("Correct, results matched.\n");
  return 0;
}