<?php

/**

 * WordPress Imagick Image Editor

 *

 * @package WordPress

 * @subpackage Image_Editor

 */

/**

 * WordPress Image Editor Class for Image Manipulation through Imagick PHP Module

 *

 * @since 3.5.0

 *

 * @see WP_Image_Editor

 */

class WP_Image_Editor_Imagick extends WP_Image_Editor {

	/**

	 * Imagick object.

	 *

	 * @var Imagick

	 */

	protected $image;

	public function __destruct() {

		if ( $this->image instanceof Imagick ) {

			// we don't need the original in memory anymore

			$this->image->clear();

			$this->image->destroy();

		}

	}

	/**

	 * Checks to see if current environment supports Imagick.

	 *

	 * We require Imagick 2.2.0 or greater, based on whether the queryFormats()

	 * method can be called statically.

	 *

	 * @since 3.5.0

	 *

	 * @param array $args

	 * @return bool

	 */

	public static function test( $args = array() ) {

		// First, test Imagick's extension and classes.

		if ( ! extension_loaded( 'imagick' ) || ! class_exists( 'Imagick', false ) || ! class_exists( 'ImagickPixel', false ) ) {

			return false;

		}

		if ( version_compare( phpversion( 'imagick' ), '2.2.0', '<' ) ) {

			return false;

		}

		$required_methods = array(

			'clear',

			'destroy',

			'valid',

			'getimage',

			'writeimage',

			'getimageblob',

			'getimagegeometry',

			'getimageformat',

			'setimageformat',

			'setimagecompression',

			'setimagecompressionquality',

			'setimagepage',

			'setoption',

			'scaleimage',

			'cropimage',

			'rotateimage',

			'flipimage',

			'flopimage',

			'readimage',

		);

		// Now, test for deep requirements within Imagick.

		if ( ! defined( 'imagick::COMPRESSION_JPEG' ) ) {

			return false;

		}

		$class_methods = array_map( 'strtolower', get_class_methods( 'Imagick' ) );

		if ( array_diff( $required_methods, $class_methods ) ) {

			return false;

		}

		// HHVM Imagick does not support loading from URL, so fail to allow fallback to GD.

		if ( defined( 'HHVM_VERSION' ) && isset( $args['path'] ) && preg_match( '|^https?://|', $args['path'] ) ) {

			return false;

		}

		return true;

	}

	/**

	 * Checks to see if editor supports the mime-type specified.

	 *

	 * @since 3.5.0

	 *

	 * @param string $mime_type

	 * @return bool

	 */

	public static function supports_mime_type( $mime_type ) {

		$imagick_extension = strtoupper( self::get_extension( $mime_type ) );

		if ( ! $imagick_extension ) {

			return false;

		}

		// setIteratorIndex is optional unless mime is an animated format.

		// Here, we just say no if you are missing it and aren't loading a jpeg.

		if ( ! method_exists( 'Imagick', 'setIteratorIndex' ) && $mime_type != 'image/jpeg' ) {

				return false;

		}

		try {

			return ( (bool) @Imagick::queryFormats( $imagick_extension ) );

		} catch ( Exception $e ) {

			return false;

		}

	}

	/**

	 * Loads image from $this->file into new Imagick Object.

	 *

	 * @since 3.5.0

	 *

	 * @return true|WP_Error True if loaded; WP_Error on failure.

	 */

	public function load() {

		if ( $this->image instanceof Imagick ) {

			return true;

		}

		if ( ! is_file( $this->file ) && ! preg_match( '|^https?://|', $this->file ) ) {

			return new WP_Error( 'error_loading_image', __( 'File doesn’t exist?' ), $this->file );

		}

		/*

		 * Even though Imagick uses less PHP memory than GD, set higher limit

		 * for users that have low PHP.ini limits.

		 */

		wp_raise_memory_limit( 'image' );

		try {

			$this->image    = new Imagick();

			$file_extension = strtolower( pathinfo( $this->file, PATHINFO_EXTENSION ) );

			$filename       = $this->file;

			if ( 'pdf' == $file_extension ) {

				$filename = $this->pdf_setup();

			}

			// Reading image after Imagick instantiation because `setResolution`

			// only applies correctly before the image is read.

			$this->image->readImage( $filename );

			if ( ! $this->image->valid() ) {

				return new WP_Error( 'invalid_image', __( 'File is not an image.' ), $this->file );

			}

			// Select the first frame to handle animated images properly

			if ( is_callable( array( $this->image, 'setIteratorIndex' ) ) ) {

				$this->image->setIteratorIndex( 0 );

			}

			$this->mime_type = $this->get_mime_type( $this->image->getImageFormat() );

		} catch ( Exception $e ) {

			return new WP_Error( 'invalid_image', $e->getMessage(), $this->file );

		}

		$updated_size = $this->update_size();

		if ( is_wp_error( $updated_size ) ) {

			return $updated_size;

		}

		return $this->set_quality();

	}

