VideoInput

//THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR //IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, //FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE //AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER //LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, //OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN //THE SOFTWARE. #include <iostream> #include "videoInput.h" #include <tchar.h> //Include Directshow stuff here so we don't worry about needing all the h files. #include <dshow.h> //#include "streams.h" #pragma include_alias( "dxtrans.h", "qedit.h" ) #define __IDxtCompositor_INTERFACE_DEFINED__ #define __IDxtAlphaSetter_INTERFACE_DEFINED__ #define __IDxtJpeg_INTERFACE_DEFINED__ #define __IDxtKey_INTERFACE_DEFINED__ #include <uuids.h> #include <aviriff.h> #include <windows.h> //for threading #include <process.h> #ifndef HEADER #define HEADER(pVideoInfo) (&(((VIDEOINFOHEADER *) (pVideoInfo))->bmiHeader)) #endif // Due to a missing qedit.h in recent Platform SDKs, we've replicated the relevant contents here // #include <qedit.h> MIDL_INTERFACE("0579154A-2B53-4994-B0D0-E773148EFF85") ISampleGrabberCB : public IUnknown { public: virtual HRESULT STDMETHODCALLTYPE SampleCB( double SampleTime, IMediaSample *pSample) = 0; virtual HRESULT STDMETHODCALLTYPE BufferCB( double SampleTime, BYTE *pBuffer, long BufferLen) = 0; }; MIDL_INTERFACE("6B652FFF-11FE-4fce-92AD-0266B5D7C78F") ISampleGrabber : public IUnknown { public: virtual HRESULT STDMETHODCALLTYPE SetOneShot( BOOL OneShot) = 0; virtual HRESULT STDMETHODCALLTYPE SetMediaType( const AM_MEDIA_TYPE *pType) = 0; virtual HRESULT STDMETHODCALLTYPE GetConnectedMediaType( AM_MEDIA_TYPE *pType) = 0; virtual HRESULT STDMETHODCALLTYPE SetBufferSamples( BOOL BufferThem) = 0; virtual HRESULT STDMETHODCALLTYPE GetCurrentBuffer( /* [out][in] */ long *pBufferSize, /* [out] */ long *pBuffer) = 0; virtual HRESULT STDMETHODCALLTYPE GetCurrentSample( /* [retval][out] */ IMediaSample **ppSample) = 0; virtual HRESULT STDMETHODCALLTYPE SetCallback( ISampleGrabberCB *pCallback, long WhichMethodToCallback) = 0; }; EXTERN_C const CLSID CLSID_SampleGrabber; EXTERN_C const IID IID_ISampleGrabber; EXTERN_C const CLSID CLSID_NullRenderer; //use videoInput::setVerbose to change static bool verbose = true; //use videoInput::setComMultiThreaded to change static bool VI_COM_MULTI_THREADED = false; /////////////////////////// HANDY FUNCTIONS ///////////////////////////// void MyFreeMediaType(AM_MEDIA_TYPE& mt){ if (mt.cbFormat != 0) { CoTaskMemFree((PVOID)mt.pbFormat); mt.cbFormat = 0; mt.pbFormat = NULL; } if (mt.pUnk != NULL) { // Unecessary because pUnk should not be used, but safest. mt.pUnk->Release(); mt.pUnk = NULL; } } void MyDeleteMediaType(AM_MEDIA_TYPE *pmt) { if (pmt != NULL) { MyFreeMediaType(*pmt); CoTaskMemFree(pmt); } } ////////////////////////////// CALLBACK //////////////////////////////// //Callback class class SampleGrabberCallback : public ISampleGrabberCB{ public: //------------------------------------------------ SampleGrabberCallback(){ InitializeCriticalSection(&critSection); freezeCheck = 0; bufferSetup = false; newFrame = false; latestBufferLength = 0; hEvent = CreateEvent(NULL, true, false, NULL); } //------------------------------------------------ ~SampleGrabberCallback(){ ptrBuffer = NULL; DeleteCriticalSection(&critSection); CloseHandle(hEvent); if(bufferSetup){ delete [] pixels; } } //------------------------------------------------ bool setupBuffer(int numBytesIn){ if(bufferSetup){ return false; }else{ numBytes = numBytesIn; pixels = new unsigned char[numBytes]; bufferSetup = true; newFrame = false; latestBufferLength = 0; } return true; } //------------------------------------------------ STDMETHODIMP_(ULONG) AddRef() { return 1; } STDMETHODIMP_(ULONG) Release() { return 2; } //------------------------------------------------ STDMETHODIMP QueryInterface(REFIID riid, void **ppvObject){ *ppvObject = static_cast<ISampleGrabberCB*>(this); return S_OK; } //This method is meant to have less overhead //------------------------------------------------ STDMETHODIMP SampleCB(double Time, IMediaSample *pSample){ if(WaitForSingleObject(hEvent, 0) == WAIT_OBJECT_0) return S_OK; HRESULT hr = pSample->GetPointer(&ptrBuffer); if(hr == S_OK){ latestBufferLength = pSample->GetActualDataLength(); if(latestBufferLength == numBytes){ EnterCriticalSection(&critSection); memcpy(pixels, ptrBuffer, latestBufferLength); newFrame = true; freezeCheck = 1; LeaveCriticalSection(&critSection); SetEvent(hEvent); }else{ printf("ERROR: SampleCB() - buffer sizes do not match\n"); } } return S_OK; } //This method is meant to have more overhead STDMETHODIMP BufferCB(double Time, BYTE *pBuffer, long BufferLen){ return E_NOTIMPL; } int freezeCheck; int latestBufferLength; int numBytes; bool newFrame; bool bufferSetup; unsigned char * pixels; unsigned char * ptrBuffer; CRITICAL_SECTION critSection; HANDLE hEvent; }; ////////////////////////////// VIDEO DEVICE //////////////////////////////// // ---------------------------------------------------------------------- // Should this class also be the callback? // // ---------------------------------------------------------------------- videoDevice::videoDevice(){ pCaptureGraph = NULL; // Capture graph builder object pGraph = NULL; // Graph builder object pControl = NULL; // Media control object pVideoInputFilter = NULL; // Video Capture filter pGrabber = NULL; // Grabs frame pDestFilter = NULL; // Null Renderer Filter pGrabberF = NULL; // Grabber Filter pMediaEvent = NULL; streamConf = NULL; pAmMediaType = NULL; //This is our callback class that processes the frame. sgCallback = new SampleGrabberCallback(); sgCallback->newFrame = false; //Default values for capture type videoType = MEDIASUBTYPE_RGB24; connection = PhysConn_Video_Composite; storeConn = 0; videoSize = 0; width = 0; height = 0; tryWidth = 0; tryHeight = 0; nFramesForReconnect= 10000; nFramesRunning = 0; myID = -1; tryDiffSize = false; useCrossbar = false; readyToCapture = false; sizeSet = false; setupStarted = false; specificFormat = false; autoReconnect = false; requestedFrameTime = -1; memset(wDeviceName, 0, sizeof(WCHAR) * 255); memset(nDeviceName, 0, sizeof(char) * 255); } // ---------------------------------------------------------------------- // The only place we are doing new // // ---------------------------------------------------------------------- void videoDevice::setSize(int w, int h){ if(sizeSet){ if(verbose)printf("SETUP: Error device size should not be set more than once \n"); } else { width = w; height = h; videoSize = w*h*3; sizeSet = true; pixels = new unsigned char[videoSize]; pBuffer = new char[videoSize]; memset(pixels, 0 , videoSize); sgCallback->setupBuffer(videoSize); } } // ---------------------------------------------------------------------- // Borrowed from the SDK, use it to take apart the graph from // the capture device downstream to the null renderer // --------------------------------------------