Assign random color per cluster through random created color-palette

Author: PascalDierich

.gitignore

@@ -1,2 +1,3 @@
 .idea
 vendor/
+image-clustering

cluster.go

@@ -5,13 +5,9 @@ import (
 	"github.com/lucasb-eyer/go-colorful"
 )
 
-func clusterImage(k int, img image.Image) image.Image {
+func clusterImage(k int, img image.Image) image.Image{
 	cls := getClusters(k, img)
-
-	// assign each cluster a random color
-	for _, c := range cls {
-		c.centroid = colorful.WarmColor() // save color in centroid
-	}
+	assignColors(cls)
 
 	// create new image
 	newImg := image.NewRGBA(img.Bounds())
@@ -22,3 +18,11 @@ func clusterImage(k int, img image.Image) image.Image {
 	}
 	return newImg
 }
+
+// assign each cluster a random color
+func assignColors(cls []*cluster) {
+	pal := colorful.FastWarmPalette(len(cls))
+	for i, c := range cls {
+		c.centroid = pal[i]
+	}
+}