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 

    16 struct 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 

    28 typedef IInArchive * (*CreateInArchiveP)();

    29 typedef IOutArchive * (*CreateOutArchiveP)();

    30 

    31 struct 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 

    41 struct 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

   101 typedef UInt32 (WINAPI *GetMethodPropertyFunc)(UInt32 index, PROPID propID, PROPVARIANT *value);

   102 typedef UInt32 (WINAPI *CreateObjectFunc)(const GUID *clsID, const GUID *interfaceID, void **outObject);

   103 

   104 

   105 struct CCodecLib

   106 {

   107   NWindows::NDLL::CLibrary Lib;

   108   GetMethodPropertyFunc GetMethodProperty;

   109   CreateObjectFunc CreateObject;

   110   #ifdef NEW_FOLDER_INTERFACE

   111   struct CIconPair

   112   {

   113     UString Ext;

   114     UInt32 IconIndex;

   115   };

   116   CSysString Path;

   117   CObjectVector<CIconPair> IconPairs;

   118   void LoadIcons();

   119   int FindIconIndex(const UString &ext) const;

   120   #endif

   121   CCodecLib(): GetMethodProperty(0) {}

   122 };

   123 #endif

   124 

   125 class CCodecs:

   126   #ifdef EXTERNAL_CODECS

   127   public ICompressCodecsInfo,

   128   #else

   129   public IUnknown,

   130   #endif

   131   public CMyUnknownImp

   132 {

   133 public:  

   134   #ifdef EXTERNAL_CODECS

   135   CObjectVector<CCodecLib> Libs;

   136   CObjectVector<CDllCodecInfo> Codecs;

   137   HRESULT LoadCodecs();

   138   HRESULT LoadFormats();

   139   HRESULT LoadDll(const CSysString &path);

   140   HRESULT LoadDllsFromFolder(const CSysString &folderPrefix);

   141 

   142   HRESULT CreateArchiveHandler(const CArcInfoEx &ai, void **archive, bool outHandler) const

   143   {

   144     return Libs[ai.LibIndex].CreateObject(&ai.ClassID, outHandler ? &IID_IOutArchive : &IID_IInArchive, (void **)archive);

   145   }

   146   #endif

   147 

   148 public:

   149   CObjectVector<CArcInfoEx> Formats;

   150   HRESULT Load();

   151   int FindFormatForArchiveName(const UString &archivePath) const;

   152   int FindFormatForArchiveType(const UString &arcType) const;

   153 

   154   MY_UNKNOWN_IMP

   155 

   156   #ifdef EXTERNAL_CODECS

   157   STDMETHOD(GetNumberOfMethods)(UINT32 *numMethods);

   158   STDMETHOD(GetProperty)(UINT32 index, PROPID propID, PROPVARIANT *value);

   159   STDMETHOD(CreateDecoder)(UINT32 index, const GUID *interfaceID, void **coder);

   160   STDMETHOD(CreateEncoder)(UINT32 index, const GUID *interfaceID, void **coder);

   161   #endif

   162 

   163   int GetCodecLibIndex(UInt32 index);

   164   bool GetCodecEncoderIsAssigned(UInt32 index);

   165   HRESULT GetCodecId(UInt32 index, UInt64 &id);

   166   UString GetCodecName(UInt32 index);

   167 

   168   HRESULT CreateInArchive(int formatIndex, CMyComPtr<IInArchive> &archive) const

   169   { 

   170     const CArcInfoEx &ai = Formats[formatIndex]; 

   171     #ifdef EXTERNAL_CODECS

   172     if (ai.LibIndex < 0)

   173     #endif

   174     {

   175       archive = ai.CreateInArchive();

   176       return S_OK;

   177     }

   178     #ifdef EXTERNAL_CODECS

   179     return CreateArchiveHandler(ai, (void **)&archive, false); 

   180     #endif

   181   }

   182   HRESULT CreateOutArchive(int formatIndex, CMyComPtr<IOutArchive> &archive) const

   183   { 

   184     const CArcInfoEx &ai = Formats[formatIndex]; 

   185     #ifdef EXTERNAL_CODECS

   186     if (ai.LibIndex < 0)

   187     #endif

   188     {

   189       archive = ai.CreateOutArchive();

   190       return S_OK;

   191     }

   192     #ifdef EXTERNAL_CODECS

   193     return CreateArchiveHandler(ai, (void **)&archive, true); 

   194     #endif

   195   }

   196   int FindOutFormatFromName(const UString &name) const

   197   {

   198     for (int i = 0; i < Formats.Size(); i++)

   199     {

   200       const CArcInfoEx &arc = Formats[i];

   201       if (!arc.UpdateEnabled)

   202         continue;

   203       if (arc.Name.CompareNoCase(name) == 0)

   204         return i;

   205     }

   206     return -1;

   207   }

   208 

   209   #ifdef EXTERNAL_CODECS

   210   HRESULT CreateCoder(const UString &name, bool encode, CMyComPtr<ICompressCoder> &coder) const;

   211   #endif

   212 

   213 };

   214 

   215 #endif