mpe_io.c

fortran并行计算包

/*
   (C) 2001 by Argonne National Laboratory.
       See COPYRIGHT in top-level directory.
*/
/*
 * This file contains a start on routines to provide some simple I/O functions
 * for MPE
 */

#include "mpe_misc_conf.h"
#include "mpe_misc.h"

#include <stdio.h>
#include <sys/types.h>
#include <fcntl.h>
#include <string.h>
#if defined( STDC_HEADERS ) || defined( HAVE_STDLIB_H )
#include <stdlib.h>
#endif
#if defined( HAVE_UNISTD_H )
/* Prototype for dup2; also defines STDOUT_FILENO */
#include <unistd.h>
#endif
#ifdef HAVE_WINDOWS_H
#include <io.h>
#endif

#ifndef STDOUT_FILENO
#define STDOUT_FILENO 1
#endif

/*@ 
  MPE_IO_Stdout_to_file - Re-direct stdout to a file
 
  Parameters:
+  name - Name of file.  If it contains '%d', this value will be replaced with
  the rank of the process in 'MPI_COMM_WORLD'.

-  mode - Mode to open the file in (see the man page for 'open').  A common
  value is '0644' (Read/Write for owner, Read for everyone else).  
  Note that this
  value is `anded` with your current 'umask' value.

  Notes:
  Some systems may complain when standard output ('stdout') is closed.  
@*/
void MPE_IO_Stdout_to_file( char *name, int mode )
{
    char fname[1024], *p;
    int  rank, fd;

    if ((p = strchr(name,'%')) && p[1] == 'd') {
	MPI_Comm_rank( MPI_COMM_WORLD, &rank );
	sprintf( fname, name, rank );
	fd = open( fname, O_WRONLY | O_CREAT, mode );
    }
    else {
	fd = open( name, O_WRONLY | O_CREAT, mode );
    }
    dup2( fd, STDOUT_FILENO );
}