HDFCFUtil.h

// Helper functions for handling dimension maps,clear memories and separating namelist
//
//  Authors:   MuQun Yang <myang6@hdfgroup.org> Eunsoo Seo
// Copyright (c) 2010-2012 The HDF Group
#ifndef HDFCFUTIL_H
#define HDFCFUTIL_H
 
#include <stdlib.h>
#include <string>
#include <iostream>
#include <sstream>
#include <vector>
#include <map>
#include <set>
#include <assert.h>
#include <iomanip>
#include <dirent.h>
#include <libgen.h>
#include <unistd.h>
#include <cerrno>
 
#include "mfhdf.h"
#include "hdf.h"
 
#ifdef USE_HDFEOS2_LIB
#include "HdfEosDef.h"
#include "HDFEOS2.h"
#endif
#include "HDFSP.h"
 
#include <TheBESKeys.h>
#include <BESUtil.h>
#include <libdap/DDS.h>
#include <libdap/DAS.h>
#include <libdap/InternalErr.h>
 
#include <libdap/escaping.h> // for escattr
 
// This is the maximum number of MODIS special values.
const int MAX_NON_SCALE_SPECIAL_VALUE=65535;
 
// This is the minimum number of MODIS special values.
const int MIN_NON_SCALE_SPECIAL_VALUE=65500;
 
 
 
// This is used to retrieve dimension map info. when retrieving the data values of HDF-EOS swath that 
// has swath dimension maps.
struct dimmap_entry
{
    // geo dimension name
    std::string geodim;
 
    // data dimension name
    std::string datadim;
    
    // offset and increment of a dimension map
    int32 offset;
    int32 inc;
};
 
 
struct HDFCFUtil
{
 
    static void close_fileid(int32 sdfd,int32 file_id,int32 gridfd,int32 swathfd,bool pass_fileid_key);
 
 
    static std::string escattr(std::string  s);
 
 
    static void Split (const char *s, int len, char sep,
        std::vector < std::string > &names);
 
    static void Split (const char *sz, char sep,
                            std::vector < std::string > &names);
 
    static bool insert_map(std::map<std::string,std::string>& m, std::string key, std::string val);
 
    static std::string get_CF_string(std::string s);
 
    static void gen_unique_name(std::string &str,std::set<std::string>& namelist, int&clash_index);
 
    static void Handle_NameClashing(std::vector<std::string>&newobjnamelist);
    static void Handle_NameClashing(std::vector<std::string>&newobjnamelist,std::set<std::string>&objnameset);
 
    static std::string print_attr(int32, int, void*);
 
    static std::string print_type(int32);
 
    static short obtain_type_size(int32);
 
 
    // Subsetting the 2-D fields
    template <typename T> static void LatLon2DSubset (T* outlatlon, int ydim, int xdim, T* latlon, const int32 * offset, const int32 * count, const int32 * step);
 
    static void correct_fvalue_type(libdap::AttrTable *at,int32 dtype);
 
 
    static void correct_scale_offset_type(libdap::AttrTable *at);
 
 
#ifdef USE_HDFEOS2_LIB
 
 
    static bool change_data_type(libdap::DAS & das, SOType scaletype, const std::string & new_field_name);
 
    static bool is_special_value(int32 dtype,float fillvalue, float realvalue);
 
    static int check_geofile_dimmap(const std::string & geofilename);
 
    static bool is_modis_dimmap_nonll_field(std::string & fieldname);
 
    static void obtain_dimmap_info(const std::string& filename, HDFEOS2::Dataset*dataset,std::vector<struct dimmap_entry>& dimmaps, std::string & modis_geofilename,bool &geofile_nas_dimmap);
 
    //Scale and offset values with other types will be converted to float and their input values must also be in float.
    //If the add_offset_found is false, the add_offset attribute will not be added. add_offset information will not be used.
    static void add_scale_offset_attrs(libdap::AttrTable* at, const std::string& s_type, float svalue_f, double svalue_d, bool add_offset_found, 
                                       const std::string& o_type, float ovalue_f, double ovalue_d);
 
