plasma_1Dt.c

Go to the documentation of this file.

#include <stdio.h>
#include <stdlib.h>
#include <math.h>
#include "../ascot5.h"
#include "../error.h"
#include "../consts.h"
#include "../print.h"
#include "plasma_1Dt.h"
 

int plasma_1Dt_init(plasma_1Dt_data* data, int nrho, int ntime, int nion,
                    real* rho, real* time, int* anum, int* znum, real* mass,
                    real* charge, real* Te, real* Ti, real* ne, real* ni,
                    real* vtor) {
 
    data->n_rho = nrho;
    data->n_time = ntime;
    data->n_species = nion + 1;
 
    data->anum = (int*) malloc( nion*sizeof(int) );
    data->znum = (int*) malloc( nion*sizeof(int) );
    data->mass = (real*) malloc( (nion+1)*sizeof(real) );
    data->charge = (real*) malloc( (nion+1)*sizeof(real) );
    for(int i = 0; i < data->n_species; i++) {
        if(i < nion) {
            data->znum[i] = znum[i];
            data->anum[i] = anum[i];
        }
        data->mass[i] = mass[i];
        data->charge[i] = charge[i];
    }
    data->rho = (real*) malloc( nrho*sizeof(real) );
    for(int i = 0; i < nrho; i++) {
        data->rho[i] = rho[i];
    }
    data->time = (real*) malloc( ntime*sizeof(real) );
    for(int i = 0; i < ntime; i++) {
        data->time[i] = time[i];
    }
    data->vtor = (real*) malloc( nrho*ntime*sizeof(real) );
    data->temp = (real*) malloc( 2*nrho*ntime*sizeof(real) );
    data->dens = (real*) malloc( (nion+1)*nrho*ntime*sizeof(real) );
    for(int i = 0; i < nrho; i++) {
        for(int j = 0; j < ntime; j++) {
            data->vtor[j*nrho + i] = vtor[j*nrho + i];
            data->temp[j*2*nrho + i] = Te[j*nrho + i];
            data->temp[(j*2+1)*nrho + i] = Ti[j*nrho + i];
            data->dens[j*nrho + i] = ne[j*nrho + i];
            for(int k = 0; k < nion; k++) {
                data->dens[(k+1)*nrho*ntime + j*nrho + i] =
                    ni[k*nrho*ntime + j*nrho + i];
            }
        }
    }
 
    print_out(VERBOSE_IO, "\n1D plasma profiles (P_1Dt)\n");
    print_out(VERBOSE_IO,
              "Min rho = %1.2le, Max rho = %1.2le,"
              " Number of rho grid points = %d\n",
              data->rho[0], data->rho[data->n_rho-1], data->n_rho);
    print_out(VERBOSE_IO,
              "Min time = %1.2le, Max time = %1.2le,"
              " Number of time points = %d\n",
              data->time[0], data->time[data->n_time-1], data->n_time);
    print_out(VERBOSE_IO, "Number of ion species = %d\n", nion);
    print_out(VERBOSE_IO,
              "Species Z/A  charge [e]/mass [amu] "
              "Density [m^-3] at Min/Max rho(t=t0)"
              "  Temperature [eV] at Min/Max rho(t=t0)\n");
    for(int i=0; i < nion; i++) {
        print_out(VERBOSE_IO,
                  " %3d  /%3d   %3d  /%7.3f             %1.2le/%1.2le     "
                  "     %1.2le/%1.2le       \n",
                  data->znum[i], data->anum[i],
                  (int)round(data->charge[i+1]/CONST_E),
                  data->mass[i+1]/CONST_U,
                  data->dens[nrho*ntime + i*nrho],
                  data->dens[nrho*ntime + (i+1)*nrho - 1],
                  data->temp[ntime*nrho] / CONST_E,
                  data->temp[ntime*nrho + nrho - 1] / CONST_E);
    }
    print_out(VERBOSE_IO,
              "[electrons]  %3d  /%7.3f             %1.2le/%1.2le          "
              "%1.2le/%1.2le       \n",
              -1, CONST_M_E/CONST_U,
              data->dens[0], data->dens[nrho-1],
              data->temp[0] / CONST_E, data->temp[nrho-1] / CONST_E);
    print_out(VERBOSE_IO, "Toroidal rotation [rad/s] at Min/Max rho: "
              "%1.2le/%1.2le\n",
              data->vtor[0], data->vtor[nrho-1]);
    real quasineutrality = 0;
    for(int k = 0; k < nrho; k++) {
        real ele_qdens =
            data->dens[ntime + nrho + k] * CONST_E;
        real ion_qdens = 0;
        for(int i=0; i < nion; i++) {
            int idx = nrho*ntime + ntime + nrho * (2+1) + k;
            ion_qdens += data->dens[idx] * data->charge[i+1];
        }
        quasineutrality = fmax( quasineutrality,
                                fabs( 1 - ion_qdens / ele_qdens ) );
    }
    print_out(VERBOSE_IO, "Quasi-neutrality is (electron / ion charge density)"
              " %.2f\n", 1+quasineutrality);
    return 0;
}

