99ab9447d1
BUG=webrtc:5259 Review URL: https://codereview.webrtc.org/1508793002 Cr-Commit-Position: refs/heads/master@{#10925}
88 lines
2.8 KiB
C++
88 lines
2.8 KiB
C++
/*
|
|
* Copyright (c) 2012 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 WEBRTC_MODULES_VIDEO_PROCESSING_CONTENT_ANALYSIS_H_
|
|
#define WEBRTC_MODULES_VIDEO_PROCESSING_CONTENT_ANALYSIS_H_
|
|
|
|
#include "webrtc/modules/include/module_common_types.h"
|
|
#include "webrtc/modules/video_processing/include/video_processing_defines.h"
|
|
#include "webrtc/typedefs.h"
|
|
#include "webrtc/video_frame.h"
|
|
|
|
namespace webrtc {
|
|
|
|
class VPMContentAnalysis {
|
|
public:
|
|
// When |runtime_cpu_detection| is true, runtime selection of an optimized
|
|
// code path is allowed.
|
|
explicit VPMContentAnalysis(bool runtime_cpu_detection);
|
|
~VPMContentAnalysis();
|
|
|
|
// Initialize ContentAnalysis - should be called prior to
|
|
// extractContentFeature
|
|
// Inputs: width, height
|
|
// Return value: 0 if OK, negative value upon error
|
|
int32_t Initialize(int width, int height);
|
|
|
|
// Extract content Feature - main function of ContentAnalysis
|
|
// Input: new frame
|
|
// Return value: pointer to structure containing content Analysis
|
|
// metrics or NULL value upon error
|
|
VideoContentMetrics* ComputeContentMetrics(const VideoFrame& inputFrame);
|
|
|
|
// Release all allocated memory
|
|
// Output: 0 if OK, negative value upon error
|
|
int32_t Release();
|
|
|
|
private:
|
|
// return motion metrics
|
|
VideoContentMetrics* ContentMetrics();
|
|
|
|
// Normalized temporal difference metric: for motion magnitude
|
|
typedef int32_t (VPMContentAnalysis::*TemporalDiffMetricFunc)();
|
|
TemporalDiffMetricFunc TemporalDiffMetric;
|
|
int32_t TemporalDiffMetric_C();
|
|
|
|
// Motion metric method: call 2 metrics (magnitude and size)
|
|
int32_t ComputeMotionMetrics();
|
|
|
|
// Spatial metric method: computes the 3 frame-average spatial
|
|
// prediction errors (1x2,2x1,2x2)
|
|
typedef int32_t (VPMContentAnalysis::*ComputeSpatialMetricsFunc)();
|
|
ComputeSpatialMetricsFunc ComputeSpatialMetrics;
|
|
int32_t ComputeSpatialMetrics_C();
|
|
|
|
#if defined(WEBRTC_ARCH_X86_FAMILY)
|
|
int32_t ComputeSpatialMetrics_SSE2();
|
|
int32_t TemporalDiffMetric_SSE2();
|
|
#endif
|
|
|
|
const uint8_t* orig_frame_;
|
|
uint8_t* prev_frame_;
|
|
int width_;
|
|
int height_;
|
|
int skip_num_;
|
|
int border_;
|
|
|
|
// Content Metrics: Stores the local average of the metrics.
|
|
float motion_magnitude_; // motion class
|
|
float spatial_pred_err_; // spatial class
|
|
float spatial_pred_err_h_; // spatial class
|
|
float spatial_pred_err_v_; // spatial class
|
|
bool first_frame_;
|
|
bool ca_Init_;
|
|
|
|
VideoContentMetrics* content_metrics_;
|
|
};
|
|
|
|
} // namespace webrtc
|
|
|
|
#endif // WEBRTC_MODULES_VIDEO_PROCESSING_CONTENT_ANALYSIS_H_
|