1 // CodecExports.cpp 2 3 #include "StdAfx.h" 4 5 #include "../../../C/CpuArch.h" 6 7 #include "../../Common/ComTry.h" 8 #include "../../Common/MyCom.h" 9 10 #include "../../Windows/PropVariant.h" 11 12 #include "../ICoder.h" 13 14 #include "../Common/RegisterCodec.h" 15 16 extern unsigned int g_NumCodecs; 17 extern const CCodecInfo *g_Codecs[]; 18 19 extern unsigned int g_NumHashers; 20 extern const CHasherInfo *g_Hashers[]; 21 22 static const UInt16 kDecodeId = 0x2790; 23 static const UInt16 kEncodeId = 0x2791; 24 static const UInt16 kHasherId = 0x2792; 25 26 DEFINE_GUID(CLSID_CCodec, 27 0x23170F69, 0x40C1, kDecodeId, 0, 0, 0, 0, 0, 0, 0, 0); 28 29 static inline HRESULT SetPropString(const char *s, unsigned int size, PROPVARIANT *value) 30 { 31 if ((value->bstrVal = ::SysAllocStringByteLen(s, size)) != 0) 32 value->vt = VT_BSTR; 33 return S_OK; 34 } 35 36 static inline HRESULT SetPropGUID(const GUID &guid, PROPVARIANT *value) 37 { 38 return SetPropString((const char *)&guid, sizeof(GUID), value); 39 } 40 41 static HRESULT SetClassID(CMethodId id, UInt16 typeId, PROPVARIANT *value) 42 { 43 GUID clsId; 44 clsId.Data1 = CLSID_CCodec.Data1; 45 clsId.Data2 = CLSID_CCodec.Data2; 46 clsId.Data3 = typeId; 47 SetUi64(clsId.Data4, id); 48 return SetPropGUID(clsId, value); 49 } 50 51 static HRESULT FindCodecClassId(const GUID *clsID, UInt32 isCoder2, bool isFilter, bool &encode, int &index) 52 { 53 index = -1; 54 if (clsID->Data1 != CLSID_CCodec.Data1 || 55 clsID->Data2 != CLSID_CCodec.Data2) 56 return S_OK; 57 encode = true; 58 if (clsID->Data3 == kDecodeId) 59 encode = false; 60 else if (clsID->Data3 != kEncodeId) 61 return S_OK; 62 UInt64 id = GetUi64(clsID->Data4); 63 for (unsigned i = 0; i < g_NumCodecs; i++) 64 { 65 const CCodecInfo &codec = *g_Codecs[i]; 66 if (id != codec.Id || encode && !codec.CreateEncoder || !encode && !codec.CreateDecoder) 67 continue; 68 if (!isFilter && codec.IsFilter || isFilter && !codec.IsFilter || 69 codec.NumInStreams != 1 && !isCoder2 || codec.NumInStreams == 1 && isCoder2) 70 return E_NOINTERFACE; 71 index = i; 72 return S_OK; 73 } 74 return S_OK; 75 } 76 77 STDAPI CreateCoder2(bool encode, int index, const GUID *iid, void **outObject) 78 { 79 COM_TRY_BEGIN 80 *outObject = 0; 81 bool isCoder = (*iid == IID_ICompressCoder) != 0; 82 bool isCoder2 = (*iid == IID_ICompressCoder2) != 0; 83 bool isFilter = (*iid == IID_ICompressFilter) != 0; 84 const CCodecInfo &codec = *g_Codecs[index]; 85 if (!isFilter && codec.IsFilter || isFilter && !codec.IsFilter || 86 codec.NumInStreams != 1 && !isCoder2 || codec.NumInStreams == 1 && isCoder2) 87 return E_NOINTERFACE; 88 if (encode) 89 { 90 if (!codec.CreateEncoder) 91 return CLASS_E_CLASSNOTAVAILABLE; 92 *outObject = codec.CreateEncoder(); 93 } 94 else 95 { 96 if (!codec.CreateDecoder) 97 return CLASS_E_CLASSNOTAVAILABLE; 98 *outObject = codec.CreateDecoder(); 99 } 100 if (*outObject) 101 { 102 if (isCoder) 103 ((ICompressCoder *)*outObject)->AddRef(); 104 else if (isCoder2) 105 ((ICompressCoder2 *)*outObject)->AddRef(); 106 else 107 ((ICompressFilter *)*outObject)->AddRef(); 108 } 109 return S_OK; 110 COM_TRY_END 111 } 112 113 STDAPI CreateCoder(const GUID *clsid, const GUID *iid, void **outObject) 114 { 115 COM_TRY_BEGIN 116 *outObject = 0; 117 bool isCoder = (*iid == IID_ICompressCoder) != 0; 118 bool isCoder2 = (*iid == IID_ICompressCoder2) != 0; 119 bool isFilter = (*iid == IID_ICompressFilter) != 0; 120 if (!isCoder && !isCoder2 && !isFilter) 121 return E_NOINTERFACE; 122 bool encode; 123 int codecIndex; 124 HRESULT res = FindCodecClassId(clsid, isCoder2, isFilter, encode, codecIndex); 125 if (res != S_OK) 126 return res; 127 if (codecIndex < 0) 128 return CLASS_E_CLASSNOTAVAILABLE; 129 130 const CCodecInfo &codec = *g_Codecs[codecIndex]; 131 if (encode) 132 *outObject = codec.CreateEncoder(); 133 else 134 *outObject = codec.CreateDecoder(); 135 if (*outObject) 136 { 137 if (isCoder) 138 ((ICompressCoder *)*outObject)->AddRef(); 139 else if (isCoder2) 140 ((ICompressCoder2 *)*outObject)->AddRef(); 141 else 142 ((ICompressFilter *)*outObject)->AddRef(); 143 } 144 return S_OK; 145 COM_TRY_END 146 } 147 148 STDAPI GetMethodProperty(UInt32 codecIndex, PROPID propID, PROPVARIANT *value) 149 { 150 ::VariantClear((VARIANTARG *)value); 151 const CCodecInfo &codec = *g_Codecs[codecIndex]; 152 switch (propID) 153 { 154 case NMethodPropID::kID: 155 value->uhVal.QuadPart = (UInt64)codec.Id; 156 value->vt = VT_UI8; 157 break; 158 case NMethodPropID::kName: 159 if ((value->bstrVal = ::SysAllocString(codec.Name)) != 0) 160 value->vt = VT_BSTR; 161 break; 162 case NMethodPropID::kDecoder: 163 if (codec.CreateDecoder) 164 return SetClassID(codec.Id, kDecodeId, value); 165 break; 166 case NMethodPropID::kEncoder: 167 if (codec.CreateEncoder) 168 return SetClassID(codec.Id, kEncodeId, value); 169 break; 170 case NMethodPropID::kInStreams: 171 if (codec.NumInStreams != 1) 172 { 173 value->vt = VT_UI4; 174 value->ulVal = (ULONG)codec.NumInStreams; 175 } 176 break; 177 } 178 return S_OK; 179 } 180 181 STDAPI GetNumberOfMethods(UINT32 *numCodecs) 182 { 183 *numCodecs = g_NumCodecs; 184 return S_OK; 185 } 186 187 188 static int FindHasherClassId(const GUID *clsID) 189 { 190 if (clsID->Data1 != CLSID_CCodec.Data1 || 191 clsID->Data2 != CLSID_CCodec.Data2 || 192 clsID->Data3 != kHasherId) 193 return -1; 194 UInt64 id = GetUi64(clsID->Data4); 195 for (unsigned i = 0; i < g_NumCodecs; i++) 196 if (id == g_Hashers[i]->Id) 197 return i; 198 return -1; 199 } 200 201 static HRESULT CreateHasher2(UInt32 index, IHasher **hasher) 202 { 203 COM_TRY_BEGIN 204 *hasher = g_Hashers[index]->CreateHasher(); 205 if (*hasher) 206 (*hasher)->AddRef(); 207 return S_OK; 208 COM_TRY_END 209 } 210 211 STDAPI CreateHasher(const GUID *clsid, IHasher **outObject) 212 { 213 COM_TRY_BEGIN 214 *outObject = 0; 215 int index = FindHasherClassId(clsid); 216 if (index < 0) 217 return CLASS_E_CLASSNOTAVAILABLE; 218 return CreateHasher2(index, outObject); 219 COM_TRY_END 220 } 221 222 STDAPI GetHasherProp(UInt32 codecIndex, PROPID propID, PROPVARIANT *value) 223 { 224 ::VariantClear((VARIANTARG *)value); 225 const CHasherInfo &codec = *g_Hashers[codecIndex]; 226 switch (propID) 227 { 228 case NMethodPropID::kID: 229 value->uhVal.QuadPart = (UInt64)codec.Id; 230 value->vt = VT_UI8; 231 break; 232 case NMethodPropID::kName: 233 if ((value->bstrVal = ::SysAllocString(codec.Name)) != 0) 234 value->vt = VT_BSTR; 235 break; 236 case NMethodPropID::kEncoder: 237 if (codec.CreateHasher) 238 return SetClassID(codec.Id, kHasherId, value); 239 break; 240 case NMethodPropID::kDigestSize: 241 value->ulVal = (ULONG)codec.DigestSize; 242 value->vt = VT_UI4; 243 break; 244 } 245 return S_OK; 246 } 247 248 class CHashers: 249 public IHashers, 250 public CMyUnknownImp 251 { 252 public: 253 MY_UNKNOWN_IMP1(IHashers) 254 255 STDMETHOD_(UInt32, GetNumHashers)(); 256 STDMETHOD(GetHasherProp)(UInt32 index, PROPID propID, PROPVARIANT *value); 257 STDMETHOD(CreateHasher)(UInt32 index, IHasher **hasher); 258 }; 259 260 STDAPI GetHashers(IHashers **hashers) 261 { 262 COM_TRY_BEGIN 263 *hashers = new CHashers; 264 if (*hashers) 265 (*hashers)->AddRef(); 266 return S_OK; 267 COM_TRY_END 268 } 269 270 STDMETHODIMP_(UInt32) CHashers::GetNumHashers() 271 { 272 return g_NumHashers; 273 } 274 275 STDMETHODIMP CHashers::GetHasherProp(UInt32 index, PROPID propID, PROPVARIANT *value) 276 { 277 return ::GetHasherProp(index, propID, value); 278 } 279 280 STDMETHODIMP CHashers::CreateHasher(UInt32 index, IHasher **hasher) 281 { 282 return ::CreateHasher2(index, hasher); 283 } 284
