/**
 * 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.
 */

#define epicsExportSharedSymbols
#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);
        }
    }
}

}
}