ref: 62139f09819f7e09a713fae90aa09c7c22db582f
dir: /test/api/DataGenerator.cpp/
#include <string.h>//memset/memcpy #include <time.h> #include "utils/DataGenerator.h" #include "utils/BufferedData.h" #include "utils/FileInputStream.h" using namespace std; bool YUVPixelDataGenerator( uint8_t* pPointer, int32_t iWidth, int32_t iHeight, int32_t iStride ) { #define SRC_FRAME_WIDTH (160) #define SRC_FRAME_HEIGHT (96) if ( SRC_FRAME_WIDTH-iWidth <= 0 || SRC_FRAME_HEIGHT-iHeight <= 0 ) { return false; } const int32_t kiFrameSize = SRC_FRAME_WIDTH*SRC_FRAME_HEIGHT; BufferedData sBuf; sBuf.SetLength(kiFrameSize); if (sBuf.Length() != kiFrameSize) { return false; } FileInputStream fileStream; if (!fileStream.Open("res/CiscoVT2people_160x96_6fps.yuv")) { return false; } if (fileStream.read(sBuf.data(), kiFrameSize) == kiFrameSize) { srand((uint32_t)time(NULL)); int32_t iStartPosX = rand()%(SRC_FRAME_WIDTH-iWidth); int32_t iStartPosY = rand()%(SRC_FRAME_HEIGHT-iHeight); uint8_t* pSrcPointer = sBuf.data() + iStartPosX + iStartPosY*SRC_FRAME_WIDTH; uint8_t* pLocalPointer = pPointer; for (int j=0;j<iHeight;j++) { memcpy(pLocalPointer, pSrcPointer, iWidth*sizeof(uint8_t)); pLocalPointer += iStride; pSrcPointer += SRC_FRAME_WIDTH; } return true; } return false; } void RandomPixelDataGenerator( uint8_t* pPointer, int32_t iWidth, int32_t iHeight, int32_t iStride ) { uint8_t* pLocalPointer = pPointer; for (int32_t j=0;j<iHeight;j++) { for (int32_t i=0;i<iWidth;i++) { pLocalPointer[i] = rand()%256; } pLocalPointer += iStride; } } void RandomResidueDataGenerator( uint16_t* pPointer, int32_t iWidth, int32_t iHeight, int32_t iStride ) { } void RandomCoeffDataGenerator( uint16_t* pPointer, int32_t iWidth, int32_t iHeight, int32_t iStride ) { }