22#include "ns3/net-device.h"
23#include "ns3/node-list.h"
25#include "ns3/nstime.h"
26#include "ns3/simulator-impl.h"
27#include "ns3/simulator.h"
155 MPI_Init(pargc, pargv);
212 uint32_t serializedSize = p->GetSerializedSize();
213 auto buffer =
new uint8_t[serializedSize + 16];
214 i->SetBuffer(buffer);
217 auto pTime =
reinterpret_cast<uint64_t*
>(buffer);
219 auto pData =
reinterpret_cast<uint32_t*
>(pTime);
223 p->Serialize(
reinterpret_cast<uint8_t*
>(pData), serializedSize);
227 uint32_t nodeSysId = destNode->GetSystemId();
229 MPI_Isend(
reinterpret_cast<void*
>(i->GetBuffer()),
257 MPI_Get_count(&status, MPI_CHAR, &count);
261 auto pTime =
reinterpret_cast<uint64_t*
>(
g_pRxBuffers[index]);
262 uint64_t time = *pTime++;
263 auto pData =
reinterpret_cast<uint32_t*
>(pTime);
269 count -=
sizeof(time) +
sizeof(node) +
sizeof(dev);
280 if (pThisDev->GetIfIndex() == dev)
317 MPI_Test(i->GetRequest(), &flag, &status);
342 MPI_Initialized(&flag);
349 NS_FATAL_ERROR(
"Cannot disable MPI environment without Initializing it first");
static void ReceiveMessages()
Check for received messages complete.
MPI_Comm GetCommunicator() override
Return the communicator used to run ns-3.
static uint32_t GetRxCount()
static bool g_freeCommunicator
Did ns-3 create the communicator? Have to free it.
uint32_t GetSystemId() override
Get the id number of this rank.
void Disable() override
Clean up the ns-3 parallel communications interface.
static void TestSendComplete()
Check for completed sends.
static bool g_mpiInitCalled
Has MPI Init been called by this interface.
void SendPacket(Ptr< Packet > p, const Time &rxTime, uint32_t node, uint32_t dev) override
Send a packet to a remote node.
static uint32_t g_size
Size of the MPI COM_WORLD group.
static bool g_enabled
Has this interface been enabled.
static uint32_t GetTxCount()
static std::list< SentBuffer > g_pendingTx
List of pending non-blocking sends.
void Enable(int *pargc, char ***pargv) override
Setup the parallel communication interface.
bool IsEnabled() override
Returns enabled state of parallel environment.
static MPI_Request * g_requests
Pending non-blocking receives.
static uint32_t g_rxCount
Total packets received.
uint32_t GetSize() override
Get the number of ranks used by ns-3.
static char ** g_pRxBuffers
Data buffers for non-blocking reads.
void Destroy() override
Deletes storage used by the parallel environment.
static uint32_t g_txCount
Total packets sent.
static MPI_Comm g_communicator
MPI communicator being used for ns-3 tasks.
static TypeId GetTypeId()
Register this type.
static uint32_t g_sid
System ID (rank) for this task.
static uint32_t GetSize()
Get the number of ranks used by ns-3.
Class to aggregate to a NetDevice if it supports MPI capability.
void Receive(Ptr< Packet > p)
Direct an incoming packet to the device Receive() method.
static Ptr< Node > GetNode(uint32_t n)
A base class which provides memory management and object aggregation.
Smart pointer class similar to boost::intrusive_ptr.
Tracks non-blocking sends.
MPI_Request m_request
The MPI request handle.
uint8_t * m_buffer
The buffer.
void SetBuffer(uint8_t *buffer)
MPI_Request * GetRequest()
static void ScheduleWithContext(uint32_t context, const Time &delay, FUNC f, Ts &&... args)
Schedule an event with the given context.
static Time Now()
Return the current simulation virtual time.
Simulation virtual time values and global simulation resolution.
int64_t GetInteger() const
Get the raw time value, in the current resolution unit.
int64_t GetTimeStep() const
Get the raw time value, in the current resolution unit.
a unique identifier for an interface.
TypeId SetParent(TypeId tid)
Set the parent TypeId.
int nDevices
Number of end device nodes to create.
Declaration of classes ns3::SentBuffer and ns3::GrantedTimeWindowMpiInterface.
#define NS_ASSERT(condition)
At runtime, in debugging builds, if this condition is not true, the program prints the source file,...
#define NS_FATAL_ERROR(msg)
Report a fatal error with a message and terminate.
#define NS_LOG_COMPONENT_DEFINE(name)
Define a Log component with a specific name.
#define NS_LOG_FUNCTION_NOARGS()
Output the name of the function.
#define NS_LOG_FUNCTION(parameters)
If log level LOG_FUNCTION is enabled, this macro will output all input parameters separated by ",...
#define NS_OBJECT_ENSURE_REGISTERED(type)
Register an Object subclass with the TypeId system.
Ptr< T > Create(Ts &&... args)
Create class instances by constructors with varying numbers of arguments and return them by Ptr.
Declaration of class ns3::MpiInterface.
ns3::MpiReceiver declaration, provides an interface to aggregate to MPI-compatible NetDevices.
Every class exported by the ns3 library is enclosed in the ns3 namespace.
const uint32_t MAX_MPI_MSG_SIZE
maximum MPI message size for easy buffer creation