blob: 32bf9cb0c52e344b4cdeaa51d718ae76c119f9fb [file] [log] [blame]
// Copyright 2018 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.
#ifndef MEDIA_CAPABILITIES_LEARNING_HELPER_H_
#define MEDIA_CAPABILITIES_LEARNING_HELPER_H_
#include <memory>
#include "base/macros.h"
#include "base/threading/sequence_bound.h"
#include "media/base/media_export.h"
#include "media/capabilities/video_decode_stats_db.h"
#include "media/learning/impl/feature_provider.h"
#include "media/learning/impl/learning_session_impl.h"
namespace media {
// Helper class to allow MediaCapabilities to log training examples to a
// media::learning LearningTask.
class MEDIA_EXPORT LearningHelper {
public:
// |feature_factory| lets us register FeatureProviders with those
// LearningTasks that include standard features.
LearningHelper(learning::FeatureProviderFactoryCB feature_factory);
~LearningHelper();
// |origin| is sent in separately since it's somewhat hacky. This should be
// provided by the FeatureProvider, but the way LearningHelper is used, it
// doesn't have access to the origin. Per-frame LearningTaskControllers will
// be able to do this much more easily.
void AppendStats(const VideoDecodeStatsDB::VideoDescKey& video_key,
learning::FeatureValue origin,
const VideoDecodeStatsDB::DecodeStatsEntry& new_stats);
private:
// Convenience function to begin and complete an observation.
void AddExample(learning::LearningTaskController* controller,
const learning::LabelledExample& example);
// Learning session for our profile. Normally, we'd not have one of these
// directly, but would instead get one that's connected to a browser profile.
// For now, however, we just instantiate one and assume that we'll be
// destroyed when the profile changes / history is cleared.
std::unique_ptr<learning::LearningSessionImpl> learning_session_;
// Controllers for each task.
std::unique_ptr<learning::LearningTaskController> base_table_controller_;
std::unique_ptr<learning::LearningTaskController> base_tree_controller_;
std::unique_ptr<learning::LearningTaskController> enhanced_tree_controller_;
std::unique_ptr<learning::LearningTaskController> unweighted_tree_controller_;
std::unique_ptr<learning::LearningTaskController> binary_tree_controller_;
};
} // namespace media
#endif // MEDIA_CAPABILITIES_LEARNING_HELPER_H_