FFmpegDecoder.h

Go to the documentation of this file.

//    Copyright 2004-2019, SenseGraphics AB, PolyDimensions
//
//    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.
//
//
//    Original file contributed by PolyDimensions.
//
//
 
#ifndef __FFMPEGDECODER_H__
#define __FFMPEGDECODER_H__
 
#include <H3D/FieldTemplates.h>
#include <H3D/SFTime.h>
#include <time.h>
 
#ifdef HAVE_FFMPEG
 
extern "C" {
#define __STDC_CONSTANT_MACROS
 
#ifdef _STDINT_H
 
#undef _STDINT_H
 
#endif
#include <avcodec.h>
#include <avformat.h>
#include <swscale.h>
}
 
#include <H3D/H3DVideoClipDecoderNode.h>
 
namespace H3D {
  class H3DAPI_API FFmpegDecoder : public H3DVideoClipDecoderNode {
    class FFmpegEventHandler: public AutoUpdate< SFTime > {
      virtual void update();
    };
 
  public:
    FFmpegDecoder();
 
    ~FFmpegDecoder();
 
    virtual bool testClip( const string &url );
 
    virtual bool loadClip( const string &url );
 
    virtual void startPlaying();
 
    virtual void stopPlaying();
 
    virtual void pausePlaying();
 
    virtual void setLooping( bool /*v*/ ) {
 
      // Quick fix - for some reason the incorrect value of v gets passed 
      looping = true;
 
/*
      if (v == true) 
        Console(LogLevel::Error) << "setLooping(true)" << endl;
      if (v == false) 
        Console(LogLevel::Error) << "setLooping(false)" << endl;
      looping = v;
*/
    }
 
    virtual H3DTime getPosition();
 
    virtual void setPosition( H3DTime pos );
 
    virtual bool setRate( double r ) { 
      rate = r;
      return 1;
    } 
 
    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 frame_bpp; }
 
    virtual unsigned int getFrameSize() { 
      return frame_height * frame_width * (frame_bpp / 8); 
    } 
 
    virtual bool theTimeIsRight();
 
    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:
 
    uint8_t *temp_buffer;
 
    struct SwsContext *imgConvertCtx;
 
    AVFormatContext *pFormatCtx;
 
    AVCodecContext *pCodecCtx;
 
    AVCodecContext *aCodecCtx;
 
    AVCodec *pCodec;
 
    AVCodec *aCodec;
 
    AVFrame *pFrame;
 
    AVFrame *pFrameRGB;
 
    unsigned char *data;
 
    AVPacket packet;
 
    double pts;
 
    int frameFinished;
 
    int videoStream;
 
    int audioStream;
 
    int numBytes;
 
    long current_frame;
 
    float video_fps;
 
    unsigned long current_time;
 
    unsigned long start_time;
  
    void cleanupFFmpeg( void );
 
    unsigned int frame_width;
 
    unsigned int frame_height;
 
    unsigned int frame_bpp;
 
    unsigned int data_size;
 
    bool have_new_frame;
 
    double rate;
  
    bool looping;
 
    H3DTime duration;
 
    bool just_a_test;
 
  };
 
}
#endif // HAVE_FFMPEG
#endif // __FFMPEGDECODER_H__