eccodes/tests/bufr_extract_headers.cc

63 lines
2.0 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 MAX_KEYS 100
int main(int argc, char* argv[])
{
char *filename, *keys;
int i, err = 0;
int num_messages = 0;
codes_bufr_header* header_array = NULL;
codes_context* c = codes_context_get_default();
const int strict_mode = 1;
int requested_print_keys_count = MAX_KEYS;
codes_values requested_print_keys[MAX_KEYS];
/* Usage: prog keys file */
assert(argc == 3);
keys = argv[1]; /* comma-separated like bufr_ls/bufr_get */
filename = argv[2];
err = codes_bufr_extract_headers_malloc(c, filename, &header_array, &num_messages, strict_mode);
if (err) {
printf("ERROR: %s\n", grib_get_error_message(err));
return 1;
}
/* Mimic the behaviour of bufr_get -f -p keys for testing */
err = parse_keyval_string(NULL, keys, 0, GRIB_TYPE_UNDEFINED, requested_print_keys, &requested_print_keys_count);
if (err) return 1;
assert(requested_print_keys_count > 0);
for (i = 0; i < num_messages; ++i) {
int j;
for (j = 0; j < requested_print_keys_count; ++j) {
size_t vlen = 0;
char value[512] = {0,};
CODES_CHECK(codes_bufr_header_get_string(&header_array[i], requested_print_keys[j].name, value, &vlen), 0);
assert(vlen > 0);
if (j > 0) printf(" ");
printf("%s", value);
}
printf("\n");
}
free(header_array);
for (i = 0; i < requested_print_keys_count; ++i) {
free((char*)requested_print_keys[i].name);
}
return 0;
}