OLD | NEW |
(Empty) | |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. |
| 4 |
| 5 #include "base/win/enum_variant.h" |
| 6 |
| 7 #include <algorithm> |
| 8 |
| 9 #include "base/logging.h" |
| 10 |
| 11 namespace base { |
| 12 namespace win { |
| 13 |
| 14 EnumVariant::EnumVariant(unsigned long count) |
| 15 : items_(new VARIANT[count]), |
| 16 count_(count), |
| 17 current_index_(0) { |
| 18 } |
| 19 |
| 20 EnumVariant::~EnumVariant() { |
| 21 } |
| 22 |
| 23 VARIANT* EnumVariant::ItemAt(unsigned long index) { |
| 24 DCHECK(index < count_); |
| 25 return &items_[index]; |
| 26 } |
| 27 |
| 28 ULONG STDMETHODCALLTYPE EnumVariant::AddRef() { |
| 29 return IUnknownImpl::AddRef(); |
| 30 } |
| 31 |
| 32 ULONG STDMETHODCALLTYPE EnumVariant::Release() { |
| 33 return IUnknownImpl::Release(); |
| 34 } |
| 35 |
| 36 STDMETHODIMP EnumVariant::QueryInterface(REFIID riid, void** ppv) { |
| 37 if (riid == IID_IEnumVARIANT) { |
| 38 *ppv = static_cast<IEnumVARIANT*>(this); |
| 39 AddRef(); |
| 40 return S_OK; |
| 41 } |
| 42 |
| 43 return IUnknownImpl::QueryInterface(riid, ppv); |
| 44 } |
| 45 |
| 46 STDMETHODIMP EnumVariant::Next(ULONG requested_count, |
| 47 VARIANT* out_elements, |
| 48 ULONG* out_elements_received) { |
| 49 unsigned long count = std::min(requested_count, count_ - current_index_); |
| 50 for (unsigned long i = 0; i < count; ++i) |
| 51 out_elements[i] = items_[current_index_ + i]; |
| 52 current_index_ += count; |
| 53 *out_elements_received = count; |
| 54 |
| 55 return (count == requested_count ? S_OK : S_FALSE); |
| 56 } |
| 57 |
| 58 STDMETHODIMP EnumVariant::Skip(ULONG skip_count) { |
| 59 unsigned long count = skip_count; |
| 60 if (current_index_ + count > count_) |
| 61 count = count_ - current_index_; |
| 62 |
| 63 current_index_ += count; |
| 64 return (count == skip_count ? S_OK : S_FALSE); |
| 65 } |
| 66 |
| 67 STDMETHODIMP EnumVariant::Reset() { |
| 68 current_index_ = 0; |
| 69 return S_OK; |
| 70 } |
| 71 |
| 72 STDMETHODIMP EnumVariant::Clone(IEnumVARIANT** out_cloned_object) { |
| 73 EnumVariant* other = new EnumVariant(count_); |
| 74 if (count_ > 0) |
| 75 memcpy(other->ItemAt(0), &items_[0], count_ * sizeof(VARIANT)); |
| 76 other->Skip(current_index_); |
| 77 other->AddRef(); |
| 78 *out_cloned_object = other; |
| 79 return S_OK; |
| 80 } |
| 81 |
| 82 } // namespace win |
| 83 } // namespace base |
OLD | NEW |