    static void add_scale_str_offset_attrs(libdap::AttrTable* at, const std::string& s_type, const std::string& s_value_str, bool add_offset_found, 
                                       const std::string& o_type, float ovalue_f, double ovalue_d);
 
 
 
    static void handle_modis_special_attrs_disable_scale_comp(libdap::AttrTable *at,const string &filename, bool is_grid, const std::string &newfname, SOType scaletype);
 
    static void handle_modis_special_attrs(libdap::AttrTable *at,const std::string &filename, bool is_grid, const std::string & newfname, SOType scaletype, bool gridname_change_valid_range, bool changedtype, bool &change_fvtype);
 
    static void handle_modis_vip_special_attrs(const std::string& valid_range_value,const std::string& scale_factor_value, float& valid_min, float & valid_max);
 
    static void handle_amsr_attrs(libdap::AttrTable *at);
 
    // The following four functions is to handle 1-D CF CV variables required by CF projection conventions for grid. 
    // Obtains the latitude and longitude dimension info. of an HDF-EOS2 grid.
    static void obtain_grid_latlon_dim_info(const HDFEOS2::GridDataset*,string &,int32 &,string &,int32 &);
 
    // Adds the 1-D cf grid projection mapping attribute to data variables
    // It is called by the function add_cf_grid_attrs. 
    static void add_cf_grid_mapping_attr(libdap::DAS &das, const HDFEOS2::GridDataset*gdset,const string& cf_projection,
                                         const string & dim0name,int32 dim0size,const string &dim1name,int32 dim1size);
 
    // This function adds the 1-D horizontal coordinate variables as well as the dummy projection variable to the grid.
    static void add_cf_grid_cvs(libdap::DDS & dds, const HDFEOS2::GridDataset *gdset);
 
    //Adds 1D grid mapping CF attributes to CV and data variables.
    static void add_cf_grid_cv_attrs(libdap::DAS &das, const HDFEOS2::GridDataset *gdset);
 
#endif 
 
    // Check OBPG attributes. Specifically, check if slope and intercept can be obtained from the file level. 
    // If having global slope and intercept,  obtain OBPG scaling, slope and intercept values.
    static void check_obpg_global_attrs(HDFSP::File*f,std::string & scaling, float & slope,bool &global_slope_flag,float & intercept, bool & global_intercept_flag);
 
    // For some OBPG files that only provide slope and intercept at the file level, 
    // global slope and intercept are needed to add to all fields and their names are needed to be changed to scale_factor and add_offset.
    // For OBPG files that provide slope and intercept at the field level,  slope and intercept are needed to rename to scale_factor and add_offset.
    static void add_obpg_special_attrs(const HDFSP::File*f,libdap::DAS &das,  const HDFSP::SDField* spsds,const std::string & scaling, float&slope,const bool &global_slope_flag,float& intercept,const bool &global_intercept_flag);
 
    // Handle HDF4 OTHERHDF products that follow SDS dimension scale model. 
    // The special handling of AVHRR data is also included.
    static void handle_otherhdf_special_attrs(const HDFSP::File *f, libdap::DAS &das);
 
    // Add  missing CF attributes for non-CV varibles
    static void add_missing_cf_attrs(const HDFSP::File*f,libdap::DAS &das);
 
    // Handle Merra and CERES attributes with the BES key EnableCERESMERRAShortName.
    static void handle_merra_ceres_attrs_with_bes_keys(const HDFSP::File*f, libdap::DAS &das,const std::string& filename);
 
    // Handle the attributes with the BES key EnableVdataDescAttr.
    static void handle_vdata_attrs_with_desc_key(const HDFSP::File*f,libdap::DAS &das);
 
    // Map AIRS version 6 or MOD08 HDF-EOS object attributes to DAP2, here we simply call Hopen and just retrieve the vgroup info. for the performance reason.
    static void map_eos2_objects_attrs(libdap::DAS &das,const string &filename);
    static void map_eos2_one_object_attrs_wrapper(libdap::DAS &das,int32 file_id,int32 vgroup_id, const string &vgroup_name, bool is_grid);
    static void map_eos2_one_object_attrs(libdap::DAS &das,int32 file_id,int32 obj_attr_group_id, const string &vgroup_name);
 
