lpms/transcoder/ffmpeg_segment_transcoder.go

package transcoder

import (
	"fmt"
	"io/ioutil"
	"math/rand"
	"os"
	"os/exec"
	"path"
	"time"

	"github.com/golang/glog"
	"github.com/livepeer/lpms/ffmpeg"
)

//SegmentTranscoder transcodes segments individually.  This is a simple wrapper for calling FFMpeg on the command line.
type FFMpegSegmentTranscoder struct {
	tProfiles  []ffmpeg.VideoProfile
	ffmpegPath string
	workDir    string
}

func NewFFMpegSegmentTranscoder(ps []ffmpeg.VideoProfile, ffmpegp, workd string) *FFMpegSegmentTranscoder {
	return &FFMpegSegmentTranscoder{tProfiles: ps, ffmpegPath: ffmpegp, workDir: workd}
}

func (t *FFMpegSegmentTranscoder) Transcode(d []byte) ([][]byte, error) {
	//Assume d is in the right format, write it to disk
	inName := randName()
	// outName := fmt.Sprintf("out%v", inName)
	if _, err := os.Stat(t.workDir); os.IsNotExist(err) {
		err := os.Mkdir(t.workDir, 0700)
		if err != nil {
			glog.Errorf("Transcoder cannot create workdir: %v", err)
			return nil, err
		}
	}

	if err := ioutil.WriteFile(path.Join(t.workDir, inName), d, 0644); err != nil {
		glog.Errorf("Transcoder cannot write file: %v", err)
		return nil, err
	}

	//Invoke ffmpeg
	var cmd *exec.Cmd
	//ffmpeg -i seg.ts -c:v libx264 -s 426:240 -r 30 -mpegts_copyts 1 -minrate 700k -maxrate 700k -bufsize 700k -threads 1 out3.ts
	args := make([]string, 0, 0)
	for i, p := range t.tProfiles {
		args = append(args, []string{"-c:v", "libx264", "-s", p.Resolution, "-minrate", p.Bitrate, "-maxrate", p.Bitrate, "-bufsize", p.Bitrate, "-r", fmt.Sprintf("%d", p.Framerate), "-threads", "1", "-copyts", path.Join(t.workDir, fmt.Sprintf("out%v%v", i, inName))}...)
	}
	cmd = exec.Command(path.Join(t.ffmpegPath, "ffmpeg"), append([]string{"-i", path.Join(t.workDir, inName)}, args...)...)
	if err := cmd.Run(); err != nil {
		glog.Errorf("Cannot start ffmpeg command: %v", err)
		return nil, err
	}

	dout := make([][]byte, len(t.tProfiles), len(t.tProfiles))
	for i, _ := range t.tProfiles {
		d, err := ioutil.ReadFile(path.Join(t.workDir, fmt.Sprintf("out%v%v", i, inName)))
		if err != nil {
			glog.Errorf("Cannot read transcode output: %v", err)
		}
		dout[i] = d
		os.Remove(path.Join(t.workDir, fmt.Sprintf("out%v%v", i, inName)))
	}

	os.Remove(path.Join(t.workDir, inName))
	return dout, nil
}

func randName() string {
	rand.Seed(time.Now().UnixNano())
	x := make([]byte, 10, 10)
	for i := 0; i < len(x); i++ {
		x[i] = byte(rand.Uint32())
	}
	return fmt.Sprintf("%x.ts", x)
}