Project restructure (#18)
* First pass at restructuring the project; untested but it does compile * Restructure builds and runs 🎉 * Add the dist folder to the gitignore * Update core/playlist/monitor.go * golint and reorganize the monitor.go file Co-authored-by: Gabe Kangas <gabek@real-ity.com>
This commit is contained in:
Bradley Hilton
200
core/ffmpeg/ffmpeg.go
Normal file
200
core/ffmpeg/ffmpeg.go
Normal file
@@ -0,0 +1,200 @@
|
||||
package ffmpeg
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"math"
|
||||
"os"
|
||||
"os/exec"
|
||||
"path"
|
||||
"strconv"
|
||||
"strings"
|
||||
|
||||
log "github.com/sirupsen/logrus"
|
||||
|
||||
"github.com/gabek/owncast/config"
|
||||
"github.com/gabek/owncast/utils"
|
||||
)
|
||||
|
||||
//ShowStreamOfflineState generates and shows the stream's offline state
|
||||
func ShowStreamOfflineState() error {
|
||||
log.Println("----- Stream offline! Showing offline state!")
|
||||
|
||||
var outputDir = config.Config.PublicHLSPath
|
||||
var variantPlaylistPath = config.Config.PublicHLSPath
|
||||
|
||||
if config.Config.IPFS.Enabled || config.Config.S3.Enabled {
|
||||
outputDir = config.Config.PrivateHLSPath
|
||||
variantPlaylistPath = config.Config.PrivateHLSPath
|
||||
}
|
||||
|
||||
outputDir = path.Join(outputDir, "%v")
|
||||
var variantPlaylistName = path.Join(variantPlaylistPath, "%v", "stream.m3u8")
|
||||
|
||||
var videoMaps = make([]string, 0)
|
||||
var streamMaps = make([]string, 0)
|
||||
var videoMapsString = ""
|
||||
var streamMappingString = ""
|
||||
if config.Config.VideoSettings.EnablePassthrough || len(config.Config.VideoSettings.StreamQualities) == 0 {
|
||||
log.Println("Enabling passthrough video for offline state")
|
||||
videoMapsString = "-b:v 1200k -b:a 128k" // Since we're compositing multiple sources we can't infer bitrate, so pick something reasonable.
|
||||
streamMaps = append(streamMaps, fmt.Sprintf("v:%d", 0))
|
||||
} else {
|
||||
for index, quality := range config.Config.VideoSettings.StreamQualities {
|
||||
maxRate := math.Floor(float64(quality.Bitrate) * 0.8)
|
||||
videoMaps = append(videoMaps, fmt.Sprintf("-map v:0 -c:v:%d libx264 -b:v:%d %dk -maxrate %dk -bufsize %dk", index, index, int(quality.Bitrate), int(maxRate), int(maxRate)))
|
||||
streamMaps = append(streamMaps, fmt.Sprintf("v:%d", index))
|
||||
videoMapsString = strings.Join(videoMaps, " ")
|
||||
}
|
||||
}
|
||||
|
||||
framerate := 25
|
||||
|
||||
streamMappingString = "-var_stream_map \"" + strings.Join(streamMaps, " ") + "\""
|
||||
|
||||
ffmpegFlags := []string{
|
||||
"-hide_banner",
|
||||
// "-stream_loop 100",
|
||||
// "-fflags", "+genpts",
|
||||
"-i", config.Config.VideoSettings.OfflineImage,
|
||||
"-i", "webroot/thumbnail.jpg",
|
||||
"-filter_complex", "\"[0:v]scale=2640:2360[bg];[bg][1:v]overlay=200:250:enable='between(t,0,3)'\"",
|
||||
videoMapsString, // All the different video variants
|
||||
"-f hls",
|
||||
// "-hls_list_size " + strconv.Itoa(config.Config.Files.MaxNumberInPlaylist),
|
||||
"-hls_time 4", // + strconv.Itoa(config.Config.VideoSettings.ChunkLengthInSeconds),
|
||||
"-hls_playlist_type", "event",
|
||||
"-master_pl_name", "stream.m3u8",
|
||||
"-strftime 1",
|
||||
"-use_localtime 1",
|
||||
"-hls_flags temp_file",
|
||||
"-tune", "zerolatency",
|
||||
"-g " + strconv.Itoa(framerate*2), " -keyint_min " + strconv.Itoa(framerate*2), // multiply your output frame rate * 2. For example, if your input is -framerate 30, then use -g 60
|
||||
"-framerate " + strconv.Itoa(framerate),
|
||||
"-preset " + config.Config.VideoSettings.EncoderPreset,
|
||||
"-sc_threshold 0", // don't create key frames on scene change - only according to -g
|
||||
"-profile:v", "main", // Main – for standard definition (SD) to 640×480, High – for high definition (HD) to 1920×1080
|
||||
// "-movflags +faststart",
|
||||
"-pix_fmt yuv420p",
|
||||
|
||||
streamMappingString,
|
||||
"-hls_segment_filename " + path.Join(outputDir, "offline-%s.ts"),
|
||||
// "-s", "720x480", // size
|
||||
variantPlaylistName,
|
||||
}
|
||||
|
||||
ffmpegFlagsString := strings.Join(ffmpegFlags, " ")
|
||||
|
||||
ffmpegCmd := config.Config.FFMpegPath + " " + ffmpegFlagsString
|
||||
|
||||
// log.Println(ffmpegCmd)
|
||||
|
||||
_, err := exec.Command("sh", "-c", ffmpegCmd).Output()
|
||||
|
||||
return err
|
||||
}
|
||||
|
||||
//Start starts the ffmpeg process
|
||||
func Start() error {
|
||||
var outputDir = config.Config.PublicHLSPath
|
||||
var variantPlaylistPath = config.Config.PublicHLSPath
|
||||
|
||||
if config.Config.IPFS.Enabled || config.Config.S3.Enabled {
|
||||
outputDir = config.Config.PrivateHLSPath
|
||||
variantPlaylistPath = config.Config.PrivateHLSPath
|
||||
}
|
||||
|
||||
outputDir = path.Join(outputDir, "%v")
|
||||
var variantPlaylistName = path.Join(variantPlaylistPath, "%v", "stream.m3u8")
|
||||
|
||||
log.Printf("Starting transcoder saving to /%s.", variantPlaylistName)
|
||||
pipePath := utils.GetTemporaryPipePath()
|
||||
|
||||
var videoMaps = make([]string, 0)
|
||||
var streamMaps = make([]string, 0)
|
||||
var audioMaps = make([]string, 0)
|
||||
var videoMapsString = ""
|
||||
var audioMapsString = ""
|
||||
var streamMappingString = ""
|
||||
var profileString = ""
|
||||
|
||||
if config.Config.VideoSettings.EnablePassthrough || len(config.Config.VideoSettings.StreamQualities) == 0 {
|
||||
log.Println("Enabling passthrough video for stream")
|
||||
streamMaps = append(streamMaps, fmt.Sprintf("v:%d,a:%d", 0, 0))
|
||||
videoMaps = append(videoMaps, "-map v:0 -c:v copy")
|
||||
videoMapsString = strings.Join(videoMaps, " ")
|
||||
audioMaps = append(audioMaps, "-map a:0")
|
||||
audioMapsString = strings.Join(audioMaps, " ") + " -c:a copy" // Pass through audio for all the variants, don't reencode
|
||||
|
||||
} else {
|
||||
for index, quality := range config.Config.VideoSettings.StreamQualities {
|
||||
maxRate := math.Floor(float64(quality.Bitrate) * 0.8)
|
||||
videoMaps = append(videoMaps, fmt.Sprintf("-map v:0 -c:v:%d libx264 -b:v:%d %dk -maxrate %dk -bufsize %dk", index, index, int(quality.Bitrate), int(maxRate), int(maxRate)))
|
||||
streamMaps = append(streamMaps, fmt.Sprintf("v:%d,a:%d", index, index))
|
||||
videoMapsString = strings.Join(videoMaps, " ")
|
||||
audioMaps = append(audioMaps, "-map a:0")
|
||||
audioMapsString = strings.Join(audioMaps, " ") + " -c:a copy" // Pass through audio for all the variants, don't reencode
|
||||
profileString = "-profile:v high" // Main – for standard definition (SD) to 640×480, High – for high definition (HD) to 1920×1080
|
||||
}
|
||||
}
|
||||
|
||||
framerate := 25
|
||||
|
||||
streamMappingString = "-var_stream_map \"" + strings.Join(streamMaps, " ") + "\""
|
||||
ffmpegFlags := []string{
|
||||
"-hide_banner",
|
||||
// "-re",
|
||||
"-fflags", "+genpts",
|
||||
"-i pipe:",
|
||||
// "-vf scale=900:-2", // Re-enable in the future with a config to togging resizing?
|
||||
// "-sws_flags fast_bilinear",
|
||||
videoMapsString, // All the different video variants
|
||||
audioMapsString,
|
||||
"-master_pl_name stream.m3u8",
|
||||
"-framerate " + strconv.Itoa(framerate),
|
||||
"-g " + strconv.Itoa(framerate*2), " -keyint_min " + strconv.Itoa(framerate*2), // multiply your output frame rate * 2. For example, if your input is -framerate 30, then use -g 60
|
||||
// "-r 25",
|
||||
"-preset " + config.Config.VideoSettings.EncoderPreset,
|
||||
"-sc_threshold 0", // don't create key frames on scene change - only according to -g
|
||||
profileString,
|
||||
"-movflags +faststart",
|
||||
"-pix_fmt yuv420p",
|
||||
"-f hls",
|
||||
"-hls_list_size " + strconv.Itoa(config.Config.Files.MaxNumberInPlaylist),
|
||||
"-hls_delete_threshold 10", // Keep 10 unreferenced segments on disk before they're deleted.
|
||||
"-hls_time " + strconv.Itoa(config.Config.VideoSettings.ChunkLengthInSeconds),
|
||||
"-strftime 1",
|
||||
"-use_localtime 1",
|
||||
"-hls_segment_filename " + path.Join(outputDir, "stream-%Y%m%d-%s.ts"),
|
||||
"-hls_flags delete_segments+program_date_time+temp_file",
|
||||
"-tune zerolatency",
|
||||
// "-s", "720x480", // size
|
||||
|
||||
streamMappingString,
|
||||
variantPlaylistName,
|
||||
}
|
||||
|
||||
ffmpegFlagsString := strings.Join(ffmpegFlags, " ")
|
||||
|
||||
ffmpegCmd := "cat " + pipePath + " | " + config.Config.FFMpegPath + " " + ffmpegFlagsString
|
||||
|
||||
// fmt.Println(ffmpegCmd)
|
||||
|
||||
_, err := exec.Command("sh", "-c", ffmpegCmd).Output()
|
||||
|
||||
return err
|
||||
}
|
||||
|
||||
//WritePlaylist writes the playlist to disk
|
||||
func WritePlaylist(data string, filePath string) error {
|
||||
f, err := os.Create(filePath)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer f.Close()
|
||||
|
||||
if _, err := f.WriteString(data); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
93
core/ffmpeg/thumbnailGenerator.go
Normal file
93
core/ffmpeg/thumbnailGenerator.go
Normal file
@@ -0,0 +1,93 @@
|
||||
package ffmpeg
|
||||
|
||||
import (
|
||||
"io/ioutil"
|
||||
"os/exec"
|
||||
"path"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
log "github.com/sirupsen/logrus"
|
||||
|
||||
"github.com/gabek/owncast/config"
|
||||
)
|
||||
|
||||
//StartThumbnailGenerator starts generating thumbnails
|
||||
func StartThumbnailGenerator(chunkPath string) {
|
||||
// Every 20 seconds create a thumbnail from the most
|
||||
// recent video segment.
|
||||
ticker := time.NewTicker(20 * time.Second)
|
||||
quit := make(chan struct{})
|
||||
|
||||
go func() {
|
||||
for {
|
||||
select {
|
||||
case <-ticker.C:
|
||||
if err := fireThumbnailGenerator(chunkPath); err != nil {
|
||||
log.Errorln("Unable to generate thumbnail:", err)
|
||||
}
|
||||
case <-quit:
|
||||
//TODO: evaluate if this is ever stopped
|
||||
log.Println("thumbnail generator has stopped")
|
||||
ticker.Stop()
|
||||
return
|
||||
}
|
||||
}
|
||||
}()
|
||||
}
|
||||
|
||||
func fireThumbnailGenerator(chunkPath string) error {
|
||||
// JPG takes less time to encode than PNG
|
||||
outputFile := path.Join("webroot", "thumbnail.jpg")
|
||||
|
||||
framePath := path.Join(chunkPath, "0")
|
||||
files, err := ioutil.ReadDir(framePath)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
var modTime time.Time
|
||||
var names []string
|
||||
for _, fi := range files {
|
||||
if path.Ext(fi.Name()) != ".ts" {
|
||||
continue
|
||||
}
|
||||
|
||||
if fi.Mode().IsRegular() {
|
||||
if !fi.ModTime().Before(modTime) {
|
||||
if fi.ModTime().After(modTime) {
|
||||
modTime = fi.ModTime()
|
||||
names = names[:0]
|
||||
}
|
||||
names = append(names, fi.Name())
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if len(names) == 0 {
|
||||
return nil
|
||||
}
|
||||
|
||||
mostRecentFile := path.Join(framePath, names[0])
|
||||
|
||||
thumbnailCmdFlags := []string{
|
||||
config.Config.FFMpegPath,
|
||||
"-y", // Overwrite file
|
||||
"-threads 1", // Low priority processing
|
||||
"-t 1", // Pull from frame 1
|
||||
"-i", mostRecentFile, // Input
|
||||
"-f image2", // format
|
||||
"-vframes 1", // Single frame
|
||||
outputFile,
|
||||
}
|
||||
|
||||
ffmpegCmd := strings.Join(thumbnailCmdFlags, " ")
|
||||
|
||||
// fmt.Println(ffmpegCmd)
|
||||
|
||||
if _, err := exec.Command("sh", "-c", ffmpegCmd).Output(); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
32
core/ffmpeg/verifyInstall.go
Normal file
32
core/ffmpeg/verifyInstall.go
Normal file
@@ -0,0 +1,32 @@
|
||||
package ffmpeg
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"fmt"
|
||||
"os"
|
||||
)
|
||||
|
||||
//VerifyFFMpegPath verifies that the path exists, is a file, and is executable
|
||||
func VerifyFFMpegPath(path string) error {
|
||||
stat, err := os.Stat(path)
|
||||
|
||||
if os.IsNotExist(err) {
|
||||
return errors.New("ffmpeg path does not exist")
|
||||
}
|
||||
|
||||
if err != nil {
|
||||
return fmt.Errorf("error while verifying the ffmpeg path: %s", err.Error())
|
||||
}
|
||||
|
||||
if stat.IsDir() {
|
||||
return errors.New("ffmpeg path can not be a folder")
|
||||
}
|
||||
|
||||
mode := stat.Mode()
|
||||
//source: https://stackoverflow.com/a/60128480
|
||||
if mode&0111 == 0 {
|
||||
return errors.New("ffmpeg path is not executable")
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
Reference in New Issue
Block a user