fixes #564 Excessive Ram usage for colors in direct / Tc mode

gdamore · Oct 17, 2022 · 7557ac2 · 7557ac2
1 parent 178ac43
commit 7557ac2
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/tscreen.go b/tscreen.go
@@ -199,9 +199,13 @@ func (t *tScreen) Init() error {
 	if os.Getenv("TCELL_TRUECOLOR") == "disable" {
 		t.truecolor = false
 	}
-	t.colors = make(map[Color]Color)
-	t.palette = make([]Color, t.nColors())
-	for i := 0; i < t.nColors(); i++ {
+	nColors := t.nColors()
+	if nColors > 256 {
+		nColors = 256 // clip to reasonable limits
+	}
+	t.colors = make(map[Color]Color, nColors)
+	t.palette = make([]Color, nColors)
+	for i := 0; i < nColors; i++ {
 		t.palette[i] = Color(i) | ColorValid
 		// identity map for our builtin colors
 		t.colors[Color(i)|ColorValid] = Color(i) | ColorValid