git.maemo.org Git - physicsfs/blob - lzma/CPP/Common/MyCom.h

   1 // MyCom.h
   2
   3 #ifndef __MYCOM_H
   4 #define __MYCOM_H
   5
   6 #include "MyWindows.h"
   7
   8 #ifndef RINOK
   9 #define RINOK(x) { HRESULT __result_ = (x); if(__result_ != S_OK) return __result_; }
  10 #endif
  11
  12 template <class T>
  13 class CMyComPtr
  14 {
  15   T* _p;
  16 public:
  17   // typedef T _PtrClass;
  18   CMyComPtr() { _p = NULL;}
  19   CMyComPtr(T* p) {if ((_p = p) != NULL) p->AddRef(); }
  20   CMyComPtr(const CMyComPtr<T>& lp)
  21   {
  22     if ((_p = lp._p) != NULL)
  23       _p->AddRef();
  24   }
  25   ~CMyComPtr() { if (_p) _p->Release(); }
  26   void Release() { if (_p) { _p->Release(); _p = NULL; } }
  27   operator T*() const {  return (T*)_p;  }
  28   // T& operator*() const {  return *_p; }
  29   T** operator&() { return &_p; }
  30   T* operator->() const { return _p; }
  31   T* operator=(T* p)
  32   {
  33     if (p != 0)
  34       p->AddRef();
  35     if (_p)
  36       _p->Release();
  37     _p = p;
  38     return p;
  39   }
  40   T* operator=(const CMyComPtr<T>& lp) { return (*this = lp._p); }
  41   bool operator!() const { return (_p == NULL); }
  42   // bool operator==(T* pT) const {  return _p == pT; }
  43   // Compare two objects for equivalence
  44   void Attach(T* p2)
  45   {
  46     Release();
  47     _p = p2;
  48   }
  49   T* Detach()
  50   {
  51     T* pt = _p;
  52     _p = NULL;
  53     return pt;
  54   }
  55   #ifdef _WIN32
  56   HRESULT CoCreateInstance(REFCLSID rclsid, REFIID iid, LPUNKNOWN pUnkOuter = NULL, DWORD dwClsContext = CLSCTX_ALL)
  57   {
  58     return ::CoCreateInstance(rclsid, pUnkOuter, dwClsContext, iid, (void**)&_p);
  59   }
  60   #endif
  61   /*
  62   HRESULT CoCreateInstance(LPCOLESTR szProgID, LPUNKNOWN pUnkOuter = NULL, DWORD dwClsContext = CLSCTX_ALL)
  63   {
  64     CLSID clsid;
  65     HRESULT hr = CLSIDFromProgID(szProgID, &clsid);
  66     ATLASSERT(_p == NULL);
  67     if (SUCCEEDED(hr))
  68       hr = ::CoCreateInstance(clsid, pUnkOuter, dwClsContext, __uuidof(T), (void**)&_p);
  69     return hr;
  70   }
  71   */
  72   template <class Q>
  73   HRESULT QueryInterface(REFGUID iid, Q** pp) const
  74   {
  75     return _p->QueryInterface(iid, (void**)pp);
  76   }
  77 };
  78
  79 //////////////////////////////////////////////////////////
  80
  81 class CMyComBSTR
  82 {
  83 public:
  84   BSTR m_str;
  85   CMyComBSTR() { m_str = NULL; }
  86   CMyComBSTR(LPCOLESTR pSrc) {  m_str = ::SysAllocString(pSrc);  }
  87   // CMyComBSTR(int nSize) { m_str = ::SysAllocStringLen(NULL, nSize); }
  88   // CMyComBSTR(int nSize, LPCOLESTR sz) { m_str = ::SysAllocStringLen(sz, nSize);  }
  89   CMyComBSTR(const CMyComBSTR& src) { m_str = src.MyCopy(); }
  90   /*
  91   CMyComBSTR(REFGUID src)
  92   {
  93     LPOLESTR szGuid;
  94     StringFromCLSID(src, &szGuid);
  95     m_str = ::SysAllocString(szGuid);
  96     CoTaskMemFree(szGuid);
  97   }
  98   */
  99   ~CMyComBSTR() { ::SysFreeString(m_str); }
 100   CMyComBSTR& operator=(const CMyComBSTR& src)
 101   {
 102     if (m_str != src.m_str)
 103     {
 104       if (m_str)
 105         ::SysFreeString(m_str);
 106       m_str = src.MyCopy();
 107     }
 108     return *this;
 109   }
 110   CMyComBSTR& operator=(LPCOLESTR pSrc)
 111   {
 112     ::SysFreeString(m_str);
 113     m_str = ::SysAllocString(pSrc);
 114     return *this;
 115   }
 116   unsigned int Length() const { return ::SysStringLen(m_str); }
 117   operator BSTR() const { return m_str; }
 118   BSTR* operator&() { return &m_str; }
 119   BSTR MyCopy() const
 120   {
 121     int byteLen = ::SysStringByteLen(m_str);
 122     BSTR res = ::SysAllocStringByteLen(NULL, byteLen);
 123     memmove(res, m_str, byteLen);
 124     return res;
 125   }
 126   void Attach(BSTR src) {  m_str = src; }
 127   BSTR Detach()
 128   {
 129     BSTR s = m_str;
 130     m_str = NULL;
 131     return s;
 132   }
 133   void Empty()
 134   {
 135     ::SysFreeString(m_str);
 136     m_str = NULL;
 137   }
 138   bool operator!() const {  return (m_str == NULL); }
 139 };
 140
 141
 142 //////////////////////////////////////////////////////////
 143
 144 class CMyUnknownImp
 145 {
 146 public:
 147   ULONG __m_RefCount;
 148   CMyUnknownImp(): __m_RefCount(0) {}
 149 };
 150
 151 #define MY_QUERYINTERFACE_BEGIN STDMETHOD(QueryInterface) \
 152     (REFGUID iid, void **outObject) {
 153
 154 #define MY_QUERYINTERFACE_ENTRY(i) if (iid == IID_ ## i) \
 155     { *outObject = (void *)(i *)this; AddRef(); return S_OK; }
 156
 157 #define MY_QUERYINTERFACE_ENTRY_UNKNOWN(i) if (iid == IID_IUnknown) \
 158     { *outObject = (void *)(IUnknown *)(i *)this; AddRef(); return S_OK; }
 159
 160 #define MY_QUERYINTERFACE_BEGIN2(i) MY_QUERYINTERFACE_BEGIN \
 161     MY_QUERYINTERFACE_ENTRY_UNKNOWN(i) \
 162     MY_QUERYINTERFACE_ENTRY(i)
 163
 164 #define MY_QUERYINTERFACE_END return E_NOINTERFACE; }
 165
 166 #define MY_ADDREF_RELEASE \
 167 STDMETHOD_(ULONG, AddRef)() { return ++__m_RefCount; } \
 168 STDMETHOD_(ULONG, Release)() { if (--__m_RefCount != 0)  \
 169   return __m_RefCount; delete this; return 0; }
 170
 171 #define MY_UNKNOWN_IMP_SPEC(i) \
 172   MY_QUERYINTERFACE_BEGIN \
 173   i \
 174   MY_QUERYINTERFACE_END \
 175   MY_ADDREF_RELEASE
 176
 177
 178 #define MY_UNKNOWN_IMP MY_QUERYINTERFACE_BEGIN \
 179   MY_QUERYINTERFACE_ENTRY_UNKNOWN(IUnknown) \
 180   MY_QUERYINTERFACE_END \
 181   MY_ADDREF_RELEASE
 182
 183 #define MY_UNKNOWN_IMP1(i) MY_UNKNOWN_IMP_SPEC( \
 184   MY_QUERYINTERFACE_ENTRY_UNKNOWN(i) \
 185   MY_QUERYINTERFACE_ENTRY(i) \
 186   )
 187
 188 #define MY_UNKNOWN_IMP2(i1, i2) MY_UNKNOWN_IMP_SPEC( \
 189   MY_QUERYINTERFACE_ENTRY_UNKNOWN(i1) \
 190   MY_QUERYINTERFACE_ENTRY(i1) \
 191   MY_QUERYINTERFACE_ENTRY(i2) \
 192   )
 193
 194 #define MY_UNKNOWN_IMP3(i1, i2, i3) MY_UNKNOWN_IMP_SPEC( \
 195   MY_QUERYINTERFACE_ENTRY_UNKNOWN(i1) \
 196   MY_QUERYINTERFACE_ENTRY(i1) \
 197   MY_QUERYINTERFACE_ENTRY(i2) \
 198   MY_QUERYINTERFACE_ENTRY(i3) \
 199   )
 200
 201 #define MY_UNKNOWN_IMP4(i1, i2, i3, i4) MY_UNKNOWN_IMP_SPEC( \
 202   MY_QUERYINTERFACE_ENTRY_UNKNOWN(i1) \
 203   MY_QUERYINTERFACE_ENTRY(i1) \
 204   MY_QUERYINTERFACE_ENTRY(i2) \
 205   MY_QUERYINTERFACE_ENTRY(i3) \
 206   MY_QUERYINTERFACE_ENTRY(i4) \
 207   )
 208
 209 #define MY_UNKNOWN_IMP5(i1, i2, i3, i4, i5) MY_UNKNOWN_IMP_SPEC( \
 210   MY_QUERYINTERFACE_ENTRY_UNKNOWN(i1) \
 211   MY_QUERYINTERFACE_ENTRY(i1) \
 212   MY_QUERYINTERFACE_ENTRY(i2) \
 213   MY_QUERYINTERFACE_ENTRY(i3) \
 214   MY_QUERYINTERFACE_ENTRY(i4) \
 215   MY_QUERYINTERFACE_ENTRY(i5) \
 216   )
 217
 218 #endif