/*
 * Copyright (c) 2017-2018 ARM Limited.
 *
 * SPDX-License-Identifier: MIT
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to
 * deal in the Software without restriction, including without limitation the
 * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
 * sell copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in all
 * copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 */
#pragma once

#ifdef CYCLE_PROFILING

#include "../perf.h"

#ifndef NO_MULTI_THREADING
#include <mutex>
#endif

namespace arm_gemm
{
#ifndef NO_MULTI_THREADING
extern std::mutex report_mutex;
#endif

class profiler
{
private:
    static const int maxevents         = 100000;
    unsigned long    times[maxevents]  = {};
    unsigned long    units[maxevents]  = {};
    int              events[maxevents] = {};
    int              currentevent      = 0;
    int              countfd           = 0;

    class ScopedProfilerClass
    {
    private:
        profiler &_parent;
        bool      legal = false;

    public:
        ScopedProfilerClass(profiler &prof, int i, unsigned long u)
            : _parent(prof)
        {
            if(prof.currentevent == maxevents)
                return;

            prof.events[prof.currentevent] = i;
            prof.units[prof.currentevent]  = u;
            legal                          = true;
            start_counter(prof.countfd);
        }

        ~ScopedProfilerClass()
        {
            if(!legal)
                return;

            long long cycs                        = stop_counter(_parent.countfd);
            _parent.times[_parent.currentevent++] = cycs;
        }
    };

public:
    profiler()
    {
        countfd = open_cycle_counter();
    }

    ~profiler()
    {
        close(countfd);
        int           tots[5];
        unsigned long counts[5];
        unsigned long tunits[5];
        const char   *descs[] = { "Prepare A", "Prepare B", "Kernel", "Merge" };

        for(int i = 1; i < 5; i++)
        {
            tots[i]   = 0;
            counts[i] = 0;
            tunits[i] = 0;
        }

        for(int i = 0; i < currentevent; i++)
        {
            //            printf("%10s: %ld\n", descs[events[i]-1], times[i]);
            tots[events[i]]++;
            counts[events[i]] += times[i];
            tunits[events[i]] += units[i];
        }

#ifdef NO_MULTI_THREADING
        printf("Profiled events:\n");
#else
        std::lock_guard<std::mutex> lock(report_mutex);
        printf("Profiled events (cpu %d):\n", sched_getcpu());
#endif

        printf("%20s  %9s %9s %9s %12s %9s\n", "", "Events", "Total", "Average", "Bytes/MACs", "Per cycle");
        for(int i = 1; i < 5; i++)
        {
            printf("%20s: %9d %9ld %9ld %12lu %9.2f\n", descs[i - 1], tots[i], counts[i], counts[i] / tots[i], tunits[i], (float)tunits[i] / counts[i]);
        }
    }

    template <typename T>
    void operator()(int i, unsigned long u, T func)
    {
        if(currentevent == maxevents)
        {
            func();
        }
        else
        {
            events[currentevent] = i;
            units[currentevent]  = u;
            start_counter(countfd);
            func();
            long long cycs        = stop_counter(countfd);
            times[currentevent++] = cycs;
        }
    }
    ScopedProfilerClass ScopedProfiler(int i, unsigned long u)
    {
        return ScopedProfilerClass(*this, i, u);
    }
};

#endif // CYCLE_PROFILING

} // namespace arm_gemm

#define PROFILE_PREPA 1
#define PROFILE_PREPB 2
#define PROFILE_KERNEL 3
#define PROFILE_MERGE 4