/home/fwph/code/wurde/rde/utils/ImageSource.cpp

Go to the documentation of this file.

#include <ImageSource.H>
#include <Logger.H>

using namespace WURDE;
using namespace std;

WURDEImage ImageSource::getImage(){

       if(m_supplier){
              m_localImage.second=NULL;
              return m_localImage;
       }

       // check if we need to release the current image
       if((m_localImage.second)){

              if(m_localImage.second->width!=info.width.getValue()||
                 m_localImage.second->height!=info.height.getValue()||
                 m_format!=info.format.getValue()||
                 m_compression!=info.compression.getValue()){
                     cvReleaseImage(&m_localImage.second);
                     m_localImage.second=NULL;
                     m_format=info.format.getValue();
                     m_compression=info.compression.getValue();
              }
       }else{
             m_format=info.format.getValue();
             m_compression=info.compression.getValue();
       }
       
       // allocate a new image
       if(!(m_localImage.second)){
              switch(m_format){
                     
              case FORMAT_CHAR_111_YUV24:
              case FORMAT_CHAR_111_RGB:
                     
                     m_localImage.second=cvCreateImage(cvSize(info.width.getValue(), info.height.getValue()), IPL_DEPTH_8U, 3);
                     break;

              case FORMAT_CHAR_11_YUV422:
                     m_localImage.second=cvCreateImage(cvSize(info.width.getValue(), info.height.getValue()), IPL_DEPTH_8U, 2);
                     break;

              case FORMAT_CHAR_1_GRAY:
                     m_localImage.second=cvCreateImage(cvSize(info.width.getValue(), info.height.getValue()), IPL_DEPTH_8U, 1);
                     break;
                     
              default:
                     char buffer[512];
                     sprintf(buffer,"Unhandled format in ImageSource: %d,%d",m_format,info.format.getValue());
                     g_warn(buffer);
                     return m_localImage;
              }
       }

       
       // copy the image data
       switch(m_compression){
       case COMPRESSION_RLE_HIGHQUALITY:
              switch(m_format){
              case FORMAT_CHAR_111_YUV24:
              case FORMAT_CHAR_111_RGB:
                     //PIX functions are written for three channel images
                     GCM_rleDecodePixMed(data.image,data.imagesize,(unsigned char *)m_localImage.second->imageData);
                     break;
              case FORMAT_CHAR_11_YUV422:
                     //YUV422 compression will probably suck
              case FORMAT_CHAR_1_GRAY:
                     GCM_rleDecodeGreyMed(data.image,data.imagesize,(unsigned char *)m_localImage.second->imageData);                
                     break;
              default:
                     g_error("Attempted to uncompress image of type not supported for compression.");
              }
              break;
       case COMPRESSION_RLE_LOWQUALITY:
              switch(m_format){
              case FORMAT_CHAR_111_YUV24:
              case FORMAT_CHAR_111_RGB:
                     //PIX functions are written for three channel images
                     GCM_rleDecodePixLow(data.image,data.imagesize,(unsigned char *)m_localImage.second->imageData);
                     break;
              case FORMAT_CHAR_11_YUV422:
                     //YUV422 compression will probably suck
              case FORMAT_CHAR_1_GRAY:
                     GCM_rleDecodeGreyLow(data.image,data.imagesize,(unsigned char *)m_localImage.second->imageData);                
                     break;
              default:
                     g_error("Attempted to uncompress image of type not supported for compression.");
              }
              break;

       case COMPRESSION_NULL:
       case COMPRESSION_UNCOMPRESSED:
              memcpy(m_localImage.second->imageData, data.image,data.imagesize);
              break;


       }

       m_localImage.first=data.timestamp;

       return m_localImage;

}

void ImageSource::publishImage(WURDEImage &image){

       if(!m_supplier){
              return;
       }

       //set the info and publish if necessary 
       if(image.second->width!=info.width.getValue()||
          image.second->height!=info.height.getValue()||
          m_format!=info.format.getValue()||
          m_compression!=info.compression.getValue()){
              info.width.setValue(image.second->width);
              info.height.setValue(image.second->height);
              info.format.setValue(m_format);
              info.compression.setValue(m_compression);
              info.timestamp.now();
              publishInfo();
              //              m_format=info.format.getValue();
              // m_compression=info.compression.getValue();
              if(data.image){
                     delete [] data.image;
                     data.image=NULL;       
              }  
       }

       //allocate the image if necessary

       switch(m_format){
              
       case FORMAT_CHAR_111_YUV24:
       case FORMAT_CHAR_111_RGB:

              if(!data.image){
                     data.image = new unsigned char[image.second->width*image.second->height*3];
                     data.imagesize=sizeof(unsigned char)*image.second->width*image.second->height*3;
              }
              memcpy(data.image,image.second->imageData,data.imagesize);

              switch(m_compression){
              case COMPRESSION_RLE_HIGHQUALITY:
                     g_logdebug << "Compressing an image" << endl;
                     data.imagesize=GCM_rleEncodePixMed(data.image,image.second->height, image.second->width);
                     break;
              case COMPRESSION_RLE_LOWQUALITY:
                     data.imagesize=GCM_rleEncodePixLow(data.image,image.second->height, image.second->width);
                     break;
              case COMPRESSION_NULL:
              case COMPRESSION_UNCOMPRESSED:
              default:
                     break;
              }
       case FORMAT_CHAR_11_YUV422:
              if(!data.image){
                     data.image = new unsigned char[image.second->width*image.second->height*2];
                     data.imagesize=sizeof(unsigned char)*image.second->width*image.second->height*2;
              }
              memcpy(data.image,image.second->imageData,data.imagesize);
              switch(m_compression){
              case COMPRESSION_RLE_HIGHQUALITY:
                     data.imagesize=GCM_rleEncodeGreyMed(data.image,image.second->height, image.second->width*2);
                     break;
              case COMPRESSION_RLE_LOWQUALITY:
                     data.imagesize=GCM_rleEncodeGreyLow(data.image,image.second->height, image.second->width*2);
                     break;
              case COMPRESSION_NULL:
              case COMPRESSION_UNCOMPRESSED:
              default:
                     break;
              }

              break;
              
       case FORMAT_CHAR_1_GRAY:
              if(!data.image){
                     data.image = new unsigned char[image.second->width*image.second->height];
                     data.imagesize=sizeof(unsigned char)*image.second->width*image.second->height;
              }
              memcpy(data.image,image.second->imageData,data.imagesize);
              switch(m_compression){
              case COMPRESSION_RLE_HIGHQUALITY:
                     data.imagesize=GCM_rleEncodeGreyMed(data.image,image.second->height, image.second->width);
                     break;
              case COMPRESSION_RLE_LOWQUALITY:
                     data.imagesize=GCM_rleEncodeGreyLow(data.image,image.second->height, image.second->width);
                     break;
              case COMPRESSION_NULL:
              case COMPRESSION_UNCOMPRESSED:
              default:
                     break;
              }
              
              break;
              
       default:
              g_error("Unhandled format in ImageSource");
       }
       data.timestamp=image.first;
       publishData();
}

---