logos-storage/nim-poseidon2
1
0
Fork 0
mirror of https://github.com/logos-storage/nim-poseidon2.git synced 2026-01-07 16:13:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix empty seq support

Browse Source
This commit is contained in:
Dmitriy Ryajov 2023-12-17 16:46:36 -06:00
parent 7be0e6dfe7
commit 59f7b0545e
No known key found for this signature in database
GPG Key ID: DA8C680CE7C657A4
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
poseidon2/io.nim
View File

@ -27,13 +27,14 @@ iterator elements*(bytes: openArray[byte], _: type F): F =
## sequences that end with 0x0's. ## sequences that end with 0x0's.
const chunkLen = 31 const chunkLen = 31
const endMarker = @[1'u8] const endMarker = @[1'u8]
let empty = bytes.len == 0
var chunkStart = 0 var chunkStart = 0
while chunkStart + chunkLen <= bytes.len: while chunkStart + chunkLen <= bytes.len:
let chunkEnd = chunkStart + chunkLen - 1 let chunkEnd = chunkStart + chunkLen - 1
let element = F.fromOpenArray(bytes.toOpenArray(chunkStart, chunkEnd)) let element = F.fromOpenArray(bytes.toOpenArray(chunkStart, chunkEnd))
yield element yield element
chunkStart += chunkLen chunkStart += chunkLen
if bytes.len - chunkStart > 0: if (bytes.len - chunkStart > 0) or empty:
let finalChunk = bytes[chunkStart..<bytes.len] & endMarker let finalChunk = bytes[chunkStart..<bytes.len] & endMarker
let finalElement = F.fromOpenArray(finalChunk) let finalElement = F.fromOpenArray(finalChunk)
yield finalElement yield finalElement