main: добавлена работа с буфером, загруженным по сети

main
TarasovEE 4 недель назад
Родитель c1136ed04e
Сommit be467a80d3

@ -1,3 +1,5 @@
#include <sstream>
#include <string>
#include <curl/curl.h> #include <curl/curl.h>
#include <iostream> #include <iostream>
#include <vector> #include <vector>
@ -33,29 +35,46 @@ input_data(istream& in, bool prompt)
return local; return local;
} }
int main(int argc, char* argv[]) Input
download(const string& address)
{ {
curl_global_init(CURL_GLOBAL_ALL); stringstream buffer;
if (argc > 1)
CURL *curl = curl_easy_init();
if(curl)
{ {
CURL *curl = curl_easy_init(); CURLcode res;
if(curl) curl_easy_setopt(curl, CURLOPT_URL, address);
res = curl_easy_perform(curl);
if (res != CURLE_OK)
{ {
CURLcode res; cerr << curl_easy_strerror(res);
curl_easy_setopt(curl, CURLOPT_URL, argv[1]); exit(1);
res = curl_easy_perform(curl);
if (res != CURLE_OK)
{
cerr << curl_easy_strerror(res);
exit(1);
}
curl_easy_cleanup(curl);
} }
return 0; curl_easy_cleanup(curl);
} }
return input_data(buffer, false);
}
size_t write_data(void* items, size_t item_size, size_t item_count, void* ctx)
{
size_t data_size = item_size * item_count;
auto in = input_data(cin, true); stringstream* buffer = reinterpret_cast<stringstream*>(ctx);
auto bins = make_histogram(in.numbers, in.bin_count); buffer->write(reinterpret_cast<const char*>(items), data_size);
return data_size;
}
int main(int argc, char* argv[])
{
Input input;
if (argc > 1)
input = download(argv[1]);
else
input = input_data(cin, true);
const auto bins = make_histogram(input.numbers, input.bin_count);
show_histogram_svg(bins); show_histogram_svg(bins);
return 0;
} }

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