blob: bb724d10b885b681ffe30f5c26e7badb4491e2a1 (
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
|
#include "filesystem.hpp"
#include <boost/iterator/filter_iterator.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 first = boost::make_filter_iterator(filter, std::filesystem::recursive_directory_iterator{source});
auto last = boost::make_filter_iterator(filter, std::filesystem::recursive_directory_iterator{});
return path_list{first, last};
}
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
|
