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

 ---------------------------------------------------------------------------*/
/* Phission must be first for some reason */
#include <NetRepeaterTest.h>
#include <phission.h>
#include <signal.h>

/* ------------------------------------------------------------------------ */
int glbl_disable_displays = 0;
int glbl_signal_caught = 0;

/* ------------------------------------------------------------------------ */
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>\tsleep 'value' seconds and then close test\n");
    printf("\t\t\t--host|-h <host>\tthe host of the NetDisplay server\n");
    printf("\t\t\t--nports|-np <value>\t the number of source/dest ports\n");
    printf("\t\t\t--srcport|-sp <port>\tthe base port to connect on the host\n");
    printf("\t\t\t--dstport|-dp <port>\tthe base port to set the output server to\n");
                
    printf("\n\n");
    exit(1);
}

/* ----------------------------------------------------------------------- */
void signal_cleanup(int signal )
{
    phFUNCTION("signal_cleanup")

    glbl_signal_caught = 1;

    exit(1);
}

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

    unsigned int    i               = 0;

    int                 displaysOpen    = 1;
    
    /* Note: For the net repeater example, if you set this to something other
     * than the input format, it will convert it to the new format. Then,
     * the new format will be sent out onto the destination port */

    /* Zlib supports any format; the following is a sample */
    //uint32_t            outputFormat    = phImageYUV9 | phImageZlib;
    //uint32_t            outputFormat    = phImageGREY8 | phImageZlib;
    //uint32_t            outputFormat    = phImageRGB24 | phImageZlib;

    /* Valid JPEG supported format combinations: */
    //uint32_t            outputFormat    = phImageRGB24;
    //uint32_t            outputFormat    = phImageRGB24 | phImageJPEG;
    //uint32_t            outputFormat    = phImageGREY8 | phImageJPEG;
    //uint32_t            outputFormat    = phImageRGB24 | phImageJPEG | phImageZlib;
    //uint32_t            outputFormat    = phImageGREY8 | phImageJPEG | phImageZlib;
    
    /* Invalid format; Contains phImageRGBA32 which isn't supported by the     
     * JPEG code. */
    //uint32_t            outputFormat    = phImageRGBA32 | phImageJPEG;

    
    uint32_t            width           = 320;
    uint32_t            height          = 240;
    phDisplayInterface  **display       = NULL;

    phNetSource    **capture = NULL;
    uint32_t        nport = 1;
    uint32_t        srcport = 22345;
    uint32_t        dstport = 22350;
    char           *host = NULL;
    char           *path = NULL;

    /* Utility class to ease the starting and stopping of displays, captures and
     * pipelines */
    phSystem        net_system;

    /* 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("--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("--srcport", &srcport, phARG_UINT32);
    phCHECK_RC(rc, NULL, "arg_parser->add");
    rc = arg_parser->add("-sp", &srcport, phARG_UINT32);
    phCHECK_RC(rc, NULL, "arg_parser->add");
    
    rc = arg_parser->add("--dstport", &dstport, phARG_UINT32);
    phCHECK_RC(rc, NULL, "arg_parser->add");
    rc = arg_parser->add("-dp", &dstport, phARG_UINT32);
    phCHECK_RC(rc, NULL, "arg_parser->add");

    rc = arg_parser->parse(argc,argv);
    phCHECK_RC(rc,NULL,"arg_parser->parse");

    /* Set up a signal handler to shut the net_system down properly */
    for (i = 0; i < 17; i++ )
    {
        signal(i,signal_cleanup);
    }
//    signal(SIGINT,signal_cleanup);
   
    /* Set up the capture stuff */
    capture = new phNetSource *[nport];
    phCHECK_NULLPTR(capture,"new","new phNetSource *[nport:%d];",nport);
 
    /* Set up the display stuff */
    display = new phDisplayInterface *[nport];
    phCHECK_NULLPTR(display,"new","new phDisplayInterface *[nport:%d];",nport);
    
    for (i = 0; i < nport; i++)
    {
        /* CAPTURE/SOURCE */
        capture[i] = new phNetSource();

        rc = phNetSource::makePath(host,srcport+i,phNetTypeTCP,&path);
        phCHECK_RC(rc,NULL,"phNetSource::makePath");
    
        rc = capture[i]->setPath(path);
        phCHECK_RC(rc,NULL,"capture[i:%d]->setPath(%s)",i,path);

        rc = net_system.addCapture(capture[i]);
        phPRINT_RC(rc,NULL,"net_system.addCapture(capture[i:%d])",i);

        /* DISPLAY */
        NetDisplay *n = new NetDisplay(width,height);
    
        display[i] = (phDisplayInterface *)n;
        display[i]->setLiveSourceInput(capture[i]->getLiveSourceOutput());
        
        phPROGRESS("Setting display[%d] to listen on port %d\n",
                 i, dstport + i);
        n->setPort(dstport + i);
        //n->setFormat(outputFormat);

        rc = net_system.addDisplay(display[i]);
        phPRINT_RC(rc,NULL,"net_system.addDisplay(display[i:%u])",i);
    }

    /* startup the net_system */
    rc = net_system.startup();
    phPRINT_RC(rc,NULL,"net_system.startup()");
   
    displaysOpen = 1;
    while ((displaysOpen) && (glbl_signal_caught == 0))
    {
        /* Sleep a while, don't loop tightly */

        phMSleep(100);
        
        /* Yielding is optional. This gives up the thread's timeslice
         * to prevent slow response in other threads. It consumes more
         * CPU cycles than sleeping. Use it instead of sleeping if
         * this loop is processing anything */
        
        /* phYield(); */

        if (nport > 0)
        {
            displaysOpen = 0;
            for (i = 0; (i < nport) && (displaysOpen == 0); i++)
            {
                if (display[i]->isOpen() == 1)
                {
                    displaysOpen = 1;
                }
            }
        }
        
        /* 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)
        { 
            displaysOpen = 0;
            phSleep(test); /* test's value should be a time (in secs) value > 0*/
        }
    }

    glbl_signal_caught = 0;
    
    rc = net_system.shutdown();
    phPRINT_RC(rc,NULL,"net_system.shutdown()");

error:
    /* Delete individual captures */
    for (i = 0; (i < nport) && (capture != NULL); i++)
    {
        phDelete(capture[i]);
    }
 
    /* Delete the allocated array */
    phDeleteArray(capture);

    /* Delete individual displays */
    for (i = 0; (i < nport) && (display != NULL); i++)
    {
        phDelete(display[i]);
    }
 
    /* Delete the allocated array */
    phDeleteArray(display);

    phDelete(arg_parser);

    return phSUCCESS;
}

Copyright (C) 2002 - 2007

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

---