From ab4bf9fff0af78dbebb35bb49f74c5ba91231e9e Mon Sep 17 00:00:00 2001 From: "laba34 (BuntovaMS)" Date: Mon, 2 Jun 2025 12:26:50 +0300 Subject: [PATCH] =?UTF-8?q?=D0=A1=D0=BE=D1=85=D1=80=D0=B0=D0=BD=D0=B5?= =?UTF-8?q?=D0=BD=D0=B8=D0=B5=20=D0=B4=D0=B0=D0=BD=D0=BD=D1=8B=D1=85=20?= =?UTF-8?q?=D0=B8=D0=B7=20=D1=81=D0=B5=D1=82=D0=B8=20=D0=B2=20=D0=B1=D1=83?= =?UTF-8?q?=D1=84=D0=B5=D1=80?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- main.cpp | 57 ++++++++++++++++++++++++++++++++------------------------ 1 file changed, 33 insertions(+), 24 deletions(-) diff --git a/main.cpp b/main.cpp index c5093ea..60b1c19 100644 --- a/main.cpp +++ b/main.cpp @@ -1,11 +1,13 @@ #include #include #include +#include +#include #include "histogram.h" #include "text.h" #include "svg.h" struct Input { - vector numbers; + std::vector numbers; size_t bin_count{}; }; @@ -37,34 +39,41 @@ Input input_data(std::istream& in, bool prompt) return data; } -int main(int argc, char* argv[]) +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& address) { - curl_global_init(CURL_GLOBAL_ALL); - using namespace std; - - if (argc > 1) + stringstream buffer; + CURL* curl = curl_easy_init(); + if (curl) { - const char *url = argv[1]; - - CURL *curl = curl_easy_init(); - if (curl) + curl_easy_setopt(curl, CURLOPT_URL, address.c_str()); + curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, write_data); + curl_easy_setopt(curl, CURLOPT_WRITEDATA, &buffer); + CURLcode res = curl_easy_perform(curl); + curl_easy_cleanup(curl); + if (res != CURLE_OK) { - curl_easy_setopt(curl, CURLOPT_URL, url); - curl_easy_perform(curl); - - CURLcode res = curl_easy_perform(curl); - if (res != CURLE_OK) - { - cerr << "curl_easy_perform() failed: " << curl_easy_strerror(res) << endl; - exit(1); - } - curl_easy_cleanup(curl); + cerr << "cURL error: " << curl_easy_strerror(res) << endl; + exit(1); } - return 0; + return input_data(buffer, false); } +} - auto in = input_data(cin,true); - auto bins = make_histogram(in.bin_count, in.numbers); +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.bin_count, input.numbers); show_histogram_svg(bins); - return 0; } +