CodeGenHelpers.h

Go to the documentation of this file.

//===----------------------------------------------------------------------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
//
// This file defines common utilities for generating C++ code.
//
//===----------------------------------------------------------------------===//
 
#ifndef LLVM_TABLEGEN_CODEGENHELPERS_H
#define LLVM_TABLEGEN_CODEGENHELPERS_H
 
#include "llvm/ADT/STLExtras.h"
#include "llvm/ADT/StringExtras.h"
#include "llvm/ADT/StringRef.h"
#include "llvm/Support/raw_ostream.h"
#include <string>
 
namespace llvm {
 
// Simple RAII helper for emitting ifdef-undef-endif scope. `LateUndef` controls
// whether the undef is emitted at the start of the scope (false) or at the end
// of the scope (true).
class IfDefEmitter {
public:
  IfDefEmitter(raw_ostream &OS, StringRef Name, bool LateUndef = false)
      : Name(Name.str()), OS(OS), LateUndef(LateUndef) {
    OS << "#ifdef " << Name << "\n";
    if (!LateUndef)
      OS << "#undef " << Name << "\n";
    OS << "\n";
  }
  ~IfDefEmitter() {
    OS << "\n";
    if (LateUndef)
      OS << "#undef " << Name << "\n";
    OS << "#endif // " << Name << "\n\n";
  }
 
private:
  std::string Name;
  raw_ostream &OS;
  bool LateUndef;
};
 
// Base class for RAII helpers for emitting various if guards.
class IfGuardEmitterBase {
protected:
  IfGuardEmitterBase(raw_ostream &OS, StringRef If, StringRef Condition)
      : Condition(Condition.str()), OS(OS) {
    OS << If << " " << Condition << "\n\n";
  }
 
  ~IfGuardEmitterBase() { OS << "\n#endif // " << Condition << "\n\n"; }
 
private:
  std::string Condition;
  raw_ostream &OS;
};
 
// RAII emitter for:
// #if <Condition>
// #endif // <Condition>
class IfGuardEmitter : private IfGuardEmitterBase {
public:
  IfGuardEmitter(raw_ostream &OS, StringRef Condition)
      : IfGuardEmitterBase(OS, "#if", Condition) {}
};
 
// RAII emitter for:
// #ifdef <Condition>
// #endif // <Condition>
class IfDefGuardEmitter : private IfGuardEmitterBase {
public:
  IfDefGuardEmitter(raw_ostream &OS, StringRef Condition)
      : IfGuardEmitterBase(OS, "#ifdef", Condition) {}
};
 
// RAII emitter for:
// #ifndef <Condition>
// #endif // <Condition>
class IfNDefGuardEmitter : private IfGuardEmitterBase {
public:
  IfNDefGuardEmitter(raw_ostream &OS, StringRef Condition)
      : IfGuardEmitterBase(OS, "#ifndef", Condition) {}
};
 
// Simple RAII helper for emitting header include guard (ifndef-define-endif).
class IncludeGuardEmitter {
public:
  IncludeGuardEmitter(raw_ostream &OS, StringRef Name)
      : Name(Name.str()), OS(OS) {
    OS << "#ifndef " << Name << "\n"
       << "#define " << Name << "\n\n";
  }
  ~IncludeGuardEmitter() { OS << "\n#endif // " << Name << "\n\n"; }
 
private:
  std::string Name;
  raw_ostream &OS;
};
 
// Simple RAII helper for emitting namespace scope. Name can be a single
// namespace or nested namespace. If the name is empty, will not generate any
// namespace scope.
class NamespaceEmitter {
public:
  NamespaceEmitter(raw_ostream &OS, const Twine &NameUntrimmed)
      : Name(trim(NameUntrimmed.str()).str()), OS(OS) {
    if (!Name.empty())
      OS << "namespace " << Name << " {\n\n";
  }
 
  ~NamespaceEmitter() {
    if (!Name.empty())
      OS << "\n} // namespace " << Name << "\n";
  }
 
private:
  // Trim "::" prefix. If the namespace specified is ""::mlir::toy", then the
  // generated namespace scope needs to use
  //
  // namespace mlir::toy {
  // }
  //
  // and cannot use "namespace ::mlir::toy".
  static StringRef trim(StringRef Name) {
    Name.consume_front("::");
    return Name;
  }
 
  std::string Name;
  raw_ostream &OS;
};
 
// Simple RAII helper for emitting anonymous namespace scope.
class AnonNamespaceEmitter {
public:
  AnonNamespaceEmitter(raw_ostream &OS) : OS(OS) { OS << "namespace {\n\n"; }
  ~AnonNamespaceEmitter() { OS << "} // namespace\n"; }
 
private:
  raw_ostream &OS;
};
 
} // end namespace llvm
 
#endif // LLVM_TABLEGEN_CODEGENHELPERS_H