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
66
67
68
69
70
71
72
|
// Copyright 2020 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "chrome/browser/language_detection/language_detection_model_service_factory.h"
#include "base/memory/scoped_refptr.h"
#include "base/no_destructor.h"
#include "base/task/sequenced_task_runner.h"
#include "base/task/task_traits.h"
#include "base/task/thread_pool.h"
#include "chrome/browser/optimization_guide/optimization_guide_keyed_service.h"
#include "chrome/browser/optimization_guide/optimization_guide_keyed_service_factory.h"
#include "chrome/browser/profiles/profile.h"
#include "components/language_detection/core/browser/language_detection_model_service.h"
#include "components/translate/core/common/translate_util.h"
// static
language_detection::LanguageDetectionModelService*
LanguageDetectionModelServiceFactory::GetForProfile(Profile* profile) {
return static_cast<language_detection::LanguageDetectionModelService*>(
GetInstance()->GetServiceForBrowserContext(profile, true));
}
// static
LanguageDetectionModelServiceFactory*
LanguageDetectionModelServiceFactory::GetInstance() {
static base::NoDestructor<LanguageDetectionModelServiceFactory> factory;
return factory.get();
}
LanguageDetectionModelServiceFactory::LanguageDetectionModelServiceFactory()
: ProfileKeyedServiceFactory(
"LanguageDetectionModelService",
ProfileSelections::Builder()
.WithRegular(ProfileSelection::kOwnInstance)
// TODO(crbug.com/40257657): Check if this service is needed in
// Guest mode.
.WithGuest(ProfileSelection::kOwnInstance)
// TODO(crbug.com/41488885): Check if this service is needed for
// Ash Internals.
.WithAshInternals(ProfileSelection::kOwnInstance)
.Build()) {
if (translate::IsTFLiteLanguageDetectionEnabled()) {
DependsOn(OptimizationGuideKeyedServiceFactory::GetInstance());
}
}
LanguageDetectionModelServiceFactory::~LanguageDetectionModelServiceFactory() =
default;
std::unique_ptr<KeyedService>
LanguageDetectionModelServiceFactory::BuildServiceInstanceForBrowserContext(
content::BrowserContext* context) const {
if (!translate::IsTFLiteLanguageDetectionEnabled()) {
return nullptr;
}
// The optimization guide service must be available for the translate model
// service to be created.
auto* opt_guide = OptimizationGuideKeyedServiceFactory::GetForProfile(
Profile::FromBrowserContext(context));
if (opt_guide) {
scoped_refptr<base::SequencedTaskRunner> background_task_runner =
base::ThreadPool::CreateSequencedTaskRunner(
{base::MayBlock(), base::TaskPriority::BEST_EFFORT});
return std::make_unique<language_detection::LanguageDetectionModelService>(
opt_guide, background_task_runner);
}
return nullptr;
}
|
