//
// Copyright © 2017 Arm Ltd. All rights reserved.
// SPDX-License-Identifier: MIT
//

#include "Layer.hpp"
#include "SubGraph.hpp"

#include <boost/numeric/conversion/cast.hpp>

namespace armnn
{

SubGraph::SubGraph()
{
}

SubGraph::SubGraph(InputSlots && inputs,
                   OutputSlots && outputs,
                   Layers && layers)
: m_InputSlots{inputs}
, m_OutputSlots{outputs}
, m_Layers{layers}
{
}

const SubGraph::InputSlots & SubGraph::GetInputSlots() const
{
    return m_InputSlots;
}

const SubGraph::OutputSlots & SubGraph::GetOutputSlots() const
{
    return m_OutputSlots;
}

const InputSlot* SubGraph::GetInputSlot(unsigned int index) const
{
    return m_InputSlots.at(index);
}

InputSlot* SubGraph::GetInputSlot(unsigned int index)
{
    return  m_InputSlots.at(index);
}

const OutputSlot* SubGraph::GetOutputSlot(unsigned int index) const
{
    return m_OutputSlots.at(index);
}

OutputSlot* SubGraph::GetOutputSlot(unsigned int index)
{
    return m_OutputSlots.at(index);
}

unsigned int SubGraph::GetNumInputSlots() const
{
    return boost::numeric_cast<unsigned int>(m_InputSlots.size());
}

unsigned int SubGraph::GetNumOutputSlots() const
{
    return boost::numeric_cast<unsigned int>(m_OutputSlots.size());
}

const SubGraph::Layers & SubGraph::GetLayers() const
{
    return m_Layers;
}

} // namespace armnn