2.php

<?php

$src = $argv[1];
$dst = $argv[2];
$codec = $argv[3];
$preset = $argv[4];
$resolution = !empty($argv[5]) ? $argv[5] : 720;
$resolution = str_replace('x', ':', $resolution);
$crf = !empty($argv[6]) ? $argv[6] : (strpos($codec, 'h265') !== false ? 19 : 17);
$tune = !empty($argv[7]) ? $argv[7] : 'grain';
$keyframes = '';
$timecodes = '';
$audio = [];
$subtitle = [];
$chapters = [];
$title = '';
$fps_mod = 0;
$fps_demod = 0;
$mkvmerge = "E:\\tmp\\media\\util\\mkvtoolnix\\mkvmerge.exe";
$attachments = [
	['desc' => 'About', 'path' => dirname(__FILE__).'/about.txt'],
	['desc' => 'common.avsi', 'path' => dirname(__FILE__).'/common.avsi']
	];

$dst = preg_replace('/\\.mkv$/i', '-'.$codec.$preset.preg_replace('/(([0-9]+):)?([0-9]+)$/', '\\3', $resolution).'crf'.$crf.'.mkv', $dst);

$resolution = explode(':', $resolution);

if(count($resolution) == 1 && is_numeric($resolution[0]))
{
	$h = (int)$resolution[0];
	
	if($h == 480 || $h == 576)
	{
		$resolution = [720, $h];
	}
	else
	{
		//$resolution = [$h * 4 / 3, $h];
		$resolution = [-1, $h];
	}
}

function is51($fn)
{
	$str = shell_exec('ffprobe -hide_banner -show_streams "'.$fn.'" -print_format json');
	$obj = json_decode($str, true);
	return $obj['streams'][0]['channels'] == 6;
}

function getfps($fn, &$mod, &$demod)
{
	$str = shell_exec('ffprobe -hide_banner -show_streams "'.$fn.'" -print_format json');
	$obj = json_decode($str, true);
	
	foreach($obj['streams'] as $stream)
	{
		if($stream['codec_type'] == 'video' && !empty($stream['avg_frame_rate']))
		{
			$fps = explode('/', $stream['avg_frame_rate'], 2);
			$mod = $fps[0];
			$demod = $fps[1];
			return true;
		}
	}
	
	return false;
}

function getchapters($fn)
{
	$str = shell_exec('ffprobe -hide_banner -show_chapters "'.$fn.'" -print_format json');
	$obj = json_decode($str, true);
	return isset($obj['chapters']) ? $obj['chapters'] : false;
}

