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, height)
|
|
for y := range b {
|
|
b[y] = make([]rune, width)
|
|
for x := range b[y] {
|
|
b[y][x] = ' '
|
|
}
|
|
}
|
|
|
|
return &Buffer{
|
|
data: b,
|
|
width: width,
|
|
height: height,
|
|
parent: nil,
|
|
}
|
|
}
|
|
|
|
func (b *Buffer) Set(x, y int, c rune) {
|
|
b.data[y][x] = c
|
|
}
|
|
|
|
func (b *Buffer) Get(x, y int) rune {
|
|
return b.data[y][x]
|
|
}
|
|
|
|
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 y, col := range b.data {
|
|
for x, char := range col {
|
|
drawFunc(x, y, char)
|
|
}
|
|
}
|
|
}
|
|
|
|
func (b *Buffer) String() string {
|
|
s := new(strings.Builder)
|
|
for ci, col := range b.data {
|
|
for _, char := range col {
|
|
s.WriteRune(char)
|
|
}
|
|
if ci != 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, h)
|
|
for dy := y; dy < y+h-1; dy++ {
|
|
col := b.data[dy]
|
|
data[dy] = col[x : x+w]
|
|
}
|
|
|
|
// make buffer
|
|
return &Buffer{
|
|
data: data,
|
|
width: w,
|
|
height: h,
|
|
parent: b,
|
|
}
|
|
}
|