1/*
2 * Copyright (C) 2014, The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 *     http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17#ifndef LATINIME_PROFILER_H
18#define LATINIME_PROFILER_H
19
20#ifdef FLAG_DO_PROFILE
21
22#include "defines.h"
23
24#include <ctime>
25#include <unordered_map>
26
27namespace latinime {
28
29class Profiler final {
30 public:
31    Profiler(const clockid_t clockId)
32            : mClockId(clockId), mStartTime(getTimeInMicroSec()), mStartTimes(), mTimes(),
33              mCounters() {}
34
35    ~Profiler() {
36        const float totalTime =
37                static_cast<float>(getTimeInMicroSec() - mStartTime) / 1000.f;
38        AKLOGI("Total time is %6.3f ms.", totalTime);
39        for (const auto &time : mTimes) {
40            AKLOGI("(%d): Used %4.2f%%, %8.4f ms. Called %d times.", time.first,
41                    time.second / totalTime * 100.0f, time.second, mCounters[time.first]);
42        }
43    }
44
45    void startTimer(const int id) {
46        mStartTimes[id] = getTimeInMicroSec();
47    }
48
49    void endTimer(const int id) {
50        mTimes[id] += static_cast<float>(getTimeInMicroSec() - mStartTimes[id]) / 1000.0f;
51        mCounters[id]++;
52    }
53
54    operator bool() const { return false; }
55
56 private:
57    DISALLOW_IMPLICIT_CONSTRUCTORS(Profiler);
58
59    const clockid_t mClockId;
60    int64_t mStartTime;
61    std::unordered_map<int, int64_t> mStartTimes;
62    std::unordered_map<int, float> mTimes;
63    std::unordered_map<int, int> mCounters;
64
65    int64_t getTimeInMicroSec() {
66        timespec time;
67        clock_gettime(mClockId, &time);
68        return static_cast<int64_t>(time.tv_sec) * 1000000
69                + static_cast<int64_t>(time.tv_nsec) / 1000;
70    }
71};
72} // namespace latinime
73
74#define PROF_INIT Profiler __LATINIME__PROFILER__(CLOCK_THREAD_CPUTIME_ID)
75#define PROF_TIMER_START(timer_id) __LATINIME__PROFILER__.startTimer(timer_id)
76#define PROF_TIMER_END(timer_id) __LATINIME__PROFILER__.endTimer(timer_id)
77
78#else // FLAG_DO_PROFILE
79
80#define PROF_INIT
81#define PROF_TIMER_START(timer_id)
82#define PROF_TIMER_END(timer_id)
83
84#endif // FLAG_DO_PROFILE
85
86#endif /* LATINIME_PROFILER_H */
87