title: ecCodes索引访问

weight: 5

先介绍与索引访问有关的几个函数。

  • 创建文件内容的索引
  1. codes_index* codes_index_new_from_file(codes_context* c, char* filename,const char* keys,int *err);
  • 从索引中获取某个 key 的维度
  1. int codes_index_get_size(codes_index* index,const char* key,size_t* size);
  • 从索引中获取 key 的不同“值”
  1. int codes_index_get_long(codes_index* index,const char* key,long* values,size_t *size);
  2. int codes_index_get_double(codes_index* index,const char* key, double* values,size_t *size);
  3. int codes_index_get_string(codes_index* index,const char* key,char** values,size_t *size);
  • 从索引中为某个 key 选择一个“值”
  1. int codes_index_select_long(codes_index* index,const char* key,long value);
  2. int codes_index_select_double(codes_index* index,const char* key,double value);
  3. int codes_index_select_string(codes_index* index,const char* key,char* value);
  • 加载与选择模型对应的 GRIB 消息
  1. codes_handle* codes_handle_new_from_index(codes_index* index,int *err);
  • 释放索引
  1. void codes_index_delete(codes_index* index);
  • 释放 GRIB 消息
  1. int codes_handle_delete(codes_handle* h);

对于“随机访问”来说，使用索引访问通常比顺序访问更快。

示例：索引访问

  1. #include <iostream>
  2. #include <eccodes.h>
  3. susing namespace std;
  5. int main(int argc, char** argv)
  6. {
  7.     if(argc < 2)
  8.     {
  9.         cout<<"Usage: "<<argv[0]<<" grib_file_path";
  10.         return 1;
  11.     }
  13.     const char* file_path = argv[1];
  15.     int err = 0;
  17.     codes_index* index = nullptr;
  18.     index = codes_index_new(nullptr, "paramId", &err);
  19.     if (err)
  20.     {
  21.         cout<<"error: "<<codes_get_error_message(err)<<endl;
  22.         exit(err);
  23.     }
  25.     err = codes_index_add_file(index, file_path);
  27.     if (err)
  28.     {
  29.         cout<<"error: " << codes_get_error_message(err)<<endl;
  30.         exit(err);
  31.     }
  32.     cout<<"end indexing..."<<endl;
  34.     size_t param_id_size;
  35.     codes_index_get_size(index,"paramId", &param_id_size);
  37.     auto param_id_array= new long[param_id_size];
  38.     codes_index_get_long(index, "paramId", param_id_array, &param_id_size);
  40.     for(int i=0; i < param_id_size; i++)
  41.     {
  42.         auto param_id = param_id_array[i];
  43.         codes_index_select_long(index, "paramId", param_id);
  45.         codes_handle* h = nullptr;
  46.         int count = 0;
  47.         while ((h = codes_handle_new_from_index(index, &err)) != nullptr)
  48.         {
  49.             count++;
  50.             if (err)
  51.             {
  52.                 printf("error: %d\n",err);
  53.                 exit(err);
  54.             }
  56.             size_t short_name_length;
  57.             codes_get_length(h, "shortName", &short_name_length);
  58.             auto short_name = new char[short_name_length];
  59.             codes_get_string(h, "shortName", short_name, &short_name_length);
  61.             long level;
  62.             codes_get_long(h,"level",&level);
  64.             long number;
  65.             codes_get_long(h,"number",&number);
  67.             long step;
  68.             codes_get_long(h,"step",&step);
  70.             printf("shortName=%s ",short_name);
  71.             printf("level=%ld ",level);
  72.             printf("number=%ld ",number);
  73.             printf("step=%ld \n",step);
  74.             codes_handle_delete(h);
  75.             delete [] short_name;
  76.         }
  77.     }
  79.     codes_index_delete(index);
  81.     return 0;
  82. }

输入/输出

索引可以保存到文件中，方便重复使用。

  • 保存索引到文件
  1. int codes_index_write(codes_index *index, const char *filename);
  • 加载使用 codes_index_write 创建的索引文件
  1. codes_index* codes_index_read(codes_context* c,const char* filename,int *err);

正如前一节的示例所展示的，我们可以将数据文件的内容添加到索引中。

  1. int codes_index_add_file(grib_index *index, const char *filename);

可以使用 ecCodes 命令行工具 grib_index_build 创建索引文件。

  1. $ grib_index_build -o cli.index levels.grib2 
  2. --- grib_index_build: processing levels.grib2
  3. --- grib_index_build: keys included in the index file cli.index:
  4. --- mars.param, mars.levelist
  5. --- mars.param = { 156, 130, 131, 132, 260238 }
  6. --- mars.levelist = { 1000, 975, 950 }
  7. --- 15 messages indexed

创建的索引文件是二进制文件。

ecCodes 索引访问 - 图1

使用 grib_dump -D <index_file> 查看索引文件的内容。

  1. $ grib_dump -D cli.index
  2. ***** FILE: cli.index 
  3. GRIB File: levels.grib2
  4. Index keys:
  5. key name = mars.param
  6. values = 156, 130, 131, 132, 260238
  7. key name = mars.levelist
  8. values = 1000, 975, 950
  9. Index count = 15
  10. no messages found in cli.index

在练习章节中会介绍更详细的内容。