Ejemplo n.º 1
0
Function* Backend::compile(const Stmt& stmt, vector<ir::Var> output){
  Environment env;

  std::set<Var> externSet;
  for (const Var& var : output) {
    if (!util::contains(externSet, var)) {  // Remove duplicates
      externSet.insert(var);
      env.addExtern(var);
    }
  }

  return compile(stmt, env, Storage());
}
Ejemplo n.º 2
0
backend::Function* Backend::compile(const Stmt& stmt, const Environment& env,
                                    const Storage& storage) {
  Environment environment = env; // copy

  // Add output and undefined variable in the stmt to the environment as externs
  std::set<Var> globalsSet;
  util::ScopedMap<Var,void*> definedVariables;

  // Add all environment variables to defined variables
  for (auto& var : environment.getConstants()) {
    globalsSet.insert(var.first);
    definedVariables.insert(var.first, nullptr);
  }
  for (auto& ext : environment.getExterns()) {
    const Var& var = ext.getVar();
    globalsSet.insert(var);
    definedVariables.insert(var, nullptr);
  }
  for (auto& tmp : environment.getTemporaries()) {
    globalsSet.insert(tmp);
    definedVariables.insert(tmp, nullptr);
  }

  // Find all undefined variables and add them to the context as globals.
  function<void(Var var)> addToGlobalsIfNotDefined = [&](Var var) {
    if (!definedVariables.contains(var) && !util::contains(globalsSet, var)) {
      environment.addExtern(var);
      globalsSet.insert(var);
      definedVariables.insert(var, nullptr);
    }
  };

  match(stmt,
    function<void(const VarExpr*)>([&](const VarExpr* op) {
      addToGlobalsIfNotDefined(op->var);
    })
    ,
    function<void(const AssignStmt*)>([&](const AssignStmt* op) {
      definedVariables.insert(op->var, nullptr);
    })
    ,
    function<void(const VarDecl*)>([&](const VarDecl* op) {
      definedVariables.insert(op->var, nullptr);
    })
    ,
    function<void(const ForRange*,Matcher*)>([&](const ForRange* op, Matcher* ctx) {
      ctx->match(op->start);
      ctx->match(op->end);
      definedVariables.insert(op->var, nullptr);
      ctx->match(op->body);
    })
    ,
    function<void(const For*,Matcher*)>([&](const For* op, Matcher* ctx) {
      definedVariables.insert(op->var, nullptr);
      ctx->match(op->body);
    })
    ,
    function<void(const Scope*,Matcher*)>([&](const Scope* op, Matcher* ctx) {
      definedVariables.scope();
      ctx->match(op->scopedStmt);
      definedVariables.unscope();
    })
  );

  Func func("main", {}, {}, stmt, environment);
  return pimpl->compile(func, storage);
}