if(preg_match('/(.*title_t[0-9]+[^-]*-)/i', $src, $m)
|| preg_match('/(.*S[0-9]+(E|R)[0-9]+-)/i', $src, $m))
{
	$dir = dirname($m[1]);
	$basename = basename($m[1]);

	$fn = rtrim($m[1], '-').'.avs';
	
	if(file_exists($fn))
	{
		$attachments[] = ['desc' => basename($fn), 'path' => $fn];
	}
	
	foreach(scandir(dirname($m[1])) as $fn)
	{
		if(!preg_match('/'.$basename.'var-.+\\.txt$/i', $fn)) continue;

		$attachments[] = ['desc' => $fn, 'path' => $dir.'/'.$fn];
	}

	$fn = $m[1].'timecodes.txt';
	
	if(file_exists($fn))
	{
		$timecodes = $fn;
		$attachments[] = ['desc' => 'VFR timecodes', 'path' => $fn];
	}
	
	$fn = $m[1].'tfm-ovr.txt';
	
	if(file_exists($fn))
	{
		$tfm_ovr = $fn;
		$attachments[] = ['desc' => 'TFM overrides', 'path' => $fn];
	}
	
	$fn = $m[1].'tdec-ovr.txt';
	
	if(file_exists($fn))
	{
		$tdec_ovr = $fn;
		$attachments[] = ['desc' => 'TDecimate overrides', 'path' => $fn];
	}

	$fn = $m[1].'frame.avi';
	
	if(file_exists($fn))
	{
		getfps($fn, $fps_mod, $fps_demod);
	}

	foreach(['eng', 'hun', 'fra', 'rus'] as $lang)
	{
		$fn2 = rtrim($m[1], '-');
		
		foreach(['mka', 'wav', 'ac3'] as $format)
		{
			$fn = $fn2.'_'.$lang.'.'.$format;
			
			if(file_exists($fn)) $audio[] = ['fn' => $fn, 'lang' => $lang, 'format' => $format, 'is51' => $format == 'wav' && is51($fn)];
		
			for($i = 1; ; $i++)
			{
				$fn = $fn2.'_'.$lang.$i.'.'.$format;
			
				if(!file_exists($fn)) break;
				
				$audio[] = ['fn' => $fn, 'lang' => $lang, 'format' => $format, 'is51' => $format == 'wav' && is51($fn)];
			}
		}
		
		foreach(['mks', 'srt', 'sub', 'ssa', 'ass'] as $format)
		{
			$fn = $fn2.'_'.$lang.'.'.$format;

			if(file_exists($fn)) $subtitle[] = ['fn' => $fn, 'lang' => $lang, 'format' => $format];
		
			for($i = 1; ; $i++)
			{
				$fn = $fn2.'_'.$lang.$i.'.'.$format;
			
				if(!file_exists($fn)) break;
			
				$subtitle[] = ['fn' => $fn, 'lang' => $lang, 'format' => $format];
			}
		}
	}
	
	foreach(['_chapters.mkv', '.mkc', '.mkv'] as $fn)
	{
		$fn = rtrim($m[1], '-').$fn;
		
		if(file_exists($fn) && ($c = getchapters($fn)) !== false)
		{
			$chapters = $c;
			break;
		}
	}
}

if(!empty($tfm_ovr) && file_exists($tfm_ovr))
{
	// all this code only to extract the title and the keyframes
	
	$frames = [];

	if(!empty($timecodes) && file_exists($timecodes))
	{
		$s = file_get_contents($timecodes);

		$pef = -1;

		foreach(explode("\n", $s) as $row)
		{
			$row = trim($row);
	
			if(!preg_match('/^([0-9]+),([0-9]+),([0-9\\.]+)/i', $row, $m)) continue;
	
			$sf = (int)$m[1];
			$ef = (int)$m[2];
	
			if($sf > $pef + 1)
			{
				$frames[] = ['sf' => $pef + 1, 'ef' => $sf - 1, 'fps_num' => 30000, 'row' => ''];
			}

			$pef = $ef;
	
			if($m[3] == '29.97') $fps_num = 30000;
			else if($m[3] == '23.976024') $fps_num = 24000;
			else if($m[3] == '17.982018') $fps_num = 18000;
			else die('unknown fps '.$m[3]);
	
			$frames[] = ['sf' => $sf, 'ef' => $ef, 'fps_num' => $fps_num, 'row' => $row];
		}

		$frames[] = ['sf' => $pef + 1, 'ef' => 1000000, 'fps_num' => 30000, 'row' => '']; // TODO: last frame number?
	}
	
	$keyframes = [];
	
	foreach(explode("\n", file_get_contents($tfm_ovr)) as $row)
	{
		if(empty($title) && preg_match('/^# *S([0-9]+)(E|R)([0-9]+) +-?(.+)$/i', $row, $m))
		{	
			$title = sprintf("Lexx S%02d%s%02d - %s", (int)$m[1], $m[2], (int)$m[3], trim($m[4]));
		}

		if(preg_match('/([0-9]+),([0-9]+).*# *keyframe/i', $row, $m))
		{
			$frame = (int)$m[1];
			
			$tfc = 0;

			foreach($frames as $f)
			{
				$fc = ($f['ef'] - $f['sf'] + 1) * 30000 / $f['fps_num'];
		
				if((int)$tfc <= $frame && $frame < (int)($tfc + $fc + 0.5))
				{
					$vfrframe = (int)($f['sf'] + ($frame - $tfc) * $f['fps_num'] / 30000 + 0.5);
					//$vfrframe = ceil($f['sf'] + ($frame - $tfc) * $f['fps_num'] / 30000);
					$vfrtime = $vfrframe * 1001 / 30000; // -r 30000 / 1001 is also set for the input later
			
					$ms = (int)(($vfrtime * 1000) % 1000);
					$ss = $vfrtime % 60; $vfrtime /= 60;
					$mm = $vfrtime % 60; $vfrtime /= 60;
					$hh = $vfrtime;
					$t = sprintf("%d:%02d:%02d.%03d", $hh, $mm, $ss, $ms);
			
					$keyframes[] = $t; //['f' => $vfrframe, 't' => $t];

					break;
				}
				
				$tfc += $fc;
			}
		}
	}
	
	$keyframes = implode(',', $keyframes);
}

