summaryrefslogtreecommitdiff
path: root/MLKEnvironment.m
blob: 8a7c2706fd19d80637b5052eb74759f64c11ce29 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
/* -*- mode: objc; coding: utf-8 -*- */
/* Étoilisp/Mulklisp, a Common Lisp subset for the Étoilé runtime.
 * Copyright (C) 2008  Matthias Andreas Benkard.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or (at
 * your option) any later version.
 *
 * This program is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#import <Foundation/NSDictionary.h>
#import <Foundation/NSArray.h>

#import "MLKEnvironment.h"
#import "MLKUndefinedVariableException.h"


@implementation MLKEnvironment
-(MLKEnvironment *) init
{
  return [self initWithParent:nil bindings:nil];
}

-(MLKEnvironment *) initWithParent:(MLKEnvironment *)parent bindings:(NSDictionary *)bindings
{
  self = [super init];
  _bindings = [[NSMutableDictionary alloc] initWithCapacity:10];
  ASSIGN (_parent, parent);
  [self addBindings: bindings];
  return self;
}

-(MLKEnvironment *) initWithParent:(MLKEnvironment *)parent
{
  return [self initWithParent:parent bindings:nil];
}

-(MLKEnvironment *) initWithBindings:(NSDictionary *)bindings
{
  return [self initWithParent:nil bindings:bindings];
}

-(MLKEnvironment *) parent
{
  return _parent;
}

-(void) setBinding:(MLKSymbol *)symbol to:(id)value
{
  [self setBinding:symbol to:value inEnvironment:self];
}

-(void) setBinding:(MLKSymbol *)symbol to:(id)value inEnvironment:(MLKEnvironment *)env
{
  if ([[_bindings allKeys] containsObject:symbol])
    [_bindings setObject:value forKey:symbol];
  else
    if (_parent)
      [_parent setBinding:symbol to:value inEnvironment:env];
    else
      [[[MLKUndefinedVariableException alloc] initWithEnvironment:env
                                              variableName:symbol]
        raise];
}

-(id) valueForBinding:(MLKSymbol *)symbol
{
  return [self valueForBinding:symbol];
}

-(id) valueForBinding:(MLKSymbol *)symbol inEnvironment:(MLKEnvironment *)env
{
  if ([[_bindings allKeys] containsObject:symbol])
    return [_bindings objectForKey:symbol];
  else
    if (_parent)
      return [_parent valueForBinding:symbol];
    else
      [[[MLKUndefinedVariableException alloc] initWithEnvironment:env
                                              variableName:symbol]
        raise];
  
  return nil;  // avoid a stupid compiler warning
}

-(void) addBindings:(NSDictionary *)bindings
{
  [_bindings addEntriesFromDictionary:bindings];
}

-(void) addBinding:(MLKSymbol *)symbol to:(id)value
{
  [_bindings setObject:value forKey:symbol];
}

-(MLKEnvironment *) environmentForBinding:(MLKSymbol *)symbol
{
  if ([[_bindings allKeys] containsObject:symbol])
    return self;
  else if (_parent)
    return [_parent environmentForBinding:symbol];
  else
    return nil;
}

-(void) dealloc
{
  RELEASE (_bindings);
  RELEASE (_parent);
  [super dealloc];
}
@end