Rearranging definitions in R

I came up with a handy little trick for programming in R.

I like to define a lot of variables all at once without worrying about what order they’re in. The goal would be something like this:

K  = -R2/R1
Wc = 1/(R2*C)
R1 = 4.7e+3
R2 = 4.7e+3
C  = 0.05e-9

But of course that doesn’t work because K can’t refer to R2 until R2 has been defined. Now a nice feature in R (or a horrific one, depending on your tastes) is that default arguments to functions are lazily evaluated and may refer to each other (they can even refer to local variables defined within the function!). So the above may be rewritten

Ckt = (function(
    K  = -R2/R1,
    Wc = 1/(R2*C),
    R1 = 4.7e+3,
    R2 = 4.7e+3,
    C  = 0.05e-9
) environment())()

And the variables can now be accessed as Ckt$K, Ckt$Wc, etc. This is often exactly what I want — it packages up these variables into Ckt so they won’t pollute the global environment. But sometimes I do want them to pollute the global environment, and not just by calling attach(). So how would you do that?

The solution I came up with is the rarely used formals() function. It gets or sets the formal parameters of a function. And I used it to build this friendly little contraption…

define = function(Defs) {
    foo = function() { environment() };
    formals(foo) = Defs;
    Parent = parent.frame();

    Results = foo()

    for (Name in ls(Results)) {
        Parent[[Name]] = Results[[Name]]
    }
}

Now the parallel definitions can be done as

define(alist(
    K  = -R2/R1,
    Wc = 1/(R2*C),
    R1 = 4.7e+3,
    R2 = 4.7e+3,
    C  = 0.05e-9
))

But it seems such a shame to lose the lazy-evaluation with the line “Parent[[Name]] = Results[[Name]]”. (I mean it doesn’t actually matter, but it was lazy up until that point!). Problem is what I wrote next didn’t work:

define = function(Defs) {
    foo = function() { environment() };
    formals(foo) = Defs;
    Parent = parent.frame();

    Results = foo()

    for (Name in ls(Results)) {
        delayedAssign(assign.env=Parent, x=Name, value=Results[[Name]])
    }
}

> print(K)
[1] 4255319

> print(Wc)
[1] 4255319

> print(R1)
[1] 4255319

> print(R2)
[1] 4255319

> print(C)
[1] 4255319

Can you figure out why? It’s very subtle.

Basically for loops (and loops in generally, including, irritatingly, lapply) in R do not create a local environment (aka closure). That means that the loop variable (‘Name’ here) gets overwritten each time. Since “Results[[Name]]” is lazily evaluated, it doesn’t have the value either, so they all take on the value of the last one.

The solution is to create a closure. This gives the final product.

define = function(Defs) {
    foo = function() { environment() };
    formals(foo) = Defs;
    Parent = parent.frame();

    Results = foo()

    for (Name in ls(Results)) (function(Name) {
        delayedAssign(assign.env=Parent, x=Name, value=Results[[Name]])
    }) (Name=Name)
}
About these ads
This entry was posted in R. Bookmark the permalink.

One Response to Rearranging definitions in R

  1. Pingback: Not Quite Functional | Struggling Through Problems

Leave a Reply

Fill in your details below or click an icon to log in:

WordPress.com Logo

You are commenting using your WordPress.com account. Log Out / Change )

Twitter picture

You are commenting using your Twitter account. Log Out / Change )

Facebook photo

You are commenting using your Facebook account. Log Out / Change )

Google+ photo

You are commenting using your Google+ account. Log Out / Change )

Connecting to %s