blob: 30bdb152a2455fa093f4541d9acd66cdbd781338 [file] [log] [blame]
Alexander Afanasyevb7ad2322012-01-17 22:54:49 -08001/* -*- Mode: C++; c-file-style: "gnu"; indent-tabs-mode:nil; -*- */
2/*
3 * Copyright (c) 2011 University of California, Los Angeles
4 *
5 * This program is free software; you can redistribute it and/or modify
6 * it under the terms of the GNU General Public License version 2 as
7 * published by the Free Software Foundation;
8 *
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
13 *
14 * You should have received a copy of the GNU General Public License
15 * along with this program; if not, write to the Free Software
16 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
17 *
18 * Author: Alexander Afanasyev <alexander.afanasyev@ucla.edu>
19 */
20
21#include "ccnx-consumer-batches.h"
22#include "ns3/ptr.h"
23#include "ns3/log.h"
24#include "ns3/simulator.h"
25#include "ns3/packet.h"
26#include "ns3/callback.h"
27#include "ns3/string.h"
28#include "ns3/uinteger.h"
29#include "ns3/double.h"
30#include "ns3/batches.h"
31
32NS_LOG_COMPONENT_DEFINE ("CcnxConsumerBatches");
33
34namespace ns3
35{
36
37NS_OBJECT_ENSURE_REGISTERED (CcnxConsumerBatches);
38
39TypeId
40CcnxConsumerBatches::GetTypeId (void)
41{
42 static TypeId tid = TypeId ("ns3::CcnxConsumerBatches")
Alexander Afanasyev6315ef72012-06-01 20:56:31 -070043 .SetGroupName ("Ccnx")
Alexander Afanasyevb7ad2322012-01-17 22:54:49 -080044 .SetParent<CcnxConsumer> ()
45 .AddConstructor<CcnxConsumerBatches> ()
46
47 .AddAttribute ("Batches", "Batches to schedule. Should be vector, containing pairs of time and amount",
48 // TypeId::ATTR_SET,
49 StringValue (""),
50 MakeBatchesAccessor (&CcnxConsumerBatches::GetBatch, &CcnxConsumerBatches::SetBatch),
51 MakeBatchesChecker ())
52 ;
53
54 return tid;
55}
56
57CcnxConsumerBatches::CcnxConsumerBatches ()
58{
59}
60
61void
62CcnxConsumerBatches::SetBatch (const Batches &batches)
63{
64 // std::cout << "Batches: " << batches << "\n";
65 for (Batches::const_iterator i = batches.begin (); i != batches.end (); i++)
66 {
67 Simulator::Schedule (i->get<0> (), &CcnxConsumerBatches::AddBatch, this, i->get<1> ());
68 }
69}
70
71void
72CcnxConsumerBatches::AddBatch (uint32_t amount)
73{
74 // std::cout << Simulator::Now () << " adding batch of " << amount << "\n";
75 m_seqMax += amount;
Alexander Afanasyev8e0d2812012-01-19 22:38:14 -080076 m_rtt->ClearSent (); // this is important, otherwise RTT estimation for the new batch will be affected by previous batch history
Alexander Afanasyevb7ad2322012-01-17 22:54:49 -080077 ScheduleNextPacket ();
78}
79
80void
81CcnxConsumerBatches::ScheduleNextPacket ()
82{
83 if (!m_sendEvent.IsRunning ())
Alexander Afanasyev1e9348f2012-01-22 17:15:54 -080084 m_sendEvent = Simulator::Schedule (Seconds (m_rtt->RetransmitTimeout ().ToDouble (Time::S) * 0.1), &CcnxConsumer::SendPacket, this);
Alexander Afanasyevb7ad2322012-01-17 22:54:49 -080085}
86
87///////////////////////////////////////////////////
88// Process incoming packets //
89///////////////////////////////////////////////////
90
91} // namespace ns3