#include "ml_profiler.h"

#include <iostream>
#include <iomanip>
#include <numeric>
#include <algorithm>


using namespace std::chrono;

void MlPerfMon::turnOn() {
    perfOn = true;
    start_time = std::chrono::high_resolution_clock::now();
}

void MlPerfMon::add_method_call(const std::string &method) {
    if (perfOn) {
        auto search = calls_counter.find(method);
        if (search != calls_counter.end()) {
            calls_counter[method] = search->second + 1;
        } else {
            calls_counter.insert({method, 1});
        }
    }
}

void MlPerfMon::print_results() {
    if (perfOn) {
        std::chrono::time_point<std::chrono::high_resolution_clock> end_time = std::chrono::high_resolution_clock::now();

        using callcount = std::pair<std::string, long>;
        std::vector<callcount> v(begin(calls_counter), end(calls_counter));
        std::sort(std::begin(v), std::end(v), [](const callcount& a, const callcount& b) { return a.second > b.second; });
        long total_calls = 0;
        for(auto const& c : v) { total_calls += c.second; }

        std::cerr << std::endl << std::endl;
        std::cerr << "perf stats" << std::endl;
        std::cerr << "run time       : " << duration_cast<milliseconds>(end_time - start_time).count() << " ms " << std::endl;
        std::cerr << "total calls    : " << total_calls << std::endl;
        std::cerr << "methods called : " << calls_counter.size() << std::endl;
        std::cerr << std::endl;

        int i {0};
        for(auto &p : v) {
            if (p.first.size() >= method_name_print_len)
                p.first.erase(method_name_print_len, std::string::npos);
             else
                p.first.insert(p.first.end(), method_name_print_len - p.first.size(), ' ');

            std::cerr << p.first << " " << p.second << std::endl;

            i++;
            if (i > print_top_methods) break;
        }

        std::cerr << std::endl;
    }
}