    // Parse TRMM V7 GridHeaders
    static void parser_trmm_v7_gridheader(const std:: vector<char>&value, int& latsize, int&lonsize, float& lat_start, float& lon_start, float& lat_res, float& lon_res, bool check_reg_orig);
 
    // Use to generate cache file name.
    // Reverse the char array order
    static void rev_str(char *str, int len);
 
    // Return the index of the char array for the integer part 
    static int int_to_str(int,char str[],int);
 
    // Convert a double number to char array
    static void dtoa(double,char *,int);
 
    // Obtain the double number in the string format
    static std::string get_double_str(double,int,int);
 
    // Obtain the integer in the string format
    static std::string get_int_str(int);
     
   
#if 0
    static size_t write_vector_to_file(const std::string &,const vector<double> &,size_t);
    static ssize_t write_vector_to_file2(const std::string &,const vector<double> &,size_t);
#endif
    // Read double-type data from a file to a vector of double type
    static ssize_t read_vector_from_file(int fd,vector<double> &,size_t);
     
    // wrap function of Unix read to a buffer. Memory for the buffer should be allocated.
    static ssize_t read_buffer_from_file(int fd,void*buf,size_t);
    static std::string obtain_cache_fname(const std::string & fprefix, const std::string & fname, const std::string &vname); 
    static size_t obtain_dds_cache_size(const HDFSP::File*);
    static void write_sp_sds_dds_cache(const HDFSP::File*,FILE*,size_t,const std::string & fname);
    static void read_sp_sds_dds_cache(FILE*,libdap::DDS * dds_ptr,const std::string &filename, const std::string &hdf_filename);
     
};
 
inline int32
INDEX_nD_TO_1D (const std::vector < int32 > &dims,
                                const std::vector < int32 > &pos)
{
#if 0
    /*
     int a[10][20][30];  // & a[1][2][3] == a + (20*30+1 + 30*2 + 1 *3);
     int b[10][2]; // &b[1][2] == b + (20*1 + 2);
    */
#endif
    assert (dims.size () == pos.size ());
    int32 sum = 0;
    int32 start = 1;
 
    for (unsigned int p = 0; p < pos.size (); p++) {
        int32 m = 1;
 
        for (unsigned int j = start; j < dims.size (); j++)
            m *= dims[j];
        sum += m * pos[p];
        start++;
    }
    return sum;
}
 
// Build a lock of a certain type.
static inline struct flock *lock(int type) {
    static struct flock lock;
    lock.l_type = (short)type;
    lock.l_whence = SEEK_SET;
    lock.l_start = 0;
    lock.l_len = 0;
    lock.l_pid = getpid();
 
    return &lock;
}
 
static inline string get_errno() {
        const char *s_err = strerror(errno);
        if (s_err)
                return s_err;
        else
                return "Unknown error.";
}
 
//
//      \param input Input variable
//       \param dim dimension info of the input
//       \param start start indexes of each dim
//       \param stride stride of each dim
//       \param edge count of each dim
//       \param poutput output variable
//      \parrm index dimension index
//       \return 0 if successful. -1 otherwise.
//
#if 0
template<typename T>
int subset(
    const T input[],
    int rank,
    vector<int> & dim,
    vector<int> & start,
    vector<int> & stride,
    vector<int> & edge,
    std::vector<T> *poutput,
    vector<int>& pos,
    int index)
{
    for(int k=0; k<edge[index]; k++)
    {
        pos[index] = start[index] + k*stride[index];
        if(index+1<rank)
            subset(input, rank, dim, start, stride, edge, poutput,pos,index+1);
        if(index==rank-1)
        {
            poutput->push_back(input[INDEX_nD_TO_1D( dim, pos)]);
        }
    } // end of for
    return 0;
} // end of template<typename T> static int subset
#endif
 
#endif