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 {
|
||||
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