70 lines
2.0 KiB
C++
70 lines
2.0 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.
|
|
*/
|
|
|
|
/*
|
|
* 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
|
|
|