[Heavy] Add scheme lib to emit c++ from nbdl dialect (import (nbdl comp))

Author: ricejasonf

heavy/include/heavy/Nbdl.h

@@ -0,0 +1,33 @@
+//===- Nbdl.h - Nbdl binding functions for HeavyScheme ----------*- C++ -*-===//
+//
+// 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 declares values and functions for the (nbdl comp) scheme library
+//
+//===----------------------------------------------------------------------===//
+
+#ifndef LLVM_HEAVY_NBDL_H
+#define LLVM_HEAVY_NBDL_H
+
+#define HEAVY_NBDL_LIB                _HEAVYL4SnbdlL4Scomp
+#define HEAVY_NBDL_LIB_(NAME)         _HEAVYL4SnbdlL4Scomp ## NAME
+// (import (nbdl comp))
+#define HEAVY_NBDL_LIB_STR            "_HEAVYL4SnbdlL4Scomp"
+#define HEAVY_NBDL_LOAD_MODULE        HEAVY_NBDL_LIB_(_load_module)
+#define HEAVY_NBDL_INIT               HEAVY_NBDL_LIB_(_init)
+
+namespace heavy {
+class Context;
+}
+
+extern "C" {
+// initialize the module for run-time independent of the compiler
+void HEAVY_NBDL_LOAD_MODULE(heavy::Context& Context);
+void HEAVY_NBDL_INIT(heavy::Context& Context);
+}
+
+#endif  // LLVM_HEAVY_NBDL_H

heavy/include/nbdl_gen/Dialect.h

@@ -13,12 +13,19 @@
 #ifndef NBDL_GEN_DIALECT_H
 #define NBDL_GEN_DIALECT_H
 
-#include "mlir/Dialect/Func/IR/FuncOps.h"
-#include "mlir/IR/BuiltinOps.h"
-#include "mlir/IR/Dialect.h"
-#include "mlir/IR/OpDefinition.h"
-#include "mlir/Interfaces/ControlFlowInterfaces.h"
-#include "mlir/Interfaces/SideEffectInterfaces.h"
+#include <mlir/Dialect/Func/IR/FuncOps.h>
+#include <mlir/IR/BuiltinOps.h>
+#include <mlir/IR/Dialect.h>
+#include <mlir/IR/OpDefinition.h>
+#include <mlir/Interfaces/ControlFlowInterfaces.h>
+#include <mlir/Interfaces/SideEffectInterfaces.h>
+
+namespace nbdl_gen {
+using mlir::func::CallOp;
+using mlir::func::FuncOp;
+using mlir::func::ReturnOp;
+using mlir::StringAttr;
+}
 
 // Include the generated header files
 

heavy/include/nbdl_gen/Nbdl.td

@@ -14,6 +14,7 @@ include "mlir/IR/SymbolInterfaces.td"
 
 def Nbdl_Dialect : Dialect {
   let name = "nbdl";
+  let cppNamespace = "nbdl_gen";
   let useDefaultTypePrinterParser = 1;
 }
 

heavy/lib/Builtins.cpp

@@ -12,6 +12,7 @@
 
 #include "heavy/Builtins.h"
 #include "heavy/Context.h"
+#include "heavy/Dialect.h"
 #include "heavy/OpGen.h"
 #include "heavy/Value.h"
 #include "llvm/ADT/StringExtras.h"
