LoadCodecs.h revision baa3858d3f5d128a5c8466b700098109edcad5f2 
1// LoadCodecs.h
2
3#ifndef __LOADCODECS_H
4#define __LOADCODECS_H
5
6#include "../../../Common/Types.h"
7#include "../../../Common/MyCom.h"
8#include "../../../Common/MyString.h"
9#include "../../../Common/Buffer.h"
10#include "../../ICoder.h"
11
12#ifdef EXTERNAL_CODECS
13#include "../../../Windows/DLL.h"
14#endif
15
16struct CDllCodecInfo
17{
18  CLSID Encoder;
19  CLSID Decoder;
20  bool EncoderIsAssigned;
21  bool DecoderIsAssigned;
22  int LibIndex;
23  UInt32 CodecIndex;
24};
25
26#include "../../Archive/IArchive.h"
27
28typedef IInArchive * (*CreateInArchiveP)();
29typedef IOutArchive * (*CreateOutArchiveP)();
30
31struct CArcExtInfo
32{
33  UString Ext;
34  UString AddExt;
35  CArcExtInfo() {}
36  CArcExtInfo(const UString &ext): Ext(ext) {}
37  CArcExtInfo(const UString &ext, const UString &addExt): Ext(ext), AddExt(addExt) {}
38};
39
40
41struct CArcInfoEx
42{
43  #ifdef EXTERNAL_CODECS
44  int LibIndex;
45  UInt32 FormatIndex;
46  CLSID ClassID;
47  #endif
48  bool UpdateEnabled;
49  CreateInArchiveP CreateInArchive;
50  CreateOutArchiveP CreateOutArchive;
51  UString Name;
52  CObjectVector<CArcExtInfo> Exts;
53  #ifndef _SFX
54  CByteBuffer StartSignature;
55  // CByteBuffer FinishSignature;
56  #ifdef NEW_FOLDER_INTERFACE
57  UStringVector AssociateExts;
58  #endif
59  #endif
60  bool KeepName;
61  UString GetMainExt() const
62  {
63    if (Exts.IsEmpty())
64      return UString();
65    return Exts[0].Ext;
66  }
67  int FindExtension(const UString &ext) const
68  {
69    for (int i = 0; i < Exts.Size(); i++)
70      if (ext.CompareNoCase(Exts[i].Ext) == 0)
71        return i;
72    return -1;
73  }
74  UString GetAllExtensions() const
75  {
76    UString s;
77    for (int i = 0; i < Exts.Size(); i++)
78    {
79      if (i > 0)
80        s += ' ';
81      s += Exts[i].Ext;
82    }
83    return s;
84  }
85
86  void AddExts(const wchar_t* ext, const wchar_t* addExt);
87
88  CArcInfoEx():
89    #ifdef EXTERNAL_CODECS
90    LibIndex(-1),
91    #endif
92    UpdateEnabled(false),
93    CreateInArchive(0), CreateOutArchive(0),
94    KeepName(false)
95    #ifndef _SFX
96    #endif
97  {}
98};
99
100#ifdef EXTERNAL_CODECS
101typedef UInt32 (WINAPI *GetMethodPropertyFunc)(UInt32 index, PROPID propID, PROPVARIANT *value);
102typedef UInt32 (WINAPI *CreateObjectFunc)(const GUID *clsID, const GUID *interfaceID, void **outObject);
103
104
105#ifdef NEW_FOLDER_INTERFACE
106struct CCodecIcons
107{
108  struct CIconPair
109  {
110    UString Ext;
111    int IconIndex;
112  };
113  CObjectVector<CIconPair> IconPairs;
114  void LoadIcons(HMODULE m);
115  bool FindIconIndex(const UString &ext, int &iconIndex) const;
116};
117#endif
118
119struct CCodecLib
120#ifdef NEW_FOLDER_INTERFACE
121: public CCodecIcons
122#endif
123{
124  NWindows::NDLL::CLibrary Lib;
125  GetMethodPropertyFunc GetMethodProperty;
126  CreateObjectFunc CreateObject;
127  #ifdef NEW_FOLDER_INTERFACE
128  CSysString Path;
129  void LoadIcons() { CCodecIcons::LoadIcons((HMODULE)Lib); }
130  #endif
131  CCodecLib(): GetMethodProperty(0) {}
132};
133#endif
134
135class CCodecs:
136  #ifdef EXTERNAL_CODECS
137  public ICompressCodecsInfo,
138  #else
139  public IUnknown,
140  #endif
141  public CMyUnknownImp
142{
143public:
144  #ifdef EXTERNAL_CODECS
145  CObjectVector<CCodecLib> Libs;
146  CObjectVector<CDllCodecInfo> Codecs;
147
148  #ifdef NEW_FOLDER_INTERFACE
149  CCodecIcons InternalIcons;
150  #endif
151
152  HRESULT LoadCodecs();
153  HRESULT LoadFormats();
154  HRESULT LoadDll(const CSysString &path, bool needCheckDll);
155  HRESULT LoadDllsFromFolder(const CSysString &folderPrefix);
156
157  HRESULT CreateArchiveHandler(const CArcInfoEx &ai, void **archive, bool outHandler) const
158  {
159    return Libs[ai.LibIndex].CreateObject(&ai.ClassID, outHandler ? &IID_IOutArchive : &IID_IInArchive, (void **)archive);
160  }
161  #endif
162
163public:
164  CObjectVector<CArcInfoEx> Formats;
165  HRESULT Load();
166
167  #ifndef _SFX
168  int FindFormatForArchiveName(const UString &arcPath) const;
169  int FindFormatForExtension(const UString &ext) const;
170  int FindFormatForArchiveType(const UString &arcType) const;
171  bool FindFormatForArchiveType(const UString &arcType, CIntVector &formatIndices) const;
172  #endif
173
174  MY_UNKNOWN_IMP
175
176  #ifdef EXTERNAL_CODECS
177  STDMETHOD(GetNumberOfMethods)(UInt32 *numMethods);
178  STDMETHOD(GetProperty)(UInt32 index, PROPID propID, PROPVARIANT *value);
179  STDMETHOD(CreateDecoder)(UInt32 index, const GUID *interfaceID, void **coder);
180  STDMETHOD(CreateEncoder)(UInt32 index, const GUID *interfaceID, void **coder);
181  #endif
182
183  int GetCodecLibIndex(UInt32 index);
184  bool GetCodecEncoderIsAssigned(UInt32 index);
185  HRESULT GetCodecId(UInt32 index, UInt64 &id);
186  UString GetCodecName(UInt32 index);
187
188  HRESULT CreateInArchive(int formatIndex, CMyComPtr<IInArchive> &archive) const
189  {
190    const CArcInfoEx &ai = Formats[formatIndex];
191    #ifdef EXTERNAL_CODECS
192    if (ai.LibIndex < 0)
193    #endif
194    {
195      archive = ai.CreateInArchive();
196      return S_OK;
197    }
198    #ifdef EXTERNAL_CODECS
199    return CreateArchiveHandler(ai, (void **)&archive, false);
200    #endif
201  }
202  HRESULT CreateOutArchive(int formatIndex, CMyComPtr<IOutArchive> &archive) const
203  {
204    const CArcInfoEx &ai = Formats[formatIndex];
205    #ifdef EXTERNAL_CODECS
206    if (ai.LibIndex < 0)
207    #endif
208    {
209      archive = ai.CreateOutArchive();
210      return S_OK;
211    }
212    #ifdef EXTERNAL_CODECS
213    return CreateArchiveHandler(ai, (void **)&archive, true);
214    #endif
215  }
216  int FindOutFormatFromName(const UString &name) const
217  {
218    for (int i = 0; i < Formats.Size(); i++)
219    {
220      const CArcInfoEx &arc = Formats[i];
221      if (!arc.UpdateEnabled)
222        continue;
223      if (arc.Name.CompareNoCase(name) == 0)
224        return i;
225    }
226    return -1;
227  }
228
229  #ifdef EXTERNAL_CODECS
230  HRESULT CreateCoder(const UString &name, bool encode, CMyComPtr<ICompressCoder> &coder) const;
231  #endif
232
233};
234
235#endif
236