forked from mirrors/gecko-dev
		
	 e226046cb8
			
		
	
	
		e226046cb8
		
	
	
	
	
		
			
			# ignore-this-changeset Depends on D28954 Differential Revision: https://phabricator.services.mozilla.com/D28956 --HG-- extra : moz-landing-system : lando
		
			
				
	
	
		
			139 lines
		
	
	
	
		
			3.2 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			139 lines
		
	
	
	
		
			3.2 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
| /* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
 | |
| /* This Source Code Form is subject to the terms of the Mozilla Public
 | |
|  * License, v. 2.0. If a copy of the MPL was not distributed with this
 | |
|  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
 | |
| 
 | |
| #include "IEnumFE.h"
 | |
| #include "nsAlgorithm.h"
 | |
| #include <algorithm>
 | |
| 
 | |
| CEnumFormatEtc::CEnumFormatEtc() : mRefCnt(0), mCurrentIdx(0) {}
 | |
| 
 | |
| // Constructor used by Clone()
 | |
| CEnumFormatEtc::CEnumFormatEtc(nsTArray<FormatEtc>& aArray)
 | |
|     : mRefCnt(0), mCurrentIdx(0) {
 | |
|   // a deep copy, calls FormatEtc's copy constructor on each
 | |
|   mFormatList.AppendElements(aArray);
 | |
| }
 | |
| 
 | |
| CEnumFormatEtc::~CEnumFormatEtc() {}
 | |
| 
 | |
| /* IUnknown impl. */
 | |
| 
 | |
| STDMETHODIMP
 | |
| CEnumFormatEtc::QueryInterface(REFIID riid, LPVOID* ppv) {
 | |
|   *ppv = nullptr;
 | |
| 
 | |
|   if (IsEqualIID(riid, IID_IUnknown) || IsEqualIID(riid, IID_IEnumFORMATETC))
 | |
|     *ppv = (LPVOID)this;
 | |
| 
 | |
|   if (*ppv == nullptr) return E_NOINTERFACE;
 | |
| 
 | |
|   // AddRef any interface we'll return.
 | |
|   ((LPUNKNOWN)*ppv)->AddRef();
 | |
|   return S_OK;
 | |
| }
 | |
| 
 | |
| STDMETHODIMP_(ULONG)
 | |
| CEnumFormatEtc::AddRef() {
 | |
|   ++mRefCnt;
 | |
|   NS_LOG_ADDREF(this, mRefCnt, "CEnumFormatEtc", sizeof(*this));
 | |
|   return mRefCnt;
 | |
| }
 | |
| 
 | |
| STDMETHODIMP_(ULONG)
 | |
| CEnumFormatEtc::Release() {
 | |
|   uint32_t refReturn;
 | |
| 
 | |
|   refReturn = --mRefCnt;
 | |
|   NS_LOG_RELEASE(this, mRefCnt, "CEnumFormatEtc");
 | |
| 
 | |
|   if (mRefCnt == 0) delete this;
 | |
| 
 | |
|   return refReturn;
 | |
| }
 | |
| 
 | |
| /* IEnumFORMATETC impl. */
 | |
| 
 | |
| STDMETHODIMP
 | |
| CEnumFormatEtc::Next(ULONG aMaxToFetch, FORMATETC* aResult,
 | |
|                      ULONG* aNumFetched) {
 | |
|   // If the method retrieves the number of items requested, the return
 | |
|   // value is S_OK. Otherwise, it is S_FALSE.
 | |
| 
 | |
|   if (aNumFetched) *aNumFetched = 0;
 | |
| 
 | |
|   // aNumFetched can be null if aMaxToFetch is 1
 | |
|   if (!aNumFetched && aMaxToFetch > 1) return S_FALSE;
 | |
| 
 | |
|   if (!aResult) return S_FALSE;
 | |
| 
 | |
|   // We're done walking the list
 | |
|   if (mCurrentIdx >= mFormatList.Length()) return S_FALSE;
 | |
| 
 | |
|   uint32_t left = mFormatList.Length() - mCurrentIdx;
 | |
| 
 | |
|   if (!aMaxToFetch) return S_FALSE;
 | |
| 
 | |
|   uint32_t count = std::min(static_cast<uint32_t>(aMaxToFetch), left);
 | |
| 
 | |
|   uint32_t idx = 0;
 | |
|   while (count > 0) {
 | |
|     // Copy out to aResult
 | |
|     mFormatList[mCurrentIdx++].CopyOut(&aResult[idx++]);
 | |
|     count--;
 | |
|   }
 | |
| 
 | |
|   if (aNumFetched) *aNumFetched = idx;
 | |
| 
 | |
|   return S_OK;
 | |
| }
 | |
| 
 | |
| STDMETHODIMP
 | |
| CEnumFormatEtc::Skip(ULONG aSkipNum) {
 | |
|   // If the method skips the number of items requested, the return value is
 | |
|   // S_OK. Otherwise, it is S_FALSE.
 | |
| 
 | |
|   if ((mCurrentIdx + aSkipNum) >= mFormatList.Length()) return S_FALSE;
 | |
| 
 | |
|   mCurrentIdx += aSkipNum;
 | |
| 
 | |
|   return S_OK;
 | |
| }
 | |
| 
 | |
| STDMETHODIMP
 | |
| CEnumFormatEtc::Reset(void) {
 | |
|   mCurrentIdx = 0;
 | |
|   return S_OK;
 | |
| }
 | |
| 
 | |
| STDMETHODIMP
 | |
| CEnumFormatEtc::Clone(LPENUMFORMATETC* aResult) {
 | |
|   // Must return a new IEnumFORMATETC interface with the same iterative state.
 | |
| 
 | |
|   if (!aResult) return E_INVALIDARG;
 | |
| 
 | |
|   CEnumFormatEtc* pEnumObj = new CEnumFormatEtc(mFormatList);
 | |
| 
 | |
|   if (!pEnumObj) return E_OUTOFMEMORY;
 | |
| 
 | |
|   pEnumObj->AddRef();
 | |
|   pEnumObj->SetIndex(mCurrentIdx);
 | |
| 
 | |
|   *aResult = pEnumObj;
 | |
| 
 | |
|   return S_OK;
 | |
| }
 | |
| 
 | |
| /* utils */
 | |
| 
 | |
| void CEnumFormatEtc::AddFormatEtc(LPFORMATETC aFormat) {
 | |
|   if (!aFormat) return;
 | |
|   FormatEtc* etc = mFormatList.AppendElement();
 | |
|   // Make a copy of aFormat
 | |
|   if (etc) etc->CopyIn(aFormat);
 | |
| }
 | |
| 
 | |
| /* private */
 | |
| 
 | |
| void CEnumFormatEtc::SetIndex(uint32_t aIdx) { mCurrentIdx = aIdx; }
 |