fix(heatmap): fix bug legend was not fully boxed (#583)

pterm · Nov 1, 2023 · 5c1ca65 · 5c1ca65
1 parent 6105d7f
commit 5c1ca65
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/heatmap_printer.go b/heatmap_printer.go
@@ -585,11 +585,14 @@ func (p HeatmapPrinter) boxLegend(buffer *bytes.Buffer, legend string, legendCol
 }
 
 func (p HeatmapPrinter) generateSeparatorRow(buffer *bytes.Buffer, legend string, legendColWidth int, top bool) {
+	p.rgbLegendValue = 11
 	steps := len(p.RGBRange)
 	if steps < p.rgbLegendValue {
 		steps = p.rgbLegendValue
 	}
-	steps *= 3
+	if p.LegendOnlyColoredCells {
+		steps *= 3
+	}
 
 	var xValue int
 	if p.EnableRGB {