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
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
|
#include <fstream>
#include <numeric>
#include <range/v3/view/split.hpp>
#include <seqan3/argument_parser/all.hpp> // includes all necessary headers
#include <seqan3/core/debug_stream.hpp> // our custom output stream
#include <seqan3/std/charconv> // includes std::from_chars
#include <seqan3/std/filesystem> // use std::filesystem::path
// This is the program!
// Take a look at it if you are interested in an example of parsing a data file.
// -----------------------------------------------------------------------------
template <typename number_type, typename range_type>
number_type to_number(range_type && range)
{
std::string str;
number_type num;
std::ranges::copy(range, std::cpp20::back_inserter(str));
auto res = std::from_chars(&str[0], &str[0] + str.size(), num);
if (res.ec != std::errc{})
{
seqan3::debug_stream << "Could not cast '" << range << "' to a valid number\n";
throw std::invalid_argument{"CAST ERROR"};
}
return num;
}
void run_program(std::filesystem::path & path, std::vector<uint8_t> sn, std::string & aggr_by, bool hd_is_set)
{
std::ifstream file{path.string()};
if (file.is_open())
{
std::vector<double> v;
std::string line;
if (hd_is_set)
std::getline(file, line); // ignore first line
//![altered_while]
while (std::getline(file, line))
{
auto splitted_line = line | std::views::split('\t');
auto it = splitted_line.begin(); // move to 1rst column
if (std::find(sn.begin(), sn.end(), to_number<uint8_t>(*it)) != sn.end())
v.push_back(to_number<double>(*std::next(it, 4)));
}
//![altered_while]
if (aggr_by == "median")
seqan3::debug_stream << ([&v] () { std::sort(v.begin(), v.end()); return v[v.size()/2]; })() << '\n';
else if (aggr_by == "mean")
seqan3::debug_stream << ([&v] () { double sum{}; for (auto i : v) sum += i; return sum / v.size(); })()
<< '\n';
else
seqan3::debug_stream << "I do not know the aggregation method " << aggr_by << '\n';
}
else
{
seqan3::debug_stream << "Error: Cannot open file for reading.\n";
}
}
// -----------------------------------------------------------------------------
//![solution]
struct cmd_arguments
{
std::filesystem::path file_path{};
std::vector<uint8_t> seasons{};
std::string aggregate_by{"mean"};
bool header_is_set{};
};
void initialise_argument_parser(seqan3::argument_parser & parser, cmd_arguments & args)
{
parser.info.author = "Cercei";
parser.info.short_description = "Aggregate average Game of Thrones viewers by season.";
parser.info.version = "1.0.0";
parser.add_positional_option(args.file_path, "Please provide a tab separated data file.");
parser.add_option(args.seasons, 's', "season", "Choose the seasons to aggregate.");
parser.add_option(args.aggregate_by, 'a', "aggregate-by", "Choose your method of aggregation: mean or median.");
parser.add_flag(args.header_is_set, 'H', "header-is-set", "Let us know whether your data file contains a "
"header to ensure correct parsing.");
}
//![solution]
int main(int argc, char ** argv)
{
seqan3::argument_parser myparser{"Game-of-Parsing", argc, argv}; // initialise myparser
cmd_arguments args{};
initialise_argument_parser(myparser, args);
try
{
myparser.parse(); // trigger command line parsing
}
catch (seqan3::argument_parser_error const & ext) // catch user errors
{
seqan3::debug_stream << "[Winter has come] " << ext.what() << "\n"; // customise your error message
return -1;
}
// parsing was successful !
// we can start running our program
run_program(args.file_path, args.seasons, args.aggregate_by, args.header_is_set);
return 0;
}
|
