#include <iostream>
#include "timer.h"


void Timer::start() {
    start_ = high_resolution_clock::now();
}

void Timer::end() {
    end_ = high_resolution_clock::now();
}

double Timer::get_time() {
    time_span = std::chrono::duration_cast<duration<double>>(end_ - start_);
    return time_span.count();
}

void Timer::start(string name) {
    auto it = start_map_.find(name);
    if (it != start_map_.end()) it -> second = high_resolution_clock::now();
    else {
        start_map_[name] = high_resolution_clock::now();
    }
}


void Timer::end(string name) {
    auto it = end_map_.find(name);
    if (it != end_map_.end()) it -> second = high_resolution_clock::now();
    else {
        end_map_[name] = high_resolution_clock::now();
    }
    double period = get_time(name);
    if(period != -1) time_counter[name] += period;
    else std::cerr << "Timer::add_up: period is minus" << std::endl;
}


void Timer::drop_counter(string name) {
    auto it = time_counter.find(name);
    if (it != time_counter.end()) time_counter.erase(it);
}

void Timer::add_up(string name, double time) {
    time_counter[name] += time;
}

double Timer::get_time(string name) {
    auto it_start = start_map_.find(name);
    auto it_end = end_map_.find(name);

    if (it_start == start_map_.end() || it_end == end_map_.end()) return -1;
    time_span = std::chrono::duration_cast<duration<double>>(it_end -> second - it_start -> second);
    return time_span.count();
}

double Timer::get_overall_time(string name) {
    auto it = time_counter.find(name);

    if (it != time_counter.end()) return it -> second;
    else return -1;
}