101 lines
1.8 KiB
Go
101 lines
1.8 KiB
Go
|
package buf2d
|
||
|
|
|
||
|
|
import (
|
||
|
|
"strings"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Buffer is a 2-dimensional rune buffer
|
||
|
|
type Buffer struct {
|
||
|
|
data [][]rune
|
||
|
|
width int
|
||
|
|
height int
|
||
|
|
parent *Buffer
|
||
|
|
}
|
||
|
|
|
||
|
|
// NewBuffer makes a new buffer with the given dimensions
|
||
|
|
func NewBuffer(width, height int) *Buffer {
|
||
|
|
b := make([][]rune, width)
|
||
|
|
for x := range b {
|
||
|
|
b[x] = make([]rune, height)
|
||
|
|
for y := range b[x] {
|
||
|
|
b[x][y] = ' '
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return &Buffer{
|
||
|
|
data: b,
|
||
|
|
width: width,
|
||
|
|
height: height,
|
||
|
|
parent: nil,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (b *Buffer) Set(x, y int, c rune) {
|
||
|
|
b.data[x][y] = c
|
||
|
|
}
|
||
|
|
|
||
|
|
func (b *Buffer) Get(x, y int) rune {
|
||
|
|
return b.data[x][y]
|
||
|
|
}
|
||
|
|
|
||
|
|
func (b *Buffer) Size() (w, h int) {
|
||
|
|
return b.width, b.height
|
||
|
|
}
|
||
|
|
|
||
|
|
func (b *Buffer) Width() int {
|
||
|
|
return b.width
|
||
|
|
}
|
||
|
|
|
||
|
|
func (b *Buffer) Height() int {
|
||
|
|
return b.height
|
||
|
|
}
|
||
|
|
|
||
|
|
// Draw calls drawFunc for every rune in this buffer
|
||
|
|
func (b *Buffer) Draw(drawFunc func(x, y int, c rune)) {
|
||
|
|
for x, row := range b.data {
|
||
|
|
for y, char := range row {
|
||
|
|
drawFunc(x, y, char)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (b *Buffer) String() string {
|
||
|
|
s := new(strings.Builder)
|
||
|
|
for ri, row := range b.data {
|
||
|
|
for _, char := range row {
|
||
|
|
s.WriteRune(char)
|
||
|
|
}
|
||
|
|
if ri != len(b.data)-1 {
|
||
|
|
s.WriteRune('\n')
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return s.String()
|
||
|
|
}
|
||
|
|
|
||
|
|
// Sub returns a buffer which is completely contained in this buffer
|
||
|
|
// Modifying the main buffer or the sub buffer will modify the other one as well
|
||
|
|
// This method can be used recursively
|
||
|
|
// If the given dimensions don't fit in the parent buffer, it will be truncated
|
||
|
|
func (b *Buffer) Sub(x, y, w, h int) *Buffer {
|
||
|
|
// sanitize inputs
|
||
|
|
x = limit(x, 0, b.width-1)
|
||
|
|
y = limit(y, 0, b.height-1)
|
||
|
|
w = limit(w, 1, b.width-x)
|
||
|
|
h = limit(h, 1, b.height-y)
|
||
|
|
|
||
|
|
// make slice references
|
||
|
|
data := make([][]rune, w)
|
||
|
|
for dx := x; dx < x+w-1; dx++ {
|
||
|
|
row := b.data[dx]
|
||
|
|
data[dx] = row[y : y+w]
|
||
|
|
}
|
||
|
|
|
||
|
|
// make buffer
|
||
|
|
return &Buffer{
|
||
|
|
data: data,
|
||
|
|
width: w,
|
||
|
|
height: h,
|
||
|
|
parent: b,
|
||
|
|
}
|
||
|
|
}
|