diff --git a/main.cpp b/main.cpp index d2149a0..98b8fbb 100644 --- a/main.cpp +++ b/main.cpp @@ -1,4 +1,6 @@ #include +#include +#include #include "histogram.h" #include "text.h" #include @@ -27,17 +29,41 @@ input_data(istream& in,bool promt = false) { in >> im.bin_count; return im; } +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) { + 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); + curl_easy_cleanup(curl); + if(res != 0){ + cerr << curl_easy_strerror(res); + exit(1); + } + } + return input_data(buffer, false); +} + int main(int argc, char* argv[]) { + curl_global_init(CURL_GLOBAL_ALL); - if(argc > 1) { - cout << "argc = " << argc << endl; - for(int i = 0; i < argc; i++){ - cout << "argv[" << i << "] = " << argv[i] << endl; - } - return 0; + + Input input; + if (argc > 1) { + input = download(argv[1]); + } else { + input = input_data(cin, true); } - auto in = input_data(cin,true); - auto bins = make_histogram(in.numbers, in.bin_count); + auto bins = make_histogram(input.numbers, input.bin_count); show_histogram_svg(bins); return 0; }