70 lines
2.2 KiB
C++
70 lines
2.2 KiB
C++
/* file LocalEnv.hpp
|
|
*
|
|
* author: Roland Conybeare, Jun 2024
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include "Environment.hpp"
|
|
|
|
namespace xo {
|
|
namespace ast {
|
|
/** @brief LocalEnv
|
|
*
|
|
* @class Local environment for a lambda.
|
|
* Lists the Variables corresponding to this lambda's formal
|
|
* parameters, but also links to @ref Environment for
|
|
* innermost enclosing @ref Lambda.
|
|
**/
|
|
class LocalEnv : public Environment {
|
|
public:
|
|
using TypeDescr = xo::reflect::TypeDescr;
|
|
|
|
public:
|
|
/** named ctor idiom. Create instance with local variables per @p argv **/
|
|
static ref::rp<LocalEnv> make(const std::vector<ref::rp<Variable>> & argv) {
|
|
return new LocalEnv(argv);
|
|
}
|
|
|
|
const std::vector<ref::rp<Variable>> & argv() const { return argv_; }
|
|
int n_arg() const { return argv_.size(); }
|
|
TypeDescr fn_arg(uint32_t i) const { return argv_[i]->valuetype(); }
|
|
|
|
/** single-assign this environment's parent **/
|
|
void assign_parent(ref::brw<Environment> p) {
|
|
assert(parent_env_.get() == nullptr);
|
|
parent_env_ = p.get();
|
|
}
|
|
|
|
// ----- Environment -----
|
|
|
|
virtual ref::brw<Variable> lookup_var(const std::string & target) const override {
|
|
for (const auto & arg : argv_) {
|
|
if (arg->name() == target)
|
|
return arg;
|
|
}
|
|
|
|
/* here: target not found in local vars,
|
|
* delegate to innermost ancestor
|
|
*/
|
|
return parent_env_->lookup_var(target);
|
|
}
|
|
|
|
private:
|
|
LocalEnv(const std::vector<ref::rp<Variable>> & argv)
|
|
: argv_(argv) {}
|
|
|
|
private:
|
|
/** formal argument names **/
|
|
std::vector<ref::rp<Variable>> argv_;
|
|
|
|
/** parent environment. Free variable in this lambda's
|
|
* body, will be resolved by referring them to @ref parent_env_.
|
|
**/
|
|
ref::rp<Environment> parent_env_;
|
|
};
|
|
} /*namespace ast*/
|
|
} /*namespace xo*/
|
|
|
|
|
|
/* end LocalEnv.hpp */
|