iridium/neko
2
2
Fork 0
mirror of https://github.com/m1k1o/neko.git synced 2024-07-24 14:40:50 +12:00
Code Issues Packages Projects Releases Wiki Activity

gst extract appsink.

Browse Source
This commit is contained in:
Miroslav Šedivý 2021-11-28 22:19:06 +01:00
parent 7b2c1570bf
commit 710d0c9cd0
5 changed files with 34 additions and 31 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
internal/capture/gst/gst.c
View File

@ -8,14 +8,7 @@ void gstreamer_init(void) {
gst_init(NULL, NULL); gst_init(NULL, NULL);
} }
GMainLoop *gstreamer_send_main_loop = NULL; static gboolean gstreamer_bus_call(GstBus *bus, GstMessage *msg, gpointer data) {
void gstreamer_send_start_mainloop(void) {
gstreamer_send_main_loop = g_main_loop_new(NULL, FALSE);
g_main_loop_run(gstreamer_send_main_loop);
}
static gboolean gstreamer_send_bus_call(GstBus *bus, GstMessage *msg, gpointer data) {
switch (GST_MESSAGE_TYPE(msg)) { switch (GST_MESSAGE_TYPE(msg)) {
case GST_MESSAGE_EOS: case GST_MESSAGE_EOS:
@ -41,7 +34,7 @@ static gboolean gstreamer_send_bus_call(GstBus *bus, GstMessage *msg, gpointer d
return TRUE; return TRUE;
} }
GstFlowReturn gstreamer_send_new_sample_handler(GstElement *object, gpointer user_data) { static GstFlowReturn gstreamer_send_new_sample_handler(GstElement *object, gpointer user_data) {
GstSample *sample = NULL; GstSample *sample = NULL;
GstBuffer *buffer = NULL; GstBuffer *buffer = NULL;
gpointer copy = NULL; gpointer copy = NULL;
@ -61,27 +54,31 @@ GstFlowReturn gstreamer_send_new_sample_handler(GstElement *object, gpointer use
return GST_FLOW_OK; return GST_FLOW_OK;
} }
void gstreamer_send_start_pipeline(GstElement *pipeline, int pipelineId) { void gstreamer_pipeline_attach_appsink(GstElement *pipeline, char *sinkName, int pipelineId) {
SampleHandlerUserData *s = calloc(1, sizeof(SampleHandlerUserData)); SampleHandlerUserData *s = calloc(1, sizeof(SampleHandlerUserData));
s->pipelineId = pipelineId; s->pipelineId = pipelineId;
GstBus *bus = gst_pipeline_get_bus(GST_PIPELINE(pipeline)); GstElement *appsink = gst_bin_get_by_name(GST_BIN(pipeline), sinkName);
gst_bus_add_watch(bus, gstreamer_send_bus_call, NULL);
gst_object_unref(bus);
GstElement *appsink = gst_bin_get_by_name(GST_BIN(pipeline), "appsink");
g_object_set(appsink, "emit-signals", TRUE, NULL); g_object_set(appsink, "emit-signals", TRUE, NULL);
g_signal_connect(appsink, "new-sample", G_CALLBACK(gstreamer_send_new_sample_handler), s); g_signal_connect(appsink, "new-sample", G_CALLBACK(gstreamer_send_new_sample_handler), s);
gst_object_unref(appsink); gst_object_unref(appsink);
}
GstElement *gstreamer_pipeline_create(char *pipelineStr, GError **error) {
GstElement *pipeline = gst_parse_launch(pipelineStr, error);
GstBus *bus = gst_pipeline_get_bus(GST_PIPELINE(pipeline));
gst_bus_add_watch(bus, gstreamer_bus_call, NULL);
gst_object_unref(bus);
return pipeline;
}
void gstreamer_pipeline_play(GstElement *pipeline) {
gst_element_set_state(pipeline, GST_STATE_PLAYING); gst_element_set_state(pipeline, GST_STATE_PLAYING);
} }
void gstreamer_send_play_pipeline(GstElement *pipeline) { void gstreamer_pipeline_stop(GstElement *pipeline) {
gst_element_set_state(pipeline, GST_STATE_PLAYING);
}
void gstreamer_send_stop_pipeline(GstElement *pipeline) {
gst_element_set_state(pipeline, GST_STATE_NULL); gst_element_set_state(pipeline, GST_STATE_NULL);
gst_object_unref(pipeline); gst_object_unref(pipeline);
} }

13
internal/capture/gst/gst.go
View File

@ -41,7 +41,7 @@ func CreatePipeline(pipelineStr string) (*Pipeline, error) {
var gstPipeline *C.GstElement var gstPipeline *C.GstElement
var gstError *C.GError var gstError *C.GError
gstPipeline = C.gst_parse_launch(pipelineStrUnsafe, &gstError) gstPipeline = C.gstreamer_pipeline_create(pipelineStrUnsafe, &gstError)
if gstError != nil { if gstError != nil {
defer C.g_error_free(gstError) defer C.g_error_free(gstError)
@ -59,16 +59,19 @@ func CreatePipeline(pipelineStr string) (*Pipeline, error) {
return p, nil return p, nil
} }
func (p *Pipeline) Start() { func (p *Pipeline) AttachAppsink(sinkName string) {
C.gstreamer_send_start_pipeline(p.Pipeline, C.int(p.id)) sinkNameUnsafe := C.CString(sinkName)
defer C.free(unsafe.Pointer(sinkNameUnsafe))
C.gstreamer_pipeline_attach_appsink(p.Pipeline, sinkNameUnsafe, C.int(p.id))
} }
func (p *Pipeline) Play() { func (p *Pipeline) Play() {
C.gstreamer_send_play_pipeline(p.Pipeline) C.gstreamer_pipeline_play(p.Pipeline)
} }
func (p *Pipeline) Stop() { func (p *Pipeline) Stop() {
C.gstreamer_send_stop_pipeline(p.Pipeline) C.gstreamer_pipeline_stop(p.Pipeline)
} }
// gst-inspect-1.0 // gst-inspect-1.0

9
internal/capture/gst/gst.h
View File

@ -5,8 +5,9 @@
extern void goHandlePipelineBuffer(void *buffer, int bufferLen, int samples, int pipelineId); extern void goHandlePipelineBuffer(void *buffer, int bufferLen, int samples, int pipelineId);
void gstreamer_send_start_pipeline(GstElement *pipeline, int pipelineId); void gstreamer_pipeline_attach_appsink(GstElement *pipeline, char *sinkName, int pipelineId);
void gstreamer_send_play_pipeline(GstElement *pipeline); GstElement *gstreamer_pipeline_create(char *pipelineStr, GError **error);
void gstreamer_send_stop_pipeline(GstElement *pipeline); void gstreamer_pipeline_play(GstElement *pipeline);
void gstreamer_send_start_mainloop(void); void gstreamer_pipeline_stop(GstElement *pipeline);
void gstreamer_init(void); void gstreamer_init(void);

3
internal/capture/screencast.go
View File

@ -170,7 +170,8 @@ func (manager *ScreencastManagerCtx) createPipeline() error {
return err return err
} }
manager.pipeline.Start() manager.pipeline.AttachAppsink("appsink")
manager.pipeline.Play()
manager.sample = manager.pipeline.Sample manager.sample = manager.pipeline.Sample
manager.sampleUpdate <- struct{}{} manager.sampleUpdate <- struct{}{}

3
internal/capture/stream.go
View File

@ -247,7 +247,8 @@ func (manager *StreamManagerCtx) createPipeline() error {
return err return err
} }
manager.pipeline.Start() manager.pipeline.AttachAppsink("appsink")
manager.pipeline.Play()
manager.sample = manager.pipeline.Sample manager.sample = manager.pipeline.Sample
manager.sampleUpdate <- struct{}{} manager.sampleUpdate <- struct{}{}