snapdvc.cpp

#include "stdafx.h"
#include <stdio.h>
#include <stdlib.h>
#include <time.h>
#include <windows.h>
#include <math.h>

#include <dvcAPI.h>



int main(int argc, char* argv[])
{       
        HANDLE hDevice ;
        int i, nStatus, nWidth, nHeight ;
        PUSHORT pRawVideoData ;
        PUSHORT pRGBVideoData ;
        BOOL bRC ;
        BOOL bIsColor ;
        char fname[256] ;
        FILE *fp ;
        
        hDevice = dvcOpenCamera(1) ;

        if(hDevice == INVALID_HANDLE_VALUE)
        {
                fprintf(stderr,"Error opening DVC Camera!\n");
                return -1 ;
        }

        for( i = 1 ; i < argc ; i++ )
        {
                if(i+1 < argc && !strcmp(argv[i],"-b")) // Set binning -b 0|1|2|3|4
                {
                        int nBin = atoi(argv[++i]) ;
                        bRC = dvcSetBinning(hDevice, nBin) ;
                        if(!bRC)
                                dvcPrintf(0,"%s: Error setting binning %d\n", argv[0],nBin);
                }
                else if(i+1 < argc && !strcmp(argv[i], "-g"))
                {
                        int nGain = atoi(argv[++i]) ;
                        bRC = dvcSetGain(hDevice, nGain) ;
                        if(!bRC)
                                dvcPrintf(0,"%s: Error setting gain %d\n", argv[0],nGain);
                }
                else if(i+1 < argc && !strcmp(argv[i], "-o"))
                {
                        int nOffset = atoi(argv[++i]) ;
                        bRC = dvcSetOffset(hDevice, nOffset) ;
                        if(!bRC)
                                dvcPrintf(0,"%s: Error setting offset %d\n", argv[0],nOffset);
                }
                else if(i+1 < argc && !strcmp(argv[i], "-e"))
                {
                        double dExposeMsec = atof(argv[++i]) ;
                        double dReturnExp =     dvcSetExposeMsec(hDevice, dExposeMsec) ;

                        if(dReturnExp != dExposeMsec)
                                dvcPrintf(0,"%s: Set expose to %g [ Not %g ]\n", argv[0], dReturnExp, dExposeMsec);
                        
                }
        }


        nStatus = dvcGetStatus(hDevice) ;
        nWidth = dvcGetXDim(hDevice) ;
        nHeight = dvcGetYDim(hDevice) ;
        bIsColor = dvcIsColor(hDevice) ;


        pRawVideoData = (PUSHORT)calloc(nWidth * nHeight, sizeof(USHORT));
        if(NULL == pRawVideoData)
        {
                dvcPrintf(0,"%s: Error allocating buffer for raw video.\n");
                dvcCloseCamera(hDevice);
                return -1 ;
        }

        if(bIsColor)
        {
                pRGBVideoData = (PUSHORT)calloc(nWidth * nHeight,  3 * sizeof(USHORT));
                if(NULL == pRGBVideoData)
                {
                        dvcPrintf(0,"%s: Error allocating buffer for RGB video.\n");
                        dvcCloseCamera(hDevice);
                        free(pRawVideoData);
                        return -1 ;
                }
        }


        if(nStatus != DVC_STATUS_RUNNING)       // If not currently streaming video..
        {
                bRC = dvcStartSequence(hDevice, 0) ;
                if(!bRC)
                {
                        dvcPrintf(0,"%s: Error starting video capture..\n");
                        dvcCloseCamera(hDevice);
                        if(bIsColor) free(pRGBVideoData);
                        free(pRawVideoData);
                        return -1 ;
                }
        }

        // Wait for a frame 
        
        bRC = dvcWaitImage(hDevice, 10000.0 ) ; // Wait up to 10 seconds..


        if(!bRC)
        {
                dvcPrintf(0,"%s: Error waiting for video capture..\n");
                dvcCloseCamera(hDevice);
                if(bIsColor) free(pRGBVideoData);
                free(pRawVideoData);
                return -1 ;
        }

        // This stops captures.  The driver internally buffers the raw
        // data so,  the read functions below return data from
        // the last frame captured.
        if(nStatus != DVC_STATUS_RUNNING)       
                dvcStopSequence(hDevice) ;

        // Read the raw video ( Bayer ) data..
        if(bRC) bRC = dvcReadImage(hDevice, 
                                pRawVideoData,
                                0, 0,                   // Left, top
                                nWidth, nHeight         // width, height
                                ) ;

        if(!bRC)
                dvcPrintf(0,"Error reading raw video!\n");

        // Read the raw video and converts to 48-bits/pixel
        // i.e.,  each pixel represented by 3 16-bit values, R,G,B,R,G,B,..
        if(bRC && bIsColor)
                        bRC = dvcReadImageRGB48(hDevice, 
                                        pRGBVideoData,
                                        0, 0,                   // Left, top
                                        nWidth, nHeight         // width, height
                                        ) ;
        if(!bRC)
                dvcPrintf(0,"Error reading RGB video!\n");
        else 
                fprintf(stdout,"Captured a %d x %d %s Image.\n",
                                nWidth, nHeight,
                                bIsColor ? "Color" : "B&W" ) ;

        if(bRC)
        {
                printf("Enter a file name to save the raw image data >") ;
                fgets(fname,255,stdin);
                if(strlen(fname) > 1)
                {
                        fname[strlen(fname)-1] = '\0' ;
                        fp = fopen(fname,"wb");
                        if(fp)
                        {
                                fwrite(
                                        (void *)pRawVideoData,
                                        nWidth * nHeight,
                                        sizeof(USHORT),
                                        fp 
                                        ) ;
                                fclose(fp) ;
                        }
                }
        }
        if(bRC && bIsColor)
        {
                printf("Enter a file name to save the RGB image data >") ;
                fgets(fname,255,stdin);
                if(strlen(fname) > 1)
                {
                        fname[strlen(fname)-1] = '\0' ;
                        fp = fopen(fname,"wb");
                        if(fp)
                        {
                                fwrite(
                                        (void *)pRGBVideoData,
                                        nWidth * nHeight,
                                        3*sizeof(USHORT),
                                        fp 
                                        ) ;
                                fclose(fp) ;
                        }
                }
        }


        free(pRawVideoData);
        if(bIsColor) free(pRGBVideoData);

        dvcCloseCamera(hDevice) ;


        return 0;
}

---