webrtc/modules/video_processing/main/source/deflickering.h

70 lines
2.0 KiB
C
Raw Normal View History

/*
* 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.
*/
/*
* deflickering.h
*/
#ifndef VPM_DEFLICKERING_H
#define VPM_DEFLICKERING_H
#include "typedefs.h"
#include "video_processing.h"
#include <cstring> // NULL
namespace webrtc {
class VPMDeflickering
{
public:
VPMDeflickering();
~VPMDeflickering();
WebRtc_Word32 ChangeUniqueId(WebRtc_Word32 id);
void Reset();
WebRtc_Word32 ProcessFrame(WebRtc_UWord8* frame,
WebRtc_UWord32 width,
WebRtc_UWord32 height,
WebRtc_UWord32 timestamp,
VideoProcessingModule::FrameStats& stats);
private:
WebRtc_Word32 PreDetection(WebRtc_UWord32 timestamp,
const VideoProcessingModule::FrameStats& stats);
WebRtc_Word32 DetectFlicker();
enum { kMeanBufferLength = 32 };
enum { kFrameHistorySize = 15 };
enum { kNumProbs = 12 };
enum { kNumQuants = kNumProbs + 2 };
enum { kMaxOnlyLength = 5 };
WebRtc_Word32 _id;
WebRtc_UWord32 _meanBufferLength;
WebRtc_UWord8 _detectionState; // 0: No flickering
// 1: Flickering detected
// 2: In flickering
WebRtc_Word32 _meanBuffer[kMeanBufferLength];
WebRtc_UWord32 _timestampBuffer[kMeanBufferLength];
WebRtc_UWord32 _frameRate;
static const WebRtc_UWord16 _probUW16[kNumProbs];
static const WebRtc_UWord16 _weightUW16[kNumQuants - kMaxOnlyLength];
WebRtc_UWord8 _quantHistUW8[kFrameHistorySize][kNumQuants];
};
} //namespace
#endif // VPM_DEFLICKERING_H