kenmirrors/ente
1
0
Fork 0
mirror of https://github.com/ente-io/ente.git synced 2025-07-03 14:06:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

rename

Browse Source
This commit is contained in:
Neeraj Gupta 2024-08-16 17:23:38 +05:30
parent 2cf8bd14f9
commit b6d5ebc5b4
4 changed files with 11 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
mobile/lib/db/ml/embeddings_db.dart → mobile/lib/db/ml/clip_db.dart
View File

@ -6,8 +6,9 @@ import "package:photos/db/ml/db.dart";
import "package:photos/db/ml/db_fields.dart"; import "package:photos/db/ml/db_fields.dart";
import "package:photos/events/embedding_updated_event.dart"; import "package:photos/events/embedding_updated_event.dart";
import "package:photos/models/ml/clip.dart"; import "package:photos/models/ml/clip.dart";
import "package:photos/models/ml/ml_versions.dart";
extension EmbeddingsDB on MLDataDB { extension ClipDB on MLDataDB {
static const databaseName = "ente.embeddings.db"; static const databaseName = "ente.embeddings.db";
Future<List<ClipEmbedding>> getAll() async { Future<List<ClipEmbedding>> getAll() async {
@ -28,10 +29,12 @@ extension EmbeddingsDB on MLDataDB {
return result; return result;
} }
Future<int> getClipIndexedFileCount() async { Future<int> getClipIndexedFileCount({
int minimumMlVersion = clipMlVersion,
}) async {
final db = await MLDataDB.instance.asyncDB; final db = await MLDataDB.instance.asyncDB;
const String query = final String query =
'SELECT COUNT(DISTINCT $fileIDColumn) as count FROM $clipTable'; 'SELECT COUNT(DISTINCT $fileIDColumn) as count FROM $clipTable WHERE $mlVersionColumn >= $minimumMlVersion';
final List<Map<String, dynamic>> maps = await db.getAll(query); final List<Map<String, dynamic>> maps = await db.getAll(query);
return maps.first['count'] as int; return maps.first['count'] as int;
} }

2
mobile/lib/services/machine_learning/face_ml/face_recognition_service.dart
View File

@ -5,8 +5,8 @@ import "dart:ui" show Image;
import "package:logging/logging.dart"; import "package:logging/logging.dart";
import "package:photos/core/event_bus.dart"; import "package:photos/core/event_bus.dart";
import "package:photos/db/ml/clip_db.dart";
import "package:photos/db/ml/db.dart"; import "package:photos/db/ml/db.dart";
import "package:photos/db/ml/embeddings_db.dart";
import "package:photos/events/diff_sync_complete_event.dart"; import "package:photos/events/diff_sync_complete_event.dart";
import "package:photos/events/people_changed_event.dart"; import "package:photos/events/people_changed_event.dart";
import "package:photos/extensions/list.dart"; import "package:photos/extensions/list.dart";

2
mobile/lib/services/machine_learning/semantic_search/semantic_search_service.dart
View File

@ -9,8 +9,8 @@ import "package:logging/logging.dart";
import "package:photos/core/cache/lru_map.dart"; import "package:photos/core/cache/lru_map.dart";
import "package:photos/core/event_bus.dart"; import "package:photos/core/event_bus.dart";
import "package:photos/db/files_db.dart"; import "package:photos/db/files_db.dart";
import "package:photos/db/ml/clip_db.dart";
import "package:photos/db/ml/db.dart"; import "package:photos/db/ml/db.dart";
import "package:photos/db/ml/embeddings_db.dart";
import 'package:photos/events/embedding_updated_event.dart'; import 'package:photos/events/embedding_updated_event.dart';
import "package:photos/models/file/file.dart"; import "package:photos/models/file/file.dart";
import "package:photos/models/ml/clip.dart"; import "package:photos/models/ml/clip.dart";

5
mobile/lib/utils/ml_util.dart
View File

@ -6,8 +6,8 @@ import "package:flutter/services.dart" show PlatformException;
import "package:logging/logging.dart"; import "package:logging/logging.dart";
import "package:photos/core/configuration.dart"; import "package:photos/core/configuration.dart";
import "package:photos/db/files_db.dart"; import "package:photos/db/files_db.dart";
import "package:photos/db/ml/clip_db.dart";
import "package:photos/db/ml/db.dart"; import "package:photos/db/ml/db.dart";
import "package:photos/db/ml/embeddings_db.dart";
import "package:photos/models/file/extensions/file_props.dart"; import "package:photos/models/file/extensions/file_props.dart";
import "package:photos/models/file/file.dart"; import "package:photos/models/file/file.dart";
import "package:photos/models/file/file_type.dart"; import "package:photos/models/file/file_type.dart";
@ -51,8 +51,7 @@ class FileMLInstruction {
Future<IndexStatus> getIndexStatus() async { Future<IndexStatus> getIndexStatus() async {
try { try {
final int indexableFiles = (await getIndexableFileIDs()).length; final int indexableFiles = (await getIndexableFileIDs()).length;
final int facesIndexedFiles = final int facesIndexedFiles = await MLDataDB.instance.getIndexedFileCount();
await MLDataDB.instance.getIndexedFileCount();
final int clipIndexedFiles = final int clipIndexedFiles =
await MLDataDB.instance.getClipIndexedFileCount(); await MLDataDB.instance.getClipIndexedFileCount();
final int indexedFiles = math.min(facesIndexedFiles, clipIndexedFiles); final int indexedFiles = math.min(facesIndexedFiles, clipIndexedFiles);