* Don't convert variable names to strings.

This commit is contained in:
Eelco Dolstra 2010-03-25 13:10:04 +00:00
parent f450384ded
commit 3c9f8fc9b6
1 changed files with 15 additions and 12 deletions

View File

@ -12,7 +12,9 @@ using namespace nix;
struct Env; struct Env;
struct Value; struct Value;
typedef std::map<string, Value> Bindings; typedef ATerm Sym;
typedef std::map<Sym, Value> Bindings;
struct Env struct Env
@ -96,7 +98,7 @@ void forceValue(Value & v)
} }
Value * lookupVar(Env * env, const string & name) Value * lookupVar(Env * env, Sym name)
{ {
for ( ; env; env = env->up) { for ( ; env; env = env->up) {
Bindings::iterator i = env->bindings.find(name); Bindings::iterator i = env->bindings.find(name);
@ -121,9 +123,9 @@ static void eval(Env * env, Expr e, Value & v)
{ {
printMsg(lvlError, format("eval: %1%") % e); printMsg(lvlError, format("eval: %1%") % e);
ATerm name; Sym name;
if (matchVar(e, name)) { if (matchVar(e, name)) {
Value * v2 = lookupVar(env, aterm2String(name)); Value * v2 = lookupVar(env, name);
forceValue(*v2); forceValue(*v2);
v = *v2; v = *v2;
return; return;
@ -143,7 +145,7 @@ static void eval(Env * env, Expr e, Value & v)
ATerm e2, pos; ATerm e2, pos;
for (ATermIterator i(es); i; ++i) { for (ATermIterator i(es); i; ++i) {
if (!matchBind(*i, name, e2, pos)) abort(); /* can't happen */ if (!matchBind(*i, name, e2, pos)) abort(); /* can't happen */
Value & v2 = (*v.attrs)[aterm2String(name)]; Value & v2 = (*v.attrs)[name];
nrValues++; nrValues++;
v2.type = tThunk; v2.type = tThunk;
v2.thunk.env = env; v2.thunk.env = env;
@ -162,7 +164,7 @@ static void eval(Env * env, Expr e, Value & v)
ATerm name, e2, pos; ATerm name, e2, pos;
for (ATermIterator i(rbnds); i; ++i) { for (ATermIterator i(rbnds); i; ++i) {
if (!matchBind(*i, name, e2, pos)) abort(); /* can't happen */ if (!matchBind(*i, name, e2, pos)) abort(); /* can't happen */
Value & v2 = env2->bindings[aterm2String(name)]; Value & v2 = env2->bindings[name];
nrValues++; nrValues++;
v2.type = tThunk; v2.type = tThunk;
v2.thunk.env = env2; v2.thunk.env = env2;
@ -176,7 +178,7 @@ static void eval(Env * env, Expr e, Value & v)
if (matchSelect(e, e2, name)) { if (matchSelect(e, e2, name)) {
eval(env, e2, v); eval(env, e2, v);
if (v.type != tAttrs) throw TypeError("expected attribute set"); if (v.type != tAttrs) throw TypeError("expected attribute set");
Bindings::iterator i = v.attrs->find(aterm2String(name)); Bindings::iterator i = v.attrs->find(name);
if (i == v.attrs->end()) throw TypeError("attribute not found"); if (i == v.attrs->end()) throw TypeError("attribute not found");
forceValue(i->second); forceValue(i->second);
v = i->second; v = i->second;
@ -203,7 +205,7 @@ static void eval(Env * env, Expr e, Value & v)
ATermList formals; ATerm ellipsis; ATermList formals; ATerm ellipsis;
if (matchVarPat(v.lambda.pat, name)) { if (matchVarPat(v.lambda.pat, name)) {
Value & vArg = env2->bindings[aterm2String(name)]; Value & vArg = env2->bindings[name];
vArg.type = tThunk; vArg.type = tThunk;
vArg.thunk.env = env; vArg.thunk.env = env;
vArg.thunk.expr = arg; vArg.thunk.expr = arg;
@ -216,7 +218,7 @@ static void eval(Env * env, Expr e, Value & v)
if (name == sNoAlias) if (name == sNoAlias)
vArg = &vArg_; vArg = &vArg_;
else else
vArg = &env2->bindings[aterm2String(name)]; vArg = &env2->bindings[name];
eval(env, arg, *vArg); eval(env, arg, *vArg);
if (vArg->type != tAttrs) throw TypeError("expected attribute set"); if (vArg->type != tAttrs) throw TypeError("expected attribute set");
@ -226,13 +228,13 @@ static void eval(Env * env, Expr e, Value & v)
argument has a default, use the default. */ argument has a default, use the default. */
unsigned int attrsUsed = 0; unsigned int attrsUsed = 0;
for (ATermIterator i(formals); i; ++i) { for (ATermIterator i(formals); i; ++i) {
Expr name, def; Expr def; Sym name;
DefaultValue def2; DefaultValue def2;
if (!matchFormal(*i, name, def2)) abort(); /* can't happen */ if (!matchFormal(*i, name, def2)) abort(); /* can't happen */
Bindings::iterator j = vArg->attrs->find(aterm2String(name)); Bindings::iterator j = vArg->attrs->find(name);
Value & v = env2->bindings[aterm2String(name)]; Value & v = env2->bindings[name];
if (j == vArg->attrs->end()) { if (j == vArg->attrs->end()) {
if (!matchDefaultValue(def2, def)) def = 0; if (!matchDefaultValue(def2, def)) def = 0;
@ -296,6 +298,7 @@ void run(Strings args)
doTest("({x, y, ...}: x) { x = 1; y = 2; z = 3; }"); doTest("({x, y, ...}: x) { x = 1; y = 2; z = 3; }");
doTest("({x, y, ...}@args: args.z) { x = 1; y = 2; z = 3; }"); doTest("({x, y, ...}@args: args.z) { x = 1; y = 2; z = 3; }");
//doTest("({x ? y, y ? x}: y) { }"); //doTest("({x ? y, y ? x}: y) { }");
doTest("let x = 1; in x");
printMsg(lvlError, format("alloced %1% values") % nrValues); printMsg(lvlError, format("alloced %1% values") % nrValues);
printMsg(lvlError, format("alloced %1% environments") % nrEnvs); printMsg(lvlError, format("alloced %1% environments") % nrEnvs);