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

/*

 * Copyright (c) 2007 University of Washington

 * 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

 */

#include <string>

#include "ns3/log.h"

#include "ns3/assert.h"

#include "ns3/packet.h"

#include "ns3/drop-tail.h"

#include "ns3/layer-connector.h"

#include "ns3/channel.h"

using namespace ns3;

NS_LOG_COMPONENT_DEFINE ("ChannelSample");

// ===========================================================================

// Cook up a simplistic Internet Node

// ===========================================================================

class FakeInternetNode : public LayerConnectorUpper

{

public:

  FakeInternetNode ();

 ~FakeInternetNode ();

  void Doit (void);

protected:

  bool UpperDoSendUp (Packet &p);

  bool UpperDoPull (Packet &p);

  DropTailQueue m_dtqOutbound;

  DropTailQueue m_dtqInbound;

};

FakeInternetNode::FakeInternetNode ()

{

  NS_LOG_FUNCTION;

}

FakeInternetNode::~FakeInternetNode ()

{

  NS_LOG_FUNCTION;

}

  void

FakeInternetNode::Doit (void)

{

  NS_LOG_FUNCTION;

  NS_LOG_INFO ("**** Send outbound packet");

  Packet p;

  m_dtqOutbound.Enqueue(p);

  UpperNotify();

}

  bool

FakeInternetNode::UpperDoSendUp (Packet &p)

{

  NS_LOG_FUNCTION;

  NS_LOG_PARAM ("(" << &p << ")");

  NS_LOG_INFO ("**** Receive inbound packet");

  m_dtqInbound.Enqueue(p);

  return m_dtqInbound.Dequeue(p);

}

  bool

FakeInternetNode::UpperDoPull (Packet &p)

{

  NS_LOG_FUNCTION;

  NS_LOG_PARAM ("(" << &p << ")");

  return m_dtqOutbound.Dequeue(p);

}

// ===========================================================================

// Cook up a simplistic Physical Layer

// ===========================================================================

class FakePhysicalLayer :

  public LayerConnectorLower,

  public LayerConnectorUpper

{

public:

  FakePhysicalLayer ();

 ~FakePhysicalLayer ();

protected:

  bool LowerDoNotify (LayerConnectorUpper *upper);

  bool UpperDoSendUp (Packet &p);

  bool UpperDoPull (Packet &p);

  DropTailQueue m_dtqInbound;

  DropTailQueue m_dtqOutbound;

};

FakePhysicalLayer::FakePhysicalLayer ()

{

  NS_LOG_FUNCTION;

}

FakePhysicalLayer::~FakePhysicalLayer ()

{

  NS_LOG_FUNCTION;

}

  bool

FakePhysicalLayer::LowerDoNotify (LayerConnectorUpper *upper)

{

  NS_LOG_FUNCTION;

  Packet p;

  NS_LOG_LOGIC ("Starting pull");

  NS_ASSERT(m_upperPartner);

  m_upperPartner->UpperPull(p);

  m_dtqOutbound.Enqueue(p);

  NS_LOG_LOGIC ("Got bits,  Notify lower");

  NS_ASSERT(m_lowerPartner);

  return m_lowerPartner->LowerNotify(this);

}

  bool

FakePhysicalLayer::UpperDoSendUp (Packet &p)

{

  NS_LOG_FUNCTION;

  NS_LOG_PARAM ("(" << &p << ")");

  NS_ASSERT(m_upperPartner);

  return m_upperPartner->UpperSendUp(p);

}

  bool

FakePhysicalLayer::UpperDoPull (Packet &p)

{

  NS_LOG_FUNCTION;

  NS_LOG_PARAM ("(" << &p << ")");

  return m_dtqOutbound.Dequeue(p);

}

// ===========================================================================

// Cook up a simplistic Channel, just to add any moby hack we feel like

// ===========================================================================

class FakeChannel : public Channel

{

public:

  FakeChannel ();

 ~FakeChannel ();

};

FakeChannel::FakeChannel ()

{

  NS_LOG_FUNCTION;

}

FakeChannel::~FakeChannel ()

{

  NS_LOG_FUNCTION;

}

int main (int argc, char *argv[])

{

  NS_LOG_INFO ("Channel Hackorama");

  FakeInternetNode      node1, node2, node3, node4;

  FakePhysicalLayer     phys1, phys2, phys3, phys4;

  FakeChannel           channel;

  Packet                pkt;

  // all the hassle above lets us do something very simple here

  node1.ConnectToLower(phys1);

  phys1.ConnectToUpper(node1);

  phys1.ConnectToLower(channel);

  channel.ConnectToUpper(phys1);

  node2.ConnectToLower(phys2);

  phys2.ConnectToUpper(node2);

  phys2.ConnectToLower(channel);

  channel.ConnectToUpper(phys2);

  node3.ConnectToLower(phys3);

  phys3.ConnectToUpper(node3);

  phys3.ConnectToLower(channel);

  channel.ConnectToUpper(phys3);

  node4.ConnectToLower(phys4);

  phys4.ConnectToUpper(node4);

  phys4.ConnectToLower(channel);

  channel.ConnectToUpper(phys4);

  node1.Doit();

  node2.Doit();

  node3.Doit();

  node4.Doit();

  return 0;

}