eccodes/tests/bufr_keys_iter.cc

72 lines
2.1 KiB
C++

/*
* (C) Copyright 2005- ECMWF.
*
* This software is licensed under the terms of the Apache Licence Version 2.0
* which can be obtained at http://www.apache.org/licenses/LICENSE-2.0.
*
* In applying this licence, ECMWF does not waive the privileges and immunities granted to it by
* virtue of its status as an intergovernmental organisation nor does it submit to any jurisdiction.
*/
#include "eccodes.h"
#undef NDEBUG
#include <assert.h>
#define ITER_ALL_KEYS 1
#define ITER_DATA_KEYS 2
int main(int argc, char* argv[])
{
int err = 0;
int i = 0;
codes_handle* h = NULL;
codes_bufr_keys_iterator* kiter = NULL;
char* input_filename = NULL;
FILE* f = NULL;
int iterator_mode = ITER_ALL_KEYS;
if (argc == 1 || argc > 3)
return 1; // usage: prog [-a|-d] infile
for (i = 1; i < argc; i++) {
if (strcmp(argv[i], "-a") == 0) {
iterator_mode = ITER_ALL_KEYS;
}
else if (strcmp(argv[i], "-d") == 0) {
iterator_mode = ITER_DATA_KEYS;
}
else {
input_filename = argv[i];
break; /* Only 1 file allowed */
}
}
assert(input_filename);
f = fopen(input_filename, "rb");
assert(f);
h = codes_handle_new_from_file(NULL, f, PRODUCT_BUFR, &err);
assert(h);
CODES_CHECK(codes_set_long(h, "unpack", 1), 0);
if (iterator_mode == ITER_ALL_KEYS) {
/*printf("Dumping ALL keys\n");*/
kiter = codes_bufr_keys_iterator_new(h, 0);
}
else {
/*printf("Dumping only DATA SECTION keys\n");*/
assert(iterator_mode == ITER_DATA_KEYS);
kiter = codes_bufr_data_section_keys_iterator_new(h);
}
while (codes_bufr_keys_iterator_next(kiter)) {
const char* kname = codes_bufr_keys_iterator_get_name(kiter);
printf("%s\n", kname);
}
codes_bufr_keys_iterator_delete(kiter);
codes_handle_delete(h);
codes_context_delete(codes_context_get_default());
return 0;
}