Ben Roberts
40fa00c034
Added plugin architecture for Worker functions, to support multiple ripping engines. Update worker script to use the new code. Update Logger to output messages to console when running under the worker script.
130 lines
4.1 KiB
PHP
130 lines
4.1 KiB
PHP
<?php
|
|
|
|
class HandBrakeCluster_Worker_Plugin_HandBrake implements HandBrakeCluster_Worker_IPlugin {
|
|
|
|
const DEINTERLACE_ALWAYS = 1;
|
|
const DEINTERLACE_SELECTIVELY = 2;
|
|
|
|
private $stdout;
|
|
private $stderr;
|
|
|
|
private $job;
|
|
|
|
private $client_job_id;
|
|
private $rip_options;
|
|
|
|
private function __construct(GearmanJob $job) {
|
|
$this->stdout = '';
|
|
$this->stderr = '';
|
|
|
|
$this->job = $job;
|
|
|
|
$this->client_job_id = $job->unique();
|
|
$this->rip_options = unserialize($job->workload());
|
|
}
|
|
|
|
public static function init() {
|
|
|
|
}
|
|
|
|
public static function workerFunctions() {
|
|
return array(
|
|
'handbrake_rip' => array(__CLASS__, 'rip'),
|
|
);
|
|
}
|
|
|
|
public static function rip(GearmanJob $job) {
|
|
$rip = new self($job);
|
|
$rip->execute();
|
|
}
|
|
|
|
public function execute() {
|
|
$config = HandBrakeCluster_Main::instance()->config();
|
|
|
|
$handbrake_cmd_raw = array(
|
|
'-n', $config->get('rips.nice'),
|
|
$config->get('rips.handbrake_binary'),
|
|
$this->evaluateOption('input_filename', '-i'),
|
|
$this->evaluateOption('output_filename', '-o'),
|
|
$this->evaluateOption('title'),
|
|
$this->evaluateOption('format', '-f'),
|
|
$this->evaluateOption('video_codec', '-e'),
|
|
$this->evaluateOption('quantizer', '-q'),
|
|
$this->evaluateOption('video_width', '-w'),
|
|
$this->evaluateOption('video_height', '-l'),
|
|
$this->evaluateOption('deinterlace'),
|
|
$this->evaluateOption('audio_tracks', '-a'),
|
|
$this->evaluateOption('audio_codec', '-E'),
|
|
$this->evaluateOption('audio_names', '-A'),
|
|
$this->evaluateOption('subtitle_tracks', '-s'),
|
|
);
|
|
|
|
$handbrake_cmd = array($config->get('rips.nice_binary'));
|
|
foreach(new RecursiveIteratorIterator(new RecursiveArrayIterator($handbrake_cmd_raw)) as $value) {
|
|
$handbrake_cmd[] = escapeshellarg($value);
|
|
}
|
|
$handbrake_cmd = join(' ', $handbrake_cmd);
|
|
|
|
return HandBrakeCluster_ForegroundTask::execute($handbrake_cmd, null, null, null, array($this, 'callbackStdout'), array($this, 'callbackStderr'), $this);
|
|
|
|
}
|
|
|
|
public function evaluateOption($name, $option = null) {
|
|
switch($name) {
|
|
case 'title': {
|
|
if (!$this->rip_options[$name] || (int)$this->rip_options[$name] < 0) {
|
|
return array('-L');
|
|
} else {
|
|
return array('-t', $this->rip_options[$name]);
|
|
}
|
|
} break;
|
|
|
|
case 'deinterlace': {
|
|
switch ($this->rip_options[$name]) {
|
|
case self::DEINTERLACE_ALWAYS:
|
|
return array('-d');
|
|
case self::DEINTERLACE_SELECTIVELY:
|
|
return array('-5');
|
|
default:
|
|
return array();
|
|
}
|
|
}
|
|
|
|
default:
|
|
return array(isset($option) ? $option : $name, $this->rip_options[$name]);
|
|
}
|
|
}
|
|
|
|
public function callbackStdout($rip, $data) {
|
|
$this->stdout .= $data;
|
|
|
|
while (count($lines = preg_split('/[\r\n]+/', $this->stdout, 2)) > 1) {
|
|
$line = $lines[0];
|
|
$this->stdout = $lines[1];
|
|
|
|
$log = HandBrakeCluster_Main::instance()->log();
|
|
$log->info($line);
|
|
}
|
|
}
|
|
|
|
public function callbackStderr($rip, $data) {
|
|
$this->stderr .= $data;
|
|
|
|
while (count($lines = preg_split('/[\r\n]+/', $this->stderr, 2)) > 1) {
|
|
$line = $lines[0];
|
|
$rip->stderr = $lines[1];
|
|
|
|
$matches = array();
|
|
if (preg_match('/Encoding: task \d+ of \d+, (\d+\.\d+) %/', $line, $matches)) {
|
|
$numerator = 100 * $matches[1];
|
|
$this->job->sendStatus($numerator, 100);
|
|
} else {
|
|
$log = HandBrakeCluster_Main::instance()->log();
|
|
$log->debug($line);
|
|
}
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
?>
|