OCA\FaceRecognition\BackgroundJob\Tasks\CreateClustersTask::getNewClusters()

Complexity

Conditions	15
Paths	18

Size

Total Lines	72
Code Lines	45

Duplication

Lines	0
Ratio	0 %

Code Coverage

Tests	21
CRAP Score	51.1227

Importance

Changes	3
Bugs	0	Features	1

<?php
	private function getNewClusters(array $faces): array {
		// Clustering parameters
		$sensitivity = $this->settingsService->getSensitivity();

		if (Requirements::pdlibLoaded()) {
			// Create edges (neighbors) for Chinese Whispers
			$edges = array();
			$faces_count = count($faces);
			for ($i = 0; $i < $faces_count; $i++) {
				$face1 = $faces[$i];
				if (!isset($face1->descriptor)) {
					$edges[] = array($i, $i);
					continue;
				}
				for ($j = $i; $j < $faces_count; $j++) {
					$face2 = $faces[$j];
					if (!isset($face2->descriptor)) {
						continue;
					}
					$distance = dlib_vector_length($face1->descriptor, $face2->descriptor);

The function dlib_vector_length was not found. Maybe you did not declare it correctly or list all dependencies?

					if ($distance < $sensitivity) {
						$edges[] = array($i, $j);
					}
				}
			}

			// Given the edges get the list of labels (found clusters) for each face.
			$newChineseClustersByIndex = dlib_chinese_whispers($edges);

The function dlib_chinese_whispers was not found. Maybe you did not declare it correctly or list all dependencies?

		} else {
			// Create edges (neighbors) for Chinese Whispers
			$edges = array();
			$faces_count = count($faces);

			for ($i = 0; $i < $faces_count; $i++) {
				$face1 = $faces[$i];
				if (!isset($face1->descriptor)) {
					$edges[] = array($i, $i);
					continue;
				}
				for ($j = $i; $j < $faces_count; $j++) {
					$face2 = $faces[$j];
					if (!isset($face2->descriptor)) {
						continue;
					}
					$distance = Euclidean::distance($face1->descriptor, $face2->descriptor);
					if ($distance < $sensitivity) {
						$edges[] = array($i, $j);
					}
				}
			}

			// The clustering algorithm actually expects ordered lists.
			$oedges = [];
			ChineseWhispers::convert_unordered_to_ordered($edges, $oedges);
			usort($oedges, function($a, $b) {
				if ($a[0] === $b[0]) return $a[1] - $b[1];
				return $a[0] - $b[0];
			});

			// Given the edges get the list of labels (found clusters) for each face.
			$newChineseClustersByIndex = [];
			ChineseWhispers::predict($oedges, $newChineseClustersByIndex);
		}

		$newClusters = array();
		for ($i = 0, $c = count($newChineseClustersByIndex); $i < $c; $i++) {
			if (!isset($newClusters[$newChineseClustersByIndex[$i]])) {
				$newClusters[$newChineseClustersByIndex[$i]] = array();
			}
			$newClusters[$newChineseClustersByIndex[$i]][] = $faces[$i]->id;
		}
		return $newClusters;
	}