-
-
Notifications
You must be signed in to change notification settings - Fork 260
/
logger.cpp
140 lines (129 loc) · 3.35 KB
/
logger.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
//===-- logger.cpp --------------------------------------------------------===//
//
// LDC – the LLVM D compiler
//
// This file is distributed under the BSD-style LDC license. See the LICENSE
// file for details.
//
//===----------------------------------------------------------------------===//
#include "mars.h"
#include "llvm/Support/CommandLine.h"
#include <cassert>
#include <cstdarg>
#include <cstdio>
#include <cstdlib>
#include <fstream>
#include <iostream>
#include <string>
#if LDC_LLVM_VER >= 303
#include "llvm/IR/GlobalValue.h"
#else
#include "llvm/GlobalValue.h"
#endif
#include "llvm/Support/Casting.h"
#include "llvm/Support/raw_os_ostream.h"
#include "llvm/Assembly/Writer.h"
#include "gen/logger.h"
#include "gen/irstate.h"
void Stream::writeType(std::ostream& OS, const llvm::Type& Ty) {
llvm::raw_os_ostream raw(OS);
Ty.print(raw);
}
void Stream::writeValue(std::ostream& OS, const llvm::Value& V) {
// Constants don't always get their types pretty-printed.
// (Only treat non-global constants like this, so that e.g. global variables
// still get their initializers printed)
llvm::raw_os_ostream raw(OS);
if (llvm::isa<llvm::Constant>(V) && !llvm::isa<llvm::GlobalValue>(V))
llvm::WriteAsOperand(raw, &V, true, gIR->module);
else
V.print(raw);
}
namespace Logger
{
static std::string indent_str;
llvm::cl::opt<bool> _enabled("vv",
llvm::cl::desc("Print front-end/glue code debug log"),
llvm::cl::ZeroOrMore);
void indent()
{
if (_enabled) {
indent_str += "* ";
}
}
void undent()
{
if (_enabled) {
assert(!indent_str.empty());
indent_str.resize(indent_str.size()-2);
}
}
Stream cout()
{
if (_enabled)
return std::cout << indent_str;
else
return 0;
}
#if defined(_MSC_VER)
static inline void
search_and_replace(std::string& str, const std::string& what, const std::string& replacement)
{
assert(!what.empty());
size_t pos = str.find(what);
while (pos != std::string::npos)
{
str.replace(pos, what.size(), replacement);
pos = str.find(what, pos + replacement.size());
}
}
#define WORKAROUND_C99_SPECIFIERS_BUG(f) \
std::string tmp = f; \
search_and_replace(tmp, std::string("%z"), std::string("%I")); \
f = tmp.c_str();
#else
#define WORKAROUND_C99_SPECIFIERS_BUG(f)
#endif
void println(const char* fmt,...)
{
if (_enabled) {
printf("%s", indent_str.c_str());
va_list va;
va_start(va,fmt);
WORKAROUND_C99_SPECIFIERS_BUG(fmt);
vprintf(fmt,va);
va_end(va);
printf("\n");
}
}
void print(const char* fmt,...)
{
if (_enabled) {
printf("%s", indent_str.c_str());
va_list va;
va_start(va,fmt);
WORKAROUND_C99_SPECIFIERS_BUG(fmt);
vprintf(fmt,va);
va_end(va);
}
}
void enable()
{
_enabled = true;
}
void disable()
{
_enabled = false;
}
bool enabled()
{
return _enabled;
}
void attention(Loc loc, const char* fmt,...)
{
va_list va;
va_start(va,fmt);
vwarning(loc,fmt,va);
va_end(va);
}
}