sashakoshka/termui
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Cleanup piechart

Browse Source
This commit is contained in:
Caleb Bassi 2018-08-16 20:22:32 -07:00
parent 10d90f04d8
commit 3bbd34011a
2 changed files with 15 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
_example/piechart.go
View File

@ -4,10 +4,11 @@ package main
import ( import (
"fmt" "fmt"
"github.com/gizak/termui"
"math" "math"
"math/rand" "math/rand"
"time" "time"
"github.com/gizak/termui"
) )
func main() { func main() {

28
piechart.go
View File

@ -35,16 +35,17 @@ type PieChart struct {
Colors []Attribute // colors to by cycled through (see defaultColors) Colors []Attribute // colors to by cycled through (see defaultColors)
BorderColor Attribute // color of the pie-border BorderColor Attribute // color of the pie-border
Label PieChartLabel // callback function for labels Label PieChartLabel // callback function for labels
Offset float64 // where to start drawing? (see piechartOffsetUp) Offset float64 // which angle to start drawing at? (see piechartOffsetUp)
} }
// Creates a new pie chart with reasonable defaults and no labels // NewPieChart Creates a new pie chart with reasonable defaults and no labels
func NewPieChart() *PieChart { func NewPieChart() *PieChart {
pc := &PieChart{Block: *NewBlock()} return &PieChart{
pc.Colors = defaultColors Block: *NewBlock(),
pc.Offset = piechartOffsetUp Colors: defaultColors,
pc.BorderColor = ColorDefault Offset: piechartOffsetUp,
return pc BorderColor: ColorDefault,
}
} }
// computes the color for a given data index // computes the color for a given data index
@ -52,18 +53,16 @@ func (pc *PieChart) colorFor(i int) Attribute {
return pc.Colors[i%len(pc.Colors)] return pc.Colors[i%len(pc.Colors)]
} }
// creates the buffer for the pie chart // Buffer creates the buffer for the pie chart
func (pc *PieChart) Buffer() Buffer { func (pc *PieChart) Buffer() Buffer {
buf := pc.Block.Buffer() buf := pc.Block.Buffer()
w, h := pc.innerArea.Dx(), pc.innerArea.Dy() w, h := pc.innerArea.Dx(), pc.innerArea.Dy()
center := image.Point{X: w / 2, Y: h / 2} center := image.Point{X: w / 2, Y: h / 2}
// radius for the border // radius for the border
r := 0.0 r := float64(w/2/xStretch) - 1.0
if h < w/xStretch { if h < w/xStretch {
r = float64(h/2) - 1.0 r = float64(h/2) - 1.0
} else {
r = float64(w/2/xStretch) - 1.0
} }
// make border // make border
@ -148,10 +147,10 @@ func fill(p image.Point, c *Cell, buf *Buffer) {
w, e, row := p.X, p.X, p.Y w, e, row := p.X, p.X, p.Y
for empty(w-1, row) { for empty(w-1, row) {
w -= 1 w--
} }
for empty(e+1, row) { for empty(e+1, row) {
e += 1 e++
} }
for x := w; x <= e; x++ { for x := w; x <= e; x++ {
buf.Set(x, row, *c) buf.Set(x, row, *c)
@ -286,7 +285,6 @@ func sum(data []float64) float64 {
func abs(x int) int { func abs(x int) int {
if x >= 0 { if x >= 0 {
return x return x
} else { }
return -x return -x
} }
}