3 * Base WordPress Image Editor
6 * @subpackage Image_Editor
10 * Base image editor class from which implementations extend
14 abstract class WP_Image_Editor {
15 protected $file = null;
16 protected $size = null;
17 protected $mime_type = null;
18 protected $default_mime_type = 'image/jpeg';
19 protected $quality = false;
20 protected $default_quality = 90;
23 * Each instance handles a single file.
25 public function __construct( $file ) {
30 * Checks to see if current environment supports the editor chosen.
31 * Must be overridden in a sub-class.
40 public static function test( $args = array() ) {
45 * Checks to see if editor supports the mime-type specified.
46 * Must be overridden in a sub-class.
52 * @param string $mime_type
55 public static function supports_mime_type( $mime_type ) {
60 * Loads image from $this->file into editor.
66 * @return boolean|WP_Error True if loaded; WP_Error on failure.
68 abstract public function load();
71 * Saves current image to file.
77 * @param string $destfilename
78 * @param string $mime_type
79 * @return array|WP_Error {'path'=>string, 'file'=>string, 'width'=>int, 'height'=>int, 'mime-type'=>string}
81 abstract public function save( $destfilename = null, $mime_type = null );
84 * Resizes current image.
86 * At minimum, either a height or width must be provided.
87 * If one of the two is set to null, the resize will
88 * maintain aspect ratio according to the provided dimension.
94 * @param int|null $max_w Image width.
95 * @param int|null $max_h Image height.
96 * @param boolean $crop
97 * @return boolean|WP_Error
99 abstract public function resize( $max_w, $max_h, $crop = false );
102 * Resize multiple images from a single source.
108 * @param array $sizes {
109 * An array of image size arrays. Default sizes are 'small', 'medium', 'large'.
111 * @type array $size {
112 * @type int $width Image width.
113 * @type int $height Image height.
114 * @type bool $crop Optional. Whether to crop the image. Default false.
117 * @return array An array of resized images metadata by size.
119 abstract public function multi_resize( $sizes );
128 * @param string|int $src The source file or Attachment ID.
129 * @param int $src_x The start x position to crop from.
130 * @param int $src_y The start y position to crop from.
131 * @param int $src_w The width to crop.
132 * @param int $src_h The height to crop.
133 * @param int $dst_w Optional. The destination width.
134 * @param int $dst_h Optional. The destination height.
135 * @param boolean $src_abs Optional. If the source crop points are absolute.
136 * @return boolean|WP_Error
138 abstract public function crop( $src_x, $src_y, $src_w, $src_h, $dst_w = null, $dst_h = null, $src_abs = false );
141 * Rotates current image counter-clockwise by $angle.
147 * @param float $angle
148 * @return boolean|WP_Error
150 abstract public function rotate( $angle );
153 * Flips current image.
159 * @param boolean $horz Flip along Horizontal Axis
160 * @param boolean $vert Flip along Vertical Axis
161 * @return boolean|WP_Error
163 abstract public function flip( $horz, $vert );
166 * Streams current image to browser.
172 * @param string $mime_type
173 * @return boolean|WP_Error
175 abstract public function stream( $mime_type = null );
178 * Gets dimensions of image.
183 * @return array {'width'=>int, 'height'=>int}
185 public function get_size() {
190 * Sets current image size.
198 protected function update_size( $width = null, $height = null ) {
200 'width' => (int) $width,
201 'height' => (int) $height
207 * Gets the Image Compression quality on a 1-100% scale.
212 * @return int $quality Compression Quality. Range: [1,100]
214 public function get_quality() {
215 if ( ! $this->quality ) {
217 * Filter the default image compression quality setting.
221 * @param int $quality Quality level between 1 (low) and 100 (high).
222 * @param string $mime_type Image mime type.
224 $quality = apply_filters( 'wp_editor_set_quality', $this->default_quality, $this->mime_type );
226 if ( 'image/jpeg' == $this->mime_type ) {
228 * Filter the JPEG compression quality for backward-compatibility.
230 * The filter is evaluated under two contexts: 'image_resize', and 'edit_image',
231 * (when a JPEG image is saved to file).
235 * @param int $quality Quality level between 0 (low) and 100 (high) of the JPEG.
236 * @param string $context Context of the filter.
238 $quality = apply_filters( 'jpeg_quality', $quality, 'image_resize' );
240 if ( ! $this->set_quality( $quality ) ) {
241 $this->quality = $this->default_quality;
246 return $this->quality;
250 * Sets Image Compression quality on a 1-100% scale.
255 * @param int $quality Compression Quality. Range: [1,100]
256 * @return boolean|WP_Error True if set successfully; WP_Error on failure.
258 public function set_quality( $quality = null ) {
259 // Allow 0, but squash to 1 due to identical images in GD, and for backwards compatibility.
260 if ( $quality == 0 ) {
264 if ( ( $quality >= 1 ) && ( $quality <= 100 ) ) {
265 $this->quality = $quality;
268 return new WP_Error( 'invalid_image_quality', __('Attempted to set image quality outside of the range [1,100].') );
273 * Returns preferred mime-type and extension based on provided
274 * file's extension and mime, or current file's extension and mime.
276 * Will default to $this->default_mime_type if requested is not supported.
278 * Provides corrected filename only if filename is provided.
283 * @param string $filename
284 * @param string $mime_type
285 * @return array { filename|null, extension, mime-type }
287 protected function get_output_format( $filename = null, $mime_type = null ) {
290 // By default, assume specified type takes priority
292 $new_ext = $this->get_extension( $mime_type );
296 $file_ext = strtolower( pathinfo( $filename, PATHINFO_EXTENSION ) );
297 $file_mime = $this->get_mime_type( $file_ext );
300 // If no file specified, grab editor's current extension and mime-type.
301 $file_ext = strtolower( pathinfo( $this->file, PATHINFO_EXTENSION ) );
302 $file_mime = $this->mime_type;
305 // Check to see if specified mime-type is the same as type implied by
306 // file extension. If so, prefer extension from file.
307 if ( ! $mime_type || ( $file_mime == $mime_type ) ) {
308 $mime_type = $file_mime;
309 $new_ext = $file_ext;
312 // Double-check that the mime-type selected is supported by the editor.
313 // If not, choose a default instead.
314 if ( ! $this->supports_mime_type( $mime_type ) ) {
316 * Filter default mime type prior to getting the file extension.
318 * @see wp_get_mime_types()
322 * @param string $mime_type Mime type string.
324 $mime_type = apply_filters( 'image_editor_default_mime_type', $this->default_mime_type );
325 $new_ext = $this->get_extension( $mime_type );
330 $info = pathinfo( $filename );
331 $dir = $info['dirname'];
333 if( isset( $info['extension'] ) )
334 $ext = $info['extension'];
336 $filename = trailingslashit( $dir ) . wp_basename( $filename, ".$ext" ) . ".{$new_ext}";
339 return array( $filename, $new_ext, $mime_type );
343 * Builds an output filename based on current file, and adding proper suffix
348 * @param string $suffix
349 * @param string $dest_path
350 * @param string $extension
351 * @return string filename
353 public function generate_filename( $suffix = null, $dest_path = null, $extension = null ) {
354 // $suffix will be appended to the destination filename, just before the extension
356 $suffix = $this->get_suffix();
358 $info = pathinfo( $this->file );
359 $dir = $info['dirname'];
360 $ext = $info['extension'];
362 $name = wp_basename( $this->file, ".$ext" );
363 $new_ext = strtolower( $extension ? $extension : $ext );
365 if ( ! is_null( $dest_path ) && $_dest_path = realpath( $dest_path ) )
368 return trailingslashit( $dir ) . "{$name}-{$suffix}.{$new_ext}";
372 * Builds and returns proper suffix for file based on height and width.
377 * @return string suffix
379 public function get_suffix() {
380 if ( ! $this->get_size() )
383 return "{$this->size['width']}x{$this->size['height']}";
387 * Either calls editor's save function or handles file as a stream.
392 * @param string|stream $filename
393 * @param callable $function
394 * @param array $arguments
397 protected function make_image( $filename, $function, $arguments ) {
398 if ( $stream = wp_is_stream( $filename ) ) {
401 // The directory containing the original file may no longer exist when using a replication plugin.
402 wp_mkdir_p( dirname( $filename ) );
405 $result = call_user_func_array( $function, $arguments );
407 if ( $result && $stream ) {
408 $contents = ob_get_contents();
410 $fp = fopen( $filename, 'w' );
415 fwrite( $fp, $contents );
427 * Returns first matched mime-type from extension,
428 * as mapped from wp_get_mime_types()
433 * @param string $extension
434 * @return string|boolean
436 protected static function get_mime_type( $extension = null ) {
440 $mime_types = wp_get_mime_types();
441 $extensions = array_keys( $mime_types );
443 foreach( $extensions as $_extension ) {
444 if ( preg_match( "/{$extension}/i", $_extension ) ) {
445 return $mime_types[$_extension];
453 * Returns first matched extension from Mime-type,
454 * as mapped from wp_get_mime_types()
459 * @param string $mime_type
460 * @return string|boolean
462 protected static function get_extension( $mime_type = null ) {
463 $extensions = explode( '|', array_search( $mime_type, wp_get_mime_types() ) );
465 if ( empty( $extensions[0] ) )
468 return $extensions[0];