FEAT: partial crush decompression

Oldes · Dec 10, 2021 · f64cd37 · f64cd37
1 parent d744cee
commit f64cd37
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/src/core/u-crush.c b/src/core/u-crush.c
@@ -359,6 +359,9 @@ static inline int get_penalty(int a, int b) {
 	ctx.bit_count = 0;
 	ctx.bit_buf = 0;
 
+	// allow partial output if specified...
+	if (limit && size > limit) size = limit;
+
 	ser = Make_Binary(size); // output series
 	buf = BIN_HEAD(ser);     // output binary
 

diff --git a/src/tests/units/compress-test.r3 b/src/tests/units/compress-test.r3
@@ -175,9 +175,9 @@ text: {Lorem ipsum dolor sit amet, consectetur adipisici elit, sed eiusmod tempo
 			--assert  text = to string! decompress compress/level text 'crush 1 'crush
 			--assert  text = to string! decompress compress/level text 'crush 2 'crush
 
-;		--test-- "CRUSH decompression with specified uncompressed size"
-;			bin: compress data 'crush
-;			--assert  #{74657374} = decompress/size bin 'crush 4
+		--test-- "CRUSH decompression with specified uncompressed size"
+			bin: compress data 'crush
+			--assert  #{74657374} = decompress/size bin 'crush 4
 
 		--test-- "CRUSH compression when input is limited"
 			--assert  #{74657374} = decompress compress/part      data   'crush  4 'crush