home *** CD-ROM | disk | FTP | other *** search
/ Tricks of the Windows Gam…ming Gurus (2nd Edition) / Disc2.iso / msdn_vcb / samples / vc98 / sdk / com / oleaut / browseh / enum.cpp < prev    next >
C/C++ Source or Header  |  1997-07-31  |  4KB  |  170 lines
  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.  
  21. #include <windows.h>
  22. #include <windowsx.h>
  23. #ifdef WIN16   
  24.   #include <ole2.h>
  25.   #include <compobj.h>    
  26.   #include <dispatch.h> 
  27.   #include <variant.h>
  28.   #include <olenls.h>  
  29. #endif 
  30. #include "browseh.h"  
  31.  
  32. /*
  33.  * CEnum::Create
  34.  *
  35.  * Purpose:
  36.  *  Creates an instance of the Enum automation object and initializes it.
  37.  *
  38.  * Parameters:       
  39.  *  ptinfo    TypeInfo of Enum.
  40.  *  ppEnum    Returns Enum automation object.
  41.  *
  42.  * Return Value:
  43.  *  HRESULT
  44.  *
  45.  */
  46. HRESULT 
  47. CEnum::Create(LPTYPEINFO ptinfo, CEnum FAR* FAR* ppEnum) 
  48. {   
  49.     HRESULT hr;
  50.     CEnum FAR* pEnum = NULL;
  51.      
  52.     *ppEnum = NULL;
  53.     
  54.     // Create object.
  55.     pEnum = new CEnum();
  56.     if (pEnum == NULL)
  57.     {
  58.         hr = E_OUTOFMEMORY; 
  59.         goto error;
  60.     }   
  61.     // Load type information for the object from type library. 
  62.     hr = pEnum->LoadTypeInfo(IID_IEnum);
  63.     if (FAILED(hr))
  64.         goto error;  
  65.     // Ask base class (CTypeInfo) to initialize.        
  66.     hr = pEnum->_InitTypeInfo(ptinfo);
  67.     if (FAILED(hr))
  68.         goto error;
  69.     
  70.     ptinfo->AddRef();
  71.     pEnum->m_ptinfo = ptinfo;
  72.  
  73. #ifdef _DEBUG  
  74.     lstrcpyn(pEnum->m_szClassName, TEXT("Enum"), 100);
  75. #endif
  76.         
  77.     *ppEnum = pEnum;
  78.     return NOERROR;
  79.     
  80. error:
  81.     if (pEnum == NULL) return E_OUTOFMEMORY;
  82.     if (pEnum->m_ptinfo) pEnum->m_ptinfo->Release();
  83.          
  84.     // Set to NULL to prevent destructor from attempting to free again  
  85.     pEnum->m_ptinfo = NULL;
  86.     
  87.     delete pEnum;
  88.     return hr;
  89. }
  90.  
  91. /*
  92.  * CEnum::CEnum
  93.  *
  94.  * Purpose:
  95.  *  Constructor for CEnum object. Initializes members to NULL.
  96.  *
  97.  */
  98. CEnum::CEnum()
  99. {
  100.     m_pdispElements = NULL;      
  101.     m_ptinfo = NULL;
  102. }
  103.  
  104. /*
  105.  * CEnum::~CEnum
  106.  *
  107.  * Purpose:
  108.  *  Destructor for CEnum object. 
  109.  *
  110.  */
  111. CEnum::~CEnum()
  112. {
  113.     if (m_pdispElements) m_pdispElements->Release();
  114.     if (m_ptinfo) m_ptinfo->Release();
  115. }  
  116.  
  117. STDMETHODIMP_(REFCLSID)
  118. CEnum::GetInterfaceID()
  119. {
  120.     return IID_IEnum;
  121. }
  122.  
  123. STDMETHODIMP_(ICollection FAR*)
  124. CEnum::get_Elements()     
  125. {    
  126.     HRESULT hr;
  127.     CConstant FAR* pConstant;
  128.     CCollection FAR* pCollection = NULL;
  129.     LPDISPATCH pdisp;
  130.     LPVARDESC pvardesc = NULL;   
  131.     LPTYPEATTR ptypeattr = NULL;
  132.     unsigned short n;
  133.     
  134.     if (m_pdispElements == NULL)
  135.     {    
  136.         hr = m_ptinfo->GetTypeAttr(&ptypeattr);
  137.         if (FAILED(hr))
  138.             {RaiseException(IDS_Unexpected); return NULL;}       
  139.         hr = CCollection::Create(ptypeattr->cVars, 0, &pCollection);  
  140.         if (FAILED(hr))
  141.             {RaiseException(IDS_Unexpected); goto error;}     
  142.         // Enumerate enum constants and return a collection of these.
  143.         for (n=0; n<ptypeattr->cVars; n++)
  144.         {       
  145.             hr = m_ptinfo->GetVarDesc(n, &pvardesc);   
  146.             if (FAILED(hr))
  147.                 {RaiseException(IDS_Unexpected); goto error;}   
  148.             hr = CConstant::Create(m_ptinfo, pvardesc, &pConstant);
  149.             if (FAILED(hr))
  150.                 {RaiseException(IDS_Unexpected); goto error;}    
  151.             m_ptinfo->ReleaseVarDesc(pvardesc); 
  152.             pvardesc = NULL;
  153.             pConstant->QueryInterface(IID_IDispatch, (void FAR* FAR*)&pdisp);
  154.             pCollection->Add(pdisp);   
  155.             pdisp->Release();
  156.         }
  157.         pCollection->QueryInterface(IID_IDispatch, (void FAR* FAR*)&pdisp);
  158.         m_pdispElements = pdisp;    
  159.         m_ptinfo->ReleaseTypeAttr(ptypeattr); 
  160.     }
  161.     m_pdispElements->AddRef();
  162.     return (ICollection FAR*)m_pdispElements;
  163.  
  164. error:  
  165.     if (ptypeattr) m_ptinfo->ReleaseTypeAttr(ptypeattr);   
  166.     if (pCollection) delete pCollection;   
  167.     if (pvardesc) m_ptinfo->ReleaseVarDesc(pvardesc);   
  168.     return NULL;
  169. }    
  170.