frame_instrumentation_evaluation.h

Enable keyboard shortcuts

/*

 * Copyright 2024 The WebRTC project authors. All rights reserved.

 * Use of this source code is governed by a BSD-style license

 * that can be found in the LICENSE file in the root of the source

 * tree. An additional intellectual property rights grant can be found

 * in the file PATENTS.  All contributing project authors may

 * be found in the AUTHORS file in the root of the source tree.

*/

#ifndef VIDEO_CORRUPTION_DETECTION_FRAME_INSTRUMENTATION_EVALUATION_H_

#define VIDEO_CORRUPTION_DETECTION_FRAME_INSTRUMENTATION_EVALUATION_H_

#include "api/video/video_content_type.h"

#include "api/video/video_frame.h"

#include "common_video/frame_instrumentation_data.h"

#include "video/corruption_detection/corruption_classifier.h"

#include "video/corruption_detection/halton_frame_sampler.h"

namespace webrtc {

class CorruptionScoreObserver {

 public:

  CorruptionScoreObserver() = default;

  virtual ~CorruptionScoreObserver() = default;

  // Results of corruption detection for a single frame, with a likelihood score

  // in the range [0.0, 1.0].

  virtual void OnCorruptionScore(double corruption_score,

                                 VideoContentType content_type) = 0;

};

class FrameInstrumentationEvaluation {

 public:

  explicit FrameInstrumentationEvaluation(CorruptionScoreObserver* observer);

  void OnInstrumentedFrame(const FrameInstrumentationData& data,

                           const VideoFrame& frame,

                           VideoContentType frame_type);

 private:

  CorruptionScoreObserver* const observer_;

  HaltonFrameSampler frame_sampler_;

  CorruptionClassifier classifier_;

};

}  // namespace webrtc

#endif  // VIDEO_CORRUPTION_DETECTION_FRAME_INSTRUMENTATION_EVALUATION_H_