MacroBuilder.h revision 202879 
1//===--- MacroBuilder.h - CPP Macro building utility ------------*- C++ -*-===//
2//
3//                     The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9//
10//  This file defines the MacroBuilder utility class.
11//
12//===----------------------------------------------------------------------===//
13
14#ifndef LLVM_CLANG_BASIC_MACROBUILDER_H
15#define LLVM_CLANG_BASIC_MACROBUILDER_H
16
17#include "llvm/ADT/Twine.h"
18#include "llvm/Support/raw_ostream.h"
19
20namespace clang {
21
22class MacroBuilder {
23  llvm::raw_ostream &Out;
24public:
25  MacroBuilder(llvm::raw_ostream &Output) : Out(Output) {}
26
27  /// Append a #define line for macro of the form "#define Name Value\n".
28  void defineMacro(const llvm::Twine &Name, const llvm::Twine &Value = "1") {
29    Out << "#define " << Name << ' ' << Value << '\n';
30  }
31
32  /// Append a #undef line for Name.  Name should be of the form XXX
33  /// and we emit "#undef XXX".
34  void undefineMacro(const llvm::Twine &Name) {
35    Out << "#undef " << Name << '\n';
36  }
37
38  /// Directly append Str and a newline to the underlying buffer.
39  void append(const llvm::Twine &Str) {
40    Out << Str << '\n';
41  }
42};
43
44}  // end namespace clang
45
46#endif
47