134 lines
5.3 KiB
C++
134 lines
5.3 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.
|
|
*/
|
|
|
|
#ifndef WEBRTC_MODULES_VIDEO_CODING_CODECS_TEST_FRAMEWORK_UNIT_TEST_H_
|
|
#define WEBRTC_MODULES_VIDEO_CODING_CODECS_TEST_FRAMEWORK_UNIT_TEST_H_
|
|
|
|
#include "test.h"
|
|
#include "event_wrapper.h"
|
|
|
|
// Disable "conditional expression is constant" warnings on the perfectly
|
|
// acceptable
|
|
// do { ... } while (0) constructions below.
|
|
// Refer to http://stackoverflow.com/questions/1946445/
|
|
// is-there-better-way-to-write-do-while0-construct-to-avoid-compiler-warnings
|
|
// for some discussion of the issue.
|
|
#pragma warning(disable : 4127)
|
|
|
|
#define VIDEO_TEST(expr) \
|
|
do \
|
|
{ \
|
|
_tests++; \
|
|
if (!(expr)) \
|
|
{ \
|
|
fprintf(stderr, "Error at line %i of %s\nAssertion failed: %s\n\n",\
|
|
__LINE__, __FILE__, #expr); \
|
|
_errors++; \
|
|
} \
|
|
} while (0)
|
|
|
|
#define VIDEO_TEST_EXIT_ON_ERR(expr) \
|
|
do \
|
|
{ \
|
|
if (!(expr)) \
|
|
{ \
|
|
fprintf(stderr, "Error at line %i of %s\nAssertion failed: %s\n", \
|
|
__LINE__, __FILE__, #expr); \
|
|
fprintf(stderr, "Exiting...\n\n"); \
|
|
exit(EXIT_FAILURE); \
|
|
} \
|
|
} while (0)
|
|
|
|
class VideoSource;
|
|
class UnitTestEncodeCompleteCallback;
|
|
class UnitTestDecodeCompleteCallback;
|
|
|
|
class UnitTest : public Test
|
|
{
|
|
public:
|
|
UnitTest();
|
|
virtual ~UnitTest();
|
|
virtual void Perform();
|
|
virtual void Print();
|
|
|
|
protected:
|
|
UnitTest(std::string name, std::string description);
|
|
virtual WebRtc_UWord32 CodecSpecific_SetBitrate(
|
|
WebRtc_UWord32 bitRate,
|
|
WebRtc_UWord32 /* frameRate */);
|
|
virtual void Setup();
|
|
virtual void Teardown();
|
|
virtual void RateControlTests();
|
|
virtual int Decode();
|
|
virtual int DecodeWithoutAssert();
|
|
virtual int SetCodecSpecificParameters() {return 0;};
|
|
|
|
virtual bool CheckIfBitExact(const void *ptrA, unsigned int aLengthBytes,
|
|
const void *ptrB, unsigned int bLengthBytes);
|
|
|
|
WebRtc_UWord32 WaitForEncodedFrame() const;
|
|
WebRtc_UWord32 WaitForDecodedFrame() const;
|
|
|
|
int _tests;
|
|
int _errors;
|
|
|
|
VideoSource* _source;
|
|
unsigned char* _refFrame;
|
|
unsigned char* _refEncFrame;
|
|
unsigned char* _refDecFrame;
|
|
int _refEncFrameLength;
|
|
FILE* _sourceFile;
|
|
|
|
UnitTestEncodeCompleteCallback* _encodeCompleteCallback;
|
|
UnitTestDecodeCompleteCallback* _decodeCompleteCallback;
|
|
enum { kMaxWaitEncTimeMs = 100 };
|
|
enum { kMaxWaitDecTimeMs = 25 };
|
|
};
|
|
|
|
class UnitTestEncodeCompleteCallback : public webrtc::EncodedImageCallback
|
|
{
|
|
public:
|
|
UnitTestEncodeCompleteCallback(TestVideoEncodedBuffer* buffer,
|
|
WebRtc_UWord32 decoderSpecificSize = 0,
|
|
void* decoderSpecificInfo = NULL) :
|
|
_encodedVideoBuffer(buffer),
|
|
_decoderSpecificSize(decoderSpecificSize),
|
|
_decoderSpecificInfo(decoderSpecificInfo),
|
|
_encodeComplete(false) {}
|
|
WebRtc_Word32 Encoded(webrtc::EncodedImage& encodedImage,
|
|
const void* codecSpecificInfo,
|
|
const webrtc::RTPFragmentationHeader*
|
|
fragmentation = NULL);
|
|
bool EncodeComplete();
|
|
// Note that this only makes sense if an encode has been completed
|
|
webrtc::VideoFrameType EncodedFrameType() const;
|
|
private:
|
|
TestVideoEncodedBuffer* _encodedVideoBuffer;
|
|
void* _decoderSpecificInfo;
|
|
WebRtc_UWord32 _decoderSpecificSize;
|
|
bool _encodeComplete;
|
|
webrtc::VideoFrameType _encodedFrameType;
|
|
};
|
|
|
|
class UnitTestDecodeCompleteCallback : public webrtc::DecodedImageCallback
|
|
{
|
|
public:
|
|
UnitTestDecodeCompleteCallback(TestVideoBuffer* buffer) :
|
|
_decodedVideoBuffer(buffer), _decodeComplete(false) {}
|
|
WebRtc_Word32 Decoded(webrtc::RawImage& image);
|
|
bool DecodeComplete();
|
|
private:
|
|
TestVideoBuffer* _decodedVideoBuffer;
|
|
bool _decodeComplete;
|
|
};
|
|
|
|
#endif // WEBRTC_MODULES_VIDEO_CODING_CODECS_TEST_FRAMEWORK_UNIT_TEST_H_
|
|
|