Prepare lexer-shell for benchmarking: process multiple input files, adjust output.

src/lexer/lexer-shell.cc

 // Copyright 2013 the V8 project authors. All rights reserved.
 // Redistribution and use in source and binary forms, with or without
 // modification, are permitted provided that the following conditions are
 // met:
 //
 //     * Redistributions of source code must retain the above copyright
 //       notice, this list of conditions and the following disclaimer.
 //     * Redistributions in binary form must reproduce the above
 //       copyright notice, this list of conditions and the following
 //       disclaimer in the documentation and/or other materials provided
@@ skipping 12 matching lines @@
 // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 
 #include <assert.h>
 #include <fcntl.h>
 #include <string.h>
 #include <stdio.h>
 #include <stdlib.h>
+#include <string>
 #include "v8.h"
 
 #include "api.h"
 #include "ast.h"
 #include "char-predicates-inl.h"
 #include "messages.h"
 #include "platform.h"
 #include "runtime.h"
 #include "scanner-character-streams.h"
 #include "scopeinfo.h"
@@ skipping 30 matching lines @@
   }
   fclose(file);
   return chars;
 }
 
 class BaselineScanner {
  public:
   BaselineScanner(const char* fname,
                   Isolate* isolate,
                   Encoding encoding,
-                  ElapsedTimer* timer) {
+                  ElapsedTimer* timer) : stream_(NULL) {
     int length = 0;
     source_ = ReadFile(fname, isolate, &length);
     unicode_cache_ = new UnicodeCache();
     scanner_ = new Scanner(unicode_cache_);
     switch (encoding) {
       case ASCII:
       case UTF8:
         stream_ = new Utf8ToUtf16CharacterStream(source_, length);
         break;
       case UTF16: {
         Handle<String> result = isolate->factory()->NewStringFromTwoByte(
             Vector<const uint16_t>(
                 reinterpret_cast<const uint16_t*>(source_),
                 length / 2));
         stream_ =
             new GenericStringUtf16CharacterStream(result, 0, result->length());
         break;
       }
       case LATIN1: {
         Handle<String> result = isolate->factory()->NewStringFromOneByte(
             Vector<const uint8_t>(source_, length));
         stream_ =
             new GenericStringUtf16CharacterStream(result, 0, result->length());
         break;
       }
-      default:
-        break;
     }
     timer->Start();
     scanner_->Initialize(stream_);
   }
 
   ~BaselineScanner() {
     delete scanner_;
     delete stream_;
     delete unicode_cache_;
     delete[] source_;
@@ skipping 79 matching lines @@
                  const std::vector<TokenWithLocation>& tokens) {
   printf("No of tokens: %d\n",
          static_cast<int>(tokens.size()));
   printf("%s:\n", name);
   for (size_t i = 0; i < tokens.size(); ++i) {
     tokens[i].Print("=>");
   }
 }
 
 
+std::pair<TimeDelta, TimeDelta> ProcessFile(
+    const char* fname,
+    Encoding encoding,
+    Isolate* isolate,
+    bool run_baseline,
+    bool run_experimental,
+    bool print_tokens) {
+  if (print_tokens) {
+    printf("Processing file %s\n", fname);
+  }
+  HandleScope handle_scope(isolate);
+  std::vector<TokenWithLocation> baseline_tokens, experimental_tokens;
+  TimeDelta baseline_time, experimental_time;
+  if (run_baseline) {
+    baseline_time = RunBaselineScanner(
+        fname, isolate, encoding, print_tokens, &baseline_tokens);
+  }
+  if (run_experimental) {
+    experimental_time = RunExperimentalScanner(
+        fname, isolate, encoding, print_tokens, &experimental_tokens);
+  }
+  if (print_tokens && !run_experimental) {
+    PrintTokens("Baseline", baseline_tokens);
+  }
+  if (print_tokens && !run_baseline) {
+    PrintTokens("Experimental", experimental_tokens);
+  }
+  if (print_tokens && run_baseline && run_experimental) {
+    printf("No of tokens in Baseline:     %d\n",
+           static_cast<int>(baseline_tokens.size()));
+    printf("No of tokens in Experimental: %d\n",
+           static_cast<int>(experimental_tokens.size()));
+    printf("Baseline and Experimental:\n");
+    for (size_t i = 0; i < experimental_tokens.size(); ++i) {
+      experimental_tokens[i].Print("=>");
+      if (experimental_tokens[i] != baseline_tokens[i]) {
+        printf("MISMATCH:\n");
+        baseline_tokens[i].Print("Expected: ");
+        experimental_tokens[i].Print("Actual:   ");
+        exit(1);
+      }
+    }
+  }
+  return std::make_pair(baseline_time, experimental_time);
+}
+
+
 int main(int argc, char* argv[]) {
   v8::V8::InitializeICU();
   v8::V8::SetFlagsFromCommandLine(&argc, argv, true);
   Encoding encoding = ASCII;
   bool print_tokens = false;
   bool run_baseline = true;
   bool run_experimental = true;
-  char* fname = argv[1];
+  std::vector<std::string> fnames;
+  std::string benchmark;
   for (int i = 0; i < argc; ++i) {
     if (strcmp(argv[i], "--latin1") == 0) {
       encoding = LATIN1;
     } else if (strcmp(argv[i], "--utf8") == 0) {
       encoding = UTF8;
     } else if (strcmp(argv[i], "--utf16") == 0) {
       encoding = UTF16;
     } else if (strcmp(argv[i], "--ascii") == 0) {
       encoding = ASCII;
     } else if (strcmp(argv[i], "--print-tokens") == 0) {
       print_tokens = true;
     } else if (strcmp(argv[i], "--no-baseline") == 0) {
       run_baseline = false;
     } else if (strcmp(argv[i], "--no-experimental") == 0) {
       run_experimental = false;
-    } else if (argv[i][0] != '-') {
-      fname = argv[i];
+    } else if (strncmp(argv[i], "--benchmark=", 12) == 0) {
+      benchmark = std::string(argv[1]).substr(12);
+    } else if (i > 0 && argv[i][0] != '-') {
+      fnames.push_back(std::string(argv[i]));
     }
   }
   v8::Isolate* isolate = v8::Isolate::GetCurrent();
   {
     v8::HandleScope handle_scope(isolate);
     v8::Handle<v8::ObjectTemplate> global = v8::ObjectTemplate::New();
     v8::Local<v8::Context> context = v8::Context::New(isolate, NULL, global);
     ASSERT(!context.IsEmpty());
     {
       v8::Context::Scope scope(context);
       Isolate* isolate = Isolate::Current();
-      HandleScope handle_scope(isolate);
-
-      std::vector<TokenWithLocation> baseline_tokens, experimental_tokens;
-      TimeDelta baseline_time, experimental_time;
+      double baseline_total = 0, experimental_total = 0;
+      for (size_t i = 0; i < fnames.size(); i++) {
+        std::pair<TimeDelta, TimeDelta> times;
+        times = ProcessFile(fnames[i].c_str(), encoding, isolate,
+                            run_baseline, run_experimental, print_tokens);
+        baseline_total += times.first.InMillisecondsF();
+        experimental_total += times.second.InMillisecondsF();
+      }
       if (run_baseline) {
-        baseline_time = RunBaselineScanner(
-            fname, isolate, encoding, print_tokens, &baseline_tokens);
+        printf("Baseline%s(RunTime): %.f ms\n", benchmark.c_str(),
+               baseline_total);
       }
       if (run_experimental) {
-        experimental_time = RunExperimentalScanner(
-            fname, isolate, encoding, print_tokens, &experimental_tokens);
-      }
-      if (print_tokens && !run_experimental) {
-        PrintTokens("Baseline", baseline_tokens);
-      }
-      if (print_tokens && !run_baseline) {
-        PrintTokens("Experimental", experimental_tokens);
-      }
-      if (print_tokens && run_baseline && run_experimental) {
-        printf("No of tokens in Baseline:     %d\n",
-               static_cast<int>(baseline_tokens.size()));
-        printf("No of tokens in Experimental: %d\n",
-               static_cast<int>(experimental_tokens.size()));
-        printf("Baseline and Experimental:\n");
-        for (size_t i = 0; i < experimental_tokens.size(); ++i) {
-          experimental_tokens[i].Print("=>");
-          if (experimental_tokens[i] != baseline_tokens[i]) {
-            printf("MISMATCH:\n");
-            baseline_tokens[i].Print("Expected: ");
-            experimental_tokens[i].Print("Actual:   ");
-            return 1;
-          }
-        }
-      }
-      if (run_baseline) {
-        printf("Baseline    : %.3f ms\n", baseline_time.InMillisecondsF());
-      }
-      if (run_experimental) {
-        printf("Experimental: %.3f ms\n", experimental_time.InMillisecondsF());
+        printf("Experimental%s(RunTime): %.f ms\n", benchmark.c_str(),
+               experimental_total);
       }
     }
   }
   v8::V8::Dispose();
   return 0;
 }