NaviTrack Tutorial:Integrating:Passing image data

From NAMIC Wiki
Jump to: navigation, search
Home < NaviTrack Tutorial:Integrating:Passing image data

Overview for passing image data

NaviTrack have a capability to send any C++ data types through NaviTrack data stream. That means, once you define a C++ class, you can transfer any kind of data even through a network, in quite same way as you did for coordinate data in previous tutorial pages.

For an image transfer, NaviTrack has already had a "Image" class to pass image data. This class has a few member variables including image size and image itself, as well as memberfunctions to serialize image data. Serialized image data is pushed into NaviTrack data flow as an event, and transfered to other nodes.

Image class is defined in include/OpenTracker/types/Image.h and src/input/types/Image.cxx.

Sending side

Modify MyTutorialModule

To send data through NaviTrack, a new event is created. The following code shows a basic idea of implementation:

 void MyTutorialModule::setImage(Image& img, int w, int h)
 {  
   if(source!=NULL)
     {
       Event event;
       event.setAttribute("image", img);
       event.setAttribute("xsize", w);
       event.setAttribute("ysize", h);
       event.timeStamp();
       source->updateObservers( event );
     }
 }


Due to NaviTrack's bug, however, above code might not work well for lage image in Windows environment (see the last section in this page). To avoid this problem, we need to split a image into several fragments as following code:

 void MyTutorialModule::setImage(Image& img, int w, int h)
 {
   //static short* pix = NULL;
   
   if(source!=NULL)
     {
       // split a image
       int nlines  = MAXIMUM_IMAGE_SIZE / (w*sizeof(short));
       int nblocks = h / nlines;
       if (h % nlines > 0) nblocks ++;
 
       for (int n = 0; n < nblocks; n ++) 
         {
           int ysize;
           int endflag = 0;
 
           if (n == nblocks-1)
             {
               ysize = h-nlines*n;
               endflag = 1;
             }
           else
             ysize = nlines;
 
           short* ba = (short*)img.image_ptr;
           Image fragment(w,ysize,sizeof(short),&ba[w*n*nlines]);
             
 
           //ot::Event event;
           ot::Event event;
           event.setAttribute("endflag",   endflag);
           event.setAttribute("xsize",     w);
           event.setAttribute("ysize",     ysize);
           event.setAttribute("totalysize",h);
           event.setAttribute("line",      n*nlines);
           event.setAttribute("image",     fragment);
           event.setAttribute("index",     n);
 
           event.timeStamp();
           source->updateObservers( event );
           this->context->loopOnce();
           //usleep(pushImageInterval);
           OSUtils::sleep(pushImageInterval);
         }
     }
 }

pushimage.cxx

The core part of image sending would be like following code:

 MyTutorialModule* mtm = (MyTutorialModule*) context.getModule("MyTutorialConfig");
 
 int stopflag = 0;
 int i = 0;
 while (stopflag == 0) {
   
    // load image data to image_data here.
     ....
   
   Image img(width, heigt, sizeof(short), (char*)image_data);
   
   if (mtm) {
     mtm->setImage(img, w, h);
   }
   
   stopflag = context.loopOnce();
   
   usleep(rate);
 }

Receiving side

Modify MyTutorialSink and MyTutorialModule

MyTutorialSink.h: Add member variables to hold the coordinate data:

 private:
   Image image;
   int width;
   int height;

MyTutorialSink.cxx (basic idea):

 void MyTutorialSink::onEventGenerated( Event& event, Node& generator)
 {
   if (event.hasAttribute("image") &&
       event.hasAttribute("xsize") &&
       event.hasAttribute("ysize"))
     {
       image = event.getAttribute((Image*)NULL, "image");
       width  = event.getAttribute(std::string("xsize"),0);
       height = event.getAttribute(std::string("ysize"),0);
     }
   else
     {
       width  = 0;
       height = 0;
     }
 }


