YetiForceCompany/YetiForcePDF

<?php

declare(strict_types=1);
/**
 * ImageStream class.
 *
 * @package   YetiForcePDF\Objects\Basic
 *
 * @copyright YetiForce Sp. z o.o
 * @license   MIT
 * @author    Rafal Pospiech <r.pospiech@yetiforce.com>
 */

namespace YetiForcePDF\Objects;

use YetiForcePDF\Math;

/**
 * Class ImageStream.
 */
class ImageStream extends \YetiForcePDF\Objects\Resource
{
    /**
     * Object name.
     *
     * @var string
     */
    protected $name = 'ImageStream';
    /**
     * Text x position at current page.
     *
     * @var int
     */
    protected $x = 0;
    /**
     * Text y position at current page.
     *
     * @var int
     */
    protected $y = 0;
    /**
     * @var string Image data
     */
    protected $imageData = '';
    /**
     * @var string image original width
     */
    protected $width = '0';
    /**
     * @var string image original height
     */
    protected $height = '0';
    /**
     * Width / Height ratio.
     *
     * @var string
     */
    protected $ratio = '0';
    /**
     * @var int bits per component
     */
    protected $bitsPerComponent = 8;

    /**
     * Convert images to jpeg.
     *
     * @param string $imageData
     *
     * @return string
     */
    protected function convertToJpg(string $imageData)
    {
        if ('data:image' === substr($imageData, 0, 10)) {
            $matches = [];
            preg_match('/data:image\/[a-z]+;base64,(.*)/', $imageData, $matches);
            $imageData = base64_decode($matches[1]);
        }
        $image = @imagecreatefromstring($imageData);
        $bg = imagecreatetruecolor(imagesx($image), imagesy($image));
        imagefill($bg, 0, 0, imagecolorallocate($bg, 255, 255, 255));
        imagealphablending($bg, true);
        imagecopy($bg, $image, 0, 0, 0, 0, imagesx($image), imagesy($image));
        ob_start();
        imagejpeg($bg, null, 90);
        $imageString = ob_get_contents();
        ob_end_clean();
        imagedestroy($image);
        imagedestroy($bg);
        return $imageString;
    }

    /**
     * Load image data.
     *
     * @param string $fileName
     *
     * @return $this
     */
    public function loadImage(string $fileName)
    {
        if ('data:image' === substr($fileName, 0, 10)) {
            $this->imageData = $this->convertToJpg($fileName);
        } elseif (filter_var($fileName, FILTER_VALIDATE_URL)) {
            try {
                $client = new \GuzzleHttp\Client();
                $res = $client->request('GET', $fileName, ['verify' => \Composer\CaBundle\CaBundle::getSystemCaRootBundlePath()]);
                if (200 === $res->getStatusCode()) {
                    $res->getHeader('content-type');
                    $this->imageData = $this->convertToJpg((string) $res->getBody());
                }
            } catch (\Exception $e) {
            }
        } else {
            if (file_exists($fileName)) {
                $this->imageData = $this->convertToJpg(file_get_contents($fileName));
            }
        }
        if ($this->imageData) {
            $info = getimagesizefromstring($this->imageData);
            $this->bitsPerComponent = (string) $info['bits'];
            $this->width = (string) $info[0];
            $this->height = (string) $info[1];
            $this->displayWidth = $this->width;
            $this->displayHeight = $this->height;
            Math::setAccurate(true);
            $this->ratio = Math::div($this->getWidth(), $this->getHeight());
            Math::setAccurate(false);
        }
        return $this;
    }

    /**
     * Get image name.
     *
     * @return string
     */
    public function getImageName()
    {
        return 'Im' . $this->id;
    }

    /**
     * Get image width.
     *
     * @return string
     */
    public function getWidth()
    {
        return $this->width;
    }

    /**
     * Get image height.
     *
     * @return string
     */
    public function getHeight()
    {
        return $this->height;
    }

    /**
     * {@inheritdoc}
     */
    public function render(): string
    {
        $stream = $this->imageData;
        return implode("\n", [
            $this->getRawId() . ' obj',
            '<<',
            '  /Type /XObject',
            '  /Subtype  /Image',
            '  /ColorSpace  /DeviceRGB',
            '  /Width ' . $this->width,
            '  /Height ' . $this->height,
            '  /BitsPerComponent ' . $this->bitsPerComponent,
            '  /Length  ' . \strlen($stream),
            '  /Filter /DCTDecode',
            '>>',
            'stream',
            $stream,
            'endstream',
            'endobj',
        ]);
    }

    /**
     * Get width / Height ratio.
     *
     * @return string
     */
    public function getRatio()
    {
        return $this->ratio;
    }

    /**
     * Set width / Height ratio.
     *
     * @param string $ratio Width / Height ratio.
     *
     * @return self
     */
    public function setRatio(string $ratio)
    {
        $this->ratio = $ratio;
        return $this;
    }

    /**
     * Get display width.
     *
     * @return string
     */
    public function getDisplayWidth()
    {
        return $this->displayWidth;
    }

    /**
     * Set display width.
     *
     * @param string $displayWidth Display width
     *
     * @return self
     */
    public function setDisplayWidth(string $displayWidth)
    {
        $this->displayWidth = $displayWidth;
        return $this;
    }

    /**
     * Get display height.
     *
     * @return string
     */
    public function getDisplayHeight()
    {
        return $this->displayHeight;
    }

    /**
     * Set display height.
     *
     * @param string $displayHeight Display height
     *
     * @return self
     */
    public function setDisplayHeight(string $displayHeight)
    {
        $this->displayHeight = $displayHeight;
        return $this;
    }
}