0ct0pu5/ente
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

[mob] minor refactor

Browse source
This commit is contained in:
Neeraj Gupta 2024-03-19 15:18:07 +05:30
parent 3e4cf4b4f2
commit e83e8cdb6e
1 changed files with 17 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
mobile/lib/services/face_ml/face_ml_service.dart
View file

@ -425,11 +425,24 @@ class FaceMlService {
int fileAnalyzedCount = 0;
int fileSkippedCount = 0;
final stopwatch = Stopwatch()..start();
final split = enteFiles.splitMatch((e) => (e.localID ?? '') == '');
final List<EnteFile> filesWithLocalID = <EnteFile>[];
final List<EnteFile> filesWithoutLocalID = <EnteFile>[];
for (final EnteFile enteFile in enteFiles) {
if (_skipAnalysisEnteFile(enteFile, alreadyIndexedFiles)) {
fileSkippedCount++;
continue;
}
if ((enteFile.localID ?? '').isEmpty) {
filesWithoutLocalID.add(enteFile);
} else {
filesWithLocalID.add(enteFile);
}
}
// list of files where files with localID are first
final sortedBylocalID = <EnteFile>[];
sortedBylocalID.addAll(split.unmatched);
sortedBylocalID.addAll(split.matched);
sortedBylocalID.addAll(filesWithLocalID);
sortedBylocalID.addAll(filesWithoutLocalID);
final List<List<EnteFile>> chunks = sortedBylocalID.chunks(kParallelism);
outerLoop:
for (final chunk in chunks) {
@ -437,9 +450,7 @@ class FaceMlService {
final List<int> fileIds = [];
// Try to find embeddings on the remote server
for (final f in chunk) {
if (!alreadyIndexedFiles.contains(f.uploadedFileID)) {
fileIds.add(f.uploadedFileID!);
}
fileIds.add(f.uploadedFileID!);
}
try {
final res =