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 | |||