Merge pull request #2 from lioncash/line

output: Shorten countLines
2014-11-27 00:13:44 -05:00 · 2014-11-27 00:13:44 -05:00 · 8186354abd
commit 8186354abd
parent 3b8451683f 5f1df4c2ff
1 changed files with 2 additions and 6 deletions
--- a/source/output.cpp
+++ b/source/output.cpp
@ -1,5 +1,6 @@
 #include "output.h"

+#include <algorithm>
 #include <string>
 #include <cstring>
 #include <cmath>
@ -18,12 +19,7 @@ static int countLines(const std::string& str)
    if (str.empty())
        return 0;

-    int cnt = 1;
-    for (const char& c : str) {
-        if (c == '\n') cnt++;
-    }
-    
-    return cnt;
+    return 1 + std::count_if(str.begin(), str.end(), [](char c) { return c == '\n'; });
 }

 static void deleteFirstLine(std::string* str)