gecko/modules/lib7z/LZMASDK/CPP/7zip/Compress/CodecExports.cpp

161 lines
4.5 KiB
C++
Raw Normal View History

// CodecExports.cpp
#include "StdAfx.h"
#include "../../Common/ComTry.h"
#include "../../Windows/PropVariant.h"
#include "../ICoder.h"
#include "../Common/RegisterCodec.h"
extern unsigned int g_NumCodecs;
extern const CCodecInfo *g_Codecs[];
static const UInt16 kDecodeId = 0x2790;
DEFINE_GUID(CLSID_CCodec,
0x23170F69, 0x40C1, kDecodeId, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00);
static inline HRESULT SetPropString(const char *s, unsigned int size, PROPVARIANT *value)
{
if ((value->bstrVal = ::SysAllocStringByteLen(s, size)) != 0)
value->vt = VT_BSTR;
return S_OK;
}
static inline HRESULT SetPropGUID(const GUID &guid, PROPVARIANT *value)
{
return SetPropString((const char *)&guid, sizeof(GUID), value);
}
static HRESULT SetClassID(CMethodId id, bool encode, PROPVARIANT *value)
{
GUID clsId = CLSID_CCodec;
for (int i = 0; i < sizeof(id); i++, id >>= 8)
clsId.Data4[i] = (Byte)(id & 0xFF);
if (encode)
clsId.Data3++;
return SetPropGUID(clsId, value);
}
static HRESULT FindCodecClassId(const GUID *clsID, UInt32 isCoder2, bool isFilter, bool &encode, int &index)
{
index = -1;
if (clsID->Data1 != CLSID_CCodec.Data1 ||
clsID->Data2 != CLSID_CCodec.Data2 ||
(clsID->Data3 & ~1) != kDecodeId)
return S_OK;
encode = (clsID->Data3 != kDecodeId);
UInt64 id = 0;
for (int j = 0; j < 8; j++)
id |= ((UInt64)clsID->Data4[j]) << (8 * j);
for (unsigned i = 0; i < g_NumCodecs; i++)
{
const CCodecInfo &codec = *g_Codecs[i];
if (id != codec.Id || encode && !codec.CreateEncoder || !encode && !codec.CreateDecoder)
continue;
if (!isFilter && codec.IsFilter || isFilter && !codec.IsFilter ||
codec.NumInStreams != 1 && !isCoder2 || codec.NumInStreams == 1 && isCoder2)
return E_NOINTERFACE;
index = i;
return S_OK;
}
return S_OK;
}
STDAPI CreateCoder2(bool encode, UInt32 index, const GUID *iid, void **outObject)
{
COM_TRY_BEGIN
*outObject = 0;
bool isCoder = (*iid == IID_ICompressCoder) != 0;
bool isCoder2 = (*iid == IID_ICompressCoder2) != 0;
bool isFilter = (*iid == IID_ICompressFilter) != 0;
const CCodecInfo &codec = *g_Codecs[index];
if (!isFilter && codec.IsFilter || isFilter && !codec.IsFilter ||
codec.NumInStreams != 1 && !isCoder2 || codec.NumInStreams == 1 && isCoder2)
return E_NOINTERFACE;
if (encode)
{
if (!codec.CreateEncoder)
return CLASS_E_CLASSNOTAVAILABLE;
*outObject = codec.CreateEncoder();
}
else
{
if (!codec.CreateDecoder)
return CLASS_E_CLASSNOTAVAILABLE;
*outObject = codec.CreateDecoder();
}
if (isCoder)
((ICompressCoder *)*outObject)->AddRef();
else if (isCoder2)
((ICompressCoder2 *)*outObject)->AddRef();
else
((ICompressFilter *)*outObject)->AddRef();
return S_OK;
COM_TRY_END
}
STDAPI CreateCoder(const GUID *clsid, const GUID *iid, void **outObject)
{
*outObject = 0;
bool isCoder = (*iid == IID_ICompressCoder) != 0;
bool isCoder2 = (*iid == IID_ICompressCoder2) != 0;
bool isFilter = (*iid == IID_ICompressFilter) != 0;
if (!isCoder && !isCoder2 && !isFilter)
return E_NOINTERFACE;
bool encode;
int codecIndex;
HRESULT res = FindCodecClassId(clsid, isCoder2, isFilter, encode, codecIndex);
if (res != S_OK)
return res;
if (codecIndex < 0)
return CLASS_E_CLASSNOTAVAILABLE;
return CreateCoder2(encode, codecIndex, iid, outObject);
}
STDAPI GetMethodProperty(UInt32 codecIndex, PROPID propID, PROPVARIANT *value)
{
::VariantClear((VARIANTARG *)value);
const CCodecInfo &codec = *g_Codecs[codecIndex];
switch(propID)
{
case NMethodPropID::kID:
{
value->uhVal.QuadPart = (UInt64)codec.Id;
value->vt = VT_UI8;
break;
}
case NMethodPropID::kName:
if ((value->bstrVal = ::SysAllocString(codec.Name)) != 0)
value->vt = VT_BSTR;
break;
case NMethodPropID::kDecoder:
if (codec.CreateDecoder)
return SetClassID(codec.Id, false, value);
break;
case NMethodPropID::kEncoder:
if (codec.CreateEncoder)
return SetClassID(codec.Id, true, value);
break;
case NMethodPropID::kInStreams:
{
if (codec.NumInStreams != 1)
{
value->vt = VT_UI4;
value->ulVal = (ULONG)codec.NumInStreams;
}
break;
}
}
return S_OK;
}
STDAPI GetNumberOfMethods(UINT32 *numCodecs)
{
*numCodecs = g_NumCodecs;
return S_OK;
}