| Line | Branch | Exec | Source | 
|---|---|---|---|
| 1 | // Copyright Contributors to the OpenVDB Project | ||
| 2 | // SPDX-License-Identifier: MPL-2.0 | ||
| 3 | |||
| 4 | #include <openvdb/tools/ParticleAtlas.h> | ||
| 5 | #include <openvdb/math/Math.h> | ||
| 6 | |||
| 7 | #include <gtest/gtest.h> | ||
| 8 | |||
| 9 | #include <vector> | ||
| 10 | #include <algorithm> | ||
| 11 | #include <cmath> | ||
| 12 | #include "util.h" // for genPoints | ||
| 13 | |||
| 14 | |||
| 15 | 1 | struct TestParticleAtlas: public ::testing::Test | |
| 16 | { | ||
| 17 | }; | ||
| 18 | |||
| 19 | |||
| 20 | //////////////////////////////////////// | ||
| 21 | |||
| 22 | namespace { | ||
| 23 | |||
| 24 | class ParticleList | ||
| 25 | { | ||
| 26 | public: | ||
| 27 | typedef openvdb::Vec3R PosType; | ||
| 28 | typedef PosType::value_type ScalarType; | ||
| 29 | |||
| 30 | ParticleList(const std::vector<PosType>& points, | ||
| 31 | const std::vector<ScalarType>& radius) | ||
| 32 | 1 | : mPoints(&points) | |
| 33 | 1/2✓ Branch 1 taken 1 times. ✗ Branch 2 not taken. | 1 | , mRadius(&radius) | 
| 34 | { | ||
| 35 | } | ||
| 36 | |||
| 37 | // Return the number of points in the array | ||
| 38 | size_t size() const { | ||
| 39 | 1/10✗ Branch 0 not taken. ✗ Branch 1 not taken. ✗ Branch 2 not taken. ✗ Branch 3 not taken. ✗ Branch 4 not taken. ✗ Branch 5 not taken. ✗ Branch 6 not taken. ✗ Branch 7 not taken. ✓ Branch 10 taken 1 times. ✗ Branch 11 not taken. | 2 | return mPoints->size(); | 
| 40 | } | ||
| 41 | |||
| 42 | // Return the world-space position for the nth particle. | ||
| 43 | void getPos(size_t n, PosType& xyz) const { | ||
| 44 | 5/12✓ Branch 0 taken 40000 times. ✗ Branch 1 not taken. ✗ Branch 2 not taken. ✗ Branch 3 not taken. ✓ Branch 4 taken 1514 times. ✓ Branch 5 taken 1433 times. ✗ Branch 6 not taken. ✗ Branch 7 not taken. ✓ Branch 8 taken 701 times. ✓ Branch 9 taken 339 times. ✗ Branch 10 not taken. ✗ Branch 11 not taken. | 43987 | xyz = (*mPoints)[n]; | 
| 45 | } | ||
| 46 | |||
| 47 | // Return the world-space radius for the nth particle. | ||
| 48 | void getRadius(size_t n, ScalarType& radius) const { | ||
| 49 | 2/6✗ Branch 0 not taken. ✓ Branch 1 taken 40000 times. ✗ Branch 2 not taken. ✓ Branch 3 taken 40000 times. ✗ Branch 5 not taken. ✗ Branch 6 not taken. | 120004 | radius = (*mRadius)[n]; | 
| 50 | } | ||
| 51 | |||
| 52 | protected: | ||
| 53 | std::vector<PosType> const * const mPoints; | ||
| 54 | std::vector<ScalarType> const * const mRadius; | ||
| 55 | }; // ParticleList | ||
| 56 | |||
| 57 | |||
| 58 | template<typename T> | ||
| 59 | 3 | bool hasDuplicates(const std::vector<T>& items) | |
| 60 | { | ||
| 61 | 3 | std::vector<T> vec(items); | |
| 62 | 3 | std::sort(vec.begin(), vec.end()); | |
| 63 | |||
| 64 | size_t duplicates = 0; | ||
| 65 | 2/2✓ Branch 0 taken 61964 times. ✓ Branch 1 taken 3 times. | 61967 | for (size_t n = 1, N = vec.size(); n < N; ++n) { | 
| 66 | 1/2✗ Branch 0 not taken. ✓ Branch 1 taken 61964 times. | 61964 | if (vec[n] == vec[n-1]) ++duplicates; | 
| 67 | } | ||
| 68 | 1/2✓ Branch 0 taken 3 times. ✗ Branch 1 not taken. | 3 | return duplicates != 0; | 
| 69 | } | ||
| 70 | |||
| 71 | } // namespace | ||
| 72 | |||
| 73 | |||
| 74 | |||
| 75 | //////////////////////////////////////// | ||
| 76 | |||
| 77 | |||
| 78 | 1/2✓ Branch 1 taken 1 times. ✗ Branch 2 not taken. | 1 | TEST_F(TestParticleAtlas, testParticleAtlas) | 
| 79 | { | ||
| 80 | // generate points | ||
| 81 | |||
| 82 | 1 | const size_t numParticle = 40000; | |
| 83 | 1/2✓ Branch 1 taken 1 times. ✗ Branch 2 not taken. | 1 | const double minVoxelSize = 0.01; | 
| 84 | |||
| 85 | std::vector<openvdb::Vec3R> points; | ||
| 86 | 1/2✓ Branch 1 taken 1 times. ✗ Branch 2 not taken. | 1 | unittest_util::genPoints(numParticle, points); | 
| 87 | |||
| 88 | std::vector<double> radius; | ||
| 89 | 2/2✓ Branch 0 taken 20000 times. ✓ Branch 1 taken 1 times. | 20001 | for (size_t n = 0, N = points.size() / 2; n < N; ++n) { | 
| 90 | 1/2✓ Branch 1 taken 20000 times. ✗ Branch 2 not taken. | 20000 | radius.push_back(minVoxelSize); | 
| 91 | } | ||
| 92 | |||
| 93 | 2/2✓ Branch 0 taken 20000 times. ✓ Branch 1 taken 1 times. | 20001 | for (size_t n = points.size() / 2, N = points.size(); n < N; ++n) { | 
| 94 | 1/4✓ Branch 1 taken 20000 times. ✗ Branch 2 not taken. ✗ Branch 3 not taken. ✗ Branch 4 not taken. | 20000 | radius.push_back(minVoxelSize * 2.0); | 
| 95 | } | ||
| 96 | |||
| 97 | ParticleList particles(points, radius); | ||
| 98 | |||
| 99 | // construct data structure | ||
| 100 | |||
| 101 | typedef openvdb::tools::ParticleAtlas<> ParticleAtlas; | ||
| 102 | |||
| 103 | 1 | ParticleAtlas atlas; | |
| 104 | |||
| 105 | EXPECT_TRUE(atlas.empty()); | ||
| 106 | EXPECT_TRUE(atlas.levels() == 0); | ||
| 107 | |||
| 108 | 1/2✓ Branch 1 taken 1 times. ✗ Branch 2 not taken. | 1 | atlas.construct(particles, minVoxelSize); | 
| 109 | |||
| 110 | 1/16✗ Branch 0 not taken. ✓ Branch 1 taken 1 times. ✗ Branch 3 not taken. ✗ Branch 4 not taken. ✗ Branch 6 not taken. ✗ Branch 7 not taken. ✗ Branch 9 not taken. ✗ Branch 10 not taken. ✗ Branch 12 not taken. ✗ Branch 13 not taken. ✗ Branch 16 not taken. ✗ Branch 17 not taken. ✗ Branch 20 not taken. ✗ Branch 21 not taken. ✗ Branch 22 not taken. ✗ Branch 23 not taken. | 1 | EXPECT_TRUE(!atlas.empty()); | 
| 111 | 1/16✗ Branch 0 not taken. ✓ Branch 1 taken 1 times. ✗ Branch 3 not taken. ✗ Branch 4 not taken. ✗ Branch 6 not taken. ✗ Branch 7 not taken. ✗ Branch 9 not taken. ✗ Branch 10 not taken. ✗ Branch 12 not taken. ✗ Branch 13 not taken. ✗ Branch 16 not taken. ✗ Branch 17 not taken. ✗ Branch 20 not taken. ✗ Branch 21 not taken. ✗ Branch 22 not taken. ✗ Branch 23 not taken. | 1 | EXPECT_TRUE(atlas.levels() == 2); | 
| 112 | |||
| 113 | 1/16✗ Branch 0 not taken. ✓ Branch 1 taken 1 times. ✗ Branch 3 not taken. ✗ Branch 4 not taken. ✗ Branch 6 not taken. ✗ Branch 7 not taken. ✗ Branch 9 not taken. ✗ Branch 10 not taken. ✗ Branch 12 not taken. ✗ Branch 13 not taken. ✗ Branch 16 not taken. ✗ Branch 17 not taken. ✗ Branch 20 not taken. ✗ Branch 21 not taken. ✗ Branch 22 not taken. ✗ Branch 23 not taken. | 1 | EXPECT_TRUE( | 
| 114 | openvdb::math::isApproxEqual(atlas.minRadius(0), minVoxelSize)); | ||
| 115 | |||
| 116 | 1/16✗ Branch 0 not taken. ✓ Branch 1 taken 1 times. ✗ Branch 3 not taken. ✗ Branch 4 not taken. ✗ Branch 6 not taken. ✗ Branch 7 not taken. ✗ Branch 9 not taken. ✗ Branch 10 not taken. ✗ Branch 12 not taken. ✗ Branch 13 not taken. ✗ Branch 16 not taken. ✗ Branch 17 not taken. ✗ Branch 20 not taken. ✗ Branch 21 not taken. ✗ Branch 22 not taken. ✗ Branch 23 not taken. | 1 | EXPECT_TRUE( | 
| 117 | openvdb::math::isApproxEqual(atlas.minRadius(1), minVoxelSize * 2.0)); | ||
| 118 | |||
| 119 | typedef openvdb::tools::ParticleAtlas<>::Iterator ParticleAtlasIterator; | ||
| 120 | |||
| 121 | 1/2✓ Branch 1 taken 1 times. ✗ Branch 2 not taken. | 2 | ParticleAtlasIterator it(atlas); | 
| 122 | |||
| 123 | 1/16✗ Branch 0 not taken. ✓ Branch 1 taken 1 times. ✗ Branch 3 not taken. ✗ Branch 4 not taken. ✗ Branch 6 not taken. ✗ Branch 7 not taken. ✗ Branch 9 not taken. ✗ Branch 10 not taken. ✗ Branch 12 not taken. ✗ Branch 13 not taken. ✗ Branch 16 not taken. ✗ Branch 17 not taken. ✗ Branch 20 not taken. ✗ Branch 21 not taken. ✗ Branch 22 not taken. ✗ Branch 23 not taken. | 1 | EXPECT_TRUE(atlas.levels() == 2); | 
| 124 | |||
| 125 | std::vector<uint32_t> indices; | ||
| 126 | 1/2✓ Branch 1 taken 1 times. ✗ Branch 2 not taken. | 1 | indices.reserve(numParticle); | 
| 127 | |||
| 128 | 1/2✓ Branch 1 taken 1 times. ✗ Branch 2 not taken. | 1 | it.updateFromLevel(0); | 
| 129 | |||
| 130 | 1/16✗ Branch 0 not taken. ✓ Branch 1 taken 1 times. ✗ Branch 3 not taken. ✗ Branch 4 not taken. ✗ Branch 6 not taken. ✗ Branch 7 not taken. ✗ Branch 9 not taken. ✗ Branch 10 not taken. ✗ Branch 12 not taken. ✗ Branch 13 not taken. ✗ Branch 16 not taken. ✗ Branch 17 not taken. ✗ Branch 20 not taken. ✗ Branch 21 not taken. ✗ Branch 22 not taken. ✗ Branch 23 not taken. | 1 | EXPECT_TRUE(it); | 
| 131 | 2/16✓ Branch 2 taken 1 times. ✗ Branch 3 not taken. ✗ Branch 4 not taken. ✓ Branch 5 taken 1 times. ✗ Branch 7 not taken. ✗ Branch 8 not taken. ✗ Branch 10 not taken. ✗ Branch 11 not taken. ✗ Branch 13 not taken. ✗ Branch 14 not taken. ✗ Branch 16 not taken. ✗ Branch 17 not taken. ✗ Branch 19 not taken. ✗ Branch 20 not taken. ✗ Branch 21 not taken. ✗ Branch 22 not taken. | 1 | EXPECT_EQ(it.size(), numParticle - (points.size() / 2)); | 
| 132 | |||
| 133 | |||
| 134 | for (; it; ++it) { | ||
| 135 | 1/2✓ Branch 1 taken 20000 times. ✗ Branch 2 not taken. | 20000 | indices.push_back(*it); | 
| 136 | } | ||
| 137 | |||
| 138 | 1/2✓ Branch 1 taken 1 times. ✗ Branch 2 not taken. | 1 | it.updateFromLevel(1); | 
| 139 | |||
| 140 | 1/16✗ Branch 0 not taken. ✓ Branch 1 taken 1 times. ✗ Branch 3 not taken. ✗ Branch 4 not taken. ✗ Branch 6 not taken. ✗ Branch 7 not taken. ✗ Branch 9 not taken. ✗ Branch 10 not taken. ✗ Branch 12 not taken. ✗ Branch 13 not taken. ✗ Branch 16 not taken. ✗ Branch 17 not taken. ✗ Branch 20 not taken. ✗ Branch 21 not taken. ✗ Branch 22 not taken. ✗ Branch 23 not taken. | 1 | EXPECT_TRUE(it); | 
| 141 | 2/16✓ Branch 2 taken 1 times. ✗ Branch 3 not taken. ✗ Branch 4 not taken. ✓ Branch 5 taken 1 times. ✗ Branch 7 not taken. ✗ Branch 8 not taken. ✗ Branch 10 not taken. ✗ Branch 11 not taken. ✗ Branch 13 not taken. ✗ Branch 14 not taken. ✗ Branch 16 not taken. ✗ Branch 17 not taken. ✗ Branch 19 not taken. ✗ Branch 20 not taken. ✗ Branch 21 not taken. ✗ Branch 22 not taken. | 1 | EXPECT_EQ(it.size(), (points.size() / 2)); | 
| 142 | |||
| 143 | |||
| 144 | for (; it; ++it) { | ||
| 145 | 1/2✓ Branch 1 taken 20000 times. ✗ Branch 2 not taken. | 20000 | indices.push_back(*it); | 
| 146 | } | ||
| 147 | |||
| 148 | 2/16✓ Branch 1 taken 1 times. ✗ Branch 2 not taken. ✗ Branch 3 not taken. ✓ Branch 4 taken 1 times. ✗ Branch 6 not taken. ✗ Branch 7 not taken. ✗ Branch 9 not taken. ✗ Branch 10 not taken. ✗ Branch 12 not taken. ✗ Branch 13 not taken. ✗ Branch 15 not taken. ✗ Branch 16 not taken. ✗ Branch 18 not taken. ✗ Branch 19 not taken. ✗ Branch 20 not taken. ✗ Branch 21 not taken. | 1 | EXPECT_EQ(numParticle, indices.size()); | 
| 149 | |||
| 150 | 2/18✓ Branch 1 taken 1 times. ✗ Branch 2 not taken. ✗ Branch 3 not taken. ✓ Branch 4 taken 1 times. ✗ Branch 6 not taken. ✗ Branch 7 not taken. ✗ Branch 9 not taken. ✗ Branch 10 not taken. ✗ Branch 12 not taken. ✗ Branch 13 not taken. ✗ Branch 15 not taken. ✗ Branch 16 not taken. ✗ Branch 19 not taken. ✗ Branch 20 not taken. ✗ Branch 23 not taken. ✗ Branch 24 not taken. ✗ Branch 25 not taken. ✗ Branch 26 not taken. | 1 | EXPECT_TRUE(!hasDuplicates(indices)); | 
| 151 | |||
| 152 | |||
| 153 | 1 | openvdb::Vec3R center = points[0]; | |
| 154 | double searchRadius = minVoxelSize * 10.0; | ||
| 155 | |||
| 156 | 1/2✓ Branch 1 taken 1 times. ✗ Branch 2 not taken. | 1 | it.worldSpaceSearchAndUpdate(center, searchRadius, particles); | 
| 157 | 1/16✗ Branch 0 not taken. ✓ Branch 1 taken 1 times. ✗ Branch 3 not taken. ✗ Branch 4 not taken. ✗ Branch 6 not taken. ✗ Branch 7 not taken. ✗ Branch 9 not taken. ✗ Branch 10 not taken. ✗ Branch 12 not taken. ✗ Branch 13 not taken. ✗ Branch 16 not taken. ✗ Branch 17 not taken. ✗ Branch 20 not taken. ✗ Branch 21 not taken. ✗ Branch 22 not taken. ✗ Branch 23 not taken. | 1 | EXPECT_TRUE(it); | 
| 158 | |||
| 159 | indices.clear(); | ||
| 160 | for (; it; ++it) { | ||
| 161 | 1/2✓ Branch 1 taken 1408 times. ✗ Branch 2 not taken. | 1408 | indices.push_back(*it); | 
| 162 | } | ||
| 163 | |||
| 164 | 2/16✓ Branch 2 taken 1 times. ✗ Branch 3 not taken. ✗ Branch 4 not taken. ✓ Branch 5 taken 1 times. ✗ Branch 7 not taken. ✗ Branch 8 not taken. ✗ Branch 10 not taken. ✗ Branch 11 not taken. ✗ Branch 13 not taken. ✗ Branch 14 not taken. ✗ Branch 16 not taken. ✗ Branch 17 not taken. ✗ Branch 19 not taken. ✗ Branch 20 not taken. ✗ Branch 21 not taken. ✗ Branch 22 not taken. | 1 | EXPECT_EQ(it.size(), indices.size()); | 
| 165 | 2/18✓ Branch 1 taken 1 times. ✗ Branch 2 not taken. ✗ Branch 3 not taken. ✓ Branch 4 taken 1 times. ✗ Branch 6 not taken. ✗ Branch 7 not taken. ✗ Branch 9 not taken. ✗ Branch 10 not taken. ✗ Branch 12 not taken. ✗ Branch 13 not taken. ✗ Branch 15 not taken. ✗ Branch 16 not taken. ✗ Branch 19 not taken. ✗ Branch 20 not taken. ✗ Branch 23 not taken. ✗ Branch 24 not taken. ✗ Branch 25 not taken. ✗ Branch 26 not taken. | 1 | EXPECT_TRUE(!hasDuplicates(indices)); | 
| 166 | |||
| 167 | |||
| 168 | openvdb::BBoxd bbox; | ||
| 169 | 2/2✓ Branch 0 taken 20000 times. ✓ Branch 1 taken 1 times. | 20001 | for (size_t n = 0, N = points.size() / 2; n < N; ++n) { | 
| 170 | 20000 | bbox.expand(points[n]); | |
| 171 | } | ||
| 172 | |||
| 173 | 1/2✓ Branch 1 taken 1 times. ✗ Branch 2 not taken. | 1 | it.worldSpaceSearchAndUpdate(bbox, particles); | 
| 174 | 1/16✗ Branch 0 not taken. ✓ Branch 1 taken 1 times. ✗ Branch 3 not taken. ✗ Branch 4 not taken. ✗ Branch 6 not taken. ✗ Branch 7 not taken. ✗ Branch 9 not taken. ✗ Branch 10 not taken. ✗ Branch 12 not taken. ✗ Branch 13 not taken. ✗ Branch 16 not taken. ✗ Branch 17 not taken. ✗ Branch 20 not taken. ✗ Branch 21 not taken. ✗ Branch 22 not taken. ✗ Branch 23 not taken. | 1 | EXPECT_TRUE(it); | 
| 175 | |||
| 176 | indices.clear(); | ||
| 177 | for (; it; ++it) { | ||
| 178 | 1/2✓ Branch 1 taken 20559 times. ✗ Branch 2 not taken. | 20559 | indices.push_back(*it); | 
| 179 | } | ||
| 180 | |||
| 181 | 2/18✓ Branch 2 taken 1 times. ✗ Branch 3 not taken. ✗ Branch 4 not taken. ✓ Branch 5 taken 1 times. ✗ Branch 7 not taken. ✗ Branch 8 not taken. ✗ Branch 10 not taken. ✗ Branch 11 not taken. ✗ Branch 13 not taken. ✗ Branch 14 not taken. ✗ Branch 16 not taken. ✗ Branch 17 not taken. ✗ Branch 19 not taken. ✗ Branch 20 not taken. ✗ Branch 21 not taken. ✗ Branch 22 not taken. ✗ Branch 23 not taken. ✗ Branch 24 not taken. | 1 | EXPECT_EQ(it.size(), indices.size()); | 
| 182 | 2/18✓ Branch 1 taken 1 times. ✗ Branch 2 not taken. ✗ Branch 3 not taken. ✓ Branch 4 taken 1 times. ✗ Branch 6 not taken. ✗ Branch 7 not taken. ✗ Branch 9 not taken. ✗ Branch 10 not taken. ✗ Branch 12 not taken. ✗ Branch 13 not taken. ✗ Branch 15 not taken. ✗ Branch 16 not taken. ✗ Branch 19 not taken. ✗ Branch 20 not taken. ✗ Branch 23 not taken. ✗ Branch 24 not taken. ✗ Branch 25 not taken. ✗ Branch 26 not taken. | 1 | EXPECT_TRUE(!hasDuplicates(indices)); | 
| 183 | 1 | } | |
| 184 | |||
| 185 |