blob: 0cf4607acdc642c9a73dae73be6d187702301b2f (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
|
#include <wanda/filesystem.hpp>
// #include <boost/iterator/filter_iterator.hpp>
#include <range/v3/all.hpp>
#include <random>
namespace wanda
{
std::optional<path_list> scan(std::filesystem::path source, bool(filter)(std::filesystem::path const &))
{
if (!std::filesystem::is_directory(source))
{
return std::nullopt;
}
auto begin = std::filesystem::recursive_directory_iterator{source};
auto end = std::filesystem::recursive_directory_iterator{};
return ranges::make_iterator_range(begin, end) | ranges::view::filter(filter);
}
std::filesystem::path random_pick(path_list const & paths)
{
static auto generator = std::mt19937{std::random_device{}()};
auto distribution = std::uniform_int_distribution<std::size_t>{0, paths.size() - 1};
return paths[distribution(generator)];
}
} // namespace wanda
|
