N0_1D.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_1D.h"
#include "../linint/linint.h"
 
int N0_1D_init_offload(N0_1D_offload_data* offload_data,
                       real** offload_array) {
    int N0_size = offload_data->n_rho;
    int T0_size = offload_data->n_rho;
 
    offload_data->offload_array_length = offload_data->n_species * N0_size
        + offload_data->n_species * T0_size;
 
    print_out(VERBOSE_IO, "\n1D neutral density and temperature (N0_1D)\n");
    print_out(VERBOSE_IO, "Grid:  nrho = %4.d   rhomin = %3.3f   rhomax = %3.3f\n",
              offload_data->n_rho,
              offload_data->rho_min, offload_data->rho_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_1D_free_offload(N0_1D_offload_data* offload_data,
                        real** offload_array) {
    free(*offload_array);
    *offload_array = NULL;
}
 
void N0_1D_init(N0_1D_data* ndata, N0_1D_offload_data* offload_data,
                real* offload_array) {
    int N0_size = offload_data->n_rho;
    int T0_size = offload_data->n_rho;
    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];
 
        linint1D_init(
            &ndata->n0[i], &offload_array[i * N0_size],
            offload_data->n_rho, NATURALBC,
            offload_data->rho_min, offload_data->rho_max);
 
        linint1D_init(
            &ndata->t0[i],
            &offload_array[i * T0_size + offload_data->n_species * N0_size],
            offload_data->n_rho, NATURALBC,
            offload_data->rho_min, offload_data->rho_max);
    }
}
 
a5err N0_1D_eval_n0(real* n0, real rho, N0_1D_data* ndata) {
    a5err err = 0;
    int interperr = 0; /* If error happened during interpolation */
    for(int i=0; i<ndata->n_species; i++) {
        interperr += linint1D_eval_f(&n0[i], &ndata->n0[i], rho);
    }
 
    if(interperr) {
        return error_raise(ERR_INPUT_EVALUATION, __LINE__, EF_N0_1D);
    }
 
    return err;
}
 
a5err N0_1D_eval_t0(real* t0, real rho, N0_1D_data* ndata) {
    a5err err = 0;
    int interperr = 0; /* If error happened during interpolation */
    for(int i=0; i<ndata->n_species; i++) {
        interperr += linint1D_eval_f(&t0[i], &ndata->t0[i], rho);
    }
 
    if(interperr) {
        return error_raise(ERR_INPUT_EVALUATION, __LINE__, EF_N0_1D);
    }
 
    return err;
}
 
int N0_1D_get_n_species(N0_1D_data* ndata) {
    return ndata->n_species;
}