#include "workshop.h"
#include <time.h>
#include <string.h>
#include <iostream>

int mag = 30;
int tamanio = 3;
double coeficientes[] = {0.25, 0.25,0.25,0.25, 0.5, 0.5, 0.5, 0.25, 0.25, 0.25, 0.25, 0.25,0.25,0.25, 0.5, 0.5, 0.5, 0.25, 0.25, 0.25, 0.25, 0.25,0.25,0.25, 0.5, 0.5, 0.5, 0.25, 0.25, 0.25, 0.25, 0.25,0.25,0.25, 0.5, 0.5, 0.5, 0.25, 0.25, 0.25, 0.25, 0.25,0.25,0.25, 0.5, 0.5, 0.5, 0.25, 0.25, 0.25, 0.25, 0.25,0.25,0.25, 0.5, 0.5, 0.5, 0.25, 0.25, 0.25, 0.25, 0.25,0.25,0.25, 0.5, 0.5, 0.5, 0.25, 0.25, 0.25, 0.25, 0.25,0.25,0.25, 0.5, 0.5, 0.5, 0.25, 0.25, 0.25,
    0.25, 0.25,0.25,0.25, 0.5, 0.5, 0.5, 0.25, 0.25, 0.25, 0.25, 0.25,0.25,0.25, 0.5, 0.5, 0.5, 0.25, 0.25, 0.25, 0.25, 0.25,0.25,0.25, 0.5, 0.5, 0.5, 0.25, 0.25, 0.25, 0.25, 0.25,0.25,0.25, 0.5, 0.5, 0.5, 0.25, 0.25, 0.25, 0.25, 0.25,0.25,0.25, 0.5, 0.5, 0.5, 0.25, 0.25, 0.25, 0.25, 0.25,0.25,0.25, 0.5, 0.5, 0.5, 0.25, 0.25, 0.25, 0.25, 0.25,0.25,0.25, 0.5, 0.5, 0.5, 0.25, 0.25, 0.25, 0.25, 0.25,0.25,0.25, 0.5, 0.5, 0.5, 0.25, 0.25, 0.25};

class FIR{
    double *coef;
    double *samples;
    int tam;

public:
    FIR(int n);
    ~FIR();

    void iniciaCoef(double C[], int n);
    double directa( double s);
    double directaMM( double s);

};

FIR::FIR(int n)
{
    tam = n;
    coef = new double[tam];
    samples = new double[tam];

}

FIR::~FIR()
{
    delete coef;
    delete samples;
}

void FIR::iniciaCoef(double C[], int n)
{
    if( tam == n)
    {
        for (int i =0; i<tam; i++)
        {
            coef[i] = C[i];
            samples[i] = 0.0;
        }
    }
    else
    {
        delete coef;
        delete samples;
        tam = n;
        coef = new double[tam];
        samples = new double[tam];
        for (int i =0; i<tam; i++)
        {
            coef[i] = C[i];
            samples[i] = 0.0;
        }

    }


}


double FIR::directa(double s)
{
    double out = 0.0;
    int i;
    samples[0] =s;

    for (i=0; i<tam; i++)
    {

        out += samples[i]*coef[i];
    }
    for (i = tam-1; i>0 ; i--)
    {

        samples[i] = samples[i-1];
    }

    return out;
}

double FIR::directaMM(double s)
{
    double out = 0.0;
    int i;
    samples[0] =s;

    for (i=0; i<tam; i++)
    {
        out += samples[i]*coef[i];
    }
    memmove(samples +1, samples, (tam-1)* sizeof(double));

    return out;
}

int main(int argc, char **argv)
{
    const int size = 512;

    int ciclos = 320000;
    srand(time(NULL));
    FIR unfiltro(160);
    unfiltro.iniciaCoef( coeficientes, 160);

    auto a = workshop::Array<float>(size);
    auto b = workshop::Array<float>(size);
    auto c = workshop::Array<float>(size);

    auto timer = workshop::start_timer();

    for (int i=0; i<tamanio; i++) {
        coeficientes[i] = coeficientes[i] / mag;
    }

    auto duration = workshop::get_duration(timer);

    timer = workshop::start_timer();

    #pragma omp simd
    for (int i=0; i<tamanio; i++) {
        coeficientes[i] = coeficientes[i] / mag;
    }

    auto vector_duration = workshop::get_duration(timer);

    std::cout << "Loop sin vector " << duration
              << " microsegundos para completar." << std::endl;

    std::cout << "Loop con vector " << vector_duration
              << " microsegundos para completar." << std::endl;

    //Medir tiempo
    std::cout<<"Filtro sin vectorizado:  ";
    auto start = std::chrono::system_clock::now();
    for (int i = 0; i<ciclos; i++ )
    {
        unfiltro.directa(rand());
    }
    auto fin = std::chrono::system_clock::now();

    std::chrono::duration<double> elapsed = fin - start;
    std::cout << "Elapsed time: " << elapsed.count() << "s"<<std::endl;


    std::cout<<"Filtro CON vectorizado:  ";
    start = std::chrono::system_clock::now();
    for (int i = 0; i<ciclos; i++ )
    {
        #pragma omp simd
        unfiltro.directa(rand());
    }
    fin = std::chrono::system_clock::now();

    elapsed = fin - start;
    std::cout << "Elapsed time: " << elapsed.count() << "s";
    std::cout<<std::endl<<"__________________________"<<std::endl;

    return 0;
}