// Copyright (C) 2003, 2004, 2005 Free Software Foundation, Inc.
//
// This file is part of the GNU ISO C++ Library.  This library is free
// software; you can redistribute it and/or modify it under the
// terms of the GNU General Public License as published by the
// Free Software Foundation; either version 2, or (at your option)
// any later version.

// This library is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.

// You should have received a copy of the GNU General Public License along
// with this library; see the file COPYING.  If not, write to the Free
// Software Foundation, 59 Temple Place - Suite 330, Boston, MA 02111-1307,
// USA.

// As a special exception, you may use this file as part of a free software
// library without restriction.  Specifically, if other files instantiate
// templates or use macros or inline functions from this file, or you compile
// this file and link it with other files to produce an executable, this
// file does not by itself cause the resulting executable to be covered by
// the GNU General Public License.  This exception does not however
// invalidate any other reasons why the executable file might be covered by
// the GNU General Public License.

/*
 * The goal with this application is to compare the performance
 * between different std::allocator implementations. The results are
 * influenced by the underlying allocator in the "C" library, malloc.
 */

// 2003-02-05 Stefan Olsson <stefan@snon.net>

#include <vector>
#include <list>
#include <map>
#include <deque>
#include <set>
#include <typeinfo>
#include <sstream>
#include <ext/mt_allocator.h>
#include <ext/new_allocator.h>
#include <ext/malloc_allocator.h>
#include <ext/bitmap_allocator.h>
#include <ext/pool_allocator.h>
#include <cxxabi.h>
#include <testsuite_performance.h>

using namespace std;

typedef int test_type;

// The number of iterations to be performed.
int iterations = 10000;

// The number of values to insert in the container, 32 will cause 5
// (re)allocations to be performed (sizes 4, 8, 16, 32 and 64)
// This means that all allocations are within _MAX_BYTES = 128 as
// defined in stl_alloc.h for __pool_alloc.  Whether or not this
// value is relevant in "the real world" or not I don't know and
// should probably be investigated in more detail.
int insert_values = 128;

template<typename TestType>
  struct value_type : public pair<TestType, TestType>
  {
    value_type() : pair<TestType, TestType>(0, 0) { }

    inline value_type operator++() { return ++this->first, *this; }
    inline operator TestType() const { return this->first; }
  };

template<typename Container>
  void
  do_loop()
  {
    Container obj;
    int test_iterations = 0;
    value_type<test_type> test_value;
    while (test_iterations < iterations)
      {
	for (int j = 0; j < insert_values; ++j)
	  obj.insert(obj.end(), ++test_value);
	++test_iterations;
      }
  }

template<typename Container>
  void*
  do_test(void* p = NULL)
  {
    do_loop<Container>();
    return p;
  }

template<typename Container>
  void
  test_container(Container obj, bool run_threaded = false)
  {
    using namespace __gnu_test;
    int status;

    time_counter time;
    resource_counter resource;
    {
      start_counters(time, resource);
      if (!run_threaded)
	{
	  do_loop<Container>();
	}
      else
	{
#if defined (_GLIBCXX_GCC_GTHR_POSIX_H) && !defined (NOTHREAD)
	  pthread_t  t1, t2, t3, t4;
	  pthread_create(&t1, 0, &do_test<Container>, 0);
	  pthread_create(&t2, 0, &do_test<Container>, 0);
	  pthread_create(&t3, 0, &do_test<Container>, 0);
	  pthread_create(&t4, 0, &do_test<Container>, 0);
	  
	  pthread_join(t1, NULL);
	  pthread_join(t2, NULL);
	  pthread_join(t3, NULL);
	  pthread_join(t4, NULL);
#endif
	}
      stop_counters(time, resource);

      std::ostringstream comment;
      if (run_threaded)
	comment << "4-way threaded iterations: " << iterations*4 << '\t';
      else
	comment << "iterations: " << iterations << '\t';
      comment << "type: " << abi::__cxa_demangle(typeid(obj).name(),
						 0, 0, &status);
      report_header(__FILE__, comment.str());
      report_performance(__FILE__, string(), time, resource);
    }
  }

