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

/*

 * Copyright (c) 2010 Network Security Lab, University of Washington, Seattle.

 *

 * 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: Sidharth Nabar <snabar@uw.edu>, He Wu <mdzz@u.washington.edu>

 */

#include "ns3/core-module.h"

#include "ns3/helper-module.h"

#include "ns3/mobility-module.h"

#include "ns3/contrib-module.h"

#include "ns3/wifi-module.h"

#include "ns3/energy-module.h"

#include <iostream>

#include <fstream>

#include <vector>

#include <string>

NS_LOG_COMPONENT_DEFINE ("EnergyExample");

using namespace ns3;

/**

 * \param socket Pointer to socket.

 *

 * Packet receiving sink.

 */

void

ReceivePacket (Ptr<Socket> socket)

{

  Ptr<Packet> packet;

  Address from;

  while (packet = socket->RecvFrom (from))

    {

      if (packet->GetSize () > 0)

        {

          InetSocketAddress iaddr = InetSocketAddress::ConvertFrom (from);

          NS_LOG_UNCOND ("--\nReceived one packet! Socket: "<< iaddr.GetIpv4 ()

                         << " port: " << iaddr.GetPort () << " at time = " <<

                         Simulator::Now ().GetSeconds () << "\n--");

        }

    }

}

/**

 * \param socket Pointer to socket.

 * \param pktSize Packet size.

 * \param n Pointer to node.

 * \param pktCount Number of packets to generate.

 * \param pktInterval Packet sending interval.

 *

 * Traffic generator.

 */

static void

GenerateTraffic (Ptr<Socket> socket, uint32_t pktSize, Ptr<Node> n,

    uint32_t pktCount, Time pktInterval)

{

  if (pktCount > 0)

    {

      socket->Send (Create<Packet> (pktSize));

      Simulator::Schedule (pktInterval, &GenerateTraffic, socket, pktSize, n,

          pktCount - 1, pktInterval);

    }

  else

    {

      socket->Close ();

    }

}

/// Trace function for remaining energy at node.

void

RemainingEnergy (double oldValue, double remainingEnergy)

{

  NS_LOG_UNCOND (Simulator::Now ().GetSeconds ()

      << "s Current remaining energy = " << remainingEnergy << "J");

}

/// Trace function for total energy consumption at node.

void

TotalEnergy (double oldValue, double totalEnergy)

{

  NS_LOG_UNCOND (Simulator::Now ().GetSeconds ()

      << "s Total energy consumed by radio = " << totalEnergy << "J");

}

int

main (int argc, char *argv[])

