B_TC.c

Go to the documentation of this file.

 
#include <stdlib.h>
#include <stdio.h>
#include <math.h>
#include "../ascot5.h"
#include "../math.h"
#include "../error.h"
#include "../print.h"
#include "B_TC.h"
 
int B_TC_init_offload(B_TC_offload_data* offload_data, real** offload_array) {
    // No initialization done
    offload_data->offload_array_length = 0;
    *offload_array = NULL;
 
    print_out(VERBOSE_IO,
              "\nTrivial cartesian magnetic field (B_TC)\n"
              "Magnetic axis at (R,z) = (%.1f,%.1f)\n"
              "psi = %.1f, rho = %.1f\n"
              "Magnetic field at origo\n"
              "B_x = %.3f B_y = %.3f B_z = %.3f\n"
              "Magnetic field gradient\n"
              "dB_x/dx = %.3f dB_x/dy = %.3f B_x/dz = %.3f\n"
              "dB_y/dx = %.3f dB_y/dy = %.3f B_y/dz = %.3f\n"
              "dB_z/dx = %.3f dB_z/dy = %.3f B_z/dz = %.3f\n",
              offload_data->axisr, offload_data->axisz,
              offload_data->psival, offload_data->rhoval,
              offload_data->B[0],  offload_data->B[1],  offload_data->B[2],
              offload_data->dB[0], offload_data->dB[1], offload_data->dB[2],
              offload_data->dB[3], offload_data->dB[4], offload_data->dB[5],
              offload_data->dB[6], offload_data->dB[7], offload_data->dB[8]);
 
    return 0;
}
 
void B_TC_free_offload(B_TC_offload_data* offload_data,
                       real** offload_array) {
    free(*offload_array);
    *offload_array = NULL;
}
 
void B_TC_init(B_TC_data* Bdata, B_TC_offload_data* offload_data,
               real* offload_array) {
    Bdata->rhoval = offload_data->rhoval;
    Bdata->psival = offload_data->psival;
    Bdata->axisr = offload_data->axisr;
    Bdata->axisz = offload_data->axisz;
 
    Bdata->B  = offload_data->B;
    Bdata->dB = offload_data->dB;
 
}
 
a5err B_TC_eval_psi(real* psi, real r, real phi, real z,
                   B_TC_data* Bdata) {
    psi[0] = Bdata->psival;
 
    return 0;
}
 
a5err B_TC_eval_psi_dpsi(real psi_dpsi[4], real r, real phi, real z,
                         B_TC_data* Bdata) {
    psi_dpsi[0] = Bdata->psival;
    psi_dpsi[1] = 0;
    psi_dpsi[2] = 0;
    psi_dpsi[3] = 0;
 
    return 0;
}
 
a5err B_TC_eval_rho_drho(real rho_drho[4], real r, real phi, real z,
                         B_TC_data* Bdata) {
    rho_drho[0] = Bdata->rhoval;
 
    rho_drho[1] = 0;
    rho_drho[2] = 0;
    rho_drho[3] = 0;
 
    return 0;
}
 
a5err B_TC_eval_B(real B[3], real r, real phi,
                 real z, B_TC_data* Bdata) {
    /* Find the Cartesian position and evaluate the field there */
    real xyz[3];
    real rpz[3] = {r, phi, z};
    math_rpz2xyz(rpz, xyz);
 
    real Bxyz[3];
    Bxyz[0] = Bdata->B[0] + Bdata->dB[0]*xyz[0] + Bdata->dB[1]*xyz[1]
        + Bdata->dB[2]*xyz[2];
    Bxyz[1] = Bdata->B[1] + Bdata->dB[3]*xyz[0] + Bdata->dB[4]*xyz[1]
        + Bdata->dB[5]*xyz[2];
    Bxyz[2] = Bdata->B[2] + Bdata->dB[6]*xyz[0] + Bdata->dB[7]*xyz[1]
        + Bdata->dB[8]*xyz[2];
 
    /* Transform the Cartesian field vector to cylindrical coordinates */
    math_vec_xyz2rpz(Bxyz, B, phi);
 
    return 0;
}
 
a5err B_TC_eval_B_dB(real B_dB[12], real r, real phi, real z,
                     B_TC_data* Bdata) {
    /* Find the Cartesian position and evaluate the field there */
    real xyz[3];
    real rpz[3] = {r, phi, z};
    math_rpz2xyz(rpz, xyz);
 
    real Bxyz[3];
    Bxyz[0] = Bdata->B[0] + Bdata->dB[0]*xyz[0] + Bdata->dB[1]*xyz[1]
        + Bdata->dB[2]*xyz[2];
    Bxyz[1] = Bdata->B[1] + Bdata->dB[3]*xyz[0] + Bdata->dB[4]*xyz[1]
        + Bdata->dB[5]*xyz[2];
    Bxyz[2] = Bdata->B[2] + Bdata->dB[6]*xyz[0] + Bdata->dB[7]*xyz[1]
        + Bdata->dB[8]*xyz[2];
 
    /* Transform the Cartesian field vector and Jacobian
       to cylindrical coordinates */
    real Brpz[3];
    math_vec_xyz2rpz(Bxyz, Brpz, phi);
    B_dB[0] = Brpz[0];
    B_dB[4] = Brpz[1];
    B_dB[8] = Brpz[2];
 
    real B_dBxyz[12] = {Bdata->B[0], Bdata->dB[0], Bdata->dB[1], Bdata->dB[2],
                        Bdata->B[1], Bdata->dB[3], Bdata->dB[4], Bdata->dB[5],
                        Bdata->B[2], Bdata->dB[6], Bdata->dB[7], Bdata->dB[8]};
    math_jac_xyz2rpz(B_dBxyz, B_dB, r, phi);
 
    return 0;
}
 
a5err B_TC_get_axis_rz(real rz[2], B_TC_data* Bdata) {
    a5err err = 0;
    rz[0] = Bdata->axisr;
    rz[1] = Bdata->axisz;
    return err;
}