util function for directives

loda-lang · Nov 3, 2024 · 6443327 · 6443327
1 parent 2d753bb
commit 6443327
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/src/lang/program.cpp b/src/lang/program.cpp
@@ -187,6 +187,12 @@ int64_t Program::getDirective(const std::string& name) const {
   return d->second;
 }
 
+int64_t Program::getDirective(const std::string& name,
+                              int64_t defaultValue) const {
+  auto d = directives.find(name);
+  return d != directives.end() ? d->second : defaultValue;
+}
+
 bool Program::operator==(const Program& p) const { return (ops == p.ops); }
 
 bool Program::operator!=(const Program& p) const { return !(*this == p); }

diff --git a/src/lang/program.hpp b/src/lang/program.hpp
@@ -127,6 +127,8 @@ class Program {
 
   int64_t getDirective(const std::string &name) const;
 
+  int64_t getDirective(const std::string &name, int64_t defaultValue) const;
+
   bool operator==(const Program &p) const;
 
   bool operator!=(const Program &p) const;