media/learning/common/experiment_helper.cc

// Copyright 2019 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.

#include "media/learning/common/experiment_helper.h"

namespace media {
namespace learning {

ExperimentHelper::ExperimentHelper(
    std::unique_ptr<LearningTaskController> controller)
    : controller_(std::move(controller)) {}

ExperimentHelper::~ExperimentHelper() {
  CancelObservationIfNeeded();
}

void ExperimentHelper::BeginObservation(const FeatureDictionary& dictionary) {
  if (!controller_)
    return;

  CancelObservationIfNeeded();

  // Get the features that our task needs.
  FeatureVector features;
  dictionary.Lookup(controller_->GetLearningTask(), &features);

  observation_id_ = base::UnguessableToken::Create();
  controller_->BeginObservation(observation_id_, features);
}

void ExperimentHelper::CompleteObservationIfNeeded(const TargetValue& target) {
  if (!observation_id_)
    return;

  controller_->CompleteObservation(observation_id_, target);
  observation_id_ = base::UnguessableToken::Null();
}

void ExperimentHelper::CancelObservationIfNeeded() {
  if (!observation_id_)
    return;

  controller_->CancelObservation(observation_id_);
  observation_id_ = base::UnguessableToken::Null();
}

}  // namespace learning
}  // namespace media