ENUM.CPP
上传用户:bangxh
上传日期:2007-01-31
资源大小:42235k
文件大小:4k
源码类别:

Windows编程

开发平台:

Visual C++

  1. /*************************************************************************
  2. **
  3. **  This is a part of the Microsoft Source Code Samples.
  4. **
  5. **  Copyright (C) 1992-1997 Microsoft Corporation. All rights reserved.
  6. **
  7. **  This source code is only intended as a supplement to Microsoft Development
  8. **  Tools and/or WinHelp documentation.  See these sources for detailed
  9. **  information regarding the Microsoft samples programs.
  10. **
  11. **  OLE Automation TypeLibrary Browse Helper Sample
  12. **
  13. **  enum.cpp
  14. **
  15. **  CEnum implementation
  16. **
  17. **  Written by Microsoft Product Support Services, Windows Developer Support
  18. **
  19. *************************************************************************/
  20. #include <windows.h>
  21. #include <windowsx.h>
  22. #ifdef WIN16   
  23.   #include <ole2.h>
  24.   #include <compobj.h>    
  25.   #include <dispatch.h> 
  26.   #include <variant.h>
  27.   #include <olenls.h>  
  28. #endif 
  29. #include "browseh.h"  
  30. /*
  31.  * CEnum::Create
  32.  *
  33.  * Purpose:
  34.  *  Creates an instance of the Enum automation object and initializes it.
  35.  *
  36.  * Parameters:       
  37.  *  ptinfo    TypeInfo of Enum.
  38.  *  ppEnum    Returns Enum automation object.
  39.  *
  40.  * Return Value:
  41.  *  HRESULT
  42.  *
  43.  */
  44. HRESULT 
  45. CEnum::Create(LPTYPEINFO ptinfo, CEnum FAR* FAR* ppEnum) 
  46. {   
  47.     HRESULT hr;
  48.     CEnum FAR* pEnum = NULL;
  49.      
  50.     *ppEnum = NULL;
  51.     
  52.     // Create object.
  53.     pEnum = new CEnum();
  54.     if (pEnum == NULL)
  55.     {
  56.         hr = E_OUTOFMEMORY; 
  57.         goto error;
  58.     }   
  59.     // Load type information for the object from type library. 
  60.     hr = pEnum->LoadTypeInfo(IID_IEnum);
  61.     if (FAILED(hr))
  62.         goto error;  
  63.     // Ask base class (CTypeInfo) to initialize.        
  64.     hr = pEnum->_InitTypeInfo(ptinfo);
  65.     if (FAILED(hr))
  66.         goto error;
  67.     
  68.     ptinfo->AddRef();
  69.     pEnum->m_ptinfo = ptinfo;
  70. #ifdef _DEBUG  
  71.     lstrcpyn(pEnum->m_szClassName, TEXT("Enum"), 100);
  72. #endif
  73.         
  74.     *ppEnum = pEnum;
  75.     return NOERROR;
  76.     
  77. error:
  78.     if (pEnum == NULL) return E_OUTOFMEMORY;
  79.     if (pEnum->m_ptinfo) pEnum->m_ptinfo->Release();
  80.          
  81.     // Set to NULL to prevent destructor from attempting to free again  
  82.     pEnum->m_ptinfo = NULL;
  83.     
  84.     delete pEnum;
  85.     return hr;
  86. }
  87. /*
  88.  * CEnum::CEnum
  89.  *
  90.  * Purpose:
  91.  *  Constructor for CEnum object. Initializes members to NULL.
  92.  *
  93.  */
  94. CEnum::CEnum()
  95. {
  96.     m_pdispElements = NULL;      
  97.     m_ptinfo = NULL;
  98. }
  99. /*
  100.  * CEnum::~CEnum
  101.  *
  102.  * Purpose:
  103.  *  Destructor for CEnum object. 
  104.  *
  105.  */
  106. CEnum::~CEnum()
  107. {
  108.     if (m_pdispElements) m_pdispElements->Release();
  109.     if (m_ptinfo) m_ptinfo->Release();
  110. }  
  111. STDMETHODIMP_(REFCLSID)
  112. CEnum::GetInterfaceID()
  113. {
  114.     return IID_IEnum;
  115. }
  116. STDMETHODIMP_(ICollection FAR*)
  117. CEnum::get_Elements()     
  118. {    
  119.     HRESULT hr;
  120.     CConstant FAR* pConstant;
  121.     CCollection FAR* pCollection = NULL;
  122.     LPDISPATCH pdisp;
  123.     LPVARDESC pvardesc = NULL;   
  124.     LPTYPEATTR ptypeattr = NULL;
  125.     unsigned short n;
  126.     
  127.     if (m_pdispElements == NULL)
  128.     {    
  129.         hr = m_ptinfo->GetTypeAttr(&ptypeattr);
  130.         if (FAILED(hr))
  131.             {RaiseException(IDS_Unexpected); return NULL;}       
  132.         hr = CCollection::Create(ptypeattr->cVars, 0, &pCollection);  
  133.         if (FAILED(hr))
  134.             {RaiseException(IDS_Unexpected); goto error;}     
  135.         // Enumerate enum constants and return a collection of these.
  136.         for (n=0; n<ptypeattr->cVars; n++)
  137.         {       
  138.             hr = m_ptinfo->GetVarDesc(n, &pvardesc);   
  139.             if (FAILED(hr))
  140.                 {RaiseException(IDS_Unexpected); goto error;}   
  141.             hr = CConstant::Create(m_ptinfo, pvardesc, &pConstant);
  142.             if (FAILED(hr))
  143.                 {RaiseException(IDS_Unexpected); goto error;}    
  144.             m_ptinfo->ReleaseVarDesc(pvardesc); 
  145.             pvardesc = NULL;
  146.             pConstant->QueryInterface(IID_IDispatch, (void FAR* FAR*)&pdisp);
  147.             pCollection->Add(pdisp);   
  148.             pdisp->Release();
  149.         }
  150.         pCollection->QueryInterface(IID_IDispatch, (void FAR* FAR*)&pdisp);
  151.         m_pdispElements = pdisp;    
  152.         m_ptinfo->ReleaseTypeAttr(ptypeattr); 
  153.     }
  154.     m_pdispElements->AddRef();
  155.     return (ICollection FAR*)m_pdispElements;
  156. error:  
  157.     if (ptypeattr) m_ptinfo->ReleaseTypeAttr(ptypeattr);   
  158.     if (pCollection) delete pCollection;   
  159.     if (pvardesc) m_ptinfo->ReleaseVarDesc(pvardesc);   
  160.     return NULL;
  161. }