// -*- Mode: C++; c-file-style: "gnu"; indent-tabs-mode:nil; -*-

//

// Copyright (c) 2006 Georgia Tech Research Corporation

// All rights reserved.

//

// This program is free software; you can redistribute it and/or modify

// it under the terms of the GNU General Public License version 2 as

// published by the Free Software Foundation;

//

// This program 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 program; if not, write to the Free Software

// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

//

// Author: George F. Riley<riley@ece.gatech.edu>

//

// Implement the L3Protocols capability for ns3.

// George F. Riley, Georgia Tech, Fall 2006

#include <sstream>

#include <string>

#include "ns3/composite-trace-resolver.h"

#include "l3-demux.h"

#include "l3-protocol.h"

namespace ns3 {

const InterfaceId L3Demux::iid = MakeInterfaceId ("L3Demux", Object::iid);

L3Demux::L3Demux (Ptr<Node> node)

  : m_node (node)

{

  SetInterfaceId (L3Demux::iid);

}

L3Demux::~L3Demux()

{}

void

L3Demux::DoDispose (void)

{

  for (L3Map_t::iterator i = m_protocols.begin(); i != m_protocols.end(); ++i)

    {

      i->second->Dispose ();

      i->second = 0;

    }

  m_protocols.clear ();

  m_node = 0;

  Object::DoDispose ();

}

TraceResolver *

L3Demux::CreateTraceResolver (TraceContext const &context) const

{

  CompositeTraceResolver *resolver = new CompositeTraceResolver (context);

  for (L3Map_t::const_iterator i = m_protocols.begin(); i != m_protocols.end(); ++i)

    {

      std::string protValue;

      std::ostringstream oss (protValue);

      oss << i->second->GetProtocolNumber ();

      ProtocolTraceType context = i->second->GetProtocolNumber ();

      resolver->Add (protValue, 

                     MakeCallback (&L3Protocol::CreateTraceResolver, PeekPointer (i->second)),

                     context);

    }

  return resolver;

}

void L3Demux::Insert(Ptr<L3Protocol> p)

{

  m_protocols.insert(L3Map_t::value_type(p->GetProtocolNumber (), p));

}

Ptr<L3Protocol>

L3Demux::GetProtocol (int p)

{ // Look up a protocol by protocol number

  L3Map_t::iterator i = m_protocols.find(p);

  if (i == m_protocols.end()) 

    {

      return 0;

    }

  return i->second; // Return the protocol

}

} //namespace ns3