~brianaker/libmemcached/1164440

« back to all changes in this revision

Viewing changes to libtest/core_count.cc

  • Committer: Continuous Integration
  • Date: 2012-10-22 05:56:09 UTC
  • mfrom: (1086.1.8 libmemcached-1.0)
  • Revision ID: ci@tangent.org-20121022055609-cbooaw9bcdal4qge
Merge lp:~tangent-org/libmemcached/1.0-build Build: jenkins-Libmemcached-1.0-87

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
1
/*  vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
 
2
 *
 
3
 *  Data Differential YATL (i.e. libtest)  library
 
4
 *
 
5
 *  Copyright (C) 2012 Data Differential, http://datadifferential.com/
 
6
 *
 
7
 *  Redistribution and use in source and binary forms, with or without
 
8
 *  modification, are permitted provided that the following conditions are
 
9
 *  met:
 
10
 *
 
11
 *      * Redistributions of source code must retain the above copyright
 
12
 *  notice, this list of conditions and the following disclaimer.
 
13
 *
 
14
 *      * Redistributions in binary form must reproduce the above
 
15
 *  copyright notice, this list of conditions and the following disclaimer
 
16
 *  in the documentation and/or other materials provided with the
 
17
 *  distribution.
 
18
 *
 
19
 *      * The names of its contributors may not be used to endorse or
 
20
 *  promote products derived from this software without specific prior
 
21
 *  written permission.
 
22
 *
 
23
 *  THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 
24
 *  "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 
25
 *  LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
 
26
 *  A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
 
27
 *  OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 
28
 *  SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 
29
 *  LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 
30
 *  DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 
31
 *  THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 
32
 *  (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 
33
 *  OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 
34
 *
 
35
 */
 
36
 
 
37
#include <config.h>
 
38
 
 
39
#include <cstdlib>
 
40
#include <fcntl.h>
 
41
#include <getopt.h>
 
42
#include <iostream>
 
43
#include <sys/stat.h>
 
44
#include <sys/types.h>
 
45
#include <unistd.h>
 
46
 
 
47
#include <libtest/cpu.hpp>
 
48
 
 
49
static void version_command(const char *command_name, int major_version, int minor_version)
 
50
{
 
51
  std::cout << command_name << " " << major_version << "." << minor_version << std::endl;
 
52
}
 
53
 
 
54
static void help_command(const char *command_name,
 
55
                         int major_version, int minor_version,
 
56
                         const struct option *long_options)
 
57
{
 
58
  std::cout << command_name << " " << major_version << "." << minor_version << std::endl;
 
59
  std::cout << "Prints the number of cores found on the local host." << std::endl << std::endl;
 
60
 
 
61
  for (uint32_t x= 0; long_options[x].name; x++)
 
62
  {
 
63
    std::cout << "\t --" << long_options[x].name << char(long_options[x].has_arg ? '=' : ' ') << std::endl;
 
64
  }
 
65
 
 
66
  std::cout << std::endl;
 
67
}
 
68
 
 
69
enum {
 
70
  OPT_HELP,
 
71
  OPT_VERSION
 
72
};
 
73
 
 
74
static void options_parse(int argc, char *argv[])
 
75
{
 
76
  static struct option long_options[]=
 
77
  {
 
78
    { "version", no_argument, NULL, OPT_VERSION},
 
79
    { "help", no_argument, NULL, OPT_HELP},
 
80
    {0, 0, 0, 0},
 
81
  };
 
82
 
 
83
  bool opt_version= false;
 
84
  bool opt_help= false;
 
85
  int option_index= 0;
 
86
 
 
87
  while (1)
 
88
  {
 
89
    int option_rv= getopt_long(argc, argv, "", long_options, &option_index);
 
90
    if (option_rv == -1) 
 
91
    {
 
92
      break;
 
93
    }
 
94
 
 
95
    switch (option_rv)
 
96
    {
 
97
    case OPT_HELP: /* --help or -h */
 
98
      opt_help= true;
 
99
      break;
 
100
 
 
101
    case OPT_VERSION: /* --version or -v */
 
102
      opt_version= true;
 
103
      break;
 
104
 
 
105
    case '?':
 
106
      /* getopt_long already printed an error message. */
 
107
      exit(EXIT_FAILURE);
 
108
 
 
109
    default:
 
110
      help_command(argv[0], 1, 0, long_options);
 
111
      exit(EXIT_FAILURE);
 
112
    }
 
113
  }
 
114
 
 
115
  if (opt_version)
 
116
  {
 
117
    version_command(argv[0], 1, 0);
 
118
    exit(EXIT_SUCCESS);
 
119
  }
 
120
 
 
121
  if (opt_help)
 
122
  {
 
123
    help_command(argv[0], 1, 0, long_options);
 
124
    exit(EXIT_SUCCESS);
 
125
  }
 
126
}
 
127
 
 
128
int main(int argc, char *argv[])
 
129
{
 
130
  options_parse(argc, argv);
 
131
 
 
132
  std::cout << libtest::number_of_cpus() << std::endl;
 
133
 
 
134
  return EXIT_SUCCESS;
 
135
}