eccodes/tools/codes_export_resource.cc

105 lines
2.8 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 "grib_api_internal.h"
typedef enum ResourceType
{
UNKNOWN,
SAMPLE,
DEFINITION
} ResourceType;
static void usage(const char* prog)
{
printf("Usage: %s [-s | -d] resource_path out_file\n", prog);
exit(1);
}
#define SIZE (1024 * 1024)
int main(int argc, char* argv[])
{
char* resource_path = NULL;
const char* resource_name = NULL;
ResourceType resource_type = UNKNOWN;
char* full_path = NULL;
char* out_file = NULL;
char* option = NULL;
grib_context* c = grib_context_get_default();
FILE* fin = NULL;
FILE* fout = NULL;
char buffer[SIZE] = {0,};
size_t bytes = 0;
if (argc != 4) usage(argv[0]);
option = argv[1];
resource_path = argv[2];
out_file = argv[3];
if (strcmp(option, "-s") == 0) {
resource_type = SAMPLE;
resource_name = "sample";
}
else if (strcmp(option, "-d") == 0) {
resource_type = DEFINITION;
resource_name = "definition";
}
else {
fprintf(stderr, "Invalid option: Specify either -s or -d\n");
return 1;
}
if (resource_type == SAMPLE) {
full_path = get_external_template_path(c, resource_path);
}
else if (resource_type == DEFINITION) {
full_path = grib_context_full_defs_path(c, resource_path);
}
if (!full_path) {
fprintf(stderr, "Failed to access %s: '%s'\n", resource_name, resource_path);
return 1;
}
fout = fopen(out_file, "wb");
if (!fout) {
fprintf(stderr, "Failed to open output file '%s'\n", out_file);
return 1;
}
fin = codes_fopen(full_path, "r");
if (!fin) {
fprintf(stderr, "Failed to open resource '%s'\n", full_path);
fclose(fout);
return 1;
}
/* write resource bytes to fout */
while (0 < (bytes = fread(buffer, 1, sizeof(buffer), fin))) {
if (fwrite(buffer, 1, bytes, fout) != bytes) {
fprintf(stderr, "Failed to write out bytes\n");
return 1;
}
}
if (fclose(fin) != 0) {
fprintf(stderr, "Call to fclose failed (input)\n");
return 1;
}
if (fclose(fout) != 0) {
fprintf(stderr, "Call to fclose failed (output)\n");
return 1;
}
grib_context_free(c, full_path);
printf("Resource exported to file '%s'.\n", out_file);
return 0;
}