mirror of
https://github.com/xfrr/goffmpeg.git
synced 2025-10-05 16:06:50 +08:00
Added input path to mediafile and seek using TS option for input
This commit is contained in:
Rohit Garg
@@ -1,87 +1,78 @@
|
||||
package transcoder
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"os"
|
||||
"goffmpeg/models"
|
||||
"os/exec"
|
||||
"fmt"
|
||||
"goffmpeg/ffmpeg"
|
||||
"goffmpeg/utils"
|
||||
"bytes"
|
||||
"encoding/json"
|
||||
"bufio"
|
||||
"strings"
|
||||
"regexp"
|
||||
"strconv"
|
||||
"errors"
|
||||
"os"
|
||||
"goffmpeg/models"
|
||||
"os/exec"
|
||||
"fmt"
|
||||
"goffmpeg/ffmpeg"
|
||||
"goffmpeg/utils"
|
||||
"bytes"
|
||||
"encoding/json"
|
||||
"bufio"
|
||||
"strings"
|
||||
"regexp"
|
||||
"strconv"
|
||||
)
|
||||
|
||||
type Transcoder struct {
|
||||
process *exec.Cmd
|
||||
inputPath string
|
||||
outputPath string
|
||||
mediafile *models.Mediafile
|
||||
configuration ffmpeg.Configuration
|
||||
process *exec.Cmd
|
||||
outputPath string
|
||||
mediafile *models.Mediafile
|
||||
configuration ffmpeg.Configuration
|
||||
}
|
||||
|
||||
func (t *Transcoder) SetProccess(v *exec.Cmd) {
|
||||
t.process = v
|
||||
}
|
||||
|
||||
func (t *Transcoder) SetInputPath(v string) {
|
||||
t.inputPath = v
|
||||
t.process = v
|
||||
}
|
||||
|
||||
func (t *Transcoder) SetOutputPath(v string) {
|
||||
t.outputPath = v
|
||||
t.outputPath = v
|
||||
}
|
||||
|
||||
func (t *Transcoder) SetMediaFile(v *models.Mediafile) {
|
||||
t.mediafile = v
|
||||
t.mediafile = v
|
||||
}
|
||||
|
||||
func (t *Transcoder) SetConfiguration(v ffmpeg.Configuration) {
|
||||
t.configuration = v
|
||||
t.configuration = v
|
||||
}
|
||||
|
||||
/*** GETTERS ***/
|
||||
|
||||
func (t Transcoder) Process() *exec.Cmd {
|
||||
return t.process
|
||||
}
|
||||
|
||||
func (t Transcoder) InputPath() string {
|
||||
return t.inputPath
|
||||
return t.process
|
||||
}
|
||||
|
||||
func (t Transcoder) OutputPath() string {
|
||||
return t.outputPath
|
||||
return t.outputPath
|
||||
}
|
||||
|
||||
func (t Transcoder) MediaFile() *models.Mediafile {
|
||||
return t.mediafile
|
||||
return t.mediafile
|
||||
}
|
||||
|
||||
func (t Transcoder) FFmpegExec() string {
|
||||
return t.configuration.FfmpegBin
|
||||
return t.configuration.FfmpegBin
|
||||
}
|
||||
|
||||
func (t Transcoder) FFprobeExec() string {
|
||||
return t.configuration.FfprobeBin
|
||||
return t.configuration.FfprobeBin
|
||||
}
|
||||
|
||||
func (t Transcoder) GetCommand() string {
|
||||
var rcommand string
|
||||
var rcommand string
|
||||
|
||||
rcommand = fmt.Sprintf("%s -y -i \"%s\" ", t.configuration.FfmpegBin, t.inputPath)
|
||||
rcommand = fmt.Sprintf("%s -y ", t.configuration.FfmpegBin)
|
||||
|
||||
media := t.mediafile
|
||||
media := t.mediafile
|
||||
|
||||
rcommand += media.ToStrCommand()
|
||||
rcommand += media.ToStrCommand()
|
||||
|
||||
rcommand += " \"" + t.outputPath + "\""
|
||||
rcommand += " \"" + t.outputPath + "\""
|
||||
|
||||
return rcommand
|
||||
return rcommand
|
||||
|
||||
}
|
||||
|
||||
@@ -89,194 +80,194 @@ func (t Transcoder) GetCommand() string {
|
||||
|
||||
func (t *Transcoder) Initialize(inputPath string, outputPath string) (error) {
|
||||
|
||||
configuration, err := ffmpeg.Configure()
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
return err
|
||||
}
|
||||
configuration, err := ffmpeg.Configure()
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
return err
|
||||
}
|
||||
|
||||
if inputPath == "" {
|
||||
return errors.New("error: transcoder.Initialize -> inputPath missing")
|
||||
}
|
||||
if inputPath == "" {
|
||||
return errors.New("error: transcoder.Initialize -> inputPath missing")
|
||||
}
|
||||
|
||||
_, err = os.Stat(inputPath)
|
||||
if os.IsNotExist(err) {
|
||||
return errors.New("error: transcoder.Initialize -> input file not found")
|
||||
}
|
||||
_, err = os.Stat(inputPath)
|
||||
if os.IsNotExist(err) {
|
||||
return errors.New("error: transcoder.Initialize -> input file not found")
|
||||
}
|
||||
|
||||
command := fmt.Sprintf("%s -i \"%s\" -print_format json -show_format -show_streams -show_error", configuration.FfprobeBin, inputPath)
|
||||
command := fmt.Sprintf("%s -i \"%s\" -print_format json -show_format -show_streams -show_error", configuration.FfprobeBin, inputPath)
|
||||
|
||||
cmd := exec.Command(configuration.ExecCmd, configuration.ExecArgs, command)
|
||||
cmd := exec.Command(configuration.ExecCmd, configuration.ExecArgs, command)
|
||||
|
||||
fmt.Println("FFprobe command: " + command)
|
||||
fmt.Println("FFprobe command: " + command)
|
||||
|
||||
var out bytes.Buffer
|
||||
var out bytes.Buffer
|
||||
|
||||
cmd.Stdout = &out
|
||||
cmd.Stdout = &out
|
||||
|
||||
err = cmd.Start()
|
||||
err = cmd.Start()
|
||||
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
_, err = cmd.Process.Wait()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
_, err = cmd.Process.Wait()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
var Metadata models.Metadata
|
||||
var Metadata models.Metadata
|
||||
|
||||
if err = json.Unmarshal([]byte(out.String()), &Metadata); err != nil {
|
||||
return err
|
||||
}
|
||||
if err = json.Unmarshal([]byte(out.String()), &Metadata); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
// Set new Mediafile
|
||||
// Set new Mediafile
|
||||
MediaFile := new(models.Mediafile)
|
||||
MediaFile.SetMetadata(Metadata)
|
||||
|
||||
MediaFile.SetInputPath(inputPath)
|
||||
// Set transcoder configuration
|
||||
t.SetInputPath(inputPath)
|
||||
|
||||
t.SetOutputPath(outputPath)
|
||||
t.SetMediaFile(MediaFile)
|
||||
t.SetConfiguration(configuration)
|
||||
t.SetConfiguration(configuration)
|
||||
|
||||
return nil
|
||||
|
||||
}
|
||||
|
||||
func (t *Transcoder) Run() (<-chan bool, error) {
|
||||
done := make(chan bool)
|
||||
var err error
|
||||
done := make(chan bool)
|
||||
var err error
|
||||
|
||||
command := t.GetCommand()
|
||||
command := t.GetCommand()
|
||||
|
||||
fmt.Println("FFmpeg command: " + command)
|
||||
fmt.Println("FFmpeg command: " + command)
|
||||
|
||||
proc := exec.Command(t.configuration.ExecCmd, t.configuration.ExecArgs, command)
|
||||
proc := exec.Command(t.configuration.ExecCmd, t.configuration.ExecArgs, command)
|
||||
|
||||
t.SetProccess(proc)
|
||||
t.SetProccess(proc)
|
||||
|
||||
go func() {
|
||||
err := proc.Start()
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
go func() {
|
||||
err := proc.Start()
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
proc.Wait()
|
||||
proc.Wait()
|
||||
|
||||
done <- true
|
||||
close(done)
|
||||
}()
|
||||
done <- true
|
||||
close(done)
|
||||
}()
|
||||
|
||||
return done, err
|
||||
return done, err
|
||||
|
||||
}
|
||||
|
||||
func (t Transcoder) Output() (<-chan models.Progress, error) {
|
||||
out := make(chan models.Progress)
|
||||
var err error
|
||||
out := make(chan models.Progress)
|
||||
var err error
|
||||
|
||||
go func() {
|
||||
defer close(out)
|
||||
go func() {
|
||||
defer close(out)
|
||||
|
||||
stderr, stderror := t.Process().StderrPipe()
|
||||
if err != nil {
|
||||
err = stderror
|
||||
return
|
||||
}
|
||||
stderr, stderror := t.Process().StderrPipe()
|
||||
if err != nil {
|
||||
err = stderror
|
||||
return
|
||||
}
|
||||
|
||||
scanner := bufio.NewScanner(stderr)
|
||||
filetype := utils.CheckFileType(t.MediaFile().Metadata().Streams)
|
||||
scanner := bufio.NewScanner(stderr)
|
||||
filetype := utils.CheckFileType(t.MediaFile().Metadata().Streams)
|
||||
|
||||
split := func(data []byte, atEOF bool) (advance int, token []byte, spliterror error) {
|
||||
split := func(data []byte, atEOF bool) (advance int, token []byte, spliterror error) {
|
||||
|
||||
if atEOF && len(data) == 0 {
|
||||
return 0, nil, nil
|
||||
}
|
||||
if atEOF && len(data) == 0 {
|
||||
return 0, nil, nil
|
||||
}
|
||||
|
||||
Iframe := strings.Index(string(data), "frame=")
|
||||
Iframe := strings.Index(string(data), "frame=")
|
||||
|
||||
if filetype == "video" {
|
||||
if Iframe > 0 {
|
||||
return Iframe + 1, data[Iframe:], nil
|
||||
}
|
||||
} else {
|
||||
if i := bytes.IndexByte(data, '\n'); i >= 0 {
|
||||
// We have a full newline-terminated line.
|
||||
return i + 1, data[0:i], nil
|
||||
}
|
||||
}
|
||||
if filetype == "video" {
|
||||
if Iframe > 0 {
|
||||
return Iframe + 1, data[Iframe:], nil
|
||||
}
|
||||
} else {
|
||||
if i := bytes.IndexByte(data, '\n'); i >= 0 {
|
||||
// We have a full newline-terminated line.
|
||||
return i + 1, data[0:i], nil
|
||||
}
|
||||
}
|
||||
|
||||
if atEOF {
|
||||
return len(data), data, nil
|
||||
}
|
||||
if atEOF {
|
||||
return len(data), data, nil
|
||||
}
|
||||
|
||||
return 0, nil, nil
|
||||
}
|
||||
return 0, nil, nil
|
||||
}
|
||||
|
||||
scanner.Split(split)
|
||||
buf := make([]byte, 2)
|
||||
scanner.Buffer(buf, bufio.MaxScanTokenSize)
|
||||
scanner.Split(split)
|
||||
buf := make([]byte, 2)
|
||||
scanner.Buffer(buf, bufio.MaxScanTokenSize)
|
||||
|
||||
var lastProgress float64
|
||||
for scanner.Scan() {
|
||||
Progress := new(models.Progress)
|
||||
line := scanner.Text()
|
||||
var lastProgress float64
|
||||
for scanner.Scan() {
|
||||
Progress := new(models.Progress)
|
||||
line := scanner.Text()
|
||||
|
||||
if strings.Contains(line, "time=") && strings.Contains(line, "bitrate=") {
|
||||
var re= regexp.MustCompile(`=\s+`)
|
||||
st := re.ReplaceAllString(line, `=`)
|
||||
if strings.Contains(line, "time=") && strings.Contains(line, "bitrate=") {
|
||||
var re= regexp.MustCompile(`=\s+`)
|
||||
st := re.ReplaceAllString(line, `=`)
|
||||
|
||||
f := strings.Fields(st)
|
||||
f := strings.Fields(st)
|
||||
|
||||
var framesProcessed string
|
||||
var currentTime string
|
||||
var currentBitrate string
|
||||
var framesProcessed string
|
||||
var currentTime string
|
||||
var currentBitrate string
|
||||
|
||||
for j := 0; j < len(f); j++ {
|
||||
field := f[j]
|
||||
fieldSplit := strings.Split(field, "=")
|
||||
for j := 0; j < len(f); j++ {
|
||||
field := f[j]
|
||||
fieldSplit := strings.Split(field, "=")
|
||||
|
||||
if len(fieldSplit) > 1 {
|
||||
fieldname := strings.Split(field, "=")[0]
|
||||
fieldvalue := strings.Split(field, "=")[1]
|
||||
if len(fieldSplit) > 1 {
|
||||
fieldname := strings.Split(field, "=")[0]
|
||||
fieldvalue := strings.Split(field, "=")[1]
|
||||
|
||||
if fieldname == "frame" {
|
||||
framesProcessed = fieldvalue
|
||||
}
|
||||
if fieldname == "frame" {
|
||||
framesProcessed = fieldvalue
|
||||
}
|
||||
|
||||
if fieldname == "time" {
|
||||
currentTime = fieldvalue
|
||||
}
|
||||
if fieldname == "time" {
|
||||
currentTime = fieldvalue
|
||||
}
|
||||
|
||||
if fieldname == "bitrate" {
|
||||
currentBitrate = fieldvalue
|
||||
}
|
||||
}
|
||||
}
|
||||
if fieldname == "bitrate" {
|
||||
currentBitrate = fieldvalue
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
timesec := utils.DurToSec(currentTime)
|
||||
dursec, _ := strconv.ParseFloat(t.MediaFile().Metadata().Format.Duration, 64)
|
||||
// Progress calculation
|
||||
progress := (timesec * 100) / dursec
|
||||
timesec := utils.DurToSec(currentTime)
|
||||
dursec, _ := strconv.ParseFloat(t.MediaFile().Metadata().Format.Duration, 64)
|
||||
// Progress calculation
|
||||
progress := (timesec * 100) / dursec
|
||||
|
||||
Progress.Progress = progress
|
||||
Progress.CurrentBitrate = currentBitrate
|
||||
Progress.FramesProcessed = framesProcessed
|
||||
Progress.CurrentTime = currentTime
|
||||
Progress.Progress = progress
|
||||
Progress.CurrentBitrate = currentBitrate
|
||||
Progress.FramesProcessed = framesProcessed
|
||||
Progress.CurrentTime = currentTime
|
||||
|
||||
if progress != lastProgress {
|
||||
lastProgress = progress
|
||||
out <- *Progress
|
||||
}
|
||||
}
|
||||
}
|
||||
if progress != lastProgress {
|
||||
lastProgress = progress
|
||||
out <- *Progress
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if scerror := scanner.Err(); err != nil {
|
||||
err = scerror
|
||||
}
|
||||
}()
|
||||
if scerror := scanner.Err(); err != nil {
|
||||
err = scerror
|
||||
}
|
||||
}()
|
||||
|
||||
return out, err
|
||||
return out, err
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user