traffic-generator-helper.cc

// Copyright (c) 2008 INRIA
// Copyright (c) 2023 Centre Tecnologic de Telecomunicacions de Catalunya (CTTC)
//
// SPDX-License-Identifier: GPL-2.0-only
 
#include "traffic-generator-helper.h"
 
#include "ns3/inet-socket-address.h"
#include "ns3/names.h"
#include "ns3/packet-socket-address.h"
#include "ns3/string.h"
 
namespace ns3
{
 
TrafficGeneratorHelper::TrafficGeneratorHelper(std::string protocol,
                                               Address address,
                                               TypeId ftpTypeId)
{
    m_factory.SetTypeId(ftpTypeId);
    m_factory.Set("Protocol", StringValue(protocol));
    m_factory.Set("Remote", AddressValue(address));
}
 
void
TrafficGeneratorHelper::SetAttribute(std::string name, const AttributeValue& value)
{
    m_factory.Set(name, value);
}
 
ApplicationContainer
TrafficGeneratorHelper::Install(Ptr<Node> node) const
{
    return ApplicationContainer(InstallPriv(node));
}
 
ApplicationContainer
TrafficGeneratorHelper::Install(std::string nodeName) const
{
    Ptr<Node> node = Names::Find<Node>(nodeName);
    return ApplicationContainer(InstallPriv(node));
}
 
ApplicationContainer
TrafficGeneratorHelper::Install(NodeContainer c) const
{
    ApplicationContainer apps;
    for (NodeContainer::Iterator i = c.Begin(); i != c.End(); ++i)
    {
        apps.Add(InstallPriv(*i));
    }
 
    return apps;
}
 
Ptr<Application>
TrafficGeneratorHelper::InstallPriv(Ptr<Node> node) const
{
    Ptr<Application> app = m_factory.Create<Application>();
    node->AddApplication(app);
 
    return app;
}
 
} // namespace ns3