image collector
This commit is contained in:
parent
ce5b36a6cb
commit
78e6288748
1 changed files with 41 additions and 0 deletions
41
image_writer.go
Normal file
41
image_writer.go
Normal file
|
@ -0,0 +1,41 @@
|
|||
package chart
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"errors"
|
||||
"image"
|
||||
"image/png"
|
||||
)
|
||||
|
||||
type RGBACollector interface {
|
||||
SetRGBA(i *image.RGBA)
|
||||
}
|
||||
|
||||
// ImageWriter is a special type of io.Writer that produces a final image.
|
||||
type ImageWriter struct {
|
||||
rgba *image.RGBA
|
||||
contents *bytes.Buffer
|
||||
}
|
||||
|
||||
func (ir *ImageWriter) Write(buffer []byte) (int, error) {
|
||||
if ir.contents == nil {
|
||||
ir.contents = bytes.NewBuffer([]byte{})
|
||||
}
|
||||
return ir.contents.Write(buffer)
|
||||
}
|
||||
|
||||
// SetRGBA sets a raw version of the image.
|
||||
func (ir *ImageWriter) SetRGBA(i *image.RGBA) {
|
||||
ir.rgba = i
|
||||
}
|
||||
|
||||
// Image returns an *image.Image for the result.
|
||||
func (ir *ImageWriter) Image() (image.Image, error) {
|
||||
if ir.rgba != nil {
|
||||
return ir.rgba, nil
|
||||
}
|
||||
if ir.contents != nil && ir.contents.Len() > 0 {
|
||||
return png.Decode(ir.contents)
|
||||
}
|
||||
return nil, errors.New("No valid sources for image data, cannot continue.")
|
||||
}
|
Loading…
Reference in a new issue