blender
/
blender
136
409
Fork 592
Code Issues 6.3k Pull Requests 775 Projects 19 Wiki Activity

BLI: Improve IndexMask::complement() performance #108331

Merged
Hans Goudey merged 15 commits from HooglyBoogly/blender:index-mask-complement-performance into main 2023-05-31 17:11:11 +02:00
Conversation 1 Commits 15 Files Changed 2 +13 -7
1 changed files with 13 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit 38a061824b - Show all commits

20
source/blender/blenlib/tests/BLI_index_mask_test.cc
View File

@ -248,7 +248,7 @@ TEST(index_mask, Complement)
}
}
TEST(index_mask, ComplementLarge)
TEST(index_mask, ComplementPerformance)
{
RandomNumberGenerator rng;
@ -256,18 +256,24 @@ TEST(index_mask, ComplementLarge)
const int64_t iter_num = 100;
const int64_t universe_size = 110000;
for (const int64_t iter : IndexRange(100)) {
for (const int64_t iter : IndexRange(iter_num)) {
const float probability = float(iter) / float(iter_num);
Array<bool> bools(mask_size);
for (const int64_t i : bools.index_range()) {
bools[i] = rng.get_float() > probability;
}
Set<int> values;
for ([[maybe_unused]] const int64_t _ : IndexRange(iter * mask_size / iter_num)) {
values.add(rng.get_int32(mask_size));
}
IndexMaskMemory memory;
const IndexMask mask = IndexMask::from_predicate(
IndexRange(mask_size), GrainSize(1024), memory, [&](const int64_t i) {
return values.contains(int(i));
});
const IndexMask mask = IndexMask::from_bools(bools, memory);
const IndexMask complement = mask.complement(IndexRange(universe_size), memory);
IndexMask complement;
{
SCOPED_TIMER("Probability: " + std::to_string(probability));
complement = mask.complement(IndexRange(universe_size), memory);
}
complement.foreach_index([&](const int64_t i) { EXPECT_FALSE(mask.contains(i)); });
mask.foreach_index([&](const int64_t i) { EXPECT_FALSE(complement.contains(i)); });
EXPECT_EQ(universe_size - mask.size(), complement.size());