minor PrimOp code enhancements

sudison · Mar 30, 2013 · b65cd86 · b65cd86
1 parent 9313b1f
commit b65cd86
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 5 deletions.
diff --git a/compiler/loader.cpp b/compiler/loader.cpp
@@ -914,8 +914,8 @@ EnvPtr ForeignStatement::getEnv() {
 
 static map<int, string> primOpNames;
 
-llvm::StringRef primOpName(const PrimOpPtr& x) {
-    map<int, string>::iterator i = primOpNames.find(x->primOpCode);
+llvm::StringRef primOpName(PrimOpCode op) {
+    map<int, string>::iterator i = primOpNames.find(op);
     assert(i != primOpNames.end());
     return i->second;
 }

diff --git a/compiler/loader.hpp b/compiler/loader.hpp
@@ -232,12 +232,16 @@ enum PrimOpCode {
 };
 
 struct PrimOp : public Object {
-    int primOpCode;
-    PrimOp(int primOpCode)
+    const PrimOpCode primOpCode;
+    PrimOp(PrimOpCode primOpCode)
         : Object(PRIM_OP), primOpCode(primOpCode) {}
 };
 
-llvm::StringRef primOpName(const PrimOpPtr& x);
+llvm::StringRef primOpName(PrimOpCode op);
+
+static inline llvm::StringRef primOpName(const PrimOpPtr& x) {
+    return primOpName(x->primOpCode);
+}
 
 }