Skip to content

Commit

Permalink
fixes nim-lang#12
Browse files Browse the repository at this point in the history
  • Loading branch information
Araq committed May 2, 2011
1 parent 07d4a27 commit d8f42bf
Show file tree
Hide file tree
Showing 3 changed files with 54 additions and 61 deletions.
101 changes: 43 additions & 58 deletions compiler/cgmeth.nim
Original file line number Diff line number Diff line change
Expand Up @@ -13,13 +13,10 @@ import
options, ast, astalgo, msgs, idents, renderer, types, magicsys

proc genConv(n: PNode, d: PType, downcast: bool): PNode =
var
dest, source: PType
diff: int
dest = skipTypes(d, abstractPtrs)
source = skipTypes(n.typ, abstractPtrs)
var dest = skipTypes(d, abstractPtrs)
var source = skipTypes(n.typ, abstractPtrs)
if (source.kind == tyObject) and (dest.kind == tyObject):
diff = inheritanceDiff(dest, source)
var diff = inheritanceDiff(dest, source)
if diff == high(int): InternalError(n.info, "cgmeth.genConv")
if diff < 0:
result = newNodeIT(nkObjUpConv, n.info, d)
Expand All @@ -36,25 +33,23 @@ proc genConv(n: PNode, d: PType, downcast: bool): PNode =
result = n

proc methodCall*(n: PNode): PNode =
var disp: PSym
result = n
disp = lastSon(result.sons[0].sym.ast).sym
var disp = lastSon(result.sons[0].sym.ast).sym
result.sons[0].sym = disp
for i in countup(1, sonsLen(result) - 1):
result.sons[i] = genConv(result.sons[i], disp.typ.sons[i], true)

var gMethods: seq[TSymSeq]

proc sameMethodBucket(a, b: PSym): bool =
var aa, bb: PType
result = false
if a.name.id != b.name.id: return
if sonsLen(a.typ) != sonsLen(b.typ):
return # check for return type:
if not sameTypeOrNil(a.typ.sons[0], b.typ.sons[0]): return
for i in countup(1, sonsLen(a.typ) - 1):
aa = a.typ.sons[i]
bb = b.typ.sons[i]
var aa = a.typ.sons[i]
var bb = b.typ.sons[i]
while true:
aa = skipTypes(aa, {tyGenericInst})
bb = skipTypes(bb, {tyGenericInst})
Expand All @@ -71,18 +66,15 @@ proc sameMethodBucket(a, b: PSym): bool =
return
result = true

proc methodDef*(s: PSym) =
var
L, q: int
disp: PSym
L = len(gMethods)
proc methodDef*(s: PSym) =
var L = len(gMethods)
for i in countup(0, L - 1):
if sameMethodBucket(gMethods[i][0], s):
add(gMethods[i], s) # store a symbol to the dispatcher:
addSon(s.ast, lastSon(gMethods[i][0].ast))
return
add(gMethods, @[s]) # create a new dispatcher:
disp = copySym(s)
var disp = copySym(s)
disp.typ = copyType(disp.typ, disp.typ.owner, false)
if disp.typ.callConv == ccInline: disp.typ.callConv = ccDefault
disp.ast = copyTree(s.ast)
Expand All @@ -92,95 +84,88 @@ proc methodDef*(s: PSym) =
addSon(s.ast, newSymNode(disp))

proc relevantCol(methods: TSymSeq, col: int): bool =
var t: PType
# returns true iff the position is relevant
t = methods[0].typ.sons[col]
var t = methods[0].typ.sons[col]
result = false
if skipTypes(t, skipPtrs).kind == tyObject:
for i in countup(1, high(methods)):
if not SameType(methods[i].typ.sons[col], t):
return true

proc cmpSignatures(a, b: PSym, relevantCols: TIntSet): int =
var
d: int
aa, bb: PType
result = 0
for col in countup(1, sonsLen(a.typ) - 1):
if intSetContains(relevantCols, col):
aa = skipTypes(a.typ.sons[col], skipPtrs)
bb = skipTypes(b.typ.sons[col], skipPtrs)
d = inheritanceDiff(aa, bb)
var aa = skipTypes(a.typ.sons[col], skipPtrs)
var bb = skipTypes(b.typ.sons[col], skipPtrs)
var d = inheritanceDiff(aa, bb)
if (d != high(int)):
return d

