void eval_potential(int nR,
double *R,
double *z,
int npot,
int * pot_type,
double * pot_args,
double *out,
int * err){
int ii;
//Set up the potentials
struct potentialArg * potentialArgs= (struct potentialArg *) malloc ( npot * sizeof (struct potentialArg) );
parse_actionAngleArgs(npot,potentialArgs,pot_type,pot_args);
//Run through and evaluate
for (ii=0; ii < nR; ii++){
*(out+ii)= evaluatePotentials(*(R+ii),*(z+ii),npot,potentialArgs);
}
for (ii=0; ii < npot; ii++) {
if ( (potentialArgs+ii)->i2d )
interp_2d_free((potentialArgs+ii)->i2d) ;
if ((potentialArgs+ii)->accx )
gsl_interp_accel_free ((potentialArgs+ii)->accx);
if ((potentialArgs+ii)->accy )
gsl_interp_accel_free ((potentialArgs+ii)->accy);
free((potentialArgs+ii)->args);
}
free(potentialArgs);
}
/*
MAIN FUNCTIONS
*/
void calc_potential(int nR,
double *R,
int nz,
double *z,
int npot,
int * pot_type,
double * pot_args,
double *out,
int * err){
int ii, jj, tid, nthreads;
#ifdef _OPENMP
nthreads = omp_get_max_threads();
#else
nthreads = 1;
#endif
double * row= (double *) malloc ( nthreads * nz * ( sizeof ( double ) ) );
//Set up the potentials
struct potentialArg * potentialArgs= (struct potentialArg *) malloc ( npot * sizeof (struct potentialArg) );
parse_actionAngleArgs(npot,potentialArgs,pot_type,pot_args);
//Run through the grid and calculate
int chunk= CHUNKSIZE;
#pragma omp parallel for schedule(static,chunk) private(ii,tid,jj) \
shared(row,npot,potentialArgs,R,z,nR,nz)
for (ii=0; ii < nR; ii++){
#ifdef _OPENMP
tid= omp_get_thread_num();
#else
tid = 0;
#endif
for (jj=0; jj < nz; jj++){
*(row+jj+tid*nz)= evaluatePotentials(*(R+ii),*(z+jj),npot,potentialArgs);
}
put_row(out,ii,row+tid*nz,nz);
}
for (ii=0; ii < npot; ii++) {
if ( (potentialArgs+ii)->i2d )
interp_2d_free((potentialArgs+ii)->i2d) ;
if ((potentialArgs+ii)->acc )
gsl_interp_accel_free ((potentialArgs+ii)->acc);
free((potentialArgs+ii)->args);
}
free(potentialArgs);
free(row);
}