Fix context size calculationHEADmaster

1 files changed, 3 insertions, 4 deletions

diff --git a/ntrop.c b/ntrop.c
index 55f6aec..9697c67 100644
--- 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++) {