ImageFactory.php

<?php
 
namespace Bitrix\Catalog\v2\Image;
 
use Bitrix\Catalog\v2\IoC\ContainerContract;
 
class ImageFactory
{
    public const MORE_PHOTO_IMAGE = MorePhotoImage::class;
    public const DETAIL_IMAGE = DetailImage::class;
    public const PREVIEW_IMAGE = PreviewImage::class;
    public const IMAGE_COLLECTION = ImageCollection::class;
 
    protected $container;
 
    public function __construct(ContainerContract $container)
    {
        $this->container = $container;
    }
 
    public function createEntity(string $type = null): BaseImage
    {
        switch ($type)
        {
            case self::PREVIEW_IMAGE:
                return $this->container->make(self::PREVIEW_IMAGE);
 
            case self::DETAIL_IMAGE:
                return $this->container->make(self::DETAIL_IMAGE);
        }
 
        return $this->container->make(self::MORE_PHOTO_IMAGE);
    }
 
    public function createCollection(): ImageCollection
    {
        return $this->container->make(self::IMAGE_COLLECTION);
    }
}