1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
|
/*********************************************************************
*
* Copyright (C) 2012, Northwestern University and Argonne National Laboratory
* See COPYRIGHT notice in top-level directory.
*
*********************************************************************/
/* $Id$ */
/* This is part of the netCDF package.
Copyright 2006 University Corporation for Atmospheric Research/Unidata.
See COPYRIGHT file for conditions of use.
This program is adopted from the netCDFexample program, URL:
http://www.unidata.ucar.edu/software/netcdf/examples/programs/SimpleXyWr.cpp
This example writes a 2D array of sample data. We create two shared
dimensions, "x" and "y", and a netCDF variable, called "data".
*/
#include <stdio.h>
#include <stdlib.h>
#include <iostream>
#include <vector>
#include <string.h> /* strcpy(), strncpy() */
#include <unistd.h> /* getopt() */
#include <pnetcdf>
using namespace std;
using namespace PnetCDF;
using namespace PnetCDF::exceptions;
// We are writing 2D data, a 6 x 12 grid.
static const int NX = 6;
static const int NY = 12;
static void
usage(char *argv0)
{
cerr <<
"Usage: %s [-h] | [-q] [file_name]\n"
" [-h] Print help\n"
" [-q] Quiet mode (reports when fail)\n"
" [filename] output netCDF file name\n"
<< argv0;
}
int main(int argc, char *argv[])
{
extern int optind;
char filename[256];
int i, rank, nprocs, verbose=1;
MPI_Init(&argc, &argv);
MPI_Comm_rank(MPI_COMM_WORLD, &rank);
MPI_Comm_size(MPI_COMM_WORLD, &nprocs);
/* get command-line arguments */
while ((i = getopt(argc, argv, "hq")) != EOF)
switch(i) {
case 'q': verbose = 0;
break;
case 'h':
default: if (rank==0) usage(argv[0]);
MPI_Finalize();
return 1;
}
if (argv[optind] == NULL) strcpy(filename, "testfile.nc");
else snprintf(filename, 256, "%s", argv[optind]);
// This is the data array we will write. It will just be filled
// with a progression of numbers for this example.
int dataOut[NX][NY];
// Create some pretend data. If this wasn't an example program, we
// would have some real data to write, for example, model output.
for (int i = 0; i < NX; i++)
for (int j = 0; j < NY; j++)
dataOut[i][j] = i * NY + j;
// The default behavior of the C++ API is to throw an exception i
// an error occurs. A try catch block is necessary.
try {
// Create the file. The Replace parameter tells netCDF to overwrite
// this file, if it already exists.
NcmpiFile dataFile(MPI_COMM_WORLD, filename, NcmpiFile::replace);
// Create netCDF dimensions
NcmpiDim xDim = dataFile.addDim("x", NX);
NcmpiDim yDim = dataFile.addDim("y", NY);
// Define the variable. The type of the variable in this case is
// ncInt (32-bit integer).
vector<NcmpiDim> dims;
dims.push_back(xDim);
dims.push_back(yDim);
NcmpiVar data = dataFile.addVar("data", ncmpiInt, dims);
// Write the data to the file. Although netCDF supports
// reading and writing subsets of data, in this case we write all
// the data in one operation.
data.putVar_all(&dataOut[0][0]);
// The file will be automatically close when the NcmpiFile object goes
// out of scope. This frees up any internal netCDF resources
// associated with the file, and flushes any buffers.
if (verbose)
cout << "*** SUCCESS writing example file simple_xy.nc!" << endl;
}
catch(NcmpiException& e) {
cout << e.what() << " error code=" << e.errorCode() << " Error!\n";
return 1;
}
/* check if there is any PnetCDF internal malloc residue */
MPI_Offset malloc_size, sum_size;
int err = ncmpi_inq_malloc_size(&malloc_size);
if (err == NC_NOERR) {
MPI_Reduce(&malloc_size, &sum_size, 1, MPI_OFFSET, MPI_SUM, 0, MPI_COMM_WORLD);
if (rank == 0 && sum_size > 0)
printf("heap memory allocated by PnetCDF internally has %lld bytes yet to be freed\n",
sum_size);
}
MPI_Finalize();
return 0;
}
|
