thunderbrewhq/squall
1
0
Fork 0
mirror of https://github.com/thunderbrewhq/squall.git synced 2025-12-12 02:22:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat(big): add SetOne

Browse source
This commit is contained in:
fallenoak 2023-01-31 22:07:33 -06:00
parent 8413214ae0
commit b2cbc02189
No known key found for this signature in database
GPG key ID: 7628F8E61AEA070D
3 changed files with 25 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

5
storm/big/Ops.cpp
View file

@ -82,6 +82,11 @@ void Mul(BigBuffer& a, const BigBuffer& b, const BigBuffer& c, BigStack& stack)
stack.UnmakeDistinct(a, aa);
}
void SetOne(BigBuffer& buffer) {
buffer.SetCount(1);
buffer[0] = 1;
}
void SetZero(BigBuffer& buffer) {
buffer.Clear();
}

2
storm/big/Ops.hpp
View file

@ -21,6 +21,8 @@ void Mul(BigBuffer& a, const BigBuffer& b, uint64_t c);
void Mul(BigBuffer& a, const BigBuffer& b, const BigBuffer& c, BigStack& stack);
void SetOne(BigBuffer& buffer);
void SetZero(BigBuffer& buffer);
void ToBinary(TSGrowableArray<uint8_t>& output, const BigBuffer& buffer);

18
test/Big.cpp
View file

@ -421,6 +421,24 @@ TEST_CASE("SBigToBinaryBuffer", "[big]") {
}
}
TEST_CASE("SetOne", "[big]") {
SECTION("sets buffer to one") {
BigData* num;
SBigNew(&num);
uint64_t data = 0x123456789ABCDEF0;
SBigFromBinary(num, reinterpret_cast<uint8_t*>(&data), sizeof(data));
CHECK(num->Primary().Count() == 2);
SetOne(num->Primary());
CHECK(num->Primary().Count() == 1);
CHECK(num->Primary()[0] == 1);
SBigDel(num);
}
}
TEST_CASE("SetZero", "[big]") {
SECTION("sets buffer to zero") {
BigData* num;