mirror of
https://github.com/stashapp/stash.git
synced 2025-12-16 21:34:14 +01:00
214a15bc40
* Add a space after // comments
For consistency, the commentFormatting lint checker suggests a space
after each // comment block. This commit handles all the spots in
the code where that is needed.
* Rewrite documentation on functions
Use the Go idiom of commenting:
* First sentence declares the purpose.
* First word is the name being declared
The reason this style is preferred is such that grep is able to find
names the user might be interested in. Consider e.g.,
go doc -all pkg/ffmpeg | grep -i transcode
in which case a match will tell you the name of the function you are
interested in.
* Remove old code comment-blocks
There are some commented out old code blocks in the code base. These are
either 3 years old, or 2 years old. By now, I don't think their use is
going to come back any time soon, and Git will track old pieces of
deleted code anyway.
Opt for deletion.
* Reorder imports
Split stdlib imports from non-stdlib imports in files we are touching.
* Use a range over an iteration variable
Probably more go-idiomatic, and the code needed comment-fixing anyway.
* Use time.After rather than rolling our own
The idiom here is common enough that the stdlib contains a function for
it. Use the stdlib function over our own variant.
* Enable the commentFormatting linter
111 lines
2.9 KiB
Go
111 lines
2.9 KiB
Go
package ffmpeg
|
|
|
|
import (
|
|
"strconv"
|
|
|
|
"github.com/stashapp/stash/pkg/models"
|
|
)
|
|
|
|
type TranscodeOptions struct {
|
|
OutputPath string
|
|
MaxTranscodeSize models.StreamingResolutionEnum
|
|
}
|
|
|
|
func calculateTranscodeScale(probeResult VideoFile, maxTranscodeSize models.StreamingResolutionEnum) string {
|
|
maxSize := 0
|
|
switch maxTranscodeSize {
|
|
case models.StreamingResolutionEnumLow:
|
|
maxSize = 240
|
|
case models.StreamingResolutionEnumStandard:
|
|
maxSize = 480
|
|
case models.StreamingResolutionEnumStandardHd:
|
|
maxSize = 720
|
|
case models.StreamingResolutionEnumFullHd:
|
|
maxSize = 1080
|
|
case models.StreamingResolutionEnumFourK:
|
|
maxSize = 2160
|
|
}
|
|
|
|
// get the smaller dimension of the video file
|
|
videoSize := probeResult.Height
|
|
if probeResult.Width < videoSize {
|
|
videoSize = probeResult.Width
|
|
}
|
|
|
|
// if our streaming resolution is larger than the video dimension
|
|
// or we are streaming the original resolution, then just set the
|
|
// input width
|
|
if maxSize >= videoSize || maxSize == 0 {
|
|
return "iw:-2"
|
|
}
|
|
|
|
// we're setting either the width or height
|
|
// we'll set the smaller dimesion
|
|
if probeResult.Width > probeResult.Height {
|
|
// set the height
|
|
return "-2:" + strconv.Itoa(maxSize)
|
|
}
|
|
|
|
return strconv.Itoa(maxSize) + ":-2"
|
|
}
|
|
|
|
func (e *Encoder) Transcode(probeResult VideoFile, options TranscodeOptions) {
|
|
scale := calculateTranscodeScale(probeResult, options.MaxTranscodeSize)
|
|
args := []string{
|
|
"-i", probeResult.Path,
|
|
"-c:v", "libx264",
|
|
"-pix_fmt", "yuv420p",
|
|
"-profile:v", "high",
|
|
"-level", "4.2",
|
|
"-preset", "superfast",
|
|
"-crf", "23",
|
|
"-vf", "scale=" + scale,
|
|
"-c:a", "aac",
|
|
"-strict", "-2",
|
|
options.OutputPath,
|
|
}
|
|
_, _ = e.runTranscode(probeResult, args)
|
|
}
|
|
|
|
// TranscodeVideo transcodes the video, and removes the audio.
|
|
// In some videos where the audio codec is not supported by ffmpeg,
|
|
// ffmpeg fails if you try to transcode the audio
|
|
func (e *Encoder) TranscodeVideo(probeResult VideoFile, options TranscodeOptions) {
|
|
scale := calculateTranscodeScale(probeResult, options.MaxTranscodeSize)
|
|
args := []string{
|
|
"-i", probeResult.Path,
|
|
"-an",
|
|
"-c:v", "libx264",
|
|
"-pix_fmt", "yuv420p",
|
|
"-profile:v", "high",
|
|
"-level", "4.2",
|
|
"-preset", "superfast",
|
|
"-crf", "23",
|
|
"-vf", "scale=" + scale,
|
|
options.OutputPath,
|
|
}
|
|
_, _ = e.runTranscode(probeResult, args)
|
|
}
|
|
|
|
// TranscodeAudio will copy the video stream as is, and transcode audio.
|
|
func (e *Encoder) TranscodeAudio(probeResult VideoFile, options TranscodeOptions) {
|
|
args := []string{
|
|
"-i", probeResult.Path,
|
|
"-c:v", "copy",
|
|
"-c:a", "aac",
|
|
"-strict", "-2",
|
|
options.OutputPath,
|
|
}
|
|
_, _ = e.runTranscode(probeResult, args)
|
|
}
|
|
|
|
// CopyVideo will copy the video stream as is, and drop the audio stream.
|
|
func (e *Encoder) CopyVideo(probeResult VideoFile, options TranscodeOptions) {
|
|
args := []string{
|
|
"-i", probeResult.Path,
|
|
"-an",
|
|
"-c:v", "copy",
|
|
options.OutputPath,
|
|
}
|
|
_, _ = e.runTranscode(probeResult, args)
|
|
}
|