mirror of https://github.com/CGAL/cgal
73 lines
1.9 KiB
C++
73 lines
1.9 KiB
C++
|
|
#define CGAL_TRACE_STREAM std::cerr
|
|
|
|
#include "util.h"
|
|
|
|
#include <CGAL/Exact_predicates_exact_constructions_kernel.h>
|
|
|
|
#include <CGAL/Octree.h>
|
|
#include <CGAL/Search_traits_3.h>
|
|
#include <CGAL/Orthogonal_k_neighbor_search.h>
|
|
|
|
#include <iostream>
|
|
#include <chrono>
|
|
|
|
using Kernel = CGAL::Exact_predicates_inexact_constructions_kernel;
|
|
using Point = Kernel::Point_3;
|
|
using Point_set = CGAL::Point_set_3<Point>;
|
|
using Point_map = Point_set::Point_map;
|
|
using Octree = CGAL::Octree<Kernel, Point_set, Point_map>;
|
|
using Kd_tree_traits = CGAL::Search_traits_3<Kernel>;
|
|
using Kd_tree_search = CGAL::Orthogonal_k_neighbor_search<Kd_tree_traits>;
|
|
using Kdtree = Kd_tree_search::Tree;
|
|
|
|
using std::chrono::high_resolution_clock;
|
|
using std::chrono::duration_cast;
|
|
using std::chrono::microseconds;
|
|
using std::chrono::milliseconds;
|
|
|
|
int main(int argc, char **argv) {
|
|
|
|
// Set output file
|
|
std::ofstream file;
|
|
file.open((argc > 1) ? argv[1] : "../construction_benchmark.csv");
|
|
|
|
// Add header for CSV
|
|
file << "Number of Points,Octree,kDTree \n";
|
|
|
|
// Perform tests for various dataset sizes
|
|
for (size_t num_points = 10; num_points < 10000000; num_points *= 1.1) {
|
|
|
|
// Create a collection of the right number of points
|
|
auto points = generate<Kernel>(num_points);
|
|
|
|
auto octreePoints = points;
|
|
auto octreeTime = bench<milliseconds>(
|
|
[&] {
|
|
// Build the tree
|
|
Octree octree(octreePoints, octreePoints.point_map());
|
|
octree.refine();
|
|
}
|
|
);
|
|
|
|
auto kdtreePoints = points;
|
|
auto kdtreeTime = bench<milliseconds>(
|
|
[&] {
|
|
// Build the tree
|
|
Kdtree kdtree(kdtreePoints.points().begin(), kdtreePoints.points().end());
|
|
kdtree.build();
|
|
}
|
|
);
|
|
|
|
file << num_points << ",";
|
|
file << octreeTime.count() << ",";
|
|
file << kdtreeTime.count() << "\n";
|
|
|
|
std::cout << num_points << std::endl;
|
|
}
|
|
|
|
file.close();
|
|
|
|
return 0;
|
|
}
|