chromium/chrome/browser/accessibility/phrase_segmentation/dependency_parser_model_loader.cc

// Copyright 2024 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/accessibility/phrase_segmentation/dependency_parser_model_loader.h"

#include "base/files/file_path.h"
#include "base/files/file_util.h"
#include "base/metrics/histogram_macros.h"
#include "components/optimization_guide/core/optimization_guide_model_provider.h"

namespace {

// Load the model file at the provided file path.
base::File LoadModelFile(const base::FilePath& model_file_path) { … }

// Close the provided model file.
void CloseModelFile(base::File model_file) { … }

// Util class for recording the result of loading the dependency parser model.
// The result is recorded when it goes out of scope and its destructor is
// called.
class ScopedModelLoadingResultRecorder { … };

// The maximum number of pending model requests allowed to be kept
// by the DependencyParserModelLoader.
constexpr int kMaxPendingRequestsAllowed = …;

}  // namespace

DependencyParserModelLoader::DependencyParserModelLoader(
    optimization_guide::OptimizationGuideModelProvider* opt_guide,
    const scoped_refptr<base::SequencedTaskRunner>& background_task_runner)
    : … { … }

DependencyParserModelLoader::~DependencyParserModelLoader() { … }

void DependencyParserModelLoader::Shutdown() { … }

void DependencyParserModelLoader::UnloadModelFile() { … }

void DependencyParserModelLoader::NotifyModelUpdatesAndClear(
    bool is_model_available) { … }

void DependencyParserModelLoader::OnModelUpdated(
    optimization_guide::proto::OptimizationTarget optimization_target,
    base::optional_ref<const optimization_guide::ModelInfo> model_info) { … }

void DependencyParserModelLoader::OnModelFileLoaded(base::File model_file) { … }

base::File DependencyParserModelLoader::GetDependencyParserModelFile() { … }

void DependencyParserModelLoader::NotifyOnModelFileAvailable(
    NotifyModelAvailableCallback callback) { … }