diff options
-rw-r--r-- | src/field.nim | 47 | ||||
-rw-r--r-- | tests/field.nim | 95 |
2 files changed, 69 insertions, 73 deletions
diff --git a/src/field.nim b/src/field.nim index 82a8575..384cb47 100644 --- a/src/field.nim +++ b/src/field.nim @@ -1,5 +1,5 @@ type - Line = ref object + Line = object x: int columns: seq[int] @@ -11,7 +11,7 @@ type func Blank*(f: var Field, x, y: int) = if y < f.y or y >= f.y+f.ly: # outside the field return - var l = f.lines[y-f.y] + var l = addr f.lines[y-f.y] if x < l.x or x >= l.x+l.columns.len: # outside the field return if x > l.x and x < l.x+l.columns.len-1: # just set the value @@ -20,14 +20,14 @@ func Blank*(f: var Field, x, y: int) = if l.columns.len == 1: # this was the last character on the line if y == f.y: # we need to trim the leading lines var i = 1 - while f.lines[i] == nil or f.lines[i].columns.len == 0: + while f.lines[i].columns.len == 0: inc i f.y += i f.lines = f.lines[i..<f.ly] f.ly -= i elif y == f.y+f.ly-1: # we need to trim the trailing lines var i = f.ly-2 - while f.lines[i] == nil or f.lines[i].columns.len == 0: + while f.lines[i].columns.len == 0: dec i f.ly = i+1 f.lines = f.lines[0..<f.ly] @@ -48,7 +48,7 @@ func Blank*(f: var Field, x, y: int) = f.x = f.lines[0].x var x2 = f.lines[0].columns.len + f.lines[0].x for i in 1..<f.ly: - if f.lines[i] == nil or f.lines[i].columns.len == 0: + if f.lines[i].columns.len == 0: continue if f.x > f.lines[i].x: f.x = f.lines[i].x @@ -71,8 +71,10 @@ proc Load*(filename: string): ref Field = if not open(file, filename): return nil defer: file.close() - var f = new(Field) - var l = new(Line) + var f: ref Field + new(f) + var l : ptr Line + var leadingSpaces = 0 var trailingSpaces = 0 var data: array[4096, char] var lastReadIsCR = false @@ -80,13 +82,12 @@ proc Load*(filename: string): ref Field = let n = file.readChars(data, 0, 4096) if n <= 0: if f.ly == 0: - if l.columns.len == 0: # we got en empty file! + if l == nil: # we got en empty file! return nil f.x = l.x - if l.columns.len > 0: + if l != nil: if f.lx < l.columns.len+l.x-f.x: f.lx = l.columns.len+l.x-f.x - f.lines.add(l) inc f.ly break var i = 0 @@ -100,17 +101,18 @@ proc Load*(filename: string): ref Field = continue if data[i] == '\n' or data[i] == '\r': if f.ly == 0: - if l.columns.len == 0: + if l == nil: return nil f.x = l.x - if l.columns.len > 0: + if l != nil: if f.x > l.x: f.x = l.x if f.lx < l.columns.len+l.x-f.x: f.lx = l.columns.len+l.x-f.x + else: + f.lines.add(Line()) inc f.ly - f.lines.add(l) - l = new(Line) + l = nil trailingSpaces = 0 if data[i] == '\r': if i+1 < n and data[i+1] == '\n': @@ -119,11 +121,16 @@ proc Load*(filename: string): ref Field = lastReadIsCR = true else: if data[i] == ' ': - if l.columns.len == 0: # trim leading spaces - inc l.x + if l == nil: # trim leading spaces + inc leadingSpaces else: inc trailingSpaces else: + if l == nil: + f.lines.add(Line()) + l = addr f.lines[^1] + l.x = leadingSpaces + leadingSpaces = 0 if trailingSpaces > 0: let oldL = l.columns.len l.columns.setlen(oldL+trailingSpaces+1) @@ -134,6 +141,7 @@ proc Load*(filename: string): ref Field = else: l.columns.add(int(data[i])) inc i + f.lines = f.lines[0..<f.ly] return f func Set*(f: var Field, x, y, v: int) = @@ -141,13 +149,10 @@ func Set*(f: var Field, x, y, v: int) = f.Blank(x, y) elif y >= f.y: if y < f.y+f.ly: # the line exists - var l = f.lines[y-f.y] - if l == nil or l.columns.len == 0: # An empty line is a special case - if l == nil: - new(l) + var l = addr f.lines[y-f.y] + if l.columns.len == 0: # An empty line is a special case l.x = x l.columns = @[v] - f.lines[y-f.y] = l if f.x > x: f.lx = f.lx+f.x-x f.x = x diff --git a/tests/field.nim b/tests/field.nim index deea574..b3c16a3 100644 --- a/tests/field.nim +++ b/tests/field.nim @@ -2,109 +2,100 @@ import unittest include ../src/field -func `==`(a, b: Field): bool = - if a.x == b.x and a.lx == b.lx and a.y == b.y and a.ly == b.ly: - for i in 0..<a.ly: - if a.lines[i] == nil and b.lines[i] == nil: - continue - if a.lines[i] == nil xor b.lines[i] == nil: - return false - if a.lines[i].x != b.lines[i].x or a.lines[i].columns != b.lines[i].columns: - return false - return true - return false +func `==`(a, b: Line): bool = a.x == b.x and a.columns == b.columns +func `==`(a, b: Field): bool = a.x == b.x and a.lx == b.lx and a.y == b.y and a.ly == b.ly and a.lines == b.lines func cols(a: openarray[char]): seq[int] = result.setlen(a.len) for i in 0..<a.len: result[i] = a[i].int() -let minimal = Field(x: 0, y: 0, lx: 1, ly: 1, lines: @[Line(x: 0, columns: @[int('@')])]) +const minimal = Field(x: 0, y: 0, lx: 1, ly: 1, lines: @[Line(x: 0, columns: @[int('@')])]) suite "Field": test "Blank": var f = Field(x: -7, y: -5, lx: 17, ly: 10, lines: @[ Line(x: -5, columns: @[int('x')]), - nil, + Line(x: 0, columns: @[]), Line(x: 8, columns: @[int('u')]), Line(x: 9, columns: @[int('e')]), - nil, + Line(x: 0, columns: @[]), Line(x: -2, columns: @[int('l'), 32, int('@'), int('z'), 32, 32, int('r')]), Line(x: -3, columns: @[int('f')]), Line(x: 5, columns: @[int('d')]), - nil, + Line(x: 0, columns: @[]), Line(x: -7, columns: @[int('y')]), ]) - let moinsz = Field(x: -7, y: -5, lx: 17, ly: 10, lines: @[ + const moinsz = Field(x: -7, y: -5, lx: 17, ly: 10, lines: @[ Line(x: -5, columns: @[int('x')]), - nil, + Line(x: 0, columns: @[]), Line(x: 8, columns: @[int('u')]), Line(x: 9, columns: @[int('e')]), - nil, + Line(x: 0, columns: @[]), Line(x: -2, columns: @[int('l'), 32, int('@'), 32, 32, 32, int('r')]), Line(x: -3, columns: @[int('f')]), Line(x: 5, columns: @[int('d')]), - nil, + Line(x: 0, columns: @[]), Line(x: -7, columns: @[int('y')]), ]) f.Blank(1, 0) check f == moinsz - let moinsy = Field(x: -5, y: -5, lx: 15, ly: 8, lines: @[ + const moinsy = Field(x: -5, y: -5, lx: 15, ly: 8, lines: @[ Line(x: -5, columns: @[int('x')]), - nil, + Line(x: 0, columns: @[]), Line(x: 8, columns: @[int('u')]), Line(x: 9, columns: @[int('e')]), - nil, + Line(x: 0, columns: @[]), Line(x: -2, columns: @[int('l'), 32, int('@'), 32, 32, 32, int('r')]), Line(x: -3, columns: @[int('f')]), Line(x: 5, columns: @[int('d')]), ]) f.Blank(-7, 4) check f == moinsy - let moinsx = Field(x: -3, y: -3, lx: 13, ly: 6, lines: @[ + const moinsx = Field(x: -3, y: -3, lx: 13, ly: 6, lines: @[ Line(x: 8, columns: @[int('u')]), Line(x: 9, columns: @[int('e')]), - nil, + Line(x: 0, columns: @[]), Line(x: -2, columns: @[int('l'), 32, int('@'), 32, 32, 32, int('r')]), Line(x: -3, columns: @[int('f')]), Line(x: 5, columns: @[int('d')]), ]) f.Blank(-5, -5) check f == moinsx - let moinsf = Field(x: -2, y: -3, lx: 12, ly: 6, lines: @[ + const moinsf = Field(x: -2, y: -3, lx: 12, ly: 6, lines: @[ Line(x: 8, columns: @[int('u')]), Line(x: 9, columns: @[int('e')]), - nil, + Line(x: 0, columns: @[]), Line(x: -2, columns: @[int('l'), 32, int('@'), 32, 32, 32, int('r')]), Line(x: -3, columns: @[]), Line(x: 5, columns: @[int('d')]), ]) f.Blank(-3, 1) check f == moinsf - let moinse = Field(x: -2, y: -3, lx: 11, ly: 6, lines: @[ + const moinse = Field(x: -2, y: -3, lx: 11, ly: 6, lines: @[ Line(x: 8, columns: @[int('u')]), Line(x: 9, columns: @[]), - nil, + Line(x: 0, columns: @[]), Line(x: -2, columns: @[int('l'), 32, int('@'), 32, 32, 32, int('r')]), Line(x: -3, columns: @[]), Line(x: 5, columns: @[int('d')]), ]) f.Blank(9, -2) check f == moinse - let moinsu = Field(x: -2, y: 0, lx: 8, ly: 3, lines: @[ + const moinsu = Field(x: -2, y: 0, lx: 8, ly: 3, lines: @[ Line(x: -2, columns: @[int('l'), 32, int('@'), 32, 32, 32, int('r')]), Line(x: -3, columns: @[]), Line(x: 5, columns: @[int('d')]), ]) f.Blank(8, -3) check f == moinsu - let moinsd = Field(x: -2, y: 0, lx: 7, ly: 1, lines: @[Line(x: -2, columns: @[int('l'), 32, int('@'), 32, 32, 32, int('r')])]) + const moinsd = Field(x: -2, y: 0, lx: 7, ly: 1, lines: @[Line(x: -2, columns: @[int('l'), 32, int('@'), 32, 32, 32, int('r')])]) f.Blank(5, 2) check f == moinsd - let moinsl = Field(x: 0, y: 0, lx: 5, ly: 1, lines: @[Line(x: 0, columns: @[int('@'), 32, 32, 32, int('r')])]) + const moinsl = Field(x: 0, y: 0, lx: 5, ly: 1, lines: @[Line(x: 0, columns: @[int('@'), 32, 32, 32, int('r')])]) f.Blank(-2, 0) check f == moinsl - let moinsr = Field(x: 0, y: 0, lx: 1, ly: 1, lines: @[Line(x: 0, columns: @[int('@')])]) + const moinsr = Field(x: 0, y: 0, lx: 1, ly: 1, lines: @[Line(x: 0, columns: @[int('@')])]) f.Blank(4, 0) check f == moinsr test "Get": @@ -146,71 +137,71 @@ suite "Field": check f == minimal f.Set(1, 0, int(' ')) check f == minimal - let xappend = Field(x: 0, y: 0, lx: 5, ly: 1, lines: @[Line(x: 0, columns: @[int('@'), 32, 32, 32, int('r')])]) + const xappend = Field(x: 0, y: 0, lx: 5, ly: 1, lines: @[Line(x: 0, columns: @[int('@'), 32, 32, 32, int('r')])]) f.Set(4, 0, int('r')) check f == xappend - let xprepend = Field(x: -2, y: 0, lx: 7, ly: 1, lines: @[Line(x: -2, columns: @[int('l'), 32, int('@'), 32, 32, 32, int('r')])]) + const xprepend = Field(x: -2, y: 0, lx: 7, ly: 1, lines: @[Line(x: -2, columns: @[int('l'), 32, int('@'), 32, 32, 32, int('r')])]) f.Set(-2, 0, int('l')) check f == xprepend - let yappend = Field(x: -2, y: 0, lx: 8, ly: 3, lines: @[ + const yappend = Field(x: -2, y: 0, lx: 8, ly: 3, lines: @[ Line(x: -2, columns: @[int('l'), 32, int('@'), 32, 32, 32, int('r')]), - nil, + Line(x: 0, columns: @[]), Line(x: 5, columns: @[int('d')]), ]) f.Set(5, 2, int('d')) check f == yappend - let yprepend = Field(x: -2, y: -3, lx: 11, ly: 6, lines: @[ + const yprepend = Field(x: -2, y: -3, lx: 11, ly: 6, lines: @[ Line(x: 8, columns: @[int('u')]), - nil, - nil, + Line(x: 0, columns: @[]), + Line(x: 0, columns: @[]), Line(x: -2, columns: @[int('l'), 32, int('@'), 32, 32, 32, int('r')]), - nil, + Line(x: 0, columns: @[]), Line(x: 5, columns: @[int('d')]), ]) f.Set(8, -3, int('u')) check f == yprepend - let xappendEmptyline = Field(x: -2, y: -3, lx: 12, ly: 6, lines: @[ + const xappendEmptyline = Field(x: -2, y: -3, lx: 12, ly: 6, lines: @[ Line(x: 8, columns: @[int('u')]), Line(x: 9, columns: @[int('e')]), - nil, + Line(x: 0, columns: @[]), Line(x: -2, columns: @[int('l'), 32, int('@'), 32, 32, 32, int('r')]), - nil, + Line(x: 0, columns: @[]), Line(x: 5, columns: @[int('d')]), ]) f.Set(9, -2, int('e')) check f == xappendEmptyline - let xprependEmptyline = Field(x: -3, y: -3, lx: 13, ly: 6, lines: @[ + const xprependEmptyline = Field(x: -3, y: -3, lx: 13, ly: 6, lines: @[ Line(x: 8, columns: @[int('u')]), Line(x: 9, columns: @[int('e')]), - nil, + Line(x: 0, columns: @[]), Line(x: -2, columns: @[int('l'), 32, int('@'), 32, 32, 32, int('r')]), Line(x: -3, columns: @[int('f')]), Line(x: 5, columns: @[int('d')]), ]) f.Set(-3, 1, int('f')) check f == xprependEmptyline - let xprependyprepend = Field(x: -5, y: -5, lx: 15, ly: 8, lines: @[ + const xprependyprepend = Field(x: -5, y: -5, lx: 15, ly: 8, lines: @[ Line(x: -5, columns: @[int('x')]), - nil, + Line(x: 0, columns: @[]), Line(x: 8, columns: @[int('u')]), Line(x: 9, columns: @[int('e')]), - nil, + Line(x: 0, columns: @[]), Line(x: -2, columns: @[int('l'), 32, int('@'), 32, 32, 32, int('r')]), Line(x: -3, columns: @[int('f')]), Line(x: 5, columns: @[int('d')]), ]) f.Set(-5, -5, int('x')) check f == xprependyprepend - let xprependyappend = Field(x: -7, y: -5, lx: 17, ly: 10, lines: @[ + const xprependyappend = Field(x: -7, y: -5, lx: 17, ly: 10, lines: @[ Line(x: -5, columns: @[int('x')]), - nil, + Line(x: 0, columns: @[]), Line(x: 8, columns: @[int('u')]), Line(x: 9, columns: @[int('e')]), - nil, + Line(x: 0, columns: @[]), Line(x: -2, columns: @[int('l'), 32, int('@'), 32, 32, 32, int('r')]), Line(x: -3, columns: @[int('f')]), Line(x: 5, columns: @[int('d')]), - nil, + Line(x: 0, columns: @[]), Line(x: -7, columns: @[int('y')]), ]) f.Set(-7, 4, int('y')) |