297 lines
8.2 KiB
Go
297 lines
8.2 KiB
Go
package layout
|
|
|
|
import (
|
|
"github.com/sqshq/sampler/component"
|
|
"github.com/sqshq/sampler/component/runchart"
|
|
"github.com/sqshq/sampler/config"
|
|
"github.com/sqshq/sampler/console"
|
|
ui "github.com/sqshq/termui"
|
|
"image"
|
|
"math"
|
|
)
|
|
|
|
type Layout struct {
|
|
ui.Block
|
|
Components []component.Component
|
|
ChangeModeEvents chan Mode
|
|
statusbar *component.StatusBar
|
|
menu *component.Menu
|
|
mode Mode
|
|
selection int
|
|
}
|
|
|
|
type Mode rune
|
|
|
|
const (
|
|
ModeDefault Mode = 0
|
|
ModePause Mode = 1
|
|
ModeComponentSelect Mode = 2
|
|
ModeMenuOptionSelect Mode = 3
|
|
ModeComponentMove Mode = 4
|
|
ModeComponentResize Mode = 5
|
|
ModeChartPinpoint Mode = 6
|
|
)
|
|
|
|
const (
|
|
columnsCount = 80
|
|
rowsCount = 40
|
|
minDimension = 3
|
|
statusbarHeight = 1
|
|
)
|
|
|
|
func NewLayout(width, height int, statusline *component.StatusBar, menu *component.Menu) *Layout {
|
|
|
|
block := *ui.NewBlock()
|
|
block.SetRect(0, 0, width, height)
|
|
statusline.SetRect(0, height-statusbarHeight, width, height)
|
|
|
|
return &Layout{
|
|
Block: block,
|
|
Components: make([]component.Component, 0),
|
|
statusbar: statusline,
|
|
menu: menu,
|
|
mode: ModeDefault,
|
|
selection: 0,
|
|
ChangeModeEvents: make(chan Mode, 10),
|
|
}
|
|
}
|
|
|
|
func (l *Layout) AddComponent(Type config.ComponentType, drawable ui.Drawable, title string, position config.Position, size config.Size, refreshRateMs int) {
|
|
l.Components = append(l.Components, component.Component{
|
|
Type: Type,
|
|
Drawable: drawable,
|
|
Title: title,
|
|
Position: position,
|
|
Size: size,
|
|
RefreshRateMs: refreshRateMs,
|
|
})
|
|
}
|
|
|
|
func (l *Layout) GetComponents(Type config.ComponentType) []component.Component {
|
|
|
|
var components []component.Component
|
|
|
|
for _, c := range l.Components {
|
|
if c.Type == Type {
|
|
components = append(components, c)
|
|
}
|
|
}
|
|
|
|
return components
|
|
}
|
|
|
|
func (l *Layout) changeMode(m Mode) {
|
|
l.mode = m
|
|
l.ChangeModeEvents <- m
|
|
}
|
|
|
|
func (l *Layout) HandleConsoleEvent(e string) {
|
|
switch e {
|
|
case console.KeyPause:
|
|
if l.mode == ModePause {
|
|
l.changeMode(ModeDefault)
|
|
} else {
|
|
if l.getSelectedComponent().Type == config.TypeRunChart {
|
|
chart := l.getSelectedComponent().Drawable.(*runchart.RunChart)
|
|
chart.DisableSelection()
|
|
}
|
|
l.menu.Idle()
|
|
l.changeMode(ModePause)
|
|
}
|
|
case console.KeyEnter:
|
|
switch l.mode {
|
|
case ModeComponentSelect:
|
|
l.menu.Choose()
|
|
l.changeMode(ModeMenuOptionSelect)
|
|
case ModeMenuOptionSelect:
|
|
option := l.menu.GetSelectedOption()
|
|
switch option {
|
|
case component.MenuOptionMove:
|
|
l.changeMode(ModeComponentMove)
|
|
l.menu.MoveOrResize()
|
|
case component.MenuOptionResize:
|
|
l.changeMode(ModeComponentResize)
|
|
l.menu.MoveOrResize()
|
|
case component.MenuOptionPinpoint:
|
|
l.changeMode(ModeChartPinpoint)
|
|
l.menu.Idle()
|
|
chart := l.getSelectedComponent().Drawable.(*runchart.RunChart)
|
|
chart.MoveSelection(0)
|
|
case component.MenuOptionResume:
|
|
l.changeMode(ModeDefault)
|
|
l.menu.Idle()
|
|
}
|
|
case ModeComponentMove:
|
|
fallthrough
|
|
case ModeComponentResize:
|
|
l.menu.Idle()
|
|
l.changeMode(ModeDefault)
|
|
}
|
|
case console.KeyEsc:
|
|
switch l.mode {
|
|
case ModeChartPinpoint:
|
|
chart := l.getSelectedComponent().Drawable.(*runchart.RunChart)
|
|
chart.DisableSelection()
|
|
fallthrough
|
|
case ModeComponentSelect:
|
|
fallthrough
|
|
case ModeMenuOptionSelect:
|
|
l.menu.Idle()
|
|
l.changeMode(ModeDefault)
|
|
}
|
|
case console.KeyLeft:
|
|
switch l.mode {
|
|
case ModeDefault:
|
|
l.changeMode(ModeComponentSelect)
|
|
l.menu.Highlight(l.getComponent(l.selection))
|
|
case ModeChartPinpoint:
|
|
chart := l.getSelectedComponent().Drawable.(*runchart.RunChart)
|
|
chart.MoveSelection(-1)
|
|
case ModeComponentSelect:
|
|
l.moveSelection(e)
|
|
l.menu.Highlight(l.getComponent(l.selection))
|
|
case ModeComponentMove:
|
|
l.getSelectedComponent().Move(-1, 0)
|
|
case ModeComponentResize:
|
|
l.getSelectedComponent().Resize(-1, 0)
|
|
}
|
|
case console.KeyRight:
|
|
switch l.mode {
|
|
case ModeDefault:
|
|
l.changeMode(ModeComponentSelect)
|
|
l.menu.Highlight(l.getComponent(l.selection))
|
|
case ModeChartPinpoint:
|
|
chart := l.getSelectedComponent().Drawable.(*runchart.RunChart)
|
|
chart.MoveSelection(1)
|
|
case ModeComponentSelect:
|
|
l.moveSelection(e)
|
|
l.menu.Highlight(l.getComponent(l.selection))
|
|
case ModeComponentMove:
|
|
l.getSelectedComponent().Move(1, 0)
|
|
case ModeComponentResize:
|
|
l.getSelectedComponent().Resize(1, 0)
|
|
}
|
|
case console.KeyUp:
|
|
switch l.mode {
|
|
case ModeDefault:
|
|
l.changeMode(ModeComponentSelect)
|
|
l.menu.Highlight(l.getComponent(l.selection))
|
|
case ModeComponentSelect:
|
|
l.moveSelection(e)
|
|
l.menu.Highlight(l.getComponent(l.selection))
|
|
case ModeMenuOptionSelect:
|
|
l.menu.Up()
|
|
case ModeComponentMove:
|
|
l.getSelectedComponent().Move(0, -1)
|
|
case ModeComponentResize:
|
|
l.getSelectedComponent().Resize(0, -1)
|
|
}
|
|
case console.KeyDown:
|
|
switch l.mode {
|
|
case ModeDefault:
|
|
l.changeMode(ModeComponentSelect)
|
|
l.menu.Highlight(l.getComponent(l.selection))
|
|
case ModeComponentSelect:
|
|
l.moveSelection(e)
|
|
l.menu.Highlight(l.getComponent(l.selection))
|
|
case ModeMenuOptionSelect:
|
|
l.menu.Down()
|
|
case ModeComponentMove:
|
|
l.getSelectedComponent().Move(0, 1)
|
|
case ModeComponentResize:
|
|
l.getSelectedComponent().Resize(0, 1)
|
|
}
|
|
}
|
|
}
|
|
|
|
func (l *Layout) ChangeDimensions(width, height int) {
|
|
l.SetRect(0, 0, width, height)
|
|
}
|
|
|
|
func (l *Layout) getComponent(i int) *component.Component {
|
|
return &l.Components[i]
|
|
}
|
|
|
|
func (l *Layout) getSelectedComponent() *component.Component {
|
|
return &l.Components[l.selection]
|
|
}
|
|
|
|
func (l *Layout) moveSelection(direction string) {
|
|
|
|
previouslySelected := *l.getSelectedComponent()
|
|
newlySelectedIndex := l.selection
|
|
|
|
for i, current := range l.Components {
|
|
|
|
if current == previouslySelected {
|
|
continue
|
|
}
|
|
|
|
if newlySelectedIndex < 0 {
|
|
newlySelectedIndex = i
|
|
}
|
|
|
|
var previouslySelectedCornerPoint image.Point
|
|
var newlySelectedCornerPoint image.Point
|
|
var currentCornerPoint image.Point
|
|
|
|
switch direction {
|
|
case console.KeyLeft:
|
|
previouslySelectedCornerPoint = component.GetRectLeftAgeCenter(previouslySelected.Drawable.GetRect())
|
|
newlySelectedCornerPoint = component.GetRectRightAgeCenter(l.getComponent(newlySelectedIndex).Drawable.GetRect())
|
|
currentCornerPoint = component.GetRectRightAgeCenter(current.Drawable.GetRect())
|
|
case console.KeyRight:
|
|
previouslySelectedCornerPoint = component.GetRectRightAgeCenter(previouslySelected.Drawable.GetRect())
|
|
newlySelectedCornerPoint = component.GetRectLeftAgeCenter(l.getComponent(newlySelectedIndex).Drawable.GetRect())
|
|
currentCornerPoint = component.GetRectLeftAgeCenter(current.Drawable.GetRect())
|
|
case console.KeyUp:
|
|
previouslySelectedCornerPoint = component.GetRectTopAgeCenter(previouslySelected.Drawable.GetRect())
|
|
newlySelectedCornerPoint = component.GetRectBottomAgeCenter(l.getComponent(newlySelectedIndex).Drawable.GetRect())
|
|
currentCornerPoint = component.GetRectBottomAgeCenter(current.Drawable.GetRect())
|
|
case console.KeyDown:
|
|
previouslySelectedCornerPoint = component.GetRectBottomAgeCenter(previouslySelected.Drawable.GetRect())
|
|
newlySelectedCornerPoint = component.GetRectTopAgeCenter(l.getComponent(newlySelectedIndex).Drawable.GetRect())
|
|
currentCornerPoint = component.GetRectTopAgeCenter(current.Drawable.GetRect())
|
|
}
|
|
|
|
if component.GetDistance(previouslySelectedCornerPoint, currentCornerPoint) <
|
|
component.GetDistance(previouslySelectedCornerPoint, newlySelectedCornerPoint) {
|
|
newlySelectedIndex = i
|
|
}
|
|
}
|
|
|
|
l.selection = newlySelectedIndex
|
|
}
|
|
|
|
func (l *Layout) Draw(buffer *ui.Buffer) {
|
|
|
|
columnWidth := float64(l.GetRect().Dx()) / float64(columnsCount)
|
|
rowHeight := float64(l.GetRect().Dy()-statusbarHeight) / float64(rowsCount)
|
|
|
|
for _, c := range l.Components {
|
|
|
|
x1 := math.Floor(float64(c.Position.X) * columnWidth)
|
|
y1 := math.Floor(float64(c.Position.Y) * rowHeight)
|
|
x2 := x1 + math.Floor(float64(c.Size.X))*columnWidth
|
|
y2 := y1 + math.Floor(float64(c.Size.Y))*rowHeight
|
|
|
|
if x2-x1 < minDimension {
|
|
x2 = x1 + minDimension
|
|
}
|
|
|
|
if y2-y1 < minDimension {
|
|
y2 = y1 + minDimension
|
|
}
|
|
|
|
c.Drawable.SetRect(int(x1), int(y1), int(x2), int(y2))
|
|
c.Drawable.Draw(buffer)
|
|
}
|
|
|
|
l.statusbar.SetRect(
|
|
0, l.GetRect().Dy()-statusbarHeight,
|
|
l.GetRect().Dx(), l.GetRect().Dy())
|
|
|
|
l.statusbar.Draw(buffer)
|
|
l.menu.Draw(buffer)
|
|
}
|