<?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',

			'readimageblob',

		);

		// 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;

		}

		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;

		}

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

				return false;

		}

		try {

			// phpcs:ignore WordPress.PHP.NoSilencedErrors.Discouraged

			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 ) && ! wp_is_stream( $this->file ) ) {

			return new WP_Error( 'error_loading_image', __( 'File does not 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 ) );

			if ( 'pdf' === $file_extension ) {

				$pdf_loaded = $this->pdf_load_source();

				if ( is_wp_error( $pdf_loaded ) ) {

					return $pdf_loaded;

				}

			} else {

				if ( wp_is_stream( $this->file ) ) {

					// Due to reports of issues with streams with `Imagick::readImageFile()`, uses `Imagick::readImageBlob()` instead.

					$this->image->readImageBlob( file_get_contents( $this->file ), $this->file );

				} else {

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

				}

			}

			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 {

			switch ( $this->mime_type ) {

				case 'image/jpeg':

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

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

					break;

				case 'image/webp':

					$webp_info = wp_get_webp_info( $this->file );

					if ( 'lossless' === $webp_info['type'] ) {

						// Use WebP lossless settings.

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

						$this->image->setOption( 'webp:lossless', 'true' );

					} else {

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

					}

					break;

				default:

					$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

	 *

	 * @return true|WP_Error

	 */

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

			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 void|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 the allowed list and the related

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

		 */

		if ( in_array( $filter_name, $allowed_filters, true ) && 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