diff --git a/main.cpp b/main.cpp index 78c7967..2a73639 100644 --- a/main.cpp +++ b/main.cpp @@ -13,21 +13,39 @@ struct Input { }; Input -input_data(istream& in) { +input_data(istream& in, bool prompt) { size_t number_count; - in >> number_count; - Input inn; - inn.numbers.resize(number_count); - for (size_t i = 0; i < number_count; i++) { - cin >> inn.numbers[i]; + if (prompt == true){ + cerr << "input number count"; + in >> number_count; + Input inn; + inn.numbers.resize(number_count); + cerr << "input numbers"; + for (size_t i = 0; i < number_count; i++) { + cin >> inn.numbers[i]; + } + cerr << "input bin count"; + in >> inn.bin_count; + return inn; + } + else{ + cout << "input number count"; + in >> number_count; + Input inn; + inn.numbers.resize(number_count); + cout << "input numbers"; + for (size_t i = 0; i < number_count; i++) { + cin >> inn.numbers[i]; + } + cerr << "input bin count"; + in >> inn.bin_count; + return inn; } - in >> inn.bin_count; - return inn; } int main(){ - auto in = input_data(cin); + auto in = input_data(cin,true); auto bins = make_histogram(in.numbers, in.bin_count); show_histogram_svg(bins); return 0;