// http://gcc.gnu.org/ml/libstdc++/2001-05/msg00105.html
// http://gcc.gnu.org/ml/libstdc++/2003-05/msg00231.html
int main(void)
{
  typedef __gnu_cxx::malloc_allocator<test_type> m_alloc_type;
  typedef __gnu_cxx::new_allocator<test_type> n_alloc_type;
  typedef __gnu_cxx::__mt_alloc<test_type> so_alloc_type;
  typedef __gnu_cxx::bitmap_allocator<test_type> bit_alloc_type;
  typedef __gnu_cxx::__pool_alloc<test_type> po_alloc_type;

#ifdef TEST_B0
  test_container(vector<test_type, m_alloc_type>());
#endif
#ifdef TEST_B1
  test_container(vector<test_type, n_alloc_type>());
#endif
#ifdef TEST_B2
  test_container(vector<test_type, so_alloc_type>());
#endif
#ifdef TEST_B3
  test_container(vector<test_type, bit_alloc_type>());
#endif
#ifdef TEST_B4
  test_container(vector<test_type, po_alloc_type>());
#endif

#ifdef TEST_B5
  test_container(list<test_type, m_alloc_type>());
#endif
#ifdef TEST_B6
  test_container(list<test_type, n_alloc_type>());
#endif
#ifdef TEST_B7
  test_container(list<test_type, so_alloc_type>());
#endif
#ifdef TEST_B8
  test_container(list<test_type, bit_alloc_type>());
#endif
#ifdef TEST_B9
  test_container(list<test_type, po_alloc_type>());
#endif

#ifdef TEST_B10
  test_container(deque<test_type, m_alloc_type>());
#endif
#ifdef TEST_B11
  test_container(deque<test_type, n_alloc_type>());
#endif
#ifdef TEST_B12
  test_container(deque<test_type, so_alloc_type>());
#endif
#ifdef TEST_B13
  test_container(deque<test_type, bit_alloc_type>());
#endif
#ifdef TEST_B14
  test_container(deque<test_type, po_alloc_type>());
#endif

  typedef less<test_type> compare_type;
  typedef pair<const test_type, test_type> pair_type;
  typedef __gnu_cxx::malloc_allocator<pair_type> m_pair_alloc_type;
  typedef __gnu_cxx::new_allocator<pair_type> n_pair_alloc_type;
  typedef __gnu_cxx::__mt_alloc<pair_type> so_pair_alloc_type;
  typedef __gnu_cxx::bitmap_allocator<pair_type> bit_pair_alloc_type;
  typedef __gnu_cxx::__pool_alloc<pair_type> po_pair_alloc_type;

#ifdef TEST_B15
  test_container(map<test_type, test_type, compare_type,
		 m_pair_alloc_type>());
#endif
#ifdef TEST_B16
  test_container(map<test_type, test_type, compare_type,
		 n_pair_alloc_type>());
#endif
#ifdef TEST_B17
  test_container(map<test_type, test_type, compare_type,
		 so_pair_alloc_type>());
#endif
#ifdef TEST_B18
  test_container(map<test_type, test_type, compare_type,
		 bit_pair_alloc_type>());
#endif
#ifdef TEST_B19
  test_container(map<test_type, test_type, compare_type,
		 po_pair_alloc_type>());
#endif

#ifdef TEST_B20
  test_container(set<test_type, compare_type, m_alloc_type>());
#endif
#ifdef TEST_B21
  test_container(set<test_type, compare_type, n_alloc_type>());
#endif
#ifdef TEST_B22
  test_container(set<test_type, compare_type, so_alloc_type>());
#endif
#ifdef TEST_B23
  test_container(set<test_type, compare_type, bit_alloc_type>());
#endif
#ifdef TEST_B24
  test_container(set<test_type, compare_type, po_alloc_type>());
#endif

#ifdef TEST_T0
  test_container(vector<test_type, m_alloc_type>(), true);
#endif
#ifdef TEST_T1
  test_container(vector<test_type, n_alloc_type>(), true);
#endif
#ifdef TEST_T2
  test_container(vector<test_type, so_alloc_type>(), true);
#endif
#ifdef TEST_T3
  test_container(vector<test_type, bit_alloc_type>(), true);
#endif
#ifdef TEST_T4
  test_container(vector<test_type, po_alloc_type>(), true);
#endif

#ifdef TEST_T5
  test_container(list<test_type, m_alloc_type>(), true);
#endif
#ifdef TEST_T6
  test_container(list<test_type, n_alloc_type>(), true);
#endif
#ifdef TEST_T7
  test_container(list<test_type, so_alloc_type>(), true);
#endif
#ifdef TEST_T8
  test_container(list<test_type, bit_alloc_type>(), true);
#endif
#ifdef TEST_T9
  test_container(list<test_type, po_alloc_type>(), true);
#endif

#ifdef TEST_T10
  test_container(deque<test_type, m_alloc_type>(), true);
#endif
#ifdef TEST_T11
  test_container(deque<test_type, n_alloc_type>(), true);
#endif
#ifdef TEST_T12
  test_container(deque<test_type, so_alloc_type>(), true);
#endif
#ifdef TEST_T13
  test_container(deque<test_type, bit_alloc_type>(), true);
#endif
#ifdef TEST_T14
  test_container(deque<test_type, po_alloc_type>(), true);
#endif

#ifdef TEST_T15
  test_container(map<test_type, test_type, compare_type,
		 m_pair_alloc_type>(), true);
#endif
#ifdef TEST_T16
  test_container(map<test_type, test_type, compare_type,
		 n_pair_alloc_type>(), true);
#endif
#ifdef TEST_T17
  test_container(map<test_type, test_type, compare_type,
		 so_pair_alloc_type>(), true);
#endif
#ifdef TEST_T18
  test_container(map<test_type, test_type, compare_type,
		 bit_pair_alloc_type>(), true);
#endif
#ifdef TEST_T19
  test_container(map<test_type, test_type, compare_type,
		 po_pair_alloc_type>(), true);
#endif

#ifdef TEST_T20
  test_container(set<test_type, compare_type, m_alloc_type>(), true);
#endif
#ifdef TEST_T21
  test_container(set<test_type, compare_type, n_alloc_type>(), true);
#endif
#ifdef TEST_T22
  test_container(set<test_type, compare_type, so_alloc_type>(), true);
#endif
#ifdef TEST_T23
  test_container(set<test_type, compare_type, bit_alloc_type>(), true);
#endif
#ifdef TEST_T24
  test_container(set<test_type, compare_type, po_alloc_type>(), true);
#endif
  return 0;
}