Loading...
tests/MallocBenchTest/MALLOC_BENCH/MallocBench/CommandLine.cpp /dev/null libmalloc-409.40.6
--- /dev/null
+++ libmalloc/libmalloc-409.40.6/tests/MallocBenchTest/MALLOC_BENCH/MallocBench/CommandLine.cpp
@@ -0,0 +1,97 @@
+/*
+ * Copyright (C) 2014 Apple Inc. All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ * 1. Redistributions of source code must retain the above copyright
+ *    notice, this list of conditions and the following disclaimer.
+ * 2. 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 with the distribution.
+ *
+ * THIS SOFTWARE IS PROVIDED BY APPLE INC. ``AS IS'' AND ANY
+ * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+ * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
+ * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE INC. OR
+ * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
+ * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
+ * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 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 "CommandLine.h"
+#include <getopt.h>
+#include <iostream>
+
+struct option CommandLine::longOptions[] =
+{
+    {"benchmark", required_argument, 0, 'b'},
+    {"detailed-report", no_argument, 0, 'd'},
+    {"no-warmup", no_argument, 0, 'n' },
+    {"parallel", no_argument, 0, 'p'},
+    {"heap", required_argument, 0, 'h'},
+    {"runs", required_argument, 0, 'r'},
+    {"use-thread-id", no_argument, 0, 't'},
+    {0, 0, 0, 0}
+};
+
+CommandLine::CommandLine(int argc, char** argv)
+    : m_argc(argc)
+    , m_argv(argv)
+    , m_detailedReport(false)
+    , m_isParallel(false)
+    , m_useThreadID(false)
+    , m_warmUp(true)
+    , m_heapSize(0)
+    , m_runs(8)
+{
+    int optionIndex = 0;
+    int ch;
+    while ((ch = getopt_long(argc, argv, "b:dnph:r:t", longOptions, &optionIndex)) != -1) {
+        switch (ch)
+        {
+            case 'b':
+                m_benchmarkName = optarg;
+                break;
+
+            case 'd':
+                m_detailedReport = true;
+                break;
+                
+            case 'n':
+                m_warmUp = false;
+                break;
+
+            case 'p':
+                m_isParallel = true;
+                break;
+                
+            case 'h':
+                m_heapSize = atoi(optarg) * 1024 * 1024;
+                break;
+
+            case 'r':
+                m_runs = atoi(optarg);
+                break;
+
+            case 't':
+                m_useThreadID = true;
+                break;
+                
+            default:
+                break;
+        }
+    }
+}
+
+void CommandLine::printUsage()
+{
+    std::string fullPath(m_argv[0]);
+    size_t pos = fullPath.find_last_of("/") + 1;
+    std::string program = fullPath.substr(pos);
+    std::cout << "Usage: " << program << " --benchmark benchmark_name [--parallel ] [--use-thread-id ] [--detailed-report] [--no-warmup] [--runs count] [--heap MB ]" << std::endl;
+}