dll/directx/ksproxy/enumpins.cpp

   1 /*
   2  * COPYRIGHT:       See COPYING in the top level directory
   3  * PROJECT:         ReactOS Network Provider for MPEG2 based networks
   4  * FILE:            dll/directx/msdvbnp/enumpins.cpp
   5  * PURPOSE:         IEnumPins interface
   6  *
   7  * PROGRAMMERS:     Johannes Anderwald (janderwald@reactos.org)
   8  */
   9 #include "precomp.h"
  10
  11 class CEnumPins : public IEnumPins
  12 {
  13 public:
  14     STDMETHODIMP QueryInterface( REFIID InterfaceId, PVOID* Interface);
  15
  16     STDMETHODIMP_(ULONG) AddRef()
  17     {
  18         InterlockedIncrement(&m_Ref);
  19         return m_Ref;
  20     }
  21     STDMETHODIMP_(ULONG) Release()
  22     {
  23         InterlockedDecrement(&m_Ref);
  24         if (!m_Ref)
  25         {
  26             delete this;
  27             return 0;
  28         }
  29         return m_Ref;
  30     }
  31
  32
  33     HRESULT STDMETHODCALLTYPE Next(ULONG cPins, IPin **ppPins, ULONG *pcFetched);
  34     HRESULT STDMETHODCALLTYPE Skip(ULONG cPins);
  35     HRESULT STDMETHODCALLTYPE Reset();
  36     HRESULT STDMETHODCALLTYPE Clone(IEnumPins **ppEnum);
  37
  38     CEnumPins(std::vector<IPin*> Pins) : m_Ref(0), m_Pins(Pins), m_Index(0){};
  39     virtual ~CEnumPins(){};
  40
  41 protected:
  42     LONG m_Ref;
  43     std::vector<IPin*> m_Pins;
  44     ULONG m_Index;
  45 };
  46
  47 HRESULT
  48 STDMETHODCALLTYPE
  49 CEnumPins::QueryInterface(
  50     IN  REFIID refiid,
  51     OUT PVOID* Output)
  52 {
  53     *Output = NULL;
  54     if (IsEqualGUID(refiid, IID_IUnknown))
  55     {
  56         *Output = PVOID(this);
  57         reinterpret_cast<IUnknown*>(*Output)->AddRef();
  58         return NOERROR;
  59     }
  60     if (IsEqualGUID(refiid, IID_IEnumPins))
  61     {
  62         *Output = (IEnumPins*)(this);
  63         reinterpret_cast<IEnumPins*>(*Output)->AddRef();
  64         return NOERROR;
  65     }
  66
  67     WCHAR Buffer[100];
  68     LPOLESTR lpstr;
  69     StringFromCLSID(refiid, &lpstr);
  70     swprintf(Buffer, L"CEnumPins::QueryInterface: NoInterface for %s\n", lpstr);
  71     OutputDebugStringW(Buffer);
  72     CoTaskMemFree(lpstr);
  73
  74 DebugBreak();
  75     return E_NOINTERFACE;
  76 }
  77
  78 HRESULT
  79 STDMETHODCALLTYPE
  80 CEnumPins::Next(
  81     ULONG cPins,
  82     IPin **ppPins,
  83     ULONG *pcFetched)
  84 {
  85     ULONG i = 0;
  86
  87     if (!ppPins)
  88         return E_POINTER;
  89
  90     if (cPins > 1 && !pcFetched)
  91         return E_INVALIDARG;
  92
  93     while(i < cPins)
  94     {
  95         if (m_Index + i >= m_Pins.size())
  96             break;
  97
  98         ppPins[i] = m_Pins[m_Index + i];
  99         m_Pins[m_Index + i]->AddRef();
 100
 101         i++;
 102     }
 103
 104     if (pcFetched)
 105     {
 106         *pcFetched = i;
 107     }
 108
 109     m_Index += i;
 110     if (i < cPins)
 111         return S_FALSE;
 112     else
 113         return S_OK;
 114 }
 115
 116 HRESULT
 117 STDMETHODCALLTYPE
 118 CEnumPins::Skip(
 119     ULONG cPins)
 120 {
 121     if (cPins + m_Index >= m_Pins.size())
 122     {
 123         return S_FALSE;
 124     }
 125
 126     m_Index += cPins;
 127     return S_OK;
 128 }
 129
 130 HRESULT
 131 STDMETHODCALLTYPE
 132 CEnumPins::Reset()
 133 {
 134     m_Index = 0;
 135     return S_OK;
 136 }
 137
 138 HRESULT
 139 STDMETHODCALLTYPE
 140 CEnumPins::Clone(
 141     IEnumPins **ppEnum)
 142 {
 143 #ifdef KSPROXY_TRACE
 144     OutputDebugStringW(L"CEnumPins::Clone : NotImplemented\n");
 145 #endif
 146
 147     return E_NOTIMPL;
 148 }
 149
 150 HRESULT
 151 WINAPI
 152 CEnumPins_fnConstructor(
 153     std::vector<IPin*> Pins,
 154     REFIID riid,
 155     LPVOID * ppv)
 156 {
 157     CEnumPins * handler = new CEnumPins(Pins);
 158
 159     if (!handler)
 160         return E_OUTOFMEMORY;
 161
 162     if (FAILED(handler->QueryInterface(riid, ppv)))
 163     {
 164         /* not supported */
 165         delete handler;
 166         return E_NOINTERFACE;
 167     }
 168
 169     return NOERROR;
 170 }