$cmd = ['ffmpeg -hide_banner'];

if($resolution[1] >= 720) 
{
	$cmd[] = '-colorspace bt709';
}

if(preg_match('/^(.+):([0-9]+)$/i', $src, $m)) 
{
	$src = $m[1];
	
	// $cmd[] = '-start_number '.$m[2];} // changing the start isn't compatible with the vfr timecode file
}

if(strpos($src, ".avi") === false && $fps_demod > 0)
{
	$cmd[] = "-r $fps_mod/$fps_demod"; // -framerate? if the fps is not set to match the real duration, the audio will stop playing at a random position, ffmpeg tries to be smart when muxing or something
}

// input

$cmd[] = '-i "'.$src.'"';

foreach($audio as $a)
{
	$cmd[] = ($a['is51'] ? '-channel_layout 5.1 ' : '').'-i "'.$a['fn'].'"';
}

foreach($subtitle as $s)
{
	$cmd[] = '-i "'.$s['fn'].'"';
}

$cmd[] = '-pix_fmt '.(strpos($codec, 'p10') !== false ? 'yuv420p10le' : 'yuv420p');

// map

$cmd[] = '-map 0:v';

foreach($audio as $index => $a)
{
	$cmd[] = '-map '.($index + 1).':a';
}

foreach($subtitle as $index => $s)
{
	$cmd[] = '-map '.($index + count($audio) + 1).':s';
}

$cmd[] = '-map_metadata -1';
$cmd[] = '-map_chapters -1';

// codecs

$cmd[] = '-c copy';

foreach($audio as $index => $a) 
{
	$cmd[] = '-metadata:s:a:'.$index.' language='.$a['lang']; 
	
	if($a['format'] == 'wav') 
	{
		$cmd[] = '-c:a:'.$index.' aac '; 
		
		if(!$a['is51'])
		{
			$cmd[] = '-b:a:'.$index.' 192k ';
		}
	}
}

if($codec == 'h264')
{
	$cmd[] = '-c:v libx264 -profile:v high -level:v 4.1';
}
else if($codec == 'h264p10')
{
	$cmd[] = '-c:v libx264 -profile:v high10 -level:v 4.1';
}
else if($codec == 'h265')
{
	$cmd[] = '-c:v libx265';
}
else if($codec == 'h265p10')
{
	$cmd[] = '-c:v libx265 -profile:v main10';
}

//if(strpos($codec, 'h265') !== false) $cmd[] = '-x265-params "strong-intra-smoothing=0"';

$cmd[] = '-preset '.$preset.' -crf '.$crf;
$cmd[] = '-vf "scale='.implode(':', $resolution).':flags=spline"'; //lanczos"';
if(!empty($tune) && $tune != 'notune') $cmd[] = '-tune '.$tune;

// misc

$cmd[] = '-aspect 4:3';
$cmd[] = '-movflags +faststart';
if(!empty($keyframes)) $cmd[] = '-force_key_frames '.$keyframes;

