85 lines
2.7 KiB
C++
85 lines
2.7 KiB
C++
/*
|
|
* Copyright (c) 2011 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.
|
|
*/
|
|
|
|
/*
|
|
* content_analysis.h
|
|
*/
|
|
|
|
#ifndef VPM_CONTENT_ANALYSIS_H
|
|
#define VPM_CONTENT_ANALYSIS_H
|
|
|
|
#include "typedefs.h"
|
|
#include "module_common_types.h"
|
|
|
|
#include "spatial_resampler.h"
|
|
|
|
namespace webrtc {
|
|
|
|
class VPMContentAnalysis
|
|
{
|
|
public:
|
|
VPMContentAnalysis();
|
|
~VPMContentAnalysis();
|
|
|
|
//Initialize ContentAnalysis - should be called prior to extractContentFeature
|
|
//Inputs: width, height
|
|
//Return value: 0 if OK, negative value upon error
|
|
WebRtc_Word32 Initialize( WebRtc_UWord16 width, WebRtc_UWord16 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
|
|
WebRtc_Word32 Release();
|
|
|
|
private:
|
|
|
|
//return motion metrics
|
|
VideoContentMetrics* ContentMetrics();
|
|
|
|
//Normalized temporal difference metric: for motion magnitude
|
|
WebRtc_Word32 TemporalDiffMetric();
|
|
|
|
//Motion metric method: call 2 metrics (magnitude and size)
|
|
WebRtc_Word32 ComputeMotionMetrics();
|
|
|
|
//Spatial metric method: computes the 3 frame-average spatial prediction errors (1x2,2x1,2x2)
|
|
WebRtc_Word32 ComputeSpatialMetrics();
|
|
|
|
const WebRtc_UWord8* _origFrame;
|
|
WebRtc_UWord8* _prevFrame;
|
|
WebRtc_UWord16 _width;
|
|
WebRtc_UWord16 _height;
|
|
|
|
|
|
//Content Metrics:
|
|
//stores the local average of the metrics
|
|
float _motionMagnitudeNZ; //for motion class
|
|
float _spatialPredErr; //for spatial class
|
|
float _spatialPredErrH; //for spatial class
|
|
float _spatialPredErrV; //for spatial class
|
|
float _sizeZeroMotion; //for motion class
|
|
float _motionPredErr; //for complexity class:
|
|
float _motionHorizontalness; //for coherence class
|
|
float _motionClusterDistortion; //for coherence class
|
|
|
|
bool _firstFrame;
|
|
bool _CAInit;
|
|
VideoContentMetrics* _cMetrics;
|
|
|
|
}; // end of VPMContentAnalysis class definition
|
|
|
|
} // namespace
|
|
|
|
#endif
|