Jackal\Giffhanger\Generator\AbstractVideoGenerator

Complexity

Total Complexity

22

Size/Duplication

Total Lines	152
Duplicated Lines	0 %

Importance

Changes	2
Bugs	1	Features	0

8 Methods

Rating	Name	Duplication	Size	Complexity
A	getFFMpeg()	0	6	1
A	getDuration()	0	6	1
A	__destruct()	0	9	5
A	addTempFileToRemove()	0	3	1
B	generate()	0	69	8
A	__construct()	0	14	3
A	getCutPoints()	0	9	2
A	getRatio()	0	6	1

<?php

namespace Jackal\Giffhanger\Generator;

use Alchemy\BinaryDriver\Exception\ExecutionFailureException;
use FFMpeg\Coordinate\Dimension;
use FFMpeg\Coordinate\FrameRate;
use FFMpeg\Coordinate\TimeCode;
use FFMpeg\Exception\RuntimeException;
use FFMpeg\FFMpeg;
use FFMpeg\Filters\Audio\SimpleFilter;
use FFMpeg\Filters\Video\ClipFilter;
use FFMpeg\Filters\Video\FrameRateFilter;
use FFMpeg\Filters\Video\ResizeFilter;
use FFMpeg\Media\Video;
use Jackal\Giffhanger\Configuration\Configuration;
use Jackal\Giffhanger\Exception\GiffhangerException;
use Jackal\Giffhanger\FFMpeg\ext\Filters\CropCenterFilter;

abstract class AbstractVideoGenerator implements GeneratorInterface
{
    abstract protected function getVideoFormat();

    protected $destination;
    protected $sourceFile;

    protected $options = [];
    private $tempFilesToRemove = [];

    public function __construct($sourceFile, $destionationFile, Configuration $options)
    {
        $this->options = $options;

        if (!is_dir($this->options->getTempFolder())) {
            if (!mkdir($this->options->getTempFolder(), 0777, true)) {
                $this->__destruct();

                throw new \Exception('Cannot create temp folder in path "' . $this->options->getTempFolder() . '"');
            }
        }

        $this->sourceFile = $sourceFile;
        $this->destination = $destionationFile;
    }

    /**
     * @return FFMpeg
     */
    protected function getFFMpeg() : FFMpeg
    {
        return FFMpeg::create([
            'ffmpeg.binaries' => $this->options->getFFMpegBinaries(),
            'ffprobe.binaries' => $this->options->getFFProbeBinaries(),
            'timeout' => 3600,
        ]);
    }

    protected function getDuration() : int
    {
        $ffmpeg = $this->getFFMpeg();
        $ffmpeg = $ffmpeg->open($this->sourceFile);

        return $ffmpeg->getFFProbe()->format($this->sourceFile)->get('duration');

The expression return $ffmpeg->getFFProbe()->format($this->sourceFile)->get('duration') could return the type null which is incompatible with the type-hinted return integer. Consider adding an additional type-check to rule them out.

    }

    protected function getCutPoints() : array
    {
        $cutPoints = [];
        $videoDuration = $this->getDuration();
        for ($i = 1;$i <= $this->options->getNumberOfFrames();$i++) {
            $cutPoints[] = (($videoDuration / $this->options->getNumberOfFrames()) - ($videoDuration / $this->options->getNumberOfFrames() / 2)) * $i;
        }

        return $cutPoints;
    }

    protected function getRatio() : float
    {
        $ffmpeg = $this->getFFMpeg();
        $ffmpeg = $ffmpeg->open($this->sourceFile);

        return $ffmpeg->getStreams()->videos()->first()->getDimensions()->getRatio()->getValue();
    }

    public function __destruct()
    {
        foreach (array_unique($this->tempFilesToRemove) as $fileToRemove) {
            if (is_file($fileToRemove)) {
                unlink($fileToRemove);
            }
            //if folder is empty, remove
            if (is_dir(dirname($fileToRemove)) and count(scandir(dirname($fileToRemove))) == 2) {

It seems like scandir(dirname($fileToRemove)) can also be of type false; however, parameter $var of count() does only seem to accept Countable|array, maybe add an additional type check?

                rmdir(dirname($fileToRemove));
            }
        }
    }

    protected function addTempFileToRemove($filePath) : void
    {
        $this->tempFilesToRemove[] = $filePath;
    }

    public function generate() : void
    {
        $ffmpeg = $this->getFFMpeg();
        $videoFormat = $this->getVideoFormat();

        try {
            $cutPoints = $this->getCutPoints();

            /** @var Video $video */
            $video = $ffmpeg->open($this->sourceFile);

            $files = [];
            foreach ($cutPoints as $k => $cutPoint) {
                $partFile = $this->options->getTempFolder() . '/' . md5($this->sourceFile) . '_' . ($k + 1) . '.avi';

                //remove audio track
                $video->addFilter(new SimpleFilter(['-an']));

                $video->addFilter(new ClipFilter(
                    TimeCode::fromSeconds($cutPoint),
                    TimeCode::fromSeconds($this->options->getOutputDuration() / count($cutPoints))
                ));

                //FIX: https://stackoverflow.com/questions/20847674/ffmpeg-libx264-height-not-divisible-by-2
                $width = (int) (ceil($this->options->getDimensionWidth() / 2) * 2);
                $height = (int) (ceil($this->options->getDimensionWidth() / $this->getRatio() / 2) * 2);

                $video->addFilter(new ResizeFilter(
                    new Dimension($width, $height)
                ));

                $video->addFilter(new FrameRateFilter(new FrameRate($this->options->getFrameRate()), 1));
                $video->save($videoFormat, $partFile);

                $files[] = $partFile;
                $this->addTempFileToRemove($partFile);
            }

            if (count($files) == 1) {
                rename($files[0], $this->destination);
            } else {
                /** @var Video $video */
                $video = $ffmpeg->open($files[0]);

                if (is_file($this->destination)) {
                    unlink($this->destination);
                }

                $video
                    ->concat($files)
                    ->saveFromSameCodecs($this->destination);
            }

            if ($this->options->getCropRatio() and ($this->options->getCropRatio() != $this->getRatio())) {
                $fileCropped = $this->options->getTempFolder() . '/' . md5($this->sourceFile) . '_cropped.avi';

                $video = $ffmpeg->open($this->destination);
                $video->addFilter(new CropCenterFilter($this->options->getCropRatio()));
                $video->save($videoFormat, $fileCropped);
                rename($fileCropped, $this->destination);
            }
        } catch (RuntimeException $e) {
            $message = $e->getMessage();

            if ($e->getPrevious() instanceof ExecutionFailureException) {
                $message = $e->getPrevious()->getMessage();
            }

            throw new GiffhangerException(sprintf('%s', $message));
        }
    }
}