tracer.hpp 603 Bytes
Newer Older
Paul's avatar
Paul committed
1
2
3
4
5
6
7
8
9
#ifndef MIGRAPH_GUARD_RTGLIB_TRACER_HPP
#define MIGRAPH_GUARD_RTGLIB_TRACER_HPP

#include <ostream>

namespace migraph {

struct swallow
{
Paul's avatar
Paul committed
10
    template <class... Ts>
Paul's avatar
Paul committed
11
    swallow(Ts&&...)
Paul's avatar
Paul committed
12
13
    {
    }
Paul's avatar
Paul committed
14
15
16
17
};

struct tracer
{
Paul's avatar
Paul committed
18
    tracer() {}
Paul's avatar
Paul committed
19

Paul's avatar
Paul committed
20
    tracer(std::ostream& s) : os(&s) {}
Paul's avatar
Paul committed
21

Paul's avatar
Paul committed
22
    bool enabled() const { return os != nullptr; }
Paul's avatar
Paul committed
23

Paul's avatar
Paul committed
24
    template <class... Ts>
Paul's avatar
Paul committed
25
26
    void operator()(const Ts&... xs) const
    {
Paul's avatar
Paul committed
27
        if(os != nullptr)
Paul's avatar
Paul committed
28
29
30
31
32
33
        {
            swallow{*os << xs...};
            *os << std::endl;
        }
    }

Paul's avatar
Paul committed
34
35
    private:
    std::ostream* os = nullptr;
Paul's avatar
Paul committed
36
37
38
39
40
};

} // namespace migraph

#endif