1 /* -*- Mode: C; c-basic-offset:4 ; indent-tabs-mode:nil ; -*- */
2 /*
3  *   Copyright (C) 1997 University of Chicago.
4  *   See COPYRIGHT notice in top-level directory.
5  */
6 
7 #include "mpioimpl.h"
8 
9 #ifdef HAVE_WEAK_SYMBOLS
10 
11 #if defined(HAVE_PRAGMA_WEAK)
12 #pragma weak MPI_File_write = PMPI_File_write
13 #elif defined(HAVE_PRAGMA_HP_SEC_DEF)
14 #pragma _HP_SECONDARY_DEF PMPI_File_write MPI_File_write
15 #elif defined(HAVE_PRAGMA_CRI_DUP)
16 #pragma _CRI duplicate MPI_File_write as PMPI_File_write
17 /* end of weak pragmas */
18 #elif defined(HAVE_WEAK_ATTRIBUTE)
19 int MPI_File_write(MPI_File fh, const void *buf, int count, MPI_Datatype datatype,
20                    MPI_Status *status) __attribute__((weak,alias("PMPI_File_write")));
21 #endif
22 
23 /* Include mapping from MPI->PMPI */
24 #define MPIO_BUILD_PROFILING
25 #include "mpioprof.h"
26 #endif
27 
28 /* status object not filled currently */
29 
30 /*@
31     MPI_File_write - Write using individual file pointer
32 
33 Input Parameters:
34 . fh - file handle (handle)
35 . buf - initial address of buffer (choice)
36 . count - number of elements in buffer (nonnegative integer)
37 . datatype - datatype of each buffer element (handle)
38 
39 Output Parameters:
40 . status - status object (Status)
41 
42 .N fortran
43 @*/
MPI_File_write(MPI_File fh,ROMIO_CONST void * buf,int count,MPI_Datatype datatype,MPI_Status * status)44 int MPI_File_write(MPI_File fh, ROMIO_CONST void *buf, int count,
45                    MPI_Datatype datatype, MPI_Status *status)
46 {
47     int error_code;
48     static char myname[] = "MPI_FILE_WRITE";
49 #ifdef MPI_hpux
50     int fl_xmpi;
51 
52     HPMP_IO_START(fl_xmpi, BLKMPIFILEWRITE, TRDTBLOCK, fh, datatype, count);
53 #endif /* MPI_hpux */
54 
55     error_code = MPIOI_File_write(fh, (MPI_Offset) 0, ADIO_INDIVIDUAL, buf,
56 				  count, datatype, myname, status);
57 
58 #ifdef MPI_hpux
59     HPMP_IO_END(fl_xmpi, fh, datatype, count);
60 #endif /* MPI_hpux */
61 
62     return error_code;
63 }
64 
65 /* prevent multiple definitions of this routine */
66 #ifdef MPIO_BUILD_PROFILING
MPIOI_File_write(MPI_File fh,MPI_Offset offset,int file_ptr_type,const void * buf,int count,MPI_Datatype datatype,char * myname,MPI_Status * status)67 int MPIOI_File_write(MPI_File fh,
68 		     MPI_Offset offset,
69 		     int file_ptr_type,
70 		     const void *buf,
71 		     int count,
72 		     MPI_Datatype datatype,
73 		     char *myname,
74 		     MPI_Status *status)
75 {
76     int error_code, buftype_is_contig, filetype_is_contig;
77     MPI_Count datatype_size;
78     ADIO_Offset off, bufsize;
79     ADIO_File adio_fh;
80     void *e32buf=NULL;
81     const void *xbuf=NULL;
82 
83     MPIU_THREAD_CS_ENTER(ALLFUNC,);
84 
85     adio_fh = MPIO_File_resolve(fh);
86 
87     /* --BEGIN ERROR HANDLING-- */
88     MPIO_CHECK_FILE_HANDLE(adio_fh, myname, error_code);
89     MPIO_CHECK_COUNT(adio_fh, count, myname, error_code);
90     MPIO_CHECK_DATATYPE(adio_fh, datatype, myname, error_code);
91 
92     if (file_ptr_type == ADIO_EXPLICIT_OFFSET && offset < 0)
93     {
94 	error_code = MPIO_Err_create_code(MPI_SUCCESS, MPIR_ERR_RECOVERABLE,
95 					  myname, __LINE__, MPI_ERR_ARG,
96 					  "**iobadoffset", 0);
97 	error_code = MPIO_Err_return_file(adio_fh, error_code);
98 	goto fn_exit;
99     }
100     /* --END ERROR HANDLING-- */
101 
102     MPI_Type_size_x(datatype, &datatype_size);
103 
104     /* --BEGIN ERROR HANDLING-- */
105     MPIO_CHECK_COUNT_SIZE(adio_fh, count, datatype_size, myname, error_code);
106     /* --END ERROR HANDLING-- */
107 
108     if (count*datatype_size == 0)
109     {
110 #ifdef HAVE_STATUS_SET_BYTES
111        MPIR_Status_set_bytes(status, datatype, 0);
112 #endif
113 	error_code = MPI_SUCCESS;
114 	goto fn_exit;
115     }
116 
117     /* --BEGIN ERROR HANDLING-- */
118     MPIO_CHECK_INTEGRAL_ETYPE(adio_fh, count, datatype_size, myname, error_code);
119     MPIO_CHECK_WRITABLE(adio_fh, myname, error_code);
120     MPIO_CHECK_NOT_SEQUENTIAL_MODE(adio_fh, myname, error_code);
121     /* --END ERROR HANDLING-- */
122 
123     ADIOI_Datatype_iscontig(datatype, &buftype_is_contig);
124     ADIOI_Datatype_iscontig(adio_fh->filetype, &filetype_is_contig);
125 
126     ADIOI_TEST_DEFERRED(adio_fh, myname, &error_code);
127 
128     xbuf = buf;
129     if (adio_fh->is_external32) {
130 	error_code = MPIU_external32_buffer_setup(buf, count, datatype, &e32buf);
131 	if (error_code != MPI_SUCCESS)
132 	    goto fn_exit;
133 
134 	xbuf = e32buf;
135     }
136 
137     if (buftype_is_contig && filetype_is_contig)
138     {
139 	/* convert bufcount and offset to bytes */
140 	bufsize = datatype_size * count;
141 	if (file_ptr_type == ADIO_EXPLICIT_OFFSET) {
142 	    off = adio_fh->disp + adio_fh->etype_size * offset;
143 	}
144 	else /* ADIO_INDIVIDUAL */ {
145 	    off = adio_fh->fp_ind;
146 	}
147 
148         /* if atomic mode requested, lock (exclusive) the region, because
149            there could be a concurrent noncontiguous request. Locking doesn't
150            work on PIOFS and PVFS, and on NFS it is done in the
151            ADIO_WriteContig.
152 	 */
153 
154         if ((adio_fh->atomicity) && ADIO_Feature(adio_fh, ADIO_LOCKS))
155 	{
156             ADIOI_WRITE_LOCK(adio_fh, off, SEEK_SET, bufsize);
157 	}
158 
159 	ADIO_WriteContig(adio_fh, xbuf, count, datatype, file_ptr_type,
160 		     off, status, &error_code);
161 
162         if ((adio_fh->atomicity) && ADIO_Feature(adio_fh, ADIO_LOCKS))
163 	{
164             ADIOI_UNLOCK(adio_fh, off, SEEK_SET, bufsize);
165 	}
166     }
167     else
168     {
169 	/* For strided and atomic mode, locking is done in ADIO_WriteStrided */
170 	ADIO_WriteStrided(adio_fh, xbuf, count, datatype, file_ptr_type,
171 			  offset, status, &error_code);
172     }
173 
174 
175     /* --BEGIN ERROR HANDLING-- */
176     if (error_code != MPI_SUCCESS)
177 	error_code = MPIO_Err_return_file(adio_fh, error_code);
178     /* --END ERROR HANDLING-- */
179 
180 fn_exit:
181     if (e32buf!= NULL) ADIOI_Free(e32buf);
182     MPIU_THREAD_CS_EXIT(ALLFUNC,);
183 
184     return error_code;
185 }
186 #endif
187