diff --git a/main.cpp b/main.cpp index 352aede..a6f8fce 100644 --- a/main.cpp +++ b/main.cpp @@ -1,11 +1,10 @@ #include -#include +#include //Файл подключает stringstream — поток, который может читать из буфера в памяти #include -#include -#include +#include #include -#include #include "hictogram.h" +#include "text.h" #include "svg.h" using namespace std; @@ -16,6 +15,7 @@ struct Input { }; Input input_data(istream& in, bool prompt) { + if (prompt) { cerr << "Enter number count, numbers and bin count: "; } @@ -35,15 +35,56 @@ Input input_data(istream& in, bool prompt) { return input; } -int main() { - curl_global_init(CURL_GLOBAL_ALL); + + +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(reinterpret_cast(items), data_size); + + return data_size; +} + +Input download(const string& address) { + stringstream buffer; + + 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); + + if (res != CURLE_OK) { + cerr << "cURL error: " << curl_easy_strerror(res) << endl; + curl_easy_cleanup(curl); + exit(1); + } + + curl_easy_cleanup(curl); + } else { + cerr << "curl_easy_init() failed" << endl; + exit(1); + } + + 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); - curl_global_cleanup(); return 0; } - -