Пример #1
0
func PlanDft2d(in, out [][]complex64, dir Direction, flag Flag) *Plan {
	// TODO: check that in and out have the same dimensions
	fftw_in := (*C.fftwf_complex)((unsafe.Pointer)(&in[0][0]))
	fftw_out := (*C.fftwf_complex)((unsafe.Pointer)(&out[0][0]))
	p := C.fftwf_plan_dft_2d((C.int)(len(in)), (C.int)(len(in[0])), fftw_in, fftw_out, C.int(dir), C.uint(flag))
	return newPlan(p)
}
Пример #2
0
func NewPlan2(in, out *Array2, dir Direction, flag Flag) *Plan {
	// TODO: check that in and out have the same dimensions
	n0, n1 := in.Dims()
	var (
		n0_   = C.int(n0)
		n1_   = C.int(n1)
		in_   = (*C.fftwf_complex)(unsafe.Pointer(in.ptr()))
		out_  = (*C.fftwf_complex)(unsafe.Pointer(out.ptr()))
		dir_  = C.int(dir)
		flag_ = C.uint(flag)
	)
	p := C.fftwf_plan_dft_2d(n0_, n1_, in_, out_, dir_, flag_)
	return &Plan{p}
}