1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87
|
package main
import (
"image"
"image/draw"
"image/png"
"os"
"strconv"
"github.com/lucasb-eyer/go-colorful"
)
// This table contains the "keypoints" of the colorgradient you want to generate.
// The position of each keypoint has to live in the range [0,1]
type GradientTable []struct {
Col colorful.Color
Pos float64
}
// This is the meat of the gradient computation. It returns a HCL-blend between
// the two colors around `t`.
// Note: It relies heavily on the fact that the gradient keypoints are sorted.
func (gt GradientTable) GetInterpolatedColorFor(t float64) colorful.Color {
for i := 0; i < len(gt)-1; i++ {
c1 := gt[i]
c2 := gt[i+1]
if c1.Pos <= t && t <= c2.Pos {
// We are in between c1 and c2. Go blend them!
t := (t - c1.Pos) / (c2.Pos - c1.Pos)
return c1.Col.BlendHcl(c2.Col, t).Clamped()
}
}
// Nothing found? Means we're at (or past) the last gradient keypoint.
return gt[len(gt)-1].Col
}
// This is a very nice thing Golang forces you to do!
// It is necessary so that we can write out the literal of the colortable below.
func MustParseHex(s string) colorful.Color {
c, err := colorful.Hex(s)
if err != nil {
panic("MustParseHex: " + err.Error())
}
return c
}
func main() {
// The "keypoints" of the gradient.
keypoints := GradientTable{
{MustParseHex("#9e0142"), 0.0},
{MustParseHex("#d53e4f"), 0.1},
{MustParseHex("#f46d43"), 0.2},
{MustParseHex("#fdae61"), 0.3},
{MustParseHex("#fee090"), 0.4},
{MustParseHex("#ffffbf"), 0.5},
{MustParseHex("#e6f598"), 0.6},
{MustParseHex("#abdda4"), 0.7},
{MustParseHex("#66c2a5"), 0.8},
{MustParseHex("#3288bd"), 0.9},
{MustParseHex("#5e4fa2"), 1.0},
}
h := 1024
w := 40
if len(os.Args) == 3 {
// Meh, I'm being lazy...
w, _ = strconv.Atoi(os.Args[1])
h, _ = strconv.Atoi(os.Args[2])
}
img := image.NewRGBA(image.Rect(0, 0, w, h))
for y := h - 1; y >= 0; y-- {
c := keypoints.GetInterpolatedColorFor(float64(y) / float64(h))
draw.Draw(img, image.Rect(0, y, w, y+1), &image.Uniform{c}, image.Point{}, draw.Src)
}
outpng, err := os.Create("gradientgen.png")
if err != nil {
panic("Error storing png: " + err.Error())
}
defer outpng.Close()
png.Encode(outpng, img)
}
|