N0_3D.c

Go to the documentation of this file.

 
#include <stdlib.h>
#include <stdio.h>
#include <math.h>
#include <string.h>
#include "../math.h"
#include "../ascot5.h"
#include "../error.h"
#include "../print.h"
#include "N0_3D.h"
#include "../linint/linint.h"
 
int N0_3D_init_offload(N0_3D_offload_data* offload_data,
                       real** offload_array) {
    int N0_size = offload_data->n_r * offload_data->n_phi * offload_data->n_z;
    int T0_size = offload_data->n_r * offload_data->n_phi * offload_data->n_z;
 
    offload_data->offload_array_length = offload_data->n_species * N0_size
        + offload_data->n_species * T0_size;
 
    print_out(VERBOSE_IO, "\n3D neutral density and temperature (N0_3D)\n");
    print_out(VERBOSE_IO, "Grid:  nR = %4.d   Rmin = %3.3f   Rmax = %3.3f\n",
              offload_data->n_r,
              offload_data->r_min, offload_data->r_max);
    print_out(VERBOSE_IO, "       nz = %4.d   zmin = %3.3f   zmax = %3.3f\n",
              offload_data->n_z,
              offload_data->z_min, offload_data->z_max);
    print_out(VERBOSE_IO, "     nphi = %4.d phimin = %3.3f phimax = %3.3f\n",
              offload_data->n_phi,
              offload_data->phi_min, offload_data->phi_max);
    print_out(VERBOSE_IO,
              " Number of neutral species = %d\n",
              offload_data->n_species);
    print_out(VERBOSE_IO,
              "Species Z/A   (Maxwellian)\n");
    for(int i=0; i < offload_data->n_species; i++) {
        print_out(VERBOSE_IO,
                  "      %3d/%3d (%1d)    \n",
                  (int)(offload_data->znum[i]),
                  (int)(offload_data->anum[i]),
                  (int)(offload_data->maxwellian[i]));
    }
 
    return 0;
}
 
void N0_3D_free_offload(N0_3D_offload_data* offload_data,
                        real** offload_array) {
    free(*offload_array);
    *offload_array = NULL;
}
 
void N0_3D_init(N0_3D_data* ndata, N0_3D_offload_data* offload_data,
                real* offload_array) {
    int N0_size = offload_data->n_r * offload_data->n_phi * offload_data->n_z;
    int T0_size = offload_data->n_r * offload_data->n_phi * offload_data->n_z;
    ndata->n_species  = offload_data->n_species;
    for(int i = 0; i < offload_data->n_species; i++) {
        ndata->anum[i]       = offload_data->anum[i];
        ndata->znum[i]       = offload_data->znum[i];
        ndata->maxwellian[i] = offload_data->maxwellian[i];
 
        linint3D_init(
            &ndata->n0[i], &offload_array[i * N0_size],
            offload_data->n_r, offload_data->n_phi, offload_data->n_z,
            NATURALBC, PERIODICBC, NATURALBC,
            offload_data->r_min, offload_data->r_max,
            offload_data->phi_min, offload_data->phi_max,
            offload_data->z_min, offload_data->z_max);
 
        linint3D_init(
            &ndata->t0[i],
            &offload_array[i * T0_size + offload_data->n_species * N0_size],
            offload_data->n_r, offload_data->n_phi, offload_data->n_z,
            NATURALBC, PERIODICBC, NATURALBC,
            offload_data->r_min, offload_data->r_max,
            offload_data->phi_min, offload_data->phi_max,
            offload_data->z_min, offload_data->z_max);
    }
}
 
a5err N0_3D_eval_n0(real* n0, real r, real phi, real z, N0_3D_data* ndata) {
    a5err err = 0;
    int interperr = 0; /* If error happened during interpolation */
    for(int i=0; i<ndata->n_species; i++) {
        interperr += linint3D_eval_f(&n0[i], &ndata->n0[i], r, phi, z);
    }
 
    if(interperr) {
        return error_raise(ERR_INPUT_EVALUATION, __LINE__, EF_N0_3D);
    }
 
    return err;
}
 
a5err N0_3D_eval_t0(real* t0, real r, real phi, real z, N0_3D_data* ndata) {
    a5err err = 0;
    int interperr = 0; /* If error happened during interpolation */
    for(int i=0; i<ndata->n_species; i++) {
        interperr += linint3D_eval_f(&t0[i], &ndata->t0[i], r, phi, z);
    }
 
    if(interperr) {
        return error_raise(ERR_INPUT_EVALUATION, __LINE__, EF_N0_3D);
    }
 
    return err;
}
 
int N0_3D_get_n_species(N0_3D_data* ndata) {
    return ndata->n_species;
}