xref: /aosp_15_r20/test/dittosuite/include/ditto/tracer.h (revision 6fa2df46f119dce7527f5beb2814eca0e6f886ac)
1*6fa2df46SAndroid Build Coastguard Worker // Copyright (C) 2023 The Android Open Source Project
2*6fa2df46SAndroid Build Coastguard Worker //
3*6fa2df46SAndroid Build Coastguard Worker // Licensed under the Apache License, Version 2.0 (the "License");
4*6fa2df46SAndroid Build Coastguard Worker // you may not use this file except in compliance with the License.
5*6fa2df46SAndroid Build Coastguard Worker // You may obtain a copy of the License at
6*6fa2df46SAndroid Build Coastguard Worker //
7*6fa2df46SAndroid Build Coastguard Worker //      http://www.apache.org/licenses/LICENSE-2.0
8*6fa2df46SAndroid Build Coastguard Worker //
9*6fa2df46SAndroid Build Coastguard Worker // Unless required by applicable law or agreed to in writing, software
10*6fa2df46SAndroid Build Coastguard Worker // distributed under the License is distributed on an "AS IS" BASIS,
11*6fa2df46SAndroid Build Coastguard Worker // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12*6fa2df46SAndroid Build Coastguard Worker // See the License for the specific language governing permissions and
13*6fa2df46SAndroid Build Coastguard Worker // limitations under the License.
14*6fa2df46SAndroid Build Coastguard Worker 
15*6fa2df46SAndroid Build Coastguard Worker #pragma once
16*6fa2df46SAndroid Build Coastguard Worker 
17*6fa2df46SAndroid Build Coastguard Worker #include <memory>
18*6fa2df46SAndroid Build Coastguard Worker #include <fstream>
19*6fa2df46SAndroid Build Coastguard Worker 
20*6fa2df46SAndroid Build Coastguard Worker #include <benchmark.pb.h>
21*6fa2df46SAndroid Build Coastguard Worker 
22*6fa2df46SAndroid Build Coastguard Worker 
23*6fa2df46SAndroid Build Coastguard Worker namespace dittosuite {
24*6fa2df46SAndroid Build Coastguard Worker 
25*6fa2df46SAndroid Build Coastguard Worker class Tracer {
26*6fa2df46SAndroid Build Coastguard Worker  public:
27*6fa2df46SAndroid Build Coastguard Worker   Tracer();
28*6fa2df46SAndroid Build Coastguard Worker   ~Tracer();
29*6fa2df46SAndroid Build Coastguard Worker   Tracer(Tracer& other) = delete;
30*6fa2df46SAndroid Build Coastguard Worker 
31*6fa2df46SAndroid Build Coastguard Worker   void StartSession(std::unique_ptr<dittosuiteproto::Benchmark> benchmark);
32*6fa2df46SAndroid Build Coastguard Worker   void Start(const std::string &splice);
33*6fa2df46SAndroid Build Coastguard Worker   void End(const std::string &splice);
34*6fa2df46SAndroid Build Coastguard Worker 
35*6fa2df46SAndroid Build Coastguard Worker  private:
36*6fa2df46SAndroid Build Coastguard Worker   std::ofstream trace_marker_;
37*6fa2df46SAndroid Build Coastguard Worker   std::string id_;
38*6fa2df46SAndroid Build Coastguard Worker };
39*6fa2df46SAndroid Build Coastguard Worker 
40*6fa2df46SAndroid Build Coastguard Worker template <class T>
trace_format__(const T & id)41*6fa2df46SAndroid Build Coastguard Worker std::string trace_format__(const T& id) {
42*6fa2df46SAndroid Build Coastguard Worker   return std::string("|") + id;
43*6fa2df46SAndroid Build Coastguard Worker }
44*6fa2df46SAndroid Build Coastguard Worker 
45*6fa2df46SAndroid Build Coastguard Worker template <class T, class... Args>
trace_format__(const T & id,Args...vars)46*6fa2df46SAndroid Build Coastguard Worker std::string trace_format__(const T& id, Args... vars) {
47*6fa2df46SAndroid Build Coastguard Worker   return std::string("|") + id + trace_format__(vars...);
48*6fa2df46SAndroid Build Coastguard Worker }
49*6fa2df46SAndroid Build Coastguard Worker 
50*6fa2df46SAndroid Build Coastguard Worker template <class T>
trace_format(const T & id)51*6fa2df46SAndroid Build Coastguard Worker std::string trace_format(const T& id) {
52*6fa2df46SAndroid Build Coastguard Worker   return id;
53*6fa2df46SAndroid Build Coastguard Worker }
54*6fa2df46SAndroid Build Coastguard Worker 
55*6fa2df46SAndroid Build Coastguard Worker template <class T, class... Args>
trace_format(const T & id,Args...vars)56*6fa2df46SAndroid Build Coastguard Worker std::string trace_format(const T& id, Args... vars) {
57*6fa2df46SAndroid Build Coastguard Worker   std::string result;
58*6fa2df46SAndroid Build Coastguard Worker 
59*6fa2df46SAndroid Build Coastguard Worker   result = trace_format(id);
60*6fa2df46SAndroid Build Coastguard Worker 
61*6fa2df46SAndroid Build Coastguard Worker   return result + trace_format__(vars...);
62*6fa2df46SAndroid Build Coastguard Worker }
63*6fa2df46SAndroid Build Coastguard Worker 
64*6fa2df46SAndroid Build Coastguard Worker 
65*6fa2df46SAndroid Build Coastguard Worker }  // namespace dittosuite
66