mpi_utils.cpp

Go to the documentation of this file.

#include <string.h>
#include "mpi_utils.h"
#include "util.h"



int mpi_bcast_vector_of_strings(vector<string> &vector_of_strings) {

    int s_size;
    char *sz;
    int proc_id=0,nump=1;

    util::get_proc_id_nump(proc_id,nump);    

   #ifdef PARALLEL

        int vec_size;
        if (proc_id==0)
                vec_size=vector_of_strings.size();
        MPI_Bcast(&vec_size,1,MPI_INT, 0, MPI_COMM_WORLD);
        if (proc_id!=0)
                vector_of_strings.resize(vec_size);
        
        for (int i=0;i<vec_size;i++) {
                
                if (proc_id==0) {
                        //convert vector_
                        string s = vector_of_strings[i];

                        sz = new char[s.length() + 1];
                        s_size=s.length();
                        sz[s_size]='\0';
                        strcpy(sz, s.c_str());                  
                }
                MPI_Bcast(&s_size,1,MPI_INT,0,MPI_COMM_WORLD);
                if (proc_id!=0)
                        sz = new char[s_size + 1]; 
                
                MPI_Bcast(&sz[0],s_size+1,MPI_CHAR,0,MPI_COMM_WORLD);
                
                if (proc_id!=0) {
                        string tmp_s(sz);
                        vector_of_strings[i]=tmp_s;
                }
        }
                
   #endif
}


int mpi_bcast_string(std::string &ss) {

    unsigned int len;
    int proc_id=0,nump=1;
    util::get_proc_id_nump(proc_id,nump);   
    if (proc_id==0)
        len = ss.length();
    mpi_bcast_scalar(len);
    if (proc_id!=0)
        ss.resize(len);
    //if ( v.length() < len ) v.insert(v.end(),(size_t)(len-v.length()), ' ' );
    //else if ( v.length() > len ) v.erase( len, v.length()-len );
    #ifdef PARALLEL
    MPI_Bcast( const_cast<char *>(ss.data()), len, MPI_CHAR, 0, MPI_COMM_WORLD);
    #endif

}