Update worker code to the point that it can run.

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.
2010-08-22 23:05:33 +01:00
parent 853c380e65
commit 40fa00c034
7 changed files with 260 additions and 67 deletions
--- a/lib/HandBrakeCluster/Worker/Plugin/HandBrake.class.php
+++ b/lib/HandBrakeCluster/Worker/Plugin/HandBrake.class.php
@@ -0,0 +1,130 @@
+<?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);
+            }
+        }
+    }
+    
+}
+
+?>