问openMP:从并行区调用并行函数

EN

int fftw_shift(fftw_complex *pulse, fftw_complex *shift_buf, int 
array_size)
{
 int j = 0;              //counter
 if ((pulse != nullptr) || (shift_buf != nullptr)){
 if (omp_in_parallel()) {

 //shift the array
 #pragma omp for private(j) //shedule(dynamic)
 for (j = 0; j < array_size / 2; j++) {
  //left to right
  shift_buf[(array_size / 2) + j][REAL] = pulse[j][REAL]; //real
  shift_buf[(array_size / 2) + j][IMAG] = pulse[j][IMAG]; //imaginary

  //right to left
  shift_buf[j][REAL] = pulse[(array_size / 2) + j][REAL]; //real
  shift_buf[j][IMAG] = pulse[(array_size / 2) + j][IMAG]; //imaginary
 }
 //rewrite the array
 #pragma omp for private(j) //shedule(dynamic)
 for (j = 0; j < array_size; j++) {
  pulse[j][REAL] = shift_buf[j][REAL]; //real
  pulse[j][IMAG] = shift_buf[j][IMAG]; //imaginary
 }

 return 0;
 }
}

....
#pragma omp parallel firstprivate(x, phase) if(array_size >= 
OMP_THREASHOLD) 
{
 // First half-step
 #pragma omp for schedule(dynamic)
 for (x = 0; x < array_size; x++) {
  ..
 }

 // Forward FTW
 fftw_shift(pulse_x, shift_buf, array_size);
 #pragma omp master
 {
  fftw_execute(dft);
 }
 #pragma omp barrier
 fftw_shift(pulse_kx, shift_buf, array_size);
 ...
}