tree structure change

internal/gopow/gopow.go

@@ -0,0 +1,109 @@
+package gopow
+
+import (
+	"fmt"
+	"image"
+	"image/png"
+	"os"
+
+	log "github.com/Sirupsen/logrus"
+	"github.com/codegangsta/cli"
+)
+
+type RunConfig struct {
+	InputFile  string
+	OutputFile string
+	Format     string
+}
+
+type GoPow struct {
+	config *RunConfig
+	image  *image.RGBA
+}
+
+func NewGoPow(c *cli.Context) (*GoPow, error) {
+
+	config := &RunConfig{
+		InputFile:  c.String("input"),
+		OutputFile: c.String("output"),
+		Format:     c.String("format"),
+	}
+
+	if config.InputFile == "" {
+		return nil, fmt.Errorf("missing input file")
+	}
+
+	if config.Format == "" {
+		config.Format = "png"
+	}
+
+	if config.OutputFile == "" {
+		config.OutputFile = config.InputFile + "." + config.Format
+	}
+
+	log.WithFields(log.Fields{
+		"input": config.InputFile,
+	}).Info("GoPow init")
+	log.WithFields(log.Fields{
+		"output": config.OutputFile,
+	}).Info("GoPow init")
+	log.WithFields(log.Fields{
+		"format": config.Format,
+	}).Info("GoPow init")
+
+	g := &GoPow{
+		config: config,
+	}
+
+	return g, nil
+}
+
+func (g *GoPow) Render() error {
+
+	log.Debug("staring render")
+
+	table, err := NewTable(g.config.InputFile)
+	if err != nil {
+		return err
+	}
+
+	g.image = table.Image()
+
+	for y, row := range table.Rows {
+		for x, _ := range row.Samples {
+			g.image.Set(x, y, table.ColorAt(x, y))
+		}
+	}
+
+	// add some frequency and date annotation
+	err = table.AnnotateXScale(g.image)
+	if err != nil {
+		return err
+	}
+
+	err = table.AnnotateYScale(g.image)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func (g *GoPow) Write() error {
+
+	log.WithFields(log.Fields{
+		"file": g.config.OutputFile,
+	}).Debug("staring output write")
+
+	out, err := os.Create(g.config.OutputFile)
+	if err != nil {
+		return err
+	}
+
+	err = png.Encode(out, g.image)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}