diff options
Diffstat (limited to 'webrtc/modules/video_processing/content_analysis.h')
-rw-r--r-- | webrtc/modules/video_processing/content_analysis.h | 87 |
1 files changed, 87 insertions, 0 deletions
diff --git a/webrtc/modules/video_processing/content_analysis.h b/webrtc/modules/video_processing/content_analysis.h new file mode 100644 index 0000000000..d3a11bd091 --- /dev/null +++ b/webrtc/modules/video_processing/content_analysis.h @@ -0,0 +1,87 @@ +/* + * 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_ |