offload.c

Go to the documentation of this file.

 
#include <stdlib.h>
#include <string.h>
#include "ascot5.h"
#include "offload.h"
 
void offload_init_offload(offload_package* o, real** offload_array,
                          int** int_offload_array) {
    *offload_array     = NULL;
    *int_offload_array = NULL;
    o->offload_array_length     = 0;
    o->int_offload_array_length = 0;
    o->unpack_pos     = 0;
    o->int_unpack_pos = 0;
}
 
 
void offload_free_offload(offload_package* o, real** offload_array,
                          int** int_offload_array) {
    free(*offload_array);
    free(*int_offload_array);
    *offload_array     = NULL;
    *int_offload_array = NULL;
    o->offload_array_length     = 0;
    o->int_offload_array_length = 0;
    o->unpack_pos     = 0;
    o->int_unpack_pos = 0;
}
 
 
void offload_pack(offload_package* o, real** offload_array, real* pack_array,
                  size_t pack_length, int** int_offload_array,
                  int* int_pack_array, size_t int_pack_length) {
 
    /* Float array */
    if( pack_length > 0 ) {
        size_t new_length = o->offload_array_length + pack_length;
        real* new_array = (real*) malloc(new_length * sizeof(real));
 
        if(o->offload_array_length > 0) {
            memcpy(new_array, *offload_array,
                   o->offload_array_length*sizeof(real));
        }
 
        memcpy(new_array+o->offload_array_length, pack_array,
               pack_length*sizeof(real));
 
        free(*offload_array);
 
        *offload_array = new_array;
        o->offload_array_length = new_length;
    }
 
    /* Int array */
    if( int_pack_length > 0 ) {
        size_t int_new_length = o->int_offload_array_length + int_pack_length;
        int* int_new_array = (int*) malloc(int_new_length * sizeof(int));
 
        if(o->int_offload_array_length > 0) {
            memcpy(int_new_array, *int_offload_array,
                   o->int_offload_array_length*sizeof(int));
        }
 
        memcpy(int_new_array+o->int_offload_array_length, int_pack_array,
               int_pack_length*sizeof(int));
 
        free(*int_offload_array);
 
        *int_offload_array = int_new_array;
        o->int_offload_array_length = int_new_length;
    }
}
 
 
void offload_unpack(offload_package* o, real* offload_array,
                    size_t pack_length, int* int_offload_array,
                    size_t int_pack_length, real** ptr, int** intptr) {
    *ptr = offload_array + o->unpack_pos;
    o->unpack_pos += pack_length;
 
    *intptr = int_offload_array + o->int_unpack_pos;
    o->int_unpack_pos += int_pack_length;
}