1dedccee8SHong Zhang 2dedccee8SHong Zhang /* 3dedccee8SHong Zhang Provides an interface to the FFT packages. 4dedccee8SHong Zhang */ 5dedccee8SHong Zhang 6c6db04a5SJed Brown #include <../src/mat/impls/fft/fft.h> /*I "petscmat.h" I*/ 7dedccee8SHong Zhang 8dedccee8SHong Zhang PetscErrorCode MatDestroy_FFT(Mat A) 9dedccee8SHong Zhang { 10dedccee8SHong Zhang PetscErrorCode ierr; 11dedccee8SHong Zhang Mat_FFT *fft = (Mat_FFT*)A->data; 12dedccee8SHong Zhang 13dedccee8SHong Zhang PetscFunctionBegin; 14dedccee8SHong Zhang if (fft->matdestroy) { 15dedccee8SHong Zhang ierr = (fft->matdestroy)(A);CHKERRQ(ierr); 16dedccee8SHong Zhang } 17dedccee8SHong Zhang ierr = PetscFree(fft->dim);CHKERRQ(ierr); 18bf0cc555SLisandro Dalcin ierr = PetscFree(A->data);CHKERRQ(ierr); 19dedccee8SHong Zhang ierr = PetscObjectChangeTypeName((PetscObject)A,0);CHKERRQ(ierr); 20dedccee8SHong Zhang PetscFunctionReturn(0); 21dedccee8SHong Zhang } 22dedccee8SHong Zhang 230a9977b2SMatthew G. Knepley /*@C 24dedccee8SHong Zhang MatCreateFFT - Creates a matrix object that provides FFT via an external package 25dedccee8SHong Zhang 26*d083f849SBarry Smith Collective 27dedccee8SHong Zhang 28dedccee8SHong Zhang Input Parameter: 29dedccee8SHong Zhang + comm - MPI communicator 30dedccee8SHong Zhang . ndim - the ndim-dimensional transform 31dedccee8SHong Zhang . dim - array of size ndim, dim[i] contains the vector length in the i-dimension 32dedccee8SHong Zhang - type - package type, e.g., FFTW or FFTCU 33dedccee8SHong Zhang 34dedccee8SHong Zhang Output Parameter: 35dedccee8SHong Zhang . A - the matrix 36dedccee8SHong Zhang 37dedccee8SHong Zhang Options Database Keys: 38dedccee8SHong Zhang + -mat_fft_type - set FFT type 39dedccee8SHong Zhang 40dedccee8SHong Zhang Level: intermediate 41dedccee8SHong Zhang 42dedccee8SHong Zhang @*/ 4319fd82e9SBarry Smith PetscErrorCode MatCreateFFT(MPI_Comm comm,PetscInt ndim,const PetscInt dim[],MatType mattype,Mat *A) 44dedccee8SHong Zhang { 45dedccee8SHong Zhang PetscErrorCode ierr; 46dedccee8SHong Zhang PetscMPIInt size; 47dedccee8SHong Zhang Mat FFT; 48dedccee8SHong Zhang PetscInt N,i; 49dedccee8SHong Zhang Mat_FFT *fft; 50dedccee8SHong Zhang 51dedccee8SHong Zhang PetscFunctionBegin; 52dedccee8SHong Zhang if (ndim < 1) SETERRQ1(comm,PETSC_ERR_USER,"ndim %d must be > 0",ndim); 53dedccee8SHong Zhang ierr = MPI_Comm_size(comm, &size);CHKERRQ(ierr); 54dedccee8SHong Zhang 55dedccee8SHong Zhang ierr = MatCreate(comm,&FFT);CHKERRQ(ierr); 56b00a9115SJed Brown ierr = PetscNewLog(FFT,&fft);CHKERRQ(ierr); 57dedccee8SHong Zhang FFT->data = (void*)fft; 58dedccee8SHong Zhang N = 1; 59dedccee8SHong Zhang for (i=0; i<ndim; i++) { 60dedccee8SHong Zhang if (dim[i] < 1) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_USER,"dim[%d]=%d must be > 0",i,dim[i]); 61dedccee8SHong Zhang N *= dim[i]; 62dedccee8SHong Zhang } 63dedccee8SHong Zhang 64785e854fSJed Brown ierr = PetscMalloc1(ndim,&fft->dim);CHKERRQ(ierr); 65dedccee8SHong Zhang ierr = PetscMemcpy(fft->dim,dim,ndim*sizeof(PetscInt));CHKERRQ(ierr); 66dedccee8SHong Zhang 67dedccee8SHong Zhang fft->ndim = ndim; 68dedccee8SHong Zhang fft->n = PETSC_DECIDE; 69dedccee8SHong Zhang fft->N = N; 700298fd71SBarry Smith fft->data = NULL; 71dedccee8SHong Zhang 72dedccee8SHong Zhang ierr = MatSetType(FFT,mattype);CHKERRQ(ierr); 7326fbe8dcSKarl Rupp 74dedccee8SHong Zhang FFT->ops->destroy = MatDestroy_FFT; 75dedccee8SHong Zhang 76dedccee8SHong Zhang /* get runtime options */ 77ce94432eSBarry Smith ierr = PetscOptionsBegin(PetscObjectComm((PetscObject)FFT),((PetscObject)FFT)->prefix,"FFT Options","Mat");CHKERRQ(ierr); 78dedccee8SHong Zhang PetscOptionsEnd(); 79dedccee8SHong Zhang 80dedccee8SHong Zhang *A = FFT; 81dedccee8SHong Zhang PetscFunctionReturn(0); 82dedccee8SHong Zhang } 83