phNetSourceTest.cpp

Go to the documentation of this file.

/* ---------------------------------------------------------------------------
    Phission : 
        Realtime Vision Processing System
    
    Copyright (C) 2003 Philip D.S. Thoren (pthoren@cs.uml.edu)
    University of Massachusetts at Lowell,
    Laboratory for Artificial Intelligence and Robotics
    
    This file is part of Phission.

 ---------------------------------------------------------------------------*/
#include <phNetSourceTest.h>
#include <stdio.h>
#include <phission.h>

/* ------------------------------------------------------------------------ */
int glbl_disable_displays = 0;
int glbl_preload_images = 1;

/* ------------------------------------------------------------------------ */
int usage()
{
    printf("\n\n\tUsage:\n");
    printf("\t\t\t--help\t\t\tdisplay usage\n");
    printf("\t\t\t--nodisplay\tdisable the allocation, opening or any use of a display.\n");
    printf("\t\t\t--test <value>\tload 'value' number of frames/files/images\n");
    printf("\t\t\t--host|-h <host>\tthe host of the NetDisplay server\n");
    printf("\t\t\t--port|-p <port>\tthe port to connect on the host\n");
    printf("\n\n");
    exit(1);
}


/* ------------------------------------------------------------------------ */
int main(int argc, char *argv[] )
{
    phFUNCTION("main")

    int             displaysOpen= 1;

    uintmax_t       count       = 0;
    unsigned int    i           = 0;

    uint32_t        port = 0;
    char            *host = NULL;
    char            *path = NULL;

    phDisplayInterface *display = NULL;

    phImage temp;
    char    temp_filename[245];
    
    /* Utility class to ease the starting and stopping of displays, captures and
     * pipelines */
    phSystem        system;
    phNetSource     *capture = new phNetSource();
   
    /* Remove the code below when using this code as an example.
     * 
     * This just checks whether "--test" has been specified with
     * a time value argument. It's for testing all the examples
     * without the need for human intervention. */
    int             test = 0;

    phArgTable      *arg_parser = new phArgTable();

    /* Setup and parse all the arguments */
    rc = arg_parser->add("--test", &test, phARG_INT);
    phCHECK_RC(rc,NULL,"arg_parser->add");
    rc = arg_parser->add("--nodisplay", (void *)&glbl_disable_displays, phARG_BOOL);
    phCHECK_RC(rc,NULL,"arg_parser->add");
    rc = arg_parser->add("--help",(void *)&usage, phARG_FUNC);
    phCHECK_RC(rc,NULL,"arg_parser->add");
   
    /* Get the host and port from the command line */
    rc = arg_parser->add("--host", &host, phARG_CHAR);
    phCHECK_RC(rc, NULL, "arg_parser->add");
    rc = arg_parser->add("-h", &host, phARG_CHAR);
    phCHECK_RC(rc, NULL, "arg_parser->add");
    rc = arg_parser->add("--port", &port, phARG_UINT32);
    phCHECK_RC(rc, NULL, "arg_parser->add");
    rc = arg_parser->add("-p", &port, phARG_UINT32);
    phCHECK_RC(rc, NULL, "arg_parser->add");
    
    rc = arg_parser->parse(argc,argv);
    phCHECK_RC(rc,NULL,"arg_parser->parse");

    capture->setSize(320,240);

    //capture->setFormat(phImageRGB24);
    
#if 0
    #if 1
    rc = capture->setHost(host);
    phCHECK_RC(rc,NULL,"capture->setHost(host:%s);",host);
        
    rc = capture->setPort(port);
    phCHECK_RC(rc,NULL,"capture->setPort(port:%d);",port);
    #else
    rc = capture->set(host,port);
    phCHECK_RC(rc,NULL,"capture->set(host:%s,port:%d);",host,port);
    #endif
#else
    rc = phNetSource::makePath(host,port,phNetTypeTCP,&path);
    phCHECK_RC(rc,NULL,"phNetSource::makePath");
    
    rc = capture->setPath(path);
    phCHECK_RC(rc,NULL,"capture->setPath(%s)",path);
#endif
    
    rc = system.add(capture);
    phCHECK_RC(rc,NULL,"system.add(capture:%p)",capture);
    
    if (glbl_disable_displays == 0)
    {
        /* this will set up with some value, but not necessarily the correct
         * value for the image size. When the capture is activated, it will
         * output images of some size and the window will resize on it's own.
         * However, one can call setSize on the capture object before here
         * for a desired size, which will not change when the new image comes
         * across */
#if defined(WIN32)
        display = new GDIDisplay(capture->getWidth(),
                                 capture->getHeight(),
                                 "phNetSourceTest");
#else
        display = new X11Display(capture->getWidth(),
                                 capture->getHeight(),
                                 "phNetSourceTest");
#endif
        phCHECK_PTR(display,"new","new X11Display failed.");
        
        display->setLiveSourceInput(capture->getLiveSourceOutput());
        
        rc = system.addDisplay(display);
    }

    temp.connect(capture->getLiveSourceOutput());

    phPROGRESS("%s:%d\n",capture->getHost(),capture->getPort());

    /* startup */
    rc = system.startup();
    phPRINT_RC(rc,NULL,"system.startup()");
    
    // now loop continuously for as long as we have a display open and a frame available from the avi file
    while ((displaysOpen) && (system.capturesActive()))
    {
        if (glbl_disable_displays == 0)
        {
            displaysOpen = 0;
            if (system.displaysOpen())
            {
                displaysOpen = 1;
            }
        }

        temp.update();
        phPROGRESS("%u\n",count);
        count++;

        /* Remove this if block when using this code as an example */
        /* Set the loop control value to end the loop when testing */
        if (test > 0)
        { 
            test--;
            if (test == 0)
                displaysOpen = 0;
        }

        phMSleep(100);
    }
    
error:
    rc = system.shutdown();
    phPRINT_RC(rc,NULL,"system.shutdown()");
    
    phDelete(display);

    // free up the filename string allocation
    phFree(host);
    phFree(path);

    phDelete(arg_parser);

    return phSUCCESS;
}

Copyright (C) 2002 - 2007

Philip D.S. Thoren ( pthoren@users.sourceforge.net ) University Of Massachusetts at Lowell Robotics Lab

---