pakhomov 3 недель назад
Родитель 10be306236
Сommit dae11e2beb

@ -3,7 +3,9 @@
#include "histogram.h" #include "histogram.h"
#include "text.h" #include "text.h"
#include "svg.h" #include "svg.h"
#include <curl/curl.h> #include "curl/include/curl/curl.h"
#include <sstream>
#include <string>
using namespace std; using namespace std;
struct Input { struct Input {
@ -17,40 +19,64 @@ Input input_data(istream& in, bool prompt) {
if (prompt){ if (prompt){
cerr << "Enter number count: "; cerr << "Enter number count: ";
} }
cin >> number_count; in >> number_count;
inp.numbers.resize(number_count); inp.numbers.resize(number_count);
if (prompt){ if (prompt){
cerr << "Enter numbers: "; cerr << "Enter numbers: ";
} }
for (size_t i = 0; i < number_count; i++) { for (size_t i = 0; i < number_count; i++) {
cin >> inp.numbers[i]; in >> inp.numbers[i];
} }
if (prompt) { if (prompt) {
cerr << "Enter bin count: "; cerr << "Enter bin count: ";
} }
cin >> inp.bin_count; in >> inp.bin_count;
return inp; return inp;
} }
size_t
write_data(void* items, size_t item_size, size_t item_count, void* ctx) {
stringstream* buffer = reinterpret_cast<stringstream*>(ctx);
size_t data_size = item_size * item_count;
(*buffer).write(reinterpret_cast<const char*>(items), data_size);
return data_size;
}
Input
download(const string& adress){
stringstream buffer;
CURL* curl = curl_easy_init();
if(curl) {
CURLcode res;
curl_easy_setopt(curl, CURLOPT_URL, adress.c_str());
curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, write_data);
curl_easy_setopt(curl, CURLOPT_WRITEDATA, &buffer);
res = curl_easy_perform(curl);
if (res != CURLE_OK){
cerr << curl_easy_strerror(res);
exit(1);
}
curl_easy_cleanup(curl);
}
return input_data(buffer, false);
}
int main(int argc, char* argv[]) { int main(int argc, char* argv[]) {
curl_global_init(CURL_GLOBAL_ALL); curl_global_init(CURL_GLOBAL_ALL);
Input input;
if (argc > 1) { if (argc > 1) {
CURL* curl = curl_easy_init(); input = download(argv[1]);
if(curl) { } else {
CURLcode res; input = input_data(cin, true);
curl_easy_setopt(curl, CURLOPT_URL, argv[1]);
res = curl_easy_perform(curl);
if (res != CURLE_OK){
cerr << curl_easy_strerror(res);
exit(1);
}
curl_easy_cleanup(curl);
}
} }
auto in = input_data(cin, true);
auto bins = make_histogram(in.numbers, in.bin_count); auto bins = make_histogram(input.numbers, input.bin_count);
show_histogram_svg(bins); show_histogram_svg(bins);
return 0; return 0;
} }

Загрузка…
Отмена
Сохранить