cpu_geometry.h

Go to the documentation of this file.

/***************************************************************************\
* Copyright (c) 2022, Claudio Pica, Sofie Martins                           *   
* All rights reserved.                                                      * 
\***************************************************************************/
 
#ifndef CPU_GEOMETRY_H
#define CPU_GEOMETRY_H
 
#include <stdlib.h>
 
#ifdef __cplusplus
extern "C" {
#endif
 
inline static int safe_mod(int x, int y) {
    if (x >= 0) {
        return (x % y);
    } else {
        return ((y - (abs(x) % y)) % y);
    }
}
 
inline static int safe_mod_alt(int x, int y) {
    while (x < 0) {
        x += y;
    }
    while (x >= y) {
        x -= y;
    }
    return x;
}
 
inline static int safe_mod_fast(int x, int y) {
    if (x < 0) {
        return x + y;
    } else {
        return (x < y) ? x : x - y;
    }
}
 
/* NB: it is assumed in the code that different directions are contiguous in memory */
#define coord_to_index(ix, mu) (((ix) << 2) | (mu))
#define index_to_coord(i, ix, mu) \
    (mu) = ((i) & 3);             \
    (ix) = ((i) >> 2)
 
//compute field element at position i, mu
#define _FIELD_AT(s, i) (((s)->ptr) + i - (s)->type->master_shift)
#define _3FIELD_AT(s, i, mu) (((s)->ptr) + ((i - (s)->type->master_shift) * 3 + mu))
#define _4FIELD_AT(s, i, mu) (((s)->ptr) + coord_to_index(i - (s)->type->master_shift, mu))
#define _6FIELD_AT(s, i, mu) (((s)->ptr) + ((i - (s)->type->master_shift) * 6 + mu))
#define _DFIELD_AT(s, i, mu, size) \
    ((size == 4) ? _4FIELD_AT(s, i, mu) : ((s)->ptr + ((i - (s)->type->master_shift) * size + mu)))
 
//same as above, but from a pointer.
#define _FIELD_AT_PTR(s, i, _master_shift) (s + i - _master_shift)
#define _4FIELD_AT_PTR(s, __i, mu, _master_shift) (s + coord_to_index(__i - _master_shift, mu))
#define _3FIELD_AT_PTR(s, i, mu, _master_shift) (s + ((i - _master_shift) * 3 + mu))
#define _6FIELD_AT_PTR(s, i, mu, _master_shift) (s + ((i - _master_shift) * 6 + mu))
#define _DFIELD_AT_PTR(s, i, mu, _master_shift, __size) \
    ((__size == 4) ? _4FIELD_AT_PTR(s, i, mu, _master_shift) : (s + ((i - _master_shift) * __size + mu)))
 
//compute start of geometry field master piece
#define _FIELD_BLK(s, i) (((s)->ptr) + ((s)->type->master_start[(i)]) - (s)->type->master_shift)
#define _4FIELD_BLK(s, i) (((s)->ptr) + 4 * ((s)->type->master_start[(i)] - (s)->type->master_shift))
#define _DFIELD_BLK(s, i, size) (((s)->ptr) + size * ((s)->type->master_start[(i)] - (s)->type->master_shift))
 
//compute start of geometry field receive buffer piece
#define _BUF_FIELD_BLK(s, i) (((s)->ptr) + ((s)->type->rbuf_start[(i)] - (s)->type->master_shift))
#define _BUF_4FIELD_BLK(s, i) (((s)->ptr) + 4 * ((s)->type->rbuf_start[(i)]))
#define _BUF_DFIELD_BLK(s, i, _size) (((s)->ptr) + (_size) * ((s)->type->rbuf_start[(i)] - (s)->type->master_shift))
 
#ifdef __cplusplus
}
#endif
#endif