update sources to v12.2.4

[ceph.git] / ceph / src / boost / libs / sort / example / keyplusdatasample.cpp

// spreadsort key and data sorting example.\r
//\r
//  Copyright Steven Ross 2009-2014.\r
//\r
// Distributed under the Boost Software License, Version 1.0.\r
//    (See accompanying file LICENSE_1_0.txt or copy at\r
//          http://www.boost.org/LICENSE_1_0.txt)\r
\r
//  See http://www.boost.org/libs/sort for library home page.\r
\r
#include <boost/sort/spreadsort/spreadsort.hpp>\r
#include <time.h>\r
#include <stdio.h>\r
#include <stdlib.h>\r
#include <algorithm>\r
#include <vector>\r
#include <string>\r
#include <fstream>\r
#include <sstream>\r
#include <iostream>\r
using namespace boost::sort::spreadsort;\r
\r
struct DATA_TYPE {\r
    int key;\r
    std::string data;\r
    };\r
//functor example\r
struct lessthan {\r
  inline bool operator()(const DATA_TYPE &x, const DATA_TYPE &y) const {\r
    return x.key < y.key;\r
  }\r
};\r
\r
struct rightshift {\r
  inline int operator()(const DATA_TYPE &x, const unsigned offset) {\r
    return x.key >> offset;\r
  }\r
};\r
\r
//Pass in an argument to test std::sort\r
int main(int argc, const char ** argv) {\r
  size_t uSize = sizeof(int);\r
  bool stdSort = false;\r
  unsigned loopCount = 1;\r
  for (int u = 1; u < argc; ++u) {\r
    if (std::string(argv[u]) == "-std")\r
      stdSort = true;\r
    else\r
      loopCount = atoi(argv[u]);\r
  }\r
  std::ifstream input("input.txt", std::ios_base::in | std::ios_base::binary);\r
  if (input.fail()) {\r
    printf("input.txt could not be opened\n");\r
    return 1;\r
  }\r
  input.seekg (0, std::ios_base::end);\r
    size_t length = input.tellg();\r
  double total = 0.0;\r
  std::vector<DATA_TYPE> array;\r
  array.reserve(length/uSize);\r
  unsigned uCount = length/uSize;\r
  //Run multiple loops, if requested\r
  for (unsigned u = 0; u < loopCount; ++u) {\r
    input.seekg (0, std::ios_base::beg);\r
    unsigned v = 0;\r
    while (input.good() && v++ < uCount) { // EOF or failure stops the reading\r
     DATA_TYPE element;\r
     input.read(reinterpret_cast<char *>(&(element.key)), sizeof(element.key));\r
     std::stringstream intstr;\r
         intstr << element.key;\r
         element.data = intstr.str();\r
         array.push_back(element);\r
    }  \r
    clock_t start, end;\r
    double elapsed;\r
    start = clock();\r
    if (stdSort)\r
      std::sort(array.begin(), array.end(), lessthan());\r
    else\r
      integer_sort(array.begin(), array.end(), rightshift(), lessthan());\r
    end = clock();\r
    elapsed = static_cast<double>(end - start) ;\r
    std::ofstream ofile;\r
    if (stdSort)\r
      ofile.open("standard_sort_out.txt", std::ios_base::out |\r
                 std::ios_base::binary | std::ios_base::trunc);\r
    else\r
      ofile.open("boost_sort_out.txt", std::ios_base::out |\r
                 std::ios_base::binary | std::ios_base::trunc);\r
    if (ofile.good()) {\r
      for (unsigned v = 0; v < array.size(); ++v) {\r
        ofile.write(reinterpret_cast<char *>(&(array[v].key)), \r
                    sizeof(array[v].key));\r
        ofile << array[v].data;\r
      }\r
      ofile.close();\r
    }\r
    total += elapsed;\r
    array.clear();\r
  }\r
  input.close();\r
  if (stdSort)\r
    printf("std::sort elapsed time %f\n", total / CLOCKS_PER_SEC);\r
  else\r
    printf("spreadsort elapsed time %f\n", total / CLOCKS_PER_SEC);\r
  return 0;\r
}\r