ntrop

commit 3cd9d8d7155f8f138c03559885b4e6099df60369
parent 7d3db45288214224caa72a883d66764d905b34cd
Author: Louis Burda <quent.burda@gmail.com>
Date:   Wed, 23 Aug 2023 16:22:37 +0200

Fix context size calculation

Diffstat:
M
ntrop.c
 | 
7
+++----

1 file changed, 3 insertions(+), 4 deletions(-)
diff --git a/ntrop.c b/ntrop.c
@@ -129,12 +129,10 @@ init_entropy_colors(Color *data_colors)
 	c.a = 255;
 	for (pos = 0; pos < data_len; pos++) {
 		uniq = 0;
-		ctx_count = entropy_ctx * 2 + 1;
+		ctx_count = 0;
 		for (i = -entropy_ctx; i <= entropy_ctx; i++) {
-			if (pos + i < 0 || pos + i > data_len) {
-				ctx_count--;
+			if (pos + i < 0 || pos + i > data_len)
 				continue;
-			}
 			for (k = 0; k < uniq; k++) {
 				if (vals[k] == file_data[pos + i]) {
 					counts[k] += 1;
@@ -146,6 +144,7 @@ init_entropy_colors(Color *data_colors)
 				vals[uniq] = file_data[pos + i];
 				uniq += 1;
 			}
+			ctx_count++;
 		}
 		entropy = 0;
 		for (k = 0; k < uniq; k++) {