Mercurial > vba-linux

annotate src/win32/7zip/7z/CPP/7zip/Compress/CodecExports.cpp @ 1:f9f4f1b99eed

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
importing src directory
author Robert McIntyre <rlm@mit.edu>
date Sat, 03 Mar 2012 10:31:27 -0600
parents
children
rev   line source
rlm@1 1 // CodecExports.cpp
rlm@1 2
rlm@1 3 #include "StdAfx.h"
rlm@1 4
rlm@1 5 #include "../../Common/ComTry.h"
rlm@1 6 #include "../../Windows/PropVariant.h"
rlm@1 7 #include "../Common/RegisterCodec.h"
rlm@1 8 #include "../ICoder.h"
rlm@1 9
rlm@1 10 extern unsigned int g_NumCodecs;
rlm@1 11 extern const CCodecInfo *g_Codecs[];
rlm@1 12
rlm@1 13 static const UInt16 kDecodeId = 0x2790;
rlm@1 14
rlm@1 15 DEFINE_GUID(CLSID_CCodec,
rlm@1 16 0x23170F69, 0x40C1, kDecodeId, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00);
rlm@1 17
rlm@1 18 static inline HRESULT SetPropString(const char *s, unsigned int size, PROPVARIANT *value)
rlm@1 19 {
rlm@1 20 if ((value->bstrVal = ::SysAllocStringByteLen(s, size)) != 0)
rlm@1 21 value->vt = VT_BSTR;
rlm@1 22 return S_OK;
rlm@1 23 }
rlm@1 24
rlm@1 25 static inline HRESULT SetPropGUID(const GUID &guid, PROPVARIANT *value)
rlm@1 26 {
rlm@1 27 return SetPropString((const char *)&guid, sizeof(GUID), value);
rlm@1 28 }
rlm@1 29
rlm@1 30 static HRESULT SetClassID(CMethodId id, bool encode, PROPVARIANT *value)
rlm@1 31 {
rlm@1 32 GUID clsId = CLSID_CCodec;
rlm@1 33 for (int i = 0; i < sizeof(id); i++, id >>= 8)
rlm@1 34 clsId.Data4[i] = (Byte)(id & 0xFF);
rlm@1 35 if (encode)
rlm@1 36 clsId.Data3++;
rlm@1 37 return SetPropGUID(clsId, value);
rlm@1 38 }
rlm@1 39
rlm@1 40 static HRESULT FindCodecClassId(const GUID *clsID, UInt32 isCoder2, bool isFilter, bool &encode, int &index)
rlm@1 41 {
rlm@1 42 index = -1;
rlm@1 43 if (clsID->Data1 != CLSID_CCodec.Data1 ||
rlm@1 44 clsID->Data2 != CLSID_CCodec.Data2 ||
rlm@1 45 (clsID->Data3 & ~1) != kDecodeId)
rlm@1 46 return S_OK;
rlm@1 47 encode = (clsID->Data3 != kDecodeId);
rlm@1 48 UInt64 id = 0;
rlm@1 49 for (int j = 0; j < 8; j++)
rlm@1 50 id |= ((UInt64)clsID->Data4[j]) << (8 * j);
rlm@1 51 for (unsigned i = 0; i < g_NumCodecs; i++)
rlm@1 52 {
rlm@1 53 const CCodecInfo &codec = *g_Codecs[i];
rlm@1 54 if (id != codec.Id || encode && !codec.CreateEncoder || !encode && !codec.CreateDecoder)
rlm@1 55 continue;
rlm@1 56 if (!isFilter && codec.IsFilter || isFilter && !codec.IsFilter ||
rlm@1 57 codec.NumInStreams != 1 && !isCoder2 || codec.NumInStreams == 1 && isCoder2)
rlm@1 58 return E_NOINTERFACE;
rlm@1 59 index = i;
rlm@1 60 return S_OK;
rlm@1 61 }
rlm@1 62 return S_OK;
rlm@1 63 }
rlm@1 64
rlm@1 65 STDAPI CreateCoder2(bool encode, UInt32 index, const GUID *iid, void **outObject)
rlm@1 66 {
rlm@1 67 COM_TRY_BEGIN
rlm@1 68 *outObject = 0;
rlm@1 69 bool isCoder = (*iid == IID_ICompressCoder) != 0;
rlm@1 70 bool isCoder2 = (*iid == IID_ICompressCoder2) != 0;
rlm@1 71 bool isFilter = (*iid == IID_ICompressFilter) != 0;
rlm@1 72 const CCodecInfo &codec = *g_Codecs[index];
rlm@1 73 if (!isFilter && codec.IsFilter || isFilter && !codec.IsFilter ||
rlm@1 74 codec.NumInStreams != 1 && !isCoder2 || codec.NumInStreams == 1 && isCoder2)
rlm@1 75 return E_NOINTERFACE;
rlm@1 76 if (encode)
rlm@1 77 {
rlm@1 78 if (!codec.CreateEncoder)
rlm@1 79 return CLASS_E_CLASSNOTAVAILABLE;
rlm@1 80 *outObject = codec.CreateEncoder();
rlm@1 81 }
rlm@1 82 else
rlm@1 83 {
rlm@1 84 if (!codec.CreateDecoder)
rlm@1 85 return CLASS_E_CLASSNOTAVAILABLE;
rlm@1 86 *outObject = codec.CreateDecoder();
rlm@1 87 }
rlm@1 88 if (isCoder)
rlm@1 89 ((ICompressCoder *)*outObject)->AddRef();
rlm@1 90 else if (isCoder2)
rlm@1 91 ((ICompressCoder2 *)*outObject)->AddRef();
rlm@1 92 else
rlm@1 93 ((ICompressFilter *)*outObject)->AddRef();
rlm@1 94 return S_OK;
rlm@1 95 COM_TRY_END
rlm@1 96 }
rlm@1 97
rlm@1 98 STDAPI CreateCoder(const GUID *clsid, const GUID *iid, void **outObject)
rlm@1 99 {
rlm@1 100 *outObject = 0;
rlm@1 101 bool isCoder = (*iid == IID_ICompressCoder) != 0;
rlm@1 102 bool isCoder2 = (*iid == IID_ICompressCoder2) != 0;
rlm@1 103 bool isFilter = (*iid == IID_ICompressFilter) != 0;
rlm@1 104 if (!isCoder && !isCoder2 && !isFilter)
rlm@1 105 return E_NOINTERFACE;
rlm@1 106 bool encode;
rlm@1 107 int codecIndex;
rlm@1 108 HRESULT res = FindCodecClassId(clsid, isCoder2, isFilter, encode, codecIndex);
rlm@1 109 if (res != S_OK)
rlm@1 110 return res;
rlm@1 111 if (codecIndex < 0)
rlm@1 112 return CLASS_E_CLASSNOTAVAILABLE;
rlm@1 113 return CreateCoder2(encode, codecIndex, iid, outObject);
rlm@1 114 }
rlm@1 115
rlm@1 116 STDAPI GetMethodProperty(UInt32 codecIndex, PROPID propID, PROPVARIANT *value)
rlm@1 117 {
rlm@1 118 ::VariantClear((VARIANTARG *)value);
rlm@1 119 const CCodecInfo &codec = *g_Codecs[codecIndex];
rlm@1 120 switch(propID)
rlm@1 121 {
rlm@1 122 case NMethodPropID::kID:
rlm@1 123 {
rlm@1 124 value->uhVal.QuadPart = (UInt64)codec.Id;
rlm@1 125 value->vt = VT_UI8;
rlm@1 126 break;
rlm@1 127 }
rlm@1 128 case NMethodPropID::kName:
rlm@1 129 if ((value->bstrVal = ::SysAllocString(codec.Name)) != 0)
rlm@1 130 value->vt = VT_BSTR;
rlm@1 131 break;
rlm@1 132 case NMethodPropID::kDecoder:
rlm@1 133 if (codec.CreateDecoder)
rlm@1 134 return SetClassID(codec.Id, false, value);
rlm@1 135 break;
rlm@1 136 case NMethodPropID::kEncoder:
rlm@1 137 if (codec.CreateEncoder)
rlm@1 138 return SetClassID(codec.Id, true, value);
rlm@1 139 break;
rlm@1 140 case NMethodPropID::kInStreams:
rlm@1 141 {
rlm@1 142 if (codec.NumInStreams != 1)
rlm@1 143 {
rlm@1 144 value->vt = VT_UI4;
rlm@1 145 value->ulVal = (ULONG)codec.NumInStreams;
rlm@1 146 }
rlm@1 147 break;
rlm@1 148 }
rlm@1 149 }
rlm@1 150 return S_OK;
rlm@1 151 }
rlm@1 152
rlm@1 153 STDAPI GetNumberOfMethods(UINT32 *numCodecs)
rlm@1 154 {
rlm@1 155 *numCodecs = g_NumCodecs;
rlm@1 156 return S_OK;
rlm@1 157 }