1 // Copyright (c) 2018, ETH Zurich and UNC Chapel Hill.
2 // All rights reserved.
3 //
4 // Redistribution and use in source and binary forms, with or without
5 // modification, are permitted provided that the following conditions are met:
6 //
7 // * Redistributions of source code must retain the above copyright
8 // notice, this list of conditions and the following disclaimer.
9 //
10 // * Redistributions in binary form must reproduce the above copyright
11 // notice, this list of conditions and the following disclaimer in the
12 // documentation and/or other materials provided with the distribution.
13 //
14 // * Neither the name of ETH Zurich and UNC Chapel Hill nor the names of
15 // its contributors may be used to endorse or promote products derived
16 // from this software without specific prior written permission.
17 //
18 // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
19 // AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
20 // IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
21 // ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDERS OR CONTRIBUTORS BE
22 // LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
23 // CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
24 // SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
25 // INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
26 // CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
27 // ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
28 // POSSIBILITY OF SUCH DAMAGE.
29 //
30 // Author: Johannes L. Schoenberger (jsch-at-demuc-dot-de)
31
32 #define TEST_NAME "mvs/consistency_graph_test"
33 #include "util/testing.h"
34
35 #include "mvs/consistency_graph.h"
36
37 using namespace colmap;
38 using namespace colmap::mvs;
39
BOOST_AUTO_TEST_CASE(TestEmpty)40 BOOST_AUTO_TEST_CASE(TestEmpty) {
41 const std::vector<int> data;
42 ConsistencyGraph consistency_graph(2, 2, data);
43 for (size_t i = 0; i < 2; ++i) {
44 for (size_t j = 0; j < 2; ++j) {
45 int num_images;
46 const int* image_idxs;
47 consistency_graph.GetImageIdxs(0, 0, &num_images, &image_idxs);
48 BOOST_CHECK_EQUAL(num_images, 0);
49 BOOST_CHECK(image_idxs == nullptr);
50 }
51 }
52 BOOST_CHECK_EQUAL(consistency_graph.GetNumBytes(), 16);
53 }
54
BOOST_AUTO_TEST_CASE(TestPartial)55 BOOST_AUTO_TEST_CASE(TestPartial) {
56 const std::vector<int> data = {0, 0, 3, 5, 7, 33};
57 ConsistencyGraph consistency_graph(2, 1, data);
58 int num_images;
59 const int* image_idxs;
60 consistency_graph.GetImageIdxs(0, 0, &num_images, &image_idxs);
61 BOOST_CHECK_EQUAL(num_images, 3);
62 BOOST_CHECK_EQUAL(image_idxs[0], 5);
63 BOOST_CHECK_EQUAL(image_idxs[1], 7);
64 BOOST_CHECK_EQUAL(image_idxs[2], 33);
65 consistency_graph.GetImageIdxs(0, 1, &num_images, &image_idxs);
66 BOOST_CHECK_EQUAL(num_images, 0);
67 BOOST_CHECK(image_idxs == nullptr);
68 BOOST_CHECK_EQUAL(consistency_graph.GetNumBytes(), 32);
69 }
70
BOOST_AUTO_TEST_CASE(TestZero)71 BOOST_AUTO_TEST_CASE(TestZero) {
72 const std::vector<int> data = {0, 0, 0};
73 ConsistencyGraph consistency_graph(2, 1, data);
74 int num_images;
75 const int* image_idxs;
76 consistency_graph.GetImageIdxs(0, 0, &num_images, &image_idxs);
77 BOOST_CHECK_EQUAL(num_images, 0);
78 BOOST_CHECK(image_idxs == nullptr);
79 consistency_graph.GetImageIdxs(0, 1, &num_images, &image_idxs);
80 BOOST_CHECK_EQUAL(num_images, 0);
81 BOOST_CHECK(image_idxs == nullptr);
82 BOOST_CHECK_EQUAL(consistency_graph.GetNumBytes(), 20);
83 }
84
BOOST_AUTO_TEST_CASE(TestFull)85 BOOST_AUTO_TEST_CASE(TestFull) {
86 const std::vector<int> data = {0, 0, 3, 5, 7, 33, 0, 1, 1, 100};
87 ConsistencyGraph consistency_graph(1, 2, data);
88 int num_images;
89 const int* image_idxs;
90 consistency_graph.GetImageIdxs(0, 0, &num_images, &image_idxs);
91 BOOST_CHECK_EQUAL(num_images, 3);
92 BOOST_CHECK_EQUAL(image_idxs[0], 5);
93 BOOST_CHECK_EQUAL(image_idxs[1], 7);
94 BOOST_CHECK_EQUAL(image_idxs[2], 33);
95 consistency_graph.GetImageIdxs(1, 0, &num_images, &image_idxs);
96 BOOST_CHECK_EQUAL(num_images, 1);
97 BOOST_CHECK_EQUAL(image_idxs[0], 100);
98 BOOST_CHECK_EQUAL(consistency_graph.GetNumBytes(), 48);
99 }
100