void cfio_put_vara_float_c_(
int *ncid, int *varid, int *ndims,
int *start, int *count, float *fp, int *ierr)
{
size_t *_start, *_count;
int i, j;
debug(DEBUG_CFIO, "ndims : %d, start :(%d, %d), count :(%d, %d)",
*ndims, start[0], start[1], count[0], count[1]);
_start = malloc((*ndims) * sizeof(size_t));
if(NULL == _start)
{
debug(DEBUG_CFIO, "malloc fail");
*ierr = CFIO_ERROR_MALLOC;
return;
}
_count = malloc((*ndims) * sizeof(size_t));
if(NULL == _count)
{
free(_start);
debug(DEBUG_CFIO, "malloc fail");
*ierr = CFIO_ERROR_MALLOC;
return;
}
for(i = 0, j = (*ndims) - 1; i < (*ndims); i ++, j --)
{
_start[i] = start[j] - 1;
_count[i] = count[j];
}
*ierr = cfio_put_vara_float(
*ncid, *varid, *ndims, _start, _count, fp);
free(_start);
free(_count);
return;
}
int main(int argc, char** argv)
{
int rank, size;
char *path = "test";
int ncidp;
int dim1,var1,i;
size_t len = 10;
char *test="test";
MPI_Comm comm = MPI_COMM_WORLD;
MPI_Init(&argc, &argv);
MPI_Comm_rank(comm, &rank);
MPI_Comm_size(comm, &size);
//assert(size == LAT_PROC * LON_PROC);
//set_debug_mask(DEBUG_IO);
size_t start[2],count[2];
start[0] = (rank % LAT_PROC) * (LAT / LAT_PROC);
start[1] = (rank / LAT_PROC) * (LON / LON_PROC);
count[0] = LAT / LAT_PROC;
count[1] = LON / LON_PROC;
float *fp = malloc(count[0] * count[1] *sizeof(float));
for( i = 0; i< count[0] * count[1]; i++)
{
fp[i] = i + rank * count[0] * count[1];
}
cfio_init( LAT_PROC, LON_PROC, ratio);
CFIO_START();
char fileName[100];
memset(fileName, 0, sizeof(fileName));
sprintf(fileName,"%s.nc",path);
int dimids[2];
cfio_create(fileName, 0, &ncidp);
debug_mark(DEBUG_USER);
int lat = LAT;
cfio_def_dim(ncidp, "lat", LAT,&dimids[0]);
cfio_def_dim(ncidp, "lon", LON,&dimids[1]);
cfio_def_dim(ncidp, "lat", LAT,&dimids[0]);
int a = 3;
double b= 4.0;
cfio_put_att(ncidp, NC_GLOBAL, "test", CFIO_INT, 1, &a);
cfio_put_att(ncidp, NC_GLOBAL, "test", CFIO_FLOAT, 1, &b);
cfio_def_var(ncidp,"time_v", CFIO_FLOAT, 2, dimids, start, count, &var1);
cfio_put_att(ncidp, var1, "test", CFIO_CHAR, strlen(test), test);
cfio_enddef(ncidp);
cfio_put_vara_float(ncidp,var1, 2,start, count,fp);
cfio_close(ncidp);
cfio_io_end();
free(fp);
CFIO_END();
cfio_finalize();
MPI_Finalize();
return 0;
}
