merging changes from DRUPAL-6--4

1 files changed, 262 insertions, 0 deletions

diff --git a/includes/conversion.inc b/includes/conversion.inc
new file mode 100644
index 0000000..3f1165e
--- /dev/null
+++ b/includes/conversion.inc
@@ -0,0 +1,262 @@
+<?php
+//$Id$
+/*
+ * @file
+ * Class file to handle video conversion using ffmpeg.
+ *
+ */
+define('VIDEO_RENDERING_PENDING', 1);
+define('VIDEO_RENDERING_ACTIVE', 5);
+define('VIDEO_RENDERING_COMPLETE', 10);
+define('VIDEO_RENDERING_FAILED', 20);
+
+class video_conversion {
+
+  /**
+   * Our main function to call when converting queued up jobs.
+   */
+  public function run_queue() {
+    if ($videos = $this->select_queue()) {
+      foreach ($videos as $video) {
+        $this->process($video);
+      }
+    }
+  }
+
+  /**
+   * Select videos from our queue
+   *
+   * @return
+   *   An array containing all the videos to be proccessed.
+   */
+  private function select_queue() {
+    $total_videos = variable_get('video_ffmpeg_instances', 5);
+    $videos = array();
+    $result = db_query_range('SELECT f.*, vf.vid, vf.nid, vf.dimensions, vf.status as video_status FROM {video_files} vf LEFT JOIN {files} f ON vf.fid = f.fid WHERE vf.status = %d AND f.status = %d ORDER BY f.timestamp',
+    VIDEO_RENDERING_PENDING, FILE_STATUS_PERMANENT, 0, $total_videos);
+
+    while ($row = db_fetch_object($result)) {
+      $videos[] = $row;
+    }
+    return $videos;
+  }
+
+  /**
+   * Process the video through ffmpeg.
+   *
+   * @param $video
+   *   This can either be the file object or the file id (fid)
+   *
+   * @return
+   *   TRUE of FALSE if video was converted successfully.
+   */
+  public function process($video) {
+    if (is_object($video) && isset($video->fid)) {
+      $return = $this->render($video);
+    }
+    else {
+      $video_object = $this->load_video($video);
+      $return = $this->render($video_object);
+    }
+    return $return;
+  }
+
+  private function render($video) {
+    if (!is_object($video)) {
+      watchdog('video_conversion', 'Video object is not present', array(), WATCHDOG_ERROR);
+      return FALSE;
+    }
+    // Make sure this video is pending or do nothing.
+    if($video->video_status == VIDEO_RENDERING_PENDING) {
+      // This will update our current video status to active.
+      $this->change_status($video->vid, VIDEO_RENDERING_ACTIVE);
+      // Get the converted file object
+      //we are going to move our video to an "original" folder
+      //we are going to transcode the video to the "converted" folder
+      $pathinfo = pathinfo($video->filepath);
+      $original = $pathinfo['dirname'] .'/original';
+      $converted = $pathinfo['dirname'] .'/converted';
+
+      if (!field_file_check_directory($original, FILE_CREATE_DIRECTORY)) {
+        watchdog('video_transcoder', 'Video conversion failed.  Could not create the directory: '.$orginal, array(), WATCHDOG_ERROR);
+        return false;
+      }
+      if (!field_file_check_directory($converted, FILE_CREATE_DIRECTORY)) {
+        watchdog('video_transcoder', 'Video conversion failed.  Could not create the directory: '.$converted, array(), WATCHDOG_ERROR);
+        return false;
+      }
+
+      $original = $original .'/'. $video->filename;
+      //lets move our video and then convert it.
+      if(file_move($video, $original)) {
+        //update our filename after the move to maintain filename uniqueness.
+        $converted = $converted .'/'. pathinfo($video->filepath, PATHINFO_FILENAME) .'.'. $this->video_extension();
+        // Update our filepath since we moved it
+        $update = drupal_write_record('files', $video, 'fid');
+        //call our transcoder
+        $command_output = $this->convert_video($video, $converted);
+        //lets check to make sure our file exists, if not error out
+        if(!file_exists($converted) || !filesize($converted)) {
+          watchdog('video_conversion', 'Video conversion failed.  FFMPEG reported the following output: '.$command_output, array(), WATCHDOG_ERROR);
+          $this->change_status($video->vid, VIDEO_RENDERING_FAILED);
+          return FALSE;
+        }
+        // Setup our converted video object
+        $video_info = pathinfo($converted);
+        //update our converted video
+        $video->converted = new stdClass();
+        $video->converted->vid = $video->vid;
+        $video->converted->filename = $video_info['basename'];
+        $video->converted->filepath = $converted;
+        $video->converted->filemime = file_get_mimetype($converted);
+        $video->converted->filesize = filesize($converted);
+        $video->converted->status = VIDEO_RENDERING_COMPLETE;
+        $video->converted->completed = time();
+        //clear our cache so our video path is updated.
+        cache_clear_all('*', 'cache_content', true);
+        // Update our video_files table with the converted video information.
+        $result = db_query("UPDATE {video_files} SET filename='%s', filepath='%s', filemime='%s', filesize=%d, status=%d, completed=%d WHERE vid=%d",
+        $video->converted->filename, $video->converted->filepath, $video->converted->filemime, $video->converted->filesize, $video->converted->status, $video->converted->completed, $video->converted->vid);
+
+        // Update our node id to published.  We do not do a node_load as it causes editing problems when saving.
+        db_query("UPDATE {node} SET status=%d WHERE nid=%d", 1, $video->nid);
+        watchdog('video_conversion', 'Successfully converted %orig to %dest', array('%orig' => $video->filepath, '%dest' => $video->converted->filepath), WATCHDOG_INFO);
+        return TRUE;
+      }
+      else {
+        watchdog('video_conversion', 'Cound not move the video to the original folder.', array(), WATCHDOG_ERROR);
+        $this->change_status($video->vid, VIDEO_RENDERING_FAILED);
+        return FALSE;
+      }
+    }
+    return NULL;
+  }
+
+  /**
+   * Calls the transcoder class to convert the video.
+   *
+   * @param $job
+   *   Video object to be transcoded
+   *
+   * @return
+   *   TRUE or FALSE
+   */
+  private function convert_video($video, $converted) {
+    //get our dimensions and pass them along.
+    $dimensions = $this->dimensions($video);
+    module_load_include('inc', 'video', '/includes/transcoder');
+    $transcoder = new video_transcoder;
+    return $transcoder->convert_video($video, $converted, $dimensions);
+  }
+
+  private function video_extension() {
+    module_load_include('inc', 'video', '/includes/transcoder');
+    $transcoder = new video_transcoder;
+    return $transcoder->video_converted_extension();
+  }
+
+  /*
+   * Function determines the dimensions you want and compares with the actual wxh of the video.
+   *
+   * If they are not exact or the aspect ratio does not match, we then figure out how much padding
+   * we should add.  We will either add a black bar on the top/bottom or on the left/right.
+   *
+   * @TODO I need to look more at this function.  I don't really like the guess work here.  Need to implement
+   * a better way to check the end WxH.  Maybe compare the final resolution to our defaults?  I don't think
+   * that just checking to make sure the final number is even is accurate enough.
+   */
+  public function dimensions($video) {
+    //lets setup our dimensions.  Make sure our aspect ratio matches the dimensions to be used, if not lets add black bars.
+    $aspect_ratio = _video_aspect_ratio($video->filepath);
+    $ratio = $aspect_ratio['ratio'];
+    $width = $aspect_ratio ['width'];
+    $height = $aspect_ratio['height'];
+
+    $wxh = explode('x', $video->dimensions);
+    $output_width = $wxh[0];
+    $output_height = $wxh[1];
+    $output_ratio = number_format($output_width / $output_height, 4);
+
+    if($output_ratio != $ratio && $width && $height) {
+      $options = array();
+      // Figure out our black bar padding.
+      if ($ratio < $output_width / $output_height) {
+        $end_width = $output_height * $ratio;
+        $end_height = $output_height;
+      }
+      else {
+        $end_height = $output_width / $ratio;
+        $end_width = $output_width;
+      }
+
+      // We need to get back to an even resolution and maybe compare with our defaults?
+      // @TODO Make this more exact on actual video dimensions instead of making sure the wxh are even numbers
+
+      if ($end_width == $output_width) {
+        // We need to pad the top/bottom of the video
+        $padding = round($output_height - $end_height);
+        $pad1 = $pad2 = floor($padding / 2);
+        if ($pad1 %2 !== 0) {
+          $pad1++;
+          $pad2--;
+        }
+        $options[] = '-padtop '. $pad1;
+        $options[] = '-padbottom '. $pad2;
+      }
+      else {
+        // We are padding the left/right of the video.
+        $padding = round($output_width - $end_width);
+        $pad1 = $pad2 = floor($padding / 2);  //@todo does padding need to be an even number?
+        if ($pad1 %2 !== 0) {
+          $pad1++;
+          $pad2--;
+        }
+        $options[] = '-padleft '. $pad1;
+        $options[] = '-padright '. $pad2;
+      }
+
+      $end_width = round($end_width) %2 !==0 ? round($end_width) + 1 : round($end_width);
+      $end_height = round($end_height) %2 !==0 ? round($end_height) + 1 : round($end_height);
+      //add our size to the beginning to make sure it hits our -s
+      array_unshift($options, $end_width .'x'. $end_height);
+      return implode(' ', $options);
+    }
+    else {
+      return $video->dimensions;
+    }
+  }
+
+  /**
+   * Load a file based on the file id ($fid)
+   *
+   * @param $fid
+   *   Integer of the file id to be loaded.
+   */
+  public function load_video($fid) {
+    $result = db_query('SELECT f.*, vf.vid, vf.nid, vf.dimensions, vf.status as video_status FROM {video_files} vf LEFT JOIN {files} f ON vf.fid = f.fid WHERE f.fid=vf.fid AND f.fid = %d', $fid);
+    return db_fetch_object($result);
+  }
+
+  /**
+   * Load a converted video based on the file id ($fid)
+   *
+   * @todo: Need to figure something out here for multiple files (HTML 5)
+   * @param $fid
+   *   Integer of the file id to be loaded.
+   */
+  public function load_converted_video($fid) {
+    $result = db_query('SELECT * FROM {video_files} WHERE fid = %d', $fid);
+    return db_fetch_object($result);
+  }
+
+  /**
+   * Change the status of the file.
+   *
+   * @param (int) $vid
+   * @param (int) $status
+   */
+  public function change_status($vid, $status) {
+    $result = db_query('UPDATE {video_files} SET status = %d WHERE vid = %d ', $status, $vid);
+  }
+}
+?>
\ No newline at end of file