/**
 * Copyright - See the COPYRIGHT that is included with this distribution.
 * pvAccessCPP is distributed subject to a Software License Agreement found
 * in file LICENSE that is included with this distribution.
 */

#include <pv/transportRegistry.h>

using namespace epics::pvData;
using namespace std;

namespace epics {
namespace pvAccess {

TransportRegistry::TransportRegistry(): _transports(), _transportCount(0), _mutex()
{

}

TransportRegistry::~TransportRegistry()
{
}

void TransportRegistry::put(Transport::shared_pointer const & transport)
{
	Lock guard(_mutex);
	//const string type = transport.getType();
	const int16 priority = transport->getPriority();
	const osiSockAddr* address = transport->getRemoteAddress();

	transportsMap_t::iterator transportsIter = _transports.find(address);
	prioritiesMapSharedPtr_t priorities;
	if(transportsIter == _transports.end())
	{
		priorities.reset(new prioritiesMap_t());
		_transports[address] = priorities;
		_transportCount++;
	}
	else
	{
		priorities = transportsIter->second;
		prioritiesMap_t::iterator prioritiesIter = priorities->find(priority);
		if(prioritiesIter == priorities->end()) //only increase transportCount if not replacing
		{
			_transportCount++;
		}
	}
	(*priorities)[priority] = transport;
}

Transport::shared_pointer TransportRegistry::get(std::string const & /*type*/, const osiSockAddr* address, const int16 priority)
{
	Lock guard(_mutex);
	transportsMap_t::iterator transportsIter = _transports.find(address);
	if(transportsIter != _transports.end())
	{
		prioritiesMapSharedPtr_t priorities = transportsIter->second;
		prioritiesMap_t::iterator prioritiesIter = priorities->find(priority);
		if(prioritiesIter != priorities->end())
		{
			return prioritiesIter->second;
		}
	}
	return Transport::shared_pointer();
}

auto_ptr<TransportRegistry::transportVector_t> TransportRegistry::get(std::string const & /*type*/, const osiSockAddr* address)
{
	Lock guard(_mutex);
	transportsMap_t::iterator transportsIter = _transports.find(address);
	if(transportsIter != _transports.end())
	{
		prioritiesMapSharedPtr_t priorities = transportsIter->second;
        auto_ptr<transportVector_t> transportArray(new transportVector_t(priorities->size()));
        int32 i = 0;
		for(prioritiesMap_t::iterator prioritiesIter = priorities->begin();
            prioritiesIter != priorities->end();
            prioritiesIter++, i++)
		{
			(*transportArray)[i] = prioritiesIter->second;
		}
		return transportArray;
	}
	return auto_ptr<transportVector_t>();
}

Transport::shared_pointer TransportRegistry::remove(Transport::shared_pointer const & transport)
{
	Lock guard(_mutex);
	const int16 priority = transport->getPriority();
	const osiSockAddr* address = transport->getRemoteAddress();
    Transport::shared_pointer retTransport;
	transportsMap_t::iterator transportsIter = _transports.find(address);
	if(transportsIter != _transports.end())
	{
		prioritiesMapSharedPtr_t priorities = transportsIter->second;
		prioritiesMap_t::iterator prioritiesIter = priorities->find(priority);
		if(prioritiesIter != priorities->end())
		{
            retTransport = prioritiesIter->second;
			priorities->erase(prioritiesIter);
            _transportCount--;
			if(priorities->size() == 0)
			{
				_transports.erase(transportsIter);
			}
		}
	}
	return retTransport;
}

void TransportRegistry::clear()
{
	Lock guard(_mutex);
	_transports.clear();
	_transportCount = 0;
}

int32 TransportRegistry::numberOfActiveTransports()
{
	Lock guard(_mutex);
	return _transportCount;
}


auto_ptr<TransportRegistry::transportVector_t> TransportRegistry::toArray(std::string const & /*type*/)
{
	// TODO support type
    return toArray();
}


auto_ptr<TransportRegistry::transportVector_t> TransportRegistry::toArray()
{
	Lock guard(_mutex);
    if (_transportCount == 0)
        return auto_ptr<transportVector_t>(0);
    
    auto_ptr<transportVector_t> transportArray(new transportVector_t(_transportCount));

    int32 i = 0;
	for (transportsMap_t::iterator transportsIter = _transports.begin();
         transportsIter != _transports.end();
         transportsIter++)
    {
		prioritiesMapSharedPtr_t priorities = transportsIter->second;
		for (prioritiesMap_t::iterator prioritiesIter = priorities->begin();
             prioritiesIter != priorities->end();
             prioritiesIter++, i++)
        {
            (*transportArray)[i] = prioritiesIter->second;
        }
    }
    
	return transportArray;
}

void TransportRegistry::toArray(transportVector_t & transportArray)
{
	Lock guard(_mutex);
    if (_transportCount == 0)
        return;
    
    transportArray.reserve(transportArray.size() + _transportCount);

	for (transportsMap_t::iterator transportsIter = _transports.begin();
         transportsIter != _transports.end();
         transportsIter++)
    {
		prioritiesMapSharedPtr_t priorities = transportsIter->second;
		for (prioritiesMap_t::iterator prioritiesIter = priorities->begin();
             prioritiesIter != priorities->end();
             prioritiesIter++)
        {
            transportArray.push_back(prioritiesIter->second);
        }
    }
}

}}