neko/internal/types/capture.go

151 lines
3.5 KiB
Go
Raw Normal View History

2020-11-02 04:09:48 +13:00
package types
2021-02-02 11:50:18 +13:00
import (
"context"
2021-08-30 03:09:13 +12:00
"errors"
2021-03-29 11:58:51 +13:00
"fmt"
2021-03-30 11:37:06 +13:00
"math"
"strings"
2021-03-29 11:58:51 +13:00
"github.com/PaesslerAG/gval"
2021-03-30 11:37:06 +13:00
"github.com/pion/webrtc/v3/pkg/media"
2021-02-02 11:50:18 +13:00
"demodesk/neko/internal/types/codec"
)
2021-08-30 03:09:13 +12:00
var (
ErrCapturePipelineAlreadyExists = errors.New("capture pipeline already exists")
)
2021-02-02 11:50:18 +13:00
type Sample media.Sample
2020-11-02 04:09:48 +13:00
type BroadcastManager interface {
Start(url string) error
Stop()
2021-02-06 02:03:53 +13:00
Started() bool
Url() string
}
2021-01-23 06:13:32 +13:00
type ScreencastManager interface {
Enabled() bool
2021-01-24 03:17:52 +13:00
Started() bool
Image() ([]byte, error)
2021-01-23 06:13:32 +13:00
}
type StreamManager interface {
Codec() codec.RTPCodec
2021-10-02 00:46:10 +13:00
AddListener(listener *func(sample Sample)) error
RemoveListener(listener *func(sample Sample)) error
MoveListenerTo(listener *func(sample Sample), targetStream StreamManager) error
2021-09-27 11:50:49 +13:00
ListenersCount() int
2021-02-06 02:03:53 +13:00
Started() bool
}
2020-11-02 04:09:48 +13:00
type CaptureManager interface {
Start()
Shutdown() error
Broadcast() BroadcastManager
2021-01-23 06:13:32 +13:00
Screencast() ScreencastManager
Audio() StreamManager
2021-02-06 06:07:58 +13:00
Video(videoID string) (StreamManager, bool)
VideoIDs() []string
2020-11-02 04:09:48 +13:00
}
2021-03-29 11:58:51 +13:00
type VideoConfig struct {
Width string `mapstructure:"width"` // expression
Height string `mapstructure:"height"` // expression
Fps string `mapstructure:"fps"` // expression
GstPrefix string `mapstructure:"gst_prefix"` // pipeline prefix, starts with !
2021-03-30 11:37:06 +13:00
GstEncoder string `mapstructure:"gst_encoder"` // gst encoder name
2021-03-29 11:58:51 +13:00
GstParams map[string]string `mapstructure:"gst_params"` // map of expressions
GstSuffix string `mapstructure:"gst_suffix"` // pipeline suffix, starts with !
GstPipeline string `mapstructure:"gst_pipeline"` // whole pipeline as a string
2021-03-29 11:58:51 +13:00
}
func (config *VideoConfig) GetPipeline(screen ScreenSize) (string, error) {
values := map[string]interface{}{
"width": screen.Width,
"height": screen.Height,
"fps": screen.Rate,
}
language := []gval.Language{
gval.Function("round", func(args ...interface{}) (interface{}, error) {
return (int)(math.Round(args[0].(float64))), nil
}),
}
// get fps pipeline
2021-03-30 09:59:07 +13:00
fpsPipeline := "! video/x-raw ! videoconvert ! queue"
2021-03-29 11:58:51 +13:00
if config.Fps != "" {
eval, err := gval.Full(language...).NewEvaluable(config.Fps)
2021-03-29 11:58:51 +13:00
if err != nil {
return "", err
}
val, err := eval.EvalFloat64(context.Background(), values)
if err != nil {
return "", err
2021-03-29 11:58:51 +13:00
}
fpsPipeline = fmt.Sprintf("! video/x-raw,framerate=%d/100 ! videoconvert ! queue", int(val*100))
2021-03-29 11:58:51 +13:00
}
// get scale pipeline
scalePipeline := ""
if config.Width != "" && config.Height != "" {
eval, err := gval.Full(language...).NewEvaluable(config.Width)
if err != nil {
return "", err
}
w, err := eval.EvalInt(context.Background(), values)
2021-03-29 11:58:51 +13:00
if err != nil {
return "", err
}
eval, err = gval.Full(language...).NewEvaluable(config.Height)
2021-03-29 11:58:51 +13:00
if err != nil {
return "", err
}
h, err := eval.EvalInt(context.Background(), values)
if err != nil {
return "", err
2021-03-29 11:58:51 +13:00
}
scalePipeline = fmt.Sprintf("! videoscale ! video/x-raw,width=%d,height=%d ! queue", w, h)
2021-03-29 11:58:51 +13:00
}
// get encoder pipeline
encPipeline := fmt.Sprintf("! %s", config.GstEncoder)
for key, expr := range config.GstParams {
if expr == "" {
continue
}
val, err := gval.Evaluate(expr, values, language...)
if err != nil {
return "", err
}
if val != nil {
encPipeline += fmt.Sprintf(" %s=%v", key, val)
} else {
encPipeline += fmt.Sprintf(" %s=%s", key, expr)
}
}
// join strings with space
return strings.Join([]string{
fpsPipeline,
scalePipeline,
config.GstPrefix,
encPipeline,
config.GstSuffix,
2021-03-30 11:37:06 +13:00
}[:], " "), nil
2021-03-29 11:58:51 +13:00
}