MpiNode Class Reference

#include <xmipp_mpi.h>

Public Member Functions
	MpiNode (int &argc, char **&argv)
	~MpiNode ()
	MpiNode (const MpiNode &)=delete
MpiNode &	operator= (const MpiNode &)=delete
bool	isMaster () const
void	barrierWait ()
template<typename T >
void	gatherMetadatas (T &MD, const FileName &rootName)

Public Attributes
size_t	rank
size_t	size
size_t	active

Protected Member Functions
size_t	getActiveNodes ()

Detailed Description

Class to wrapp some MPI common calls in an work node.

Definition at line 47 of file xmipp_mpi.h.

Constructor & Destructor Documentation

MpiNode() [1/2]

MpiNode::MpiNode	(	int &	argc,
		char **&	argv
	)

Definition at line 144 of file xmipp_mpi.cpp.

{
    MPI_Init(&argc, &argv);
    int irank, isize;
    MPI_Comm_rank(MPI_COMM_WORLD, &irank);
    MPI_Comm_size(MPI_COMM_WORLD, &isize);
    rank=irank;
    size=isize;
    //comm = new MPI_Comm;
    //MPI_Comm_dup(MPI_COMM_WORLD, comm);
    active = 1;
    //activeNodes = size;
}

~MpiNode()

MpiNode::~MpiNode

(

)

Definition at line 158 of file xmipp_mpi.cpp.

{
    //active = 0;
    //updateComm();
    //std::cerr << "Send Finalize to: " << rank << std::endl;
    MPI_Finalize();
}

MpiNode() [2/2]

MpiNode::MpiNode ( const MpiNode &  )

delete

Member Function Documentation

barrierWait()

void MpiNode::barrierWait

(

)

Wait on a barrier for the other MPI nodes

Definition at line 171 of file xmipp_mpi.cpp.

{
    MPI_Barrier(MPI_COMM_WORLD);
}

gatherMetadatas()

template<typename T >

template void MpiNode::gatherMetadatas< MetaDataDb >	(	T &	MD,
		const FileName &	rootName
	)

Gather metadatas

Wait for all workers write results

Definition at line 200 of file xmipp_mpi.cpp.

{
    if (size == 1)
        return;

    FileName fn;

    if (!isMaster()) //workers just write down partial results
    {
        fn = formatString("%s_node%d.xmd", rootname.c_str(), rank);
        MD.write(fn);
    }
    barrierWait();
    if (isMaster()) //master should collect and join workers results
    {
        MetaDataDb mdAll(MD), mdSlave;
        for (size_t nodeRank = 1; nodeRank < size; nodeRank++)
        {
            fn = formatString("%s_node%d.xmd", rootname.c_str(), nodeRank);
            mdSlave.read(fn);
            //make sure file is not empty
            if (!mdSlave.isEmpty())
                mdAll.unionAll(mdSlave);
            //Remove blockname
            fn = fn.removeBlockName();
            remove(fn.c_str());
        }
        //remove first metadata
        fn = formatString("%s_node%d.xmd", rootname.c_str(), 1);
        fn = fn.removeBlockName();
        remove(fn.c_str());
        MD = T(mdAll);
    }
}

getActiveNodes()

size_t MpiNode::getActiveNodes ( )

protected

Update the MPI communicator to connect the currently active nodes Calculate the number of still active nodes

isMaster()

bool MpiNode::isMaster

(

)

const

Check if the node is master

Definition at line 166 of file xmipp_mpi.cpp.

{
    return rank == 0;
}

operator=()

MpiNode& MpiNode::operator= ( const MpiNode &  )

delete

Member Data Documentation

active

size_t MpiNode::active

Definition at line 52 of file xmipp_mpi.h.

rank

size_t MpiNode::rank

Definition at line 52 of file xmipp_mpi.h.

size

size_t MpiNode::size

Definition at line 52 of file xmipp_mpi.h.

---

The documentation for this class was generated from the following files:

• xmipp/libraries/parallel/xmipp_mpi.h
• xmipp/libraries/parallel/xmipp_mpi.cpp