VideoFrameLoader.h

Go to the documentation of this file.

#pragma once

#include <mfapi.h>
#include <mfidl.h>
#include <mfreadwrite.h>

#include <list>



#define MAXVideoURLLenght                   256

#define VFL_OK                              0
#define VFL_ERROR                           -1
#define VFL_NO_FILE_URL_SET                 1
#define VFL_URL_TO_LONG                     2
#define VFL_PATH_TO_LONG                    3
#define VFL_NO_VIDEO_SOURCE                 4
#define VFL_VIDEO_ATTRIBUTES_NOT_SET        5
#define VFL_VIDEO_BUFFER_IS_NULL            6
#define VFL_COULD_NOT_CONVERT_PATH_TO_URL   25
#define VFL_COULD_NOT_READ_SOURCE_FILE      26
#define VFL_COULD_NOT_DECODE_FILE           27
#define VFL_COULD_NOT_GET_MEDIA_TYPE        28
#define VFL_COULD_NOT_GET_MEDIA_ATTRIBUTES  29
#define VFL_COULD_NOT_READ_ANY_FRAMES       30
#define VFL_NO_MORE_FRAMES                  31

typedef int VFLResult;

using namespace std;


class CVideoFrameLoader
{
public:

    CVideoFrameLoader(void);


    CVideoFrameLoader(PWSTR filePath);


    ~CVideoFrameLoader(void);


    VFLResult setVideoPath(PWSTR filePath);


    VFLResult setVideoURL(PWSTR fileURL);


    VFLResult loadAllFramesIntoBuffer();


    VFLResult loadAllFramesIntoBuffer(int offset, int distanceBetween);


    VFLResult loadAllFramesIntoBuffer(int offset, int distanceBetween, int lastFrame);


    VFLResult addNextFrameToBuffer();


    VFLResult addNextFrameToBuffer(int distanceFromLast);


    VFLResult getVideoResolution(int *width, int *height);


    VFLResult getVideoBuffer(PBYTE *videoArray);


    VFLResult getBufferFrameCount(int *frameCount);


    void reset();


    HRESULT setPosition(const LONGLONG& hnsPosition);
protected:
    PBYTE p_videoArray; 

    virtual VFLResult resetProperties();


    void copyVideoListToVideoArray();
private:
    PWSTR p_videoURL;
    IMFSourceReader *p_videoReader;
    int m_frameCount;
    int m_totalFrameCount;
    UINT32 m_videoWidth, m_videoHeight;
    bool m_videoBufferTransfered;
    bool m_comInitialisationSuccessful;
    bool m_mediaFoundationInitalisationSuccessful;

    list<PBYTE, allocator<PBYTE>> *p_videoList;

    void initializeFrameLoader();
    VFLResult setVideoPath(PWSTR filePath, bool reset);
    VFLResult setVideoURL(PWSTR fileURL, bool reset);
    VFLResult openFileInSourceReader();
    HRESULT configureDecoder(IMFSourceReader *pReader);
    VFLResult copyFrameToBuffer(IMFSample *pSample);
    void SafeRelease(IUnknown **ppT);
};