{

  /*

  LogComponentEnable ("EnergySource", LOG_LEVEL_DEBUG);

  LogComponentEnable ("BasicEnergySource", LOG_LEVEL_DEBUG);

  LogComponentEnable ("DeviceEnergyModel", LOG_LEVEL_DEBUG);

  LogComponentEnable ("WifiRadioEnergyModel", LOG_LEVEL_DEBUG);

   */

  std::string phyMode ("DsssRate1Mbps");

  double Prss = -80;            // dBm

  uint32_t PpacketSize = 200;   // bytes

  bool verbose = false;

  // simulation parameters

  uint32_t numPackets = 10000;  // number of packets to send

  double interval = 1;          // seconds

  double startTime = 0.0;       // seconds

  double distanceToRx = 100.0;  // meters

  /*

   * This is a magic number used to set the transmit power, based on other

   * configuration.

   */

  double offset = 81;

  CommandLine cmd;

  cmd.AddValue ("phyMode", "Wifi Phy mode", phyMode);

  cmd.AddValue ("Prss", "Intended primary RSS (dBm)", Prss);

  cmd.AddValue ("PpacketSize", "size of application packet sent", PpacketSize);

  cmd.AddValue ("numPackets", "Total number of packets to send", numPackets);

  cmd.AddValue ("startTime", "Simulation start time", startTime);

  cmd.AddValue ("distanceToRx", "X-Axis distance between nodes", distanceToRx);

  cmd.AddValue ("verbose", "Turn on all device log components", verbose);

  cmd.Parse (argc, argv);

  // Convert to time object

  Time interPacketInterval = Seconds (interval);

  // disable fragmentation for frames below 2200 bytes

  Config::SetDefault ("ns3::WifiRemoteStationManager::FragmentationThreshold",

                      StringValue ("2200"));

  // turn off RTS/CTS for frames below 2200 bytes

  Config::SetDefault ("ns3::WifiRemoteStationManager::RtsCtsThreshold",

                      StringValue ("2200"));

  // Fix non-unicast data rate to be the same as that of unicast

  Config::SetDefault ("ns3::WifiRemoteStationManager::NonUnicastMode",

                      StringValue (phyMode));

  NodeContainer c;

  c.Create (2);     // create 2 nodes

  NodeContainer networkNodes;

  networkNodes.Add (c.Get (0));

  networkNodes.Add (c.Get (1));

  // The below set of helpers will help us to put together the wifi NICs we want

  WifiHelper wifi;

  if (verbose)

    {

      wifi.EnableLogComponents ();

    }

  wifi.SetStandard (WIFI_PHY_STANDARD_80211b);

  /** Wifi PHY **/

  /***************************************************************************/

  YansWifiPhyHelper wifiPhy = YansWifiPhyHelper::Default ();

  wifiPhy.Set ("RxGain", DoubleValue (-10));

  wifiPhy.Set ("TxGain", DoubleValue (offset + Prss));

  wifiPhy.Set ("CcaMode1Threshold", DoubleValue (0.0));

  /***************************************************************************/

  /** wifi channel **/

  YansWifiChannelHelper wifiChannel;

  wifiChannel.SetPropagationDelay ("ns3::ConstantSpeedPropagationDelayModel");

  wifiChannel.AddPropagationLoss ("ns3::FriisPropagationLossModel");

  // create wifi channel

  Ptr<YansWifiChannel> wifiChannelPtr = wifiChannel.Create ();

  wifiPhy.SetChannel (wifiChannelPtr);

  /** MAC layer **/

  // Add a non-QoS upper MAC, and disable rate control

  NqosWifiMacHelper wifiMac = NqosWifiMacHelper::Default ();

  wifi.SetRemoteStationManager ("ns3::ConstantRateWifiManager", "DataMode",

                                StringValue (phyMode), "ControlMode",

                                StringValue (phyMode));

  // Set it to ad-hoc mode

  wifiMac.SetType ("ns3::AdhocWifiMac");

  /** install PHY + MAC **/

  NetDeviceContainer devices = wifi.Install (wifiPhy, wifiMac, networkNodes);

  /** mobility **/

  MobilityHelper mobility;

  Ptr<ListPositionAllocator> positionAlloc = CreateObject<ListPositionAllocator> ();

  positionAlloc->Add (Vector (0.0, 0.0, 0.0));

  positionAlloc->Add (Vector (2 * distanceToRx, 0.0, 0.0));

  mobility.SetPositionAllocator (positionAlloc);

  mobility.SetMobilityModel ("ns3::ConstantPositionMobilityModel");

  mobility.Install (c);

  /** Energy Model **/

  /***************************************************************************/

  /* energy source */

  BasicEnergySourceHelper basicSourceHelper;

  // configure energy source

  basicSourceHelper.Set ("BasicEnergySourceInitialEnergyJ", DoubleValue (0.1));

  // install source

  EnergySourceContainer sources = basicSourceHelper.Install (c);

  /* device energy model */

  WifiRadioEnergyModelHelper radioEnergyHelper;

  // configure radio energy model

  radioEnergyHelper.Set ("TxCurrentA", DoubleValue (0.0174));

  // install device model

  DeviceEnergyModelContainer deviceModels = radioEnergyHelper.Install (devices, sources);

  /***************************************************************************/

  /** Internet stack **/

  InternetStackHelper internet;

  internet.Install (networkNodes);

  Ipv4AddressHelper ipv4;

  NS_LOG_INFO ("Assign IP Addresses.");

  ipv4.SetBase ("10.1.1.0", "255.255.255.0");

  Ipv4InterfaceContainer i = ipv4.Assign (devices);

  TypeId tid = TypeId::LookupByName ("ns3::UdpSocketFactory");

  Ptr<Socket> recvSink = Socket::CreateSocket (networkNodes.Get (1), tid);  // node 1, receiver

  InetSocketAddress local = InetSocketAddress (Ipv4Address::GetAny (), 80);

  recvSink->Bind (local);

  recvSink->SetRecvCallback (MakeCallback (&ReceivePacket));

  Ptr<Socket> source = Socket::CreateSocket (networkNodes.Get (0), tid);    // node 0, sender

  InetSocketAddress remote = InetSocketAddress (Ipv4Address::GetBroadcast (), 80);

  source->SetAllowBroadcast (true);

  source->Connect (remote);

  /** connect trace sources **/

  /***************************************************************************/

  // all sources are connected to node 1

  // energy source

  Ptr<BasicEnergySource> basicSourcePtr = DynamicCast<BasicEnergySource> (sources.Get (1));

  basicSourcePtr->TraceConnectWithoutContext ("RemainingEnergy", MakeCallback (&RemainingEnergy));

  // device energy model

  Ptr<DeviceEnergyModel> basicRadioModelPtr =

    basicSourcePtr->FindDeviceEnergyModels ("ns3::WifiRadioEnergyModel").Get (0);

  NS_ASSERT (basicRadioModelPtr != NULL);

  basicRadioModelPtr->TraceConnectWithoutContext ("TotalEnergyConsumption", MakeCallback (&TotalEnergy));

  /***************************************************************************/

  /** simulation setup **/

  // start traffic

  Simulator::Schedule (Seconds (startTime), &GenerateTraffic, source, PpacketSize,

                       networkNodes.Get (0), numPackets, interPacketInterval);

  Simulator::Stop (Seconds (10.0));

  Simulator::Run ();

  Simulator::Destroy ();

  return 0;

}