Skip to content

Commit

Permalink
fix nim-lang#9872: setLen now works properly at CT [backport]
Browse files Browse the repository at this point in the history
(cherry picked from commit f838b1e)
  • Loading branch information
timotheecour authored and narimiran committed Dec 15, 2018
1 parent 28946ed commit c16fa63
Show file tree
Hide file tree
Showing 2 changed files with 32 additions and 16 deletions.
18 changes: 2 additions & 16 deletions compiler/vm.nim
Expand Up @@ -413,26 +413,12 @@ proc recSetFlagIsRef(arg: PNode) =
arg.sons[i].recSetFlagIsRef

proc setLenSeq(c: PCtx; node: PNode; newLen: int; info: TLineInfo) =
# FIXME: this doesn't attempt to solve incomplete
# support of tyPtr, tyRef in VM.
let typ = node.typ.skipTypes(abstractInst+{tyRange}-{tyTypeDesc})
let typeEntry = typ.sons[0].skipTypes(abstractInst+{tyRange}-{tyTypeDesc})
let typeKind = case typeEntry.kind
of tyUInt..tyUInt64: nkUIntLit
of tyRange, tyEnum, tyBool, tyChar, tyInt..tyInt64: nkIntLit
of tyFloat..tyFloat128: nkFloatLit
of tyString: nkStrLit
of tyObject: nkObjConstr
of tySequence: nkNilLit
of tyProc, tyTuple: nkTupleConstr
else: nkEmpty

let oldLen = node.len
setLen(node.sons, newLen)
if oldLen < newLen:
# TODO: This is still not correct for tyPtr, tyRef default value
for i in oldLen ..< newLen:
node.sons[i] = newNodeI(typeKind, info)
node.sons[i] = getNullValue(typ.sons[0], info, c.config)

const
errIndexOutOfBounds = "index out of bounds"
Expand All @@ -458,7 +444,7 @@ proc rawExecute(c: PCtx, start: int, tos: PStackFrame): TFullReg =
#if c.traceActive:
when traceCode:
echo "PC ", pc, " ", c.code[pc].opcode, " ra ", ra, " rb ", instr.regB, " rc ", instr.regC
# message(c.config, c.debug[pc], warnUser, "Trace")
# message(c.config, c.debug[pc], warnUser, "Trace")

case instr.opcode
of opcEof: return regs[ra]
Expand Down
30 changes: 30 additions & 0 deletions tests/vm/tsetlen.nim
@@ -0,0 +1,30 @@
type Foo = object
index: int

block:
proc fun[T]() =
var foo: T
var n = 10

var foos: seq[T]
foos.setLen n

n.inc
foos.setLen n

for i in 0 ..< n:
let temp = foos[i]
when T is object:
doAssert temp.index == 0
when T is ref object:
doAssert temp == nil
doAssert temp == foo

static:
fun[Foo]()
fun[int]()
fun[float]()
fun[string]()
fun[(int, string)]()
fun[ref Foo]()
fun[seq[int]]()

0 comments on commit c16fa63

Please sign in to comment.