summary refs log tree commit diff stats
path: root/nim/semtempl.pas
blob: c07a7bd13a29b2fa3a36cedd72d9e9272dd075ff (plain) (blame)
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
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
//
//
//           The Nimrod Compiler
//        (c) Copyright 2008 Andreas Rumpf
//
//    See the file "copying.txt", included in this
//    distribution, for details about the copyright.
//

function isExpr(n: PNode): bool;
// returns true if ``n`` looks like an expression
var
  i: int;
begin
  if n = nil then begin result := false; exit end;
  case n.kind of
    nkIdent..nkNilLit: result := true;
    nkCall..nkPassAsOpenArray: begin
      for i := 0 to sonsLen(n)-1 do
        if not isExpr(n.sons[i]) then begin
          result := false; exit
        end;
      result := true
    end
    else result := false
  end
end;

function isTypeDesc(n: PNode): bool;
// returns true if ``n`` looks like a type desc
var
  i: int;
begin
  if n = nil then begin result := false; exit end;
  case n.kind of
    nkIdent, nkSym, nkType: result := true;
    nkDotExpr, nkQualified, nkBracketExpr: begin
      for i := 0 to sonsLen(n)-1 do
        if not isTypeDesc(n.sons[i]) then begin
          result := false; exit
        end;
      result := true
    end;
    nkTypeOfExpr..nkEnumTy: result := true;
    else result := false
  end
end;

function evalTemplateAux(c: PContext; templ, actual: PNode;
                         sym: PSym): PNode;
var
  i: int;
begin
  if templ = nil then begin result := nil; exit end;
  case templ.kind of
    nkSym: begin
      if (templ.sym.kind = skParam) and (templ.sym.owner.id = sym.id) then
        result := copyTree(actual.sons[templ.sym.position+1]) // BUGFIX: +1
      else
        result := copyNode(templ)
    end;
    nkNone..nkIdent, nkType..nkNilLit: // atom
      result := copyNode(templ);
    else begin
      result := copyNode(templ);
      newSons(result, sonsLen(templ));
      for i := 0 to sonsLen(templ)-1 do
        result.sons[i] := evalTemplateAux(c, templ.sons[i], actual, sym);
    end
  end
end;

var
  evalTemplateCounter: int = 0; // to prevend endless recursion in templates
                                // instantation

function evalTemplate(c: PContext; n: PNode; sym: PSym): PNode;
var
  r: PNode;
begin
  inc(evalTemplateCounter);
  if evalTemplateCounter > 100 then
    liMessage(n.info, errTemplateInstantiationTooNested);
  // replace each param by the corresponding node:
  r := sym.ast.sons[paramsPos].sons[0];
  if (r.kind <> nkIdent) then InternalError(r.info, 'evalTemplate');
  result := evalTemplateAux(c, sym.ast.sons[codePos], n, sym);
  if r.ident.id = ord(wExpr) then result := semExpr(c, result)
  else result := semStmt(c, result);
  dec(evalTemplateCounter);
end;

function resolveTemplateParams(c: PContext; n: PNode): PNode;
var
  i: int;
  s: PSym;
begin
  if n = nil then begin result := nil; exit end;
  case n.kind of
    nkIdent: begin
      s := SymTabLocalGet(c.Tab, n.ident);
      if (s <> nil) then begin
        result := newSymNode(s);
        result.info := n.info
      end
      else
        result := n
    end;
    nkSym..nkNilLit: // atom
      result := n;
    else begin
      result := n;
      for i := 0 to sonsLen(n)-1 do
        result.sons[i] := resolveTemplateParams(c, n.sons[i]);
    end
  end
end;

function semTemplateParamKind(c: PContext; n, p: PNode): PNode;
begin
  if (p = nil) or (p.kind <> nkIdent) then
    liMessage(n.info, errInvalidParamKindX, renderTree(p))
  else begin
    case p.ident.id of
      ord(wExpr), ord(wStmt), ord(wTypeDesc): begin end;
      else
        liMessage(p.info, errInvalidParamKindX, p.ident.s)
    end
  end;
  result := p;
end;

function transformToExpr(n: PNode): PNode;
var
  i, realStmt: int;
begin
  result := n;
  case n.kind of
    nkStmtList: begin
      realStmt := -1;
      for i := 0 to sonsLen(n)-1 do begin
        case n.sons[i].kind of
          nkCommentStmt, nkEmpty, nkNilLit: begin end;
          else begin
            if realStmt = -1 then realStmt := i
            else realStmt := -2
          end
        end
      end;
      if realStmt >= 0 then
        result := transformToExpr(n.sons[realStmt])
      else
        n.kind := nkStmtListExpr;
    end;
    nkBlockStmt: n.kind := nkBlockExpr;
    //nkIfStmt: n.kind := nkIfExpr; // this is not correct!
    else begin end
  end
end;

function semTemplateDef(c: PContext; n: PNode): PNode;
var
  s, param: PSym;
  i, j, len, counter: int;
  params, p, paramKind: PNode;
begin
  if c.p.owner = nil then begin
    s := semIdentVis(c, skTemplate, n.sons[0], {@set}[sfStar]);
    include(s.flags, sfGlobal);
  end
  else
    s := semIdentVis(c, skTemplate, n.sons[0], {@set}[]);
  if sfStar in s.flags then include(s.flags, sfInInterface);
  // check parameter list:
  pushOwner(s);
  openScope(c.tab);
  params := n.sons[paramsPos];
  counter := 0;
  for i := 1 to sonsLen(params)-1 do begin
    p := params.sons[i];
    len := sonsLen(p);
    paramKind := semTemplateParamKind(c, p, p.sons[len-2]);
    if (p.sons[len-1] <> nil) then
      liMessage(p.sons[len-1].info, errDefaultArgumentInvalid);
    for j := 0 to len-3 do begin
      param := newSymS(skParam, p.sons[j], c);
      param.position := counter;
      param.ast := paramKind;
      inc(counter);
      addDecl(c, param)
    end;
  end;
  params.sons[0] := semTemplateParamKind(c, params, params.sons[0]);
  n.sons[namePos] := newSymNode(s);

  // check that no pragmas exist:
  if n.sons[pragmasPos] <> nil then
    liMessage(n.info, errNoPragmasAllowedForX, 'template');
  // check that no generic parameters exist:
  if n.sons[genericParamsPos] <> nil then
    liMessage(n.info, errNoGenericParamsAllowedForX, 'template');
  // resolve parameters:
  n.sons[codePos] := resolveTemplateParams(c, n.sons[codePos]);
  if params.sons[0].ident.id = ord(wExpr) then
    n.sons[codePos] := transformToExpr(n.sons[codePos]);

  // only parameters are resolved, no type checking is performed
  closeScope(c.tab);
  popOwner();
  s.ast := n;

  result := n;
  if n.sons[codePos] = nil then
    liMessage(n.info, errImplOfXexpected, s.name.s);
  // add identifier of template as a last step to not allow
  // recursive templates
  addInterfaceDecl(c, s);
end;