MyTutorialSink.cxx (implemented for this tutorial):

 void MyTutorialSink::onEventGenerated( Event& event, Node& generator)
 {
   //std::cerr << "MyTutorialSink::onEventGenerated()" << std::endl;
   if (event.hasAttribute("position"))
     for(int i = 0; i < 3; i ++)
       position[i]=event.getPosition()[i];
   else
     {
       position[0]=0.0;
       position[1]=0.0;
       position[2]=0.0;
     }
 
   if (event.hasAttribute("orientation"))
     {
       for  (int i = 0; i < 4; i ++) {
         orientation[i]= event.getOrientation()[i];
       }
     }
   else
     {
       orientation[0]=0.0;
       orientation[1]=0.0;
       orientation[2]=0.0;
       orientation[3]=0.0;
     }
     
 
   if (event.hasAttribute("image") &&
       event.hasAttribute("xsize") &&
       event.hasAttribute("ysize") &&
       event.hasAttribute("totalysize") &&
       event.hasAttribute("line") //&&
       //event.hasAttribute("endflag")
       )
     {
       Image segimg   = event.getAttribute((Image*)NULL, "image");
       width_buf      = event.getAttribute(std::string("xsize"),0);
       height_buf     = event.getAttribute(std::string("totalysize"),0);
       int segheight  = event.getAttribute(std::string("ysize"),0);
       int line       = event.getAttribute(std::string("line"),0);
       int endflag    = event.getAttribute(std::string("endflag"), 0);
       int index      = event.getAttribute(std::string("index"), 0);
   
       if (!image_buf) {
         image_buf = new Image();
         image_buf->SetSize(width_buf, height_buf, sizeof(short));
         image_buf->image_ptr = (void*)malloc(image_buf->size());
       } else if (image_buf->size() != width_buf * height_buf * sizeof(short)) {
         free(image_buf->image_ptr);
         image_buf->SetSize(width_buf, height_buf, sizeof(short));
         image_buf->image_ptr = (void*)malloc(image_buf->size());
       }
       short* dest = (short*)image_buf->image_ptr;
 
       if (width_buf*segheight > 0)
         memcpy(&dest[width_buf*line], segimg.image_ptr, width_buf*segheight*sizeof(short));
       if (endflag)
         {
           // switch image buffer
           Image* tmp;
           int    tmp_width, tmp_height;
           tmp       = image;
           image     = image_buf;
           image_buf = tmp;
 
           tmp_width = width;
           width     = width_buf;
           width_buf = tmp_width;
 
           tmp_height= height;
           height    = height_buf;
           height_buf= tmp_height;
 
           timestamp = event.time;
         }
     }
   else
     {
       width  = 0;
       height = 0;
     }
 }

pullimage.cxx

 MyTutorialModule* mtm = (MyTutorialModule*) context.getModule("MyTutorialConfig");
 
 while (stopflag == 0) {
   
   stopflag = context.loopOnce();   // push and pull Event
   
   if (mtm) {
     Image img;
     int w, h;
     
     mtm->getImage(img, w, h);
     if (img.size() != 0) {
        
       // do something for the received image here
        ...
     
     }
   }
 
   usleep(rate);
 }

Testing

In terminal 1 (sending side: pushimage)

$ ./pushimage tutorial_source.xml

In terminal 2 (receiving side: pullimage)

$ ./pullimage tutorial_sink.xml

Current Probelms on Image Transfer using NaviTrack

Microsoft Windows issue

In case that the image size exceeds approx 8000 bytes (uncertain), data transfer from NetworkSink to NetworkSource might fail in Windows environment. This is thought to be due to an implementation of OpenTracker or ACE library, which NaviTrack depends on, but we are not sure the reason, and we continue to work on this issue. At this moment, to avoid this problem, the programs used in this tutorial splits image data into several fragments with size of less than 8000 bytes in a pushimage program, and reconstruct a complete image from fragments in a pullimage program.

UDP issue

You may encounter application errors, while passing image data through a network using NaviTrack NetworkModule. This is because NaviTrack NetworkModule uses UDP for a communication between Sink and Source. Each data has header information in this communication including entire size of the data, and packet loss causes inconsistency between data size information in header and actual data size. Currently, NaviTrack has no capability to recover this kind of errors.

Back to Integrating into your application.