	/**

	 * Sets Image Compression quality on a 1-100% scale.

	 *

	 * @since 3.5.0

	 *

	 * @param int $quality Compression Quality. Range: [1,100]

	 * @return true|WP_Error True if set successfully; WP_Error on failure.

	 */

	public function set_quality( $quality = null ) {

		$quality_result = parent::set_quality( $quality );

		if ( is_wp_error( $quality_result ) ) {

			return $quality_result;

		} else {

			$quality = $this->get_quality();

		}

		try {

			if ( 'image/jpeg' == $this->mime_type ) {

				$this->image->setImageCompressionQuality( $quality );

				$this->image->setImageCompression( imagick::COMPRESSION_JPEG );

			} else {

				$this->image->setImageCompressionQuality( $quality );

			}

		} catch ( Exception $e ) {

			return new WP_Error( 'image_quality_error', $e->getMessage() );

		}

		return true;

	}

	/**

	 * Sets or updates current image size.

	 *

	 * @since 3.5.0

	 *

	 * @param int $width

	 * @param int $height

	 *

	 * @return true|WP_Error

	 */

	protected function update_size( $width = null, $height = null ) {

		$size = null;

		if ( ! $width || ! $height ) {

			try {

				$size = $this->image->getImageGeometry();

			} catch ( Exception $e ) {

				return new WP_Error( 'invalid_image', __( 'Could not read image size.' ), $this->file );

			}

		}

		if ( ! $width ) {

			$width = $size['width'];

		}

		if ( ! $height ) {

			$height = $size['height'];

		}

		return parent::update_size( $width, $height );

	}

	/**

	 * Resizes current image.

	 *

	 * At minimum, either a height or width must be provided.

	 * If one of the two is set to null, the resize will

	 * maintain aspect ratio according to the provided dimension.

	 *

	 * @since 3.5.0

	 *

	 * @param  int|null $max_w Image width.

	 * @param  int|null $max_h Image height.

	 * @param  bool     $crop

	 * @return bool|WP_Error

	 */

	public function resize( $max_w, $max_h, $crop = false ) {

		if ( ( $this->size['width'] == $max_w ) && ( $this->size['height'] == $max_h ) ) {

			return true;

		}

		$dims = image_resize_dimensions( $this->size['width'], $this->size['height'], $max_w, $max_h, $crop );

		if ( ! $dims ) {

			return new WP_Error( 'error_getting_dimensions', __( 'Could not calculate resized image dimensions' ) );

		}

		list( $dst_x, $dst_y, $src_x, $src_y, $dst_w, $dst_h, $src_w, $src_h ) = $dims;

		if ( $crop ) {

			return $this->crop( $src_x, $src_y, $src_w, $src_h, $dst_w, $dst_h );

		}

		// Execute the resize

		$thumb_result = $this->thumbnail_image( $dst_w, $dst_h );

		if ( is_wp_error( $thumb_result ) ) {

			return $thumb_result;

		}

		return $this->update_size( $dst_w, $dst_h );

	}

	/**

	 * Efficiently resize the current image

	 *

	 * This is a WordPress specific implementation of Imagick::thumbnailImage(),

	 * which resizes an image to given dimensions and removes any associated profiles.

	 *

	 * @since 4.5.0

	 *

	 * @param int    $dst_w       The destination width.

	 * @param int    $dst_h       The destination height.

	 * @param string $filter_name Optional. The Imagick filter to use when resizing. Default 'FILTER_TRIANGLE'.

	 * @param bool   $strip_meta  Optional. Strip all profiles, excluding color profiles, from the image. Default true.

	 * @return bool|WP_Error

	 */

