termui/_examples/linechart.go

// Copyright 2017 Zack Guo <zack.y.guo@gmail.com>. All rights reserved.
// Use of this source code is governed by a MIT license that can
// be found in the LICENSE file.

// +build ignore

package main

import (
	"log"
	"math"

	ui "github.com/gizak/termui"
	"github.com/gizak/termui/widgets"
)

func main() {
	if err := ui.Init(); err != nil {
		log.Fatalf("failed to initialize termui: %v", err)
	}
	defer ui.Close()

	sinData := func() [][]float64 {
		n := 220
		data := make([][]float64, 2)
		data[0] = make([]float64, n)
		data[1] = make([]float64, n)
		for i := 0; i < n; i++ {
			data[0][i] = 1 + math.Sin(float64(i)/5)
			data[1][i] = 1 + math.Cos(float64(i)/5)
		}
		return data
	}()

	lc0 := widgets.NewLineChart()
	lc0.Title = "braille-mode Line Chart"
	lc0.Data = sinData
	lc0.SetRect(0, 0, 50, 15)
	lc0.AxesColor = ui.ColorWhite
	lc0.LineColors[0] = ui.ColorGreen

	lc1 := widgets.NewLineChart()
	lc1.Title = "custom Line Chart"
	lc1.LineType = widgets.DotLine
	lc1.Data = [][]float64{[]float64{1, 2, 3, 4, 5}}
	lc1.SetRect(50, 0, 75, 10)
	lc1.DotChar = '+'
	lc1.AxesColor = ui.ColorWhite
	lc1.LineColors[0] = ui.ColorYellow
	lc1.DrawDirection = widgets.DrawLeft

	lc2 := widgets.NewLineChart()
	lc2.Title = "dot-mode Line Chart"
	lc2.LineType = widgets.DotLine
	lc2.Data = make([][]float64, 2)
	lc2.Data[0] = sinData[0][4:]
	lc2.Data[1] = sinData[1][4:]
	lc2.SetRect(0, 15, 50, 30)
	lc2.AxesColor = ui.ColorWhite
	lc2.LineColors[0] = ui.ColorCyan

	ui.Render(lc0, lc1, lc2)

	uiEvents := ui.PollEvents()
	for {
		e := <-uiEvents
		switch e.ID {
		case "q", "<C-c>":
			return
		}
	}
}