Add Try/catch (#193)

* Add Try

* Fix placement new

Author: wsmoses

tools/mlir-clang/Lib/CGStmt.cc

@@ -978,6 +978,12 @@ ValueCategory MLIRScanner::VisitGotoStmt(clang::GotoStmt *stmt) {
   return nullptr;
 }
 
+ValueCategory MLIRScanner::VisitCXXTryStmt(clang::CXXTryStmt *stmt) {
+  llvm::errs() << "warning, not performing catches for try: ";
+  stmt->dump();
+  return Visit(stmt->getTryBlock());
+}
+
 ValueCategory MLIRScanner::VisitReturnStmt(clang::ReturnStmt *stmt) {
   IfScope scope(*this);
   bool isArrayReturn = false;

tools/mlir-clang/Lib/clang-mlir.cc

@@ -1107,7 +1107,23 @@ ValueCategory MLIRScanner::VisitCXXNewExpr(clang::CXXNewExpr *expr) {
 
   mlir::Value alloc;
   mlir::Value arrayCons;
-  if (auto mt = ty.dyn_cast<mlir::MemRefType>()) {
+  if (!expr->placement_arguments().empty()) {
+    mlir::Value val = Visit(*expr->placement_arg_begin()).getValue(builder);
+    if (auto mt = ty.dyn_cast<mlir::MemRefType>()) {
+      arrayCons = alloc =
+          builder.create<polygeist::Pointer2MemrefOp>(loc, mt, val);
+    } else {
+      arrayCons = alloc = builder.create<mlir::LLVM::BitcastOp>(loc, ty, val);
+      auto PT = ty.cast<LLVM::LLVMPointerType>();
+      if (expr->isArray())
+        arrayCons = builder.create<mlir::LLVM::BitcastOp>(
+            loc,
+            LLVM::LLVMPointerType::get(
+                LLVM::LLVMArrayType::get(PT.getElementType(), 0),
+                PT.getAddressSpace()),
+            alloc);
+    }
+  } else if (auto mt = ty.dyn_cast<mlir::MemRefType>()) {
     auto shape = std::vector<int64_t>(mt.getShape());
     mlir::Value args[1] = {count};
     arrayCons = alloc = builder.create<mlir::memref::AllocOp>(loc, mt, args);
@@ -1116,7 +1132,7 @@ ValueCategory MLIRScanner::VisitCXXNewExpr(clang::CXXNewExpr *expr) {
     auto typeSize = getTypeSize(expr->getAllocatedType());
     mlir::Value args[1] = {builder.create<arith::MulIOp>(loc, typeSize, count)};
     args[0] = builder.create<IndexCastOp>(loc, i64, args[0]);
-    alloc = builder.create<mlir::LLVM::BitcastOp>(
+    arrayCons = alloc = builder.create<mlir::LLVM::BitcastOp>(
         loc, ty,
         builder
             .create<mlir::LLVM::CallOp>(loc, Glob.GetOrCreateMallocFunction(),

tools/mlir-clang/Lib/clang-mlir.h

@@ -249,6 +249,8 @@ class MLIRScanner : public StmtVisitor<MLIRScanner, ValueCategory> {
 
   ValueCategory VisitCXXTypeidExpr(clang::CXXTypeidExpr *expr);
 
+  ValueCategory VisitCXXTryStmt(clang::CXXTryStmt *stmt);
+
   ValueCategory VisitStringLiteral(clang::StringLiteral *expr);
 
   ValueCategory VisitParenExpr(clang::ParenExpr *expr);