Implemented function to set a field value
This commit is contained in:
parent
e8662451d4
commit
fc132c3dea
2 changed files with 340 additions and 0 deletions
|
|
@ -13,3 +13,86 @@ func (f Field) Get(x, y int) int {
|
||||||
func (f Field) isIn(x, y int) bool {
|
func (f Field) isIn(x, y int) bool {
|
||||||
return x >= f.x && x < f.x+f.lx && y >= f.y && y < f.y+f.ly
|
return x >= f.x && x < f.x+f.lx && y >= f.y && y < f.y+f.ly
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (f *Field) Set(x, y, v int) {
|
||||||
|
if y >= f.y {
|
||||||
|
if y < f.y+f.ly {
|
||||||
|
l := &f.lines[y-f.y]
|
||||||
|
if l.l == 0 {
|
||||||
|
// an empty line is a special case
|
||||||
|
l.x = x
|
||||||
|
l.l = 1
|
||||||
|
l.columns = append(l.columns, v)
|
||||||
|
} else if x >= l.x {
|
||||||
|
if x < l.x+l.l {
|
||||||
|
// just set the value
|
||||||
|
l.columns[x-l.x] = v
|
||||||
|
} else {
|
||||||
|
// append columns
|
||||||
|
newL := l.l + x - l.x
|
||||||
|
for i := l.l; i < newL-1; i++ {
|
||||||
|
l.columns = append(l.columns, ' ')
|
||||||
|
}
|
||||||
|
l.columns = append(l.columns, v)
|
||||||
|
l.l = newL
|
||||||
|
if f.lx < l.l-l.x {
|
||||||
|
f.lx = l.l - l.x
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
// prepend columns
|
||||||
|
newL := l.l + l.x - x
|
||||||
|
c := make([]int, newL)
|
||||||
|
c[0] = v
|
||||||
|
for i := 0; i < l.x-x; i++ {
|
||||||
|
c[i+1] = ' '
|
||||||
|
}
|
||||||
|
for j := 0; j < l.l; j++ {
|
||||||
|
c[j+l.x-x] = l.columns[j]
|
||||||
|
}
|
||||||
|
l.columns = c
|
||||||
|
l.x = x
|
||||||
|
l.l = newL
|
||||||
|
if f.x > x {
|
||||||
|
f.x = x
|
||||||
|
f.lx = newL
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
// append lines
|
||||||
|
newLy := y - f.y + 1
|
||||||
|
for i := f.ly; i < newLy-1; i++ {
|
||||||
|
f.lines = append(f.lines, Line{})
|
||||||
|
}
|
||||||
|
f.lines = append(f.lines, Line{x: x, l: 1, columns: []int{v}})
|
||||||
|
f.ly = newLy
|
||||||
|
if f.x > x {
|
||||||
|
f.lx += f.x - x
|
||||||
|
f.x = x
|
||||||
|
} else if f.lx-f.x < x {
|
||||||
|
f.lx = x - f.x
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
// prepend lines
|
||||||
|
newLy := f.ly + f.y - y
|
||||||
|
lines := make([]Line, newLy)
|
||||||
|
lines[0] = Line{
|
||||||
|
x: x,
|
||||||
|
l: 1,
|
||||||
|
columns: []int{v},
|
||||||
|
}
|
||||||
|
for j := 0; j < f.ly; j++ {
|
||||||
|
lines[j+f.y-y] = f.lines[j]
|
||||||
|
}
|
||||||
|
f.lines = lines
|
||||||
|
f.y = y
|
||||||
|
f.ly = newLy
|
||||||
|
if f.x > x {
|
||||||
|
f.lx += f.x - x
|
||||||
|
f.x = x
|
||||||
|
} else if f.lx-f.x < x {
|
||||||
|
f.lx = x - f.x + 1
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
|
||||||
|
|
@ -72,3 +72,260 @@ func TestIsIn(t *testing.T) {
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestSetMinimalAppendOne(t *testing.T) {
|
||||||
|
hashField := Field{
|
||||||
|
x: 0,
|
||||||
|
y: 0,
|
||||||
|
lx: 1,
|
||||||
|
ly: 1,
|
||||||
|
lines: []Line{
|
||||||
|
Line{x: 0, l: 1, columns: []int{'#'}},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
xAppendField := Field{
|
||||||
|
x: 0,
|
||||||
|
y: 0,
|
||||||
|
lx: 2,
|
||||||
|
ly: 1,
|
||||||
|
lines: []Line{
|
||||||
|
Line{x: 0, l: 2, columns: []int{'@', '#'}},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
xPrependField := Field{
|
||||||
|
x: -1,
|
||||||
|
y: 0,
|
||||||
|
lx: 2,
|
||||||
|
ly: 1,
|
||||||
|
lines: []Line{
|
||||||
|
Line{x: -1, l: 2, columns: []int{'#', '@'}},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
yAppendField := Field{
|
||||||
|
x: 0,
|
||||||
|
y: 0,
|
||||||
|
lx: 1,
|
||||||
|
ly: 2,
|
||||||
|
lines: []Line{
|
||||||
|
Line{x: 0, l: 1, columns: []int{'@'}},
|
||||||
|
Line{x: 0, l: 1, columns: []int{'#'}},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
yPrependField := Field{
|
||||||
|
x: 0,
|
||||||
|
y: -1,
|
||||||
|
lx: 1,
|
||||||
|
ly: 2,
|
||||||
|
lines: []Line{
|
||||||
|
Line{x: 0, l: 1, columns: []int{'#'}},
|
||||||
|
Line{x: 0, l: 1, columns: []int{'@'}},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
// Test cases
|
||||||
|
testCases := []struct {
|
||||||
|
name string
|
||||||
|
input string
|
||||||
|
inputX int
|
||||||
|
inputY int
|
||||||
|
inputV int
|
||||||
|
expected *Field
|
||||||
|
}{
|
||||||
|
{"simple", "test_data/minimal.b98", 0, 0, '#', &hashField},
|
||||||
|
{"xappend", "test_data/minimal.b98", 1, 0, '#', &xAppendField},
|
||||||
|
{"xprepend", "test_data/minimal.b98", -1, 0, '#', &xPrependField},
|
||||||
|
{"yappend", "test_data/minimal.b98", 0, 1, '#', &yAppendField},
|
||||||
|
{"yprepend", "test_data/minimal.b98", 0, -1, '#', &yPrependField},
|
||||||
|
}
|
||||||
|
for _, tc := range testCases {
|
||||||
|
t.Run(tc.name, func(t *testing.T) {
|
||||||
|
file, err := os.Open(tc.input)
|
||||||
|
require.NoError(t, err, "Failed to open file")
|
||||||
|
defer file.Close()
|
||||||
|
field, err := LoadFile(file)
|
||||||
|
field.Set(tc.inputX, tc.inputY, tc.inputV)
|
||||||
|
require.NoError(t, err)
|
||||||
|
require.Equal(t, tc.expected, field, "Invalid value")
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSetMinimalAppendTwo(t *testing.T) {
|
||||||
|
bottomRight := Field{
|
||||||
|
x: 0,
|
||||||
|
y: 0,
|
||||||
|
lx: 5,
|
||||||
|
ly: 3,
|
||||||
|
lines: []Line{
|
||||||
|
Line{x: 0, l: 1, columns: []int{'@'}},
|
||||||
|
Line{},
|
||||||
|
Line{x: 5, l: 1, columns: []int{'#'}},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
bottomLeft := Field{
|
||||||
|
x: -5,
|
||||||
|
y: 0,
|
||||||
|
lx: 6,
|
||||||
|
ly: 4,
|
||||||
|
lines: []Line{
|
||||||
|
Line{x: 0, l: 1, columns: []int{'@'}},
|
||||||
|
Line{},
|
||||||
|
Line{},
|
||||||
|
Line{x: -5, l: 1, columns: []int{'#'}},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
topRight := Field{
|
||||||
|
x: 0,
|
||||||
|
y: -3,
|
||||||
|
lx: 9,
|
||||||
|
ly: 4,
|
||||||
|
lines: []Line{
|
||||||
|
Line{x: 8, l: 1, columns: []int{'#'}},
|
||||||
|
Line{},
|
||||||
|
Line{},
|
||||||
|
Line{x: 0, l: 1, columns: []int{'@'}},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
topLeft := Field{
|
||||||
|
x: -12,
|
||||||
|
y: -4,
|
||||||
|
lx: 13,
|
||||||
|
ly: 5,
|
||||||
|
lines: []Line{
|
||||||
|
Line{x: -12, l: 1, columns: []int{'#'}},
|
||||||
|
Line{},
|
||||||
|
Line{},
|
||||||
|
Line{},
|
||||||
|
Line{x: 0, l: 1, columns: []int{'@'}},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
// Test cases
|
||||||
|
testCases := []struct {
|
||||||
|
name string
|
||||||
|
input string
|
||||||
|
inputX int
|
||||||
|
inputY int
|
||||||
|
inputV int
|
||||||
|
expected *Field
|
||||||
|
}{
|
||||||
|
{"bottomRight", "test_data/minimal.b98", 5, 2, '#', &bottomRight},
|
||||||
|
{"bottomLeft", "test_data/minimal.b98", -5, 3, '#', &bottomLeft},
|
||||||
|
{"topRight", "test_data/minimal.b98", 8, -3, '#', &topRight},
|
||||||
|
{"topLeft", "test_data/minimal.b98", -12, -4, '#', &topLeft},
|
||||||
|
}
|
||||||
|
for _, tc := range testCases {
|
||||||
|
t.Run(tc.name, func(t *testing.T) {
|
||||||
|
file, err := os.Open(tc.input)
|
||||||
|
require.NoError(t, err, "Failed to open file")
|
||||||
|
defer file.Close()
|
||||||
|
field, err := LoadFile(file)
|
||||||
|
field.Set(tc.inputX, tc.inputY, tc.inputV)
|
||||||
|
require.NoError(t, err)
|
||||||
|
require.Equal(t, tc.expected, field, "Invalid value")
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSetMinimalAppendThree(t *testing.T) {
|
||||||
|
xAppendField := Field{
|
||||||
|
x: 0,
|
||||||
|
y: 0,
|
||||||
|
lx: 4,
|
||||||
|
ly: 1,
|
||||||
|
lines: []Line{
|
||||||
|
Line{x: 0, l: 4, columns: []int{'@', ' ', ' ', '#'}},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
xPrependField := Field{
|
||||||
|
x: -3,
|
||||||
|
y: 0,
|
||||||
|
lx: 4,
|
||||||
|
ly: 1,
|
||||||
|
lines: []Line{
|
||||||
|
Line{x: -3, l: 4, columns: []int{'#', ' ', ' ', '@'}},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
yAppendField := Field{
|
||||||
|
x: 0,
|
||||||
|
y: 0,
|
||||||
|
lx: 1,
|
||||||
|
ly: 4,
|
||||||
|
lines: []Line{
|
||||||
|
Line{x: 0, l: 1, columns: []int{'@'}},
|
||||||
|
Line{},
|
||||||
|
Line{},
|
||||||
|
Line{x: 0, l: 1, columns: []int{'#'}},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
yPrependField := Field{
|
||||||
|
x: 0,
|
||||||
|
y: -3,
|
||||||
|
lx: 1,
|
||||||
|
ly: 4,
|
||||||
|
lines: []Line{
|
||||||
|
Line{x: 0, l: 1, columns: []int{'#'}},
|
||||||
|
Line{},
|
||||||
|
Line{},
|
||||||
|
Line{x: 0, l: 1, columns: []int{'@'}},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
// Test cases
|
||||||
|
testCases := []struct {
|
||||||
|
name string
|
||||||
|
input string
|
||||||
|
inputX int
|
||||||
|
inputY int
|
||||||
|
inputV int
|
||||||
|
expected *Field
|
||||||
|
}{
|
||||||
|
{"xappend", "test_data/minimal.b98", 3, 0, '#', &xAppendField},
|
||||||
|
{"xprepend", "test_data/minimal.b98", -3, 0, '#', &xPrependField},
|
||||||
|
{"yappend", "test_data/minimal.b98", 0, 3, '#', &yAppendField},
|
||||||
|
{"yprepend", "test_data/minimal.b98", 0, -3, '#', &yPrependField},
|
||||||
|
}
|
||||||
|
for _, tc := range testCases {
|
||||||
|
t.Run(tc.name, func(t *testing.T) {
|
||||||
|
file, err := os.Open(tc.input)
|
||||||
|
require.NoError(t, err, "Failed to open file")
|
||||||
|
defer file.Close()
|
||||||
|
field, err := LoadFile(file)
|
||||||
|
field.Set(tc.inputX, tc.inputY, tc.inputV)
|
||||||
|
require.NoError(t, err)
|
||||||
|
require.Equal(t, tc.expected, field, "Invalid value")
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// get and put on an empty line
|
||||||
|
func TestGetAndSetOnEmptyLines(t *testing.T) {
|
||||||
|
f := Field{
|
||||||
|
x: 0,
|
||||||
|
y: -4,
|
||||||
|
lx: 1,
|
||||||
|
ly: 8,
|
||||||
|
lines: []Line{
|
||||||
|
Line{x: 0, l: 1, columns: []int{'b'}},
|
||||||
|
Line{},
|
||||||
|
Line{},
|
||||||
|
Line{x: 0, l: 1, columns: []int{'c'}},
|
||||||
|
Line{x: 0, l: 1, columns: []int{'@'}},
|
||||||
|
Line{},
|
||||||
|
Line{x: 0, l: 1, columns: []int{'a'}},
|
||||||
|
Line{x: 0, l: 1, columns: []int{'#'}},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
file, err := os.Open("test_data/minimal.b98")
|
||||||
|
require.NoError(t, err, "Failed to open file")
|
||||||
|
defer file.Close()
|
||||||
|
field, err := LoadFile(file)
|
||||||
|
field.Set(0, 3, '#')
|
||||||
|
v := field.Get(0, 3)
|
||||||
|
require.Equal(t, v, int('#'))
|
||||||
|
v = field.Get(0, 2)
|
||||||
|
require.Equal(t, v, int(' '))
|
||||||
|
field.Set(0, 2, 'a')
|
||||||
|
field.Set(0, -4, 'b')
|
||||||
|
v = field.Get(0, -1)
|
||||||
|
require.Equal(t, v, int(' '))
|
||||||
|
field.Set(0, -1, 'c')
|
||||||
|
require.Equal(t, field, &f)
|
||||||
|
}
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue