72 lines
1.3 KiB
Go
72 lines
1.3 KiB
Go
package ffmpeg
|
|
|
|
/*
|
|
#cgo LDFLAGS: -lavformat -lavutil -lavcodec -lavresample
|
|
#include "ffmpeg.h"
|
|
void ffinit() {
|
|
av_register_all();
|
|
}
|
|
*/
|
|
import "C"
|
|
import (
|
|
"runtime"
|
|
"unsafe"
|
|
)
|
|
|
|
const (
|
|
QUIET = int(C.AV_LOG_QUIET)
|
|
PANIC = int(C.AV_LOG_PANIC)
|
|
FATAL = int(C.AV_LOG_FATAL)
|
|
ERROR = int(C.AV_LOG_ERROR)
|
|
WARNING = int(C.AV_LOG_WARNING)
|
|
INFO = int(C.AV_LOG_INFO)
|
|
VERBOSE = int(C.AV_LOG_VERBOSE)
|
|
DEBUG = int(C.AV_LOG_DEBUG)
|
|
TRACE = int(C.AV_LOG_TRACE)
|
|
)
|
|
|
|
func HasEncoder(name string) bool {
|
|
return C.avcodec_find_encoder_by_name(C.CString(name)) != nil
|
|
}
|
|
|
|
func HasDecoder(name string) bool {
|
|
return C.avcodec_find_decoder_by_name(C.CString(name)) != nil
|
|
}
|
|
|
|
//func EncodersList() []string
|
|
//func DecodersList() []string
|
|
|
|
func SetLogLevel(level int) {
|
|
C.av_log_set_level(C.int(level))
|
|
}
|
|
|
|
func init() {
|
|
C.ffinit()
|
|
}
|
|
|
|
type ffctx struct {
|
|
ff C.FFCtx
|
|
}
|
|
|
|
func newFFCtxByCodec(codec *C.AVCodec) (ff *ffctx, err error) {
|
|
ff = &ffctx{}
|
|
ff.ff.codec = codec
|
|
ff.ff.codecCtx = C.avcodec_alloc_context3(codec)
|
|
runtime.SetFinalizer(ff, freeFFCtx)
|
|
return
|
|
}
|
|
|
|
func freeFFCtx(self *ffctx) {
|
|
ff := &self.ff
|
|
if ff.frame != nil {
|
|
C.av_frame_free(&ff.frame)
|
|
ff.frame = nil
|
|
}
|
|
if ff.codecCtx != nil {
|
|
C.avcodec_close(ff.codecCtx)
|
|
C.av_free(unsafe.Pointer(ff.codecCtx))
|
|
ff.codecCtx = nil
|
|
}
|
|
}
|
|
|