logos-storage/logos-storage-nim
1
0
Fork 0
mirror of https://github.com/logos-storage/logos-storage-nim.git synced 2026-02-04 13:43:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

Implements cell index collection

Browse Source
This commit is contained in:
benbierens 2023-11-22 10:19:51 +01:00 committed by Dmitriy Ryajov
parent c320dd2e7d
commit d395328ed4
No known key found for this signature in database
GPG Key ID: DA8C680CE7C657A4
2 changed files with 41 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
codex/proof/datasampler.nim
View File

@ -1,6 +1,7 @@
import ../contracts/requests
import std/bitops
import std/sugar
import pkg/constantine/math/arithmetic
import pkg/poseidon2/types
@ -52,11 +53,11 @@ proc findCellIndex*(
return index
# func findCellIndices*(
# slot: Slot,
# slotRootHash: DSFieldElement,
# challenge: DSFieldElement,
# nSamples: int): seq[int] =
# # Computes nSamples cell indices.
# let numberOfCells = getNumberOfCellsInSlot(slot)
# return collect(newSeq, (for i in 1..nSamples: findCellIndex(slotRootHash, challenge, toF(i), numberOfCells)))
func findCellIndices*(
slot: Slot,
slotRootHash: DSFieldElement,
challenge: DSFieldElement,
nSamples: int): seq[DSCellIndex] =
# Computes nSamples cell indices.
let numberOfCells = getNumberOfCellsInSlot(slot)
return collect(newSeq, (for i in 1..nSamples: findCellIndex(slotRootHash, challenge, toF(i), numberOfCells)))

38
tests/codex/proof/testdatasampler.nim
View File

@ -1,6 +1,7 @@
import std/os
import std/strutils
import std/sequtils
import std/sugar
import pkg/questionable
import pkg/questionable/results
@ -73,7 +74,8 @@ asyncchecksuite "Test proof datasampler":
setup:
await createSlotBlocks()
test "number of cells is a power of two":
test "Number of cells is a power of two":
# This is to check that the data used for testing is sane.
proc isPow2(value: int): bool =
let log2 = ceilingLog2(value)
return (1 shl log2) == value
@ -105,11 +107,35 @@ asyncchecksuite "Test proof datasampler":
expectedNumberOfCells == 512
expectedNumberOfCells == getNumberOfCellsInSlot(slot)
let knownIndices = @[178.uint64, 277.uint64, 366.uint64]
test "Can find single cell index":
let
counter: DSFieldElement = toF(1)
numberOfCells = getNumberOfCellsInSlot(slot)
cellIndex = findCellIndex(slotRootHash, challenge, counter, numberOfCells)
let numberOfCells = getNumberOfCellsInSlot(slot)
proc cellIndex(i: int): DSCellIndex =
let counter: DSFieldElement = toF(i)
return findCellIndex(slotRootHash, challenge, counter, numberOfCells)
proc getExpectedIndex(i: int): DSCellIndex =
let hash = Sponge.digest(@[slotRootHash, challenge, toF(i)], rate = 2)
return extractLowBits(hash.toBig(), ceilingLog2(numberOfCells))
check:
cellIndex == 2
cellIndex(1) == getExpectedIndex(1)
cellIndex(1) == knownIndices[0]
cellIndex(2) == getExpectedIndex(2)
cellIndex(2) == knownIndices[1]
cellIndex(3) == getExpectedIndex(3)
cellIndex(3) == knownIndices[2]
test "Can find sequence of cell indices":
proc cellIndices(n: int): seq[DSCellIndex] =
findCellIndices(slot, slotRootHash, challenge, n)
let numberOfCells = getNumberOfCellsInSlot(slot)
proc getExpectedIndices(n: int): seq[DSCellIndex] =
return collect(newSeq, (for i in 1..n: findCellIndex(slotRootHash, challenge, toF(i), numberOfCells)))
check:
cellIndices(3) == getExpectedIndices(3)
cellIndices(3) == knownIndices