#ifndef WELSVP_COMPLEXITYANALYSIS_H
#define WELSVP_COMPLEXITYANALYSIS_H
#include "util.h"
#include "memory.h"
#include "WelsFrameWork.h"
#include "IWelsVP.h"
#include "common.h"
WELSVP_NAMESPACE_BEGIN
typedef void (GOMSadFunc) (uint32_t* pGomSad, int32_t* pGomForegroundBlockNum, int32_t* pSad8x8,
uint8_t pBackgroundMbFlag);
typedef GOMSadFunc* PGOMSadFunc;
GOMSadFunc GomSampleSad;
GOMSadFunc GomSampleSadExceptBackground;
class CComplexityAnalysis : public IStrategy {
public:
CComplexityAnalysis (int32_t iCpuFlag);
~CComplexityAnalysis();
EResult Process (int32_t iType, SPixMap* pSrc, SPixMap* pRef);
EResult Set (int32_t iType, void* pParam);
EResult Get (int32_t iType, void* pParam);
private:
void AnalyzeFrameComplexityViaSad (SPixMap* pSrc, SPixMap* pRef);
int32_t GetFrameSadExcludeBackground (SPixMap* pSrc, SPixMap* pRef);
void AnalyzeGomComplexityViaSad (SPixMap* pSrc, SPixMap* pRef);
void AnalyzeGomComplexityViaVar (SPixMap* pSrc, SPixMap* pRef);
private:
PGOMSadFunc m_pfGomSad;
SComplexityAnalysisParam m_sComplexityAnalysisParam;
};
class CComplexityAnalysisScreen : public IStrategy {
public:
CComplexityAnalysisScreen (int32_t cpu_flag);
~CComplexityAnalysisScreen();
EResult Process (int32_t nType, SPixMap* src, SPixMap* ref);
EResult Set (int32_t nType, void* pParam);
EResult Get (int32_t nType, void* pParam);
private:
void GomComplexityAnalysisIntra (SPixMap* pSrc);
void GomComplexityAnalysisInter (SPixMap* pSrc, SPixMap* pRef, bool bScrollFlag);
private:
PSad16x16Func m_pSadFunc;
GetIntraPredPtr m_pIntraFunc[2];
SComplexityAnalysisScreenParam m_ComplexityAnalysisParam;
};
WELSVP_NAMESPACE_END
#endif