/* * 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. */ // testAPI.cpp : Defines the entry point for the console application. // #include #if defined(_WIN32) #include #include #include #include #include #include #include #include #elif defined(WEBRTC_LINUX) #include #include #include #include #elif defined(WEBRTC_MAC_INTEL) #import #import #import #import #include #include using namespace std; @class CocoaRenderView; #endif #include "common_types.h" #include "process_thread.h" #include "module_common_types.h" #include "video_render_defines.h" #include "video_render.h" #include "tick_util.h" #include "trace.h" using namespace webrtc; void GetTestVideoFrame(WebRtc_UWord8* frame, WebRtc_Word32 width, WebRtc_Word32 height, WebRtc_UWord8 startColor); int TestSingleStream(VideoRender* renderModule); int TestFullscreenStream(VideoRender* &renderModule, void* window, const VideoRenderType videoRenderType); int TestBitmapText(VideoRender* renderModule); int TestMultipleStreams(VideoRender* renderModule); int TestExternalRender(VideoRender* renderModule); #define TEST_FRAME_RATE 30 #define TEST_TIME_SECOND 5 #define TEST_FRAME_NUM (TEST_FRAME_RATE*TEST_TIME_SECOND) #define TEST_STREAM0_START_COLOR 0 #define TEST_STREAM1_START_COLOR 64 #define TEST_STREAM2_START_COLOR 128 #define TEST_STREAM3_START_COLOR 192 #if defined(_WIN32) && defined(_DEBUG) // #include "vld.h" #define SLEEP(x) ::Sleep(x) #elif defined(WEBRTC_LINUX) #define GET_TIME_IN_MS timeGetTime() #define SLEEP(x) Sleep(x) void Sleep(unsigned long x) { timespec t; t.tv_sec = x/1000; t.tv_nsec = (x-(x/1000)*1000)*1000000; nanosleep(&t,NULL); } unsigned long timeGetTime() { struct timeval tv; struct timezone tz; unsigned long val; gettimeofday(&tv, &tz); val= tv.tv_sec*1000+ tv.tv_usec/1000; return(val); } #elif defined(WEBRTC_MAC_INTEL) #define GET_TIME_IN_MS timeGetTime() #define SLEEP(x) usleep(x * 1000) unsigned long timeGetTime() { return 0; } #else #define GET_TIME_IN_MS ::timeGetTime() #define SLEEP(x) ::Sleep(x) #endif using namespace std; #if defined(_WIN32) LRESULT CALLBACK WebRtcWinProc( HWND hWnd,UINT uMsg,WPARAM wParam,LPARAM lParam) { switch(uMsg) { case WM_DESTROY: break; case WM_COMMAND: break; } return DefWindowProc(hWnd,uMsg,wParam,lParam); } int WebRtcCreateWindow(HWND &hwndMain,int winNum, int width, int height) { HINSTANCE hinst = GetModuleHandle(0); WNDCLASSEX wcx; wcx.hInstance = hinst; wcx.lpszClassName = TEXT("VideoRenderTest"); wcx.lpfnWndProc = (WNDPROC)WebRtcWinProc; wcx.style = CS_DBLCLKS; wcx.hIcon = LoadIcon (NULL, IDI_APPLICATION); wcx.hIconSm = LoadIcon (NULL, IDI_APPLICATION); wcx.hCursor = LoadCursor (NULL, IDC_ARROW); wcx.lpszMenuName = NULL; wcx.cbSize = sizeof (WNDCLASSEX); wcx.cbClsExtra = 0; wcx.cbWndExtra = 0; wcx.hbrBackground = GetSysColorBrush(COLOR_3DFACE); // Register our window class with the operating system. // If there is an error, exit program. if ( !RegisterClassEx (&wcx) ) { MessageBox( 0, TEXT("Failed to register window class!"),TEXT("Error!"), MB_OK|MB_ICONERROR ); return 0; } // Create the main window. hwndMain = CreateWindowEx( 0, // no extended styles TEXT("VideoRenderTest"), // class name TEXT("VideoRenderTest Window"), // window name WS_OVERLAPPED |WS_THICKFRAME, // overlapped window 800, // horizontal position 0, // vertical position width, // width height, // height (HWND) NULL, // no parent or owner window (HMENU) NULL, // class menu used hinst, // instance handle NULL); // no window creation data if (!hwndMain) { int error = GetLastError(); return -1; } // Show the window using the flag specified by the program // that started the application, and send the application // a WM_PAINT message. ShowWindow(hwndMain, SW_SHOWDEFAULT); UpdateWindow(hwndMain); return 0; } #elif defined(WEBRTC_LINUX) int WebRtcCreateWindow(Window *outWindow, Display **outDisplay, int winNum, int width, int height) // unsigned char* title, int titleLength) { int screen, xpos = 10, ypos = 10; XEvent evnt; XSetWindowAttributes xswa; // window attribute struct XVisualInfo vinfo; // screen visual info struct unsigned long mask; // attribute mask // get connection handle to xserver Display* _display = XOpenDisplay( NULL ); // get screen number screen = DefaultScreen(_display); // put desired visual info for the screen in vinfo if( XMatchVisualInfo(_display, screen, 24, TrueColor, &vinfo) != 0 ) { //printf( "Screen visual info match!\n" ); } // set window attributes xswa.colormap = XCreateColormap(_display, DefaultRootWindow(_display), vinfo.visual, AllocNone); xswa.event_mask = StructureNotifyMask | ExposureMask; xswa.background_pixel = 0; xswa.border_pixel = 0; // value mask for attributes mask = CWBackPixel | CWBorderPixel | CWColormap | CWEventMask; switch( winNum ) { case 0: xpos = 200; ypos = 200; break; case 1: xpos = 300; ypos = 200; break; default: break; } // create a subwindow for parent (defroot) Window _window = XCreateWindow(_display, DefaultRootWindow(_display), xpos, ypos, width, height, 0, vinfo.depth, InputOutput, vinfo.visual, mask, &xswa); // Set window name if( winNum == 0 ) { XStoreName(_display, _window, "VE MM Local Window"); XSetIconName(_display, _window, "VE MM Local Window"); } else if( winNum == 1 ) { XStoreName(_display, _window, "VE MM Remote Window"); XSetIconName(_display, _window, "VE MM Remote Window"); } // make x report events for mask XSelectInput(_display, _window, StructureNotifyMask); // map the window to the display XMapWindow(_display, _window); // wait for map event do { XNextEvent(_display, &evnt); } while (evnt.type != MapNotify || evnt.xmap.event != _window); *outWindow = _window; *outDisplay = _display; return 0; } #elif defined(WEBRTC_MAC_INTEL) // LINUX int WebRtcCreateWindow(CocoaRenderView*& cocoaRenderer, int winNum, int width, int height) // unsigned char* title, int titleLength) { // In Cocoa, rendering is not done directly to a window like in Windows and Linux. // It is rendererd to a Subclass of NSOpenGLView // create cocoa container window NSRect outWindowFrame = NSMakeRect(200, 800, width + 20, height + 20); NSWindow* outWindow = [[NSWindow alloc] initWithContentRect:outWindowFrame styleMask:NSTitledWindowMask backing:NSBackingStoreBuffered defer:NO]; [outWindow orderOut:nil]; [outWindow setTitle:@"Cocoa Renderer"]; [outWindow setBackgroundColor:[NSColor blueColor]]; // create renderer and attach to window NSRect cocoaRendererFrame = NSMakeRect(10, 10, width, height); cocoaRenderer = [[CocoaRenderView alloc] initWithFrame:cocoaRendererFrame]; [[outWindow contentView] addSubview:cocoaRenderer]; [outWindow makeKeyAndOrderFront:NSApp]; return 0; } #endif class MyRenderCallback: public VideoRenderCallback { public: MyRenderCallback() : _cnt(0) { } ; ~MyRenderCallback() { } ; virtual WebRtc_Word32 RenderFrame(const WebRtc_UWord32 streamId, VideoFrame& videoFrame) { _cnt++; if (_cnt % 100 == 0) { printf("Render callback %d \n",_cnt); } return 0; } WebRtc_Word32 _cnt; }; void GetTestVideoFrame(WebRtc_UWord8* frame, WebRtc_Word32 width, WebRtc_Word32 height, WebRtc_UWord8 startColor) { // changing color static WebRtc_UWord8 color = startColor; WebRtc_UWord8* destY = frame; WebRtc_UWord8* destU = &frame[width*height]; WebRtc_UWord8* destV = &frame[width*height*5/4]; //Y for (WebRtc_Word32 y=0; y<(width*height); y++) { destY[y] = color; } //U for (WebRtc_Word32 u=0; u<(width*height/4); u++) { destU[u] = color; } //V for (WebRtc_Word32 v=0; v<(width*height/4); v++) { destV[v] = color; } color++; } int TestSingleStream(VideoRender* renderModule) { int error = 0; // Add settings for a stream to render printf("Add stream 0 to entire window\n"); const int streamId0 = 0; VideoRenderCallback* renderCallback0 = renderModule->AddIncomingRenderStream(streamId0, 0, 0.0f, 0.0f, 1.0f, 1.0f); assert(renderCallback0 != NULL); #if defined (WEBRTC_VIDEO_EXTERNAL_CAPTURE_AND_RENDER) MyRenderCallback externalRender; renderModule->AddExternalRenderCallback(streamId0, &externalRender); #endif printf("Start render\n"); error = renderModule->StartRender(streamId0); assert(error == 0); // Loop through an I420 file and render each frame const WebRtc_UWord32 width = 352; const WebRtc_UWord32 height = 288; const WebRtc_UWord32 numBytes = (WebRtc_UWord32)(1.5 * width * height); VideoFrame videoFrame0; videoFrame0.VerifyAndAllocate(numBytes); const WebRtc_UWord32 renderDelayMs = 500; for (int i=0; iRenderFrame(streamId0, videoFrame0); SLEEP(1000/TEST_FRAME_RATE); } videoFrame0.Free(); // Shut down printf("Closing...\n"); error = renderModule->StopRender(streamId0); assert(error == 0); error = renderModule->DeleteIncomingRenderStream(streamId0); assert(error == 0); return 0; } int TestFullscreenStream(VideoRender* &renderModule, void* window, const VideoRenderType videoRenderType) { int error = 0; VideoRender::DestroyVideoRender(renderModule); renderModule = VideoRender::CreateVideoRender(12345, window, true, videoRenderType); TestSingleStream(renderModule); VideoRender::DestroyVideoRender(renderModule); renderModule = VideoRender::CreateVideoRender(12345, window, false, videoRenderType); return 0; } int TestBitmapText(VideoRender* renderModule) { #if defined(WIN32) int error = 0; // Add settings for a stream to render printf("Add stream 0 to entire window\n"); const int streamId0 = 0; VideoRenderCallback* renderCallback0 = renderModule->AddIncomingRenderStream(streamId0, 0, 0.0f, 0.0f, 1.0f, 1.0f); assert(renderCallback0 != NULL); printf("Adding Bitmap\n"); DDCOLORKEY ColorKey; // black ColorKey.dwColorSpaceHighValue = RGB(0, 0, 0); ColorKey.dwColorSpaceLowValue = RGB(0, 0, 0); HBITMAP hbm = (HBITMAP)LoadImage(NULL, (LPCTSTR)_T("renderStartImage.bmp"), IMAGE_BITMAP, 0, 0, LR_LOADFROMFILE); renderModule->SetBitmap(hbm, 0, &ColorKey, 0.0f, 0.0f, 0.3f, 0.3f); printf("Adding Text\n"); renderModule->SetText(1, (WebRtc_UWord8*) "WebRtc Render Demo App", 20, RGB(255, 0, 0), RGB(0, 0, 0), 0.25f, 0.1f, 1.0f, 1.0f); printf("Start render\n"); error = renderModule->StartRender(streamId0); assert(error == 0); // Loop through an I420 file and render each frame const WebRtc_UWord32 width = 352; const WebRtc_UWord32 height = 288; const WebRtc_UWord32 numBytes = (WebRtc_UWord32)(1.5 * width * height); VideoFrame videoFrame0; videoFrame0.VerifyAndAllocate(numBytes); const WebRtc_UWord32 renderDelayMs = 500; for (int i=0; iRenderFrame(streamId0, videoFrame0); SLEEP(1000/TEST_FRAME_RATE); } videoFrame0.Free(); // Sleep and let all frames be rendered before closing SLEEP(renderDelayMs*2); // Shut down printf("Closing...\n"); ColorKey.dwColorSpaceHighValue = RGB(0,0,0); ColorKey.dwColorSpaceLowValue = RGB(0,0,0); renderModule->SetBitmap(NULL, 0, &ColorKey, 0.0f, 0.0f, 0.0f, 0.0f); renderModule->SetText(1, NULL, 20, RGB(255,255,255), RGB(0,0,0), 0.0f, 0.0f, 0.0f, 0.0f); error = renderModule->StopRender(streamId0); assert(error == 0); error = renderModule->DeleteIncomingRenderStream(streamId0); assert(error == 0); #endif return 0; } int TestMultipleStreams(VideoRender* renderModule) { int error = 0; // Add settings for a stream to render printf("Add stream 0\n"); const int streamId0 = 0; VideoRenderCallback* renderCallback0 = renderModule->AddIncomingRenderStream(streamId0, 0, 0.0f, 0.0f, 0.45f, 0.45f); assert(renderCallback0 != NULL); printf("Add stream 1\n"); const int streamId1 = 1; VideoRenderCallback* renderCallback1 = renderModule->AddIncomingRenderStream(streamId1, 0, 0.55f, 0.0f, 1.0f, 0.45f); assert(renderCallback1 != NULL); printf("Add stream 2\n"); const int streamId2 = 2; VideoRenderCallback* renderCallback2 = renderModule->AddIncomingRenderStream(streamId2, 0, 0.0f, 0.55f, 0.45f, 1.0f); assert(renderCallback2 != NULL); printf("Add stream 3\n"); const int streamId3 = 3; VideoRenderCallback* renderCallback3 = renderModule->AddIncomingRenderStream(streamId3, 0, 0.55f, 0.55f, 1.0f, 1.0f); assert(renderCallback3 != NULL); assert(renderModule->StartRender(streamId0) == 0); assert(renderModule->StartRender(streamId1) == 0); assert(renderModule->StartRender(streamId2) == 0); assert(renderModule->StartRender(streamId3) == 0); // Loop through an I420 file and render each frame const WebRtc_UWord32 width = 352; const WebRtc_UWord32 height = 288; const WebRtc_UWord32 numBytes = (WebRtc_UWord32)(1.5 * width * height); VideoFrame videoFrame0; videoFrame0.VerifyAndAllocate(numBytes); VideoFrame videoFrame1; videoFrame1.VerifyAndAllocate(numBytes); VideoFrame videoFrame2; videoFrame2.VerifyAndAllocate(numBytes); VideoFrame videoFrame3; videoFrame3.VerifyAndAllocate(numBytes); const WebRtc_UWord32 renderDelayMs = 500; for (int i=0; iRenderFrame(streamId0, videoFrame0); GetTestVideoFrame(videoFrame1.Buffer(), width, height, TEST_STREAM1_START_COLOR); videoFrame1.SetRenderTime(TickTime::MillisecondTimestamp() + renderDelayMs); // Render this frame with the specified delay videoFrame1.SetWidth(width); videoFrame1.SetHeight(height); videoFrame1.SetLength(numBytes); renderCallback1->RenderFrame(streamId1, videoFrame1); GetTestVideoFrame(videoFrame2.Buffer(), width, height, TEST_STREAM2_START_COLOR); videoFrame2.SetRenderTime(TickTime::MillisecondTimestamp() + renderDelayMs); // Render this frame with the specified delay videoFrame2.SetWidth(width); videoFrame2.SetHeight(height); videoFrame2.SetLength(numBytes); renderCallback2->RenderFrame(streamId2, videoFrame2); GetTestVideoFrame(videoFrame3.Buffer(), width, height, TEST_STREAM3_START_COLOR); videoFrame3.SetRenderTime(TickTime::MillisecondTimestamp() + renderDelayMs); // Render this frame with the specified delay videoFrame3.SetWidth(width); videoFrame3.SetHeight(height); videoFrame3.SetLength(numBytes); renderCallback3->RenderFrame(streamId3, videoFrame3); SLEEP(1000/TEST_FRAME_RATE); } videoFrame0.Free(); videoFrame1.Free(); videoFrame2.Free(); videoFrame3.Free(); // Shut down printf("Closing...\n"); assert(renderModule->StopRender(streamId0) == 0); assert(renderModule->DeleteIncomingRenderStream(streamId0) == 0); assert(renderModule->StopRender(streamId1) == 0); assert(renderModule->DeleteIncomingRenderStream(streamId1) == 0); assert(renderModule->StopRender(streamId2) == 0); assert(renderModule->DeleteIncomingRenderStream(streamId2) == 0); assert(renderModule->StopRender(streamId3) == 0); assert(renderModule->DeleteIncomingRenderStream(streamId3) == 0); return 0; } int TestExternalRender(VideoRender* renderModule) { MyRenderCallback *externalRender = new MyRenderCallback(); const int streamId0 = 0; VideoRenderCallback* renderCallback0 = renderModule->AddIncomingRenderStream(streamId0, 0, 0.0f, 0.0f, 1.0f, 1.0f); assert(renderCallback0 != NULL); assert(renderModule->AddExternalRenderCallback(streamId0, externalRender) == 0); assert(renderModule->StartRender(streamId0) == 0); const WebRtc_UWord32 width = 352; const WebRtc_UWord32 height = 288; const WebRtc_UWord32 numBytes = (WebRtc_UWord32) (1.5 * width * height); VideoFrame videoFrame0; videoFrame0.VerifyAndAllocate(numBytes); const WebRtc_UWord32 renderDelayMs = 500; int frameCount = TEST_FRAME_NUM; for (int i=0; iRenderFrame(streamId0, videoFrame0); SLEEP(33); } // Sleep and let all frames be rendered before closing SLEEP(2*renderDelayMs); videoFrame0.Free(); assert(renderModule->StopRender(streamId0) == 0); assert(renderModule->DeleteIncomingRenderStream(streamId0) == 0); assert(frameCount == externalRender->_cnt); delete externalRender; externalRender = NULL; return 0; } #if defined(_WIN32) int _tmain(int argc, _TCHAR* argv[]) #elif defined(WEBRTC_LINUX) int main(int argc, char* argv[]) #elif defined(WEBRTC_MAC_INTEL) int main (int argc, const char * argv[]) //int begin() #endif { #ifdef WEBRTC_MAC_INTEL NSAutoreleasePool * pool = [[NSAutoreleasePool alloc] init]; [NSApplication sharedApplication]; #endif int myId = 12345; // Create a window for testing void* window = NULL; #if defined (_WIN32) HWND testHwnd; WebRtcCreateWindow(testHwnd, 0, 352, 288); window = (void*)testHwnd; VideoRenderType windowType = kRenderWindows; #elif defined(WEBRTC_LINUX) Window testWindow; Display* display; WebRtcCreateWindow(&testWindow, &display, 0, 352, 288); VideoRenderType windowType = kRenderX11; window = (void*)testWindow; #elif defined(WEBRTC_MAC_INTEL) CocoaRenderView* testWindow; WebRtcCreateWindow(testWindow, 0, 352, 288); VideoRenderType windowType = kRenderCocoa; window = (void*)testWindow; #endif #if defined (WEBRTC_VIDEO_EXTERNAL_CAPTURE_AND_RENDER) windowType = kRenderExternal; #endif // Create the render module printf("Create render module\n"); VideoRender* renderModule = NULL; renderModule = VideoRender::CreateVideoRender(myId, window, false, windowType); assert(renderModule != NULL); // ##### Test single stream rendering #### printf("#### TestSingleStream ####\n"); if (TestSingleStream(renderModule) != 0) { printf ("TestSingleStream failed\n"); } // ##### Test fullscreen rendering #### printf("#### TestFullscreenStream ####\n"); if (TestFullscreenStream(renderModule, window, windowType) != 0) { printf ("TestFullscreenStream failed\n"); } // ##### Test bitmap and text #### printf("#### TestBitmapText ####\n"); if (TestBitmapText(renderModule) != 0) { printf ("TestBitmapText failed\n"); } // ##### Test multiple streams #### printf("#### TestMultipleStreams ####\n"); if (TestMultipleStreams(renderModule) != 0) { printf ("TestMultipleStreams failed\n"); } // ##### Test multiple streams #### printf("#### TestExternalRender ####\n"); if (TestExternalRender(renderModule) != 0) { printf ("TestExternalRender failed\n"); } delete renderModule; renderModule = NULL; printf("VideoRender unit tests passed.\n"); #if defined(WEBRTC_MAC_INTEL) [pool release]; #endif return 0; }