IRDeviceDS.h

/******************************************************************************
 * Copyright (c) 2012-2017 All Rights Reserved, http://www.evocortex.com      *
 *  Evocortex GmbH                                                            *
 *  Emilienstr. 1                                                             *
 *  90489 Nuremberg                                                            *
 *  Germany                                                                   *
 *                                                                            *
 * Contributors:                                                              *
 *  Initial version for Linux 64-Bit platform supported by Fraunhofer IPA,    *
 *  http://www.ipa.fraunhofer.de                                              *
 *****************************************************************************/
 
#pragma once
#pragma comment(lib, "strmiids")
 
#define SKIP_DXTRANS
//#define SHOW_DEBUG_RENDERER
#define _CRT_SECURE_NO_WARNINGS
 
/* If you are having the problem you can't open dxtrans.h:
 * Open qedit.h and add this to the start of the file:
 *
 * #ifdef SKIP_DXTRANS
 * #define __IDxtAlphaSetter_INTERFACE_DEFINED__
 * #define __IDxtJpeg_INTERFACE_DEFINED__
 * #define __IDxtKey_INTERFACE_DEFINED__
 * #define __IDxtCompositor_INTERFACE_DEFINED__
 * #endif
 *
 * Also replace the line 
 * #include "dxtrans.h"
 * with:
 * #ifndef SKIP_DXTRANS
 * #include "dxtrans.h"
 * #endif
 */
 
#include "devices/directshow/SampleGrabber.h"
#include <windows.h>
#include <map>
#include <atomic>
#include "IRDevice.h"
#include "irdirectsdk_defs.h"
#include "Timer.h"
 
namespace evo
{
 
#ifndef MAXLONGLONG
#define MAXLONGLONG 0x7FFFFFFFFFFFFFFF
#endif
 
#ifndef MAX_DEVICE_NAME
#define MAX_DEVICE_NAME 80
#endif
 
#ifndef BITS_PER_PIXEL
#define BITS_PER_PIXEL 16
#endif
 
class IRDeviceDS;
struct ISampleGrabber;
class CallbackHandler;
 
#define SAFE_DS_RELEASE(x) { if (x) x->Release(); x = NULL; }
 
class __IRDIRECTSDK_API__ IRDeviceDS : public IRDevice
{
  friend class CallbackHandler;
 
public:
 
  ~IRDeviceDS();
 
  static IRDeviceDS*      createInstance(IRDeviceParams &params);
 
  static bool  findSerial(unsigned long &query, unsigned int skipNResults = 0);
 
  int                     startStreaming();
 
  int                     stopStreaming();
 
  const char*             getFriendlyName();
 
  virtual void            run();
 
  virtual void            exit();
 
  bool isOpen();
 
  virtual IRDeviceError   getFrame(unsigned char* buffer, double* timestamp = NULL, int timeoutMilliseconds = -1);
 
private:
  IRDeviceDS(unsigned short bufferQueueSize);
 
  static IRDeviceDS*      createVideoGraph(unsigned long serial, int videoFormatIndex, unsigned short bufferQueueSize);
 
  static void             removeDownstream(IBaseFilter *pf, IFilterGraph2* pFg);
 
  static HRESULT          findCaptureDevice(IRDeviceDS ** device, unsigned long serial, unsigned short bufferQueueSize);
 
  HRESULT                 initializeGraph();
 
  void                    onRawFrame(unsigned char* data, int len);
 
  static unsigned long    determineSerialNumber(wchar_t* deviceName);
 
  static void             deleteMediaType(AM_MEDIA_TYPE *pmt);
 
  char*                   _friendlyName;
 
  WCHAR*                  _filterName;
 
  IBaseFilter*            _sourceFilter;
 
  IBaseFilter*            _sampleGrabberFilter;
 
  IBaseFilter*            _nullRenderer;
 
  IEvoSampleGrabber*      _sampleGrabber;
 
  ICaptureGraphBuilder2*  _builder;
 
  IMediaControl*          _control;
 
  IFilterGraph2*          _graph;
 
  CallbackHandler*        _callbackHandler;
 
  IMediaEventEx*          _eventEx;
 
  unsigned char**         _rawBuffer;
 
  double*                 _rawBufferTimestamps;
 
  byte                    _rawBufferIndexWrite;
 
  byte                    _rawBufferIndexRead;
 
  byte                    _newFrameCount;
 
  CRITICAL_SECTION        _critSec;
 
  bool                    _run;
 
  Timer                   _timer;
 
  bool                    _isOpen;
 
  HANDLE                  _available;
 
  const unsigned short    _bufferQueueSize = 4;
};
 
} // namespace