github.com/gnolang/gno@v0.0.0-20240520182011-228e9d0192ce/misc/logos/buffer.go

package logos

import (
	"fmt"
	"strings"

	"github.com/gdamore/tcell/v2"
)

// ----------------------------------------
// Buffer

// A Buffer is a buffer area in which to draw.
type Buffer struct {
	Size
	Cells []Cell
}

func NewBuffer(sz Size) *Buffer {
	return &Buffer{
		Size:  sz,
		Cells: make([]Cell, sz.Width*sz.Height),
	}
}

func (bb *Buffer) Reset() {
	sz := bb.Size
	bb.Cells = make([]Cell, sz.Width*sz.Height)
}

func (bb *Buffer) GetCell(x, y int) *Cell {
	if bb.Width <= x {
		panic(fmt.Sprintf(
			"index x=%d out of bounds, width=%d",
			x, bb.Width))
	}
	if bb.Height <= y {
		panic(fmt.Sprintf(
			"index y=%d out of bounds, height=%d",
			y, bb.Height))
	}
	return &bb.Cells[y*bb.Width+x]
}

func (bb *Buffer) Sprint() string {
	lines := []string{}
	for y := 0; y < bb.Height; y++ {
		parts := []string{}
		for x := 0; x < bb.Width; x++ {
			cell := bb.GetCell(x, y)
			parts = append(parts, cell.Character)
			// NOTE: hacky way to debug.
			if true {
				attrs := cell.GetAttrs()
				flags := attrs.GetAttrFlags()
				parts = append(parts,
					fmt.Sprintf("%X", flags))
			}
		}
		line := strings.Join(parts, "")
		lines = append(lines, line)
	}
	return strings.Join(lines, "\n")
}

func (bb *Buffer) DrawToScreen(s tcell.Screen) {
	sw, sh := s.Size()
	if bb.Size.Width != sw || bb.Size.Height != sh {
		panic("buffer doesn't match screen size")
	}
	for y := 0; y < sh; y++ {
		for x := 0; x < sw; x++ {
			cell := bb.GetCell(x, y)
			if x == 0 && y == 0 {
				// NOTE: to thwart some inexplicable bugs.
				s.SetContent(0, 0, tcell.RunePlus, nil, gDefaultSpaceTStyle)
			} else {
				mainc, combc, tstyle := cell.GetTCellContent()
				s.SetContent(x, y, mainc, combc, tstyle)
			}
		}
	}
}

// ----------------------------------------
// Cell

// A terminal character cell.
type Cell struct {
	Character string // 1 unicode character, or " ".
	Width     int
	*Style
	Attrs
	Ref Elem // reference to element
}

// NOTE: there is a difference in behavior
// from SetValue(...,c2) when c2 has
// attributes not present in c2.Ref, so the
// two are not equivalent.
func (cc *Cell) SetValueFromCell(c2 *Cell) {
	cc.SetValue(c2.Character, c2.Width, c2.Style, c2.Ref)
	cc.Character = c2.Character
	cc.Width = c2.Width
	if c2.Style != nil {
		cc.Style = c2.Style
	}
	cc.Attrs.Merge(c2.GetAttrs())
	if c2.Ref != nil {
		cc.Ref = c2.Ref
	}
}

func (cc *Cell) SetValue(chs string, w int, st *Style, el Elem) {
	cc.Character = chs
	cc.Width = w
	if st != nil {
		cc.Style = st
	}
	if el != nil {
		cc.Attrs.Merge(el.GetAttrs())
		cc.Ref = el
	}
}

func (cc *Cell) Reset() {
	*cc = Cell{}
}

var gDefaultSpaceTStyle = tcell.StyleDefault.
	Dim(true).
	Background(tcell.ColorGray)

// This is where a bit of dynamic logic is performed,
// namely where the attr is used to derive the final style.
func (cc *Cell) GetTCellContent() (mainc rune, combc []rune, tstyle tcell.Style) {
	style := cc.Style
	attrs := &cc.Attrs
	if cc.Character == "" {
		mainc = '?' // for debugging
		if style == nil {
			// special case
			tstyle = gDefaultSpaceTStyle
		} else {
			tstyle = style.WithAttrs(attrs).GetTStyle()
		}
	} else {
		rz := toRunes(cc.Character)
		mainc = rz[0]
		combc = rz[1:]
		if style == nil {
			tstyle = gDefaultStyle.WithAttrs(attrs).GetTStyle()
		} else {
			tstyle = style.WithAttrs(attrs).GetTStyle()
		}
	}
	return
}

// ----------------------------------------
// View
// analogy: "Buffer:View :: array:slice".

// Offset and Size must be within bounds of *Buffer.
type View struct {
	Base   *Buffer
	Offset Coord // offset within Buffer
	Bounds Size  // total size of slice
}

// offset elements must be 0 or positive.
func (bb *Buffer) NewView(offset Coord) View {
	if !offset.IsNonNegative() {
		panic("should not happen")
	}
	return View{
		Base:   bb,
		Offset: offset,
		Bounds: bb.Size,
	}
}

func (bs View) NewView(offset Coord) View {
	return View{
		Base:   bs.Base,
		Offset: bs.Offset.Add(offset),
		Bounds: bs.Bounds.SubCoord(offset),
	}
}

