CImage 从内存中读取图像

CImage 的CImage::Load( IStream* pStream) 从内存中读取图像时，需要提供实现了IStream的对象。一般都是采用CreateStreamOnHGlobal创建IStream对象，但这需要重新分配内存，再将内存中图像复制到新分配的内存中，完了还要释放，多了很多操作，也影响性能。
下面这个类就是实现了IStream，可以实现从内存直接读取图像，省了上述多余的操作。

Stream.h文件

#pragma once
#include <windows.h> 
class CStream : public IStream
{
public:
    // 
    // IUnknown members 
    // 
    HRESULT __stdcall QueryInterface(REFIID iid, void ** ppvObject);
    ULONG   __stdcall AddRef(void);
    ULONG   __stdcall Release(void);

    // 
    // ISequentialStream members 
    // 
    HRESULT __stdcall Read(void *pv, ULONG cb, ULONG *pcbRead);
    HRESULT __stdcall Write(const void *pv, ULONG cb, ULONG *pcbWritten);


    // 
    // IStream members 
    // 
    HRESULT __stdcall Seek(LARGE_INTEGER  dlibMove, DWORD dwOrigin, ULARGE_INTEGER *plibNewPosition);
    HRESULT __stdcall SetSize(ULARGE_INTEGER libNewSize);
    HRESULT __stdcall CopyTo(IStream *pstm, ULARGE_INTEGER cb, ULARGE_INTEGER *pcbRead, ULARGE_INTEGER *pcbWritten);
    HRESULT __stdcall Commit(DWORD grfCommitFlags);
    HRESULT __stdcall Revert(void);
    HRESULT __stdcall LockRegion(ULARGE_INTEGER libOffset, ULARGE_INTEGER cb, DWORD dwLockType);
    HRESULT __stdcall UnlockRegion(ULARGE_INTEGER libOffset, ULARGE_INTEGER cb, DWORD dwLockType);
    HRESULT __stdcall Stat(STATSTG *pstatstg, DWORD grfStatFlag);
    HRESULT __stdcall Clone(IStream **ppstm);


    // 
    // Constructor / Destructor 
    // 
    CStream(BYTE *pData, ULONG nSize);
    ~CStream();

private:
    // 
    // private members and functions 
    // 
    LONG            m_lRefCount;


    STATSTG         m_statstg;          // each IStream needs one of these 
    ULARGE_INTEGER  m_nOffset;          // offset within the stream 
    ULARGE_INTEGER  m_nLength;          // length of the stream 
    BYTE*           m_pData;            // stream data from 
};

Stream.cpp文件

// 
//  IStream.cpp 
//  Implementation of the IStream COM interface 
// 
#include "stdafx.h"
#include "Stream.h"

#define DEFNAME L"Memory IStream" 

// 
//  Constructor 
// 
CStream::CStream(BYTE *pData, ULONG nSize)
{
    m_lRefCount = 1;
    m_pData = pData;

    // stream metrics 
    m_nOffset.QuadPart = 0;
    m_nLength.QuadPart = nSize;//len; 


                           // stream status 
    m_statstg.type = STGTY_STREAM;      // IStream object 
    m_statstg.cbSize.QuadPart = 0;//len;                // Set to the length of our stream object 
    m_statstg.grfLocksSupported = 0;                // Region locking not supported 
    m_statstg.grfMode = 0;              // access mode 
    m_statstg.clsid = CLSID_NULL;       // not used for IStreams 
    m_statstg.grfStateBits = 0;             // not used for IStreams 
    m_statstg.reserved = 0;             // reserved for 


    CoFileTimeNow(&m_statstg.ctime);                // creation time 
    CoFileTimeNow(&m_statstg.atime);                // last access time 
    CoFileTimeNow(&m_statstg.mtime);                // last modify time 
}


// 
//  Destructor 
// 
CStream::~CStream()
{
}


// 
//  IUnknown::AddRef 
// 
ULONG __stdcall CStream::AddRef(void)
{
    // increment object reference count 
    return InterlockedIncrement(&m_lRefCount);
}


// 
//  IUnknown::Release 
// 
ULONG __stdcall CStream::Release(void)
{
    // decrement object reference count 
    LONG count = InterlockedDecrement(&m_lRefCount);

    if (count == 0)
    {
        delete this;
        return 0;
    }
    else
    {
        return count;
    }
}


// 
//  IUnknown::QueryInterface 
// 
HRESULT __stdcall CStream::QueryInterface(REFIID iid, void **ppvObject)
{
    // check to see what interface has been requested 
    if (iid == IID_IStream || iid == IID_IUnknown || iid == IID_ISequentialStream)
    {
        AddRef();
        *ppvObject = this;
        return S_OK;
    }
    else
    {
        *ppvObject = 0;
        return E_NOINTERFACE;
    }
}


