map.h

Go to the documentation of this file.

#ifndef GHOST_MAP_H
#define GHOST_MAP_H

#include "config.h"
#include "types.h"
#include "sparsemat_src.h"

typedef enum
{
    GHOST_MAP_NONE,
    GHOST_MAP_ROW,
    GHOST_MAP_COL
} 
ghost_maptype;

typedef enum {
    GHOST_SPARSEMAT_SRC_FILE,
    GHOST_SPARSEMAT_SRC_MM,
    GHOST_SPARSEMAT_SRC_FUNC,
    GHOST_SPARSEMAT_SRC_NONE
} ghost_sparsemat_src;

typedef enum {
    GHOST_MAP_DIST_NNZ,
    GHOST_MAP_DIST_NROWS
} ghost_map_dist_type;
    
typedef enum {
    GHOST_MAP_DEFAULT=0,
    GHOST_PERM_NO_DISTINCTION=1
} ghost_map_flags;

typedef struct 
{
    ghost_gidx gdim;
    ghost_gidx *goffs;
    ghost_gidx offs;
    ghost_lidx *ldim;
    ghost_lidx dim;
    ghost_lidx dimhalo;
    ghost_lidx dimpad;
    ghost_lidx nhalo;
    ghost_lidx *loc_perm;
    ghost_lidx *loc_perm_inv;
    ghost_gidx *glb_perm;
    ghost_gidx *glb_perm_inv;
    ghost_lidx *cu_loc_perm;
    ghost_maptype type;
    ghost_mpi_comm mpicomm;
    ghost_map_flags flags;
    int ref_count;
} 
ghost_map;

#ifdef __cplusplus
extern "C" {
#endif

    ghost_error ghost_map_create_distribution(ghost_map *map, ghost_sparsemat_src_rowfunc *matsrc, double weight, ghost_map_dist_type distType,ghost_lidx *el_per_rank);
    ghost_error ghost_map_create(ghost_map **map, ghost_gidx gdim, ghost_mpi_comm comm, ghost_maptype type, ghost_map_flags flags);
    ghost_map *ghost_map_create_light(ghost_lidx dim, ghost_mpi_comm mpicomm);
    void ghost_map_destroy(ghost_map *map);
    int ghost_rank_of_row(ghost_map *map, ghost_gidx row);
#ifdef __cplusplus
}
#endif

#endif