pcp-slave.c

来自「The example for Boor USING MPI2」· C语言 代码 · 共 52 行

/* pcp from the Scalable Unix Tools, in MPI */
#include "mpi.h"
#include <stdio.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#define BUFSIZE    256*1024
#define CMDSIZE    80
int main( int argc, char *argv[] )
{
    int   mystatus, allstatus, done, numread;
    char  outfilename[128], controlmsg[80];
    int   outfd;
    char  buf[BUFSIZE];

    MPI_Init( &argc, &argv );

    MPI_Bcast( controlmsg, CMDSIZE, MPI_CHAR, 0, MPI_COMM_WORLD );
    if ( strcmp( controlmsg, "exit" ) == 0 ) {
        MPI_Finalize();
	return -1;
    }

    MPI_Bcast( controlmsg, CMDSIZE, MPI_CHAR, 0, MPI_COMM_WORLD );
    if ( (outfd = open( controlmsg, O_CREAT|O_WRONLY|O_TRUNC, S_IRWXU ) ) == -1 ) 
        mystatus = -1;
    else
        mystatus = 0;
    MPI_Allreduce( &mystatus, &allstatus, 1, MPI_INT, MPI_MIN,
		   MPI_COMM_WORLD );
    if ( allstatus == -1 ) {
	MPI_Finalize();
	return( -1 );
    }

    /* at this point all files have been successfully opened */
    
    done = 0;
    while ( !done ) {
	MPI_Bcast( &numread, 1, MPI_INT, 0, MPI_COMM_WORLD );
	if ( numread > 0 ) {
	    MPI_Bcast( buf, numread, MPI_BYTE, 0, MPI_COMM_WORLD );
	    write( outfd, buf, numread ); 
	}
	else {	  
	    close( outfd );
	    done = 1;
	}
    }
    MPI_Finalize();
}