/* -*-  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 "ns3/core-module.h"
#include "ns3/network-module.h"
#include "ns3/point-to-point-module.h"
#include "ns3/NDNabstraction-module.h"
#include <ns3/point-to-point-grid.h>
#include "ns3/ipv4-global-routing-helper.h"

#include <iostream>
#include <sstream>

#include "ns3/visualizer-module.h"
#include "ns3/ccnx.h"


using namespace ns3;

NS_LOG_COMPONENT_DEFINE ("SyncTopologyNDNabstraction");

int 
main (int argc, char *argv[])
{
  GlobalValue::Bind ("SimulatorImplementationType", StringValue
                       ("ns3::VisualSimulatorImpl"));
    
  // Set up some default values for the simulation.  Use the 
  
  Config::SetDefault ("ns3::OnOffApplication::PacketSize", UintegerValue (210));
  Config::SetDefault ("ns3::OnOffApplication::DataRate", StringValue ("448kb/s"));
  
    Packet::EnableChecking();
    Packet::EnablePrinting();

  // Allow the user to override any of the defaults and the above
  // DefaultValue::Bind ()s at run-time, via command-line arguments
  CommandLine cmd;
  cmd.Parse (argc, argv);
  
  // Here, we will explicitly create seven nodes. 
  NS_LOG_INFO ("Create nodes.");
  NodeContainer c;
  c.Create (7);
    Names::Add ("1", c.Get (0));
    Names::Add ("2", c.Get (1));
    Names::Add ("3", c.Get (2));
    Names::Add ("4", c.Get (3));
    Names::Add ("5", c.Get (4));
    Names::Add ("6", c.Get (5));
    Names::Add ("7", c.Get (6));
    
    
  NodeContainer n13 = NodeContainer (c.Get (0), c.Get (2));
  NodeContainer n23 = NodeContainer (c.Get (1), c.Get (2));
  NodeContainer n35 = NodeContainer (c.Get (2), c.Get (4));
  NodeContainer n34 = NodeContainer (c.Get (2), c.Get (3));
  NodeContainer n45 = NodeContainer (c.Get (3), c.Get (4));
  NodeContainer n56 = NodeContainer (c.Get (4), c.Get (5));
  NodeContainer n57 = NodeContainer (c.Get (4), c.Get (6));
  
  //Ipv4StaticRoutingHelper staticRouting;
  
  //Ipv4ListRoutingHelper list;
  //list.Add (staticRouting, 1);
  
  //Add static routing
  //InternetStackHelper internet;
  //internet.SetRoutingHelper (list); // has effect on the next Install ()
  //internet.Install (c);
  
  // We create the channels first without any IP addressing information
  NS_LOG_INFO ("Create channels.");
  PointToPointHelper p2p;
  p2p.SetDeviceAttribute ("DataRate", StringValue ("10Mbps"));
  p2p.SetChannelAttribute ("Delay", StringValue ("1ms"));
  NetDeviceContainer nd13 = p2p.Install (n13);
  NetDeviceContainer nd23 = p2p.Install (n23);
  NetDeviceContainer nd56 = p2p.Install (n56);
  
  p2p.SetDeviceAttribute ("DataRate", StringValue ("10Mbps"));
  p2p.SetChannelAttribute ("Delay", StringValue ("50ms"));
  NetDeviceContainer nd57 = p2p.Install (n57);
  
  p2p.SetDeviceAttribute ("DataRate", StringValue ("1Mbps"));
  p2p.SetChannelAttribute ("Delay", StringValue ("1ms"));
  NetDeviceContainer nd34 = p2p.Install (n34);
  NetDeviceContainer nd45 = p2p.Install (n45);
  
  p2p.SetDeviceAttribute ("DataRate", StringValue ("1Mbps"));
  p2p.SetChannelAttribute ("Delay", StringValue ("50ms"));
  NetDeviceContainer nd35 = p2p.Install (n35);
  
    InternetStackHelper stack;
    Ipv4GlobalRoutingHelper ipv4RoutingHelper;
    // Ptr<Ipv4RoutingHelper> ipv4RoutingHelper = stack.GetRoutingHelper ();
    stack.SetRoutingHelper (ipv4RoutingHelper);
    stack.Install(c);
    // // Create router nodes, initialize routing database and set up the routing
    // // tables in the nodes.
    Ipv4GlobalRoutingHelper::PopulateRoutingTables ();
    
  // Later, we add IP addresses.
  NS_LOG_INFO ("Assign IP Addresses.");
  Ipv4AddressHelper ipv4;
  ipv4.SetBase ("192.168.1.0", "255.255.255.0");
  Ipv4InterfaceContainer i13 = ipv4.Assign (nd13);
  Ipv4InterfaceContainer i23 = ipv4.Assign (nd23);
  Ipv4InterfaceContainer i35 = ipv4.Assign (nd35);
  Ipv4InterfaceContainer i34 = ipv4.Assign (nd34);
  Ipv4InterfaceContainer i45 = ipv4.Assign (nd45);
  Ipv4InterfaceContainer i56 = ipv4.Assign (nd56);
  Ipv4InterfaceContainer i57 = ipv4.Assign (nd57);
  
    
    CcnxStackHelper ccnx(Ccnx::NDN_FLOODING/*Ccnx::NDN_BESTROUTE*/);
    Ptr<CcnxFaceContainer> cf = ccnx.Install (c);
    
    NS_LOG_INFO ("Installing Applications");
    CcnxConsumerHelper helper ("/3");
    ApplicationContainer app = helper.Install (c.Get(1));
    app.Start (Seconds (1.0));
    app.Stop (Seconds (1000.05));
    
    /*CcnxConsumerHelper helper2 ("/4");
    ApplicationContainer app2 = helper2.Install(c.Get(5));
    app2.Start (Seconds (1.0));
    app2.Stop (Seconds (1000.05));
    */
    CcnxProducerHelper helper3 ("/3",120);
    ApplicationContainer app3 = helper3.Install(c.Get(6));
    app3.Start(Seconds(0.0));
    app3.Stop(Seconds(1500.0));
    /*
    CcnxProducerHelper helper4 ("/4",150);
    ApplicationContainer app4 = helper4.Install(c.Get(0));
    app4.Start(Seconds(0.0));
    app4.Stop(Seconds(1500.0));
  */
    
    ccnx.AddRoute("1","/3",0,1);
    ccnx.AddRoute("3","/3",2,1);
    ccnx.AddRoute("3","/3",3,1);
    ccnx.AddRoute("4","/3",1,1);
    ccnx.AddRoute("5","/3",2,1);
    
    /*ccnx.AddRoute ("1", "/3", 0, 1);
    ccnx.AddRoute ("1", "/3", 1, 1);
    
    ccnx.AddRoute ("2", "/3", 1, 1);
    
    ccnx.AddRoute ("3", "/3", 1, 1);
    
    ccnx.AddRoute ("4", "/3", 2, 1);
    
    ccnx.AddRoute ("6", "/3", 2, 1);
    
    ccnx.AddRoute ("7", "/3", 1, 1);
    
    ccnx.AddRoute ("8", "/3", 1, 1);
  */
    
  // Create the OnOff application to send UDP datagrams of size
  // 210 bytes at a rate of 448 Kb/s from n0 to n4
  /*NS_LOG_INFO ("Create Applications.");
  uint16_t port = 9;   // Discard port (RFC 863)
  
  std::string sendsizeattr = "SendSize";
  //flow2 7-->2
  BulkSendHelper bulksend0 ("ns3::UdpSocketFactory", InetSocketAddress (i23.GetAddress (0), port));
  //bulksend0.SetAttribute(sendsizeattr, AttributeValue(ConstantVariable(2560)));
  bulksend0.SetAttribute("MaxBytes", UintegerValue(2560));
  ApplicationContainer apps = bulksend0.Install(c.Get(6));
  apps.Start(Seconds (1.0));
  apps.Stop(Seconds (10.0));
  
  // Create a packet sink to receive these packets
  PacketSinkHelper sink0 ("ns3::UdpSocketFactory", InetSocketAddress(Ipv4Address::GetAny (), port));
  apps = sink0.Install(c.Get(1));
  apps.Start(Seconds(0.0));
  apps.Stop(Seconds(20.0));
  
  //flow1 1-->6
  BulkSendHelper bulksend ("ns3::UdpSocketFactory", InetSocketAddress (i56.GetAddress (1), port));
  //bulksend.SetAttribute(sendsizeattr, AttributeValue( ConstantVariable(2560)));
  bulksend0.SetAttribute("MaxBytes", UintegerValue(2560));
  apps = bulksend.Install (c.Get (0));
  apps.Start (Seconds (6.0));
  apps.Stop (Seconds (20.0));
  
  // Create a packet sink to receive these packets
  PacketSinkHelper sink ("ns3::UdpSocketFactory", InetSocketAddress (Ipv4Address::GetAny (), port));
  apps = sink.Install (c.Get (5));
  apps.Start(Seconds(0.0));
  apps.Stop(Seconds(20.0));
  
  AsciiTraceHelper ascii;
  p2p.EnableAsciiAll (ascii.CreateFileStream ("sync-topology-ndnabstraction.tr"));
  p2p.EnablePcapAll ("sync-topology-ndnabstraction");*/
  
  Simulator::Stop (Seconds (2000));
  
  NS_LOG_INFO ("Run Simulation.");
  Simulator::Run ();
  Simulator::Destroy ();
  NS_LOG_INFO ("Done.");
  
	return 0;
}