MNN/source/core/AutoTime.cpp

//
//  AutoTime.cpp
//  MNN
//
//  Created by MNN on 2018/07/14.
//  Copyright © 2018, Alibaba Group Holding Limited
//

#include <stdlib.h>
#include <string.h>
#if defined(_MSC_VER)
#include <Windows.h>
#else
#include <sys/time.h>
#endif
#include <MNN/AutoTime.hpp>
#include "core/Macro.h"

namespace MNN {

Timer::Timer() {
    reset();
}

Timer::~Timer() {
    // do nothing
}

void Timer::reset() {
#if defined(_MSC_VER)
    LARGE_INTEGER time, freq;
    QueryPerformanceFrequency(&freq);
    QueryPerformanceCounter(&time);
    uint64_t sec   = time.QuadPart / freq.QuadPart;
    uint64_t usec  = (time.QuadPart % freq.QuadPart) * 1000000 / freq.QuadPart;
    mLastResetTime = sec * 1000000 + usec;
#else
    struct timeval Current;
    gettimeofday(&Current, nullptr);
    mLastResetTime = Current.tv_sec * 1000000 + Current.tv_usec;
#endif
}

uint64_t Timer::durationInUs() {
#if defined(_MSC_VER)
    LARGE_INTEGER time, freq;
    QueryPerformanceCounter(&time);
    QueryPerformanceFrequency(&freq);
    uint64_t sec  = time.QuadPart / freq.QuadPart;
    uint64_t usec = (time.QuadPart % freq.QuadPart) * 1000000 / freq.QuadPart;
    auto lastTime = sec * 1000000 + usec;
#else
    struct timeval Current;
    gettimeofday(&Current, nullptr);
    auto lastTime = Current.tv_sec * 1000000 + Current.tv_usec;
#endif

    return lastTime - mLastResetTime;
}

AutoTime::AutoTime(int line, const char* func) : Timer() {
    mName = ::strdup(func);
    mLine = line;
}
AutoTime::~AutoTime() {
    auto timeInUs = durationInUs();
    MNN_PRINT("%s, %d, cost time: %f ms\n", mName, mLine, (float)timeInUs / 1000.0f);
    free(mName);
}

} // namespace MNN