eccodes/examples/C/grib_sections_copy.c

106 lines
3.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"
#include <ctype.h>
static void usage(const char* prog)
{
printf("usage: %s in1.grib in2.grib what out.grib\n", prog);
printf("in1.grib The grib in whose sections we are interested, i.e. the source of the sections (read-only)\n");
printf("in2.grib The input grib (read-only)\n");
printf("what The section(s) to copy: p(Product), g(Grid), l(Local), d(Data), b(Bitmap)\n");
printf("out.grib The output file\n");
exit(1);
}
int main(int argc, char* argv[])
{
codes_handle *hfrom, *hto, *h;
FILE* in;
char *in_name1, *in_name2, *what_str, *out_name;
int i, err = 0, what = 0;
if (argc < 5) usage(argv[0]);
in_name1 = argv[1];
in_name2 = argv[2];
what_str = argv[3];
out_name = argv[4];
in = fopen(in_name1, "rb");
if (!in) {
perror(in_name1);
exit(1);
}
hfrom = codes_handle_new_from_file(0, in, PRODUCT_GRIB, &err);
CODES_CHECK(err, 0);
fclose(in);
in = fopen(in_name2, "rb");
if (!in) {
perror(in_name2);
exit(1);
}
hto = codes_handle_new_from_file(0, in, PRODUCT_GRIB, &err);
CODES_CHECK(err, 0);
fclose(in);
/* the sections for the "what" argument are:
* CODES_SECTION_PRODUCT
* CODES_SECTION_GRID
* CODES_SECTION_LOCAL
* CODES_SECTION_DATA
* CODES_SECTION_BITMAP
* one can bitwise-OR them to have more than one section copied
* e.g. what = CODES_SECTION_PRODUCT | CODES_SECTION_LOCAL;
*/
for (i = 0; i < strlen(what_str); ++i) {
if (what_str[i] == 'p') {
printf("Copying the PRODUCT section\n");
what |= CODES_SECTION_PRODUCT;
}
else if (what_str[i] == 'g') {
printf("Copying the GRID section\n");
what |= CODES_SECTION_GRID;
}
else if (what_str[i] == 'l') {
printf("Copying the LOCAL section\n");
what |= CODES_SECTION_LOCAL;
}
else if (what_str[i] == 'd') {
printf("Copying the DATA section\n");
what |= CODES_SECTION_DATA;
}
else if (what_str[i] == 'b') {
printf("Copying the BITMAP section\n");
what |= CODES_SECTION_BITMAP;
}
else if (isspace(what_str[i]) || what_str[i] == ',') {
/* ignore spaces and comma separator */
}
else {
fprintf(stderr, "Invalid option: '%c'. Ignoring.\n", what_str[i]);
}
}
h = codes_grib_util_sections_copy(hfrom, hto, what, &err);
CODES_CHECK(err, 0);
err = codes_write_message(h, out_name, "w");
codes_handle_delete(hfrom);
codes_handle_delete(hto);
codes_handle_delete(h);
return err;
}