commit 37a5894e41ad6847b95c1cfdf2cde5fd51c3dd8c Author: Tai Groot Date: Thu Jun 16 06:00:27 2022 -0600 work on svg generation diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..9c4f66c --- /dev/null +++ b/.gitignore @@ -0,0 +1,2 @@ +*.svg +gitgraph diff --git a/go.mod b/go.mod new file mode 100644 index 0000000..3859143 --- /dev/null +++ b/go.mod @@ -0,0 +1,18 @@ +module github.com/taigrr/gitgraph + +go 1.18 + +replace github.com/taigrr/go-colorpallettes => ../go-colorpallettes + +require ( + github.com/ajstarks/svgo v0.0.0-20211024235047-1546f124cd8b + github.com/srwiley/oksvg v0.0.0-20220128195007-1f435e4c2b44 + github.com/srwiley/rasterx v0.0.0-20220615024203-67b7089efd25 + github.com/taigrr/go-colorpallettes v0.0.0-00010101000000-000000000000 +) + +require ( + golang.org/x/image v0.0.0-20211028202545-6944b10bf410 // indirect + golang.org/x/net v0.0.0-20211118161319-6a13c67c3ce4 // indirect + golang.org/x/text v0.3.6 // indirect +) diff --git a/go.sum b/go.sum new file mode 100644 index 0000000..0bcf8d6 --- /dev/null +++ b/go.sum @@ -0,0 +1,39 @@ +github.com/BurntSushi/toml v0.3.1/go.mod h1:xHWCNGjB5oqiDr8zfno3MHue2Ht5sIBksp03qcyfWMU= +github.com/ajstarks/deck v0.0.0-20200831202436-30c9fc6549a9/go.mod h1:JynElWSGnm/4RlzPXRlREEwqTHAN3T56Bv2ITsFT3gY= +github.com/ajstarks/deck/generate v0.0.0-20210309230005-c3f852c02e19/go.mod h1:T13YZdzov6OU0A1+RfKZiZN9ca6VeKdBdyDV+BY97Tk= +github.com/ajstarks/svgo v0.0.0-20211024235047-1546f124cd8b h1:slYM766cy2nI3BwyRiyQj/Ud48djTMtMebDqepE95rw= +github.com/ajstarks/svgo v0.0.0-20211024235047-1546f124cd8b/go.mod h1:1KcenG0jGWcpt8ov532z81sp/kMMUG485J2InIOyADM= +github.com/kisielk/gotool v1.0.0/go.mod h1:XhKaO+MFFWcvkIS/tQcRk01m1F5IRFswLeQ+oQHNcck= +github.com/srwiley/oksvg v0.0.0-20220128195007-1f435e4c2b44 h1:XPYXKIuH/n5zpUoEWk2jWV/SjEMNYmqDYmTgbjmhtaI= +github.com/srwiley/oksvg v0.0.0-20220128195007-1f435e4c2b44/go.mod h1:cNQ3dwVJtS5Hmnjxy6AgTPd0Inb3pW05ftPSX7NZO7Q= +github.com/srwiley/rasterx v0.0.0-20220615024203-67b7089efd25 h1:uyi1c8AxAw2b9MTr639iqz45FMFb4dDR7rB++NaVooo= +github.com/srwiley/rasterx v0.0.0-20220615024203-67b7089efd25/go.mod h1:nXTWP6+gD5+LUJ8krVhhoeHjvHTutPxMYl5SvkcnJNE= +github.com/yuin/goldmark v1.2.1/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74= +golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w= +golang.org/x/crypto v0.0.0-20191011191535-87dc89f01550/go.mod h1:yigFU9vqHzYiE8UmvKecakEJjdnWj3jj499lnFckfCI= +golang.org/x/crypto v0.0.0-20200622213623-75b288015ac9/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto= +golang.org/x/image v0.0.0-20211028202545-6944b10bf410 h1:hTftEOvwiOq2+O8k2D5/Q7COC7k5Qcrgc2TFURJYnvQ= +golang.org/x/image v0.0.0-20211028202545-6944b10bf410/go.mod h1:023OzeP/+EPmXeapQh35lcL3II3LrY8Ic+EFFKVhULM= +golang.org/x/mod v0.3.0/go.mod h1:s0Qsj1ACt9ePp/hMypM3fl4fZqREWJwdYDEqhRiZZUA= +golang.org/x/net v0.0.0-20190404232315-eb5bcb51f2a3/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg= +golang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s= +golang.org/x/net v0.0.0-20201021035429-f5854403a974/go.mod h1:sp8m0HH+o8qH0wwXwYZr8TS3Oi6o0r6Gce1SSxlDquU= +golang.org/x/net v0.0.0-20211118161319-6a13c67c3ce4 h1:DZshvxDdVoeKIbudAdFEKi+f70l51luSy/7b76ibTY0= +golang.org/x/net v0.0.0-20211118161319-6a13c67c3ce4/go.mod h1:9nx3DQGgdP8bBQD5qxJ1jj9UTztislL4KSBs9R2vV5Y= +golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sync v0.0.0-20201020160332-67f06af15bc9/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= +golang.org/x/sys v0.0.0-20190412213103-97732733099d/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20200930185726-fdedc70b468f/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20210119212857-b64e53b001e4/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ= +golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ= +golang.org/x/text v0.3.6 h1:aRYxNxv6iGQlyVaZmk6ZgYEDa+Jg18DxebPSrd6bg1M= +golang.org/x/text v0.3.6/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ= +golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ= +golang.org/x/tools v0.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo= +golang.org/x/tools v0.1.0/go.mod h1:xkSsbof2nBLbhDlRMhhhyNLN/zl3eTqcnHD5viDpcZ0= +golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= +golang.org/x/xerrors v0.0.0-20191011141410-1b5146add898/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= +golang.org/x/xerrors v0.0.0-20200804184101-5ec99f83aff1/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= +honnef.co/go/tools v0.1.3/go.mod h1:NgwopIslSNH47DimFoV78dnkksY2EFtX0ajyb3K/las= diff --git a/graph/graph.go b/graph/graph.go new file mode 100644 index 0000000..75ce734 --- /dev/null +++ b/graph/graph.go @@ -0,0 +1,116 @@ +package graph + +import ( + "bufio" + "bytes" + "fmt" + "image" + "image/color" + "image/png" + "os" + "sync" + + svg "github.com/ajstarks/svgo" + "github.com/srwiley/oksvg" + "github.com/srwiley/rasterx" + sc "github.com/taigrr/go-colorpallettes/simplecolor" +) + +var colorsLoaded sync.Once +var colorScheme []sc.SimpleColor + +func CreateGraph() bytes.Buffer { + var x bytes.Buffer + return x +} + +func init() { + colors := []string{"#767960", "#a7297f", "#e8ca89", "#f5efd6", "#858966"} + for _, c := range colors { + color := sc.FromHexString(c) + _ = c + colorScheme = append(colorScheme, color) + } + +} + +func SetColorScheme(c []color.Color) { + for _, c := range c { + colorScheme = append(colorScheme, sc.FromRGBA(c.RGBA())) + } +} + +func ColorForFrequency(freq, min, max int) color.Color { + spread := max - min + interval := float64(spread) / float64(len(colorScheme)) + colorIndex := 0 + for i := float64(min); i < float64(freq); i += float64(interval) { + colorIndex++ + } + if colorIndex > len(colorScheme)-1 { + colorIndex = len(colorScheme) - 1 + } + return colorScheme[colorIndex] +} +func GetImage(frequencies []int) bytes.Buffer { + squareColors := []color.Color{} + min, max := minmax(frequencies) + for _, f := range frequencies { + squareColors = append(squareColors, ColorForFrequency(f, min, max)) + } + return drawImage(squareColors) +} + +func svgToPng() { + w, h := 512, 512 + + in, err := os.Open("in.svg") + if err != nil { + panic(err) + } + defer in.Close() + + icon, _ := oksvg.ReadIconStream(in) + icon.SetTarget(0, 0, float64(w), float64(h)) + rgba := image.NewRGBA(image.Rect(0, 0, w, h)) + icon.Draw(rasterx.NewDasher(w, h, rasterx.NewScannerGV(w, h, rgba, rgba.Bounds())), 1) + + out, err := os.Create("out.png") + if err != nil { + panic(err) + } + defer out.Close() + + err = png.Encode(out, rgba) + if err != nil { + panic(err) + } +} + +func drawImage(c []sc.SimpleColor) bytes.Buffer { + //TODO here, draw suqares in appropriate colors, hopefully as an svg + var sb bytes.Buffer + sbw := bufio.NewWriter(&sb) + width := 500 + height := 500 + canvas := svg.New(sbw) + canvas.Start(width, height) + canvas.Square(width/20, height/300, width/10, fmt.Sprintf("fill:%s", colorScheme[0].HexString())) + canvas.Circle(width/2, height/2, 100) + canvas.Text(width/2, height/2, "Hello, SVG", "text-anchor:middle;font-size:30px;fill:white") + canvas.End() + sbw.Flush() + return sb +} + +func minmax(f []int) (int, int) { + min, max := 0, 0 + for _, x := range f { + if x < min { + min = x + } else if x > max { + max = x + } + } + return min, max +} diff --git a/main.go b/main.go new file mode 100644 index 0000000..6af0efa --- /dev/null +++ b/main.go @@ -0,0 +1,21 @@ +package main + +import ( + "fmt" + "os" + + "github.com/taigrr/gitgraph/graph" +) + +type DayCount [366]int + +func main() { + svg := graph.GetImage([]int{1, 2, 5, 6, 5, 4, 5, 8, 7, 43, 2, 3}) + f, err := os.Create("out.svg") + if err != nil { + fmt.Printf("Error: %v\n", err) + return + } + defer f.Close() + svg.WriteTo(f) +}