E_field.c

Go to the documentation of this file.

 
#include <stdio.h>
#include "ascot5.h"
#include "error.h"
#include "print.h"
#include "E_field.h"
#include "B_field.h"
#include "Efield/E_TC.h"
#include "Efield/E_1DS.h"
 
int E_field_init_offload(E_field_offload_data* offload_data,
                          real** offload_array) {
    int err = 0;
 
    switch(offload_data->type) {
 
        case E_field_type_1DS:
            err = E_1DS_init_offload(&(offload_data->E1DS), offload_array);
            offload_data->offload_array_length =
                offload_data->E1DS.offload_array_length;
            break;
 
        case E_field_type_TC:
            err = E_TC_init_offload(&(offload_data->ETC), offload_array);
            offload_data->offload_array_length =
                offload_data->ETC.offload_array_length;
            break;
 
        default:
            /* Unregonized input. Produce error. */
            print_err("Error: Unregonized electric field type.");
            err = 1;
            break;
    }
 
    if(!err) {
        print_out(VERBOSE_IO, "Estimated memory usage %.1f MB\n",
                  offload_data->offload_array_length
                  * sizeof(real) / (1024.0*1024.0) );
    }
 
    return err;
}
 
void E_field_free_offload(E_field_offload_data* offload_data,
                          real** offload_array) {
    switch(offload_data->type) {
 
        case E_field_type_1DS:
            E_1DS_free_offload(&(offload_data->E1DS), offload_array);
            break;
 
        case E_field_type_TC:
            E_TC_free_offload(&(offload_data->ETC), offload_array);
            break;
    }
}
 
int E_field_init(E_field_data* Edata, E_field_offload_data* offload_data,
                  real* offload_array) {
    int err = 0;
 
    switch(offload_data->type) {
 
        case E_field_type_1DS:
            E_1DS_init(&(Edata->E1DS), &(offload_data->E1DS), offload_array);
            break;
 
        case E_field_type_TC:
            E_TC_init(&(Edata->ETC), &(offload_data->ETC), offload_array);
            break;
 
        default:
            /* Unregonized input. Produce error. */
            print_err("Error: Unregonized electric field type.\n");
            err = 1;
            break;
    }
    Edata->type = offload_data->type;
 
    return err;
}
 
a5err E_field_eval_E(real E[3], real r, real phi, real z, real t,
                     E_field_data* Edata, B_field_data* Bdata) {
    a5err err = 0;
 
    switch(Edata->type) {
 
        case E_field_type_1DS:
            err = E_1DS_eval_E(E, r, phi, z, &(Edata->E1DS), Bdata);
            break;
 
        case E_field_type_TC:
            err = E_TC_eval_E(E, r, phi, z, &(Edata->ETC), Bdata);
            break;
 
        default:
            /* Unregonized input. Produce error. */
            err = error_raise( ERR_UNKNOWN_INPUT, __LINE__, EF_E_FIELD );
            break;
    }
 
    return err;
}