Showing the changes to test/api/DataGenerator.cpp (1 of 23 files). Show all
Unified · Split
| @@ -0,0 +1,66 @@ | ||
| 1 | +#include <string.h>//memset/memcpy | |
| 2 | +#include <time.h> | |
| 3 | +#include "utils/DataGenerator.h" | |
| 4 | +#include "utils/BufferedData.h" | |
| 5 | +#include "utils/FileInputStream.h" | |
| 6 | + | |
| 7 | +using namespace std; | |
| 8 | + | |
| 9 | + | |
| 10 | +bool YUVPixelDataGenerator( uint8_t* pPointer, int32_t iWidth, int32_t iHeight, int32_t iStride ) | |
| 11 | +{ | |
| 12 | +#define SRC_FRAME_WIDTH (160) | |
| 13 | +#define SRC_FRAME_HEIGHT (96) | |
| 14 | + | |
| 15 | + if ( SRC_FRAME_WIDTH-iWidth <= 0 || SRC_FRAME_HEIGHT-iHeight <= 0 ) { | |
| 16 | + return false; | |
| 17 | + } | |
| 18 | + | |
| 19 | + const int32_t kiFrameSize = SRC_FRAME_WIDTH*SRC_FRAME_HEIGHT; | |
| 20 | + BufferedData sBuf; | |
| 21 | + sBuf.SetLength(kiFrameSize); | |
| 22 | + if (sBuf.Length() != kiFrameSize) { | |
| 23 | + return false; | |
| 24 | + } | |
| 25 | + | |
| 26 | + FileInputStream fileStream; | |
| 27 | + if (!fileStream.Open("res/CiscoVT2people_160x96_6fps.yuv")) { | |
| 28 | + return false; | |
| 29 | + } | |
| 30 | + if (fileStream.read(sBuf.data(), kiFrameSize) == kiFrameSize) { | |
| 31 | + srand((uint32_t)time(NULL)); | |
| 32 | + int32_t iStartPosX = rand()%(SRC_FRAME_WIDTH-iWidth); | |
| 33 | + int32_t iStartPosY = rand()%(SRC_FRAME_HEIGHT-iHeight); | |
| 34 | + uint8_t* pSrcPointer = sBuf.data() + iStartPosX + iStartPosY*SRC_FRAME_WIDTH; | |
| 35 | + uint8_t* pLocalPointer = pPointer; | |
| 36 | + | |
| 37 | + for (int j=0;j<iHeight;j++) { | |
| 38 | + memcpy(pLocalPointer, pSrcPointer, iWidth*sizeof(uint8_t)); | |
| 39 | + pLocalPointer += iStride; | |
| 40 | + pSrcPointer += SRC_FRAME_WIDTH; | |
| 41 | + } | |
| 42 | + return true; | |
| 43 | + } | |
| 44 | + return false; | |
| 45 | +} | |
| 46 | + | |
| 47 | +void RandomPixelDataGenerator( uint8_t* pPointer, int32_t iWidth, int32_t iHeight, int32_t iStride ) | |
| 48 | +{ | |
| 49 | + uint8_t* pLocalPointer = pPointer; | |
| 50 | + for (int32_t j=0;j<iHeight;j++) { | |
| 51 | + for (int32_t i=0;i<iWidth;i++) { | |
| 52 | + pLocalPointer[i] = rand()%256; | |
| 53 | + } | |
| 54 | + pLocalPointer += iStride; | |
| 55 | + } | |
| 56 | +} | |
| 57 | + | |
| 58 | + | |
| 59 | +void RandomResidueDataGenerator( uint16_t* pPointer, int32_t iWidth, int32_t iHeight, int32_t iStride ) | |
| 60 | +{ | |
| 61 | +} | |
| 62 | + | |
| 63 | +void RandomCoeffDataGenerator( uint16_t* pPointer, int32_t iWidth, int32_t iHeight, int32_t iStride ) | |
| 64 | +{ | |
| 65 | +} | |
| 66 | + | |