void plasma_1Dt_free(plasma_1Dt_data* data) {
    free(data->mass);
    free(data->charge);
    free(data->anum);
    free(data->znum);
    free(data->rho);
    free(data->time);
    free(data->temp);
    free(data->dens);
}

void plasma_1Dt_offload(plasma_1Dt_data* data) {
    GPU_MAP_TO_DEVICE(
        data->mass[0:data->n_species], data->charge[0:data->n_species], \
        data->anum[0:data->n_species-1], data->znum[0:data->n_species-1], \
        data->rho[0:data->n_rho], data->time[0:data->n_time], \
        data->vtor[0:data->n_time*data->n_rho], \
        data->temp[0:data->n_time*data->n_rho*2], \
        data->dens[0:data->n_rho*data->n_species*data->n_time]
    )
}

a5err plasma_1Dt_eval_temp(real* temp, real rho, real t, int species,
                           plasma_1Dt_data* pls_data) {
 
    real temp_dens[MAX_SPECIES], temp_temp[MAX_SPECIES];
 
    a5err err = plasma_1Dt_eval_densandtemp(temp_dens, temp_temp, rho, t,
                                            pls_data);
 
    *temp = temp_temp[species];
 
    return err;
 
}

a5err plasma_1Dt_eval_dens(real* dens, real rho, real t, int species,
                           plasma_1Dt_data* pls_data) {
    real temp_dens[MAX_SPECIES], temp_temp[MAX_SPECIES];
 
    a5err err = plasma_1Dt_eval_densandtemp(temp_dens, temp_temp, rho, t,
                                            pls_data);
 
    *dens = temp_dens[species];
 
    return err;
 
}

