update simplemediasource sample (#614)

* update simplemediasource wrl to winrtcpp COM, add mediatype NV12 (SimpleFrameGenerator), fix Mediatype missing attribute (bitrate), add IMFActivate, IMFSampleAllocator interface and implementation, update hardcode arg with named variables, cleanup sample for StreamIndex and StreamId, add sample KSControl
* address PR-it1 comments
* address comment from PR-it2, replace manual define of IKSControl by ksproxy.h
* remove unuse header
* address comment from PR
* fix sdk to latest, add driversign algorithm
* remove nudget, use submodule

1 files changed, 104 insertions, 65 deletions

diff --git a/general/SimpleMediaSource/MediaSource/SimpleMediaSource.h b/general/SimpleMediaSource/MediaSource/SimpleMediaSource.h
index 019926c8..10ac200b 100644
--- a/general/SimpleMediaSource/MediaSource/SimpleMediaSource.h
+++ b/general/SimpleMediaSource/MediaSource/SimpleMediaSource.h
@@ -1,81 +1,120 @@
+//
+// Copyright (C) Microsoft Corporation. All rights reserved.
+//
 #pragma once
+#ifndef SIMPLEMEDIASOURCE_H
+#define SIMPLEMEDIASOURCE_H
 
-// Represents a simple media source that only has one stream and outputs a static image
+// Example Custom Property implemented by SimpleMediaSource
+// 
+// {0CE2EF73-4800-4F53-9B8E-8C06790FC0C7}
+static const GUID PROPSETID_SIMPLEMEDIASOURCE_CUSTOMCONTROL =
+{ 0xce2ef73, 0x4800, 0x4f53, { 0x9b, 0x8e, 0x8c, 0x6, 0x79, 0xf, 0xc0, 0xc7 } };
 
-#include "stdafx.h"
+enum
+{
+    KSPROPERTY_SIMPLEMEDIASOURCE_CUSTOMCONTROL_COLORMODE = 0,
+    KSPROPERTY_SIMPLEMEDIASOURCE_CUSTOMCONTROL_END  // all ids must define before this.
+};
 
-class SimpleMediaStream;
+#define KSPROPERTY_SIMPLEMEDIASOURCE_CUSTOMCONTROL_COLORMODE_GRAYSCALE        0x00FFFFFFL
+#define KSPROPERTY_SIMPLEMEDIASOURCE_CUSTOMCONTROL_COLORMODE_RED              0x00FF0000L
+#define KSPROPERTY_SIMPLEMEDIASOURCE_CUSTOMCONTROL_COLORMODE_GREEN            0x0000FF00L
+#define KSPROPERTY_SIMPLEMEDIASOURCE_CUSTOMCONTROL_COLORMODE_BLUE             0x000000FFL
 
-class __declspec(uuid("{9812588D-5CE9-4E4C-ABC1-049138D10DCE}"))
-SimpleMediaSource : public RuntimeClass<
-    RuntimeClassFlags<WinRtClassicComMix>,
-    IMFMediaEventGenerator,
-    IMFMediaSource,
-    IMFMediaSourceEx,
-    IMFGetService,
-    IKsControl>
-{
+typedef struct {
+    ULONG      StremID;
+    ULONG      ColorMode;
+} KSPROPERTY_SIMPLEMEDIASOURCE_CUSTOMCONTROL_COLORMODE_S, * PKSPROPERTY_SIMPLEMEDIASOURCE_CUSTOMCONTROL_COLORMODE_S;
 
-    enum class SourceState
+namespace winrt::WindowsSample::implementation
+{
+    // forward declaration
+    struct SimpleMediaStream;
+    
+    struct SimpleMediaSource : winrt::implements<SimpleMediaSource, IMFMediaSourceEx, IMFGetService, IKsControl, IMFSampleAllocatorControl>
     {
-        Invalid, Stopped, Started, Shutdown
-    };
+        SimpleMediaSource() = default;
 
-public:
-    // IMFMediaEventGenerator
-    IFACEMETHOD(BeginGetEvent)(_In_ IMFAsyncCallback *pCallback, _In_ IUnknown *punkState);
-    IFACEMETHOD(EndGetEvent)(_In_ IMFAsyncResult *pResult, _Out_ IMFMediaEvent **ppEvent);
-    IFACEMETHOD(GetEvent)(DWORD dwFlags, _Out_ IMFMediaEvent **ppEvent);
-    IFACEMETHOD(QueueEvent)(MediaEventType met, REFGUID guidExtendedType, HRESULT hrStatus, _In_ const PROPVARIANT *pvValue);
+    private:
+        enum class SourceState
+        {
+            Invalid, Stopped, Started, Shutdown
+        };
 
-    // IMFMediaSource
-    IFACEMETHOD(CreatePresentationDescriptor)(_Out_ IMFPresentationDescriptor **ppPresentationDescriptor);
-    IFACEMETHOD(GetCharacteristics)(_Out_ DWORD *pdwCharacteristics);
-    IFACEMETHOD(Pause)();
-    IFACEMETHOD(Shutdown)();
-    IFACEMETHOD(Start)(_In_ IMFPresentationDescriptor *pPresentationDescriptor, _In_ const GUID *pguidTimeFormat, _In_ const PROPVARIANT *pvarStartPosition);
-    IFACEMETHOD(Stop)();
+    public:
+        // IMFMediaEventGenerator (inherits by IMFMediaSource)
+        IFACEMETHODIMP BeginGetEvent(_In_ IMFAsyncCallback* pCallback, _In_ IUnknown* punkState) override;
+        IFACEMETHODIMP EndGetEvent(_In_ IMFAsyncResult* pResult, _Out_ IMFMediaEvent** ppEvent) override;
+        IFACEMETHODIMP GetEvent(DWORD dwFlags, _Out_ IMFMediaEvent** ppEvent) override;
+        IFACEMETHODIMP QueueEvent(MediaEventType met, REFGUID guidExtendedType, HRESULT hrStatus, _In_ const PROPVARIANT* pvValue) override;
 
-    // IMFMediaSourceEx
-    IFACEMETHOD(GetSourceAttributes)(_COM_Outptr_ IMFAttributes **ppAttributes);
-    IFACEMETHOD(GetStreamAttributes)(DWORD dwStreamIdentifier, _COM_Outptr_ IMFAttributes **ppAttributes);
-    IFACEMETHOD(SetD3DManager)(_In_opt_ IUnknown *pManager);
+        // IMFMediaSource (inherits by IMFMediaSourceEx)
+        IFACEMETHODIMP CreatePresentationDescriptor(_Out_ IMFPresentationDescriptor** ppPresentationDescriptor) override;
+        IFACEMETHODIMP GetCharacteristics(_Out_ DWORD* pdwCharacteristics) override;
+        IFACEMETHODIMP Pause() override;
+        IFACEMETHODIMP Shutdown() override;
+        IFACEMETHODIMP Start(_In_ IMFPresentationDescriptor* pPresentationDescriptor, _In_ const GUID* pguidTimeFormat, _In_ const PROPVARIANT* pvarStartPosition) override;
+        IFACEMETHODIMP Stop() override;
 
-    // IMFGetService
-    IFACEMETHOD(GetService)(_In_ REFGUID guidService, _In_ REFIID riid, _Out_ LPVOID *ppvObject);
+        // IMFMediaSourceEx
+        IFACEMETHODIMP GetSourceAttributes(_COM_Outptr_ IMFAttributes** ppAttributes) override;
+        IFACEMETHODIMP GetStreamAttributes(DWORD dwStreamIdentifier, _COM_Outptr_ IMFAttributes** ppAttributes) override;
+        IFACEMETHODIMP SetD3DManager(_In_opt_ IUnknown* pManager) override;
 
-    // IKsControl
-    IFACEMETHOD(KsProperty)(_In_reads_bytes_(ulPropertyLength) PKSPROPERTY pProperty,
-                            _In_ ULONG ulPropertyLength,
-                            _Inout_updates_to_(ulDataLength, *pBytesReturned) LPVOID pPropertyData,
-                            _In_ ULONG ulDataLength,
-                            _Out_ ULONG* pBytesReturned);
-    IFACEMETHOD(KsMethod)(_In_reads_bytes_(ulMethodLength) PKSMETHOD pMethod,
-                          _In_ ULONG ulMethodLength,
-                          _Inout_updates_to_(ulDataLength, *pBytesReturned) LPVOID pMethodData,
-                          _In_ ULONG ulDataLength,
-                          _Out_ ULONG* pBytesReturned);
-    IFACEMETHOD(KsEvent)(_In_reads_bytes_opt_(ulEventLength) PKSEVENT pEvent,
-                         _In_ ULONG ulEventLength,
-                         _Inout_updates_to_(ulDataLength, *pBytesReturned) LPVOID pEventData,
-                         _In_ ULONG ulDataLength,
-                         _Out_opt_ ULONG* pBytesReturned);
-public:
-    HRESULT RuntimeClassInitialize();
+        // IMFGetService
+        IFACEMETHODIMP GetService(_In_ REFGUID guidService, _In_ REFIID riid, _Out_ LPVOID* ppvObject) override;
 
-private:
-    HRESULT _CheckShutdownRequiresLock();
-    HRESULT _ValidatePresentationDescriptor(_In_ IMFPresentationDescriptor *pPresentationDescriptor);
+        // IKsControl
+        IFACEMETHODIMP KsProperty(
+            _In_reads_bytes_(ulPropertyLength) PKSPROPERTY pProperty,
+            _In_ ULONG ulPropertyLength,
+            _Inout_updates_to_(ulDataLength, *pBytesReturned) LPVOID pPropertyData,
+            _In_ ULONG ulDataLength,
+            _Out_ ULONG* pBytesReturned) override;
+        IFACEMETHODIMP KsMethod(
+            _In_reads_bytes_(ulMethodLength) PKSMETHOD pMethod,
+            _In_ ULONG ulMethodLength,
+            _Inout_updates_to_(ulDataLength, *pBytesReturned) LPVOID pMethodData,
+            _In_ ULONG ulDataLength,
+            _Out_ ULONG* pBytesReturned) override;
+        IFACEMETHODIMP KsEvent(
+            _In_reads_bytes_opt_(ulEventLength) PKSEVENT pEvent,
+            _In_ ULONG ulEventLength,
+            _Inout_updates_to_(ulDataLength, *pBytesReturned) LPVOID pEventData,
+            _In_ ULONG ulDataLength,
+            _Out_opt_ ULONG* pBytesReturned) override;
 
-    CriticalSection _critSec;
-    SourceState _sourceState{ SourceState::Invalid };
-    ComPtr<IMFMediaEventQueue> _spEventQueue;
-    ComPtr<IMFPresentationDescriptor> _spPresentationDescriptor;
-    ComPtr<IMFAttributes> _spAttributes;
+        // IMFSampleAllocatorControl
+        IFACEMETHODIMP SetDefaultAllocator(
+            _In_  DWORD dwOutputStreamID,
+            _In_  IUnknown* pAllocator) override;
 
-    bool _wasStreamPreviouslySelected; // maybe makes more sense as a property of the stream
-    const DWORD NUM_STREAMS = 1;
-    ComPtr<SimpleMediaStream> _stream;
-};
+        IFACEMETHODIMP GetAllocatorUsage(
+            _In_  DWORD dwOutputStreamID,
+            _Out_  DWORD* pdwInputStreamID,
+            _Out_  MFSampleAllocatorUsage* peUsage) override;
+
+        // Non-Interface functions
+        HRESULT Initialize();
+
+    private:
+        HRESULT _CheckShutdownRequiresLock();
+        HRESULT _ValidatePresentationDescriptor(_In_ IMFPresentationDescriptor* pPresentationDescriptor);
+        HRESULT _CreateSourceAttributes();
+        HRESULT _GetStreamDescriptorByStreamId(_In_ DWORD dwStreamId, _Out_ DWORD* pdwStreamIdx, _Out_ bool* pSelected, _COM_Outptr_ IMFStreamDescriptor** ppStreamDescriptor);
+        HRESULT _GetMediaStreamById(_In_ DWORD dwStreamId, _COM_Outptr_ SimpleMediaStream** ppMediaStream);
+
+        winrt::slim_mutex m_Lock;
+        SourceState m_sourceState{ SourceState::Invalid };
+
+        wil::com_ptr_nothrow<IMFMediaEventQueue> m_spEventQueue;
+        wil::com_ptr_nothrow<IMFPresentationDescriptor> m_spPresentationDescriptor;
+        wil::com_ptr_nothrow<IMFAttributes> m_spAttributes;
+        wil::unique_cotaskmem_array_ptr<wil::com_ptr_nothrow<SimpleMediaStream>> m_streamList;
+
+        const DWORD NUM_STREAMS = 1;
+    };
+}
 
-CoCreatableClass(SimpleMediaSource);
+#endif