diff --git a/main.cpp b/main.cpp index 9ccb94a..6e34731 100644 --- a/main.cpp +++ b/main.cpp @@ -4,43 +4,74 @@ #include "text.h" #include "svg.h" #include - - +#include +#include using namespace std; - - struct Input { - std::vector numbers; + vector numbers; size_t bin_count; }; -// Ввод данных - -Input input_data(istream& in, bool prompt = false) { - Input in_data; +Input input_data(istream& in, bool prompt) { + Input inp; size_t number_count; - if (prompt) cerr << "Enter number count: "; + if (prompt){ + cerr << "Enter number count: "; + } in >> number_count; - in_data.numbers.resize(number_count); - if (prompt && number_count > 0) cerr << "Enter numbers: "; + inp.numbers.resize(number_count); + if (prompt){ + cerr << "Enter numbers: "; + } for (size_t i = 0; i < number_count; i++) { - in >> in_data.numbers[i]; + in >> inp.numbers[i]; + } + + if (prompt) { + cerr << "Enter bin count: "; } - if (prompt) cerr << "Enter bin count: "; - in >> in_data.bin_count; - return in_data; + in >> inp.bin_count; + return inp; } -int main() { -curl_global_init(CURL_GLOBAL_ALL); +size_t +write_data(void* items, size_t item_size, size_t item_count, void* ctx) { + stringstream* buffer = reinterpret_cast(ctx); + size_t data_size = item_size * item_count; + (*buffer).write(reinterpret_cast(items), data_size); + return data_size; +} +Input +download(const string& adress){ + stringstream buffer; + CURL* curl = curl_easy_init(); + if(curl) { - // Читаем данные с консоли с выводом подсказок - Input in_data = input_data(cin, true); // true - подсказки выводятся + 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); + } - auto bins = make_histogram(in_data.numbers, in_data.bin_count); - show_histogram_svg(bins); + 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); + } + auto bins = make_histogram(input.numbers, input.bin_count); + show_histogram_svg(bins); return 0; }