Mirror/MarginaliaSearch
1
0
Fork 0
mirror of https://github.com/MarginaliaSearch/MarginaliaSearch.git synced 2025-02-24 05:18:58 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

(gamma) Fix readCount() behavior in EGC

Browse Source
This commit is contained in:
Viktor Lofgren 2024-06-25 22:17:54 +02:00
parent dae22ccbe0
commit b805f6daa8
2 changed files with 11 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
code/libraries/coded-sequence/java/nu/marginalia/sequence/EliasGammaCodec.java
View File

@ -37,12 +37,12 @@ public class EliasGammaCodec implements IntIterator {
public static int readCount(ByteBuffer buffer) { public static int readCount(ByteBuffer buffer) {
var reader = new BitReader(buffer); var reader = new BitReader(buffer);
if (reader.getCurrentValue() > 0) {
int bits = reader.takeWhileZero(); int bits = reader.takeWhileZero();
return reader.get(bits); if (!reader.hasMore()) {
return 0;
} }
else { else {
return 0; return reader.get(bits);
} }
} }

7
code/libraries/coded-sequence/test/nu/marginalia/sequence/EliasGammaCodecTest.java
View File

@ -30,6 +30,13 @@ class EliasGammaCodecTest {
assertEquals(expected, decoded); assertEquals(expected, decoded);
} }
@Test
public void valueCount() {
var ret = EliasGammaCodec.encode(work, new int[] { 1, 3, 5, 16, 32, 64 });
var count = EliasGammaCodec.readCount(ret);
assertEquals(6, count);
}
@Test @Test
public void testCodec2() { public void testCodec2() {
var ret = EliasGammaCodec.encode(work, new int[] { 1, 256 }); var ret = EliasGammaCodec.encode(work, new int[] { 1, 256 });