JMergeSort.cc

Go to the documentation of this file.

 
#include <iostream>
#include <iomanip>
#include <functional>
#include <limits>
#include <vector>
#include <algorithm>
 
#include "TRandom3.h"
 
#include "JTools/JMergeSort.hh"
#include "JLang/JSTDToolkit.hh"
 
#include "Jeep/JTimer.hh"
#include "Jeep/JParser.hh"
#include "Jeep/JMessage.hh"
 
 
namespace JTOOLS {
 
  /**
   * Merge multiple sorted ranges.
   *
   * \param  __begin            begin of data
   * \param  N                  number of delimiters (equals number of ranges plus one)
   * \param  delimiter          incremental delimiters
   */
  template<class T>
  void inplace_merge(T __begin, const size_t N, const size_t* delimiter)
  {
    switch (N - 1) {
 
    case 0:
    case 1:
      break;
 
    case 2:
 
      std::inplace_merge(__begin + delimiter[0],
                         __begin + delimiter[1],
                         __begin + delimiter[2]);
 
      break;
 
    default:
      
      const size_t n = (N - 1) >> 1;
 
      inplace_merge(__begin, n + 1, delimiter);
      inplace_merge(__begin, N - n, delimiter + n);
 
      std::inplace_merge(__begin + delimiter[ 0 ],
                         __begin + delimiter[ n ],
                         __begin + delimiter[N-1]);
 
      break;
    };
  }
 
 
  /**
   * Auxiliary class to convert value to element.
   */
  template<class T>
  struct JElement
  {
    /**
     * Default constructor.
     */
    JElement()
    {}
    
 
    /**
     * Type conversion operator.
     *
     * \param  x                  value
     * \return                    element
     */
    inline T operator()(const double x) const
    {
      return (T) x;
    }
 
 
    /**
     * Get end marker.
     *
     * \return                    element
     */
    static inline T getEndMarker()
    {
      return std::numeric_limits<T>::max();
    }
  };
}
 
 
/**
 * \file
 *
 * Example program to test performance of JTOOLS::JMergeSort.
 * \author mdejong
 */
int main(int argc, char **argv)
{
  using namespace std;
  using namespace JPP;
 
  double         rate_Hz;
  int            debug;
 
  try {
 
    JParser<> zap("Example program to test performance of merge sort.");
 
    zap['B'] = make_field(rate_Hz);
    zap['d'] = make_field(debug)               = 0;
 
    zap(argc, argv);
  }
  catch(const exception &error) {
    FATAL(error.what() << endl);
  }
 
 
  if (rate_Hz <= 0.0) {
    FATAL("Rate " << rate_Hz << " < 0 Hz." << endl);
  }
 
  const double period = 1.0e9 / rate_Hz;      // [ns]
 
  const double Tmin   = 0.0;                  // [ns]
  const double Tmax   = 1.0e8;                // [ns]
 
  const int NUMBER_OF_PMTS    =       31;
  const int NUMBER_OF_MODULES = 115 * 18;
 
 
  //typedef int                        hit_type; 
  typedef double                     hit_type; 
  typedef JElement<hit_type>         JElement_t;
  const   JElement_t                 getElement;
  typedef std::vector<hit_type>      JBuffer1D;
  typedef std::vector<JBuffer1D>     JBuffer2D;
 
 
  // generate random data
 
  JBuffer2D input(NUMBER_OF_PMTS);
 
  size_t number_of_hits = 0;
 
  for (int i = 0; i != NUMBER_OF_PMTS; ++i) {
 
    for (double t1 = Tmin + gRandom->Exp(period); t1 < Tmax; t1 += gRandom->Exp(period), ++number_of_hits) {
      input[i].push_back(getElement(t1));
    }
 
    NOTICE("PMT[" << setw(2) << i << "]  " << input[i].size() << endl);
 
    putEndMarker(input[i], JElement_t::getEndMarker());
  }
 
 
  JTimer timer[] = {
    JTimer("std::sort"),
    JTimer("std::inplace_merge"),
    JTimer("JMergeSort")
  };
 
 
  {
    JBuffer1D buffer;
 
    for (int i = 0; i != NUMBER_OF_MODULES; ++i) {
 
      timer[0].start();
 
      buffer.clear();
 
      for (int i = 0; i != NUMBER_OF_PMTS; ++i) {
        copy(input[i].begin(), input[i].end(), back_inserter(buffer));
      }
 
      std::sort(buffer.begin(), buffer.end());
 
      timer[0].stop();
 
      ASSERT(buffer.size() == number_of_hits,         "Test std::sort.");
      ASSERT(is_sorted(buffer.begin(), buffer.end()), "Test std::sort.");
    }
  }
 
  {
    JBuffer1D buffer;
 
    vector<size_t> delimiter(NUMBER_OF_PMTS + 1);
 
    for (int i = 0; i != NUMBER_OF_MODULES; ++i) {
 
      timer[1].start();
 
      size_t n = 0;
 
      for (int i = 0; i != NUMBER_OF_PMTS; ++i) {
        n += input[i].size();
      }
 
      buffer.resize(n);
 
      JBuffer1D::iterator out = buffer.begin();      
 
      delimiter[0] = 0;
 
      for (int i = 0; i != NUMBER_OF_PMTS; ++i) {
 
        out = copy(input[i].begin(), input[i].end(), out);
 
        delimiter[i+1] = distance(buffer.begin(), out);
      }
 
      inplace_merge(buffer.begin(), delimiter.size(), delimiter.data());
 
      timer[1].stop();
 
      ASSERT(buffer.size() == number_of_hits,         "Test std::inplace_merge.");
      ASSERT(is_sorted(buffer.begin(), buffer.end()), "Test std::inplace_merge.");
    }
  }
 
  {
    JBuffer1D            buffer;
    JMergeSort<hit_type> merge;
 
    for (int i = 0; i != NUMBER_OF_MODULES; ++i) {
 
      timer[2].start();
 
      buffer.clear();
      
      merge(input, buffer);
 
      timer[2].stop();
 
      ASSERT(buffer.size() == number_of_hits,         "Test JMergeSort.");
      ASSERT(is_sorted(buffer.begin(), buffer.end()), "Test JMergeSort.");
    }
  }
 
 
  for (int i = 0; i != sizeof(timer)/sizeof(timer[0]); ++i) {
    timer[i].print(cout, false);
  }
}