diff --git a/project/main.cpp b/project/main.cpp index f149c8c..e559e85 100644 --- a/project/main.cpp +++ b/project/main.cpp @@ -5,6 +5,8 @@ #include "text.h" #include "svg.h" #include +#include +#include using namespace std; struct Input @@ -27,37 +29,58 @@ input_data(istream& in, bool promt) ik.numbers.resize(number_count); for (size_t i = 0; i < number_count; i++) { - cin >> ik.numbers[i]; + in >> ik.numbers[i]; } cerr << "Enter bin count: "; - cin>> ik.bin_count; + in>> ik.bin_count; return ik; } +size_t write_data(void* items, size_t item_size, size_t item_count, void* ctx) +{ + size_t data_size = item_size * item_count; + stringstream* buffer = reinterpret_cast(ctx); + buffer->write(static_cast(items), data_size); + return 0; +} -int main(int argc, char* argv[]) +Input +download(const string& address) { - if (argc > 1) + stringstream buffer; + CURL *curl = curl_easy_init(); + if(curl) { - CURL *curl = curl_easy_init(); - if(curl) + CURLcode res; + curl_easy_setopt(curl, CURLOPT_URL, address); + curl_easy_setopt(curl, CURLOPT_WRITEDATA, &buffer); + res = curl_easy_perform(curl); + curl_easy_cleanup(curl); + if (res != CURLE_OK) { - CURLcode res; - curl_easy_setopt(curl, CURLOPT_URL, argv[1]); - res = curl_easy_perform(curl); - curl_easy_cleanup(curl); - if (res != CURLE_OK){ - cout << curl_easy_strerror; - exit(1); - } + cout << curl_easy_strerror; + exit(1); } - return 0; } - curl_global_init(CURL_GLOBAL_ALL); - auto in = input_data(cin, true); - auto bins = make_histogram(in.numbers, in.bin_count); + curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, write_data); + + return input_data(buffer, false); +} + + +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); - getch(); - return 0; }