package goffmpeg

/*
#include <stdlib.h>
#include "libcffmpeg.h"
*/
import "C"
import "unsafe"

// BuildDecoder build decoder
func (h *GoFFMPEG) BuildDecoder() {
	C.wrap_fn_decoder(unsafe.Pointer(libcffmpeg), h.ffmpeg)
}

// GetYUV get yuv data
func (h *GoFFMPEG) GetYUV() ([]byte, int, int, int64) {
	var fid C.long
	var length C.int
	var srcW, srcH, srcF C.int

	p := C.wrap_fn_decoder_pic(unsafe.Pointer(libcffmpeg), h.ffmpeg, &srcW, &srcH, &srcF, &length, &fid)
	if p == nil || length == 0 || srcW == 0 || srcH == 0 {
		return nil, 0, 0, 0
	}
	defer C.free(unsafe.Pointer(p))

	return C.GoBytes(p, length), int(srcW), int(srcH), int64(fid)
}

// GetYUV2 get yuv data
func (h *GoFFMPEG) GetYUV2() (unsafe.Pointer, []byte, int, int, int64) {
	var fid C.long
	var length C.int
	var srcW, srcH, srcF C.int

	p := C.wrap_fn_decoder_pic(unsafe.Pointer(libcffmpeg), h.ffmpeg, &srcW, &srcH, &srcF, &length, &fid)
	if p == nil || length == 0 || srcW == 0 || srcH == 0 {
		return nil, nil, 0, 0, 0
	}
	const maxLen = 0x7fffffff
	size := int(length)
	data := (*[maxLen]byte)(unsafe.Pointer(p))[:size:size]

	return unsafe.Pointer(p), data, int(srcW), int(srcH), int64(fid)
}