blob: b7406ae06ec58dcbbae895dedf9af0cb8ad2b1c9 [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
Alexander Afanasyev0c395372014-12-20 15:54:02 -080021#include "ndn-consumer-batches.hpp"
Alexander Afanasyevb7ad2322012-01-17 22:54:49 -080022#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"
Alexander Afanasyev669cafd2012-11-06 10:17:33 -080030
Alexander Afanasyev0c395372014-12-20 15:54:02 -080031#include "../utils/batches.hpp"
Alexander Afanasyevb7ad2322012-01-17 22:54:49 -080032
Alexander Afanasyevbe55cf62014-12-20 17:51:09 -080033NS_LOG_COMPONENT_DEFINE("ndn.ConsumerBatches");
Alexander Afanasyevb7ad2322012-01-17 22:54:49 -080034
Alexander Afanasyev2b4c9472012-08-09 15:00:38 -070035namespace ns3 {
36namespace ndn {
Alexander Afanasyevb7ad2322012-01-17 22:54:49 -080037
Alexander Afanasyevbe55cf62014-12-20 17:51:09 -080038NS_OBJECT_ENSURE_REGISTERED(ConsumerBatches);
39
40TypeId
41ConsumerBatches::GetTypeId(void)
42{
43 static TypeId tid =
44 TypeId("ns3::ndn::ConsumerBatches")
45 .SetGroupName("Ndn")
46 .SetParent<Consumer>()
47 .AddConstructor<ConsumerBatches>()
48
49 .AddAttribute("Batches",
50 "Batches to schedule. Should be vector, containing pairs of time and amount",
51 // TypeId::ATTR_SET,
52 StringValue(""), MakeBatchesAccessor(&ConsumerBatches::m_batches),
53 MakeBatchesChecker());
Alexander Afanasyevb7ad2322012-01-17 22:54:49 -080054
55 return tid;
56}
57
Alexander Afanasyevbe55cf62014-12-20 17:51:09 -080058ConsumerBatches::ConsumerBatches()
59 : m_initial(true)
Alexander Afanasyevb7ad2322012-01-17 22:54:49 -080060{
61}
62
63void
Alexander Afanasyevbe55cf62014-12-20 17:51:09 -080064ConsumerBatches::StartApplication()
Alexander Afanasyevb7ad2322012-01-17 22:54:49 -080065{
Alexander Afanasyevbe55cf62014-12-20 17:51:09 -080066 Consumer::StartApplication();
67
Alexander Afanasyevb7ad2322012-01-17 22:54:49 -080068 // std::cout << "Batches: " << batches << "\n";
Alexander Afanasyevbe55cf62014-12-20 17:51:09 -080069 for (Batches::const_iterator i = m_batches.begin(); i != m_batches.end(); i++) {
70 Simulator::ScheduleWithContext(GetNode()->GetId(), i->get<0>(), &ConsumerBatches::AddBatch,
71 this, i->get<1>());
72 }
Alexander Afanasyevb7ad2322012-01-17 22:54:49 -080073}
74
75void
Alexander Afanasyevbe55cf62014-12-20 17:51:09 -080076ConsumerBatches::AddBatch(uint32_t amount)
Alexander Afanasyevb7ad2322012-01-17 22:54:49 -080077{
78 // std::cout << Simulator::Now () << " adding batch of " << amount << "\n";
79 m_seqMax += amount;
Alexander Afanasyevbe55cf62014-12-20 17:51:09 -080080 m_rtt->ClearSent(); // this is important, otherwise RTT estimation for the new batch will be
81 // affected by previous batch history
Alexander Afanasyev3476edf2012-08-14 11:26:00 -070082 m_initial = true;
Alexander Afanasyevbe55cf62014-12-20 17:51:09 -080083 ScheduleNextPacket();
Alexander Afanasyevb7ad2322012-01-17 22:54:49 -080084}
85
86void
Alexander Afanasyevbe55cf62014-12-20 17:51:09 -080087ConsumerBatches::ScheduleNextPacket()
Alexander Afanasyevb7ad2322012-01-17 22:54:49 -080088{
Alexander Afanasyevbe55cf62014-12-20 17:51:09 -080089 if (!m_sendEvent.IsRunning()) {
90 Time delay = Seconds(0);
91 if (!m_initial)
92 delay = m_rtt->RetransmitTimeout();
93
94 m_initial = false;
95 m_sendEvent = Simulator::Schedule(delay, &Consumer::SendPacket, this);
96 }
Alexander Afanasyevb7ad2322012-01-17 22:54:49 -080097}
98
99///////////////////////////////////////////////////
100// Process incoming packets //
101///////////////////////////////////////////////////
102
Alexander Afanasyev2b4c9472012-08-09 15:00:38 -0700103} // namespace ndn
Alexander Afanasyevb7ad2322012-01-17 22:54:49 -0800104} // namespace ns3