	protected function thumbnail_image( $dst_w, $dst_h, $filter_name = 'FILTER_TRIANGLE', $strip_meta = true ) {

		$allowed_filters = array(

			'FILTER_POINT',

			'FILTER_BOX',

			'FILTER_TRIANGLE',

			'FILTER_HERMITE',

			'FILTER_HANNING',

			'FILTER_HAMMING',

			'FILTER_BLACKMAN',

			'FILTER_GAUSSIAN',

			'FILTER_QUADRATIC',

			'FILTER_CUBIC',

			'FILTER_CATROM',

			'FILTER_MITCHELL',

			'FILTER_LANCZOS',

			'FILTER_BESSEL',

			'FILTER_SINC',

		);

		/**

		 * Set the filter value if '$filter_name' name is in our whitelist and the related

		 * Imagick constant is defined or fall back to our default filter.

		 */

		if ( in_array( $filter_name, $allowed_filters ) && defined( 'Imagick::' . $filter_name ) ) {

			$filter = constant( 'Imagick::' . $filter_name );

		} else {

			$filter = defined( 'Imagick::FILTER_TRIANGLE' ) ? Imagick::FILTER_TRIANGLE : false;

		}

		/**

		 * Filters whether to strip metadata from images when they're resized.

		 *

		 * This filter only applies when resizing using the Imagick editor since GD

		 * always strips profiles by default.

		 *

		 * @since 4.5.0

		 *

		 * @param bool $strip_meta Whether to strip image metadata during resizing. Default true.

		 */

		if ( apply_filters( 'image_strip_meta', $strip_meta ) ) {

			$this->strip_meta(); // Fail silently if not supported.

		}

		try {

			/*

			 * To be more efficient, resample large images to 5x the destination size before resizing

			 * whenever the output size is less that 1/3 of the original image size (1/3^2 ~= .111),

			 * unless we would be resampling to a scale smaller than 128x128.

			 */

			if ( is_callable( array( $this->image, 'sampleImage' ) ) ) {

				$resize_ratio  = ( $dst_w / $this->size['width'] ) * ( $dst_h / $this->size['height'] );

				$sample_factor = 5;

				if ( $resize_ratio < .111 && ( $dst_w * $sample_factor > 128 && $dst_h * $sample_factor > 128 ) ) {

					$this->image->sampleImage( $dst_w * $sample_factor, $dst_h * $sample_factor );

				}

			}

			/*

			 * Use resizeImage() when it's available and a valid filter value is set.

			 * Otherwise, fall back to the scaleImage() method for resizing, which

			 * results in better image quality over resizeImage() with default filter

			 * settings and retains backward compatibility with pre 4.5 functionality.

			 */

			if ( is_callable( array( $this->image, 'resizeImage' ) ) && $filter ) {

				$this->image->setOption( 'filter:support', '2.0' );

				$this->image->resizeImage( $dst_w, $dst_h, $filter, 1 );

			} else {

				$this->image->scaleImage( $dst_w, $dst_h );

			}

			// Set appropriate quality settings after resizing.

			if ( 'image/jpeg' == $this->mime_type ) {

				if ( is_callable( array( $this->image, 'unsharpMaskImage' ) ) ) {

					$this->image->unsharpMaskImage( 0.25, 0.25, 8, 0.065 );

				}

				$this->image->setOption( 'jpeg:fancy-upsampling', 'off' );

			}

			if ( 'image/png' === $this->mime_type ) {

				$this->image->setOption( 'png:compression-filter', '5' );

				$this->image->setOption( 'png:compression-level', '9' );

				$this->image->setOption( 'png:compression-strategy', '1' );

				$this->image->setOption( 'png:exclude-chunk', 'all' );

			}

			/*

			 * If alpha channel is not defined, set it opaque.

			 *

			 * Note that Imagick::getImageAlphaChannel() is only available if Imagick

			 * has been compiled against ImageMagick version 6.4.0 or newer.

			 */

			if ( is_callable( array( $this->image, 'getImageAlphaChannel' ) )

				&& is_callable( array( $this->image, 'setImageAlphaChannel' ) )

				&& defined( 'Imagick::ALPHACHANNEL_UNDEFINED' )

				&& defined( 'Imagick::ALPHACHANNEL_OPAQUE' )

			) {

				if ( $this->image->getImageAlphaChannel() === Imagick::ALPHACHANNEL_UNDEFINED ) {

					$this->image->setImageAlphaChannel( Imagick::ALPHACHANNEL_OPAQUE );

				}

			}

			// Limit the bit depth of resized images to 8 bits per channel.

			if ( is_callable( array( $this->image, 'getImageDepth' ) ) && is_callable( array( $this->image, 'setImageDepth' ) ) ) {

				if ( 8 < $this->image->getImageDepth() ) {

					$this->image->setImageDepth( 8 );

				}

			}

			if ( is_callable( array( $this->image, 'setInterlaceScheme' ) ) && defined( 'Imagick::INTERLACE_NO' ) ) {

				$this->image->setInterlaceScheme( Imagick::INTERLACE_NO );

			}

		} catch ( Exception $e ) {

			return new WP_Error( 'image_resize_error', $e->getMessage() );