diff options
| -rw-r--r-- | examples/sinewave/sinewave_example.cpp | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/examples/sinewave/sinewave_example.cpp b/examples/sinewave/sinewave_example.cpp index 10be073..1815222 100644 --- a/examples/sinewave/sinewave_example.cpp +++ b/examples/sinewave/sinewave_example.cpp @@ -14,13 +14,13 @@ using Clock = std::chrono::high_resolution_clock; int main(int argc, char* argv[]) { std::chrono::milliseconds interval{ 10ms }; - int samples, opt, i = 0; + int samples, opt, i = 1; bool end = false; const std::function infinite = []() {}; const std::function limited = [&end, &i, &samples]() { - if (i > samples) end = true; i++; + if (i > samples) end = true; }; auto generatorGuard = infinite; @@ -33,6 +33,7 @@ int main(int argc, char* argv[]) { break; case 'n': samples = std::stoi(optarg); + if (samples == 0) end = true; generatorGuard = limited; break; default: |