func (bs View) GetCell(x, y int) *Cell {
	if bs.Bounds.Width <= x {
		panic("should not happen")
	}
	if bs.Bounds.Height <= y {
		panic("should not happen")
	}
	return bs.Base.GetCell(
		bs.Offset.X+x,
		bs.Offset.Y+y,
	)
}

// ----------------------------------------
// BufferedView

// A view onto an element.
// Somewhat like a slice onto an array
// (as a view is onto an elem),
// except cells are allocated here.
type BufferedElemView struct {
	Coord
	Size
	*Style
	Attrs         // e.g. to focus on a scrollbar
	Base    Elem  // the underlying elem
	Offset  Coord // within elem for pagination
	*Buffer       // view's internal draw screen
}

// Returns a new *BufferedElemView that spans the whole elem.
// If size is zero, the elem is measured first to get the full
// buffer size. The result must still be rendered before drawing.  The
// *BufferedElemView inherits the style and coordinates of the elem, and
// the elem's coord is set to zero.
func NewBufferedElemView(elem Elem, size Size) *BufferedElemView {
	if size.IsZero() {
		size = elem.Measure()
	}
	bpv := &BufferedElemView{
		Size:   size,
		Style:  elem.GetStyle(),
		Base:   elem,
		Offset: Coord{0, 0},
		// NOTE: be lazy, size may change.
		// Buffer: NewBuffer(size),
	}
	bpv.SetCoord(elem.GetCoord())
	bpv.SetIsDirty(true)
	elem.SetParent(bpv)
	elem.SetCoord(Coord{}) // required for abs calc.
	return bpv
}

func (bpv *BufferedElemView) StringIndented(indent string) string {
	return fmt.Sprintf("Buffered%v@%p\n%s    %s",
		bpv.Size,
		bpv,
		indent,
		bpv.Base.StringIndented(indent+"    "))
}

func (bpv *BufferedElemView) String() string {
	return fmt.Sprintf("Buffered%v{%v}@%p",
		bpv.Size,
		bpv.Base,
		bpv)
}

// Pass on style to the base elem.
func (bpv *BufferedElemView) SetStyle(style *Style) {
	bpv.Style = style
	bpv.Base.SetStyle(style)
	bpv.SetIsDirty(true)
}

func (bpv *BufferedElemView) SetSize(size Size) {
	bpv.Size = size
	bpv.Buffer = nil
	bpv.SetIsDirty(true)
}

// Cursor status pierces the buffered elem view;
// but a few key events are still consumed before the base.
func (bpv *BufferedElemView) SetIsCursor(ic bool) {
	bpv.Attrs.SetIsCursor(ic)
	bpv.Base.SetIsCursor(ic)
}

// BufferedElemView's size is simply defined by .Size.
func (bpv *BufferedElemView) Measure() Size {
	return bpv.Size
}

// Renders the elem onto the internal buffer.
// Assumes buffered elem view's elem was already rendered.
// TODO: this function could be optimized to reduce
// redundant background cell modifications.
func (bpv *BufferedElemView) Render() (updated bool) {
	if !bpv.GetIsDirty() {
		return
	} else {
		defer bpv.SetIsDirty(false)
	}
	// Get or initialize buffer.
	buffer := bpv.Buffer
	if buffer == nil {
		buffer = NewBuffer(bpv.Size)
		bpv.Buffer = buffer
	}
	// Reset the buffer's cells.  We could
	// use Buffer.Reset(), but this helps
	// distinguish between undefined cells
	// and those of an empty buffer.
	for x := 0; x < buffer.Size.Width; x++ {
		for y := 0; y < buffer.Size.Height; y++ {
			cell := buffer.GetCell(x, y)
			cell.Reset()
			// Draw Logos star background.
			cell.SetValue("\u2606", 1, bpv.Style, bpv)
		}
	}
	// Then, render and draw elem.
	bpv.Base.Render()
	bpv.Base.Draw(bpv.Offset, buffer.NewView(Coord{}))
	return true
}

func (bpv *BufferedElemView) Draw(offset Coord, view View) {
	minX, maxX, minY, maxY := computeIntersection(bpv.Size, offset, view.Bounds)
	for y := minY; y < maxY; y++ {
		for x := minX; x < maxX; x++ {
			bcell := bpv.Buffer.GetCell(x, y)
			vcell := view.GetCell(x-offset.X, y-offset.Y)
			vcell.SetValueFromCell(bcell)
		}
	}
}

func (bpv *BufferedElemView) ProcessEventKey(ev *EventKey) bool {
	// Pagination is outer-greedy, and so Logos
	// generally just likes infinite areas.
	switch evr := ev.Rune(); evr {
	case 'a': // left
		bpv.Scroll(Coord{16, 0})
	case 's': // down
		bpv.Scroll(Coord{0, -8})
	case 'd': // right
		bpv.Scroll(Coord{-16, 0})
	case 'w': // up
		bpv.Scroll(Coord{0, 8})
	default:
		// Try to get the base to handle it.
		if bpv.Base.ProcessEventKey(ev) {
			return true
		}
		return false
	}
	return true // convenience for cases.
}

func (bpv *BufferedElemView) Scroll(dir Coord) {
	bpv.Offset = bpv.Offset.Add(dir)
	bpv.SetIsDirty(true)
}