package/gmf
1
0
Fork 0
mirror of https://github.com/3d0c/gmf synced 2025-12-24 10:40:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

AVFilter, examples

Browse Source
This commit is contained in:
alex
2019-01-07 17:37:35 +03:00
parent 91aa3b021c
commit bbdeeb1452
4 changed files with 0 additions and 220 deletions
Download Patch File Download Diff File Expand all files Collapse all files

0
examples/opencv.go
View File

198
examples/overlay.go
View File

@@ -1,198 +0,0 @@
package main
/*
#cgo pkg-config: libavcodec libavutil
#cgo CFLAGS: -I/usr/local/include/opencv
#cgo CFLAGS: -I/usr/local/include/opencv2
#cgo LDFLAGS: -lopencv_imgproc -lopencv_core -lopencv_highgui
#include <stdio.h>
#include <strings.h>
#include <stdlib.h>
#include <stdint.h>
#include "libavcodec/avcodec.h"
#include "libavutil/frame.h"
#include "opencv2/highgui/highgui_c.h"
#include "cv.h"
typedef struct {
unsigned char *data;
unsigned int length;
} Blob;
void blender(AVFrame *frame, Blob *imgdata, CvRect *roi) {
// init backgroud
int flen = frame->width*frame->height*3;
CvMat *bgMat = cvCreateMat(1, flen, CV_8UC1);
bgMat->data.ptr = frame->data[0];
// init foreground
CvMat *fgMat = cvCreateMat(1, imgdata->length, CV_8UC1);
fgMat->data.ptr = imgdata->data;
// CvMat *fgMat = cvDecodeImageM(fgBuf, CV_LOAD_IMAGE_UNCHANGED);
// cvReleaseMat(&fgBuf);
int loc_x = (roi->x > 0) ? roi->x : 0;
int loc_y = (roi->y > 0) ? roi->y : 0;
int y, x, c;
for(y = loc_y; y < bgMat->rows; y++) {
int fY = y - roi->y;
if(fY >= fgMat->rows) {
break;
}
for(x = loc_x; x < bgMat->cols; x++) {
int fX = x - roi->x;
if(fX >= fgMat->cols) {
break;
}
double opacity;
opacity = ((double)fgMat->data.ptr[fY * fgMat->step + fX * 3 + 3]) / 255.;
for(c = 0; opacity > 0 && c < bgMat->nChannels; c++) {
unsigned char foregroundPx = fgMat->data.ptr[fY * fgMat->step + fX * fgMat->nChannels + c];
unsigned char backgroundPx = bgMat->data.ptr[y * bgMat->step + x * bgMat->nChannels + c];
resultMat->data.ptr[y*resultMat->step + bgMat->nChannels*x + c] = backgroundPx * (1.-opacity) + foregroundPx * opacity;
}
}
}
}
*/
import "C"
import (
"errors"
"flag"
"fmt"
"io/ioutil"
"log"
"os"
"github.com/3d0c/gmf"
)
type CvRect _Ctype_CvRect
func main() {
var (
img, dst string
octx *gmf.FmtCtx
swsCtx, swsYUV *gmf.SwsCtx
err error
width, height int = 640, 480
)
flag.StringVar(&img, "img", "", "overlay with image, e.g. -img=logo.png")
flag.StringVar(&dst, "dst", "", "destination file, e.g. -dst=result.mp4")
flag.Parse()
if img == "" || dst == "" {
log.Fatalf("-img or -dst not set\n")
}
f, err := os.OpenFile(img, 0644, os.O_RDONLY)
if err != nil {
log.Fatalf("%s\n", err)
}
defer f.Close()
imgdata := ioutil.ReadAll(f)
if octx, err = gmf.NewOutputCtx(dst); err != nil {
log.Fatalf("%s\n", err)
}
codec, err := gmf.FindEncoder("libx264")
if err != nil {
log.Fatalf("%s\n", err)
}
encCtx := NewCodecCtx(codec)
if encCtx == nil {
log.Fatalf("unable to create codec context\n")
}
defer encCtx.Free()
encCtx.
SetBitRate(100000).
SetWidth(width).
SetHeight(height).
SetTimeBase(gmf.AVR{Num: 1, Den: 25}).
SetPixFmt(gmf.AV_PIX_FMT_YUV420P).
SetProfile(gmf.FF_PROFILE_MPEG4_SIMPLE).
SetMbDecision(gmf.FF_MB_DECISION_RD)
ost := octx.NewStream(codec)
if ost == nil {
log.Fatalf("Unable to create output stream for %s\n", codec.Name())
}
defer ost.Free()
ost.SetCodecCtx(encCtx)
if swsCtx = NewSwsCtxExplicit(width, height, encCtx.PixFmt(), width, height, gmf.AV_PIX_FMT_RGB24, 0); swsCtx == nil {
log.Fatalf("Unable to create sws context\n")
}
if swsYUV = NewSwsCtxExplicit(width, height, gmf.AV_PIX_FMT_RGB24, width, height, encCtx.PixFmt(), 0); swsCtx == nil {
log.Fatalf("Unable to create sws context\n")
}
if err = octx.WriteHeader(); err != nil {
log.Fatalf("%s\n", err)
}
var (
srcFrame *gmf.Frame
rgbFrame *gmf.Frame = gmf.NewFrame().SetWidth(width).SetHeight(heigh).SetFormat(gmf.AV_PIX_FMT_BGR24)
)
if err = rgbFrame.ImgAlloc(); err != nil {
log.Fatalf("%s\n", err)
}
for frame = range GenSyntVideoNewFrame(videoEncCtx.Width(), videoEncCtx.Height(), videoEncCtx.PixFmt()) {
frame.SetPts(i)
swsCtx.Scale(srcFrame, rgbFrame)
blender(rgbFrame, (*C.uchar)(unsafe.Pointer(&imgdata[0])), &CvRect{0, 0, 0, 0})
if p, err := frame.Encode(ost.CodecCtx()); p != nil {
if p.Pts() != AV_NOPTS_VALUE {
p.SetPts(RescaleQ(p.Pts(), ost.CodecCtx().TimeBase(), ost.TimeBase()))
}
if p.Dts() != AV_NOPTS_VALUE {
p.SetDts(RescaleQ(p.Dts(), ost.CodecCtx().TimeBase(), ost.TimeBase()))
}
if err := outputCtx.WritePacket(p); err != nil {
fatal(err)
}
n++
log.Printf("Write frame=%d size=%v pts=%v dts=%v\n", frame.Pts(), p.Size(), p.Pts(), p.Dts())
Release(p)
} else if err != nil {
fatal(err)
}
Release(frame)
i++
}
}

