Imported Upstream version 1.4.0
[platform/core/ml/nnfw.git] / runtime / libs / misc / include / misc / EventRecorder.h
1 /*
2  * Copyright (c) 2019 Samsung Electronics Co., Ltd. All Rights Reserved
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 __EVENT_RECORDER_H__
18 #define __EVENT_RECORDER_H__
19
20 #include <map>
21 #include <memory>
22 #include <mutex>
23
24 #include <ostream>
25 #include <sstream>
26
27 struct Event
28 {
29   std::string name;
30   std::string tid;
31   std::string ph; /* REQUIRED */
32   std::string ts; /* REQUIRED */
33 };
34
35 struct DurationEvent : public Event
36 {
37   // TO BE FILLED
38 };
39
40 struct CounterEvent : public Event
41 {
42   std::map<std::string, std::string> values;
43 };
44
45 //
46 // Record Event as Chrome Trace Event File Format
47 //
48 // Refrence: https://docs.google.com/document/d/1CvAClvFfyA5R-PhYUmn5OOQtYMH4h6I0nSsKchNAySU/edit
49 //
50 class EventRecorder
51 {
52 public:
53   EventRecorder() = default;
54
55 public:
56   void emit(const DurationEvent &evt);
57   void emit(const CounterEvent &evt);
58
59 public:
60   bool empty() { return _ss.str().empty(); }
61   void writeToFile(std::ostream &os);
62
63 private:
64   std::mutex _mu;
65   std::stringstream _ss;
66 };
67
68 #endif // __EVENT_RECORDER_H__