DirectShowDecoder.h

Go to the documentation of this file.

//    Copyright 2004, 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 "FieldTemplates.h"
#include "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>

#ifdef RESTORE__DEBUG
#define _DEBUG
#undef RESTORE__DEBUG
#endif

#ifdef RESTORE_DEBUG
#define DEBUG
#undef RESTORE_DEBUG
#endif

#ifdef _MSC_VER
#pragma comment( lib, "strmbase.lib" )
#endif

#include "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 ) {
      HRESULT hr;
      if( g_pMS ) 
        hr = g_pMS->SetRate( r ); 
      rate = r;
      return hr == S_OK;
    }

    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_width * frame_height * 3; 
    }

    virtual Image::PixelType getFramePixelType() { return Image::RGB; }
    
    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 data_size;

    unsigned char *data;

    bool have_new_frame;

    double rate;
  
    CFrameGrabber *frame_grabber;

    bool looping;

    H3DTime duration;

    auto_ptr< DShowEventHandler > event_handler;
  };

}
#endif
#endif

---