DirectShowDecoder.h

Go to the documentation of this file.

//    Copyright 2004-2019, SenseGraphics AB
//
//    This file is part of H3D API.
//
//    H3D API is free software; you can redistribute it and/or modify
//    it under the terms of the GNU General Public License as published by
//    the Free Software Foundation; either version 2 of the License, or
//    (at your option) any later version.
//
//    H3D API is distributed in the hope that it will be useful,
//    but WITHOUT ANY WARRANTY; without even the implied warranty of
//    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
//    GNU General Public License for more details.
//
//    You should have received a copy of the GNU General Public License
//    along with H3D API; if not, write to the Free Software
//    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
//
//    A commercial license is also available. Please contact us at 
//    www.sensegraphics.com for more information.
//
//
//
 
#ifndef __DIRECTSHOWDECODER_H__
#define __DIRECTSHOWDECODER_H__
 
#include <H3D/FieldTemplates.h>
#include <H3D/SFTime.h>
 
#ifdef HAVE_DSHOW
 
// Include from DirectShow BaseClasses
// Remove the DEBUG defines when including DShow classes in order
// to always use the release version. The debug version caused us
// problems.
#ifdef _DEBUG
#define RESTORE__DEBUG
#undef _DEBUG
#endif
 
#ifdef DEBUG
#define RESTORE_DEBUG
#undef DEBUG
#endif
#include <streams.h>
#include <atlcomcli.h>
#include <strsafe.h>
 
#ifdef RESTORE__DEBUG
#define _DEBUG
#undef RESTORE__DEBUG
#endif
 
#ifdef RESTORE_DEBUG
#define DEBUG
#undef RESTORE_DEBUG
#endif
 
#include <H3D/H3DVideoClipDecoderNode.h>
 
namespace H3D {
 
struct __declspec(uuid("{71771540-2017-11cf-ae26-0020afd79767}")) CLSID_TextureRenderer;
 
  class H3DAPI_API DirectShowDecoder : public H3DVideoClipDecoderNode {
 
    class DShowEventHandler: public AutoUpdate< SFTime > {
      virtual void update();
    };
 
    class CFrameGrabber: public CBaseVideoRenderer {
    public:
      CFrameGrabber(LPUNKNOWN pUnk, HRESULT *phr, DirectShowDecoder *d );
 
      // CBaseVideoRenderer functions
 
      HRESULT CheckMediaType(const CMediaType *pmt );
 
      HRESULT SetMediaType(const CMediaType *pmt );
    
      HRESULT DoRenderSample(IMediaSample *pMediaSample); 
 
      void OnReceiveFirstSample(IMediaSample *pSample) { 
        DoRenderSample( pSample ); 
      }
 
      DirectShowDecoder *decoder;
    };
 
    friend class CFrameGrabber;
 
  public:
    DirectShowDecoder();
 
    ~DirectShowDecoder();
 
    virtual bool loadClip( const string &url );
 
    virtual void startPlaying();
 
    virtual void stopPlaying();
 
    virtual void pausePlaying();
 
    virtual void setLooping( bool v ) {
      looping = v;
    }
 
    virtual H3DTime getPosition();
 
    virtual void setPosition( H3DTime pos );
 
    virtual bool setRate( double r ) {
      rate = r;
      if( g_pMS ) {
        HRESULT hr;
        hr = g_pMS->SetRate( r );
        return hr == S_OK;
      }
      return false;
    }
 
    virtual H3DTime getDuration() {
      return duration;
    }
 
    virtual bool haveNewFrame() { return have_new_frame; }
 
    virtual void getNewFrame( unsigned char *buffer );
 
    virtual unsigned int getFrameWidth() { return frame_width; }
 
    virtual unsigned int getFrameHeight() { return frame_height; }
 
    virtual unsigned int getFrameBitsPerPixel() { return 24; }
 
    virtual unsigned int getFrameSize() { 
      return frame_size;
    }
 
    virtual unsigned int getFrameByteAlignment() {
      return 4;
    }
 
    virtual Image::PixelType getFramePixelType() { return Image::BGR; }
    
    virtual Image::PixelComponentType getFramePixelComponentType() { 
      return Image::UNSIGNED;
    }
 
    static H3DNodeDatabase database;  
  
    static DecoderRegistration reader_registration;
 
    static bool supportsFileType( const string &url );
  protected:
    // DirectShow pointers
    CComPtr<IGraphBuilder>  g_pGB;          // GraphBuilder
    CComPtr<IMediaControl>  g_pMC;          // Media Control
    CComPtr<IMediaSeeking> g_pMS;          // Media Seeking
    CComPtr<IMediaEvent>    g_pME;          // Media Event
    CComPtr<IBaseFilter>    g_pRenderer;    // our custom renderer
  
    HRESULT initDShowTextureRenderer( const string &url );
 
    void cleanupDShow( void );
 
    unsigned int frame_width;
 
    unsigned int frame_height;
 
    unsigned int frame_size;
 
    unsigned int data_size;
 
    unsigned char *data;
 
    bool have_new_frame;
 
    double rate;
  
    CFrameGrabber *frame_grabber;
 
    bool looping;
 
    H3DTime duration;
 
    H3DUniquePtr< DShowEventHandler > event_handler;
  };
 
}
#endif
#endif