// 
//  ISequentialStream::Read 
// 
HRESULT __stdcall CStream::Read(void *pv, ULONG cb, ULONG *pcbRead)
{
    ULONG available;


    if (pv == 0)
        return STG_E_INVALIDPOINTER;

    available = min(cb, (ULONG)(m_nLength.QuadPart - m_nOffset.QuadPart));

    memcpy(pv, m_pData + m_nOffset.QuadPart, available);

    m_nOffset.QuadPart += available;

    if (pcbRead)
        *pcbRead = available;

    return S_OK;
}


// 
//  ISequentialStream::Write 
// 
HRESULT __stdcall CStream::Write(const void *pv, ULONG cb, ULONG *pcbWritten)
{
    if (pv == 0)
        return STG_E_INVALIDPOINTER;


    return E_NOTIMPL;
}


// 
//  IStream::Seek 
// 
HRESULT __stdcall CStream::Seek(LARGE_INTEGER dlibMove, DWORD dwOrigin, ULARGE_INTEGER *plibNewPosition)
{
    switch (dwOrigin)
    {
    case STREAM_SEEK_SET:   m_nOffset.QuadPart = dlibMove.QuadPart;                      break;
    case STREAM_SEEK_CUR:   m_nOffset.QuadPart = m_nOffset.QuadPart + dlibMove.QuadPart; break;
    case STREAM_SEEK_END:   m_nOffset.QuadPart = m_nLength.QuadPart - dlibMove.QuadPart; break;
    }

    if (plibNewPosition)
        *plibNewPosition = m_nOffset;


    return S_OK;
}


// 
//  IStream::SetSize 
// 
HRESULT __stdcall CStream::SetSize(ULARGE_INTEGER libNewSize)
{
    return S_OK;
}


// 
//  IStream::CopyTo 
// 
HRESULT __stdcall CStream::CopyTo(IStream *pstm, ULARGE_INTEGER cb, ULARGE_INTEGER *pcbRead, ULARGE_INTEGER *pcbWritten)
{
    DWORD len, written;
    len = (ULONG)min(cb.QuadPart, m_nLength.QuadPart);
    pstm->Write(m_pData, len, &written);

    if (pcbRead)
        pcbRead->QuadPart = len;
    if (pcbWritten)
        pcbWritten->QuadPart = written;
    return S_OK;
}


// 
//  IStream::Commit 
// 
HRESULT __stdcall CStream::Commit(DWORD grfCommitFlags)
{
    // Transacted mode is not supported 
    return S_OK;
}


// 
//  IStream::Revert 
// 
HRESULT __stdcall CStream::Revert()
{
    // Transacted mode is not supported 
    return S_OK;
}


// 
//  IStream::LockRegion 
// 
HRESULT __stdcall CStream::LockRegion(ULARGE_INTEGER libOffset, ULARGE_INTEGER cb, DWORD dwLockType)
{
    // locking is not supported 
    return STG_E_INVALIDFUNCTION;
}


// 
//  IStream::UnlockRegion 
// 
HRESULT __stdcall CStream::UnlockRegion(ULARGE_INTEGER libOffset, ULARGE_INTEGER cb, DWORD dwLockType)
{
    // locking is not supported 
    return STG_E_INVALIDFUNCTION;
}


// 
//  IStream::Stat 
// 
HRESULT __stdcall CStream::Stat(STATSTG *pstatstg, DWORD grfStatFlag)
{
    if (pstatstg == 0)
        return STG_E_INVALIDPOINTER;


    // return our STATSTG to the caller 
    m_statstg.cbSize.QuadPart = m_nLength.QuadPart;
    *pstatstg = m_statstg;


    switch (grfStatFlag)
    {
    case STATFLAG_DEFAULT:
        // allocate a new buffer for the name 
        if ((pstatstg->pwcsName = (WCHAR *)CoTaskMemAlloc(sizeof(DEFNAME))) == 0)
            return STG_E_INSUFFICIENTMEMORY;


        lstrcpyW(pstatstg->pwcsName, DEFNAME);
        break;


    case STATFLAG_NONAME:
        pstatstg->pwcsName = 0;
        break;


    default:
        return STG_E_INVALIDFLAG;
    }


    return S_OK;
}


// 
//  IStream::Clone 
// 
HRESULT __stdcall CStream::Clone(IStream **ppstm)
{
    return E_NOTIMPL;
}

使用就很简单了：
BYTE* pData ; // 图像数据
DWORD nSize; // 图像数据的长度
…

CStream stream(pData, nSize);
CImage img;
img.Load(&stream);