roboforum.ru

#pragma comment(lib,"Wininet.lib")

#include "stdafx.h"
#include <cv.h>
#include <highgui.h>
#include <stdlib.h>
#include <stdio.h>
#include <Windows.h>
#include "WinInet.h"
#include <string.h>
#include "cxcore.h"

#define strncasecmp _strnicmp

struct tmjpeg
{
   unsigned char *pbuff;
   DWORD ibuffsize;
   unsigned char *pmess;
   DWORD imesssize;
   IplImage* image;
   int width;
   int height;
   HINTERNET hSession;
   HINTERNET hConnect;
   HINTERNET hRequest;
   char szUrl[MAX_PATH];
};
tmjpeg mjpeg;

static int PeekData(void)
{
   int ilen;
   
   // узнаём размер следующего пакета
   InternetQueryDataAvailable(mjpeg.hRequest,&mjpeg.imesssize,0,0);
   // выходим если размер нулевой
   if(mjpeg.imesssize<=0)
      return -1;

   // зарезервируем массив байт для очередной порции
   mjpeg.pmess=new unsigned char[mjpeg.imesssize];
   if(mjpeg.pmess==NULL)
      return -1;
   // эта переменка будет получать число считанных байт
   DWORD nbr=0;
   // Если считывание удается
   if (!InternetReadFile(mjpeg.hRequest,mjpeg.pmess,mjpeg.imesssize,&nbr))
      return -1;
   //Формируем временный буфер размером старый буфер+ новый пакет
   ilen=mjpeg.ibuffsize+mjpeg.imesssize;
   unsigned char *ps= new unsigned char[ilen];
   if(ps==NULL)
      return -1;
   //Если это не первичный запуск
   if(mjpeg.ibuffsize>0)
   {
      // Копируем старый буфер
      memcpy(ps,mjpeg.pbuff,mjpeg.ibuffsize);
      // Освобождаем память старого буфера
      free(mjpeg.pbuff);
   }
   // Копируем новый пакет
   memcpy(ps+mjpeg.ibuffsize,mjpeg.pmess,mjpeg.imesssize);
   // Освобождаем память пакета
   free(mjpeg.pmess);

   //Переназначаем буфер
   mjpeg.pbuff=ps;
   // Устанавливаем новый размер буфера
   mjpeg.ibuffsize=ilen;

   return mjpeg.imesssize;
}


bool CreateMJPEGCameraCapture(char *szUrl, int width, int heigth)
{
   mjpeg.hSession = InternetOpen(TEXT("MJPG listener"),INTERNET_OPEN_TYPE_DIRECT, NULL, NULL, 0);
   mjpeg.width=width;
   mjpeg.height=heigth;
   strcpy(mjpeg.szUrl,szUrl);
   if(mjpeg.hSession)
      return true;
   else
      return false;

}

IplImage* QueryMJPEGFrame(void)
{
   mjpeg.ibuffsize=0;
   mjpeg.imesssize=0;
   mjpeg.pmess=NULL;


   mjpeg.hRequest = InternetOpenUrlA(mjpeg.hSession, (LPCSTR)mjpeg.szUrl, NULL, 0, 0, 0);
   if(mjpeg.hRequest)
   {
      DWORD Size=0;
      // Выясняем по сколько портал нам будет высылать байт в очередной порции
      InternetQueryDataAvailable(mjpeg.hRequest,&Size,0,0);
      
      while(PeekData()>0);
      //printf("Bufflen=%d\n\n",mjpeg.ibuffsize);

      CvMat* mat = cvCreateMatHeader(1,mjpeg.ibuffsize, CV_8UC1 );
      cvCreateData( mat );
      cvSetData(mat,mjpeg.pbuff,mjpeg.ibuffsize);

      // декодируем массив в картинку
      if(mjpeg.image!=NULL)
         cvReleaseImage(& mjpeg.image);

      mjpeg.image = cvDecodeImage(mat);//,CV_LOAD_IMAGE_COLOR);
      
      cvReleaseData(mat);

      if(mjpeg.pbuff)
         free(mjpeg.pbuff);
      mjpeg.pbuff=NULL;
   

      InternetCloseHandle(mjpeg.hRequest);
   } else
      return NULL;


   return mjpeg.image;
}

bool ReleaseMJPEGCapture(void)
{
   if(mjpeg.hSession)
   {
      InternetCloseHandle(mjpeg.hSession);
   }

   if(mjpeg.image!=NULL)
        cvReleaseImage(&mjpeg.image);
   if(mjpeg.pbuff!=NULL)
      free(mjpeg.pbuff);

   return true;
}

int _tmain(int argc, _TCHAR* argv[])
{

   IplImage* frame;

   if(!CreateMJPEGCameraCapture("http://192.168.1.120:8080/?action=snapshot",640,480))
   {
      return 0;
   }
   cvNamedWindow("original",CV_WINDOW_AUTOSIZE);
        while(true)
   {
            // получаем кадр
         frame=QueryMJPEGFrame();
         if(frame==NULL)
         {
            printf("Error: No frame\n");
            break;
         }
            // показываем
            cvShowImage("original",frame);
            char c = cvWaitKey(33);
            if (c == 27)
         { // нажата ESC
                    break;
            }
   }

   
   // удаляем окно
   cvDestroyWindow("original");
   // освобождаем ресурсы
   ReleaseMJPEGCapture();
   return 0;
}

int main(int argc, char* argv[])
{
        IplImage* frame=0;
        CvCapture* capture = cvCreateCameraCapture(CV_CAP_ANY);
        cvNamedWindow("capture", CV_WINDOW_AUTOSIZE);
        while(true){
                frame = cvQueryFrame( capture );
                cvShowImage("capture", frame);
                char c = cvWaitKey(33);
                if (c == 27) break;
        }
        cvReleaseCapture( &capture );
        cvDestroyWindow("capture");
        return 0;
}

int _tmain(int argc, _TCHAR* argv[])
{
   IplImage* frame;
   CreateMJPEGCameraCapture("http://192.168.1.120:8080/?action=snapshot",640,480);
   cvNamedWindow("original",CV_WINDOW_AUTOSIZE);
   while(true)
   {
         frame=QueryMJPEGFrame();
         cvShowImage("original",frame);
         char c = cvWaitKey(33);
         if (c == 27) break;
   }
   cvDestroyWindow("original");
   ReleaseMJPEGCapture();
   return 0;
}

      if (! cinfo->entropy->insufficient_data) {
   //WARNMS(cinfo, JWRN_HIT_MARKER);
   cinfo->entropy->insufficient_data = TRUE;
      }