Source codes

1 files changed, 192 insertions, 0 deletions

diff --git a/Examples/WhisperDesktop/AppState.cpp b/Examples/WhisperDesktop/AppState.cpp
new file mode 100644
index 0000000..6697e6a
--- /dev/null
+++ b/Examples/WhisperDesktop/AppState.cpp
@@ -0,0 +1,192 @@
+#include "stdafx.h"
+#include "AppState.h"
+#include "Utils/miscUtils.h"
+#include <commctrl.h>
+#pragma comment(lib, "Comctl32.lib")
+// #pragma comment(linker,"/manifestdependency:\"type='win32' name='Microsoft.Windows.Common-Controls' version='6.0.0.0' processorArchitecture='amd64' publicKeyToken='6595b64144ccf1df' language='*'\"")
+#include "CircleIndicator.h"
+
+namespace
+{
+	static const HKEY regKeyRoot = HKEY_CURRENT_USER;
+	const LPCTSTR regKey = LR"(SOFTWARE\const.me\WhisperDesktop)";
+	const LPCTSTR regValPath = L"modelPath";
+	const LPCTSTR regValImpl = L"modelImpl";
+	const LPCTSTR regValLang = L"language";
+	const LPCTSTR regValLastScreen = L"screen";
+
+	static HRESULT readString( CRegKey& k, LPCTSTR name, CString& rdi )
+	{
+		ULONG nChars = 0;
+		LSTATUS lss = k.QueryStringValue( name, nullptr, &nChars );
+		if( lss != ERROR_SUCCESS )
+			return HRESULT_FROM_WIN32( lss );
+		if( nChars == 0 )
+		{
+			rdi = L"";
+			return S_FALSE;
+		}
+
+		lss = k.QueryStringValue( name, rdi.GetBufferSetLength( nChars ), &nChars );
+		rdi.ReleaseBuffer();
+		if( lss != ERROR_SUCCESS )
+			return HRESULT_FROM_WIN32( lss );
+
+		return S_OK;
+	}
+
+	using Whisper::eModelImplementation;
+}
+
+HRESULT AppState::startup()
+{
+	HRESULT hr = CoInitializeEx( nullptr, COINIT_MULTITHREADED );
+	if( FAILED( hr ) )
+	{
+		reportFatalError( "CoInitializeEx failed", hr );
+		return hr;
+	}
+	coInit = true;
+
+	LSTATUS lss = registryKey.Create( regKeyRoot, regKey );
+	if( lss != ERROR_SUCCESS )
+	{
+		hr = HRESULT_FROM_WIN32( lss );
+		reportFatalError( "Unable to open the registry key", hr );
+		return hr;
+	}
+
+	INITCOMMONCONTROLSEX init;
+	init.dwSize = sizeof( init );
+	init.dwICC = ICC_LINK_CLASS | ICC_PROGRESS_CLASS | ICC_STANDARD_CLASSES | ICC_TAB_CLASSES;
+	const BOOL icc = InitCommonControlsEx( &init );
+	if( !icc )
+	{
+		reportFatalError( "InitCommonControlsEx failed", HRESULT_FROM_WIN32( GetLastError() ) );
+		return E_FAIL;
+	}
+
+	hr = initMediaFoundation( &mediaFoundation );
+	if( FAILED( hr ) )
+	{
+		reportFatalError( "Unable to initialize Media Foundation runtime", hr );
+		return hr;
+	}
+
+	hr = console.initialize();
+	if( FAILED( hr ) )
+	{
+		reportFatalError( "Unable to initialize logging", hr );
+		return hr;
+	}
+
+	hr = CircleIndicator::registerClass();
+	if( FAILED( hr ) )
+	{
+		reportFatalError( "Unable to register custom controls", hr );
+		return hr;
+	}
+	appIcon.LoadIcon( IDI_WHISPERDESKTOP );
+	return S_OK;
+}
+
+AppState::~AppState()
+{
+	if( coInit )
+	{
+		CoUninitialize();
+		coInit = false;
+	}
+}
+
+HRESULT AppState::findModelSource()
+{
+	CHECK( readString( registryKey, regValPath, source.path ) );
+
+	{
+		CAtlFile file;
+		CHECK( file.Create( source.path, GENERIC_READ, FILE_SHARE_READ, OPEN_EXISTING ) );
+		ULONGLONG len;
+		CHECK( file.GetSize( len ) );
+		source.sizeInBytes = len;
+	}
+
+	CString impl;
+	CHECK( readString( registryKey, regValImpl, impl ) );
+	CHECK( implParse( impl, source.impl ) );
+	source.found = true;
+	return S_OK;
+}
+
+HRESULT AppState::saveModelSource()
+{
+	LSTATUS lss = registryKey.SetStringValue( regValPath, source.path );
+	if( lss != ERROR_SUCCESS )
+		return HRESULT_FROM_WIN32( lss );
+
+	LPCTSTR impl = implString( source.impl );
+	if( nullptr == impl )
+		return E_INVALIDARG;
+	lss = registryKey.SetStringValue( regValImpl, impl );
+	if( lss != ERROR_SUCCESS )
+		return HRESULT_FROM_WIN32( lss );
+
+	return S_OK;
+}
+
+uint32_t AppState::languageRead()
+{
+	DWORD dw;
+	LSTATUS lss = registryKey.QueryDWORDValue( regValLang, dw );
+	if( lss == ERROR_SUCCESS )
+		return dw;
+	return UINT_MAX;
+}
+
+void AppState::languageWrite( uint32_t key )
+{
+	registryKey.SetDWORDValue( regValLang, key );
+}
+
+CString AppState::stringLoad( LPCTSTR name )
+{
+	CString res;
+	readString( registryKey, name, res );
+	return res;
+}
+void AppState::stringStore( LPCTSTR name, LPCTSTR value )
+{
+	registryKey.SetStringValue( name, value );
+}
+uint32_t AppState::dwordLoad( LPCTSTR name, uint32_t fallback )
+{
+	DWORD dw;
+	LSTATUS lss = registryKey.QueryDWORDValue( name, dw );
+	if( lss == ERROR_SUCCESS )
+		return dw;
+	return fallback;
+}
+void AppState::dwordStore( LPCTSTR name, uint32_t value )
+{
+	registryKey.SetDWORDValue( name, value );
+}
+
+void AppState::lastScreenSave( HRESULT code )
+{
+	dwordStore( regValLastScreen, (uint32_t)code );
+}
+
+HRESULT AppState::lastScreenLoad()
+{
+	return (HRESULT)dwordLoad( regValLastScreen, SCREEN_TRANSCRIBE );
+}
+
+void AppState::setupIcon( CWindow* wnd )
+{
+	HICON ic = appIcon;
+	if( nullptr != ic )
+	{
+		wnd->SendMessage( WM_SETICON, ICON_SMALL, (LPARAM)ic );
+		wnd->SendMessage( WM_SETICON, ICON_BIG, (LPARAM)ic );
+	}
+}
\ No newline at end of file