107 lines
3.5 KiB
C
107 lines
3.5 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.
|
||
|
|
*/
|
||
|
|
|
||
|
|
/*
|
||
|
|
* video_processing_impl.h
|
||
|
|
*/
|
||
|
|
#ifndef WEBRTC_MODULE_VIDEO_PROCESSING_IMPL_H
|
||
|
|
#define WEBRTC_MODULE_VIDEO_PROCESSING_IMPL_H
|
||
|
|
|
||
|
|
#include "video_processing.h"
|
||
|
|
#include "brightness_detection.h"
|
||
|
|
#include "color_enhancement.h"
|
||
|
|
#include "deflickering.h"
|
||
|
|
#include "denoising.h"
|
||
|
|
#include "frame_preprocessor.h"
|
||
|
|
|
||
|
|
namespace webrtc {
|
||
|
|
class CriticalSectionWrapper;
|
||
|
|
|
||
|
|
class VideoProcessingModuleImpl : public VideoProcessingModule
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
|
||
|
|
VideoProcessingModuleImpl(WebRtc_Word32 id);
|
||
|
|
|
||
|
|
virtual ~VideoProcessingModuleImpl();
|
||
|
|
|
||
|
|
virtual WebRtc_Word32 Version(WebRtc_Word8* version,
|
||
|
|
WebRtc_UWord32& remainingBufferInBytes,
|
||
|
|
WebRtc_UWord32& position) const;
|
||
|
|
|
||
|
|
WebRtc_Word32 Id() const;
|
||
|
|
|
||
|
|
virtual WebRtc_Word32 ChangeUniqueId(const WebRtc_Word32 id);
|
||
|
|
|
||
|
|
virtual void Reset();
|
||
|
|
|
||
|
|
virtual WebRtc_Word32 Deflickering(WebRtc_UWord8* frame,
|
||
|
|
WebRtc_UWord32 width,
|
||
|
|
WebRtc_UWord32 height,
|
||
|
|
WebRtc_UWord32 timestamp,
|
||
|
|
FrameStats& stats);
|
||
|
|
|
||
|
|
virtual WebRtc_Word32 Deflickering(VideoFrame& frame,
|
||
|
|
FrameStats& stats);
|
||
|
|
|
||
|
|
virtual WebRtc_Word32 Denoising(WebRtc_UWord8* frame,
|
||
|
|
WebRtc_UWord32 width,
|
||
|
|
WebRtc_UWord32 height);
|
||
|
|
|
||
|
|
virtual WebRtc_Word32 Denoising(VideoFrame& frame);
|
||
|
|
|
||
|
|
virtual WebRtc_Word32 BrightnessDetection(const WebRtc_UWord8* frame,
|
||
|
|
WebRtc_UWord32 width,
|
||
|
|
WebRtc_UWord32 height,
|
||
|
|
const FrameStats& stats);
|
||
|
|
|
||
|
|
virtual WebRtc_Word32 BrightnessDetection(const VideoFrame& frame,
|
||
|
|
const FrameStats& stats);
|
||
|
|
|
||
|
|
|
||
|
|
//Frame pre-processor functions
|
||
|
|
|
||
|
|
//Enable temporal decimation
|
||
|
|
virtual void EnableTemporalDecimation(bool enable);
|
||
|
|
|
||
|
|
virtual void SetInputFrameResampleMode(VideoFrameResampling resamplingMode);
|
||
|
|
|
||
|
|
//Enable content analysis
|
||
|
|
virtual void EnableContentAnalysis(bool enable);
|
||
|
|
|
||
|
|
//Set max frame rate
|
||
|
|
virtual WebRtc_Word32 SetMaxFrameRate(WebRtc_UWord32 maxFrameRate);
|
||
|
|
|
||
|
|
// Set Target Resolution: frame rate and dimension
|
||
|
|
virtual WebRtc_Word32 SetTargetResolution(WebRtc_UWord32 width, WebRtc_UWord32 height, WebRtc_UWord32 frameRate);
|
||
|
|
|
||
|
|
// Get decimated values: frame rate/dimension
|
||
|
|
virtual WebRtc_UWord32 DecimatedFrameRate();
|
||
|
|
virtual WebRtc_UWord32 DecimatedWidth() const;
|
||
|
|
virtual WebRtc_UWord32 DecimatedHeight() const;
|
||
|
|
|
||
|
|
// Preprocess:
|
||
|
|
virtual WebRtc_Word32 PreprocessFrame(const VideoFrame* frame, VideoFrame** processedFrame);
|
||
|
|
virtual VideoContentMetrics* ContentMetrics() const;
|
||
|
|
|
||
|
|
private:
|
||
|
|
WebRtc_Word32 _id;
|
||
|
|
CriticalSectionWrapper& _mutex;
|
||
|
|
|
||
|
|
VPMDeflickering _deflickering;
|
||
|
|
VPMDenoising _denoising;
|
||
|
|
VPMBrightnessDetection _brightnessDetection;
|
||
|
|
VPMFramePreprocessor _framePreProcessor;
|
||
|
|
};
|
||
|
|
|
||
|
|
} // namespace
|
||
|
|
|
||
|
|
#endif
|