FFTW_ComplexArray.hpp

Go to the documentation of this file.

#ifndef FFTW_ComplexArray_HPP
#define FFTW_ComplexArray_HPP

template<class T>
void FFTW_ComplexArray::copy(const complex<T>* f,const tUIndex& n) {
    tUIndex i;
    
    if (n==0) {
        desallocate();
        return;
    }
    
    //allocate the vector
    if (n>mCapacity) {
        mSize=n;
        allocate(n);
    }
    
    //copy the vector
    if (n==0) return;
    
    const complex<T>* pf=&f[0];
    tFFTWComplex * vs=mValues;
    OMP_PARALLEL_PRIVATE_SHARED_DEFAULT(private(i),shared(n,pf,vs),default(none)) {
        // id of the  thread
        tUInteger threadId=OMP_GET_THREAD_ID();
        // number of threads 
        tUInteger nThreads=OMP_GET_THREADS_NUMBER();
        
        // compute the loop index of each thread
        tUIndex start=threadId*n/nThreads;
        tUIndex end=(threadId+1)*n/nThreads;
        
        const complex<T> *fi=&pf[start];
        tFFTWComplex* vi=&vs[start];

        for (i=start;i<end;i++) {
            (*vi)[0]=fi->real();
            (*vi)[1]=fi->imag();
            vi++;
            fi++;
            
        }
    }//end parallel loop
}


#endif