17
filter.go
View File

@@ -9,23 +9,6 @@ package gmf
#include <libavfilter/buffersrc.h>
#include <libavutil/pixdesc.h>
int gmf_create_filter(AVFilterContext **filt_ctx, const AVFilter *filt, const char *name, const char *args, void *opaque, AVFilterGraph *graph_ctx, int i) {
return avfilter_graph_create_filter(&filt_ctx[i], filt, name, args, opaque, graph_ctx);
}
int gmf_av_buffersrc_add_frame_flags(AVFilterContext **filt_ctx, AVFrame *frame, int flags, int i) {
return av_buffersrc_add_frame_flags(filt_ctx[i], frame, flags);
}
// XXX PTS! Pass actual pts instead of 0.
int gmf_av_buffersrc_close(AVFilterContext **filt_ctx, int i) {
return av_buffersrc_close(filt_ctx[i], 0, AV_BUFFERSRC_FLAG_PUSH);
}
AVFilterContext *gmf_get_current(AVFilterContext **filt_ctx, int i) {
return filt_ctx[i];
}
*/
import "C"

5
frame.go
View File

@@ -229,8 +229,3 @@ func (f *Frame) IsNil() bool {
func (f *Frame) Time(timebase AVRational) int {
return int(float64(timebase.AVR().Num) / float64(timebase.AVR().Den) * float64(f.Pts()))
}
// temp. export unexported. remove it
func (f *Frame) GetAVFrame() *C.struct_AVFrame {
return f.avFrame
}