#include "test_common.hpp"
#include "test_perf.hpp"

void PERF_TEST::Test_MatMul(aclCxt *acl_context)
{
    int val, type;
    int valmax = 8192;
    double begin, end, time, acltime;
    Common_Test test;

    type = CV_32FC1;

    for (val = 8; val <= valmax; val *= 2)
    {
        Mat mat_src(val, val, type);
        Mat mat_src1(val, val, type);
        Mat mat_dest(val, val, type);
        Mat mat_dest1(val, val, type);

        test.SetDataRange(mat_src, 32); 
        test.SetDataRange(mat_src1, 32); 
        test.SetDataRange(mat_dest, 32); 

        aclMat aclmat_src(val, val, type, mat_src.data, acl_context);
        aclMat aclmat_src1(val, val, type, mat_src1.data, acl_context);
        aclMat aclmat_dest(val, val, type, mat_dest.data, acl_context);
        int n = 100;

        begin = static_cast<double>(getTickCount());
        while (n--)
            mat_dest = mat_src * mat_src1;
        end = static_cast<double>(getTickCount());
        time = (end - begin) / getTickFrequency();

        n = 100;
        begin = static_cast<double>(getTickCount());
        while (n--)
            MatMul(aclmat_src1, aclmat_src, aclmat_dest);
        end = static_cast<double>(getTickCount());
        acltime = (end - begin) / getTickFrequency();

        aclmat_dest.download(mat_dest1);
        bool ret = test.Test_Diff(mat_dest, mat_dest1);
        ASSERT_TRUE(ret);
        if (val < 128)
            cout << "Shape: " << val << " x " << val << "\t\t";
        else
            cout << "Shape: " << val << " x " << val << "\t";
        cout << "CpuTimes: " << time << "\tAclTimes: " << acltime << "\tRate: " << time / acltime << endl;
    }
}

void PERF_TEST::Test_Convolution(aclCxt *acl_context)
{
    int val, type;
    int valmax = 8192;
    double begin, end, time, acltime;
    Common_Test test;

    type = CV_32FC1;

    for (val = 8; val <= valmax; val *= 2)
    {
        Mat mat_src(val, val, type, Scalar{1, 2});
        Mat mat_kernel(3, 3, type, Scalar(1, 4));
        Mat mat_dest(val, val, type, Scalar{6});

        aclMat aclmat_src(val, val, type, mat_src.data, acl_context);
        aclMat aclmat_kernel(3, 3, type, mat_kernel.data, acl_context);
        aclMat aclmat_dest(val, val, type, mat_dest.data, acl_context);
        int n = 1;

        begin = static_cast<double>(getTickCount());
        while (n--)
            filter2D(mat_src, mat_dest, -1, mat_kernel);
        end = static_cast<double>(getTickCount());
        time = (end - begin) / getTickFrequency();

        n = 1;
        begin = static_cast<double>(getTickCount());
        vector<int64_t> strides{1, 1, 1, 1};
        vector<int64_t> pads{1, 1, 1, 1};
        while (n--)
            Convolution(aclmat_src, aclmat_kernel, aclmat_dest, strides, pads);
        end = static_cast<double>(getTickCount());
        Mat mat_dest1(aclmat_dest.rows, aclmat_dest.cols, type);
        acltime = (end - begin) / getTickFrequency();

        aclmat_dest.download(mat_dest1);
        cout << mat_dest << endl;
        cout << mat_dest1 << endl;
        /*
        bool ret = test.Test_Diff(mat_dest, mat_dest1);
        ASSERT_TRUE(ret);
        */
        if (val < 128)
            cout << "Shape: " << val << " x " << val << "\t\t";
        else
            cout << "Shape: " << val << " x " << val << "\t";
        cout << "CpuTimes: " << time << "\tAclTimes: " << acltime << "\tRate: " << time / acltime << endl;
    }
}