Completed
Push — master ( 2ee167...4713e8 )
by Matias
23s queued 13s
created

ImageMapper::resetErrors()   A

Complexity

Conditions 1
Paths 1

Size

Total Lines 9
Code Lines 8

Duplication

Lines 0
Ratio 0 %

Code Coverage

Tests 0
CRAP Score 2

Importance

Changes 0
Metric Value
cc 1
eloc 8
nc 1
nop 1
dl 0
loc 9
ccs 0
cts 9
cp 0
crap 2
rs 10
c 0
b 0
f 0
1
<?php
2
/**
3
 * @copyright Copyright (c) 2017-2020, Matias De lellis <[email protected]>
4
 * @copyright Copyright (c) 2018-2019, Branko Kokanovic <[email protected]>
5
 *
6
 * @author Branko Kokanovic <[email protected]>
7
 *
8
 * @license GNU AGPL version 3 or any later version
9
 *
10
 * This program is free software: you can redistribute it and/or modify
11
 * it under the terms of the GNU Affero General Public License as
12
 * published by the Free Software Foundation, either version 3 of the
13
 * License, or (at your option) any later version.
14
 *
15
 * This program is distributed in the hope that it will be useful,
16
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
17
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
18
 * GNU Affero General Public License for more details.
19
 *
20
 * You should have received a copy of the GNU Affero General Public License
21
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
22
 *
23
 */
