//
//  HashSetTest.cpp
//
//  Created by Setsu on 5/2/10.
//  Copyright 2010 RoundSquare Inc. All rights reserved.
//
// Permission is hereby granted, free of charge, to any person or organization
// obtaining a copy of the software and accompanying documentation covered by
// this license (the "Software") to use, reproduce, display, distribute,
// execute, and transmit the Software, and to prepare derivative works of the
// Software, and to permit third-parties to whom the Software is furnished to
// do so, all subject to the following:
// 
// The copyright notices in the Software and this entire statement, including
// the above license grant, this restriction and the following disclaimer,
// must be included in all copies of the Software, in whole or in part, and
// all derivative works of the Software, unless such copies or derivative
// works are solely in the form of machine-executable object code generated by
// a source language processor.
// 
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE, TITLE AND NON-INFRINGEMENT. IN NO EVENT
// SHALL THE COPYRIGHT HOLDERS OR ANYONE DISTRIBUTING THE SOFTWARE BE LIABLE
// FOR ANY DAMAGES OR OTHER LIABILITY, WHETHER IN CONTRACT, TORT OR OTHERWISE,
// ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
// DEALINGS IN THE SOFTWARE.
//

//----------------------------------------
//	include
//----------------------------------------
#include <Poco/Logger.h>
#include <Poco/PatternFormatter.h>
#include <Poco/FormattingChannel.h>
#include <Poco/ConsoleChannel.h>
#include <Poco/Format.h>
#include <Poco/Random.h>
#include <Poco/HashSet.h>

#include <string>
#include <vector>
#include <set>
#include <iostream>

#include "ScopedElapsedTime.h"

//----------------------------------------
//	enum
//----------------------------------------
enum MethodSel
{
	eInsert = 0
,	eFind
,	eErase
,	eNumMethod
};

//----------------------------------------
//	TestInsert
//----------------------------------------
template <class Tc, class Tk>
void TestInsert(Tc& container, const Tk& key)
{
	container.insert(key);
}

//----------------------------------------
//	TestFind
//----------------------------------------
template <class Tc, class Tk>
void TestFind(Tc& container, const Tk& key)
{
	container.find(key);
}

//----------------------------------------
//	TestErase
//----------------------------------------
template <class Tc, class Tk>
void TestErase(Tc& container, const Tk& key)
{
	container.erase(key);
}

//----------------------------------------
//	TestMethod
//----------------------------------------
template <class Tc, class Tk>
void TestMethod(	Tc& container
				,	MethodSel sel
				,	const std::string& title
				,	const std::vector<Tk>& keyVector
				,	bool needsNewLine=false)
{
	if(eNumMethod <= sel)	return;

	typedef void (*CallMethod)(Tc& container, const Tk& key);

	CallMethod methodVector[] = {	TestInsert<Tc, Tk>
								,	TestFind<Tc, Tk>
								,	TestErase<Tc, Tk>	};

	CallMethod method = methodVector[sel];

	{
		ScopedElapsedTime msg(title, "start", "end" + std::string(needsNewLine ? "\n":""));

		for(std::size_t i=0; i<keyVector.size(); ++i)
		{
			method(container, keyVector[i]);
		}
	}
}

//----------------------------------------
//	TestAll
//----------------------------------------
template <class Tk>
void TestAll(std::vector<Tk>& keyVector)
{
	std::set<Tk> stdSet;
	Poco::HashSet<Tk> pocoHashSet;

	TestMethod(stdSet,		eInsert,	"std::set insert ",	keyVector);
	TestMethod(pocoHashSet,	eInsert,	"HashSet  insert ",	keyVector, true);

	TestMethod(stdSet,		eFind,		"std::set find ",	keyVector);
	TestMethod(pocoHashSet,	eFind,		"HashSet  find ",	keyVector, true);

	TestMethod(stdSet,		eErase,		"std::set erase ",	keyVector);
	TestMethod(pocoHashSet,	eErase,		"HashSet  erase ",	keyVector, true);
}

//----------------------------------------
//	PrepareConsoleLogger
//----------------------------------------
void PrepareConsoleLogger(const std::string& name, int level=Poco::Message::PRIO_INFORMATION)
{
	Poco::FormattingChannel* pFCConsole = new Poco::FormattingChannel(new Poco::PatternFormatter("%t"));
	pFCConsole->setChannel(new Poco::ConsoleChannel);
	pFCConsole->open();

	Poco::Logger::create(name, pFCConsole, level);
}

//----------------------------------------
//	main
//----------------------------------------
int main(int /*argc*/, char** /*argv*/)
{
	PrepareConsoleLogger(Poco::Logger::ROOT, Poco::Message::PRIO_INFORMATION);

	const std::string::size_type kNumKeys = 262144;

	std::vector<Poco::UInt32>	intVector(kNumKeys);
	std::vector<std::string>	strVector(kNumKeys);

	Poco::Random random;

	for(std::size_t i=0; i<kNumKeys; ++i)
	{
		intVector[i] = random.next();
		strVector[i] = Poco::format("%08x", intVector[i]);
	}

	std::cout << "------------------------------------" << std::endl;
	std::cout << "Comparison for key type Poco::UInt32" << std::endl;
	std::cout << "------------------------------------" << std::endl;
	TestAll(intVector);

	std::cout << "----------------------------------------------" << std::endl;
	std::cout << "Comparison for key type std::string (length=8)" << std::endl;
	std::cout << "----------------------------------------------" << std::endl;
	TestAll(strVector);

	return 0;
}