M4Image.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
#pragma once
 
#include "vdecmpeg4.h"
#include "M4Decoder.h"
 
namespace vdecmpeg4
{
 
#define EDGE_SIZE 32
 
#define EDGE_SIZE2  (EDGE_SIZE/2)
 
#define ALIGNMENT_TRASH (64)
 
 
class M4Image
{
public:
    VIDImage    mImage;
 
    VIDImageInfo mImageInfo;
 
private:
    M4Image()
    {
    }
 
public:
    //
    //--- Public Interface
    //
 
    static M4Image* create(M4Decoder* decoder, int16 width, int16 height, bool poolMem = true);
 
    static void destroy(M4Image*& pImage, M4Decoder* decoder);
 
    void Black();
 
    void RefAdd()
    {
        ++mRefCount;
    }
 
    void RefRemove()
    {
        if (--mRefCount == 0)
        {
            // Actually, we need to get a copy of the member variable, because the destructor call
            // in the next line frees the instance possibily overwriting the mDecoder member.
            M4Decoder* pDecoder = mDecoder;
            this->~M4Image();
            pDecoder->mMemSys.free(this);
        }
    }
 
    int32 RefGet() const
    {
        return mRefCount;
    }
 
    M4Decoder& getDecoder() const
    {
        return *mDecoder;
    }
 
    static void poolMemCb(void* memPtr, uint32 size, uint32 poolFlags, void* userData);
 
    void* operator new(size_t sz, M4MemHandler& memSys)
    {
        void* pAddr = memSys.malloc(sz, 16);
        if (pAddr)
        {
            FMemory::Memzero(pAddr, sz);
        }
        return pAddr;
    }
 
    void operator delete(void* ptr)
    {
        ((M4Image*)ptr)->mDecoder->mMemSys.free(ptr);
    }
 
private:
    uint8*          mBaseMem;
 
    M4Decoder*      mDecoder;
 
    volatile int32 mRefCount;
 
 
    ~M4Image();
 
    void setup(uint8* ptr, uint32 flags);
 
    M4Image(const M4Image& pObj);
 
    const M4Image &operator=(const M4Image& pObj);
 
    M4_MEMORY_HANDLER
 
    friend class M4Decoder;
    friend struct VIDImage;
 
 
#ifdef _M4_ENABLE_BMP_OUT
    void saveBMP(const char* pBaseName, uint32 frameCounter);
 
private:
    void getRGB(uint8* dst, int32 dstStride);
 
    void initImageSave();
 
    static int32 mTabRgbY[256];
    static int32 mTabBU[256];
    static int32 mTabGU[256];
    static int32 mTabGV[256];
    static int32 mTabRV[256];
    static bool mInitImageSave;
#endif
};
 
/******************************************************************************
 *  INLINE FUNCTIONS
 ******************************************************************************/
 
inline void swapM4Image(M4Image*& img1, M4Image*& img2)
{
    M4Image* tmp = img1;
    img1 = img2;
    img2 = tmp;
}
 
}