/*
PROGRAMA: SumaVectores_OpenMP.cpp
DESCRPCIÓN: Suma dos vectores con tipo de dato float y almacena el resultado 
            en un tercer vector del mismo tipo. 
			Paralelizamos la suma usando directivas de OpenMP.
AUTOR: Francisco J. Hernández-López
Fecha Inicial: 16/Marzo/2016
Fecha Actualización:
*/

#include <stdio.h>
#include <conio.h>
#include <stdlib.h>
#include <iostream>
#include <math.h>
#include <sys\timeb.h> 
//OpenMP
#include <omp.h>


void Suma_vectores(float *c, float *a, float *b, long int N)
{
	long int i;
#pragma omp parallel shared(a,b,c,N) private(i)
	{
#pragma omp for nowait
		for (i = 0; i<N; i++){
			//c[i] = a[i] + b[i];
			c[i] = cos(a[i]) + (int)(b[i])%3;
		}
	}
}

// Código principal que se ejecuta en el Host
int main(void){
	float *a_h, *b_h, *c_h; //Punteros a arreglos en el Host 
	long int N = 100000000;  //Número de elementos en los arreglos
	long int i;
	struct timeb start, end;
	int diff;
	size_t size_float = N * sizeof(float);

	a_h = (float *)malloc(size_float); // Pedimos memoria en el Host
	b_h = (float *)malloc(size_float);
	c_h = (float *)malloc(size_float);//También se puede con cudaMallocHost

	omp_set_num_threads(4);//Fijamos cuantos hilos vamos a lanzar

	//Inicializamos los arreglos a,b en el Host
/*#pragma omp parallel shared(a_h,b_h,N) private(i)
	{
#pragma omp for nowait*/
		for (i = 0; i<N; i++){
			a_h[i] = (float)i;
			b_h[i] = (float)(i + 1);
		}
	//}
	
	printf("\nArreglo a:\n");
	for (i=0; i<20; i++) printf("%f ", a_h[i]);
	printf("\n\nArreglo b:\n");
	for (i=0; i<20; i++) printf("%f ", b_h[i]);
	
	printf("\n\nComenzando el procesamiento...");
	ftime(&start);
	Suma_vectores(c_h, a_h, b_h, N);
	ftime(&end);
	diff = (int)(1000.0 * (end.time - start.time)
		+ (end.millitm - start.millitm));
	printf("\nProcesamiento Finalizado...");
	printf("\nTiempo: %u milisegundos...\n", diff);

	//Resultado
	printf("\n\nArreglo c:\n");
	for (i = 0; i<20; i++)
		printf("%f ", c_h[i]);
	

	system("pause");

	// Liberamos la memoria del Host
	free(a_h);
	free(b_h);
	free(c_h);
	return(0);
}