AhmadMayahi\Vision\Detectors\Landmark

Complexity

Total Complexity

5

Size/Duplication

Total Lines	35
Duplicated Lines	0 %

Importance

Changes	2
Bugs	0	Features	0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	detect()	0	21	4
A	getOriginalResponse()	0	7	1

<?php

declare(strict_types=1);

namespace AhmadMayahi\Vision\Detectors;

use AhmadMayahi\Vision\Contracts\Detectable;
use AhmadMayahi\Vision\Data\Landmark as LandmarkData;
use AhmadMayahi\Vision\Data\Location as LocationData;
use AhmadMayahi\Vision\Support\AbstractDetector;
use AhmadMayahi\Vision\Traits\Arrayable;
use AhmadMayahi\Vision\Traits\Jsonable;
use Generator;
use Google\Cloud\Vision\V1\EntityAnnotation;
use Google\Cloud\Vision\V1\LocationInfo;
use Google\Protobuf\Internal\RepeatedField;

/**
 * @see https://cloud.google.com/vision/docs/detecting-landmarks
 */
class Landmark extends AbstractDetector implements Detectable
{
    use Arrayable;
    use Jsonable;

    public function getOriginalResponse(): RepeatedField
    {
        $response = $this
            ->imageAnnotatorClient
            ->landmarkDetection($this->file->toVisionFile());

        return $response->getLandmarkAnnotations();
    }

    public function detect(): ?Generator
    {
        $response = $this->getOriginalResponse();

        if (0 === $response->count()) {
            return null;
        }

        /** @var EntityAnnotation $entity */
        foreach ($response as $entity) {
            yield new LandmarkData(
                name: $entity->getDescription(),
                locations: array_map(function (LocationInfo $location) {
                    $info = $location->getLatLng();

                    if (is_null($info)) {
                        return null;
                    }

                    return new LocationData($info->getLatitude(), $info->getLongitude());
                }, iterator_to_array($entity->getLocations()->getIterator()))
            );
        }
    }
}