AutoEnumString.cpp
上传用户:huashan618
上传日期:2013-03-26
资源大小:261k
文件大小:3k
源码类别:

浏览器

开发平台:

Visual C++

  1. // AutoEnumString.cpp: implementation of the CAutoEnumString class.
  2. //
  3. //////////////////////////////////////////////////////////////////////
  4. #include "stdafx.h"
  5. #include "AutoEnumString.h"
  6. #ifdef _DEBUG
  7. #undef THIS_FILE
  8. static char THIS_FILE[]=__FILE__;
  9. #define new DEBUG_NEW
  10. #endif
  11. //////////////////////////////////////////////////////////////////////
  12. // Construction/Destruction
  13. //////////////////////////////////////////////////////////////////////
  14. CAutoEnumString::CAutoEnumString()
  15. {
  16. m_Ref = 0;
  17. m_iterCur = 0;
  18. }
  19. CAutoEnumString::~CAutoEnumString()
  20. {
  21. }
  22. int CAutoEnumString::AddString(LPCSTR lpszStr)
  23. {
  24. return m_arString.Add(lpszStr);
  25. }
  26. ULONG CAutoEnumString::AddRef()
  27. {
  28. m_Ref ++;
  29. return  (ULONG) m_Ref;
  30. }
  31. ULONG CAutoEnumString::Release()
  32. {
  33. m_Ref --;
  34. if (m_Ref == 0 ) {
  35. delete this;
  36. return 0;
  37. }
  38. return  (ULONG) m_Ref;
  39. }
  40. HRESULT CAutoEnumString::QueryInterface(const IID &iid, void **ppv)
  41. {
  42. if ( iid == IID_IUnknown )
  43. {
  44. *ppv = (IEnumString *) this ;
  45. ((IEnumString *)(*ppv))->AddRef() ;
  46. } else if ( iid == IID_IEnumString ) 
  47. {
  48. *ppv = (IEnumString *) this ;
  49. ((IEnumString *)(*ppv))->AddRef() ;
  50. }
  51. else
  52. {
  53. *ppv = NULL;
  54. return E_NOINTERFACE ;
  55. }
  56. return S_OK;
  57. }
  58. HRESULT CAutoEnumString::Next(ULONG celt, LPOLESTR *rgelt, ULONG *pceltFetched)
  59. {
  60. if (rgelt == NULL || (celt != 1 && pceltFetched == NULL)) 
  61.         return E_POINTER; 
  62.     if (m_iterCur == m_arString.GetSize()) 
  63.         return E_FAIL; 
  64.     ULONG nRem = (ULONG)(m_arString.GetSize() - m_iterCur); 
  65.     HRESULT hRes = S_OK; 
  66.     if (nRem < celt) 
  67.         hRes = S_FALSE; 
  68.     ULONG nMin = min(celt, nRem); 
  69.     if (pceltFetched != NULL) 
  70.         *pceltFetched = nMin; 
  71.     while(nMin--) {
  72. CString str = m_arString.GetAt(m_iterCur++);
  73. BSTR bstr = str.AllocSysString();
  74.         Copy(rgelt++, &bstr); 
  75. }
  76.     return hRes; 
  77. }
  78. HRESULT CAutoEnumString::Copy(LPOLESTR* p1, LPOLESTR* p2)
  79. {
  80. HRESULT hr = S_OK;
  81. (*p1) = (LPOLESTR)CoTaskMemAlloc(sizeof(OLECHAR)*(ocslen(*p2)+1));
  82. if (*p1 == NULL)
  83. hr = E_OUTOFMEMORY;
  84. else
  85. ocscpy(*p1,*p2);
  86. return hr;
  87. }
  88. HRESULT CAutoEnumString::Reset(void)
  89. {
  90. m_iterCur = 0;
  91. return S_OK;
  92. }
  93. HRESULT CAutoEnumString::Skip(ULONG celt)
  94. {
  95. if (m_arString.GetSize() - m_iterCur <  (long)celt)
  96. {
  97. m_iterCur = (int)m_arString.GetSize();
  98. return S_FALSE;
  99. }
  100. m_iterCur += celt;
  101. return S_FALSE;
  102. }
  103. HRESULT CAutoEnumString::Clone(IEnumString** ppEnum)
  104. {
  105. HRESULT hRes = E_POINTER;
  106. if (ppEnum != NULL)
  107. {
  108. CAutoEnumString* p = NULL;
  109. p = new CAutoEnumString;
  110. if (p == NULL)
  111. {
  112. *ppEnum = NULL;
  113. hRes = E_OUTOFMEMORY;
  114. }
  115. else
  116. {
  117. for(int i=0;i<m_arString.GetSize();i++)
  118. p->AddString(m_arString.GetAt(i));
  119. hRes = p->QueryInterface(IID_IEnumString, (void**)ppEnum);
  120. if (FAILED(hRes))
  121. delete p;
  122. }
  123. }
  124. return hRes;
  125. }