@@ -812,6 +813,7 @@ void is_source_value(Context& C, ValueRefs Args) {
 
 // initialize the module for run-time independent of the compiler
 void HEAVY_BASE_INIT(heavy::Context& Context) {
+  Context.DialectRegistry->insert<heavy::Dialect>();
   // syntax
   HEAVY_BASE_VAR(define)          = heavy::base::define;
   HEAVY_BASE_VAR(define_syntax)   = heavy::base::define_syntax;

heavy/lib/CMakeLists.txt

@@ -15,7 +15,9 @@ add_heavy_library(heavy
   Quasiquote.cpp
   SourceFileStorage.cpp
   SourceManager.cpp
+  Nbdl.cpp
   NbdlDialect.cpp
+  NbdlWriter.cpp
   # TODO Do not use Clangs CharInfo.
   ${CLANG_LIB_DIR}/Basic/CharInfo.cpp
 

heavy/lib/Context.cpp

@@ -16,6 +16,7 @@
 #include "heavy/Lexer.h"
 #include "heavy/Mangle.h"
 #include "heavy/Mlir.h"
+#include "heavy/Nbdl.h"
 #include "heavy/OpGen.h"
 #include "heavy/Parser.h"
 #include "heavy/Source.h"
@@ -85,6 +86,7 @@ Context::Context()
                   _HEAVY_load_module);
   RegisterModule(HEAVY_BASE_LIB_STR, HEAVY_BASE_LOAD_MODULE);
   RegisterModule(HEAVY_MLIR_LIB_STR, HEAVY_MLIR_LOAD_MODULE);
+  RegisterModule(HEAVY_NBDL_LIB_STR, HEAVY_NBDL_LOAD_MODULE);
   HEAVY_BASE_INIT(*this);
 }
 

heavy/lib/Mlir.cpp

@@ -10,7 +10,6 @@
 //
 //===----------------------------------------------------------------------===//
 
-#include <nbdl_gen/Dialect.h>
 #include <heavy/Context.h>
 #include <heavy/Dialect.h>
 #include <heavy/Mlir.h>
@@ -775,11 +774,6 @@ void verify(Context& C, heavy::ValueRefs Args) {
 extern "C" {
 // initialize the module for run-time independent of the compiler
 void HEAVY_MLIR_INIT(heavy::Context& C) {
-  // TODO Register dialects in their corresponding
-  //      scheme modules instead of here.
-  C.DialectRegistry->insert<heavy::Dialect,
-                            nbdl::NbdlDialect>();
-
   mlir::MLIRContext* MC = C.MLIRContext.get();
   heavy::Value MC_Val = CreateTagged(C, kind::mlir_context, MC);
   heavy::Value BuilderVal = CreateTagged(C, kind::mlir_builder,

heavy/lib/Nbdl.cpp

@@ -0,0 +1,72 @@
+//===--- Nbdl.cpp - Nbdl binding syntax for HeavyScheme ---------*- C++ -*-===//
+//
+// 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 syntax (nbdl impl) bindings for HeavyScheme.
+//
+//===----------------------------------------------------------------------===//
+
+#include <heavy/Context.h>
+#include <heavy/Nbdl.h>
+#include <heavy/Value.h>
+#include <nbdl_gen/Dialect.h>
+#include <llvm/Support/Casting.h>
+#include <memory>
+#include <tuple>
+
+namespace nbdl_gen {
+std::tuple<std::string, mlir::Location, mlir::Operation*>
+translate_cpp(llvm::raw_ostream& OS, mlir::Operation* Op);
+}
+
+namespace heavy::nbdl_bind_var {
+heavy::ExternFunction write_cpp;
+}
+
+namespace heavy::nbdl_bind {
+// Translate a nbdl dialect operation to C++.
+void write_cpp(Context& C, ValueRefs Args) {
+  if (Args.size() != 2)
+    return C.RaiseError("invalid arity");
+  auto* Op = dyn_cast<mlir::Operation>(Args[0]);
+  // Do not captured the emphemeral Tagged object.
+  auto* Tagged = dyn_cast<heavy::Tagged>(Args[1]);
+  heavy::Symbol* KindSym = C.CreateSymbol("::llvm::raw_ostream");
+  if (!Op)
+    return C.RaiseError("expecting mlir.operation");
+  if (!Tagged || !Tagged->isa(KindSym))
+    return C.RaiseError("expecting ::llvm::raw_ostream");
+
+  auto& OS = Tagged->cast<llvm::raw_ostream>();
+  auto&& [ErrMsg, ErrLoc, Irritant] = nbdl_gen::translate_cpp(OS, Op);
+  if (!ErrMsg.empty()) {
+    auto Loc = heavy::SourceLocation(mlir::OpaqueLoc
+      ::getUnderlyingLocationOrNull<heavy::SourceLocationEncoding*>(
+        mlir::dyn_cast<mlir::OpaqueLoc>(ErrLoc)));
+    heavy::Error* Err = C.CreateError(Loc, ErrMsg, 
+        Irritant ? heavy::Value(Irritant) : Undefined());
+    return C.Raise(Err);
+  }
+  C.Cont();
+}
+}
+
+extern "C" {
+// initialize the module for run-time independent of the compiler
+void HEAVY_NBDL_INIT(heavy::Context& C) {
+  C.DialectRegistry->insert<nbdl_gen::NbdlDialect>();
+
+  heavy::nbdl_bind_var::write_cpp = heavy::nbdl_bind::write_cpp;
+}
+
+void HEAVY_NBDL_LOAD_MODULE(heavy::Context& C) {
+  HEAVY_NBDL_INIT(C);
+  heavy::initModuleNames(C, HEAVY_NBDL_LIB_STR, {
+    {"write-cpp", heavy::nbdl_bind_var::write_cpp},
+  });
+}
+}

heavy/lib/NbdlDialect.cpp

@@ -26,7 +26,7 @@
 #define GET_OP_CLASSES
 #include "nbdl_gen/NbdlOps.cpp.inc"
 
-void nbdl::NbdlDialect::initialize() {
+void nbdl_gen::NbdlDialect::initialize() {
   addTypes<
 #define GET_TYPEDEF_LIST
 #include "nbdl_gen/NbdlTypes.cpp.inc"