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
|
/*
* Demonstration of using the netCDF module from another C module.
*
* Written by Konrad Hinsen
* last revision: 2001-1-3
*/
#include "Python.h"
#include "numpy/arrayobject.h"
#include "netcdf/netcdfmodule.h"
/*
* Create a file with two dimensions and one variable.
*
* Attention: there is no error checking in this code! All return
* values ought to be tested.
*/
static PyObject *
create_file(self, args)
PyObject *self; /* Not used */
PyObject *args; /* Not used either */
{
/* Pointer to file object */
PyNetCDFFileObject *file;
/* Dimension names for variable foo */
char *dimensions[] = {"n", "xyz", "string_length"};
/* Pointer to variable object */
PyNetCDFVariableObject *foo, *bar;
/* Pointer to indices */
PyNetCDFIndex *indices;
/* Open file */
file = PyNetCDFFile_Open("demo.nc", "w");
/* Add file attribute */
PyNetCDFFile_SetAttributeString(file, "title", "useless data");
/* Add history line */
PyNetCDFFile_AddHistoryLine(file, "Created some day");
/* Create two dimensions */
PyNetCDFFile_CreateDimension(file, dimensions[0], 10);
PyNetCDFFile_CreateDimension(file, dimensions[1], 3);
PyNetCDFFile_CreateDimension(file, dimensions[2], 100);
/* Create variable */
foo = PyNetCDFFile_CreateVariable(file, "foo", 'l', dimensions, 2);
/* Add variable attribute */
PyNetCDFVariable_SetAttributeString(foo, "units", "arbitrary");
/* Create index array */
indices = PyNetCDFVariable_Indices(foo);
/* Write zeros everywhere */
PyNetCDFVariable_WriteArray(foo, indices, PyInt_FromLong(0));
/* Create variable */
bar = PyNetCDFFile_CreateVariable(file, "bar", 'c', dimensions+2, 1);
/* Write string */
PyNetCDFVariable_WriteString(bar,
(PyStringObject *)PyString_FromString("nothing important"));
/* Close file */
PyNetCDFFile_Close(file);
/* Return None */
Py_INCREF(Py_None);
return Py_None;
}
/* Table of functions defined in the module */
static PyMethodDef demo_methods[] = {
{"createDemoFile", create_file, 1},
{NULL, NULL} /* sentinel */
};
/* Module initialization */
PyMODINIT_FUNC initdemo(void)
{
PyObject *module;
PyObject *netcdf, *netcdf_dict;
PyObject *c_api_object;
/* Create the module and add the functions */
module = Py_InitModule("demo", demo_methods);
/* Import the array module */
import_array();
/* Import netcdf and retrieve its C API address array */
netcdf = PyImport_ImportModule("Scientific.IO.NetCDF");
if (netcdf != NULL) {
netcdf_dict = PyModule_GetDict(netcdf);
c_api_object = PyDict_GetItemString(netcdf_dict, "_C_API");
if (PyCObject_Check(c_api_object)) {
PyNetCDF_API = (void **)PyCObject_AsVoidPtr(c_api_object);
}
}
/* Check for errors */
if (PyErr_Occurred())
Py_FatalError("can't initialize module demo");
}
|
