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 }
|