diff options
| -rw-r--r-- | aws.go | 11 | ||||
| -rw-r--r-- | graph.go | 4 | ||||
| -rw-r--r-- | pdf.go | 1 | 
3 files changed, 12 insertions, 4 deletions
| @@ -20,7 +20,7 @@ import (  	"github.com/aws/aws-sdk-go/service/sqs"  ) -const PreprocPattern = `_bin[0-9].[0-9].png` +const defaultAwsRegion = `eu-west-2`  type Qmsg struct {  	Id, Handle, Body string @@ -35,8 +35,11 @@ type ObjMeta struct {  	Date time.Time  } +// AwsConn contains the necessary things to interact with various AWS +// services in ways useful for the bookpipeline package. It is +// designed to be generic enough to swap in other backends easily.  type AwsConn struct { -	// these need to be set before running Init() +	// these should be set before running Init(), or left to defaults  	Region string  	Logger *log.Logger @@ -53,10 +56,10 @@ type AwsConn struct {  // MinimalInit does the bare minimum to initialise aws services  func (a *AwsConn) MinimalInit() error {  	if a.Region == "" { -		return errors.New("No Region set") +		a.Region = defaultAwsRegion  	}  	if a.Logger == nil { -		return errors.New("No logger set") +		a.Logger = log.New(os.Stdout, "", 0)  	}  	var err error @@ -32,6 +32,8 @@ type GraphConf struct {  	Pgnum, Conf float64  } +// createLine creates a horizontal line with a particular y value for +// a graph  func createLine(xvalues []float64, y float64, c drawing.Color) chart.ContinuousSeries {  	var yvalues []float64  	for range xvalues { @@ -46,10 +48,12 @@ func createLine(xvalues []float64, y float64, c drawing.Color) chart.ContinuousS  	}  } +// Graph creates a graph of the confidence of each page in a book  func Graph(confs map[string]*Conf, bookname string, w io.Writer) error {  	return GraphOpts(confs, bookname, "Page number", true, w)  } +// Graph creates a graph of confidences  func GraphOpts(confs map[string]*Conf, bookname string, xaxis string, guidelines bool, w io.Writer) error {  	if len(confs) < 2 {  		return errors.New("Not enough valid confidences") @@ -31,6 +31,7 @@ func pxToPt(i int) float64 {  	return float64(i) / pageWidth  } +// Fpdf abstracts the gofpdf.Fpdf adding some useful methods  type Fpdf struct {  	fpdf *gofpdf.Fpdf  } | 
