From 78812b52361520c54a308dfe1c18222a1df3183d Mon Sep 17 00:00:00 2001 From: lzwdgc Date: Tue, 6 Oct 2015 12:20:50 +0300 Subject: [PATCH] Add option to name generator how many names to generate. --- src/name_generator/name_generator.cpp | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/src/name_generator/name_generator.cpp b/src/name_generator/name_generator.cpp index 93cedf9..d3b81df 100644 --- a/src/name_generator/name_generator.cpp +++ b/src/name_generator/name_generator.cpp @@ -82,9 +82,11 @@ private: } }; -int main() +int main(int argc, char *argv[]) { + int n = argc == 1 ? 1 : std::stoi(argv[1]); NameGenerator ng; - std::cout << ng.generate(); + for (int i = 0; i < n; i++) + std::cout << ng.generate() << (i != n-1 ? "\n" : ""); return 0; }