예제 #1
0
func Alloc1d(n int) []complex128 {
	buffer := (unsafe.Pointer)(C.fftw_malloc((C.size_t)(16 * n)))
	var slice []complex128
	header := (*reflect.SliceHeader)(unsafe.Pointer(&slice))
	*header = reflect.SliceHeader{uintptr(buffer), n, n}
	return slice
}
예제 #2
0
파일: fftw.go 프로젝트: jtc25/go-fftw
func Alloc1d(n int) []complex128 {
	// Try to allocate memory.
	buffer, err := C.fftw_malloc(C.size_t(16 * n))
	if err != nil {
		// If malloc failed, invoke garbage collector and try again.
		runtime.GC()
		buffer, err = C.fftw_malloc(C.size_t(16 * n))
		if err != nil {
			// If it still failed, then panic.
			panic(fmt.Sprint("Could not fftw_malloc for ", n, " elements: ", err))
		}
	}
	// Create a slice header for the memory.
	var slice []complex128
	header := (*reflect.SliceHeader)(unsafe.Pointer(&slice))
	header.Data = uintptr(buffer)
	header.Len = n
	header.Cap = n
	// In the spirit of Go, initialize all memory to zero.
	for i := 0; i < n; i++ {
		slice[i] = 0
	}
	return slice
}
예제 #3
0
파일: fftw.go 프로젝트: delfvad/hi
// Alloc2D allocates a complex matrix for fftw
func Alloc2D(m, n int) ([][]complex128, error) {
	if m <= 0 || n <= 0 {
		return [][]complex128{}, fmt.Errorf("positive dimensions required")
	}

	samples := m * n
	var slice []complex128
	header := (*reflect.SliceHeader)(unsafe.Pointer(&slice))
	*header = reflect.SliceHeader{
		Data: uintptr(C.fftw_malloc((C.size_t)(16 * samples))),
		Len:  samples,
		Cap:  samples,
	}

	r := make([][]complex128, m)
	for i := range r {
		r[i] = slice[i*n : (i+1)*n]
	}

	return r, nil
}