// meta

$cmd[] = '-metadata title="'.$title.'"';
$cmd[] = '-metadata description="https://github.com/gabest11/lexx-ivtc" ';
$cmd[] = '-metadata:s title= ';

foreach($subtitle as $index => $s) 
{
	$cmd[] = '-metadata:s:s:'.$index.' language='.$s['lang'];
}

/*
// does not work correctly, ffmpeg chops off the audio early, calculated from the vfr rate, do a 2pass
if(!empty($timecodes) && preg_match('/# TDecimate Mode [0-9]+: +Last Frame = ([0-9]+)/i', file_get_contents($timecodes), $m))
{
	$cmd[] = '-frames:v '.($m[1] + 1);
}
*/

$cmd[] = '"'.$dst.'"';

$cmd = implode(' ', $cmd);

echo $cmd.PHP_EOL;

$ret = 0;
passthru($cmd, $ret);
if(!empty($ret)) die(sprintf("ffmpeg %d\n", $ret));

if(!empty($timecodes))
{

$dstvfr = preg_replace('/\\.mkv$/i', '-vfr.mkv', $dst);

$cmd = <<<EOT
$mkvmerge ^
--output "$dstvfr" ^
--language 0:und ^
--default-track 0:yes ^
--default-track 1:yes ^
--timestamps "0:$timecodes" ^
--fix-bitstream-timing-information 0:1 ^

EOT;

foreach($subtitle as $index => $s) 
{
	$cmd .= '--default-track '.($index + count($audio) + 1).':no ^'.PHP_EOL;
}

foreach($attachments as $a)
{
	if(file_exists($a['path']))
	{
		$cmd .= '--attachment-description "'.$a['desc'].'" ^'.PHP_EOL;
		$cmd .= '--attachment-mime-type text/plain ^'.PHP_EOL;
		$cmd .= '--attach-file "'.$a['path'].'" ^'.PHP_EOL;
	}
}

$cmd .= '"'.$dst.'"';

$cmd = preg_replace('/[\r\n]+/', ' ', $cmd);

echo $cmd.PHP_EOL;
	
$ret = 0;
passthru($cmd, $ret);
if(!empty($ret)) die(sprintf("ffmpeg %d\n", $ret));

echo 'del '.$dst.PHP_EOL;

if(!unlink($dst)) echo 'cannot delete...'.PHP_EOL;

$dst = $dstvfr;

}

// TODO: chapters for keyframes

if(!empty($chapters))
{
	$dstwch = preg_replace('/\\.mkv$/i', '-with-chapters.mkv', $dst);
	
	$sl = [];
	$i = 1;
	
	foreach($chapters as $c)
	{
		$t = (int)($c['start_time'] * 1000);
		if($t < 1000) $t = 0;
		$ms = $t % 1000; $t = (int)($t / 1000);
		$ss = $t % 60; $t = (int)($t / 60);
		$mm = $t % 60; $t = (int)($t / 60);
		$hh = $t;
		$sl[] = sprintf("CHAPTER%02d=%02d:%02d:%02d.%03d", $i, $hh, $mm, $ss, $ms);
		$sl[] = sprintf("CHAPTER%02dNAME=%s", $i, sprintf("Chapter %02d", $i));
		$i++;
	}
	
	$tmp = tempnam(dirname($dst), 'chapters');
	file_put_contents($tmp, implode("\n", $sl));
	register_shutdown_function(function() use($tmp) { unlink($tmp); });

	$cmd = "$mkvmerge --output \"$dstwch\" --chapters \"$tmp\" \"$dst\" ";

	echo $cmd.PHP_EOL;
	
	$ret = 0;
	passthru($cmd, $ret);
	if(!empty($ret)) die(sprintf("ffmpeg %d\n", $ret));

	echo 'del '.$dst.PHP_EOL;

	if(!unlink($dst)) echo 'cannot delete...'.PHP_EOL;
}

?>