MpiHost.h

Go to the documentation of this file.

/*
 * Copyright (C) 2020 Niek Linnenbank
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * 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, see <http://www.gnu.org/licenses/>.
 */
 
#ifndef __LIB_LIBMPI_MPIHOST_H
#define __LIB_LIBMPI_MPIHOST_H
 
#include <arpa/inet.h>
#include <netinet/in.h>
#include <Types.h>
#include <Index.h>
#include <List.h>
#include "MpiBackend.h"
 
class MpiHost : public MpiBackend
{
  private:
 
    static const Size MaximumNodes = 512;
 
  private:
 
    struct Node
    {
        in_addr_t ipAddress; 
        u16 udpPort;         
        u32 coreId;          
    };
 
    struct Packet
    {
        u8 *data;            
        Size size;           
    };
 
  public:
 
    MpiHost();
 
    virtual Result initialize(int *argc,
                              char ***argv);
 
    virtual Result terminate();
 
    virtual Result getCommRank(MPI_Comm comm,
                               int *rank);
 
    virtual Result getCommSize(MPI_Comm comm,
                               int *size);
 
    virtual Result send(const void *buf,
                        int count,
                        MPI_Datatype datatype,
                        int dest,
                        int tag,
                        MPI_Comm comm);
 
    virtual Result receive(void *buf,
                           int count,
                           MPI_Datatype datatype,
                           int source,
                           int tag,
                           MPI_Comm comm,
                           MPI_Status *status);
 
  private:
 
    Result parseHostsFile(const char *hostsfile);
 
    Result startProcesses(int argc,
                          char **argv);
 
    Result sendPacket(const Size nodeId,
                      const void *packet,
                      const Size size) const;
 
    Result receivePacket(const Size nodeId,
                         const MpiProxy::Operation operation,
                         void *packet,
                         Size & size);
 
  private:
 
    int m_sock;
 
    Index<Node, MaximumNodes> m_nodes;
 
    Index<List<Packet *>, MaximumNodes> m_packetBuffers;
};
 
#endif /* __LIB_LIBMPI_MPIHOST_H */