-
Notifications
You must be signed in to change notification settings - Fork 3.9k
/
firebase_ml_model_downloader.dart
65 lines (54 loc) 路 2.3 KB
/
firebase_ml_model_downloader.dart
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
// Copyright 2021 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
part of firebase_ml_model_downloader;
class FirebaseModelDownloader extends FirebasePluginPlatform {
FirebaseModelDownloader._({required this.app})
: super(app.name, 'plugins.flutter.io/firebase_ml_model_downloader');
// Cached and lazily loaded instance of [FirebaseModelDownloaderPlatform] to avoid
// creating a [MethodChannelFirebaseFunctions] when not needed or creating an
// instance with the default app before a user specifies an app.
FirebaseModelDownloaderPlatform? _delegatePackingProperty;
/// The [FirebaseApp] for this current [FirebaseModelDownloader] instance.
final FirebaseApp app;
static final Map<String, FirebaseModelDownloader> _cachedInstances = {};
/// Returns the underlying [FirebaseModelDownloaderPlatform] delegate for this
/// [FirebaseModelDownloader] instance. This is useful for testing purposes only.
@visibleForTesting
FirebaseModelDownloaderPlatform get delegate {
return _delegatePackingProperty ??=
FirebaseModelDownloaderPlatform.instanceFor(app: app);
}
/// Returns an instance using the default [FirebaseApp].
static FirebaseModelDownloader get instance {
return FirebaseModelDownloader.instanceFor(
app: Firebase.app(),
);
}
/// Returns an instance using a specified [FirebaseApp].
factory FirebaseModelDownloader.instanceFor({required FirebaseApp app}) {
return _cachedInstances.putIfAbsent(app.name, () {
return FirebaseModelDownloader._(app: app);
});
}
/// Gets the downloaded model file based on download type and conditions.
Future<FirebaseCustomModel> getModel(
String modelName,
FirebaseModelDownloadType downloadType, [
FirebaseModelDownloadConditions? conditions,
]) {
return delegate.getModel(
modelName,
downloadType,
conditions ?? FirebaseModelDownloadConditions(),
);
}
/// Lists all models downloaded to device.
Future<List<FirebaseCustomModel>> listDownloadedModels() {
return delegate.listDownloadedModels();
}
/// Deletes a locally downloaded model by name.
Future<void> deleteDownloadedModel(String modelName) {
return delegate.deleteDownloadedModel(modelName);
}
}