OCA\FaceRecognition\BackgroundJob\Tasks\ImageProcessingTask::execute()

Complexity

Conditions	10
Paths	532

Size

Total Lines	97
Code Lines	53

Duplication

Lines	0
Ratio	0 %

Code Coverage

Tests	43
CRAP Score	10.5186

Importance

Changes	2
Bugs	0	Features	0

<?php
	public function execute(FaceRecognitionContext $context) {
		$this->setContext($context);

		$this->logInfo('NOTE: Starting face recognition. If you experience random crashes after this point, please look FAQ at https://github.com/matiasdelellis/facerecognition/wiki/FAQ');

		// Get current model.
		$this->model = $this->modelManager->getCurrentModel();

		// Open model.
		$this->model->open();

		$images = $context->propertyBag['images'];
		foreach($images as $image) {
			yield;

			$startMillis = round(microtime(true) * 1000);

			try {
				// Get a image lock
				$lockKey = 'facerecognition/' . $image->getId();
				$lockType = ILockingProvider::LOCK_EXCLUSIVE;
				$this->lockingProvider->acquireLock($lockKey, $lockType);

				$dbImage = $this->imageMapper->find($image->getUser(), $image->getId());
				if ($dbImage->getIsProcessed()) {
					$this->logInfo('Faces found: 0. Image will be skipped since it was already processed.');
					// Release lock of file.
					$this->lockingProvider->releaseLock($lockKey, $lockType);
					continue;
				}


				// Get an temp Image to process this image.
				$tempImage = $this->getTempImage($image);

				if (is_null($tempImage)) {
					// If we cannot find a file probably it was deleted out of our control and we must clean our tables.
					$this->settingsService->setNeedRemoveStaleImages(true, $image->user);
					$this->logInfo('File with ID ' . $image->file . ' doesn\'t exist anymore, skipping it');
					// Release lock of file.
					$this->lockingProvider->releaseLock($lockKey, $lockType);
					continue;
				}

				if ($tempImage->getSkipped() === true) {
					$this->logInfo('Faces found: 0 (image will be skipped because it is too small)');
					$this->imageMapper->imageProcessed($image, array(), 0);
					// Release lock of file.
					$this->lockingProvider->releaseLock($lockKey, $lockType);
					continue;
				}

				// Get faces in the temporary image
				$tempImagePath = $tempImage->getTempPath();
				$rawFaces = $this->model->detectFaces($tempImagePath);

				$this->logInfo('Faces found: ' . count($rawFaces));

				$faces = array();
				foreach ($rawFaces as $rawFace) {
					// Normalize face and landmarks from model to original size
					$normFace = $this->getNormalizedFace($rawFace, $tempImage->getRatio());
					// Convert from dictionary of face to our Face Db Entity.
					$face = Face::fromModel($image->getId(), $normFace);
					// Save the normalized Face to insert on database later.
					$faces[] = $face;
				}

				// Save new faces fo database
				$endMillis = round(microtime(true) * 1000);
				$duration = (int) max($endMillis - $startMillis, 0);
				$this->imageMapper->imageProcessed($image, $faces, $duration);

				// Release lock of file.
				$this->lockingProvider->releaseLock($lockKey, $lockType);
			} catch (\OCP\Lock\LockedException $e) {
				$this->logInfo('Faces found: 0. Image will be skipped because it is locked');
			} catch (\Exception $e) {
				if ($e->getMessage() === "std::bad_alloc") {
					throw new \RuntimeException("Not enough memory to run face recognition! Please look FAQ at https://github.com/matiasdelellis/facerecognition/wiki/FAQ");
				}
				$this->logInfo('Faces found: 0. Image will be skipped because of the following error: ' . $e->getMessage());
				$this->logDebug((string) $e);

				// Save an empty entry so it can be analyzed again later
				$this->imageMapper->imageProcessed($image, array(), 0, $e);
			} finally {
				// Clean temporary image.
				if (isset($tempImage)) {
					$tempImage->clean();
				}
				// If there are temporary files from external files, they must also be cleaned.
				$this->fileService->clean();
			}
		}

		return true;
	}
}