proc sortBucket(a: var TSymSeq, relevantCols: TIntSet) =
# we use shellsort here; fast and simple
var
N, j, h: int
v: PSym
N = len(a)
h = 1
var N = len(a)
var h = 1
while true:
h = 3 * h + 1
if h > N: break
while true:
h = h div 3
for i in countup(h, N - 1):
v = a[i]
j = i
var v = a[i]
var j = i
while cmpSignatures(a[j - h], v, relevantCols) >= 0:
a[j] = a[j - h]
j = j - h
if j < h: break
a[j] = v
if h == 1: break

proc genDispatcher(methods: TSymSeq, relevantCols: TIntSet): PSym =
var
disp, cond, call, ret, a, isn: PNode
base, curr, ands, iss: PSym
paramLen: int
base = lastSon(methods[0].ast).sym
proc genDispatcher(methods: TSymSeq, relevantCols: TIntSet): PSym =
var base = lastSon(methods[0].ast).sym
result = base
paramLen = sonsLen(base.typ)
disp = newNodeI(nkIfStmt, base.info)
ands = getSysSym("and")
iss = getSysSym("is")
for meth in countup(0, high(methods)):
curr = methods[meth] # generate condition:
cond = nil
for col in countup(1, paramLen - 1):
if IntSetContains(relevantCols, col):
isn = newNodeIT(nkCall, base.info, getSysType(tyBool))
var paramLen = sonsLen(base.typ)
var disp = newNodeI(nkIfStmt, base.info)
var ands = getSysSym("and")
var iss = getSysSym("is")
for meth in countup(0, high(methods)):
var curr = methods[meth] # generate condition:
var cond: PNode = nil
for col in countup(1, paramLen - 1):
if IntSetContains(relevantCols, col):
var isn = newNodeIT(nkCall, base.info, getSysType(tyBool))
addSon(isn, newSymNode(iss))
addSon(isn, newSymNode(base.typ.n.sons[col].sym))
addSon(isn, newNodeIT(nkType, base.info, curr.typ.sons[col]))
if cond != nil:
a = newNodeIT(nkCall, base.info, getSysType(tyBool))
var a = newNodeIT(nkCall, base.info, getSysType(tyBool))
addSon(a, newSymNode(ands))
addSon(a, cond)
addSon(a, isn)
cond = a
else:
else:
cond = isn
call = newNodeI(nkCall, base.info)
var call = newNodeI(nkCall, base.info)
addSon(call, newSymNode(curr))
for col in countup(1, paramLen - 1):
addSon(call, genConv(newSymNode(base.typ.n.sons[col].sym),
curr.typ.sons[col], false))
if base.typ.sons[0] != nil:
a = newNodeI(nkAsgn, base.info)
var ret: PNode
if base.typ.sons[0] != nil:
var a = newNodeI(nkAsgn, base.info)
addSon(a, newSymNode(base.ast.sons[resultPos].sym))
addSon(a, call)
ret = newNodeI(nkReturnStmt, base.info)
addSon(ret, a)
else:
else:
ret = call
a = newNodeI(nkElifBranch, base.info)
addSon(a, cond)
addSon(a, ret)
addSon(disp, a)
if cond != nil:
var a = newNodeI(nkElifBranch, base.info)
addSon(a, cond)
addSon(a, ret)
addSon(disp, a)
else:
disp = ret
result.ast.sons[codePos] = disp

proc generateMethodDispatchers*(): PNode =
Expand Down
10 changes: 10 additions & 0 deletions tests/accept/run/tmethods1.nim
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@
discard """
output: "do nothing"
"""

method somethin(obj: TObject) =
echo "do nothing"

var o: TObject
o.somethin()

4 changes: 1 addition & 3 deletions todo.txt
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@
High priority (version 0.9.0)
=============================

- tests: run modules that contain "when isMainModule", compile the other
- tests: run modules that contain "#RUN_ME", compile the other
modules; run the GC tests
- fix implicit generic routines
- fix the streams implementation so that it uses methods
Expand Down Expand Up @@ -44,8 +44,6 @@ Low priority
------------
- resizing of strings/sequences could take into account the memory that
is allocated
- implicit conversions from ``ptr/ref T`` to ``var T`` (from
``ptr/ref T`` to ``T``)?
- typeAllowed() for parameters...
- find a way to reintroduce the cleanup() pass for C code generation: this
is hard because of partial evaluation --> symbol files will fix this as
Expand Down

0 comments on commit d8f42bf

Please sign in to comment.