/* -*-  Mode: C++; c-file-style: "gnu"; indent-tabs-mode:nil; -*- */
/*
 * Copyright (c) 2011 University of California, Los Angeles
 *
 * 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: Ilya Moiseenko <iliamo@cs.ucla.edu>
 */

#include "annotated-topology-reader.h"

#include "ns3/nstime.h"
#include "ns3/log.h"
#include "ns3/assert.h"
#include "ns3/names.h"
#include "ns3/net-device-container.h"
#include "ns3/point-to-point-helper.h"
#include "ns3/point-to-point-net-device.h"
#include "ns3/internet-stack-helper.h"
#include "ns3/ipv4-address-helper.h"
#include "ns3/ipv4-global-routing-helper.h"
#include "ns3/drop-tail-queue.h"
#include "ns3/ipv4-interface.h"
#include "ns3/ipv4.h"
#include "ns3/string.h"
#include "ns3/pointer.h"
#include "ns3/uinteger.h"
#include "ns3/ipv4-address.h"

#include "ns3/constant-position-mobility-model.h"
#include "ns3/random-variable.h"

#include <boost/foreach.hpp>
#include <boost/lexical_cast.hpp>

using namespace std;

namespace ns3 
{    

NS_LOG_COMPONENT_DEFINE ("AnnotatedTopologyReader");
    
NS_OBJECT_ENSURE_REGISTERED (AnnotatedTopologyReader);
    
TypeId AnnotatedTopologyReader::GetTypeId (void)
{
  static TypeId tid = TypeId ("ns3::AnnotatedTopologyReader")
    .SetParent<Object> ()
    ;
  return tid;
}
    
AnnotatedTopologyReader::AnnotatedTopologyReader (const std::string &path)
  : m_path (path)
  , m_ulx (0)
  , m_uly (0)
  , m_lrx (100.0)
  , m_lry (100.0)
{
  NS_LOG_FUNCTION (this);
}

void
AnnotatedTopologyReader::SetBoundingBox (double ulx, double uly, double lrx, double lry)
{
  NS_LOG_FUNCTION (this << ulx << uly << lrx << lry);
  
  m_ulx = ulx;
  m_uly = uly;
  m_lrx = lrx;
  m_lry = lry;
}

AnnotatedTopologyReader::~AnnotatedTopologyReader ()
{
  NS_LOG_FUNCTION (this);
}
    
NodeContainer
AnnotatedTopologyReader::Read (void)
{
  ifstream topgen;
  topgen.open (GetFileName ().c_str ());
  NodeContainer nodes;
        
  if (!topgen.is_open ())
    {
      NS_LOG_ERROR ("Cannot open file " << GetFileName () << " for reading");
      return nodes;
    }

  int linksNumber = 0;
  int nodesNumber = 0;
        
  string line;
  getline (topgen,line);
  istringstream headerLineBuffer (line);
        
  int totnode;
  int totlink;
  headerLineBuffer >> totnode;
  headerLineBuffer >> totlink;
  
  NS_LOG_INFO ("Annotated topology should have " << totnode << " nodes and " << totlink << " links");

  for (int i = 0; i < totlink && !topgen.eof (); i++)
    {
      getline (topgen, line);
      istringstream lineBuffer (line);
            
      string from;
      string to;
      lineBuffer >> from;
      lineBuffer >> to;

      if ( (!from.empty ()) && (!to.empty ()) )
        {
          NS_LOG_INFO ( linksNumber << " From: " << from << " to: " << to );

          Ptr<Node> fromNode = Names::Find<Node> (m_path, from);
          Ptr<Node> toNode   = Names::Find<Node> (m_path, to);
          
          if (fromNode == 0)
            {
              fromNode = CreateObject<Node> ();
              Names::Add (m_path, from, fromNode);
              nodes.Add (fromNode);
              nodesNumber++;
            }
                
          if (toNode == 0)
            {
              toNode = CreateObject<Node> ();
              Names::Add (m_path, to, toNode);
              nodes.Add (toNode);
              nodesNumber++;
            }
                
          Link link (fromNode, from, toNode, to);
                
          string dataRate;
          lineBuffer >> dataRate;

          string ospf;
          lineBuffer >> ospf;

          string delay;
          lineBuffer >> delay;

          string queueSizeNode1;
          lineBuffer >> queueSizeNode1;
          
          string queueSizeNode2;
          lineBuffer >> queueSizeNode2;

          if (dataRate.empty () ||
              ospf.empty () ||
              delay.empty () ||
              queueSizeNode1.empty () ||
              queueSizeNode2.empty ())
            {
              NS_LOG_ERROR ("File [" << GetFileName () << ":" << i+2 << " wrong format, skipping");
              continue;
            }
          
          link.SetAttribute ("DataRate", dataRate);
          link.SetAttribute ("OSPF", ospf);
          link.SetAttribute ("Delay", delay);
          link.SetAttribute ("QueueSizeNode1", queueSizeNode1);
          link.SetAttribute ("QueueSizeNode2", queueSizeNode2);
                
          AddLink (link);
                
          linksNumber++;
        }
    }

  NS_ASSERT (nodesNumber == totnode && linksNumber == totlink);
        
  NS_LOG_INFO ("Annotated topology created with " << nodesNumber << " nodes and " << linksNumber << " links");
  topgen.close ();

  ApplySettings ();
  AssignCoordinates ();
  
  return nodes;
}

void
AnnotatedTopologyReader::AssignIpv4Addresses (Ipv4Address base)
{
  Ipv4AddressHelper address (base, Ipv4Mask ("/24"));

  BOOST_FOREACH (const Link &link, m_linksList)
    {
      address.Assign (NetDeviceContainer (link.GetFromNetDevice (),
                                          link.GetToNetDevice ()));
      
      base = Ipv4Address (base.Get () + 256);
      address.SetBase (base, Ipv4Mask ("/24"));
    }

  ApplyOspfMetric ();
}

void
AnnotatedTopologyReader::ApplyOspfMetric ()
{
  BOOST_FOREACH (const Link &link, m_linksList)
    {
      uint16_t metric = boost::lexical_cast<uint16_t> (link.GetAttribute ("OSPF"));
      
      {
        Ptr<Ipv4> ipv4 = link.GetFromNode ()->GetObject<Ipv4> ();
        NS_ASSERT (ipv4 != 0);

        int32_t interfaceId = ipv4->GetInterfaceForDevice (link.GetFromNetDevice ());
        NS_ASSERT (interfaceId >= 0);

        ipv4->SetMetric (interfaceId,metric);
      }

      {
        Ptr<Ipv4> ipv4 = link.GetToNode ()->GetObject<Ipv4> ();
        NS_ASSERT (ipv4 != 0);
        
        int32_t interfaceId = ipv4->GetInterfaceForDevice (link.GetToNetDevice ());
        NS_ASSERT (interfaceId >= 0);

        ipv4->SetMetric (interfaceId,metric);
      }
    }
}

void
AnnotatedTopologyReader::ApplySettings ()
{
  PointToPointHelper p2p;

  // temporary queue, will be changed later
  p2p.SetQueue ("ns3::DropTailQueue",
                "MaxPackets", StringValue("100"));
  
  BOOST_FOREACH (Link &link, m_linksList)
    {
      NS_LOG_INFO ("DataRate = " + link.GetAttribute("DataRate")+"Kbps");
      p2p.SetDeviceAttribute ("DataRate", StringValue(link.GetAttribute("DataRate")+"Kbps"));

      NS_LOG_INFO ("Delay = " + link.GetAttribute("Delay")+"ms");
      p2p.SetChannelAttribute ("Delay", StringValue(link.GetAttribute("Delay")+"ms"));
      
      NetDeviceContainer nd = p2p.Install(link.GetFromNode (), link.GetToNode ());
      link.SetNetDevices (nd.Get (0), nd.Get (1));

      NS_LOG_INFO ("Queue: " << link.GetAttribute("QueueSizeNode1") << " <==> " << link.GetAttribute("QueueSizeNode2"));
      
      PointerValue txQueueFrom;
      link.GetFromNetDevice ()->GetAttribute ("TxQueue", txQueueFrom);
      NS_ASSERT (txQueueFrom.Get<DropTailQueue> () != 0);

      PointerValue txQueueTo;
      link.GetToNetDevice ()->GetAttribute ("TxQueue", txQueueTo);
      NS_ASSERT (txQueueTo.Get<DropTailQueue> () != 0);
      
      txQueueFrom.Get<DropTailQueue> ()->SetAttribute ("MaxPackets", StringValue (link.GetAttribute("QueueSizeNode1")));
      txQueueTo.  Get<DropTailQueue> ()->SetAttribute ("MaxPackets", StringValue (link.GetAttribute("QueueSizeNode2")));
    }
}

void
AnnotatedTopologyReader::AssignCoordinates ()
{
  UniformVariable randX (m_ulx, m_lrx);
  double x = 0.0;
  UniformVariable randY (m_uly, m_lry);
  double y = 0.0;

  BOOST_FOREACH (Link &link, m_linksList)
    {
      Ptr<ConstantPositionMobilityModel> loc = link.GetFromNode ()->GetObject<ConstantPositionMobilityModel> ();
      if (loc != 0)
        continue; // no need to assign twice

      loc = CreateObject<ConstantPositionMobilityModel> ();
      link.GetFromNode ()->AggregateObject (loc);
        
      x = randX.GetValue();
      y = randY.GetValue();
      NS_LOG_INFO("X = "<<x <<"Y = "<<y);

      loc->SetPosition (Vector (x, y, 0));
    }
}

}