a5err plasma_1Dt_eval_densandtemp(real* dens, real* temp, real rho, real t,
                                  plasma_1Dt_data* pls_data) {
 
    a5err err = 0;
    if(rho < pls_data->rho[0]) {
        err = error_raise( ERR_INPUT_EVALUATION, __LINE__, EF_PLASMA_1D );
    }
    else if(rho >= pls_data->rho[pls_data->n_rho-1]) {
        err = error_raise( ERR_INPUT_EVALUATION, __LINE__, EF_PLASMA_1D );
    }
    else {
        int i_rho = 0;
        while(i_rho < pls_data->n_rho-1 && pls_data->rho[i_rho] <= rho) {
            i_rho++;
        }
        i_rho--;
 
        real t_rho = (rho - pls_data->rho[i_rho])
                 / (pls_data->rho[i_rho+1] - pls_data->rho[i_rho]);
 
        int i_time = 0;
        while(i_time < pls_data->n_time-1 && pls_data->time[i_time] <= t) {
            i_time++;
        }
        i_time--;
 
        real t_time = (t - pls_data->time[i_time])
                 / (pls_data->time[i_time+1] - pls_data->time[i_time]);
 
        if(i_time < 0) {
            /* time < t[0], use first profile */
            i_time = 0;
            t_time = 0;
        }
        else if(i_time >= pls_data->n_time-2) {
            /* time > t[n_time-1], use last profile */
            i_time = pls_data->n_time-2;
            t_time = 1;
        }
 
        for(int i = 0; i < pls_data->n_species; i++) {
            real p11, p12, p21, p22, p1, p2;
 
            p11 = pls_data->dens[i_time*pls_data->n_species*pls_data->n_rho
                                 + i*pls_data->n_rho
                                 + i_rho];
            p12 = pls_data->dens[i_time*pls_data->n_species*pls_data->n_rho
                                 + i*pls_data->n_rho
                                 + i_rho + 1];
            p21 = pls_data->dens[(i_time+1)*pls_data->n_species*pls_data->n_rho
                                 + i*pls_data->n_rho
                                 + i_rho];
            p22 = pls_data->dens[(i_time+1)*pls_data->n_species*pls_data->n_rho
                                 + i*pls_data->n_rho
                                 + i_rho + 1];
 
            p1 = p11 + t_rho * (p12 - p11);
            p2 = p21 + t_rho * (p22 - p21);
 
            dens[i] = p1 + t_time * (p2 - p1);
 
            if(i < 2) {
                /* Electron and ion temperature */
                p11 = pls_data->temp[i_time*2*pls_data->n_rho
                                     + i*pls_data->n_rho
                                     +i_rho];
                p12 = pls_data->temp[i_time*2*pls_data->n_rho
                                     + i*pls_data->n_rho
                                     + i_rho + 1];
                p21 = pls_data->temp[(i_time+1)*2*pls_data->n_rho
                                     + i*pls_data->n_rho
                                     + i_rho];
                p22 = pls_data->temp[(i_time+1)*2*pls_data->n_rho
                                     + i*pls_data->n_rho
                                     + i_rho + 1];
 
                p1 = p11 + t_rho * (p12 - p11);
                p2 = p21 + t_rho * (p22 - p21);
 
                temp[i] = p1 + t_time * (p2 - p1);
            }
            else {
                /* Temperature is same for all ion species */
                temp[i] = temp[1];
            }
        }
    }
 
    return err;
}

a5err plasma_1Dt_eval_flow(real* vflow, real rho, real t, real r,
                           plasma_1Dt_data* pls_data) {
    a5err err = 0;
    if(rho < pls_data->rho[0]) {
        err = error_raise( ERR_INPUT_EVALUATION, __LINE__, EF_PLASMA_1D );
    }
    else if(rho >= pls_data->rho[pls_data->n_rho-1]) {
        err = error_raise( ERR_INPUT_EVALUATION, __LINE__, EF_PLASMA_1D );
    }
    else {
        int i_rho = 0;
        while(i_rho < pls_data->n_rho-1 && pls_data->rho[i_rho] <= rho) {
            i_rho++;
        }
        i_rho--;
 
        real t_rho = (rho - pls_data->rho[i_rho])
                 / (pls_data->rho[i_rho+1] - pls_data->rho[i_rho]);
 
        int i_time = 0;
        while(i_time < pls_data->n_time-1 && pls_data->time[i_time] <= t) {
            i_time++;
        }
        i_time--;
 
        real t_time = (t - pls_data->time[i_time])
                 / (pls_data->time[i_time+1] - pls_data->time[i_time]);
 
        if(i_time < 0) {
            /* time < t[0], use first profile */
            i_time = 0;
            t_time = 0;
        }
        else if(i_time >= pls_data->n_time-2) {
            /* time > t[n_time-1], use last profile */
            i_time = pls_data->n_time-2;
            t_time = 1;
        }
 
        real p11, p12, p21, p22, p1, p2;
 
        p11 = pls_data->vtor[i_time*pls_data->n_rho + i_rho];
        p12 = pls_data->vtor[(i_time+1)*pls_data->n_rho + i_rho];
        p21 = pls_data->vtor[i_time*pls_data->n_rho + i_rho + 1];
        p22 = pls_data->vtor[(i_time+1)*pls_data->n_rho + i_rho + 1];
 
        p1 = p11 + t_rho * (p12 - p11);
        p2 = p21 + t_rho * (p22 - p21);
 
        *vflow = p1 + t_time * (p2 - p1);
    }
    *vflow *= r;
    return err;
}