extern "C" {
    extern void dot_test();
}

#include "allocator.h"

#include <dsppp/arch.hpp>
#include <dsppp/fixed_point.hpp>
#include <dsppp/matrix.hpp>

#include <iostream>

#include <cmsis_tests.h>

#include "dsp/basic_math_functions.h"
#include "dsp/basic_math_functions_f16.h"





template<typename T,int NB,typename O>
static void complex_test(const T scale)
{
   std::cout << "----\r\n" << "N = " << NB << "\r\n";
   #if defined(STATIC_TEST)
   PVector<T,NB> a;
   PVector<T,NB> b;
   PVector<T,NB> c;
   PVector<T,NB> d;

   PVector<T,NB> res;
   #else 
   PVector<T> a(NB);
   PVector<T> b(NB);
   PVector<T> c(NB);
   PVector<T> d(NB);

   PVector<T> res(NB);
   #endif

   
   init_array(a,NB);
   init_array(b,NB);
   init_array(c,NB);
   init_array(d,NB);
    
   INIT_SYSTICK;
   START_CYCLE_MEASUREMENT;
   startSectionNB(1);
   O result = dot(scale*(a+b),c*d);
   stopSectionNB(1);
   STOP_CYCLE_MEASUREMENT;
   
   O ref;
   PVector<T,NB> tmp1;
   PVector<T,NB> tmp2;
   INIT_SYSTICK;
   START_CYCLE_MEASUREMENT;
   cmsisdsp_dot_expr(a.const_ptr(),
                     b.const_ptr(),
                     c.const_ptr(),
                     d.const_ptr(),
                     tmp1.ptr(),
                     tmp2.ptr(),
                     scale,
                     ref,NB);
   STOP_CYCLE_MEASUREMENT;

   if (!validate(result,ref))
   {
      printf("dot expr failed \r\n");

   }

   std::cout << "=====\r\n";

}


template<typename T,int NB,typename O>
static void test()
{
   std::cout << "----\r\n" << "N = " << NB << "\r\n";
   #if defined(STATIC_TEST)
   PVector<T,NB> a;
   PVector<T,NB> b;

   PVector<T,NB> res;
   #else
   PVector<T> a(NB);
   PVector<T> b(NB);

   PVector<T> res(NB);
   #endif

   init_array(a,NB);
   init_array(b,NB);

   INIT_SYSTICK;
   START_CYCLE_MEASUREMENT;
   startSectionNB(1);
   O result = dot(a,b);
   stopSectionNB(1);
   STOP_CYCLE_MEASUREMENT;


   O ref;
   INIT_SYSTICK;
   START_CYCLE_MEASUREMENT;
   cmsisdsp_dot(a.const_ptr(),b.const_ptr(),ref,NB);
   STOP_CYCLE_MEASUREMENT;

   if (!validate(result,ref))
   {
      printf("dot failed \r\n");

   }

   std::cout << "=====\r\n";

}


template<typename T>
void all_dot_test()
{

   const int nb_tails = TailForTests<T>::tail;
   const int nb_loops = TailForTests<T>::loop;

    using ACC = typename number_traits<T>::accumulator;
    constexpr auto v = TestConstant<T>::v;

    title<T>("Dot product");

    
    test<T,NBVEC_4,ACC>();
    test<T,NBVEC_8,ACC>();
    test<T,NBVEC_9,ACC>();
    test<T,NBVEC_16,ACC>();
    test<T,NBVEC_32,ACC>();
    test<T,NBVEC_64,ACC>();
    test<T,NBVEC_128,ACC>();
    test<T,NBVEC_256,ACC>();
    test<T,NBVEC_258,ACC>();
    test<T,NBVEC_512,ACC>();
    test<T,NBVEC_1024,ACC>();
    if constexpr (!std::is_same<T,double>::value)
    {
       test<T,NBVEC_2048,ACC>();
    }

    test<T,1,ACC>();
    test<T,nb_tails,ACC>();
    test<T,nb_loops,ACC>();
    test<T,nb_loops+1,ACC>();
    test<T,nb_loops+nb_tails,ACC>();


    title<T>("Dot product with expressions");

    
    complex_test<T,NBVEC_4,ACC>(v);
    complex_test<T,NBVEC_8,ACC>(v);
    complex_test<T,NBVEC_9,ACC>(v);
    complex_test<T,NBVEC_32,ACC>(v);
    complex_test<T,NBVEC_64,ACC>(v);
    complex_test<T,NBVEC_128,ACC>(v);
    
    complex_test<T,NBVEC_256,ACC>(v);
    
    complex_test<T,NBVEC_258,ACC>(v);
    complex_test<T,NBVEC_512,ACC>(v);
    complex_test<T,NBVEC_1024,ACC>(v);
    if constexpr (!std::is_same<T,double>::value)
    {
       complex_test<T,NBVEC_2048,ACC>(v);
    }

    complex_test<T,1,ACC>(v);
    complex_test<T,nb_tails,ACC>(v);
    complex_test<T,nb_loops,ACC>(v);
    complex_test<T,nb_loops+1,ACC>(v);
    complex_test<T,nb_loops+nb_tails,ACC>(v);

    //print_map("Stats",max_stats);

}

void dot_test()
{
#if defined(DOT_TEST)
   #if defined(F64_DT)
   all_dot_test<double>();
   #endif
   #if defined(F32_DT)
   all_dot_test<float>();
   #endif
   #if defined(F16_DT) && !defined(DISABLEFLOAT16)
   all_dot_test<float16_t>();
   #endif
   #if defined(Q31_DT)
   all_dot_test<Q31>();
   #endif
   #if defined(Q15_DT)
   all_dot_test<Q15>();
   #endif
   #if defined(Q7_DT)
   all_dot_test<Q7>();
   #endif
#endif
}