nc_assembler.hh

Go to the documentation of this file.

/*
** SEAGNAL PROJECT, 2024
** netcdf-assembler
** File description:
** The header file of the project
*/
#include <algorithm>
#include <cerrno>
#include <cmath>
#include <cstdbool>
#include <cstddef>
#include <cstdint>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <fcntl.h>
#include <iostream>
#include <libgen.h>
#include <linux/limits.h>
#include <netcdf.h>
#include <stdexcept>
#include <unistd.h>
#include <variant>
#include <vector>
#include <wait.h>
 
#ifndef NC_ASSEMBLER_HH_
    #define NC_ASSEMBLER_HH_
 
    #define RED "\033[0;31m"
    #define YELLOW "\033[0;33m"
    #define BOLD "\033[1m"
    #define UNDERLINE "\033[4m"
    #define RESET "\033[0m"
    #ifdef DEBUG_MODE
        #define DEBUG (std::cout << YELLOW << "FILE: " << __FILE__ << ":" << __LINE__ << ", FUNCTION: " << __PRETTY_FUNCTION__ << RESET << std::endl)
    #else
        #define DEBUG
    #endif
    #define ERROR(ec) check_error(ec, __FILE__, __LINE__, __PRETTY_FUNCTION__)
 
/* The variable types */
using var_type = std::variant<signed char, char, short, int, float, double, unsigned char, unsigned short, unsigned int, long long, unsigned long long, char *>;
 
/* The dimension information */
typedef struct dimension_information_s {
    int32_t i_output_id = -1; /* The output id */
    char ac_dim_name[NC_MAX_NAME + 1] = {0}; /* The dimension name */
    size_t i_dim_len = 0; /* The dimension length */
} dimension_information_t;
 
/* The variable information */
typedef struct variable_information_s {
    int32_t i_id = 0; /* The variable id */
    char ac_var_name[NC_MAX_NAME + 1] = {0}; /* The variable name */
    nc_type i_type = 0; /* The variable type */
    int32_t i_ndims = 0; /* The number of dimensions */
    int32_t ai_dimids[NC_MAX_VAR_DIMS] = {0}; /* The dimensions ids */
    int32_t i_natts = 0; /* The number of attributes */
    size_t ai_dims_size[NC_MAX_VAR_DIMS] = {0}; /* The dimensions size */
    size_t i_data_size = 0; /* The data size */
    int32_t i_dim_id = -1; /* The dimension id (for dimension variable) */
    int32_t i_output_id = -1; /* The output id */
} variable_information_t;
 
/* The file information */
typedef struct file_information_s {
    char *ac_path = {0}; /* The file path */
    int32_t i_file_id = 0; /* The file id */
    int32_t i_nb_dimensions = 0; /* The number of dimensions */
    int32_t i_nb_variables = 0; /* The number of variables */
    int32_t i_nb_attributes = 0; /* The number of attributes */
    int32_t i_first_unlimited_dimensions_id = 0; /* The first unlimited dimension id */
    std::vector<dimension_information_t> vs_dims; /* The dimensions */
    std::vector<variable_information_t> vs_variables; /* The variables */
} file_information_t;
 
class assembler {
    protected:
        std::vector<file_information_t> _vs_input_files;
        file_information_t _s_output_file = {0};
 
    public:
        assembler(int argc, char **argv);
 
        ~assembler();
 
 
 
            /* Attributes functions */
 
        void copy_attributes(int32_t in_i_input_file, variable_information_t & in_s_ref_var,
                             variable_information_t & in_s_fill_var);
 
        void add_globals_attributes(void);
 
 
 
            /* Dimensions functions */
 
        variable_information_t get_variable_from_dim_id(file_information_t & in_s_file,
                                                        size_t in_i_dim_id);
 
        void copy_dimensions(void);
 
 
 
            /* Variables functions */
 
        void sort_variable(variable_information_t & in_s_current_var);
 
        size_t *get_start_from_input(size_t in_i_file, variable_information_t & in_s_input_var,
                                     variable_information_t & in_s_output_var, size_t *in_ai_input_start);
 
        void add_data_to_dim_variable(size_t in_i_file,
                                      variable_information_t & in_s_input_var,
                                      variable_information_t & in_s_output_var);
 
        void add_data_to_variable(size_t in_i_file,
                                  variable_information_t & in_s_input_var,
                                  variable_information_t & in_s_output_var);
 
        void copy_variables(void);
};
 
 
 
    /* Generale functions */
 
void check_error(int in_i_error, const char *in_ac_file,
                 const int in_i_line, const char *in_ac_func);
 
void open_file(file_information_t & in_s_file_info, int in_i_mode);
 
void create_file(file_information_t & in_s_file_info, int in_i_mode);
 
void get_info(file_information_t & in_s_file_info);
 
void close_file(file_information_t & in_s_file_info);
 
size_t get_dimension_size(file_information_t & in_s_file_info, size_t in_i_dim_id);
 
 
 
#endif /* NC_ASSEMBLER_HH_ */