vector.int32.go

package blackcl

/*
#cgo CFLAGS: -I CL
#cgo !darwin LDFLAGS: -lOpenCL
#cgo darwin LDFLAGS: -framework OpenCL

#define CL_SILENCE_DEPRECATION
#ifdef __APPLE__
#include <OpenCL/opencl.h>
#else
#include <CL/cl.h>
#endif
*/
import "C"
import (
	"errors"
	"unsafe"
)

//VectorInt32 is a memory buffer on device that holds []int32
type VectorInt32 struct {
	buf *buffer
}

//Length the length of the vector
func (v *VectorInt32) Length() int {
	return v.buf.size / int32CLSize
}

//Release releases the buffer on the device
func (v *VectorInt32) Release() error {
	return v.buf.Release()
}

//NewVectorInt32 allocates new vector buffer with specified length
func (d *Device) NewVectorInt32(length int) (*VectorInt32, error) {
	size := length * int32CLSize
	buf, err := newBuffer(d, size)
	if err != nil {
		return nil, err
	}
	return &VectorInt32{buf: &buffer{memobj: buf, device: d, size: size}}, nil
}

//NewVectorInt32 allocates new vector buffer with specified length
func (d *Device) NewVectorInt32With(data []int32) (*VectorInt32, error) {
	size := len(data) * int32CLSize
	buf, err := newBuffer(d, size)
	if err != nil {
		return nil, err
	}
	v := &VectorInt32{buf: &buffer{memobj: buf, device: d, size: size}}
	if err = <-v.Copy(data); err != nil {
		return nil, err
	}
	return v, nil
}

//Copy copies the float32 data from host data to device buffer
//it's a non-blocking call, channel will return an error or nil if the data transfer is complete
func (v *VectorInt32) Copy(data []int32) <-chan error {
	if v.Length() != len(data) {
		ch := make(chan error, 1)
		ch <- errors.New("vector length not equal to data length")
		return ch
	}
	return v.buf.copy(len(data)*int32CLSize, unsafe.Pointer(&data[0]))
}

//Data gets int32 data from device, it's a blocking call
func (v *VectorInt32) Data() ([]int32, error) {
	data := make([]int32, v.buf.size/int32CLSize)
	err := toErr(C.clEnqueueReadBuffer(
		v.buf.device.queue,
		v.buf.memobj,
		C.CL_TRUE,
		0,
		C.size_t(v.buf.size),
		unsafe.Pointer(&data[0]),
		0,
		nil,
		nil,
	))
	if err != nil {
		return nil, err
	}
	return data, nil
}

//Map applies an map kernel on all elements of the vector
func (v *VectorInt32) Map(k *Kernel) <-chan error {
	return k.Global(v.Length()).Local(1).Run(v)
}