ffmpegFile.h

Go to the documentation of this file.

#ifndef ffmpegFile_H
#define ffmpegFile_H

#include "NDPluginFile.h"
#include "ffmpegCommon.h"
#ifdef WIN32
#define snprintf _snprintf
#endif

#define ffmpegFileBitrateString "FFMPEG_BITRATE"  /* (asynFloat64, r/w) File bitrate */
#define ffmpegFileFPSString     "FFMPEG_FPS"      /* (asynInt32, r/w) Frames per second */
#define ffmpegFileHeightString  "FFMPEG_HEIGHT"   /* (asynInt32, r/w) Video Height */
#define ffmpegFileWidthString   "FFMPEG_WIDTH"    /* (asynInt32, r/w) Video Width */

/* Minimum integer exactly representable in IEEE 754 double (-2^53) */
#define EXACT_INT_DBL_MIN -9007199254740992.0
/* Maximum integer exactly representable in IEEE 754 double (2^53) */
#define EXACT_INT_DBL_MAX 9007199254740992.0

class ffmpegFile : public NDPluginFile {
public:
    ffmpegFile(const char *portName, int queueSize, int blockingCallbacks,
               const char *NDArrayPort, int NDArrayAddr,
               int priority, int stackSize);
    /* The methods that this class implements */
    virtual asynStatus openFile(const char *fileName, NDFileOpenMode_t openMode, NDArray *pArray);
    virtual asynStatus readFile(NDArray **pArray);
    virtual asynStatus writeFile(NDArray *pArray);
    virtual asynStatus closeFile();
    virtual asynStatus writeFloat64(asynUser *pasynUser, epicsFloat64 value);

protected:
    int ffmpegFileBitrate;
    #define FIRST_FFMPEG_FILE_PARAM ffmpegFileBitrate
    int ffmpegFileFPS;
    int ffmpegFileHeight;
    int ffmpegFileWidth;
    #define LAST_FFMPEG_FILE_PARAM ffmpegFileWidth

private:
    FILE *outFile;
    AVCodec *codec;
    enum AVCodecID codec_id;
    AVCodecContext *c;
    AVFrame *inPicture;
    AVFrame *scPicture;    
    NDArray *scArray;
    NDArray *outArray;
    struct SwsContext *ctx;      
    size_t outSize;
    int needStop;      
    int sheight, swidth;
    enum AVPixelFormat spix_fmt;
    AVOutputFormat *fmt;
    AVFormatContext *oc;
    AVStream *video_st;
    double video_pts;   
};
#define NUM_FFMPEG_FILE_PARAMS (int)(&LAST_FFMPEG_FILE_PARAM - &FIRST_FFMPEG_FILE_PARAM + 1)   

#endif