174 lines
3.3 KiB
C++
174 lines
3.3 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;
|
|
}
|