Generate Builder.foo for all math ops on load

parrot · Mar 8, 2011 · 4f7f1ed · 4f7f1ed
1 parent 3b39ab4
commit 4f7f1ed
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 0 deletions.
diff --git a/runtime/parrot/library/LLVM/Builder.pm b/runtime/parrot/library/LLVM/Builder.pm
@@ -193,6 +193,26 @@ class LLVM::Builder is LLVM::Opaque {
 #            LLVMBuildIsNotNull => "pppt",
 #            LLVMBuildPtrDiff => "ppppt",
 
+INIT {
+    my $HOW  := LLVM::Builder.HOW;
+    my $WHAT := LLVM::Builder.WHAT;
+
+    # Generate binary ops
+    for <Add NSWAdd FAdd Sub FSub Mul FMul UDiv SDiv ExactSDiv FDiv URem SRem FRem Shl LShr AShr And Or Xor> -> $name {
+        my $call    := "Build" ~ $name;
+        my $subname := pir::downcase($name);
+        $HOW.add_method(
+            $subname,
+            method (LLVM::Value $left, LLVM::Value $right, :$name?) {
+                LLVM::Value.create(
+                    LLVM::call($call, self, $left, $right, $name)
+                )
+            },
+            to => $WHAT
+        );
+    }
+}
+
 };
 
 # vim: ft=perl6
diff --git a/t/library/llvm.t b/t/library/llvm.t
@@ -42,6 +42,14 @@ ok( 1, "Call created with args");
 
 my $answer := LLVM::Constant::integer(42);
 ok( 1, "Constant created");
+
+# Simple math
+$answer := $builder.add(
+    LLVM::Constant::integer(40),
+    LLVM::Constant::integer(2)
+);
+ok( 1, "Simple add created");
+
 $builder.ret($answer);
 ok( 1, "return created");