24
namespace OCA\FaceRecognition\Db;
25
26
use OCP\IDBConnection;
27
use OCP\IUser;
28
29
use OCP\AppFramework\Db\QBMapper;
30
use OCP\AppFramework\Db\DoesNotExistException;
31
use OCP\DB\QueryBuilder\IQueryBuilder;
32
33
class ImageMapper extends QBMapper {
34
	/** @var FaceMapper Face mapper*/
35
	private $faceMapper;
36
37
	public function __construct(IDBConnection $db, FaceMapper $faceMapper) {
38
		parent::__construct($db, 'facerecog_images', '\OCA\FaceRecognition\Db\Image');
39
		$this->faceMapper = $faceMapper;
40
	}
41
42
	/**
43
	 * @param string $userId Id of user
44
	 * @param int $imageId Id of Image to get
45
	 */
46 4
	public function find(string $userId, int $imageId) {
47 4
		$qb = $this->db->getQueryBuilder();
48 4
		$qb->select('id', 'file', 'is_processed', 'error', 'last_processed_time', 'processing_duration')
49 4
			->from($this->getTableName(), 'i')
50 4
			->where($qb->expr()->eq('user', $qb->createNamedParameter($userId)))
51 4
			->andWhere($qb->expr()->eq('id', $qb->createNamedParameter($imageId)));
52
		try {
53 4
			return $this->findEntity($qb);
54
		} catch (DoesNotExistException $e) {
55
			return null;
56
		}
57
	}
58
59
	/**
60
	 * @param string $userId Id of user
61
	 * @param int $modelId Id of model to get
62
	 */
63
	public function findAll(string $userId, int $modelId) {
64
		$qb = $this->db->getQueryBuilder();
65
		$qb->select('id', 'file', 'is_processed', 'error', 'last_processed_time', 'processing_duration')
66
			->from($this->getTableName())
67
			->where($qb->expr()->eq('user', $qb->createNamedParameter($userId)))
68
			->andWhere($qb->expr()->eq('model', $qb->createNamedParameter($modelId)));
69
		return $this->findEntities($qb);
70
	}
71
72
	/**
73
	 * @param string $userId Id of user
74
	 * @param int $modelId Id of model
75
	 * @param int $fileId Id of file to get Image
76
	 */
77
	public function findFromFile(string $userId, int $modelId, int $fileId) {
78
		$qb = $this->db->getQueryBuilder();
79
		$qb->select('id', 'is_processed', 'error')
80
			->from($this->getTableName(), 'i')
81
			->where($qb->expr()->eq('user', $qb->createNamedParameter($userId)))
82
			->andwhere($qb->expr()->eq('model', $qb->createNamedParameter($modelId)))
83
			->andWhere($qb->expr()->eq('file', $qb->createNamedParameter($fileId)));
84
85
		try {
86
			return $this->findEntity($qb);
87
		} catch (DoesNotExistException $e) {
88
			return null;
89
		}
90
	}
91
92 7
	public function imageExists(Image $image) {
93 7
		$qb = $this->db->getQueryBuilder();
94
		$query = $qb
95 7
			->select(['id'])
96 7
			->from($this->getTableName())
97 7
			->where($qb->expr()->eq('user', $qb->createParameter('user')))
98 7
			->andWhere($qb->expr()->eq('file', $qb->createParameter('file')))
99 7
			->andWhere($qb->expr()->eq('model', $qb->createParameter('model')))
100 7
			->setParameter('user', $image->getUser())
101 7
			->setParameter('file', $image->getFile())
102 7
			->setParameter('model', $image->getModel());
103 7
		$resultStatement = $query->execute();
104 7
		$row = $resultStatement->fetch();
105 7
		$resultStatement->closeCursor();
106 7
		return $row ? (int)$row['id'] : null;
107
	}
108
109
	public function countImages(int $model): int {
110
		$qb = $this->db->getQueryBuilder();
111
		$query = $qb
112
			->select($qb->createFunction('COUNT(' . $qb->getColumnName('id') . ')'))
113
			->from($this->getTableName())
114
			->where($qb->expr()->eq('model', $qb->createParameter('model')))
115
			->setParameter('model', $model);
116
		$resultStatement = $query->execute();
117
		$data = $resultStatement->fetch(\PDO::FETCH_NUM);
118
		$resultStatement->closeCursor();
119
120
		return (int)$data[0];
121
	}
122
123
	public function countProcessedImages(int $model): int {
124
		$qb = $this->db->getQueryBuilder();
125
		$query = $qb
126
			->select($qb->createFunction('COUNT(' . $qb->getColumnName('id') . ')'))
127
			->from($this->getTableName())
128
			->where($qb->expr()->eq('model', $qb->createParameter('model')))
129
			->andWhere($qb->expr()->eq('is_processed', $qb->createParameter('is_processed')))
130
			->setParameter('model', $model)
131
			->setParameter('is_processed', True);
132
		$resultStatement = $query->execute();
133
		$data = $resultStatement->fetch(\PDO::FETCH_NUM);
134
		$resultStatement->closeCursor();
135
136
		return (int)$data[0];
137
	}
138
139
	public function avgProcessingDuration(int $model): int {
140
		$qb = $this->db->getQueryBuilder();
141
		$query = $qb
142
			->select($qb->createFunction('AVG(' . $qb->getColumnName('processing_duration') . ')'))
143
			->from($this->getTableName())
144
			->where($qb->expr()->eq('model', $qb->createParameter('model')))
145
			->andWhere($qb->expr()->eq('is_processed', $qb->createParameter('is_processed')))
146
			->setParameter('model', $model)
147
			->setParameter('is_processed', True);
148
		$resultStatement = $query->execute();
149
		$data = $resultStatement->fetch(\PDO::FETCH_NUM);
150
		$resultStatement->closeCursor();
151
152
		return (int)$data[0];
153
	}
154
155 3
	public function countUserImages(string $userId, int $model, $processed = false): int {
156 3
		$qb = $this->db->getQueryBuilder();
157
		$query = $qb
158 3
			->select($qb->createFunction('COUNT(' . $qb->getColumnName('id') . ')'))
159 3
			->from($this->getTableName())
160 3
			->where($qb->expr()->eq('user', $qb->createParameter('user')))
161 3
			->andWhere($qb->expr()->eq('model', $qb->createParameter('model')))
162 3
			->setParameter('user', $userId)
163 3
			->setParameter('model', $model);
164
165 3
		if ($processed) {
166 1
			$query->andWhere($qb->expr()->eq('is_processed', $qb->createParameter('is_processed')))
167 1
			      ->setParameter('is_processed', true);
168
		}
169
170 3
		$resultStatement = $query->execute();
171 3
		$data = $resultStatement->fetch(\PDO::FETCH_NUM);
172 3
		$resultStatement->closeCursor();
173
174 3
		return (int)$data[0];
175
	}
176
177
	/**
178
	 * @param IUser|null $user User for which to get images for. If not given, all images from instance are returned.
179
	 * @param int $modelId Model Id to get images for.
180
	 */
181 9
	public function findImagesWithoutFaces(IUser $user = null, int $modelId): array {
182 9
		$qb = $this->db->getQueryBuilder();
183
		$qb
184 9
			->select(['id', 'user', 'file', 'model'])
185 9
			->from($this->getTableName())
186 9
			->where($qb->expr()->eq('is_processed',  $qb->createParameter('is_processed')))
187 9
			->andWhere($qb->expr()->eq('model', $qb->createNamedParameter($modelId)))
188 9
			->setParameter('is_processed', false, IQueryBuilder::PARAM_BOOL);
189 9
		if (!is_null($user)) {
190 9
			$qb->andWhere($qb->expr()->eq('user', $qb->createNamedParameter($user->getUID())));
191
		}
192 9
		return $this->findEntities($qb);
193
	}
194
195 6
	public function findImages(string $userId, int $model): array {
196 6
		$qb = $this->db->getQueryBuilder();
197 6
		$qb->select('i.id', 'i.file')
198 6
			->from($this->getTableName(), 'i')
199 6
			->where($qb->expr()->eq('user', $qb->createNamedParameter($userId)))
200 6
			->andWhere($qb->expr()->eq('model', $qb->createNamedParameter($model)));
201
202 6
		$images = $this->findEntities($qb);
203 6
		return $images;
204
	}
205
206
	public function findFromPersonLike(string $userId, int $model, string $name, $offset = null, $limit = null): array {
207
		$qb = $this->db->getQueryBuilder();
208
		$qb->select('i.id', 'i.file')
209
			->from($this->getTableName(), 'i')
210
			->innerJoin('i', 'facerecog_faces', 'f', $qb->expr()->eq('f.image', 'i.id'))
211
			->innerJoin('i', 'facerecog_persons', 'p', $qb->expr()->eq('f.person', 'p.id'))
212
			->where($qb->expr()->eq('p.user', $qb->createNamedParameter($userId)))
213
			->andWhere($qb->expr()->eq('model', $qb->createNamedParameter($model)))
214
			->andWhere($qb->expr()->eq('is_processed', $qb->createNamedParameter(True)))
215
			->andWhere($qb->expr()->like($qb->func()->lower('p.name'), $qb->createParameter('query')));
216
217
		$query = '%' . $this->db->escapeLikeParameter(strtolower($name)) . '%';
218
		$qb->setParameter('query', $query);
219
220
		$qb->setFirstResult($offset);
221
		$qb->setMaxResults($limit);
222
223
		return $this->findEntities($qb);
224
	}
225
226
	/**
227
	 * Writes to DB that image has been processed. Previously found faces are deleted and new ones are inserted.
228
	 * If there is exception, its stack trace is also updated.
229
	 *
230
	 * @param Image $image Image to be updated
231
	 * @param Face[] $faces Faces to insert
232
	 * @param int $duration Processing time, in milliseconds
233
	 * @param \Exception|null $e Any exception that happened during image processing
234
	 */
235 4
	public function imageProcessed(Image $image, array $faces, int $duration, \Exception $e = null) {
236 4
		$this->db->beginTransaction();
237
		try {
238
			// Update image itself
239
			//
240 4
			$error = null;
241 4
			if ($e !== null) {
242 1
				$error = substr($e->getMessage(), 0, 1024);
243
			}
244
245 4
			$qb = $this->db->getQueryBuilder();
246 4
			$qb->update($this->getTableName())
247 4
				->set("is_processed", $qb->createNamedParameter(true, IQueryBuilder::PARAM_BOOL))
248 4
				->set("error", $qb->createNamedParameter($error))
249 4
				->set("last_processed_time", $qb->createNamedParameter(new \DateTime(), IQueryBuilder::PARAM_DATE))
250 4
				->set("processing_duration", $qb->createNamedParameter($duration))
251 4
				->where($qb->expr()->eq('id', $qb->createNamedParameter($image->id)))
252 4
				->execute();
253
254
			// Delete all previous faces
255
			//
256 4
			$qb = $this->db->getQueryBuilder();
257 4
			$qb->delete('facerecog_faces')
258 4
				->where($qb->expr()->eq('image', $qb->createNamedParameter($image->id)))
259 4
				->execute();
260
261
			// Insert all faces
262
			//
263 4
			foreach ($faces as $face) {
264 1
				$this->faceMapper->insertFace($face, $this->db);
265
			}
266
267 4
			$this->db->commit();
268
		} catch (\Exception $e) {
269
			$this->db->rollBack();
270
			throw $e;
271
		}
272 4
	}
273
274
	/**
275
	 * Resets image by deleting all associated faces and prepares it to be processed again
276
	 *
277
	 * @param Image $image Image to reset
278
	 */
279
	public function resetImage(Image $image) {
280
		$qb = $this->db->getQueryBuilder();
281
		$qb->update($this->getTableName())
282
			->set("is_processed", $qb->createNamedParameter(false, IQueryBuilder::PARAM_BOOL))
283
			->set("error", $qb->createNamedParameter(null))
284
			->set("last_processed_time", $qb->createNamedParameter(null))
285
			->where($qb->expr()->eq('user', $qb->createNamedParameter($image->getUser())))
286
			->andWhere($qb->expr()->eq('file', $qb->createNamedParameter($image->getFile())))
287
			->andWhere($qb->expr()->eq('model', $qb->createNamedParameter($image->getModel())))
288
			->execute();
289
	}
290
291
	/**
292
	 * Resets all image with error from that user and prepares it to be processed again
293
	 *
294
	 * @param string $userId User to reset errors
295
	 */
296
	public function resetErrors(string $userId) {
297
		$qb = $this->db->getQueryBuilder();
298
		$qb->update($this->getTableName())
299
			->set("is_processed", $qb->createNamedParameter(false, IQueryBuilder::PARAM_BOOL))
300
			->set("error", $qb->createNamedParameter(null))
301
			->set("last_processed_time", $qb->createNamedParameter(null))
302
			->where($qb->expr()->eq('user', $qb->createNamedParameter($userId)))
303
			->andWhere($qb->expr()->isNotNull('error'))
304
			->execute();
305
	}
306
307
	/**
308
	 * Deletes all images from that user.
309
	 *
310
	 * @param string $userId User to drop images from table.
311
	 */
312 28
	public function deleteUserImages(string $userId) {
313 28
		$qb = $this->db->getQueryBuilder();
314 28
		$qb->delete($this->getTableName())
315 28
			->where($qb->expr()->eq('user', $qb->createNamedParameter($userId)))
316 28
			->execute();
317 28
	}
318
319
	/**
320
	 * Deletes all images from that user and Model
321
	 *
322
	 * @param string $userId User to drop images from table.
323
	 * @param int $modelId model to drop images from table.
324
	 */
325
	public function deleteUserModel(string $userId, $modelId) {
326
		$qb = $this->db->getQueryBuilder();
327
		$qb->delete($this->getTableName())
328
			->where($qb->expr()->eq('user', $qb->createNamedParameter($userId)))
329
			->andWhere($qb->expr()->eq('model', $qb->createNamedParameter($modelId)))
330
			->execute();
331
	}
332
333
}
334