vetVideo4Linux.cpp

Go to the documentation of this file.

#include "vetVideo4Linux.h"
#include "../vetUtility.h"

#include <unistd.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <stdio.h>
#include <sys/ioctl.h>
#include <stdlib.h>
#include <linux/types.h>

vetVideo4Linux::vetVideo4Linux(char* device) : vetInput(0)
 {
        fd = -1;
        videoBuffer = NULL;
        reset();
        connect(device);
 }

vetVideo4Linux::~vetVideo4Linux()
 {
        disconnect();
        delete [] videoBuffer;
 }

VETRESULT vetVideo4Linux::connect(char* device)
 {
        if ( device == NULL )
                return VETRET_PARAM_ERR;

        if ( fd != -1 )
                return VETRET_ILLEGAL_USE;      //must disconnect first

        // Open the video4link device for reading.
        fd = open(device,O_RDONLY);

        if (fd < 0)
                return VETRET_INTERNAL_ERR;     //throw "Cannot open device";

        //Make sure this is a video4linux device
        if (ioctl(fd,VIDIOCGCAP, &cap) < 0) {
                close(fd);
                return VETRET_INTERNAL_ERR;     //throw "VIDEOGCAP, not a video4linux device";
         }

        //Get the video overlay window
        if (ioctl(fd,VIDIOCGWIN, &win) < 0) {
                close(fd);
                return VETRET_INTERNAL_ERR;     //throw "VIDIOCGWIN";
         }

        //Get the picture properties
        if (ioctl (fd,VIDIOCGPICT, &vpic) < 0) {
                close(fd);
                return VETRET_INTERNAL_ERR;     //throw "VIDIOCGPICT";
         }

        //Now set the color resolution on the camera
        if (cap.type & VID_TYPE_MONOCHROME)
         {
                vpic.depth=8;
                vpic.palette=VIDEO_PALETTE_GREY;
                if(ioctl(fd,VIDIOCSPICT,&vpic) < 0)
                 {
                        vpic.depth=6;
                        if(ioctl(fd,VIDIOCSPICT, &vpic) < 0)
                         {
                                vpic.depth=4;
                                if (ioctl(fd,VIDIOCSPICT, &vpic) <0) {
                                        close(fd);
                                        return VETRET_INTERNAL_ERR;     //throw "Unable to find a supported Capture format";
                                 }
                         }
                 }
         }
        else
         {
                vpic.depth = 24;
                vpic.palette = VIDEO_PALETTE_RGB24;

                if (ioctl(fd,VIDIOCSPICT,&vpic) <0)
                 {
                        vpic.palette=VIDEO_PALETTE_RGB565;
                        vpic.depth=16;

                        if(ioctl(fd,VIDIOCSPICT,&vpic) == -1)
                         {
                                vpic.palette = VIDEO_PALETTE_RGB555;
                                vpic.depth=15;

                                if(ioctl(fd, VIDIOCSPICT, &vpic) == -1) {
                                                close(fd);
                                                throw "Unable to find a supported Capture format";
                                 }
                         }
                 }
         }
        bpp = vpic.depth;

        if (win.width && win.height)
         {
                if (videoBuffer != NULL)
                        delete [] videoBuffer;

                videoBuffer = (unsigned char*) malloc(win.width * win.height * 3);

                if (videoBuffer == NULL)
                        return VETRET_INTERNAL_ERR;
         }

        return VETRET_OK;
 }

VETRESULT vetVideo4Linux::disconnect()
 {
        if ( fd==-1 )
                return VETRET_ILLEGAL_USE;

        close(fd);
        fd=-1;
        win.height = 0;
        win.width = 0;
        win.height = 0;
        vpic.palette = 0;

        return VETRET_OK;
 }

VETRESULT vetVideo4Linux::reset()
 {
        disconnect();

        if (videoBuffer != NULL)
                delete [] videoBuffer;

        videoBuffer = NULL;
        fd = -1;
        win.height = 0;
        win.width = 0;
        win.height = 0;
        vpic.palette = 0;

        return VETRET_OK;
 }


bool vetVideo4Linux::EoF()
 {
        if ( fd == -1 ) // not connected
                return true;

        return false;
 }


VETRESULT vetVideo4Linux::extractTo(vetFrameYUV420& img)
 {
        if ( fd == -1 )
                return VETRET_ILLEGAL_USE;

        if (img.width != win.width || img.height != win.height)
         {
                 img.reAllocCanvas(win.width, win.height);
         }

        if ( vpic.palette == VIDEO_PALETTE_RGB24 )
         {
                return VETRET_NOT_IMPLEMENTED;
         }
        else if ( vpic.palette == VIDEO_PALETTE_RGB565 )
         {
                return VETRET_NOT_IMPLEMENTED;
         }
        else if ( vpic.palette == VIDEO_PALETTE_RGB555 )
         {
                return VETRET_NOT_IMPLEMENTED;
         }

        return VETRET_OK;
 }

VETRESULT vetVideo4Linux::extractTo(vetFrameRGB24& img)
 {
        if ( fd == -1 )
                return VETRET_ILLEGAL_USE;

        if (img.width != win.width || img.height != win.height)
         {
                 img.reAllocCanvas(win.width, win.height);
         }

        if ( vpic.palette == VIDEO_PALETTE_RGB24 )
         {
                if (videoBuffer == NULL)
                        return VETRET_INTERNAL_ERR;

                read(fd,videoBuffer, win.width * win.height * 3 * sizeof(unsigned char) ); //BGR
                vetUtility::conv_bgr_rgb((unsigned char*)img.data[0], videoBuffer, win.width, win.height);
         }
        else if ( vpic.palette == VIDEO_PALETTE_RGB565 )
         {
                return VETRET_NOT_IMPLEMENTED;
         }
        else if ( vpic.palette == VIDEO_PALETTE_RGB555 )
         {
                return VETRET_NOT_IMPLEMENTED;
         }

        return VETRET_OK;
 }

VETRESULT vetVideo4Linux::extractTo(vetFrameT<unsigned char>& img)
 {
        if ( fd == -1 )
                return VETRET_ILLEGAL_USE;

        if (win.width != img.width || win.height != img.height)
                img.reAllocCanvas(win.width, win.height);

        if ( vpic.palette == VIDEO_PALETTE_RGB24  && img.profile == vetFrame::VETFRAME_BGR24 )
         {
                read(fd, img.data, win.width * win.height * 3 * sizeof(unsigned char) );//BGR
                return VETRET_OK;
         }
        if ( vpic.palette == VIDEO_PALETTE_RGB24  && img.profile == vetFrame::VETFRAME_RGB24 )
         {
                if (videoBuffer == NULL)
                        return VETRET_INTERNAL_ERR;

                read(fd,videoBuffer, win.width * win.height * 3 * sizeof(unsigned char) ); //BGR
                vetUtility::conv_bgr_rgb(img.data, videoBuffer, win.width, win.height);

         }

        return VETRET_NOT_IMPLEMENTED;
 }

---