Получение аргументов программы

master
victoriaCS 2 месяцев назад
Родитель a2d7dad6a8
Сommit 3ecb25b0b9

@ -1,8 +1,9 @@
#include <curl/curl.h>
#define CURL_STATICLIB
#include <iostream> #include <iostream>
#include <sstream>
#include <string>
#include <vector> #include <vector>
#include "histogram.h" #include <curl/curl.h>
#include "hictogram.h"
#include "text.h" #include "text.h"
#include "svg.h" #include "svg.h"
@ -11,63 +12,73 @@ using namespace std;
struct Input { struct Input {
vector<double> numbers; vector<double> numbers;
size_t bin_count{}; size_t bin_count{};
size_t block_width{};
}; };
Input input_data(istream& in_stream = cin, bool prompt = true) { Input input_data(istream& in, bool prompt) {
Input in;
size_t number_count;
if (prompt) { if (prompt) {
cerr << "Enter number count: "; cerr << "Enter number count, numbers and bin count: ";
} }
in_stream >> number_count;
in.numbers.resize(number_count); size_t number_count;
in >> number_count;
Input input;
input.numbers.resize(number_count);
if (prompt) {
cerr << "Enter " << number_count << " numbers: ";
}
for (size_t i = 0; i < number_count; i++) { for (size_t i = 0; i < number_count; i++) {
in_stream >> in.numbers[i]; in >> input.numbers[i];
} }
if (prompt) { in >> input.bin_count;
cerr << "Enter bin count: ";
}
in_stream >> in.bin_count;
return in; return input;
} }
vector<size_t> make_histogram(const vector<double>& numbers, size_t bin_count) { size_t write_data(void* items, size_t item_size, size_t item_count, void* ctx) {
vector<size_t> bins(bin_count, 0); size_t data_size = item_size * item_count;
for (double num : numbers) {
size_t index = min(bin_count - 1, static_cast<size_t>(num)); // ïðèìåð stringstream* buffer = reinterpret_cast<stringstream*>(ctx);
bins[index]++; buffer->write(reinterpret_cast<const char*>(items), data_size);
}
return bins; 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[]) { int main(int argc, char* argv[]) {
if (argc > 1) { if (argc > 1) {
cout << "argc = " << argc << endl; std::cout << "argc = " << argc << std::endl;
for (int i = 0; i < argc; ++i) { for (int i = 0; i < argc; ++i) {
cout << "argv[" << i << "] = \"" << argv[i] << "\"" << endl; std::cout << "argv[" << i << "] = " << argv[i] << std::endl;
} }
return 0; return 0;
} }
// Åñëè àðãóìåíòîâ íåò, ïðîãðàììà íè÷åãî íå äåëàåò è ïðîñòî çàâåðøàåò
CURLcode res = curl_global_init(CURL_GLOBAL_ALL); return 0;
if (res != CURLE_OK) {
cerr << "cURL initialization failed: " << curl_easy_strerror(res) << endl;
return 1;
}
auto in = input_data();
auto bins = make_histogram(in.numbers, in.bin_count);
show_histogram_svg(bins, in.block_width);
curl_global_cleanup();
} }

Загрузка…
Отмена
Сохранить