mirror of
https://github.com/microsoft/PowerToys.git
synced 2026-07-03 09:00:04 +02:00
Compare commits
39 Commits
integrate/
...
yuleng/kbm
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
c27ce19ce2 | ||
|
|
46e3215c10 | ||
|
|
f0a828ee22 | ||
|
|
b2bd24db0d | ||
|
|
28d6fe1615 | ||
|
|
4d01062f76 | ||
|
|
d7d1e543ae | ||
|
|
272b725ff0 | ||
|
|
7884f4217a | ||
|
|
92014c81b9 | ||
|
|
d57096af20 | ||
|
|
f136a4fe04 | ||
|
|
fbc1a0c3da | ||
|
|
20df1fd96e | ||
|
|
71d91a9616 | ||
|
|
bd97ba31e8 | ||
|
|
245a6db963 | ||
|
|
314f9fe751 | ||
|
|
832db1bfea | ||
|
|
412028c861 | ||
|
|
dda2a89aa6 | ||
|
|
164ac6074a | ||
|
|
09cb927356 | ||
|
|
78c0e3e131 | ||
|
|
ebc44a0e9d | ||
|
|
d88dca2c1e | ||
|
|
f893fc7a77 | ||
|
|
7c3c5514ee | ||
|
|
3a012d4cf1 | ||
|
|
9d58b1bdd1 | ||
|
|
6fd36f9579 | ||
|
|
a8b79158f1 | ||
|
|
ff578d15a3 | ||
|
|
bccceba97b | ||
|
|
e03d048e8a | ||
|
|
806ff3c07a | ||
|
|
c1ecdda60c | ||
|
|
43e530d2e1 | ||
|
|
016e0732b0 |
3
.github/actions/spell-check/allow/code.txt
vendored
3
.github/actions/spell-check/allow/code.txt
vendored
@@ -432,3 +432,6 @@ SHELLEXPERIENCEHOST
|
||||
SHELLHOST
|
||||
STARTMENUEXPERIENCEHOST
|
||||
WIDGETBOARD
|
||||
|
||||
# URIs
|
||||
actioncenter
|
||||
|
||||
212
.github/actions/spell-check/allow/zoomit.txt
vendored
212
.github/actions/spell-check/allow/zoomit.txt
vendored
@@ -1,23 +1,51 @@
|
||||
accelscroll
|
||||
acq
|
||||
ADDTO
|
||||
ADDTOOL
|
||||
adr
|
||||
Adr
|
||||
ALWAYSTIP
|
||||
APPLYTOSUBMENUS
|
||||
ARCHMASK
|
||||
archs
|
||||
AUDCLNT
|
||||
autocorr
|
||||
avx
|
||||
axisdefer
|
||||
axisflip
|
||||
axisstart
|
||||
backlight
|
||||
BEOS
|
||||
bfi
|
||||
BFIN
|
||||
bfly
|
||||
BGRX
|
||||
bitmaps
|
||||
bitrev
|
||||
blits
|
||||
Borgerding
|
||||
Borland
|
||||
breakc
|
||||
BREAKSCR
|
||||
BUFFERFLAGS
|
||||
bugzilla
|
||||
Cands
|
||||
capturepath
|
||||
cbs
|
||||
centiseconds
|
||||
cexp
|
||||
cfx
|
||||
cfy
|
||||
cgem
|
||||
cifx
|
||||
cify
|
||||
CLASSW
|
||||
coeffs
|
||||
colblocks
|
||||
constantbuffer
|
||||
coprime
|
||||
cpuid
|
||||
cpx
|
||||
CREATEDIBSECTION
|
||||
CREATESTRUCTW
|
||||
crossfades
|
||||
@@ -28,109 +56,216 @@ CTLCOLORDLG
|
||||
CTLCOLOREDIT
|
||||
CTLCOLORLISTBOX
|
||||
CTrim
|
||||
CVTEPI
|
||||
DBuffer
|
||||
dcl
|
||||
dct
|
||||
ddx
|
||||
ddy
|
||||
Deinterleave
|
||||
denoise
|
||||
denoised
|
||||
DEVSOURCE
|
||||
DFCS
|
||||
DIVSCALAR
|
||||
DJGPP
|
||||
dlg
|
||||
dlu
|
||||
dnn
|
||||
DONTCARE
|
||||
downsample
|
||||
DRAWITEM
|
||||
DRAWITEMSTRUCT
|
||||
droppedband
|
||||
Droppedband
|
||||
DSPs
|
||||
dsum
|
||||
dupburst
|
||||
dupsegments
|
||||
DWLP
|
||||
eband
|
||||
ebx
|
||||
ECX
|
||||
EDITCONTROL
|
||||
EDSP
|
||||
emmintrin
|
||||
EMX
|
||||
ENABLEHOOK
|
||||
endloop
|
||||
ENDOFSTREAM
|
||||
ener
|
||||
enh
|
||||
ettings
|
||||
expectedlock
|
||||
expf
|
||||
fabs
|
||||
fabsf
|
||||
facbuf
|
||||
fastscroll
|
||||
FDE
|
||||
ffast
|
||||
FIXDIV
|
||||
floorf
|
||||
fmadd
|
||||
fout
|
||||
fstride
|
||||
fxc
|
||||
GETCHANNELRECT
|
||||
GETCHECK
|
||||
GETCOUNT
|
||||
GETDISPINFO
|
||||
GETSCREENSAVEACTIVE
|
||||
GETSCREENSAVETIMEOUT
|
||||
GETTHUMBRECT
|
||||
GIFs
|
||||
glu
|
||||
groupshared
|
||||
gru
|
||||
hcfdark
|
||||
hcfwhitespace
|
||||
hlsl
|
||||
Hsieh
|
||||
hstride
|
||||
HTBOTTOMRIGHT
|
||||
HTHEME
|
||||
htol
|
||||
ICONINFORMATION
|
||||
ICONWARNING
|
||||
idct
|
||||
IDIn
|
||||
IDISHWND
|
||||
ifft
|
||||
igc
|
||||
ilog
|
||||
imad
|
||||
imax
|
||||
imin
|
||||
immintrin
|
||||
Inj
|
||||
interp
|
||||
inttypes
|
||||
ishl
|
||||
itof
|
||||
jumprecover
|
||||
kfft
|
||||
kheight
|
||||
kissfft
|
||||
KSDATAFORMAT
|
||||
ksize
|
||||
ktime
|
||||
lastg
|
||||
latestcapture
|
||||
ldx
|
||||
LEFTNOWORDWRAP
|
||||
legitjumps
|
||||
lenmem
|
||||
letterbox
|
||||
lld
|
||||
lldx
|
||||
llu
|
||||
llums
|
||||
logfont
|
||||
lookback
|
||||
lpc
|
||||
lpcnet
|
||||
LPNMHDR
|
||||
LPNMTTDISPINFO
|
||||
lround
|
||||
lte
|
||||
luma
|
||||
Luma
|
||||
maj
|
||||
manualdrop
|
||||
maskcache
|
||||
maxabs
|
||||
maxcorr
|
||||
MAXFACTORS
|
||||
maxperiod
|
||||
maxstep
|
||||
memalign
|
||||
memid
|
||||
memneeded
|
||||
MENUINFO
|
||||
MFSTARTUP
|
||||
mfxhw
|
||||
mic
|
||||
middledrop
|
||||
minperiod
|
||||
MIPSr
|
||||
MJPEG
|
||||
MMRESULT
|
||||
momentumreversal
|
||||
movc
|
||||
mrate
|
||||
mrt
|
||||
MULBYSCALAR
|
||||
MULC
|
||||
MWERKS
|
||||
mycfg
|
||||
narrowstrip
|
||||
nbak
|
||||
nbytes
|
||||
ncapture
|
||||
nchw
|
||||
ncm
|
||||
nduplicates
|
||||
nfft
|
||||
NHWC
|
||||
niterations
|
||||
nmonitor
|
||||
nnet
|
||||
NONCLIENTMETRICS
|
||||
NONOTIFY
|
||||
nonvle
|
||||
normf
|
||||
nredraw
|
||||
nstop
|
||||
nsubpixel
|
||||
ntorn
|
||||
numthreads
|
||||
nvw
|
||||
Octasic
|
||||
osc
|
||||
OSCE
|
||||
ovflw
|
||||
OWNERDRAW
|
||||
PBGRA
|
||||
periodictrap
|
||||
pillarbox
|
||||
pfdc
|
||||
pillarbox
|
||||
playhead
|
||||
pnmh
|
||||
pointerreuse
|
||||
PPW
|
||||
prereq
|
||||
PSHR
|
||||
pstdint
|
||||
PSWA
|
||||
pwfx
|
||||
QCONST
|
||||
qpc
|
||||
Qpc
|
||||
quantums
|
||||
qweight
|
||||
RCSEGMODEL
|
||||
RCZOOMITSCR
|
||||
readback
|
||||
READERF
|
||||
realcapture
|
||||
REFKNOWNFOLDERID
|
||||
relu
|
||||
reposted
|
||||
RETURNCMD
|
||||
rnn
|
||||
rnnoise
|
||||
rotateleft
|
||||
rsqrt
|
||||
rtcd
|
||||
RTEXT
|
||||
RTH
|
||||
rtvs
|
||||
SCALEIN
|
||||
SCALEOUT
|
||||
SCREENSAVE
|
||||
SCRNSAVE
|
||||
SCRNSAVECONFIGURE
|
||||
@@ -138,43 +273,80 @@ scrnsavw
|
||||
Scrnsavw
|
||||
scrollramp
|
||||
SCROLLSIZEGRIP
|
||||
selfie
|
||||
selftest
|
||||
SETBARCOLOR
|
||||
SETBKCOLOR
|
||||
SETDEFID
|
||||
SETRECT
|
||||
SETSCREENSAVETIMEOUT
|
||||
SETTIPSIDE
|
||||
sgem
|
||||
sgemv
|
||||
sgv
|
||||
SHAREMODE
|
||||
SHAREVIOLATION
|
||||
shortlist
|
||||
simde
|
||||
siv
|
||||
slowthenfast
|
||||
smallstart
|
||||
SNIPOCR
|
||||
softmax
|
||||
sqrtf
|
||||
SROUND
|
||||
srvs
|
||||
ssi
|
||||
startuprecovery
|
||||
stdint
|
||||
stf
|
||||
stopafter
|
||||
STREAMFLAGS
|
||||
SUBFROM
|
||||
subias
|
||||
submix
|
||||
sxx
|
||||
sxy
|
||||
symbian
|
||||
synthesising
|
||||
syy
|
||||
tallportal
|
||||
TBTS
|
||||
tci
|
||||
tcsicmp
|
||||
TEXTCALLBACK
|
||||
TEXTMETRIC
|
||||
tgsm
|
||||
THIRDPARTY
|
||||
tinystep
|
||||
tme
|
||||
toolbars
|
||||
TOOLINFO
|
||||
TRACKMOUSEEVENT
|
||||
TRIANGLELIST
|
||||
TTM
|
||||
TTN
|
||||
TWID
|
||||
UADD
|
||||
uav
|
||||
uavs
|
||||
uge
|
||||
Unadvise
|
||||
upscaled
|
||||
upscales
|
||||
USUB
|
||||
utof
|
||||
vad
|
||||
vaddq
|
||||
vaddvq
|
||||
valgrind
|
||||
Valin
|
||||
vandq
|
||||
vblank
|
||||
vcgeq
|
||||
vdup
|
||||
vectorizer
|
||||
VERTID
|
||||
VIDCAP
|
||||
vld
|
||||
vle
|
||||
@@ -184,6 +356,7 @@ vminq
|
||||
vmlal
|
||||
vmull
|
||||
vqaddq
|
||||
VSHR
|
||||
vshrn
|
||||
vsntprintf
|
||||
vsnwprintf
|
||||
@@ -194,7 +367,9 @@ WAVEFORMATEXTENSIBLE
|
||||
webcam
|
||||
Webcam
|
||||
webcams
|
||||
Wextra
|
||||
wfopen
|
||||
WGC
|
||||
wideportal
|
||||
wil
|
||||
WMU
|
||||
@@ -202,11 +377,46 @@ wrapjump
|
||||
wtol
|
||||
WTSSESSION
|
||||
WTSUn
|
||||
wxyz
|
||||
xchg
|
||||
xcorr
|
||||
XEnd
|
||||
Xfl
|
||||
Xiang
|
||||
Xiph
|
||||
xmmintrin
|
||||
xptr
|
||||
xshift
|
||||
XStart
|
||||
XStep
|
||||
xxxy
|
||||
xxyx
|
||||
xxyz
|
||||
xyw
|
||||
xywx
|
||||
xyxx
|
||||
xyxz
|
||||
xyzw
|
||||
xyzx
|
||||
xzwx
|
||||
xzxx
|
||||
Yfl
|
||||
YInternal
|
||||
yshift
|
||||
YUV
|
||||
yyyx
|
||||
yyzw
|
||||
yzw
|
||||
yzwy
|
||||
yzyy
|
||||
Zhou
|
||||
Zhu
|
||||
ZMBS
|
||||
zncc
|
||||
Zncc
|
||||
ZNCC
|
||||
zrh
|
||||
zwzz
|
||||
zyzw
|
||||
zzwz
|
||||
zzzw
|
||||
|
||||
4
.github/actions/spell-check/excludes.txt
vendored
4
.github/actions/spell-check/excludes.txt
vendored
@@ -112,8 +112,6 @@
|
||||
^src/modules/cmdpal/Microsoft\.CmdPal\.UI/Settings/InternalPage\.SampleData\.cs$
|
||||
^src/modules/cmdpal/Tests/Microsoft\.CmdPal\.Common\.UnitTests/.*\.TestData\.cs$
|
||||
^src/modules/cmdpal/Tests/Microsoft\.CmdPal\.Common\.UnitTests/Text/.*\.cs$
|
||||
^src/modules/cmdpal/Tests/Microsoft\.CommandPalette\.Extensions\.Toolkit\.UnitTests/FuzzyMatcherComparisonTests.cs$
|
||||
^src/modules/cmdpal/Tests/Microsoft\.CommandPalette\.Extensions\.Toolkit\.UnitTests/FuzzyMatcherDiacriticsTests.cs$
|
||||
^src/modules/colorPicker/ColorPickerUI/Shaders/GridShader\.cso$
|
||||
^src/modules/launcher/Plugins/Microsoft\.PowerToys\.Run\.Plugin\.TimeDate/Properties/
|
||||
^src/modules/MouseUtils/MouseJumpUI/MainForm\.resx$
|
||||
@@ -137,6 +135,8 @@
|
||||
^src/modules/previewpane/SvgPreviewHandler/SvgHTMLPreviewGenerator\.cs$
|
||||
^src/modules/previewpane/UnitTests-MarkdownPreviewHandler/HelperFiles/MarkdownWithHTMLImageTag\.txt$
|
||||
^src/modules/registrypreview/RegistryPreviewUILib/Controls/HexBox/.*$
|
||||
^src/modules/ZoomIt/ZoomIt/rnnoise/
|
||||
^src/modules/ZoomIt/ZoomIt/selfie_segmentation\.onnx$
|
||||
^src/modules/ZoomIt/ZoomIt/ZoomIt\.idc$
|
||||
^src/Monaco/
|
||||
^tools/project_template/ModuleTemplate/resource\.h$
|
||||
|
||||
9
.github/actions/spell-check/expect.txt
vendored
9
.github/actions/spell-check/expect.txt
vendored
@@ -113,7 +113,6 @@ azman
|
||||
azureaiinference
|
||||
azureinference
|
||||
azureopenai
|
||||
Backlight
|
||||
backticks
|
||||
Badflags
|
||||
Badmode
|
||||
@@ -472,7 +471,6 @@ DWMWINDOWATTRIBUTE
|
||||
DWMWINDOWMAXIMIZEDCHANGE
|
||||
DWORDLONG
|
||||
dworigin
|
||||
DWRITE
|
||||
dxgi
|
||||
Dxva
|
||||
eab
|
||||
@@ -998,7 +996,6 @@ luid
|
||||
lusrmgr
|
||||
LVDS
|
||||
LWA
|
||||
LWIN
|
||||
LZero
|
||||
MAGTRANSFORM
|
||||
makeappx
|
||||
@@ -1208,7 +1205,6 @@ nonclient
|
||||
NONCLIENTMETRICSW
|
||||
NONELEVATED
|
||||
nonspace
|
||||
nonstd
|
||||
NOOWNERZORDER
|
||||
NOPARENTNOTIFY
|
||||
NOPREFIX
|
||||
@@ -1400,6 +1396,7 @@ POWERRENAMECONTEXTMENU
|
||||
powerrenameinput
|
||||
POWERRENAMETEST
|
||||
POWERTOYNAME
|
||||
powertoyscli
|
||||
powertoyssetup
|
||||
powertoysusersetup
|
||||
Powrprof
|
||||
@@ -1422,7 +1419,6 @@ Prefixer
|
||||
Premul
|
||||
prependpath
|
||||
prepopulate
|
||||
Prereq
|
||||
prevhost
|
||||
previewer
|
||||
PREVIEWHANDLERFRAMEINFO
|
||||
@@ -1561,6 +1557,7 @@ RESIZETOFIT
|
||||
resmimetype
|
||||
RESOURCEID
|
||||
RESTORETOMAXIMIZED
|
||||
retargets
|
||||
RETURNONLYFSDIRS
|
||||
Revalidates
|
||||
RGBQUAD
|
||||
@@ -2000,7 +1997,6 @@ valuegenerator
|
||||
VARTYPE
|
||||
vbcscompiler
|
||||
vcamp
|
||||
VCENTER
|
||||
vcgtq
|
||||
VCINSTALLDIR
|
||||
vcp
|
||||
@@ -2038,7 +2034,6 @@ vorrq
|
||||
VOS
|
||||
vpaddlq
|
||||
vqsubq
|
||||
VREDRAW
|
||||
vreinterpretq
|
||||
VSC
|
||||
VSCBD
|
||||
|
||||
6
.github/actions/spell-check/patterns.txt
vendored
6
.github/actions/spell-check/patterns.txt
vendored
@@ -312,3 +312,9 @@ ms-windows-store://\S+
|
||||
|
||||
# ANSI color codes
|
||||
(?:\\(?:u00|x)1[Bb]|\\03[1-7]|\x1b|\\u\{1[Bb]\})\[\d+(?:;\d+)*m
|
||||
|
||||
# Special licenses text from RNNoise (BSD-style disclaimer: ``AS IS'')
|
||||
``AS IS''
|
||||
|
||||
# Old school moniker for macOS from RNNoise
|
||||
MacOS
|
||||
|
||||
3
.gitignore
vendored
3
.gitignore
vendored
@@ -19,6 +19,9 @@
|
||||
[Rr]eleases/
|
||||
x64/
|
||||
x86/
|
||||
!**/rnnoise/
|
||||
!**/rnnoise/x86/
|
||||
!**/rnnoise/x86/**
|
||||
ARM64/
|
||||
bld/
|
||||
[Bb]in/
|
||||
|
||||
@@ -500,6 +500,10 @@
|
||||
<Platform Solution="*|ARM64" Project="ARM64" />
|
||||
<Platform Solution="*|x64" Project="x64" />
|
||||
</Project>
|
||||
<Project Path="src/modules/keyboardmanager/KeyboardManagerEditorUI.UnitTests/KeyboardManagerEditorUI.UnitTests.csproj">
|
||||
<Platform Solution="*|ARM64" Project="ARM64" />
|
||||
<Platform Solution="*|x64" Project="x64" />
|
||||
</Project>
|
||||
<Project Path="src/modules/keyboardmanager/KeyboardManagerEngine/KeyboardManagerEngine.vcxproj" Id="ba661f5b-1d5a-4ffc-9bf1-fc39df280bdd" />
|
||||
<Project Path="src/modules/keyboardmanager/KeyboardManagerEngineLibrary/KeyboardManagerEngineLibrary.vcxproj" Id="e496b7fc-1e99-4bab-849b-0e8367040b02" />
|
||||
</Folder>
|
||||
|
||||
Binary file not shown.
|
Before Width: | Height: | Size: 258 KiB After Width: | Height: | Size: 134 KiB |
@@ -73,7 +73,8 @@
|
||||
<PrecompiledHeaderOutputFile>$(IntDir)pch.pch</PrecompiledHeaderOutputFile>
|
||||
<WarningLevel>Level4</WarningLevel>
|
||||
<AdditionalOptions>%(AdditionalOptions) /bigobj</AdditionalOptions>
|
||||
<PreprocessorDefinitions>_WINRT_DLL;WINRT_LEAN_AND_MEAN;%(PreprocessorDefinitions)</PreprocessorDefinitions>
|
||||
<!-- TODO: _SILENCE_EXPERIMENTAL_COROUTINE_DEPRECATION_WARNINGS: suppress VS 2026 STL hard error for <experimental/coroutine> until the code is ported to <coroutine> -->
|
||||
<PreprocessorDefinitions>_WINRT_DLL;WINRT_LEAN_AND_MEAN;_SILENCE_EXPERIMENTAL_COROUTINE_DEPRECATION_WARNINGS;%(PreprocessorDefinitions)</PreprocessorDefinitions>
|
||||
<AdditionalIncludeDirectories>../../..;%(AdditionalIncludeDirectories)</AdditionalIncludeDirectories>
|
||||
<AdditionalUsingDirectories>$(WindowsSDK_WindowsMetadata);$(AdditionalUsingDirectories)</AdditionalUsingDirectories>
|
||||
</ClCompile>
|
||||
|
||||
@@ -102,19 +102,18 @@ namespace ShortcutGuide.Helpers
|
||||
/// <summary>
|
||||
/// Retrieves all application IDs that should be displayed, based on the foreground window and background processes.
|
||||
/// </summary>
|
||||
/// <param name="foregroundWindowHandle">The window handle captured before Shortcut Guide UI takes focus.</param>
|
||||
/// <returns>
|
||||
/// A dictionary mapping each application ID to the full path of the executable
|
||||
/// that caused the match (used for icon extraction), or <c>null</c> when no
|
||||
/// specific executable is associated (for example, wildcard filters like the
|
||||
/// default shell).
|
||||
/// </returns>
|
||||
public static Dictionary<string, string?> GetAllCurrentApplicationIds()
|
||||
public static Dictionary<string, string?> GetAllCurrentApplicationIds(nint foregroundWindowHandle)
|
||||
{
|
||||
nint handle = NativeMethods.GetForegroundWindow();
|
||||
|
||||
Dictionary<string, string?> applicationIds = new(StringComparer.Ordinal);
|
||||
|
||||
if (NativeMethods.GetWindowThreadProcessId(handle, out uint processId) > 0)
|
||||
if (NativeMethods.GetWindowThreadProcessId(foregroundWindowHandle, out uint processId) > 0)
|
||||
{
|
||||
string? name = null;
|
||||
string? executablePath = null;
|
||||
|
||||
@@ -21,9 +21,12 @@ namespace ShortcutGuide
|
||||
{
|
||||
public static Thread CopyAndIndexGenerationThread { get; private set; } = null!;
|
||||
|
||||
public static nint ForegroundWindowHandle { get; private set; } = nint.Zero;
|
||||
|
||||
[STAThread]
|
||||
public static void Main(string[] args)
|
||||
{
|
||||
ForegroundWindowHandle = NativeMethods.GetForegroundWindow();
|
||||
Logger.InitializeLogger("\\ShortcutGuide\\Logs");
|
||||
|
||||
// The module interface passes: <powertoys_pid> [telemetry]
|
||||
|
||||
@@ -53,7 +53,7 @@ namespace ShortcutGuide
|
||||
_getAppIdsTask = Task.Run(() =>
|
||||
{
|
||||
Program.CopyAndIndexGenerationThread.Join();
|
||||
_currentApplicationIds = ManifestInterpreter.GetAllCurrentApplicationIds();
|
||||
_currentApplicationIds = ManifestInterpreter.GetAllCurrentApplicationIds(Program.ForegroundWindowHandle);
|
||||
return _currentApplicationIds;
|
||||
});
|
||||
|
||||
|
||||
@@ -34,15 +34,17 @@ namespace winrt
|
||||
using namespace Windows::Devices::Enumeration;
|
||||
}
|
||||
|
||||
AudioSampleGenerator::AudioSampleGenerator(bool captureMicrophone, bool captureSystemAudio, bool micMonoMix)
|
||||
AudioSampleGenerator::AudioSampleGenerator(bool captureMicrophone, bool captureSystemAudio, bool mixMicrophoneMono, bool useNoiseCancellation)
|
||||
: m_captureMicrophone(captureMicrophone)
|
||||
, m_captureSystemAudio(captureSystemAudio)
|
||||
, m_micMonoMix(micMonoMix)
|
||||
, m_mixMicrophoneMono(mixMicrophoneMono)
|
||||
, m_useNoiseCancellation(useNoiseCancellation)
|
||||
{
|
||||
OutputDebugStringA(("AudioSampleGenerator created, captureMicrophone=" +
|
||||
std::string(captureMicrophone ? "true" : "false") +
|
||||
", captureSystemAudio=" + std::string(captureSystemAudio ? "true" : "false") +
|
||||
", micMonoMix=" + std::string(micMonoMix ? "true" : "false") + "\n").c_str());
|
||||
", mixMicrophoneMono=" + std::string(mixMicrophoneMono ? "true" : "false") +
|
||||
", useNoiseCancellation=" + std::string(useNoiseCancellation ? "true" : "false") + "\n").c_str());
|
||||
m_audioEvent.create(wil::EventOptions::ManualReset);
|
||||
m_endEvent.create(wil::EventOptions::ManualReset);
|
||||
m_startEvent.create(wil::EventOptions::ManualReset);
|
||||
@@ -158,8 +160,24 @@ winrt::IAsyncAction AudioSampleGenerator::InitializeAsync()
|
||||
throw winrt::hresult_error(E_FAIL, L"Failed to initialize loopback audio capture!");
|
||||
}
|
||||
|
||||
// Initialize noise suppressor for microphone audio if enabled
|
||||
if (m_useNoiseCancellation && m_captureMicrophone)
|
||||
{
|
||||
m_noiseSuppressor = std::make_unique<NoiseSuppressor>();
|
||||
OutputDebugStringA("Noise cancellation enabled for microphone\n");
|
||||
}
|
||||
|
||||
m_audioGraph.QuantumStarted({ this, &AudioSampleGenerator::OnAudioQuantumStarted });
|
||||
|
||||
// Start the AudioGraph now so the microphone device begins warming up
|
||||
// during the remaining recording initialization (transcoder setup, etc.).
|
||||
// OnAudioQuantumStarted returns early while m_started is false, so audio
|
||||
// samples are discarded until Start() is called. The side-effect of
|
||||
// starting the graph early is that the system mic-active icon appears
|
||||
// sooner, which also triggers a desktop-content change that helps
|
||||
// unblock the WGC frame pool wait in OnMediaStreamSourceStarting.
|
||||
m_audioGraph.Start();
|
||||
|
||||
m_asyncInitialized.SetEvent();
|
||||
}
|
||||
}
|
||||
@@ -205,67 +223,65 @@ std::optional<winrt::MediaStreamSample> AudioSampleGenerator::TryGetNextSample()
|
||||
}
|
||||
}
|
||||
|
||||
// Wait for audio samples to become available, retrying on spurious wakes
|
||||
// (e.g. when OnAudioQuantumStarted signals m_audioEvent but the quantum
|
||||
// produced an empty buffer so m_samples is still empty).
|
||||
for (;;)
|
||||
{
|
||||
auto lock = m_lock.lock_exclusive();
|
||||
if (m_samples.empty() && m_endEvent.is_signaled())
|
||||
{
|
||||
auto lock = m_lock.lock_exclusive();
|
||||
if (m_samples.empty() && m_endEvent.is_signaled())
|
||||
{
|
||||
return std::nullopt;
|
||||
}
|
||||
else if (!m_samples.empty())
|
||||
{
|
||||
std::optional result(m_samples.front());
|
||||
m_samples.pop_front();
|
||||
return result;
|
||||
}
|
||||
}
|
||||
|
||||
m_audioEvent.ResetEvent();
|
||||
std::vector<HANDLE> events = { m_endEvent.get(), m_audioEvent.get() };
|
||||
auto waitResult = WaitForMultipleObjectsEx(static_cast<DWORD>(events.size()), events.data(), false, INFINITE, false);
|
||||
auto eventIndex = -1;
|
||||
switch (waitResult)
|
||||
{
|
||||
case WAIT_OBJECT_0:
|
||||
case WAIT_OBJECT_0 + 1:
|
||||
eventIndex = waitResult - WAIT_OBJECT_0;
|
||||
break;
|
||||
}
|
||||
WINRT_VERIFY(eventIndex >= 0);
|
||||
|
||||
auto signaledEvent = events[eventIndex];
|
||||
if (signaledEvent == m_endEvent.get())
|
||||
{
|
||||
// End was signaled, but check for any remaining samples before returning nullopt
|
||||
auto lock = m_lock.lock_exclusive();
|
||||
if (!m_samples.empty())
|
||||
{
|
||||
std::optional result(m_samples.front());
|
||||
m_samples.pop_front();
|
||||
return result;
|
||||
}
|
||||
return std::nullopt;
|
||||
}
|
||||
else if (!m_samples.empty())
|
||||
{
|
||||
std::optional result(m_samples.front());
|
||||
m_samples.pop_front();
|
||||
return result;
|
||||
}
|
||||
}
|
||||
|
||||
m_audioEvent.ResetEvent();
|
||||
std::vector<HANDLE> events = { m_endEvent.get(), m_audioEvent.get() };
|
||||
auto waitResult = WaitForMultipleObjectsEx(static_cast<DWORD>(events.size()), events.data(), false, INFINITE, false);
|
||||
auto eventIndex = -1;
|
||||
switch (waitResult)
|
||||
{
|
||||
case WAIT_OBJECT_0:
|
||||
case WAIT_OBJECT_0 + 1:
|
||||
eventIndex = waitResult - WAIT_OBJECT_0;
|
||||
break;
|
||||
}
|
||||
WINRT_VERIFY(eventIndex >= 0);
|
||||
|
||||
auto signaledEvent = events[eventIndex];
|
||||
if (signaledEvent == m_endEvent.get())
|
||||
{
|
||||
// End was signaled, but check for any remaining samples before returning nullopt
|
||||
auto lock = m_lock.lock_exclusive();
|
||||
if (!m_samples.empty())
|
||||
{
|
||||
std::optional result(m_samples.front());
|
||||
m_samples.pop_front();
|
||||
return result;
|
||||
}
|
||||
return std::nullopt;
|
||||
}
|
||||
else
|
||||
{
|
||||
auto lock = m_lock.lock_exclusive();
|
||||
if (m_samples.empty())
|
||||
{
|
||||
// Spurious wake or race - no samples available
|
||||
// If end is signaled, return nullopt
|
||||
return m_endEvent.is_signaled() ? std::nullopt : std::optional<winrt::MediaStreamSample>{};
|
||||
}
|
||||
std::optional result(m_samples.front());
|
||||
m_samples.pop_front();
|
||||
return result;
|
||||
// m_audioEvent was signaled — loop back to check m_samples again.
|
||||
// If the quantum produced an empty buffer, m_samples will still be
|
||||
// empty and we'll wait for the next quantum.
|
||||
}
|
||||
}
|
||||
|
||||
void AudioSampleGenerator::Start()
|
||||
void AudioSampleGenerator::Start(int64_t videoStartTimestamp)
|
||||
{
|
||||
CheckInitialized();
|
||||
m_videoStartTimestamp = videoStartTimestamp;
|
||||
auto expected = false;
|
||||
if (m_started.compare_exchange_strong(expected, true))
|
||||
{
|
||||
OutputDebugStringW( L"[AudioGen] Start(): m_started set to true, setting m_startEvent\n" );
|
||||
m_endEvent.ResetEvent();
|
||||
m_startEvent.SetEvent();
|
||||
|
||||
@@ -284,7 +300,7 @@ void AudioSampleGenerator::Start()
|
||||
m_loopbackCapture->Start();
|
||||
}
|
||||
|
||||
m_audioGraph.Start();
|
||||
// AudioGraph was already started in InitializeAsync for mic warmup.
|
||||
}
|
||||
}
|
||||
|
||||
@@ -611,12 +627,21 @@ void AudioSampleGenerator::CombineQueuedSamples()
|
||||
|
||||
void AudioSampleGenerator::OnAudioQuantumStarted(winrt::AudioGraph const& sender, winrt::IInspectable const& args)
|
||||
{
|
||||
// Don't process if we're not actively recording
|
||||
// Don't process if we're not actively recording, but DO drain the
|
||||
// output node so stale audio doesn't accumulate during mic warmup.
|
||||
// Without this, the first GetFrame() after m_started becomes true
|
||||
// would return several seconds of buffered audio, confusing the
|
||||
// transcoder's A/V interleaving.
|
||||
if (!m_started.load())
|
||||
{
|
||||
auto frame = m_audioOutputNode.GetFrame();
|
||||
(void)frame; // discard
|
||||
return;
|
||||
}
|
||||
|
||||
static int s_quantumCount = 0;
|
||||
s_quantumCount++;
|
||||
|
||||
{
|
||||
auto lock = m_lock.lock_exclusive();
|
||||
|
||||
@@ -628,6 +653,14 @@ void AudioSampleGenerator::OnAudioQuantumStarted(winrt::AudioGraph const& sender
|
||||
auto sampleBuffer = winrt::Buffer::CreateCopyFromMemoryBuffer(audioBuffer);
|
||||
sampleBuffer.Length(audioBuffer.Length());
|
||||
|
||||
if( s_quantumCount <= 5 )
|
||||
{
|
||||
wchar_t dbg[256];
|
||||
swprintf_s( dbg, L"[AudioGen] quantum #%d: audioBuffer.Length=%u sampleBuffer.Length=%u started=%d\n",
|
||||
s_quantumCount, audioBuffer.Length(), sampleBuffer.Length(), m_started.load() ? 1 : 0 );
|
||||
OutputDebugStringW( dbg );
|
||||
}
|
||||
|
||||
// Calculate expected samples per quantum (~10ms at graph sample rate)
|
||||
// AudioGraph uses 10ms quantums by default
|
||||
uint32_t expectedSamplesPerQuantum = (m_graphSampleRate / 100) * m_graphChannels;
|
||||
@@ -636,7 +669,7 @@ void AudioSampleGenerator::OnAudioQuantumStarted(winrt::AudioGraph const& sender
|
||||
// Apply mono mixing to microphone audio if enabled
|
||||
// This converts stereo mic input (with same signal on both channels) to true mono
|
||||
// by averaging the channels and writing the result to both channels
|
||||
if (m_micMonoMix && m_captureMicrophone && numMicSamples > 0 && m_graphChannels >= 2)
|
||||
if (m_mixMicrophoneMono && m_captureMicrophone && numMicSamples > 0 && m_graphChannels >= 2)
|
||||
{
|
||||
float* micData = reinterpret_cast<float*>(sampleBuffer.data());
|
||||
uint32_t numFrames = numMicSamples / m_graphChannels;
|
||||
@@ -656,6 +689,12 @@ void AudioSampleGenerator::OnAudioQuantumStarted(winrt::AudioGraph const& sender
|
||||
}
|
||||
}
|
||||
}
|
||||
// Apply noise suppression to microphone audio before mixing with loopback
|
||||
if (m_noiseSuppressor && m_captureMicrophone && numMicSamples > 0)
|
||||
{
|
||||
float* micData = reinterpret_cast<float*>(sampleBuffer.data());
|
||||
m_noiseSuppressor->Process(micData, numMicSamples, m_graphChannels);
|
||||
}
|
||||
|
||||
// Drain loopback samples regardless of whether we have mic audio
|
||||
if (m_loopbackCapture)
|
||||
@@ -733,13 +772,25 @@ void AudioSampleGenerator::OnAudioQuantumStarted(winrt::AudioGraph const& sender
|
||||
|
||||
if (sampleBuffer.Length() > 0)
|
||||
{
|
||||
auto sample = winrt::MediaStreamSample::CreateFromBuffer(sampleBuffer, timestamp.value());
|
||||
// Rebase audio timestamps to the video's SystemRelativeTime domain.
|
||||
// AudioGraph RelativeTime starts near 0 (or a few hundred ms after
|
||||
// warmup draining), while video uses absolute SRT (~hours since boot).
|
||||
// Without rebasing, the transcoder sees audio far behind video and
|
||||
// starves video while trying to fill the gap with audio.
|
||||
if (!m_hasTimestampOffset && timestamp.has_value())
|
||||
{
|
||||
m_timestampOffset = m_videoStartTimestamp - timestamp.value().count();
|
||||
m_hasTimestampOffset = true;
|
||||
}
|
||||
auto adjustedTs = winrt::TimeSpan{ timestamp.value().count() + m_timestampOffset };
|
||||
|
||||
auto sample = winrt::MediaStreamSample::CreateFromBuffer(sampleBuffer, adjustedTs);
|
||||
m_samples.push_back(sample);
|
||||
|
||||
const uint32_t sampleCount = sampleBuffer.Length() / sizeof(float);
|
||||
const uint32_t frames = (m_graphChannels > 0) ? (sampleCount / m_graphChannels) : 0;
|
||||
const int64_t durationTicks = (m_graphSampleRate > 0) ? (static_cast<int64_t>(frames) * 10000000LL / m_graphSampleRate) : 0;
|
||||
m_lastSampleTimestamp = timestamp.value();
|
||||
m_lastSampleTimestamp = adjustedTs;
|
||||
m_lastSampleDuration = winrt::TimeSpan{ durationTicks };
|
||||
m_hasLastSampleTimestamp = true;
|
||||
}
|
||||
|
||||
@@ -1,18 +1,23 @@
|
||||
#pragma once
|
||||
|
||||
#include "LoopbackCapture.h"
|
||||
#include "NoiseSuppressor.h"
|
||||
|
||||
#include <deque>
|
||||
#include <optional>
|
||||
#include <memory>
|
||||
|
||||
class AudioSampleGenerator
|
||||
{
|
||||
public:
|
||||
AudioSampleGenerator(bool captureMicrophone = true, bool captureSystemAudio = true, bool micMonoMix = false);
|
||||
AudioSampleGenerator(bool captureMicrophone = true, bool captureSystemAudio = true, bool mixMicrophoneMono = false, bool useNoiseCancellation = false);
|
||||
~AudioSampleGenerator();
|
||||
|
||||
winrt::Windows::Foundation::IAsyncAction InitializeAsync();
|
||||
winrt::Windows::Media::MediaProperties::AudioEncodingProperties GetEncodingProperties();
|
||||
|
||||
std::optional<winrt::Windows::Media::Core::MediaStreamSample> TryGetNextSample();
|
||||
void Start();
|
||||
void Start(int64_t videoStartTimestamp = 0);
|
||||
void Stop();
|
||||
|
||||
private:
|
||||
@@ -70,5 +75,14 @@ private:
|
||||
std::atomic<bool> m_started = false;
|
||||
bool m_captureMicrophone = true;
|
||||
bool m_captureSystemAudio = true;
|
||||
bool m_micMonoMix = false;
|
||||
};
|
||||
bool m_mixMicrophoneMono = false;
|
||||
bool m_useNoiseCancellation = false;
|
||||
std::unique_ptr<NoiseSuppressor> m_noiseSuppressor;
|
||||
|
||||
// Timestamp rebasing: audio RelativeTime → video SystemRelativeTime domain.
|
||||
// Without this, the transcoder sees audio timestamps (~350ms) far below video
|
||||
// timestamps (~111 billion ticks) and starves video while trying to fill the gap.
|
||||
int64_t m_videoStartTimestamp = 0;
|
||||
int64_t m_timestampOffset = 0;
|
||||
bool m_hasTimestampOffset = false;
|
||||
};
|
||||
|
||||
859
src/modules/ZoomIt/ZoomIt/BackgroundBlur.cpp
Normal file
859
src/modules/ZoomIt/ZoomIt/BackgroundBlur.cpp
Normal file
@@ -0,0 +1,859 @@
|
||||
//==============================================================================
|
||||
//
|
||||
// BackgroundBlur.cpp
|
||||
//
|
||||
// Windows ML-based person segmentation and background blur for the
|
||||
// webcam overlay. Uses the built-in Windows.AI.MachineLearning API
|
||||
// to load an ONNX segmentation model (e.g. MediaPipe SelfieSegmentation)
|
||||
// and produce a per-pixel person mask, then blurs or replaces the
|
||||
// background using an iterated box blur or a user-chosen image.
|
||||
//
|
||||
// Copyright (C) Mark Russinovich
|
||||
// Sysinternals - www.sysinternals.com
|
||||
//
|
||||
//==============================================================================
|
||||
#include "pch.h"
|
||||
#include "BackgroundBlur.h"
|
||||
#include <algorithm>
|
||||
#include <cstring>
|
||||
#include <wincodec.h>
|
||||
#include <wil/com.h>
|
||||
|
||||
namespace winml = winrt::Windows::AI::MachineLearning;
|
||||
namespace wf = winrt::Windows::Foundation::Collections;
|
||||
|
||||
// Defined in Zoomit.cpp; compiles to nothing in Release builds.
|
||||
void OutputDebug(const TCHAR* format, ...);
|
||||
|
||||
//----------------------------------------------------------------------------
|
||||
// BackgroundBlur::Initialize
|
||||
//
|
||||
// Loads the ONNX segmentation model via Windows ML and inspects its
|
||||
// input/output tensor shapes to auto-configure preprocessing.
|
||||
//----------------------------------------------------------------------------
|
||||
bool BackgroundBlur::Initialize( const wchar_t* modelPath )
|
||||
{
|
||||
try
|
||||
{
|
||||
// Load the model from file.
|
||||
m_model = winml::LearningModel::LoadFromFilePath( modelPath );
|
||||
|
||||
// Try GPU (DirectML) first for faster inference; fall back to CPU
|
||||
// if no suitable GPU is available.
|
||||
try
|
||||
{
|
||||
winml::LearningModelDevice gpuDevice( winml::LearningModelDeviceKind::DirectXHighPerformance );
|
||||
m_session = winml::LearningModelSession( m_model, gpuDevice );
|
||||
m_usingGpu = true;
|
||||
OutputDebug( L"[BackgroundBlur] Using DirectML (GPU) for inference\n" );
|
||||
}
|
||||
catch( ... )
|
||||
{
|
||||
winml::LearningModelDevice cpuDevice( winml::LearningModelDeviceKind::Cpu );
|
||||
m_session = winml::LearningModelSession( m_model, cpuDevice );
|
||||
m_usingGpu = false;
|
||||
OutputDebug( L"[BackgroundBlur] GPU unavailable, falling back to CPU\n" );
|
||||
}
|
||||
m_binding = winml::LearningModelBinding( m_session );
|
||||
|
||||
// Get input feature descriptor.
|
||||
auto inputFeatures = m_model.InputFeatures();
|
||||
if( inputFeatures.Size() == 0 )
|
||||
{
|
||||
OutputDebug( L"[BackgroundBlur] Model has no input features\n" );
|
||||
return false;
|
||||
}
|
||||
auto inputDesc = inputFeatures.GetAt( 0 );
|
||||
m_inputName = inputDesc.Name();
|
||||
|
||||
// Inspect input tensor shape.
|
||||
auto tensorDesc = inputDesc.as<winml::ITensorFeatureDescriptor>();
|
||||
auto shape = tensorDesc.Shape();
|
||||
if( shape.Size() == 4 )
|
||||
{
|
||||
if( shape.GetAt( 1 ) == 3 || shape.GetAt( 1 ) == 1 )
|
||||
{
|
||||
// NCHW layout.
|
||||
m_inputIsNchw = true;
|
||||
m_modelInputChannels = shape.GetAt( 1 );
|
||||
m_modelInputHeight = shape.GetAt( 2 ) > 0 ? shape.GetAt( 2 ) : 256;
|
||||
m_modelInputWidth = shape.GetAt( 3 ) > 0 ? shape.GetAt( 3 ) : 256;
|
||||
}
|
||||
else
|
||||
{
|
||||
// NHWC layout.
|
||||
m_inputIsNchw = false;
|
||||
m_modelInputHeight = shape.GetAt( 1 ) > 0 ? shape.GetAt( 1 ) : 256;
|
||||
m_modelInputWidth = shape.GetAt( 2 ) > 0 ? shape.GetAt( 2 ) : 256;
|
||||
m_modelInputChannels = shape.GetAt( 3 );
|
||||
}
|
||||
}
|
||||
|
||||
// Get output feature name.
|
||||
auto outputFeatures = m_model.OutputFeatures();
|
||||
if( outputFeatures.Size() == 0 )
|
||||
{
|
||||
OutputDebug( L"[BackgroundBlur] Model has no output features\n" );
|
||||
return false;
|
||||
}
|
||||
m_outputName = outputFeatures.GetAt( 0 ).Name();
|
||||
|
||||
OutputDebug( L"[BackgroundBlur] Model loaded: input=%s %lldx%lld (ch=%lld, %s)\n",
|
||||
m_inputName.c_str(), m_modelInputWidth, m_modelInputHeight,
|
||||
m_modelInputChannels, m_inputIsNchw ? L"NCHW" : L"NHWC" );
|
||||
|
||||
// Pre-allocate input tensor buffer.
|
||||
m_inputTensor.resize( static_cast<size_t>( m_modelInputChannels * m_modelInputHeight * m_modelInputWidth ) );
|
||||
|
||||
return true;
|
||||
}
|
||||
catch( winrt::hresult_error const& ex )
|
||||
{
|
||||
OutputDebug( L"[BackgroundBlur] Initialize failed: %s (0x%08X)\n", ex.message().c_str(), ex.code().value );
|
||||
m_session = nullptr;
|
||||
m_model = nullptr;
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
//----------------------------------------------------------------------------
|
||||
// BackgroundBlur::RunSegmentation
|
||||
//
|
||||
// Resizes the BGRA frame to the model's expected input size, converts
|
||||
// to float RGB, runs inference via Windows ML, and produces a float mask
|
||||
// in m_mask where 1.0 = person, 0.0 = background.
|
||||
//----------------------------------------------------------------------------
|
||||
bool BackgroundBlur::RunSegmentation( const uint8_t* bgraPixels, uint32_t width, uint32_t height,
|
||||
bool modelResOnly )
|
||||
{
|
||||
const int64_t mW = m_modelInputWidth;
|
||||
const int64_t mH = m_modelInputHeight;
|
||||
const int64_t mC = m_modelInputChannels;
|
||||
|
||||
// Resize BGRA → model-sized float RGB using nearest-neighbor.
|
||||
for( int64_t y = 0; y < mH; y++ )
|
||||
{
|
||||
uint32_t srcY = static_cast<uint32_t>( y * height / mH );
|
||||
for( int64_t x = 0; x < mW; x++ )
|
||||
{
|
||||
uint32_t srcX = static_cast<uint32_t>( x * width / mW );
|
||||
const uint8_t* px = bgraPixels + ( static_cast<size_t>( srcY ) * width + srcX ) * 4;
|
||||
float b = px[0] / 255.0f;
|
||||
float g = px[1] / 255.0f;
|
||||
float r = px[2] / 255.0f;
|
||||
|
||||
if( m_inputIsNchw )
|
||||
{
|
||||
m_inputTensor[static_cast<size_t>(0ll * mH * mW + y * mW + x)] = r;
|
||||
if( mC > 1 ) m_inputTensor[static_cast<size_t>(1ll * mH * mW + y * mW + x)] = g;
|
||||
if( mC > 2 ) m_inputTensor[static_cast<size_t>(2ll * mH * mW + y * mW + x)] = b;
|
||||
}
|
||||
else
|
||||
{
|
||||
size_t idx = static_cast<size_t>( (y * mW + x) * mC );
|
||||
m_inputTensor[idx + 0] = r;
|
||||
if( mC > 1 ) m_inputTensor[idx + 1] = g;
|
||||
if( mC > 2 ) m_inputTensor[idx + 2] = b;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
try
|
||||
{
|
||||
// Create the input tensor shape.
|
||||
std::vector<int64_t> inputShape;
|
||||
if( m_inputIsNchw )
|
||||
inputShape = { 1, mC, mH, mW };
|
||||
else
|
||||
inputShape = { 1, mH, mW, mC };
|
||||
|
||||
// Create a TensorFloat from our data.
|
||||
auto inputTensor = winml::TensorFloat::CreateFromArray(
|
||||
inputShape, winrt::array_view<const float>( m_inputTensor.data(),
|
||||
m_inputTensor.data() + m_inputTensor.size() ) );
|
||||
|
||||
// Bind input and evaluate.
|
||||
m_binding.Clear();
|
||||
m_binding.Bind( m_inputName, inputTensor );
|
||||
|
||||
auto result = m_session.Evaluate( m_binding, L"" );
|
||||
|
||||
// Extract output tensor — bulk-copy to a raw float array so we
|
||||
// avoid per-element WinRT/COM dispatch in the hot loop.
|
||||
auto outputTensor = result.Outputs().Lookup( m_outputName ).as<winml::TensorFloat>();
|
||||
auto outputShape = outputTensor.Shape();
|
||||
auto outputView = outputTensor.GetAsVectorView();
|
||||
const uint32_t outputSize = outputView.Size();
|
||||
m_outputBuf.resize( outputSize );
|
||||
outputView.GetMany( 0, m_outputBuf );
|
||||
const float* outData = m_outputBuf.data();
|
||||
|
||||
// Determine output mask dimensions.
|
||||
int64_t outH = mH, outW = mW;
|
||||
int64_t numClasses = 1;
|
||||
if( outputShape.Size() == 4 )
|
||||
{
|
||||
if( outputShape.GetAt( 1 ) <= 2 && outputShape.GetAt( 2 ) > 2 )
|
||||
{
|
||||
// [1, classes, H, W]
|
||||
numClasses = outputShape.GetAt( 1 );
|
||||
outH = outputShape.GetAt( 2 );
|
||||
outW = outputShape.GetAt( 3 );
|
||||
}
|
||||
else
|
||||
{
|
||||
// [1, H, W, classes]
|
||||
outH = outputShape.GetAt( 1 );
|
||||
outW = outputShape.GetAt( 2 );
|
||||
numClasses = outputShape.GetAt( 3 );
|
||||
}
|
||||
}
|
||||
else if( outputShape.Size() == 3 )
|
||||
{
|
||||
outH = outputShape.GetAt( 1 );
|
||||
outW = outputShape.GetAt( 2 );
|
||||
}
|
||||
|
||||
// Store actual output dimensions for GetModelMaskWidth/Height.
|
||||
m_modelOutputWidth = outW;
|
||||
m_modelOutputHeight = outH;
|
||||
|
||||
// Build model-resolution mask first, apply sigmoid sharpening
|
||||
// at model resolution (e.g. 256×256 = 65K pixels), then upscale
|
||||
// to frame resolution.
|
||||
const size_t modelPixels = static_cast<size_t>( outH * outW );
|
||||
m_erodeBuf.resize( modelPixels );
|
||||
|
||||
// Extract person scores at model resolution from the raw array.
|
||||
// Apply a hard threshold to produce a binary mask. This is much
|
||||
// faster than a sigmoid (no expf) and eliminates the partial-blur
|
||||
// halo that was bleeding onto body/head edges.
|
||||
for( int64_t y = 0; y < outH; y++ )
|
||||
{
|
||||
for( int64_t x = 0; x < outW; x++ )
|
||||
{
|
||||
float personScore;
|
||||
if( numClasses == 1 )
|
||||
{
|
||||
personScore = outData[y * outW + x];
|
||||
}
|
||||
else
|
||||
{
|
||||
float bg = outData[0 * outH * outW + y * outW + x];
|
||||
float fg = outData[1 * outH * outW + y * outW + x];
|
||||
personScore = ( fg > bg ) ? 1.0f : 0.0f;
|
||||
}
|
||||
|
||||
m_erodeBuf[static_cast<size_t>( y * outW + x )] = ( personScore > 0.5f ) ? 1.0f : 0.0f;
|
||||
}
|
||||
}
|
||||
|
||||
// ── GPU path: model-resolution post-processing only ────────
|
||||
// When modelResOnly is true, apply feathering and temporal
|
||||
// smoothing at model resolution (e.g. 256×256 = 65K pixels)
|
||||
// and return early. The GPU's hardware bilinear sampler will
|
||||
// handle upscaling to frame resolution for free.
|
||||
if( modelResOnly )
|
||||
{
|
||||
// Small box blur on m_erodeBuf for edge feathering.
|
||||
// Radius 1 at 256×256 provides similar smoothing to
|
||||
// radius 3 at 960×540 after bilinear upscale.
|
||||
const int modelBlurRadius = 1;
|
||||
const int modelDiam = modelBlurRadius * 2 + 1;
|
||||
m_maskBlurBuf.resize( modelPixels );
|
||||
|
||||
// Horizontal pass.
|
||||
for( int64_t y = 0; y < outH; y++ )
|
||||
{
|
||||
const float* srcRow = m_erodeBuf.data() + y * outW;
|
||||
float* dstRow = m_maskBlurBuf.data() + y * outW;
|
||||
float sum = 0.0f;
|
||||
for( int i = -modelBlurRadius; i <= modelBlurRadius; i++ )
|
||||
sum += srcRow[(std::max)( static_cast<int64_t>(0), (std::min)( outW - 1, static_cast<int64_t>( i ) ) )];
|
||||
for( int64_t x = 0; x < outW; x++ )
|
||||
{
|
||||
dstRow[x] = sum / modelDiam;
|
||||
int64_t remX = (std::max)( static_cast<int64_t>(0), x - modelBlurRadius );
|
||||
int64_t addX = (std::min)( outW - 1, x + modelBlurRadius + 1 );
|
||||
sum += srcRow[addX] - srcRow[remX];
|
||||
}
|
||||
}
|
||||
|
||||
// Vertical pass.
|
||||
for( int64_t x = 0; x < outW; x++ )
|
||||
{
|
||||
float sum = 0.0f;
|
||||
for( int i = -modelBlurRadius; i <= modelBlurRadius; i++ )
|
||||
{
|
||||
int64_t iy = (std::max)( static_cast<int64_t>(0), (std::min)( outH - 1, static_cast<int64_t>( i ) ) );
|
||||
sum += m_maskBlurBuf[static_cast<size_t>( iy * outW + x )];
|
||||
}
|
||||
for( int64_t y = 0; y < outH; y++ )
|
||||
{
|
||||
m_erodeBuf[static_cast<size_t>( y * outW + x )] = sum / modelDiam;
|
||||
int64_t remY = (std::max)( static_cast<int64_t>(0), y - modelBlurRadius );
|
||||
int64_t addY = (std::min)( outH - 1, y + modelBlurRadius + 1 );
|
||||
sum += m_maskBlurBuf[static_cast<size_t>( addY * outW + x )] -
|
||||
m_maskBlurBuf[static_cast<size_t>( remY * outW + x )];
|
||||
}
|
||||
}
|
||||
|
||||
// Temporal smoothing at model resolution.
|
||||
if( m_prevModelMask.size() == modelPixels )
|
||||
{
|
||||
constexpr float alpha = 0.6f;
|
||||
constexpr float beta = 0.4f;
|
||||
for( size_t i = 0; i < modelPixels; i++ )
|
||||
m_erodeBuf[i] = alpha * m_erodeBuf[i] + beta * m_prevModelMask[i];
|
||||
}
|
||||
m_prevModelMask = m_erodeBuf;
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
// Upscale processed mask to frame dimensions via bilinear interpolation
|
||||
// to produce smooth edges instead of staircase artifacts.
|
||||
const size_t maskPixels = static_cast<size_t>( width ) * height;
|
||||
m_mask.resize( maskPixels );
|
||||
for( uint32_t y = 0; y < height; y++ )
|
||||
{
|
||||
float srcYf = ( y + 0.5f ) * outH / static_cast<float>( height ) - 0.5f;
|
||||
srcYf = (std::max)( 0.0f, (std::min)( srcYf, static_cast<float>( outH - 1 ) ) );
|
||||
int64_t y0 = static_cast<int64_t>( srcYf );
|
||||
int64_t y1 = (std::min)( y0 + 1, outH - 1 );
|
||||
float fy = srcYf - y0;
|
||||
|
||||
for( uint32_t x = 0; x < width; x++ )
|
||||
{
|
||||
float srcXf = ( x + 0.5f ) * outW / static_cast<float>( width ) - 0.5f;
|
||||
srcXf = (std::max)( 0.0f, (std::min)( srcXf, static_cast<float>( outW - 1 ) ) );
|
||||
int64_t x0 = static_cast<int64_t>( srcXf );
|
||||
int64_t x1 = (std::min)( x0 + 1, outW - 1 );
|
||||
float fx = srcXf - x0;
|
||||
|
||||
float v00 = m_erodeBuf[static_cast<size_t>(y0 * outW + x0)];
|
||||
float v01 = m_erodeBuf[static_cast<size_t>(y0 * outW + x1)];
|
||||
float v10 = m_erodeBuf[static_cast<size_t>(y1 * outW + x0)];
|
||||
float v11 = m_erodeBuf[static_cast<size_t>(y1 * outW + x1)];
|
||||
|
||||
m_mask[static_cast<size_t>( y ) * width + x] =
|
||||
v00 * ( 1.0f - fx ) * ( 1.0f - fy ) +
|
||||
v01 * fx * ( 1.0f - fy ) +
|
||||
v10 * ( 1.0f - fx ) * fy +
|
||||
v11 * fx * fy;
|
||||
}
|
||||
}
|
||||
|
||||
// Apply a small box blur to the upscaled mask to feather edges.
|
||||
const int maskBlurRadius = 3;
|
||||
const int maskDiam = maskBlurRadius * 2 + 1;
|
||||
m_maskBlurBuf.resize( maskPixels );
|
||||
|
||||
// Horizontal pass.
|
||||
for( uint32_t y = 0; y < height; y++ )
|
||||
{
|
||||
const float* srcRow = m_mask.data() + static_cast<size_t>( y ) * width;
|
||||
float* dstRow = m_maskBlurBuf.data() + static_cast<size_t>( y ) * width;
|
||||
float sum = 0.0f;
|
||||
|
||||
for( int i = -maskBlurRadius; i <= maskBlurRadius; i++ )
|
||||
sum += srcRow[(std::max)( 0, (std::min)( static_cast<int>( width ) - 1, i ) )];
|
||||
|
||||
for( uint32_t x = 0; x < width; x++ )
|
||||
{
|
||||
dstRow[x] = sum / maskDiam;
|
||||
int remX = (std::max)( 0, static_cast<int>( x ) - maskBlurRadius );
|
||||
int addX = (std::min)( static_cast<int>( width ) - 1, static_cast<int>( x ) + maskBlurRadius + 1 );
|
||||
sum += srcRow[addX] - srcRow[remX];
|
||||
}
|
||||
}
|
||||
|
||||
// Vertical pass.
|
||||
for( uint32_t x = 0; x < width; x++ )
|
||||
{
|
||||
float sum = 0.0f;
|
||||
|
||||
for( int i = -maskBlurRadius; i <= maskBlurRadius; i++ )
|
||||
{
|
||||
int iy = (std::max)( 0, (std::min)( static_cast<int>( height ) - 1, i ) );
|
||||
sum += m_maskBlurBuf[static_cast<size_t>( iy ) * width + x];
|
||||
}
|
||||
|
||||
for( uint32_t y = 0; y < height; y++ )
|
||||
{
|
||||
m_mask[static_cast<size_t>( y ) * width + x] = sum / maskDiam;
|
||||
int remY = (std::max)( 0, static_cast<int>( y ) - maskBlurRadius );
|
||||
int addY = (std::min)( static_cast<int>( height ) - 1, static_cast<int>( y ) + maskBlurRadius + 1 );
|
||||
sum += m_maskBlurBuf[static_cast<size_t>( addY ) * width + x] -
|
||||
m_maskBlurBuf[static_cast<size_t>( remY ) * width + x];
|
||||
}
|
||||
}
|
||||
|
||||
// Temporal smoothing: blend the current mask with the previous
|
||||
// frame's mask to stabilize edges and reduce flicker. A weight
|
||||
// of 0.6 current + 0.4 previous keeps edges responsive while
|
||||
// dampening the frame-to-frame jitter around fine details like
|
||||
// ears, hair, and fingers.
|
||||
{
|
||||
const size_t maskPixelsInner = static_cast<size_t>( width ) * height;
|
||||
if( m_prevMask.size() == maskPixelsInner )
|
||||
{
|
||||
constexpr float alpha = 0.6f; // current frame weight
|
||||
constexpr float beta = 0.4f; // previous frame weight
|
||||
for( size_t i = 0; i < maskPixelsInner; i++ )
|
||||
{
|
||||
m_mask[i] = alpha * m_mask[i] + beta * m_prevMask[i];
|
||||
}
|
||||
}
|
||||
m_prevMask = m_mask;
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
catch( winrt::hresult_error const& ex )
|
||||
{
|
||||
OutputDebug( L"[BackgroundBlur] Evaluate failed: %s (0x%08X)\n", ex.message().c_str(), ex.code().value );
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
//----------------------------------------------------------------------------
|
||||
// HorizontalBoxBlur / VerticalBoxBlur
|
||||
//
|
||||
// Separable box blur passes used to build an approximate Gaussian.
|
||||
//----------------------------------------------------------------------------
|
||||
static void HorizontalBoxBlur(
|
||||
const uint8_t* src, uint8_t* dst, uint32_t width, uint32_t height, int radius )
|
||||
{
|
||||
const int diameter = radius * 2 + 1;
|
||||
for( uint32_t y = 0; y < height; y++ )
|
||||
{
|
||||
int rSum = 0, gSum = 0, bSum = 0;
|
||||
const uint8_t* row = src + static_cast<size_t>( y ) * width * 4;
|
||||
|
||||
// Initialize window with clamped left edge.
|
||||
for( int i = -radius; i <= radius; i++ )
|
||||
{
|
||||
int ix = (std::max)( 0, (std::min)( static_cast<int>( width ) - 1, i ) );
|
||||
const uint8_t* px = row + ix * 4;
|
||||
bSum += px[0];
|
||||
gSum += px[1];
|
||||
rSum += px[2];
|
||||
}
|
||||
|
||||
uint8_t* dstRow = dst + static_cast<size_t>( y ) * width * 4;
|
||||
for( uint32_t x = 0; x < width; x++ )
|
||||
{
|
||||
dstRow[x * 4 + 0] = static_cast<uint8_t>( bSum / diameter );
|
||||
dstRow[x * 4 + 1] = static_cast<uint8_t>( gSum / diameter );
|
||||
dstRow[x * 4 + 2] = static_cast<uint8_t>( rSum / diameter );
|
||||
dstRow[x * 4 + 3] = 0xFF;
|
||||
|
||||
// Slide window: add right, remove left.
|
||||
int removeX = (std::max)( 0, static_cast<int>( x ) - radius );
|
||||
int addX = (std::min)( static_cast<int>( width ) - 1, static_cast<int>( x ) + radius + 1 );
|
||||
const uint8_t* remPx = row + removeX * 4;
|
||||
const uint8_t* addPx = row + addX * 4;
|
||||
bSum += addPx[0] - remPx[0];
|
||||
gSum += addPx[1] - remPx[1];
|
||||
rSum += addPx[2] - remPx[2];
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
static void VerticalBoxBlur(
|
||||
const uint8_t* src, uint8_t* dst, uint32_t width, uint32_t height, int radius )
|
||||
{
|
||||
const int diameter = radius * 2 + 1;
|
||||
for( uint32_t x = 0; x < width; x++ )
|
||||
{
|
||||
int rSum = 0, gSum = 0, bSum = 0;
|
||||
|
||||
// Initialize window with clamped top edge.
|
||||
for( int i = -radius; i <= radius; i++ )
|
||||
{
|
||||
int iy = (std::max)( 0, (std::min)( static_cast<int>( height ) - 1, i ) );
|
||||
const uint8_t* px = src + ( static_cast<size_t>( iy ) * width + x ) * 4;
|
||||
bSum += px[0];
|
||||
gSum += px[1];
|
||||
rSum += px[2];
|
||||
}
|
||||
|
||||
for( uint32_t y = 0; y < height; y++ )
|
||||
{
|
||||
uint8_t* dstPx = dst + ( static_cast<size_t>( y ) * width + x ) * 4;
|
||||
dstPx[0] = static_cast<uint8_t>( bSum / diameter );
|
||||
dstPx[1] = static_cast<uint8_t>( gSum / diameter );
|
||||
dstPx[2] = static_cast<uint8_t>( rSum / diameter );
|
||||
dstPx[3] = 0xFF;
|
||||
|
||||
int removeY = (std::max)( 0, static_cast<int>( y ) - radius );
|
||||
int addY = (std::min)( static_cast<int>( height ) - 1, static_cast<int>( y ) + radius + 1 );
|
||||
const uint8_t* remPx = src + ( static_cast<size_t>( removeY ) * width + x ) * 4;
|
||||
const uint8_t* addPx = src + ( static_cast<size_t>( addY ) * width + x ) * 4;
|
||||
bSum += addPx[0] - remPx[0];
|
||||
gSum += addPx[1] - remPx[1];
|
||||
rSum += addPx[2] - remPx[2];
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
//----------------------------------------------------------------------------
|
||||
// BackgroundBlur::ApplyBlurWithMask
|
||||
//
|
||||
// Downscales the frame to a small working size, blurs there, then
|
||||
// performs a single full-resolution pass that blends the original
|
||||
// pixels with the upscaled blurred pixels according to the mask.
|
||||
//----------------------------------------------------------------------------
|
||||
void BackgroundBlur::ApplyBlurWithMask( uint8_t* bgraPixels, uint32_t width, uint32_t height, int blurRadius )
|
||||
{
|
||||
const size_t frameBytes = static_cast<size_t>( width ) * height * 4;
|
||||
m_blurredFrame.resize( frameBytes );
|
||||
m_tempFrame.resize( frameBytes );
|
||||
|
||||
// The input is already capped at 960×540 by WebcamCapture, so blur
|
||||
// directly — no need for a secondary downscale.
|
||||
int effectiveRadius = (std::max)( 3, blurRadius );
|
||||
|
||||
// 2 iterations of box blur → approximate Gaussian.
|
||||
HorizontalBoxBlur( bgraPixels, m_blurredFrame.data(), width, height, effectiveRadius );
|
||||
VerticalBoxBlur( m_blurredFrame.data(), m_tempFrame.data(), width, height, effectiveRadius );
|
||||
HorizontalBoxBlur( m_tempFrame.data(), m_blurredFrame.data(), width, height, effectiveRadius );
|
||||
VerticalBoxBlur( m_blurredFrame.data(), m_tempFrame.data(), width, height, effectiveRadius );
|
||||
|
||||
// Blend pass with alpha support for smooth mask edges.
|
||||
const uint8_t* blurData = m_tempFrame.data();
|
||||
for( uint32_t y = 0; y < height; y++ )
|
||||
{
|
||||
uint8_t* dstRow = bgraPixels + static_cast<size_t>( y ) * width * 4;
|
||||
const uint8_t* blurRow = blurData + static_cast<size_t>( y ) * width * 4;
|
||||
const float* maskRow = m_mask.data() + static_cast<size_t>( y ) * width;
|
||||
|
||||
for( uint32_t x = 0; x < width; x++ )
|
||||
{
|
||||
float maskVal = maskRow[x];
|
||||
|
||||
// Fast path: fully person → keep original pixel untouched.
|
||||
if( maskVal >= 1.0f )
|
||||
continue;
|
||||
|
||||
uint8_t* dp = dstRow + x * 4;
|
||||
const uint8_t* bp = blurRow + x * 4;
|
||||
|
||||
// Fast path: fully background → copy blurred pixel.
|
||||
if( maskVal <= 0.0f )
|
||||
{
|
||||
*reinterpret_cast<uint32_t*>( dp ) = *reinterpret_cast<const uint32_t*>( bp );
|
||||
continue;
|
||||
}
|
||||
|
||||
// Edge pixel → alpha blend original and blurred.
|
||||
float inv = 1.0f - maskVal;
|
||||
dp[0] = static_cast<uint8_t>( dp[0] * maskVal + bp[0] * inv + 0.5f );
|
||||
dp[1] = static_cast<uint8_t>( dp[1] * maskVal + bp[1] * inv + 0.5f );
|
||||
dp[2] = static_cast<uint8_t>( dp[2] * maskVal + bp[2] * inv + 0.5f );
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
//----------------------------------------------------------------------------
|
||||
// BackgroundBlur::SetBackgroundImage
|
||||
//
|
||||
// Loads an image file via WIC and stores it as a BGRA pixel buffer.
|
||||
//----------------------------------------------------------------------------
|
||||
bool BackgroundBlur::SetBackgroundImage( const wchar_t* imagePath )
|
||||
{
|
||||
m_bgImage.clear();
|
||||
m_bgImageWidth = 0;
|
||||
m_bgImageHeight = 0;
|
||||
m_scaledBgImage.clear();
|
||||
m_scaledBgW = 0;
|
||||
m_scaledBgH = 0;
|
||||
|
||||
if( !imagePath || !*imagePath )
|
||||
return false;
|
||||
|
||||
auto factory = wil::CoCreateInstance<IWICImagingFactory>( CLSID_WICImagingFactory );
|
||||
if( !factory )
|
||||
return false;
|
||||
|
||||
wil::com_ptr<IWICBitmapDecoder> decoder;
|
||||
HRESULT hr = factory->CreateDecoderFromFilename(
|
||||
imagePath, nullptr, GENERIC_READ, WICDecodeMetadataCacheOnDemand, &decoder );
|
||||
if( FAILED( hr ) )
|
||||
{
|
||||
OutputDebug( L"[BackgroundBlur] Failed to decode image: %s (hr=0x%08X)\n", imagePath, hr );
|
||||
return false;
|
||||
}
|
||||
|
||||
wil::com_ptr<IWICBitmapFrameDecode> frame;
|
||||
hr = decoder->GetFrame( 0, &frame );
|
||||
if( FAILED( hr ) )
|
||||
return false;
|
||||
|
||||
// Convert to BGRA 32bpp.
|
||||
wil::com_ptr<IWICFormatConverter> converter;
|
||||
hr = factory->CreateFormatConverter( &converter );
|
||||
if( FAILED( hr ) )
|
||||
return false;
|
||||
|
||||
hr = converter->Initialize(
|
||||
frame.get(), GUID_WICPixelFormat32bppBGRA,
|
||||
WICBitmapDitherTypeNone, nullptr, 0.0, WICBitmapPaletteTypeCustom );
|
||||
if( FAILED( hr ) )
|
||||
return false;
|
||||
|
||||
UINT w = 0, h = 0;
|
||||
converter->GetSize( &w, &h );
|
||||
if( w == 0 || h == 0 )
|
||||
return false;
|
||||
|
||||
m_bgImage.resize( static_cast<size_t>( w ) * h * 4 );
|
||||
hr = converter->CopyPixels( nullptr, w * 4, static_cast<UINT>( m_bgImage.size() ), m_bgImage.data() );
|
||||
if( FAILED( hr ) )
|
||||
{
|
||||
m_bgImage.clear();
|
||||
return false;
|
||||
}
|
||||
|
||||
m_bgImageWidth = w;
|
||||
m_bgImageHeight = h;
|
||||
|
||||
OutputDebug( L"[BackgroundBlur] Background image loaded: %ux%u from %s\n", w, h, imagePath );
|
||||
return true;
|
||||
}
|
||||
|
||||
//----------------------------------------------------------------------------
|
||||
// BackgroundBlur::EnsureScaledBgImage
|
||||
//
|
||||
// Scales the loaded background image to the specified dimensions using
|
||||
// nearest-neighbor. The result is cached and only recomputed when the
|
||||
// target dimensions change. The image is center-cropped to preserve
|
||||
// aspect ratio (like "cover" scaling).
|
||||
//----------------------------------------------------------------------------
|
||||
void BackgroundBlur::EnsureScaledBgImage( uint32_t width, uint32_t height )
|
||||
{
|
||||
if( m_scaledBgW == width && m_scaledBgH == height && !m_scaledBgImage.empty() )
|
||||
return;
|
||||
|
||||
m_scaledBgImage.resize( static_cast<size_t>( width ) * height * 4 );
|
||||
m_scaledBgW = width;
|
||||
m_scaledBgH = height;
|
||||
|
||||
// Compute center-crop of the source image to match the target aspect ratio.
|
||||
double targetAspect = static_cast<double>( width ) / height;
|
||||
double srcAspect = static_cast<double>( m_bgImageWidth ) / m_bgImageHeight;
|
||||
|
||||
uint32_t cropW, cropH, cropX, cropY;
|
||||
if( srcAspect > targetAspect )
|
||||
{
|
||||
// Source is wider — crop horizontally.
|
||||
cropH = m_bgImageHeight;
|
||||
cropW = static_cast<uint32_t>( m_bgImageHeight * targetAspect + 0.5 );
|
||||
cropX = ( m_bgImageWidth - cropW ) / 2;
|
||||
cropY = 0;
|
||||
}
|
||||
else
|
||||
{
|
||||
// Source is taller — crop vertically.
|
||||
cropW = m_bgImageWidth;
|
||||
cropH = static_cast<uint32_t>( m_bgImageWidth / targetAspect + 0.5 );
|
||||
cropX = 0;
|
||||
cropY = ( m_bgImageHeight - cropH ) / 2;
|
||||
}
|
||||
|
||||
for( uint32_t y = 0; y < height; y++ )
|
||||
{
|
||||
uint32_t srcY = cropY + y * cropH / height;
|
||||
for( uint32_t x = 0; x < width; x++ )
|
||||
{
|
||||
uint32_t srcX = cropX + x * cropW / width;
|
||||
size_t srcIdx = ( static_cast<size_t>( srcY ) * m_bgImageWidth + srcX ) * 4;
|
||||
size_t dstIdx = ( static_cast<size_t>( y ) * width + x ) * 4;
|
||||
m_scaledBgImage[dstIdx + 0] = m_bgImage[srcIdx + 0];
|
||||
m_scaledBgImage[dstIdx + 1] = m_bgImage[srcIdx + 1];
|
||||
m_scaledBgImage[dstIdx + 2] = m_bgImage[srcIdx + 2];
|
||||
m_scaledBgImage[dstIdx + 3] = 0xFF;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
//----------------------------------------------------------------------------
|
||||
// BackgroundBlur::ApplyImageWithMask
|
||||
//
|
||||
// Replaces background pixels with the loaded background image using the
|
||||
// segmentation mask. Person pixels are preserved, background pixels come
|
||||
// from the scaled image.
|
||||
//----------------------------------------------------------------------------
|
||||
void BackgroundBlur::ApplyImageWithMask( uint8_t* bgraPixels, uint32_t width, uint32_t height )
|
||||
{
|
||||
EnsureScaledBgImage( width, height );
|
||||
|
||||
const uint8_t* bgData = m_scaledBgImage.data();
|
||||
|
||||
for( uint32_t y = 0; y < height; y++ )
|
||||
{
|
||||
uint8_t* dstRow = bgraPixels + static_cast<size_t>( y ) * width * 4;
|
||||
const uint8_t* bgRow = bgData + static_cast<size_t>( y ) * width * 4;
|
||||
const float* maskRow = m_mask.data() + static_cast<size_t>( y ) * width;
|
||||
|
||||
for( uint32_t x = 0; x < width; x++ )
|
||||
{
|
||||
float maskVal = maskRow[x];
|
||||
|
||||
// Fully person → keep original pixel.
|
||||
if( maskVal >= 1.0f )
|
||||
continue;
|
||||
|
||||
uint8_t* dp = dstRow + x * 4;
|
||||
const uint8_t* bp = bgRow + x * 4;
|
||||
|
||||
// Fully background → copy background image pixel.
|
||||
if( maskVal <= 0.0f )
|
||||
{
|
||||
*reinterpret_cast<uint32_t*>( dp ) = *reinterpret_cast<const uint32_t*>( bp );
|
||||
continue;
|
||||
}
|
||||
|
||||
// Edge pixel → alpha blend person and background image.
|
||||
float inv = 1.0f - maskVal;
|
||||
dp[0] = static_cast<uint8_t>( dp[0] * maskVal + bp[0] * inv + 0.5f );
|
||||
dp[1] = static_cast<uint8_t>( dp[1] * maskVal + bp[1] * inv + 0.5f );
|
||||
dp[2] = static_cast<uint8_t>( dp[2] * maskVal + bp[2] * inv + 0.5f );
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
//----------------------------------------------------------------------------
|
||||
// BackgroundBlur::ShouldRunInference
|
||||
//
|
||||
// Decides whether segmentation inference should run this frame.
|
||||
// Uses a combination of periodic fallback and motion detection:
|
||||
// motion is estimated by comparing luminance at a sparse grid of
|
||||
// sample points with the previous frame. When the scene changes
|
||||
// quickly (fast head movement), inference runs every frame.
|
||||
//----------------------------------------------------------------------------
|
||||
bool BackgroundBlur::ShouldRunInference( const uint8_t* bgraPixels, uint32_t width, uint32_t height )
|
||||
{
|
||||
// Always run if no cached mask or dimensions changed.
|
||||
if( !m_hasCachedMask || m_lastMaskWidth != width || m_lastMaskHeight != height )
|
||||
return true;
|
||||
|
||||
// Periodic fallback: run at least every N frames.
|
||||
const uint32_t pixels = width * height;
|
||||
const int inferenceInterval = ( pixels > 500000 ) ? 6 : 3;
|
||||
if( ( m_frameCounter % inferenceInterval ) == 0 )
|
||||
return true;
|
||||
|
||||
// Motion detection: sample luminance at a sparse grid and compare
|
||||
// with the previous frame.
|
||||
constexpr int gridSize = MOTION_GRID_SIZE;
|
||||
constexpr int numSamples = gridSize * gridSize;
|
||||
float curSamples[numSamples];
|
||||
|
||||
for( int gy = 0; gy < gridSize; gy++ )
|
||||
{
|
||||
uint32_t sy = ( gy * 2 + 1 ) * height / ( gridSize * 2 );
|
||||
for( int gx = 0; gx < gridSize; gx++ )
|
||||
{
|
||||
uint32_t sx = ( gx * 2 + 1 ) * width / ( gridSize * 2 );
|
||||
const uint8_t* px = bgraPixels + ( static_cast<size_t>( sy ) * width + sx ) * 4;
|
||||
curSamples[gy * gridSize + gx] = 0.299f * px[2] + 0.587f * px[1] + 0.114f * px[0];
|
||||
}
|
||||
}
|
||||
|
||||
float motionScore = 0.0f;
|
||||
if( m_hasPrevSamples )
|
||||
{
|
||||
for( int i = 0; i < numSamples; i++ )
|
||||
{
|
||||
float diff = curSamples[i] - m_prevSamples[i];
|
||||
motionScore += diff > 0.0f ? diff : -diff;
|
||||
}
|
||||
motionScore /= numSamples;
|
||||
}
|
||||
|
||||
memcpy( m_prevSamples, curSamples, sizeof( curSamples ) );
|
||||
m_hasPrevSamples = true;
|
||||
|
||||
// Average per-sample luminance change > 5/255 indicates significant motion.
|
||||
return motionScore > 5.0f;
|
||||
}
|
||||
|
||||
//----------------------------------------------------------------------------
|
||||
// BackgroundBlur::ApplyImageReplacement
|
||||
//
|
||||
// Main entry point for background image replacement mode.
|
||||
//----------------------------------------------------------------------------
|
||||
bool BackgroundBlur::ApplyImageReplacement( uint8_t* bgraPixels, uint32_t width, uint32_t height )
|
||||
{
|
||||
if( !m_session || !bgraPixels || width == 0 || height == 0 )
|
||||
return false;
|
||||
|
||||
if( m_bgImage.empty() )
|
||||
return false;
|
||||
|
||||
if( ShouldRunInference( bgraPixels, width, height ) )
|
||||
{
|
||||
if( !RunSegmentation( bgraPixels, width, height ) )
|
||||
return false;
|
||||
m_lastMaskWidth = width;
|
||||
m_lastMaskHeight = height;
|
||||
m_hasCachedMask = true;
|
||||
}
|
||||
m_frameCounter++;
|
||||
|
||||
ApplyImageWithMask( bgraPixels, width, height );
|
||||
return true;
|
||||
}
|
||||
|
||||
//----------------------------------------------------------------------------
|
||||
// BackgroundBlur::Apply
|
||||
//
|
||||
// Main entry point: runs segmentation and applies blur to the background.
|
||||
//----------------------------------------------------------------------------
|
||||
bool BackgroundBlur::Apply( uint8_t* bgraPixels, uint32_t width, uint32_t height, int blurRadius )
|
||||
{
|
||||
if( !m_session || !bgraPixels || width == 0 || height == 0 )
|
||||
return false;
|
||||
|
||||
if( ShouldRunInference( bgraPixels, width, height ) )
|
||||
{
|
||||
if( !RunSegmentation( bgraPixels, width, height ) )
|
||||
return false;
|
||||
m_lastMaskWidth = width;
|
||||
m_lastMaskHeight = height;
|
||||
m_hasCachedMask = true;
|
||||
}
|
||||
m_frameCounter++;
|
||||
|
||||
ApplyBlurWithMask( bgraPixels, width, height, blurRadius );
|
||||
return true;
|
||||
}
|
||||
|
||||
//----------------------------------------------------------------------------
|
||||
// BackgroundBlur::RunSegmentationOnly
|
||||
//
|
||||
// Runs the segmentation model and produces the mask, but does NOT blur
|
||||
// or modify the pixel buffer. Used when the GPU compute shader will
|
||||
// perform the box blur instead of the CPU.
|
||||
//----------------------------------------------------------------------------
|
||||
bool BackgroundBlur::RunSegmentationOnly( const uint8_t* bgraPixels, uint32_t width, uint32_t height )
|
||||
{
|
||||
if( !m_session || !bgraPixels || width == 0 || height == 0 )
|
||||
return false;
|
||||
|
||||
if( ShouldRunInference( bgraPixels, width, height ) )
|
||||
{
|
||||
// Model-resolution only: skip CPU upscale+feather at frame
|
||||
// resolution — the GPU bilinear sampler handles that.
|
||||
if( !RunSegmentation( bgraPixels, width, height, /*modelResOnly=*/ true ) )
|
||||
return false;
|
||||
m_lastMaskWidth = width;
|
||||
m_lastMaskHeight = height;
|
||||
m_hasCachedMask = true;
|
||||
}
|
||||
m_frameCounter++;
|
||||
|
||||
return m_hasCachedMask;
|
||||
}
|
||||
160
src/modules/ZoomIt/ZoomIt/BackgroundBlur.h
Normal file
160
src/modules/ZoomIt/ZoomIt/BackgroundBlur.h
Normal file
@@ -0,0 +1,160 @@
|
||||
//==============================================================================
|
||||
//
|
||||
// BackgroundBlur.h
|
||||
//
|
||||
// Performs person segmentation using Windows ML (Windows.AI.MachineLearning)
|
||||
// and applies either a Gaussian blur or a custom background image to the
|
||||
// background of a BGRA webcam frame. The segmentation model runs on CPU
|
||||
// via the Windows ML default device, keeping the GPU free for recording.
|
||||
//
|
||||
// Copyright (C) Mark Russinovich
|
||||
// Sysinternals - www.sysinternals.com
|
||||
//
|
||||
//==============================================================================
|
||||
#pragma once
|
||||
|
||||
#include <vector>
|
||||
#include <string>
|
||||
#include <cstdint>
|
||||
#include <winrt/Windows.AI.MachineLearning.h>
|
||||
|
||||
// Background processing mode for the webcam overlay.
|
||||
enum class WebcamBackgroundMode : uint32_t
|
||||
{
|
||||
None = 0, // No background processing
|
||||
Blur = 1, // Gaussian blur on the background
|
||||
Image = 2, // Replace background with a user-chosen image
|
||||
};
|
||||
|
||||
class BackgroundBlur
|
||||
{
|
||||
public:
|
||||
BackgroundBlur() = default;
|
||||
~BackgroundBlur() = default;
|
||||
|
||||
// Initialize the ONNX model. modelPath must point to a valid .onnx
|
||||
// segmentation model file. Returns true on success.
|
||||
bool Initialize( const wchar_t* modelPath );
|
||||
|
||||
// Load a background replacement image from the given file path.
|
||||
// The image is decoded via WIC and stored as a BGRA buffer.
|
||||
// Returns true on success.
|
||||
bool SetBackgroundImage( const wchar_t* imagePath );
|
||||
|
||||
// Returns true if a background image has been loaded.
|
||||
bool HasBackgroundImage() const { return !m_bgImage.empty(); }
|
||||
|
||||
// Apply background blur to a BGRA pixel buffer in-place.
|
||||
// width/height are the frame dimensions. blurRadius controls
|
||||
// the strength of the Gaussian blur (in pixels).
|
||||
// Returns true if segmentation + blur was applied successfully.
|
||||
bool Apply( uint8_t* bgraPixels, uint32_t width, uint32_t height, int blurRadius = 21 );
|
||||
|
||||
// Apply background image replacement to a BGRA pixel buffer in-place.
|
||||
// Uses the previously loaded background image (via SetBackgroundImage).
|
||||
// Returns true if segmentation + image replacement was applied.
|
||||
bool ApplyImageReplacement( uint8_t* bgraPixels, uint32_t width, uint32_t height );
|
||||
|
||||
// Returns true if the model has been loaded successfully.
|
||||
bool IsInitialized() const { return m_session != nullptr; }
|
||||
|
||||
// Access the segmentation mask after Apply()/ApplyImageReplacement().
|
||||
// The mask has one float [0..1] per pixel at the processing resolution
|
||||
// (1.0 = person / foreground, 0.0 = background).
|
||||
const std::vector<float>& GetMask() const { return m_mask; }
|
||||
uint32_t GetMaskWidth() const { return m_lastMaskWidth; }
|
||||
uint32_t GetMaskHeight() const { return m_lastMaskHeight; }
|
||||
bool HasCachedMask() const { return m_hasCachedMask; }
|
||||
|
||||
// Run segmentation only (no CPU blur or mask blend). Use this when
|
||||
// the blur will be performed on the GPU via a compute shader.
|
||||
// Populates the mask (GetMask) but does NOT touch bgraPixels.
|
||||
bool RunSegmentationOnly( const uint8_t* bgraPixels, uint32_t width, uint32_t height );
|
||||
|
||||
// Access the fully-blurred frame after Apply().
|
||||
// Contains all pixels blurred (before mask-based compositing).
|
||||
// Only valid after Apply() — NOT after ApplyImageReplacement().
|
||||
const std::vector<uint8_t>& GetBlurredFrame() const { return m_tempFrame; }
|
||||
|
||||
// Access the model-resolution mask before upscaling (e.g. 256×256).
|
||||
// Useful when the GPU handles upscaling via hardware bilinear filtering.
|
||||
const std::vector<float>& GetModelMask() const { return m_erodeBuf; }
|
||||
int64_t GetModelMaskWidth() const { return m_modelOutputWidth; }
|
||||
int64_t GetModelMaskHeight() const { return m_modelOutputHeight; }
|
||||
|
||||
private:
|
||||
// Run the segmentation model and produce a float mask [0..1] per pixel.
|
||||
// When modelResOnly is true, stops after model-resolution post-processing
|
||||
// (feathering + temporal smoothing at 256×256) and skips the CPU upscale
|
||||
// to frame resolution — the GPU bilinear sampler handles that instead.
|
||||
bool RunSegmentation( const uint8_t* bgraPixels, uint32_t width, uint32_t height,
|
||||
bool modelResOnly = false );
|
||||
|
||||
// Apply box blur (iterated for Gaussian approximation) to bgraPixels
|
||||
// only where the mask indicates background.
|
||||
void ApplyBlurWithMask( uint8_t* bgraPixels, uint32_t width, uint32_t height, int blurRadius );
|
||||
|
||||
// Replace background pixels with the loaded background image.
|
||||
void ApplyImageWithMask( uint8_t* bgraPixels, uint32_t width, uint32_t height );
|
||||
|
||||
// Scale the loaded background image to the given dimensions (cached).
|
||||
void EnsureScaledBgImage( uint32_t width, uint32_t height );
|
||||
|
||||
// Decide whether inference is needed this frame (periodic + motion-adaptive).
|
||||
bool ShouldRunInference( const uint8_t* bgraPixels, uint32_t width, uint32_t height );
|
||||
|
||||
// Windows ML objects.
|
||||
winrt::Windows::AI::MachineLearning::LearningModel m_model{ nullptr };
|
||||
winrt::Windows::AI::MachineLearning::LearningModelSession m_session{ nullptr };
|
||||
winrt::Windows::AI::MachineLearning::LearningModelBinding m_binding{ nullptr };
|
||||
winrt::hstring m_inputName;
|
||||
winrt::hstring m_outputName;
|
||||
|
||||
// Model metadata (detected from the loaded model).
|
||||
int64_t m_modelInputWidth = 256;
|
||||
int64_t m_modelInputHeight = 256;
|
||||
int64_t m_modelInputChannels = 3;
|
||||
bool m_inputIsNchw = true; // true = [1,C,H,W], false = [1,H,W,C]
|
||||
bool m_usingGpu = false; // true if DirectML session is active
|
||||
|
||||
// Actual model output dimensions (may differ from input dimensions).
|
||||
int64_t m_modelOutputWidth = 256;
|
||||
int64_t m_modelOutputHeight = 256;
|
||||
|
||||
// Reusable buffers to avoid per-frame allocations.
|
||||
std::vector<float> m_inputTensor; // RGB float [1,3,H,W] or [1,H,W,3]
|
||||
std::vector<float> m_outputBuf; // Raw copy of output tensor data
|
||||
std::vector<float> m_mask; // Segmentation mask [width*height]
|
||||
std::vector<float> m_erodeBuf; // Model-resolution mask buffer
|
||||
std::vector<float> m_maskBlurBuf; // Temp buffer for mask edge smoothing
|
||||
std::vector<uint8_t> m_blurredFrame; // Temporary blurred copy
|
||||
std::vector<uint8_t> m_tempFrame; // Second temp buffer for blur passes
|
||||
|
||||
// Background image (original resolution, BGRA).
|
||||
std::vector<uint8_t> m_bgImage;
|
||||
uint32_t m_bgImageWidth = 0;
|
||||
uint32_t m_bgImageHeight = 0;
|
||||
|
||||
// Scaled background image (cached at overlay dimensions).
|
||||
std::vector<uint8_t> m_scaledBgImage;
|
||||
uint32_t m_scaledBgW = 0;
|
||||
uint32_t m_scaledBgH = 0;
|
||||
|
||||
// Frame-skipping: reuse the segmentation mask for N frames.
|
||||
int m_frameCounter = 0;
|
||||
uint32_t m_lastMaskWidth = 0;
|
||||
uint32_t m_lastMaskHeight = 0;
|
||||
bool m_hasCachedMask = false;
|
||||
|
||||
// Motion detection: luminance samples from the previous frame.
|
||||
static constexpr int MOTION_GRID_SIZE = 8; // 8×8 = 64 sample points
|
||||
float m_prevSamples[MOTION_GRID_SIZE * MOTION_GRID_SIZE] = {};
|
||||
bool m_hasPrevSamples = false;
|
||||
|
||||
// Temporal smoothing: previous frame's mask blended with current
|
||||
// to stabilize edges and reduce flicker.
|
||||
std::vector<float> m_prevMask;
|
||||
|
||||
// Model-resolution previous mask for GPU path temporal smoothing.
|
||||
std::vector<float> m_prevModelMask;
|
||||
};
|
||||
606
src/modules/ZoomIt/ZoomIt/BoxBlurCS.h
Normal file
606
src/modules/ZoomIt/ZoomIt/BoxBlurCS.h
Normal file
@@ -0,0 +1,606 @@
|
||||
#if 0
|
||||
//
|
||||
// Generated by Microsoft (R) HLSL Shader Compiler 10.1
|
||||
//
|
||||
//
|
||||
// Buffer Definitions:
|
||||
//
|
||||
// cbuffer BlurConstants
|
||||
// {
|
||||
//
|
||||
// uint Direction; // Offset: 0 Size: 4
|
||||
// int Radius; // Offset: 4 Size: 4
|
||||
// uint Width; // Offset: 8 Size: 4
|
||||
// uint Height; // Offset: 12 Size: 4
|
||||
//
|
||||
// }
|
||||
//
|
||||
//
|
||||
// Resource Bindings:
|
||||
//
|
||||
// Name Type Format Dim HLSL Bind Count
|
||||
// ------------------------------ ---------- ------- ----------- -------------- ------
|
||||
// InputTex texture float4 2d t0 1
|
||||
// OutputTex UAV float4 2d u0 1
|
||||
// BlurConstants cbuffer NA NA cb0 1
|
||||
//
|
||||
//
|
||||
//
|
||||
// Input signature:
|
||||
//
|
||||
// Name Index Mask Register SysValue Format Used
|
||||
// -------------------- ----- ------ -------- -------- ------- ------
|
||||
// no Input
|
||||
//
|
||||
// Output signature:
|
||||
//
|
||||
// Name Index Mask Register SysValue Format Used
|
||||
// -------------------- ----- ------ -------- -------- ------- ------
|
||||
// no Output
|
||||
cs_5_0
|
||||
dcl_globalFlags refactoringAllowed
|
||||
dcl_constantbuffer CB0[1], immediateIndexed
|
||||
dcl_resource_texture2d (float,float,float,float) t0
|
||||
dcl_uav_typed_texture2d (float,float,float,float) u0
|
||||
dcl_input vThreadGroupID.xy
|
||||
dcl_input vThreadIDInGroup.x
|
||||
dcl_temps 4
|
||||
dcl_tgsm_structured g0, 16, 320
|
||||
dcl_thread_group 256, 1, 1
|
||||
imin r0.x, cb0[0].y, l(32)
|
||||
ishl r0.y, r0.x, l(1)
|
||||
iadd r0.z, r0.y, l(256)
|
||||
if_z cb0[0].x
|
||||
uge r0.w, vThreadGroupID.y, cb0[0].w
|
||||
if_nz r0.w
|
||||
ret
|
||||
endif
|
||||
ishl r0.w, vThreadGroupID.x, l(8)
|
||||
iadd r1.x, cb0[0].z, l(-1)
|
||||
mov r2.y, vThreadGroupID.y
|
||||
mov r2.zw, l(0,0,0,0)
|
||||
mov r1.y, vThreadIDInGroup.x
|
||||
loop
|
||||
ige r1.z, r1.y, r0.z
|
||||
breakc_nz r1.z
|
||||
iadd r1.z, r0.w, r1.y
|
||||
iadd r1.z, -r0.x, r1.z
|
||||
imax r1.z, r1.z, l(0)
|
||||
imin r2.x, r1.x, r1.z
|
||||
ld_indexable(texture2d)(float,float,float,float) r3.xyzw, r2.xyzw, t0.xyzw
|
||||
store_structured g0.xyzw, r1.y, l(0), r3.xyzw
|
||||
iadd r1.y, r1.y, l(256)
|
||||
endloop
|
||||
sync_g_t
|
||||
imad r1.x, vThreadGroupID.x, l(256), vThreadIDInGroup.x
|
||||
uge r0.w, r1.x, cb0[0].z
|
||||
if_nz r0.w
|
||||
ret
|
||||
endif
|
||||
mov r2.xyzw, l(0,0,0,0)
|
||||
mov r0.w, l(0)
|
||||
loop
|
||||
ilt r3.x, r0.y, r0.w
|
||||
breakc_nz r3.x
|
||||
iadd r3.x, r0.w, vThreadIDInGroup.x
|
||||
ld_structured r3.xyzw, r3.x, l(0), g0.xyzw
|
||||
add r2.xyzw, r2.xyzw, r3.xyzw
|
||||
iadd r0.w, r0.w, l(1)
|
||||
endloop
|
||||
bfi r0.w, l(31), l(1), r0.x, l(1)
|
||||
itof r0.w, r0.w
|
||||
div r2.xyzw, r2.xyzw, r0.wwww
|
||||
mov r1.yzw, vThreadGroupID.yyyy
|
||||
store_uav_typed u0.xyzw, r1.xyzw, r2.xyzw
|
||||
else
|
||||
uge r0.w, vThreadGroupID.y, cb0[0].z
|
||||
if_nz r0.w
|
||||
ret
|
||||
endif
|
||||
ishl r0.w, vThreadGroupID.x, l(8)
|
||||
iadd r1.x, cb0[0].w, l(-1)
|
||||
mov r2.x, vThreadGroupID.y
|
||||
mov r2.zw, l(0,0,0,0)
|
||||
mov r1.y, vThreadIDInGroup.x
|
||||
loop
|
||||
ige r1.z, r1.y, r0.z
|
||||
breakc_nz r1.z
|
||||
iadd r1.z, r0.w, r1.y
|
||||
iadd r1.z, -r0.x, r1.z
|
||||
imax r1.z, r1.z, l(0)
|
||||
imin r2.y, r1.x, r1.z
|
||||
ld_indexable(texture2d)(float,float,float,float) r3.xyzw, r2.xyzw, t0.xyzw
|
||||
store_structured g0.xyzw, r1.y, l(0), r3.xyzw
|
||||
iadd r1.y, r1.y, l(256)
|
||||
endloop
|
||||
sync_g_t
|
||||
imad r1.yzw, vThreadGroupID.xxxx, l(0, 256, 256, 256), vThreadIDInGroup.xxxx
|
||||
uge r0.z, r1.w, cb0[0].w
|
||||
if_nz r0.z
|
||||
ret
|
||||
endif
|
||||
mov r2.xyzw, l(0,0,0,0)
|
||||
mov r0.z, l(0)
|
||||
loop
|
||||
ilt r0.w, r0.y, r0.z
|
||||
breakc_nz r0.w
|
||||
iadd r0.w, r0.z, vThreadIDInGroup.x
|
||||
ld_structured r3.xyzw, r0.w, l(0), g0.xyzw
|
||||
add r2.xyzw, r2.xyzw, r3.xyzw
|
||||
iadd r0.z, r0.z, l(1)
|
||||
endloop
|
||||
bfi r0.x, l(31), l(1), r0.x, l(1)
|
||||
itof r0.x, r0.x
|
||||
div r0.xyzw, r2.xyzw, r0.xxxx
|
||||
mov r1.x, vThreadGroupID.y
|
||||
store_uav_typed u0.xyzw, r1.xyzw, r0.xyzw
|
||||
endif
|
||||
ret
|
||||
// Approximately 89 instruction slots used
|
||||
#endif
|
||||
|
||||
const BYTE g_BoxBlurCS[] =
|
||||
{
|
||||
68, 88, 66, 67, 109, 156,
|
||||
172, 79, 78, 198, 69, 61,
|
||||
87, 5, 27, 232, 85, 229,
|
||||
69, 181, 1, 0, 0, 0,
|
||||
212, 10, 0, 0, 5, 0,
|
||||
0, 0, 52, 0, 0, 0,
|
||||
80, 2, 0, 0, 96, 2,
|
||||
0, 0, 112, 2, 0, 0,
|
||||
56, 10, 0, 0, 82, 68,
|
||||
69, 70, 20, 2, 0, 0,
|
||||
1, 0, 0, 0, 192, 0,
|
||||
0, 0, 3, 0, 0, 0,
|
||||
60, 0, 0, 0, 0, 5,
|
||||
83, 67, 0, 1, 0, 0,
|
||||
233, 1, 0, 0, 82, 68,
|
||||
49, 49, 60, 0, 0, 0,
|
||||
24, 0, 0, 0, 32, 0,
|
||||
0, 0, 40, 0, 0, 0,
|
||||
36, 0, 0, 0, 12, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
156, 0, 0, 0, 2, 0,
|
||||
0, 0, 5, 0, 0, 0,
|
||||
4, 0, 0, 0, 255, 255,
|
||||
255, 255, 0, 0, 0, 0,
|
||||
1, 0, 0, 0, 13, 0,
|
||||
0, 0, 165, 0, 0, 0,
|
||||
4, 0, 0, 0, 5, 0,
|
||||
0, 0, 4, 0, 0, 0,
|
||||
255, 255, 255, 255, 0, 0,
|
||||
0, 0, 1, 0, 0, 0,
|
||||
13, 0, 0, 0, 175, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 1, 0,
|
||||
0, 0, 1, 0, 0, 0,
|
||||
73, 110, 112, 117, 116, 84,
|
||||
101, 120, 0, 79, 117, 116,
|
||||
112, 117, 116, 84, 101, 120,
|
||||
0, 66, 108, 117, 114, 67,
|
||||
111, 110, 115, 116, 97, 110,
|
||||
116, 115, 0, 171, 171, 171,
|
||||
175, 0, 0, 0, 4, 0,
|
||||
0, 0, 216, 0, 0, 0,
|
||||
16, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
120, 1, 0, 0, 0, 0,
|
||||
0, 0, 4, 0, 0, 0,
|
||||
2, 0, 0, 0, 136, 1,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
255, 255, 255, 255, 0, 0,
|
||||
0, 0, 255, 255, 255, 255,
|
||||
0, 0, 0, 0, 172, 1,
|
||||
0, 0, 4, 0, 0, 0,
|
||||
4, 0, 0, 0, 2, 0,
|
||||
0, 0, 184, 1, 0, 0,
|
||||
0, 0, 0, 0, 255, 255,
|
||||
255, 255, 0, 0, 0, 0,
|
||||
255, 255, 255, 255, 0, 0,
|
||||
0, 0, 220, 1, 0, 0,
|
||||
8, 0, 0, 0, 4, 0,
|
||||
0, 0, 2, 0, 0, 0,
|
||||
136, 1, 0, 0, 0, 0,
|
||||
0, 0, 255, 255, 255, 255,
|
||||
0, 0, 0, 0, 255, 255,
|
||||
255, 255, 0, 0, 0, 0,
|
||||
226, 1, 0, 0, 12, 0,
|
||||
0, 0, 4, 0, 0, 0,
|
||||
2, 0, 0, 0, 136, 1,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
255, 255, 255, 255, 0, 0,
|
||||
0, 0, 255, 255, 255, 255,
|
||||
0, 0, 0, 0, 68, 105,
|
||||
114, 101, 99, 116, 105, 111,
|
||||
110, 0, 100, 119, 111, 114,
|
||||
100, 0, 0, 0, 19, 0,
|
||||
1, 0, 1, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 130, 1,
|
||||
0, 0, 82, 97, 100, 105,
|
||||
117, 115, 0, 105, 110, 116,
|
||||
0, 171, 0, 0, 2, 0,
|
||||
1, 0, 1, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 179, 1,
|
||||
0, 0, 87, 105, 100, 116,
|
||||
104, 0, 72, 101, 105, 103,
|
||||
104, 116, 0, 77, 105, 99,
|
||||
114, 111, 115, 111, 102, 116,
|
||||
32, 40, 82, 41, 32, 72,
|
||||
76, 83, 76, 32, 83, 104,
|
||||
97, 100, 101, 114, 32, 67,
|
||||
111, 109, 112, 105, 108, 101,
|
||||
114, 32, 49, 48, 46, 49,
|
||||
0, 171, 171, 171, 73, 83,
|
||||
71, 78, 8, 0, 0, 0,
|
||||
0, 0, 0, 0, 8, 0,
|
||||
0, 0, 79, 83, 71, 78,
|
||||
8, 0, 0, 0, 0, 0,
|
||||
0, 0, 8, 0, 0, 0,
|
||||
83, 72, 69, 88, 192, 7,
|
||||
0, 0, 80, 0, 5, 0,
|
||||
240, 1, 0, 0, 106, 8,
|
||||
0, 1, 89, 0, 0, 4,
|
||||
70, 142, 32, 0, 0, 0,
|
||||
0, 0, 1, 0, 0, 0,
|
||||
88, 24, 0, 4, 0, 112,
|
||||
16, 0, 0, 0, 0, 0,
|
||||
85, 85, 0, 0, 156, 24,
|
||||
0, 4, 0, 224, 17, 0,
|
||||
0, 0, 0, 0, 85, 85,
|
||||
0, 0, 95, 0, 0, 2,
|
||||
50, 16, 2, 0, 95, 0,
|
||||
0, 2, 18, 32, 2, 0,
|
||||
104, 0, 0, 2, 4, 0,
|
||||
0, 0, 160, 0, 0, 5,
|
||||
0, 240, 17, 0, 0, 0,
|
||||
0, 0, 16, 0, 0, 0,
|
||||
64, 1, 0, 0, 155, 0,
|
||||
0, 4, 0, 1, 0, 0,
|
||||
1, 0, 0, 0, 1, 0,
|
||||
0, 0, 37, 0, 0, 8,
|
||||
18, 0, 16, 0, 0, 0,
|
||||
0, 0, 26, 128, 32, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 1, 64, 0, 0,
|
||||
32, 0, 0, 0, 41, 0,
|
||||
0, 7, 34, 0, 16, 0,
|
||||
0, 0, 0, 0, 10, 0,
|
||||
16, 0, 0, 0, 0, 0,
|
||||
1, 64, 0, 0, 1, 0,
|
||||
0, 0, 30, 0, 0, 7,
|
||||
66, 0, 16, 0, 0, 0,
|
||||
0, 0, 26, 0, 16, 0,
|
||||
0, 0, 0, 0, 1, 64,
|
||||
0, 0, 0, 1, 0, 0,
|
||||
31, 0, 0, 4, 10, 128,
|
||||
32, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 80, 0,
|
||||
0, 7, 130, 0, 16, 0,
|
||||
0, 0, 0, 0, 26, 16,
|
||||
2, 0, 58, 128, 32, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 31, 0, 4, 3,
|
||||
58, 0, 16, 0, 0, 0,
|
||||
0, 0, 62, 0, 0, 1,
|
||||
21, 0, 0, 1, 41, 0,
|
||||
0, 6, 130, 0, 16, 0,
|
||||
0, 0, 0, 0, 10, 16,
|
||||
2, 0, 1, 64, 0, 0,
|
||||
8, 0, 0, 0, 30, 0,
|
||||
0, 8, 18, 0, 16, 0,
|
||||
1, 0, 0, 0, 42, 128,
|
||||
32, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 1, 64,
|
||||
0, 0, 255, 255, 255, 255,
|
||||
54, 0, 0, 4, 34, 0,
|
||||
16, 0, 2, 0, 0, 0,
|
||||
26, 16, 2, 0, 54, 0,
|
||||
0, 8, 194, 0, 16, 0,
|
||||
2, 0, 0, 0, 2, 64,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
54, 0, 0, 4, 34, 0,
|
||||
16, 0, 1, 0, 0, 0,
|
||||
10, 32, 2, 0, 48, 0,
|
||||
0, 1, 33, 0, 0, 7,
|
||||
66, 0, 16, 0, 1, 0,
|
||||
0, 0, 26, 0, 16, 0,
|
||||
1, 0, 0, 0, 42, 0,
|
||||
16, 0, 0, 0, 0, 0,
|
||||
3, 0, 4, 3, 42, 0,
|
||||
16, 0, 1, 0, 0, 0,
|
||||
30, 0, 0, 7, 66, 0,
|
||||
16, 0, 1, 0, 0, 0,
|
||||
58, 0, 16, 0, 0, 0,
|
||||
0, 0, 26, 0, 16, 0,
|
||||
1, 0, 0, 0, 30, 0,
|
||||
0, 8, 66, 0, 16, 0,
|
||||
1, 0, 0, 0, 10, 0,
|
||||
16, 128, 65, 0, 0, 0,
|
||||
0, 0, 0, 0, 42, 0,
|
||||
16, 0, 1, 0, 0, 0,
|
||||
36, 0, 0, 7, 66, 0,
|
||||
16, 0, 1, 0, 0, 0,
|
||||
42, 0, 16, 0, 1, 0,
|
||||
0, 0, 1, 64, 0, 0,
|
||||
0, 0, 0, 0, 37, 0,
|
||||
0, 7, 18, 0, 16, 0,
|
||||
2, 0, 0, 0, 10, 0,
|
||||
16, 0, 1, 0, 0, 0,
|
||||
42, 0, 16, 0, 1, 0,
|
||||
0, 0, 45, 0, 0, 137,
|
||||
194, 0, 0, 128, 67, 85,
|
||||
21, 0, 242, 0, 16, 0,
|
||||
3, 0, 0, 0, 70, 14,
|
||||
16, 0, 2, 0, 0, 0,
|
||||
70, 126, 16, 0, 0, 0,
|
||||
0, 0, 168, 0, 0, 9,
|
||||
242, 240, 17, 0, 0, 0,
|
||||
0, 0, 26, 0, 16, 0,
|
||||
1, 0, 0, 0, 1, 64,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
70, 14, 16, 0, 3, 0,
|
||||
0, 0, 30, 0, 0, 7,
|
||||
34, 0, 16, 0, 1, 0,
|
||||
0, 0, 26, 0, 16, 0,
|
||||
1, 0, 0, 0, 1, 64,
|
||||
0, 0, 0, 1, 0, 0,
|
||||
22, 0, 0, 1, 190, 24,
|
||||
0, 1, 35, 0, 0, 7,
|
||||
18, 0, 16, 0, 1, 0,
|
||||
0, 0, 10, 16, 2, 0,
|
||||
1, 64, 0, 0, 0, 1,
|
||||
0, 0, 10, 32, 2, 0,
|
||||
80, 0, 0, 8, 130, 0,
|
||||
16, 0, 0, 0, 0, 0,
|
||||
10, 0, 16, 0, 1, 0,
|
||||
0, 0, 42, 128, 32, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 31, 0, 4, 3,
|
||||
58, 0, 16, 0, 0, 0,
|
||||
0, 0, 62, 0, 0, 1,
|
||||
21, 0, 0, 1, 54, 0,
|
||||
0, 8, 242, 0, 16, 0,
|
||||
2, 0, 0, 0, 2, 64,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
54, 0, 0, 5, 130, 0,
|
||||
16, 0, 0, 0, 0, 0,
|
||||
1, 64, 0, 0, 0, 0,
|
||||
0, 0, 48, 0, 0, 1,
|
||||
34, 0, 0, 7, 18, 0,
|
||||
16, 0, 3, 0, 0, 0,
|
||||
26, 0, 16, 0, 0, 0,
|
||||
0, 0, 58, 0, 16, 0,
|
||||
0, 0, 0, 0, 3, 0,
|
||||
4, 3, 10, 0, 16, 0,
|
||||
3, 0, 0, 0, 30, 0,
|
||||
0, 6, 18, 0, 16, 0,
|
||||
3, 0, 0, 0, 58, 0,
|
||||
16, 0, 0, 0, 0, 0,
|
||||
10, 32, 2, 0, 167, 0,
|
||||
0, 9, 242, 0, 16, 0,
|
||||
3, 0, 0, 0, 10, 0,
|
||||
16, 0, 3, 0, 0, 0,
|
||||
1, 64, 0, 0, 0, 0,
|
||||
0, 0, 70, 254, 17, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 7, 242, 0, 16, 0,
|
||||
2, 0, 0, 0, 70, 14,
|
||||
16, 0, 2, 0, 0, 0,
|
||||
70, 14, 16, 0, 3, 0,
|
||||
0, 0, 30, 0, 0, 7,
|
||||
130, 0, 16, 0, 0, 0,
|
||||
0, 0, 58, 0, 16, 0,
|
||||
0, 0, 0, 0, 1, 64,
|
||||
0, 0, 1, 0, 0, 0,
|
||||
22, 0, 0, 1, 140, 0,
|
||||
0, 11, 130, 0, 16, 0,
|
||||
0, 0, 0, 0, 1, 64,
|
||||
0, 0, 31, 0, 0, 0,
|
||||
1, 64, 0, 0, 1, 0,
|
||||
0, 0, 10, 0, 16, 0,
|
||||
0, 0, 0, 0, 1, 64,
|
||||
0, 0, 1, 0, 0, 0,
|
||||
43, 0, 0, 5, 130, 0,
|
||||
16, 0, 0, 0, 0, 0,
|
||||
58, 0, 16, 0, 0, 0,
|
||||
0, 0, 14, 0, 0, 7,
|
||||
242, 0, 16, 0, 2, 0,
|
||||
0, 0, 70, 14, 16, 0,
|
||||
2, 0, 0, 0, 246, 15,
|
||||
16, 0, 0, 0, 0, 0,
|
||||
54, 0, 0, 4, 226, 0,
|
||||
16, 0, 1, 0, 0, 0,
|
||||
86, 21, 2, 0, 164, 0,
|
||||
0, 7, 242, 224, 17, 0,
|
||||
0, 0, 0, 0, 70, 14,
|
||||
16, 0, 1, 0, 0, 0,
|
||||
70, 14, 16, 0, 2, 0,
|
||||
0, 0, 18, 0, 0, 1,
|
||||
80, 0, 0, 7, 130, 0,
|
||||
16, 0, 0, 0, 0, 0,
|
||||
26, 16, 2, 0, 42, 128,
|
||||
32, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 31, 0,
|
||||
4, 3, 58, 0, 16, 0,
|
||||
0, 0, 0, 0, 62, 0,
|
||||
0, 1, 21, 0, 0, 1,
|
||||
41, 0, 0, 6, 130, 0,
|
||||
16, 0, 0, 0, 0, 0,
|
||||
10, 16, 2, 0, 1, 64,
|
||||
0, 0, 8, 0, 0, 0,
|
||||
30, 0, 0, 8, 18, 0,
|
||||
16, 0, 1, 0, 0, 0,
|
||||
58, 128, 32, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
1, 64, 0, 0, 255, 255,
|
||||
255, 255, 54, 0, 0, 4,
|
||||
18, 0, 16, 0, 2, 0,
|
||||
0, 0, 26, 16, 2, 0,
|
||||
54, 0, 0, 8, 194, 0,
|
||||
16, 0, 2, 0, 0, 0,
|
||||
2, 64, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 54, 0, 0, 4,
|
||||
34, 0, 16, 0, 1, 0,
|
||||
0, 0, 10, 32, 2, 0,
|
||||
48, 0, 0, 1, 33, 0,
|
||||
0, 7, 66, 0, 16, 0,
|
||||
1, 0, 0, 0, 26, 0,
|
||||
16, 0, 1, 0, 0, 0,
|
||||
42, 0, 16, 0, 0, 0,
|
||||
0, 0, 3, 0, 4, 3,
|
||||
42, 0, 16, 0, 1, 0,
|
||||
0, 0, 30, 0, 0, 7,
|
||||
66, 0, 16, 0, 1, 0,
|
||||
0, 0, 58, 0, 16, 0,
|
||||
0, 0, 0, 0, 26, 0,
|
||||
16, 0, 1, 0, 0, 0,
|
||||
30, 0, 0, 8, 66, 0,
|
||||
16, 0, 1, 0, 0, 0,
|
||||
10, 0, 16, 128, 65, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
42, 0, 16, 0, 1, 0,
|
||||
0, 0, 36, 0, 0, 7,
|
||||
66, 0, 16, 0, 1, 0,
|
||||
0, 0, 42, 0, 16, 0,
|
||||
1, 0, 0, 0, 1, 64,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
37, 0, 0, 7, 34, 0,
|
||||
16, 0, 2, 0, 0, 0,
|
||||
10, 0, 16, 0, 1, 0,
|
||||
0, 0, 42, 0, 16, 0,
|
||||
1, 0, 0, 0, 45, 0,
|
||||
0, 137, 194, 0, 0, 128,
|
||||
67, 85, 21, 0, 242, 0,
|
||||
16, 0, 3, 0, 0, 0,
|
||||
70, 14, 16, 0, 2, 0,
|
||||
0, 0, 70, 126, 16, 0,
|
||||
0, 0, 0, 0, 168, 0,
|
||||
0, 9, 242, 240, 17, 0,
|
||||
0, 0, 0, 0, 26, 0,
|
||||
16, 0, 1, 0, 0, 0,
|
||||
1, 64, 0, 0, 0, 0,
|
||||
0, 0, 70, 14, 16, 0,
|
||||
3, 0, 0, 0, 30, 0,
|
||||
0, 7, 34, 0, 16, 0,
|
||||
1, 0, 0, 0, 26, 0,
|
||||
16, 0, 1, 0, 0, 0,
|
||||
1, 64, 0, 0, 0, 1,
|
||||
0, 0, 22, 0, 0, 1,
|
||||
190, 24, 0, 1, 35, 0,
|
||||
0, 10, 226, 0, 16, 0,
|
||||
1, 0, 0, 0, 6, 16,
|
||||
2, 0, 2, 64, 0, 0,
|
||||
0, 0, 0, 0, 0, 1,
|
||||
0, 0, 0, 1, 0, 0,
|
||||
0, 1, 0, 0, 6, 32,
|
||||
2, 0, 80, 0, 0, 8,
|
||||
66, 0, 16, 0, 0, 0,
|
||||
0, 0, 58, 0, 16, 0,
|
||||
1, 0, 0, 0, 58, 128,
|
||||
32, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 31, 0,
|
||||
4, 3, 42, 0, 16, 0,
|
||||
0, 0, 0, 0, 62, 0,
|
||||
0, 1, 21, 0, 0, 1,
|
||||
54, 0, 0, 8, 242, 0,
|
||||
16, 0, 2, 0, 0, 0,
|
||||
2, 64, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 54, 0, 0, 5,
|
||||
66, 0, 16, 0, 0, 0,
|
||||
0, 0, 1, 64, 0, 0,
|
||||
0, 0, 0, 0, 48, 0,
|
||||
0, 1, 34, 0, 0, 7,
|
||||
130, 0, 16, 0, 0, 0,
|
||||
0, 0, 26, 0, 16, 0,
|
||||
0, 0, 0, 0, 42, 0,
|
||||
16, 0, 0, 0, 0, 0,
|
||||
3, 0, 4, 3, 58, 0,
|
||||
16, 0, 0, 0, 0, 0,
|
||||
30, 0, 0, 6, 130, 0,
|
||||
16, 0, 0, 0, 0, 0,
|
||||
42, 0, 16, 0, 0, 0,
|
||||
0, 0, 10, 32, 2, 0,
|
||||
167, 0, 0, 9, 242, 0,
|
||||
16, 0, 3, 0, 0, 0,
|
||||
58, 0, 16, 0, 0, 0,
|
||||
0, 0, 1, 64, 0, 0,
|
||||
0, 0, 0, 0, 70, 254,
|
||||
17, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 7, 242, 0,
|
||||
16, 0, 2, 0, 0, 0,
|
||||
70, 14, 16, 0, 2, 0,
|
||||
0, 0, 70, 14, 16, 0,
|
||||
3, 0, 0, 0, 30, 0,
|
||||
0, 7, 66, 0, 16, 0,
|
||||
0, 0, 0, 0, 42, 0,
|
||||
16, 0, 0, 0, 0, 0,
|
||||
1, 64, 0, 0, 1, 0,
|
||||
0, 0, 22, 0, 0, 1,
|
||||
140, 0, 0, 11, 18, 0,
|
||||
16, 0, 0, 0, 0, 0,
|
||||
1, 64, 0, 0, 31, 0,
|
||||
0, 0, 1, 64, 0, 0,
|
||||
1, 0, 0, 0, 10, 0,
|
||||
16, 0, 0, 0, 0, 0,
|
||||
1, 64, 0, 0, 1, 0,
|
||||
0, 0, 43, 0, 0, 5,
|
||||
18, 0, 16, 0, 0, 0,
|
||||
0, 0, 10, 0, 16, 0,
|
||||
0, 0, 0, 0, 14, 0,
|
||||
0, 7, 242, 0, 16, 0,
|
||||
0, 0, 0, 0, 70, 14,
|
||||
16, 0, 2, 0, 0, 0,
|
||||
6, 0, 16, 0, 0, 0,
|
||||
0, 0, 54, 0, 0, 4,
|
||||
18, 0, 16, 0, 1, 0,
|
||||
0, 0, 26, 16, 2, 0,
|
||||
164, 0, 0, 7, 242, 224,
|
||||
17, 0, 0, 0, 0, 0,
|
||||
70, 14, 16, 0, 1, 0,
|
||||
0, 0, 70, 14, 16, 0,
|
||||
0, 0, 0, 0, 21, 0,
|
||||
0, 1, 62, 0, 0, 1,
|
||||
83, 84, 65, 84, 148, 0,
|
||||
0, 0, 89, 0, 0, 0,
|
||||
4, 0, 0, 0, 0, 0,
|
||||
0, 0, 2, 0, 0, 0,
|
||||
4, 0, 0, 0, 27, 0,
|
||||
0, 0, 4, 0, 0, 0,
|
||||
7, 0, 0, 0, 8, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 2, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
12, 0, 0, 0, 0, 0,
|
||||
0, 0, 2, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
2, 0, 0, 0, 0, 0,
|
||||
0, 0, 2, 0, 0, 0
|
||||
};
|
||||
115
src/modules/ZoomIt/ZoomIt/BoxBlurCS.hlsl
Normal file
115
src/modules/ZoomIt/ZoomIt/BoxBlurCS.hlsl
Normal file
@@ -0,0 +1,115 @@
|
||||
//==============================================================================
|
||||
//
|
||||
// BoxBlurCS.hlsl
|
||||
//
|
||||
// D3D11 compute shader for separable box blur. Each dispatch performs
|
||||
// either a horizontal or a vertical pass controlled by the Direction
|
||||
// constant. Run four dispatches (H→V→H→V) to approximate a Gaussian.
|
||||
//
|
||||
// Uses group-shared memory so each texel is loaded once per thread
|
||||
// group, then reused across the sliding window.
|
||||
//
|
||||
// Entry point:
|
||||
// CSMain (cs_5_0)
|
||||
//
|
||||
// Recompile with:
|
||||
// fxc /T cs_5_0 /E CSMain /Fh BoxBlurCS.h /Vn g_BoxBlurCS BoxBlurCS.hlsl
|
||||
//
|
||||
// Copyright (C) Mark Russinovich
|
||||
// Sysinternals - www.sysinternals.com
|
||||
//
|
||||
//==============================================================================
|
||||
|
||||
// Input texture (read-only).
|
||||
Texture2D<float4> InputTex : register(t0);
|
||||
|
||||
// Output texture (write-only).
|
||||
RWTexture2D<float4> OutputTex : register(u0);
|
||||
|
||||
cbuffer BlurConstants : register(b0)
|
||||
{
|
||||
uint Direction; // 0 = horizontal, 1 = vertical
|
||||
int Radius; // Box blur radius in pixels
|
||||
uint Width; // Image width
|
||||
uint Height; // Image height
|
||||
};
|
||||
|
||||
// Thread group: 256 threads along the blur axis.
|
||||
#define GROUP_SIZE 256
|
||||
|
||||
// Max radius we support. Shared memory = (GROUP_SIZE + 2*MAX_RADIUS) * 4 floats * 4 bytes
|
||||
// = (256 + 64) * 16 = ~5 KB, well within the 32 KB limit.
|
||||
#define MAX_RADIUS 32
|
||||
|
||||
// Shared memory tile: enough for the group + apron on both sides.
|
||||
groupshared float4 tile[GROUP_SIZE + 2 * MAX_RADIUS];
|
||||
|
||||
[numthreads(GROUP_SIZE, 1, 1)]
|
||||
void CSMain( uint3 groupId : SV_GroupID,
|
||||
uint3 groupTid : SV_GroupThreadID,
|
||||
uint3 dispatchId : SV_DispatchThreadID )
|
||||
{
|
||||
int r = min( Radius, MAX_RADIUS );
|
||||
int tileSize = GROUP_SIZE + 2 * r;
|
||||
int tid = (int)groupTid.x;
|
||||
|
||||
if( Direction == 0 )
|
||||
{
|
||||
// ── Horizontal pass ────────────────────────────────────────
|
||||
int row = (int)groupId.y;
|
||||
if( (uint)row >= Height )
|
||||
return;
|
||||
|
||||
int groupStart = (int)groupId.x * GROUP_SIZE;
|
||||
|
||||
// Load tile: each thread loads its primary texel + apron.
|
||||
for( int i = tid; i < tileSize; i += GROUP_SIZE )
|
||||
{
|
||||
int srcX = clamp( groupStart + i - r, 0, (int)Width - 1 );
|
||||
tile[i] = InputTex[int2( srcX, row )];
|
||||
}
|
||||
GroupMemoryBarrierWithGroupSync();
|
||||
|
||||
int outX = groupStart + tid;
|
||||
if( (uint)outX >= Width )
|
||||
return;
|
||||
|
||||
// Sum the window from shared memory.
|
||||
float4 sum = (float4)0;
|
||||
int windowStart = tid; // tile index = tid + r - r
|
||||
for( int k = 0; k <= 2 * r; k++ )
|
||||
{
|
||||
sum += tile[windowStart + k];
|
||||
}
|
||||
OutputTex[int2( outX, row )] = sum / (float)( 2 * r + 1 );
|
||||
}
|
||||
else
|
||||
{
|
||||
// ── Vertical pass ──────────────────────────────────────────
|
||||
int col = (int)groupId.y;
|
||||
if( (uint)col >= Width )
|
||||
return;
|
||||
|
||||
int groupStart = (int)groupId.x * GROUP_SIZE;
|
||||
|
||||
// Load tile.
|
||||
for( int i = tid; i < tileSize; i += GROUP_SIZE )
|
||||
{
|
||||
int srcY = clamp( groupStart + i - r, 0, (int)Height - 1 );
|
||||
tile[i] = InputTex[int2( col, srcY )];
|
||||
}
|
||||
GroupMemoryBarrierWithGroupSync();
|
||||
|
||||
int outY = groupStart + tid;
|
||||
if( (uint)outY >= Height )
|
||||
return;
|
||||
|
||||
float4 sum = (float4)0;
|
||||
int windowStart = tid;
|
||||
for( int k = 0; k <= 2 * r; k++ )
|
||||
{
|
||||
sum += tile[windowStart + k];
|
||||
}
|
||||
OutputTex[int2( col, outY )] = sum / (float)( 2 * r + 1 );
|
||||
}
|
||||
}
|
||||
@@ -77,6 +77,7 @@ std::optional<CaptureFrame> CaptureFrameWait::TryGetNextFrame()
|
||||
if (m_currentFrame != nullptr)
|
||||
{
|
||||
m_currentFrame.Close();
|
||||
m_currentFrame = nullptr; // Prevent double-Close on subsequent calls
|
||||
}
|
||||
m_nextFrameEvent.ResetEvent();
|
||||
|
||||
@@ -107,6 +108,33 @@ std::optional<CaptureFrame> CaptureFrameWait::TryGetNextFrame()
|
||||
}
|
||||
|
||||
|
||||
//----------------------------------------------------------------------------
|
||||
//
|
||||
// CaptureFrameWait::PeekCurrentFrame
|
||||
//
|
||||
// Returns the frame that is currently held (if any) without closing
|
||||
// it and without waiting for a new one. This is useful during
|
||||
// recording startup: the constructor captured a frame when the
|
||||
// session began, and OnMediaStreamSourceStarting can use it
|
||||
// immediately instead of blocking until the next desktop change.
|
||||
// The frame remains alive in the pool until the next
|
||||
// TryGetNextFrame() call closes it.
|
||||
//
|
||||
//----------------------------------------------------------------------------
|
||||
std::optional<CaptureFrame> CaptureFrameWait::PeekCurrentFrame() const
|
||||
{
|
||||
if (m_currentFrame != nullptr)
|
||||
{
|
||||
return std::optional<CaptureFrame>(
|
||||
{
|
||||
m_currentFrame.Surface(),
|
||||
m_currentFrame.ContentSize(),
|
||||
m_currentFrame.SystemRelativeTime(),
|
||||
});
|
||||
}
|
||||
return std::nullopt;
|
||||
}
|
||||
|
||||
//----------------------------------------------------------------------------
|
||||
//
|
||||
// CaptureFrameWait::TryGetNextFrame (with timeout)
|
||||
@@ -121,6 +149,7 @@ std::optional<CaptureFrame> CaptureFrameWait::TryGetNextFrame( DWORD timeoutMs )
|
||||
if( m_currentFrame != nullptr )
|
||||
{
|
||||
m_currentFrame.Close();
|
||||
m_currentFrame = nullptr; // Prevent double-Close on subsequent calls
|
||||
}
|
||||
m_nextFrameEvent.ResetEvent();
|
||||
|
||||
|
||||
@@ -108,6 +108,7 @@ public:
|
||||
|
||||
std::optional<CaptureFrame> TryGetNextFrame();
|
||||
std::optional<CaptureFrame> TryGetNextFrame( DWORD timeoutMs );
|
||||
std::optional<CaptureFrame> PeekCurrentFrame() const;
|
||||
void StopCapture();
|
||||
void EnableCursorCapture( bool enable = true )
|
||||
{
|
||||
|
||||
113
src/modules/ZoomIt/ZoomIt/NoiseSuppressor.cpp
Normal file
113
src/modules/ZoomIt/ZoomIt/NoiseSuppressor.cpp
Normal file
@@ -0,0 +1,113 @@
|
||||
#include "pch.h"
|
||||
#include "NoiseSuppressor.h"
|
||||
|
||||
extern "C" {
|
||||
#include "rnnoise/rnnoise.h"
|
||||
}
|
||||
|
||||
// RNNoise processes 480 mono samples per frame (10ms at 48kHz)
|
||||
static constexpr uint32_t RNNOISE_FRAME_SIZE = 480;
|
||||
|
||||
// RNNoise expects samples in PCM16 range (-32768 to 32767), not normalized float [-1, 1]
|
||||
static constexpr float PCM16_SCALE = 32768.0f;
|
||||
static constexpr float PCM16_SCALE_INV = 1.0f / 32768.0f;
|
||||
|
||||
NoiseSuppressor::NoiseSuppressor()
|
||||
{
|
||||
}
|
||||
|
||||
NoiseSuppressor::~NoiseSuppressor()
|
||||
{
|
||||
for (auto& channel : m_channels)
|
||||
{
|
||||
if (channel.state)
|
||||
{
|
||||
rnnoise_destroy(channel.state);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
void NoiseSuppressor::EnsureChannelCount(uint32_t channels)
|
||||
{
|
||||
if (m_channels.size() == channels)
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
// Channel count changed (or first call): rebuild per-channel RNNoise state.
|
||||
for (auto& channel : m_channels)
|
||||
{
|
||||
if (channel.state)
|
||||
{
|
||||
rnnoise_destroy(channel.state);
|
||||
}
|
||||
}
|
||||
|
||||
m_channels.clear();
|
||||
m_channels.resize(channels);
|
||||
for (auto& channel : m_channels)
|
||||
{
|
||||
channel.state = rnnoise_create(nullptr);
|
||||
}
|
||||
}
|
||||
|
||||
void NoiseSuppressor::Process(float* samples, uint32_t sampleCount, uint32_t channels)
|
||||
{
|
||||
if (sampleCount == 0 || channels == 0)
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
EnsureChannelCount(channels);
|
||||
|
||||
uint32_t frameCount = sampleCount / channels;
|
||||
|
||||
// Denoise each channel independently so the original channel layout is
|
||||
// preserved (e.g. a mic wired only to the left channel stays on the left
|
||||
// and silent channels stay silent instead of being filled with the voice).
|
||||
for (uint32_t ch = 0; ch < channels; ch++)
|
||||
{
|
||||
ChannelState& channel = m_channels[ch];
|
||||
if (!channel.state)
|
||||
{
|
||||
continue;
|
||||
}
|
||||
|
||||
uint32_t residualCount = static_cast<uint32_t>(channel.residual.size());
|
||||
uint32_t totalSamples = residualCount + frameCount;
|
||||
|
||||
channel.work.resize(totalSamples);
|
||||
|
||||
// Copy residual from previous call
|
||||
if (residualCount > 0)
|
||||
{
|
||||
memcpy(channel.work.data(), channel.residual.data(), residualCount * sizeof(float));
|
||||
}
|
||||
|
||||
// Deinterleave this channel and scale to PCM16 range for RNNoise
|
||||
for (uint32_t i = 0; i < frameCount; i++)
|
||||
{
|
||||
channel.work[residualCount + i] = samples[i * channels + ch] * PCM16_SCALE;
|
||||
}
|
||||
|
||||
// Process complete 480-sample frames through RNNoise
|
||||
uint32_t processedSamples = 0;
|
||||
while (processedSamples + RNNOISE_FRAME_SIZE <= totalSamples)
|
||||
{
|
||||
rnnoise_process_frame(channel.state, &channel.work[processedSamples], &channel.work[processedSamples]);
|
||||
processedSamples += RNNOISE_FRAME_SIZE;
|
||||
}
|
||||
|
||||
// Save unprocessed residual for next call
|
||||
channel.residual.assign(
|
||||
channel.work.begin() + processedSamples,
|
||||
channel.work.end());
|
||||
|
||||
// Write denoised samples back to the interleaved buffer, scaling back to
|
||||
// normalized float. Only this call's input region is written back.
|
||||
for (uint32_t i = 0; i < frameCount; i++)
|
||||
{
|
||||
samples[i * channels + ch] = channel.work[residualCount + i] * PCM16_SCALE_INV;
|
||||
}
|
||||
}
|
||||
}
|
||||
37
src/modules/ZoomIt/ZoomIt/NoiseSuppressor.h
Normal file
37
src/modules/ZoomIt/ZoomIt/NoiseSuppressor.h
Normal file
@@ -0,0 +1,37 @@
|
||||
#pragma once
|
||||
|
||||
#include <vector>
|
||||
#include <stdint.h>
|
||||
|
||||
struct DenoiseState;
|
||||
|
||||
class NoiseSuppressor
|
||||
{
|
||||
public:
|
||||
NoiseSuppressor();
|
||||
~NoiseSuppressor();
|
||||
|
||||
NoiseSuppressor(const NoiseSuppressor&) = delete;
|
||||
NoiseSuppressor& operator=(const NoiseSuppressor&) = delete;
|
||||
|
||||
// Process interleaved multi-channel float samples in-place.
|
||||
// Each channel is denoised independently through its own RNNoise state in
|
||||
// 480-sample frames, preserving the original channel layout (e.g. a mic
|
||||
// wired only to the left channel stays on the left and is not duplicated
|
||||
// onto the right).
|
||||
void Process(float* samples, uint32_t sampleCount, uint32_t channels);
|
||||
|
||||
private:
|
||||
// Per-channel RNNoise state and buffers so each channel is denoised
|
||||
// independently and the channel layout is preserved.
|
||||
struct ChannelState
|
||||
{
|
||||
DenoiseState* state = nullptr;
|
||||
std::vector<float> work; // Working buffer for the current quantum
|
||||
std::vector<float> residual; // Leftover samples from previous quantum
|
||||
};
|
||||
|
||||
void EnsureChannelCount(uint32_t channels);
|
||||
|
||||
std::vector<ChannelState> m_channels;
|
||||
};
|
||||
@@ -32,6 +32,9 @@ extern DWORD g_WebcamPosition;
|
||||
extern DWORD g_WebcamSize;
|
||||
extern DWORD g_WebcamShape;
|
||||
extern TCHAR g_WebcamDeviceSymLink[MAX_PATH];
|
||||
extern DWORD g_WebcamBackgroundMode;
|
||||
extern TCHAR g_WebcamBackgroundImage[];
|
||||
extern DWORD g_WebcamBrightness;
|
||||
extern class ClassRegistry reg;
|
||||
extern REG_SETTING RegSettings[];
|
||||
extern HINSTANCE g_hInstance;
|
||||
@@ -106,6 +109,34 @@ static double RecDiagElapsedMs()
|
||||
return static_cast<double>( now.QuadPart - s_origin.QuadPart ) * 1000.0 / s_freq.QuadPart;
|
||||
}
|
||||
|
||||
static FILE* s_recDiagFile = nullptr;
|
||||
|
||||
static void RecDiagOpenFile()
|
||||
{
|
||||
if( !s_recDiagFile )
|
||||
{
|
||||
wchar_t path[MAX_PATH];
|
||||
if( ExpandEnvironmentStringsW( L"%TEMP%\\ZoomIt_RecDiag.log", path, MAX_PATH ) )
|
||||
{
|
||||
_wfopen_s( &s_recDiagFile, path, L"a" );
|
||||
if( s_recDiagFile )
|
||||
{
|
||||
fwprintf( s_recDiagFile, L"\n===== NEW SESSION =====\n" );
|
||||
fflush( s_recDiagFile );
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
static void RecDiagCloseFile()
|
||||
{
|
||||
if( s_recDiagFile )
|
||||
{
|
||||
fclose( s_recDiagFile );
|
||||
s_recDiagFile = nullptr;
|
||||
}
|
||||
}
|
||||
|
||||
static void RecDiag( const wchar_t* fmt, ... )
|
||||
{
|
||||
wchar_t buf[512];
|
||||
@@ -123,8 +154,19 @@ static void RecDiag( const wchar_t* fmt, ... )
|
||||
_vsnwprintf_s( buf + offset, _countof( buf ) - offset, _TRUNCATE, fmt, va );
|
||||
va_end( va );
|
||||
OutputDebugStringW( buf );
|
||||
|
||||
RecDiagOpenFile();
|
||||
if( s_recDiagFile )
|
||||
{
|
||||
fwprintf( s_recDiagFile, L"%s", buf );
|
||||
fflush( s_recDiagFile );
|
||||
}
|
||||
}
|
||||
|
||||
static int s_diagVideoCount = 0;
|
||||
static int s_diagAudioCount = 0;
|
||||
static int64_t s_diagStartTs = 0; // SystemRelativeTime from OnStarting
|
||||
|
||||
static bool IsGifPath(const std::wstring& path)
|
||||
{
|
||||
try
|
||||
@@ -930,12 +972,20 @@ VideoRecordingSession::VideoRecordingSession(
|
||||
winrt::GraphicsCaptureItem const& item,
|
||||
RECT const cropRect,
|
||||
uint32_t frameRate,
|
||||
bool captureAudio,
|
||||
bool captureSystemAudio,
|
||||
bool micMonoMix,
|
||||
std::unique_ptr<AudioSampleGenerator> audioGenerator,
|
||||
winrt::IAsyncAction audioInitAction,
|
||||
winrt::Streams::IRandomAccessStream const& stream)
|
||||
{
|
||||
RecDiag( L"Constructor: entry\n" );
|
||||
|
||||
// Take ownership of pre-created audio generator. Its InitializeAsync
|
||||
// was started in StartRecordingAsync so it runs in parallel with all
|
||||
// the D3D, capture-item, and webcam setup below.
|
||||
m_audioGenerator = std::move( audioGenerator );
|
||||
m_audioInitAction = audioInitAction;
|
||||
RecDiag( L"Constructor: audio generator received (init %s)\n",
|
||||
m_audioInitAction ? L"pending" : L"none" );
|
||||
|
||||
m_device = device;
|
||||
m_d3dDevice = GetDXGIInterfaceFromObject<ID3D11Device>(m_device);
|
||||
m_d3dDevice->GetImmediateContext(m_d3dContext.put());
|
||||
@@ -1038,8 +1088,15 @@ VideoRecordingSession::VideoRecordingSession(
|
||||
probeCapture.Close();
|
||||
return true;
|
||||
}
|
||||
catch( winrt::hresult_error const& ex )
|
||||
{
|
||||
RecDiag( L"Constructor: webcam probe failed hr=0x%08X: %s\n",
|
||||
static_cast<unsigned>( ex.code() ), ex.message().c_str() );
|
||||
return false;
|
||||
}
|
||||
catch( ... )
|
||||
{
|
||||
RecDiag( L"Constructor: webcam probe failed with unknown exception\n" );
|
||||
return false;
|
||||
}
|
||||
});
|
||||
@@ -1067,7 +1124,10 @@ VideoRecordingSession::VideoRecordingSession(
|
||||
static_cast<WebcamCapture::Position>( g_WebcamPosition ),
|
||||
static_cast<WebcamCapture::Size>( g_WebcamSize ),
|
||||
webcamShape,
|
||||
isFullScreenRecording );
|
||||
isFullScreenRecording,
|
||||
static_cast<WebcamBackgroundMode>( g_WebcamBackgroundMode ),
|
||||
g_WebcamBackgroundImage,
|
||||
static_cast<int>( g_WebcamBrightness ) );
|
||||
m_webcamCapture->Start();
|
||||
RecDiag( L"Constructor: WebcamCapture::Start() returned\n" );
|
||||
}
|
||||
@@ -1089,12 +1149,12 @@ VideoRecordingSession::VideoRecordingSession(
|
||||
// Store frame interval for timeout-based frame production when webcam is active.
|
||||
m_frameIntervalTicks = ( frameRate > 0 ) ? ( 10'000'000LL / frameRate ) : 333'333LL;
|
||||
|
||||
if (captureAudio || captureSystemAudio)
|
||||
if (m_audioGenerator)
|
||||
{
|
||||
// Always set up audio profile for loopback capture (stereo AAC)
|
||||
auto audio = m_encodingProfile.Audio();
|
||||
audio = winrt::AudioEncodingProperties::CreateAac(48000, 2, 192000);
|
||||
m_encodingProfile.Audio(audio);
|
||||
auto audioProps = m_audioGenerator->GetEncodingProperties();
|
||||
m_encodingProfile.Audio(winrt::AudioEncodingProperties::CreateAac(
|
||||
audioProps.SampleRate(), audioProps.ChannelCount(), 192000));
|
||||
}
|
||||
|
||||
// Describe our input: uncompressed BGRA8 buffers
|
||||
@@ -1113,15 +1173,6 @@ VideoRecordingSession::VideoRecordingSession(
|
||||
DXGI_FORMAT_B8G8R8A8_UNORM,
|
||||
2);
|
||||
|
||||
if (captureAudio || captureSystemAudio)
|
||||
{
|
||||
m_audioGenerator = std::make_unique<AudioSampleGenerator>(captureAudio, captureSystemAudio, micMonoMix);
|
||||
}
|
||||
else
|
||||
{
|
||||
m_audioGenerator = nullptr;
|
||||
}
|
||||
|
||||
// Wait for the webcam's first frame now that all other setup is done.
|
||||
// The camera was started early, so most of its ~850 ms sensor warmup has
|
||||
// overlapped with the encoding profile, swap chain, and audio generator
|
||||
@@ -1155,6 +1206,7 @@ VideoRecordingSession::VideoRecordingSession(
|
||||
VideoRecordingSession::~VideoRecordingSession()
|
||||
{
|
||||
Close();
|
||||
RecDiagCloseFile();
|
||||
}
|
||||
|
||||
|
||||
@@ -1173,8 +1225,13 @@ winrt::IAsyncAction VideoRecordingSession::StartAsync()
|
||||
// Create our MediaStreamSource
|
||||
if(m_audioGenerator) {
|
||||
|
||||
RecDiag( L"StartAsync: co_await InitializeAsync...\n" );
|
||||
co_await m_audioGenerator->InitializeAsync();
|
||||
RecDiag( L"StartAsync: co_await audio init...\n" );
|
||||
if (m_audioInitAction) {
|
||||
co_await m_audioInitAction; // started in constructor
|
||||
m_audioInitAction = nullptr;
|
||||
} else {
|
||||
co_await m_audioGenerator->InitializeAsync();
|
||||
}
|
||||
RecDiag( L"StartAsync: audio initialized\n" );
|
||||
m_streamSource = winrt::MediaStreamSource(m_videoDescriptor, winrt::AudioStreamDescriptor(m_audioGenerator->GetEncodingProperties()));
|
||||
}
|
||||
@@ -1236,6 +1293,9 @@ winrt::IAsyncAction VideoRecordingSession::StartAsync()
|
||||
//----------------------------------------------------------------------------
|
||||
void VideoRecordingSession::Close()
|
||||
{
|
||||
RecDiag( L"Close: totalVideoFrames=%d totalAudioSamples=%d\n",
|
||||
s_diagVideoCount, s_diagAudioCount );
|
||||
|
||||
// Stop webcam capture before closing the main session.
|
||||
if( m_webcamCapture )
|
||||
{
|
||||
@@ -1285,28 +1345,45 @@ void VideoRecordingSession::OnMediaStreamSourceStarting(
|
||||
winrt::MediaStreamSource const&,
|
||||
winrt::MediaStreamSourceStartingEventArgs const& args)
|
||||
{
|
||||
RecDiag( L"OnStarting: entry, calling TryGetNextFrame...\n" );
|
||||
auto frame = m_frameWait->TryGetNextFrame();
|
||||
// Close the stale frame captured in the constructor (~1-2 seconds
|
||||
// ago) and grab a FRESH frame via TryGetNextFrame. This gives us
|
||||
// both current visual content and a current SystemRelativeTime,
|
||||
// avoiding the frozen-first-frame artefact. WGC delivers a new
|
||||
// frame within one vblank (~16 ms) after the old frame is released
|
||||
// back to the pool, so a 200 ms timeout is very generous.
|
||||
RecDiag( L"OnStarting: calling TryGetNextFrame(200) for fresh frame...\n" );
|
||||
auto frame = m_frameWait->TryGetNextFrame( 200 );
|
||||
|
||||
int64_t startSRT = 0;
|
||||
|
||||
if (frame) {
|
||||
RecDiag( L"OnStarting: got frame, SystemRelativeTime=%lld (%.1fms)\n",
|
||||
frame->SystemRelativeTime.count(),
|
||||
frame->SystemRelativeTime.count() / 10000.0 );
|
||||
args.Request().SetActualStartPosition(frame->SystemRelativeTime);
|
||||
startSRT = frame->SystemRelativeTime.count();
|
||||
RecDiag( L"OnStarting: got fresh frame, SRT=%lld (%.1fms)\n",
|
||||
startSRT, startSRT / 10000.0 );
|
||||
|
||||
// Cache this frame so it can be served as the very first video
|
||||
// sample in OnMediaStreamSourceSampleRequested. Without this,
|
||||
// the frame is discarded and the first encoded sample comes from
|
||||
// the *next* capture, creating a visible timestamp gap.
|
||||
args.Request().SetActualStartPosition( frame->SystemRelativeTime );
|
||||
m_cachedStartingFrame = frame;
|
||||
|
||||
if (m_audioGenerator) {
|
||||
|
||||
RecDiag( L"OnStarting: calling AudioSampleGenerator::Start()\n" );
|
||||
m_audioGenerator->Start();
|
||||
RecDiag( L"OnStarting: audio started\n" );
|
||||
}
|
||||
m_adjustedStartSRT = startSRT;
|
||||
} else {
|
||||
RecDiag( L"OnStarting: TryGetNextFrame returned nullopt!\n" );
|
||||
// Timeout (very unlikely). Fall back to QPC-derived SRT.
|
||||
// Use double for the intermediate product to avoid int64
|
||||
// overflow (naive integer multiply overflows after ~25.6 h).
|
||||
RecDiag( L"OnStarting: TryGetNextFrame timed out, using QPC fallback\n" );
|
||||
LARGE_INTEGER qpcFreq, qpcNow;
|
||||
QueryPerformanceFrequency( &qpcFreq );
|
||||
QueryPerformanceCounter( &qpcNow );
|
||||
startSRT = static_cast<int64_t>(
|
||||
static_cast<double>( qpcNow.QuadPart ) * 10'000'000.0 / qpcFreq.QuadPart );
|
||||
args.Request().SetActualStartPosition( winrt::TimeSpan{ startSRT } );
|
||||
m_adjustedStartSRT = startSRT;
|
||||
}
|
||||
|
||||
// Start audio capture. Pass the video start SRT so audio
|
||||
// timestamps can be rebased to the same domain as video.
|
||||
if (m_audioGenerator) {
|
||||
RecDiag( L"OnStarting: calling AudioSampleGenerator::Start(videoStartSRT=%lld)\n", startSRT );
|
||||
m_audioGenerator->Start( startSRT );
|
||||
RecDiag( L"OnStarting: audio started\n" );
|
||||
}
|
||||
RecDiag( L"OnStarting: exit\n" );
|
||||
}
|
||||
@@ -1321,12 +1398,11 @@ std::shared_ptr<VideoRecordingSession> VideoRecordingSession::Create(
|
||||
winrt::GraphicsCaptureItem const& item,
|
||||
RECT const& crop,
|
||||
uint32_t frameRate,
|
||||
bool captureAudio,
|
||||
bool captureSystemAudio,
|
||||
bool micMonoMix,
|
||||
std::unique_ptr<AudioSampleGenerator> audioGenerator,
|
||||
winrt::IAsyncAction audioInitAction,
|
||||
winrt::Streams::IRandomAccessStream const& stream)
|
||||
{
|
||||
return std::shared_ptr<VideoRecordingSession>(new VideoRecordingSession(device, item, crop, frameRate, captureAudio, captureSystemAudio, micMonoMix, stream));
|
||||
return std::shared_ptr<VideoRecordingSession>(new VideoRecordingSession(device, item, crop, frameRate, std::move(audioGenerator), audioInitAction, stream));
|
||||
}
|
||||
|
||||
//----------------------------------------------------------------------------
|
||||
@@ -1338,10 +1414,6 @@ void VideoRecordingSession::OnMediaStreamSourceSampleRequested(
|
||||
winrt::MediaStreamSource const&,
|
||||
winrt::MediaStreamSourceSampleRequestedEventArgs const& args)
|
||||
{
|
||||
static int s_diagVideoCount = 0;
|
||||
static int s_diagAudioCount = 0;
|
||||
static int64_t s_diagStartTs = 0; // SystemRelativeTime from OnStarting
|
||||
|
||||
auto request = args.Request();
|
||||
auto streamDescriptor = request.StreamDescriptor();
|
||||
if (auto videoStreamDescriptor = streamDescriptor.try_as<winrt::VideoStreamDescriptor>())
|
||||
@@ -1412,7 +1484,16 @@ void VideoRecordingSession::OnMediaStreamSourceSampleRequested(
|
||||
else
|
||||
{
|
||||
// New desktop frame — crop and copy to back buffer.
|
||||
timeStamp = frame->SystemRelativeTime;
|
||||
// If this is the cached starting frame, use the adjusted
|
||||
// SRT (computed from QPC in OnStarting) instead of the
|
||||
// stale SRT from the constructor. The stale SRT is ~2-3s
|
||||
// behind, creating a massive timestamp gap between frame
|
||||
// #1 and #2 that causes the transcoder to starve video
|
||||
// while filling the gap with audio.
|
||||
if( cachedFrame.has_value() && m_adjustedStartSRT != 0 )
|
||||
timeStamp = winrt::TimeSpan{ m_adjustedStartSRT };
|
||||
else
|
||||
timeStamp = frame->SystemRelativeTime;
|
||||
auto contentSize = frame->ContentSize;
|
||||
auto frameTexture = GetDXGIInterfaceFromObject<ID3D11Texture2D>(frame->FrameTexture);
|
||||
D3D11_TEXTURE2D_DESC desc = {};
|
||||
@@ -1448,7 +1529,7 @@ void VideoRecordingSession::OnMediaStreamSourceSampleRequested(
|
||||
m_d3dContext->CopyResource( m_cachedDesktopTex.get(), backBuffer.get() );
|
||||
}
|
||||
|
||||
// Log first 10 video frames with timing.
|
||||
// Log first 50 video frames with timing.
|
||||
if( !m_hasVideoSample.load() )
|
||||
{
|
||||
s_diagVideoCount = 0;
|
||||
@@ -1462,13 +1543,14 @@ void VideoRecordingSession::OnMediaStreamSourceSampleRequested(
|
||||
m_hasQpcOrigin = true;
|
||||
}
|
||||
s_diagVideoCount++;
|
||||
if( s_diagVideoCount <= 10 )
|
||||
if( s_diagVideoCount <= 50 )
|
||||
{
|
||||
RecDiag( L"SampleReq VIDEO #%d: sysRelTime=%lld deltaFromStart=%.1fms repeat=%d\n",
|
||||
RecDiag( L"SampleReq VIDEO #%d: sysRelTime=%lld deltaFromStart=%.1fms repeat=%d cached=%d\n",
|
||||
s_diagVideoCount,
|
||||
timeStamp.count(),
|
||||
( timeStamp.count() - s_diagStartTs ) / 10000.0,
|
||||
isRepeatFrame ? 1 : 0 );
|
||||
isRepeatFrame ? 1 : 0,
|
||||
( s_diagVideoCount == 1 && cachedFrame.has_value() ) ? 1 : 0 );
|
||||
}
|
||||
|
||||
#if _DEBUG
|
||||
@@ -1492,6 +1574,11 @@ void VideoRecordingSession::OnMediaStreamSourceSampleRequested(
|
||||
// Composite webcam overlay onto the back buffer.
|
||||
if( m_webcamCapture )
|
||||
{
|
||||
if( s_diagVideoCount <= 3 )
|
||||
{
|
||||
RecDiag( L"SampleReq VIDEO #%d: compositing LIVE webcam frame\n",
|
||||
s_diagVideoCount );
|
||||
}
|
||||
m_webcamCapture->CompositeOnto( backBuffer.get() );
|
||||
}
|
||||
|
||||
@@ -1541,7 +1628,9 @@ void VideoRecordingSession::OnMediaStreamSourceSampleRequested(
|
||||
}
|
||||
catch (winrt::hresult_error const& error)
|
||||
{
|
||||
OutputDebugStringW(error.message().c_str());
|
||||
RecDiag( L"SampleReq VIDEO EXCEPTION on frame #%d: hr=0x%08X %s\n",
|
||||
s_diagVideoCount, static_cast<unsigned>(error.code()),
|
||||
error.message().c_str() );
|
||||
request.Sample(nullptr);
|
||||
CloseInternal();
|
||||
return;
|
||||
@@ -1551,26 +1640,53 @@ void VideoRecordingSession::OnMediaStreamSourceSampleRequested(
|
||||
{
|
||||
try
|
||||
{
|
||||
static int s_audioReqCount = 0;
|
||||
if( !m_hasVideoSample.load() )
|
||||
s_audioReqCount = 0;
|
||||
s_audioReqCount++;
|
||||
|
||||
if( s_audioReqCount <= 5 )
|
||||
{
|
||||
RecDiag( L"SampleReq AUDIO req #%d: calling TryGetNextSample (started=%d)...\n",
|
||||
s_audioReqCount, m_audioGenerator ? 1 : 0 );
|
||||
}
|
||||
|
||||
LARGE_INTEGER tBefore, tAfter, tFreq;
|
||||
QueryPerformanceFrequency( &tFreq );
|
||||
QueryPerformanceCounter( &tBefore );
|
||||
|
||||
if (auto sample = m_audioGenerator ? m_audioGenerator->TryGetNextSample() : std::optional<winrt::MediaStreamSample>{}; sample.has_value())
|
||||
{
|
||||
QueryPerformanceCounter( &tAfter );
|
||||
double waitMs = static_cast<double>( tAfter.QuadPart - tBefore.QuadPart ) * 1000.0 / tFreq.QuadPart;
|
||||
|
||||
s_diagAudioCount++;
|
||||
if( s_diagAudioCount <= 10 )
|
||||
if( s_diagAudioCount <= 50 )
|
||||
{
|
||||
RecDiag( L"SampleReq AUDIO #%d: timestamp=%lld (%.1fms)\n",
|
||||
s_diagAudioCount,
|
||||
sample.value().Timestamp().count(),
|
||||
sample.value().Timestamp().count() / 10000.0 );
|
||||
auto ts = sample.value().Timestamp().count();
|
||||
auto dur = sample.value().Duration().count();
|
||||
RecDiag( L"SampleReq AUDIO #%d (req %d): timestamp=%lld (%.1fms) duration=%lld (%.1fms) waitMs=%.1f\n",
|
||||
s_diagAudioCount, s_audioReqCount,
|
||||
ts, ts / 10000.0,
|
||||
dur, dur / 10000.0,
|
||||
waitMs );
|
||||
}
|
||||
request.Sample(sample.value());
|
||||
}
|
||||
else
|
||||
{
|
||||
QueryPerformanceCounter( &tAfter );
|
||||
double waitMs = static_cast<double>( tAfter.QuadPart - tBefore.QuadPart ) * 1000.0 / tFreq.QuadPart;
|
||||
RecDiag( L"SampleReq AUDIO req #%d: TryGetNextSample returned EMPTY after %.1fms → end-of-audio-stream\n",
|
||||
s_audioReqCount, waitMs );
|
||||
request.Sample(nullptr);
|
||||
}
|
||||
}
|
||||
catch (winrt::hresult_error const& error)
|
||||
{
|
||||
OutputDebugStringW(error.message().c_str());
|
||||
RecDiag( L"SampleReq AUDIO EXCEPTION on sample #%d: hr=0x%08X %s\n",
|
||||
s_diagAudioCount, static_cast<unsigned>(error.code()),
|
||||
error.message().c_str() );
|
||||
request.Sample(nullptr);
|
||||
CloseInternal();
|
||||
return;
|
||||
@@ -1659,7 +1775,7 @@ public:
|
||||
}
|
||||
return S_OK;
|
||||
}
|
||||
|
||||
|
||||
IFACEMETHODIMP OnFolderChanging(IFileDialog*, IShellItem*) { return S_OK; }
|
||||
IFACEMETHODIMP OnSelectionChange(IFileDialog*) { return S_OK; }
|
||||
IFACEMETHODIMP OnShareViolation(IFileDialog*, IShellItem*, FDE_SHAREVIOLATION_RESPONSE*) { return S_OK; }
|
||||
@@ -2838,7 +2954,7 @@ static void HandlePlaybackCommand(int controlId, VideoRecordingSession::TrimDial
|
||||
case IDC_TRIM_REWIND:
|
||||
{
|
||||
StopPlayback(hDlg, pData, false);
|
||||
// Use 1 second step for timelines < 20 seconds, 2 seconds
|
||||
// Use 1 second step for timelines < 20 seconds, 2 seconds
|
||||
const int64_t duration = pData->trimEnd.count() - pData->trimStart.count();
|
||||
const int64_t stepTicks = (duration < 200'000'000) ? 10'000'000 : kJogStepTicks;
|
||||
const int64_t newTicks = (std::max)(pData->trimStart.count(), pData->currentPosition.count() - stepTicks);
|
||||
@@ -2854,7 +2970,7 @@ static void HandlePlaybackCommand(int controlId, VideoRecordingSession::TrimDial
|
||||
case IDC_TRIM_FORWARD:
|
||||
{
|
||||
StopPlayback(hDlg, pData, false);
|
||||
// Use 1 second step for timelines < 20 seconds, 2 seconds
|
||||
// Use 1 second step for timelines < 20 seconds, 2 seconds
|
||||
const int64_t duration = pData->trimEnd.count() - pData->trimStart.count();
|
||||
const int64_t stepTicks = (duration < 200'000'000) ? 10'000'000 : kJogStepTicks;
|
||||
const int64_t newTicks = (std::min)(pData->trimEnd.count(), pData->currentPosition.count() + stepTicks);
|
||||
@@ -5517,7 +5633,7 @@ INT_PTR CALLBACK VideoRecordingSession::TrimDialogProc(HWND hDlg, UINT message,
|
||||
const auto relativePos = winrt::TimeSpan{ (std::max)(pData->currentPosition.count() - pData->trimStart.count(), int64_t{ 0 }) };
|
||||
SetTimeText(hDlg, IDC_TRIM_POSITION_LABEL, relativePos, true);
|
||||
}
|
||||
|
||||
|
||||
if (elapsedMs >= frameDurationMs)
|
||||
{
|
||||
// Time to advance to next frame
|
||||
|
||||
@@ -27,9 +27,8 @@ public:
|
||||
winrt::GraphicsCaptureItem const& item,
|
||||
RECT const& cropRect,
|
||||
uint32_t frameRate,
|
||||
bool captureAudio,
|
||||
bool captureSystemAudio,
|
||||
bool micMonoMix,
|
||||
std::unique_ptr<AudioSampleGenerator> audioGenerator,
|
||||
winrt::Windows::Foundation::IAsyncAction audioInitAction,
|
||||
winrt::Streams::IRandomAccessStream const& stream);
|
||||
~VideoRecordingSession();
|
||||
|
||||
@@ -217,9 +216,8 @@ private:
|
||||
winrt::Capture::GraphicsCaptureItem const& item,
|
||||
RECT const cropRect,
|
||||
uint32_t frameRate,
|
||||
bool captureAudio,
|
||||
bool captureSystemAudio,
|
||||
bool micMonoMix,
|
||||
std::unique_ptr<AudioSampleGenerator> audioGenerator,
|
||||
winrt::Windows::Foundation::IAsyncAction audioInitAction,
|
||||
winrt::Streams::IRandomAccessStream const& stream);
|
||||
void CloseInternal();
|
||||
|
||||
@@ -279,5 +277,9 @@ private:
|
||||
LARGE_INTEGER m_qpcFreq{};
|
||||
LARGE_INTEGER m_qpcRecordingStart{}; // QPC at first sample
|
||||
int64_t m_startSystemRelativeTime = 0; // SystemRelativeTime of first frame
|
||||
int64_t m_adjustedStartSRT = 0; // QPC-based current SRT set in OnStarting
|
||||
bool m_hasQpcOrigin = false;
|
||||
|
||||
// Audio initialization started in the constructor, awaited in StartAsync.
|
||||
winrt::Windows::Foundation::IAsyncAction m_audioInitAction{ nullptr };
|
||||
};
|
||||
File diff suppressed because it is too large
Load Diff
@@ -18,11 +18,37 @@
|
||||
#include <mfreadwrite.h>
|
||||
#include <atomic>
|
||||
#include <condition_variable>
|
||||
#include <memory>
|
||||
#include "BackgroundBlur.h"
|
||||
#include <mutex>
|
||||
#include <thread>
|
||||
#include <vector>
|
||||
#include <winrt/base.h>
|
||||
|
||||
class BackgroundBlur;
|
||||
|
||||
// Must match CompositeConstants cbuffer layout in WebcamComposite.hlsl.
|
||||
struct GpuCompositeConstants
|
||||
{
|
||||
float CropOffsetX, CropOffsetY; // Camera crop UV offset
|
||||
float CropScaleX, CropScaleY; // Camera crop UV scale
|
||||
float Gamma; // Gamma correction exponent
|
||||
float CornerRadius; // Corner radius in output pixels
|
||||
float OutputW, OutputH; // Output dimensions
|
||||
UINT ShapeType; // 0=Square, 1=RoundedRect, 2=RoundedSquare, 3=Circle
|
||||
UINT HasMask; // 1 if mask texture valid
|
||||
float Pad[2];
|
||||
};
|
||||
|
||||
// Must match BlurConstants cbuffer layout in BoxBlurCS.hlsl.
|
||||
struct GpuBlurConstants
|
||||
{
|
||||
UINT Direction; // 0 = horizontal, 1 = vertical
|
||||
INT Radius; // Box blur radius in pixels
|
||||
UINT Width; // Image width
|
||||
UINT Height; // Image height
|
||||
};
|
||||
|
||||
class WebcamCapture
|
||||
{
|
||||
public:
|
||||
@@ -44,7 +70,10 @@ public:
|
||||
Position position,
|
||||
Size size,
|
||||
Shape shape,
|
||||
bool fullScreenRecording = false );
|
||||
bool fullScreenRecording = false,
|
||||
WebcamBackgroundMode backgroundMode = WebcamBackgroundMode::None,
|
||||
const wchar_t* backgroundImagePath = nullptr,
|
||||
int brightness = 50 );
|
||||
~WebcamCapture();
|
||||
|
||||
// Start/stop the capture thread.
|
||||
@@ -106,6 +135,19 @@ private:
|
||||
bool InitSourceReader();
|
||||
RECT ComputeDestRect() const;
|
||||
void ComputeOverlayDimensions();
|
||||
bool InitGpuComposite();
|
||||
bool GpuComposite( const UINT32* cameraPixels, UINT camW, UINT camH,
|
||||
const UINT32* blurPixels, UINT blurW, UINT blurH,
|
||||
const float* mask, UINT maskW, UINT maskH,
|
||||
UINT outW, UINT outH,
|
||||
UINT srcCropX, UINT srcCropY, UINT srcCropW, UINT srcCropH,
|
||||
float gamma, Shape shape, float cornerRadius,
|
||||
ID3D11ShaderResourceView* preBlurSRV = nullptr );
|
||||
|
||||
// GPU box blur: runs 4 compute-shader dispatches (H→V→H→V) on the
|
||||
// processing-resolution frame. The result stays GPU-resident in
|
||||
// m_blurPingPong[0] for direct use by GpuComposite.
|
||||
bool GpuBoxBlur( const UINT32* pixels, UINT width, UINT height, int radius );
|
||||
|
||||
winrt::com_ptr<ID3D11Device> m_d3dDevice;
|
||||
winrt::com_ptr<ID3D11DeviceContext> m_d3dContext;
|
||||
@@ -135,6 +177,7 @@ private:
|
||||
// Reusable frame buffer for the capture thread (avoids per-frame alloc).
|
||||
std::vector<BYTE> m_framePixels;
|
||||
std::vector<BYTE> m_scaledPixels;
|
||||
std::vector<BYTE> m_upscalePixels;
|
||||
|
||||
UINT m_overlayW = 0;
|
||||
UINT m_overlayH = 0;
|
||||
@@ -142,6 +185,11 @@ private:
|
||||
UINT m_camHeight = 0;
|
||||
RECT m_destRect = {};
|
||||
|
||||
// Brightness correction (user-controlled, fixed gamma LUT).
|
||||
int m_brightness = 50; // 0=dark, 50=neutral, 100=bright
|
||||
std::array<uint8_t, 256> m_gammaLUT = {}; // current LUT
|
||||
double m_lutGamma = 1.0; // gamma used for m_gammaLUT
|
||||
|
||||
// Output dimensions (recording output after crop+scale).
|
||||
UINT m_outputWidth = 0;
|
||||
UINT m_outputHeight = 0;
|
||||
@@ -163,8 +211,57 @@ private:
|
||||
std::condition_variable m_readyCV;
|
||||
bool m_firstFrameCaptured = false;
|
||||
|
||||
// Background processing.
|
||||
WebcamBackgroundMode m_backgroundMode = WebcamBackgroundMode::None;
|
||||
std::wstring m_backgroundImagePath;
|
||||
std::unique_ptr<BackgroundBlur> m_backgroundBlur;
|
||||
|
||||
// Debug counters for CompositeOnto logging.
|
||||
int m_compositeCount = 0;
|
||||
int m_lockFailCount = 0;
|
||||
int m_uploadCount = 0;
|
||||
|
||||
// ── GPU composite pipeline ──────────────────────────────
|
||||
// Separate D3D device for capture thread (avoids contention
|
||||
// with the recording session's device/context).
|
||||
winrt::com_ptr<ID3D11Device> m_gpuDevice;
|
||||
winrt::com_ptr<ID3D11DeviceContext> m_gpuContext;
|
||||
winrt::com_ptr<ID3D11VertexShader> m_compositeVS;
|
||||
winrt::com_ptr<ID3D11PixelShader> m_compositePS;
|
||||
winrt::com_ptr<ID3D11Buffer> m_compositeCB;
|
||||
winrt::com_ptr<ID3D11SamplerState> m_bilinearSampler;
|
||||
winrt::com_ptr<ID3D11RasterizerState> m_gpuRasterState;
|
||||
winrt::com_ptr<ID3D11BlendState> m_gpuBlendState;
|
||||
|
||||
// Input textures + SRVs (recreated when dimensions change).
|
||||
winrt::com_ptr<ID3D11Texture2D> m_gpuCameraTex;
|
||||
winrt::com_ptr<ID3D11ShaderResourceView> m_gpuCameraSRV;
|
||||
UINT m_gpuCameraW = 0, m_gpuCameraH = 0;
|
||||
|
||||
winrt::com_ptr<ID3D11Texture2D> m_gpuBlurTex;
|
||||
winrt::com_ptr<ID3D11ShaderResourceView> m_gpuBlurSRV;
|
||||
UINT m_gpuBlurW = 0, m_gpuBlurH = 0;
|
||||
|
||||
winrt::com_ptr<ID3D11Texture2D> m_gpuMaskTex;
|
||||
winrt::com_ptr<ID3D11ShaderResourceView> m_gpuMaskSRV;
|
||||
UINT m_gpuMaskW = 0, m_gpuMaskH = 0;
|
||||
|
||||
// Render target + staging for readback.
|
||||
winrt::com_ptr<ID3D11Texture2D> m_gpuRenderTarget;
|
||||
winrt::com_ptr<ID3D11RenderTargetView> m_gpuRTV;
|
||||
winrt::com_ptr<ID3D11Texture2D> m_gpuStaging;
|
||||
UINT m_gpuRTW = 0, m_gpuRTH = 0;
|
||||
|
||||
bool m_gpuCompositeReady = false;
|
||||
|
||||
// ── GPU box-blur compute pipeline ───────────────────────
|
||||
winrt::com_ptr<ID3D11ComputeShader> m_blurCS;
|
||||
winrt::com_ptr<ID3D11Buffer> m_blurCB;
|
||||
|
||||
// Ping-pong textures with SRV + UAV for blur passes.
|
||||
winrt::com_ptr<ID3D11Texture2D> m_blurPingPong[2];
|
||||
winrt::com_ptr<ID3D11ShaderResourceView> m_blurPingSRV[2];
|
||||
winrt::com_ptr<ID3D11UnorderedAccessView> m_blurPingUAV[2];
|
||||
UINT m_blurPPW = 0, m_blurPPH = 0;
|
||||
bool m_gpuBlurReady = false;
|
||||
};
|
||||
|
||||
143
src/modules/ZoomIt/ZoomIt/WebcamComposite.hlsl
Normal file
143
src/modules/ZoomIt/ZoomIt/WebcamComposite.hlsl
Normal file
@@ -0,0 +1,143 @@
|
||||
//==============================================================================
|
||||
//
|
||||
// WebcamComposite.hlsl
|
||||
//
|
||||
// GPU composite shader for webcam overlay.
|
||||
// Composites sharp foreground from full-resolution camera with
|
||||
// blurred background from processing-resolution blur buffer, using
|
||||
// a segmentation mask to blend between the two sources.
|
||||
//
|
||||
// The GPU's hardware texture sampler provides free bilinear filtering,
|
||||
// making this orders of magnitude faster than the equivalent CPU loop.
|
||||
//
|
||||
// Entry points:
|
||||
// VSMain (vs_5_0) — full-screen triangle, no vertex buffer needed
|
||||
// PSMain (ps_5_0) — composite camera + blur + mask + shape mask
|
||||
//
|
||||
// Recompile with:
|
||||
// fxc /T vs_5_0 /E VSMain /Fh WebcamCompositeVS.h /Vn g_WebcamCompositeVS WebcamComposite.hlsl
|
||||
// fxc /T ps_5_0 /E PSMain /Fh WebcamCompositePS.h /Vn g_WebcamCompositePS WebcamComposite.hlsl
|
||||
//
|
||||
// Copyright (C) Mark Russinovich
|
||||
// Sysinternals - www.sysinternals.com
|
||||
//
|
||||
//==============================================================================
|
||||
|
||||
// Camera frame at full resolution (e.g. 1920x1080), B8G8R8A8_UNORM.
|
||||
// Shader sees RGBA due to hardware swizzle.
|
||||
Texture2D CameraTex : register(t0);
|
||||
|
||||
// Blurred processing buffer at reduced resolution (e.g. 960x540), B8G8R8A8_UNORM.
|
||||
// Already gamma-corrected from CPU downsample.
|
||||
Texture2D BlurTex : register(t1);
|
||||
|
||||
// Segmentation mask from ONNX model, R32_FLOAT.
|
||||
// 1.0 = foreground (person), 0.0 = background.
|
||||
Texture2D MaskTex : register(t2);
|
||||
|
||||
// Bilinear sampler with clamp addressing — used for all textures.
|
||||
SamplerState BilinearSamp : register(s0);
|
||||
|
||||
cbuffer CompositeConstants : register(b0)
|
||||
{
|
||||
float2 CropOffset; // Camera crop UV offset (srcCropX/camW, srcCropY/camH)
|
||||
float2 CropScale; // Camera crop UV scale (srcCropW/camW, srcCropH/camH)
|
||||
float Gamma; // Gamma correction exponent (< 1 brightens)
|
||||
float CornerRadius; // Corner radius in output pixels
|
||||
float OutputW; // Output width in pixels
|
||||
float OutputH; // Output height in pixels
|
||||
uint ShapeType; // 0=Square, 1=RoundedRect, 2=RoundedSquare, 3=Circle
|
||||
uint HasMask; // 1 if segmentation mask is valid
|
||||
float2 Pad;
|
||||
};
|
||||
|
||||
struct VSOutput
|
||||
{
|
||||
float4 Position : SV_POSITION;
|
||||
float2 TexCoord : TEXCOORD0;
|
||||
};
|
||||
|
||||
//----------------------------------------------------------------------------
|
||||
// Vertex shader: full-screen triangle from SV_VertexID (no vertex buffer).
|
||||
// Draw(3, 0) to invoke. The triangle covers [-1,1] clip space.
|
||||
//----------------------------------------------------------------------------
|
||||
VSOutput VSMain( uint vertexId : SV_VertexID )
|
||||
{
|
||||
VSOutput output;
|
||||
output.TexCoord = float2( (vertexId << 1) & 2, vertexId & 2 );
|
||||
output.Position = float4( output.TexCoord * float2(2.0, -2.0) + float2(-1.0, 1.0), 0.0, 1.0 );
|
||||
return output;
|
||||
}
|
||||
|
||||
//----------------------------------------------------------------------------
|
||||
// Pixel shader: composite camera foreground with blurred background.
|
||||
// - Shape masking (circle, rounded rect) produces alpha = 0 outside.
|
||||
// - Segmentation mask blends camera (foreground) with blur (background).
|
||||
// - Gamma correction applied to camera samples only (blur already corrected).
|
||||
// - Hardware bilinear filtering on all texture samples (free).
|
||||
//----------------------------------------------------------------------------
|
||||
float4 PSMain( VSOutput input ) : SV_TARGET
|
||||
{
|
||||
float2 uv = input.TexCoord;
|
||||
float px = uv.x * OutputW;
|
||||
float py = uv.y * OutputH;
|
||||
|
||||
// ── Shape mask ─────────────────────────────────────────────────────
|
||||
if( ShapeType == 3 ) // Circle
|
||||
{
|
||||
float halfW = OutputW * 0.5;
|
||||
float halfH = OutputH * 0.5;
|
||||
float radius = min( halfW, halfH );
|
||||
float dx = ( px - halfW ) / radius;
|
||||
float dy = ( py - halfH ) / radius;
|
||||
if( dx * dx + dy * dy > 1.0 )
|
||||
return float4( 0, 0, 0, 0 );
|
||||
}
|
||||
else if( ShapeType >= 1 ) // RoundedRect or RoundedSquare
|
||||
{
|
||||
float cx = 0, cy = 0;
|
||||
bool inCorner = false;
|
||||
|
||||
if( px < CornerRadius && py < CornerRadius )
|
||||
{ cx = CornerRadius; cy = CornerRadius; inCorner = true; }
|
||||
else if( px > OutputW - CornerRadius && py < CornerRadius )
|
||||
{ cx = OutputW - CornerRadius; cy = CornerRadius; inCorner = true; }
|
||||
else if( px < CornerRadius && py > OutputH - CornerRadius )
|
||||
{ cx = CornerRadius; cy = OutputH - CornerRadius; inCorner = true; }
|
||||
else if( px > OutputW - CornerRadius && py > OutputH - CornerRadius )
|
||||
{ cx = OutputW - CornerRadius; cy = OutputH - CornerRadius; inCorner = true; }
|
||||
|
||||
if( inCorner )
|
||||
{
|
||||
float ddx = px - cx;
|
||||
float ddy = py - cy;
|
||||
if( ddx * ddx + ddy * ddy > CornerRadius * CornerRadius )
|
||||
return float4( 0, 0, 0, 0 );
|
||||
}
|
||||
}
|
||||
|
||||
// ── Composite ──────────────────────────────────────────────────────
|
||||
if( HasMask )
|
||||
{
|
||||
// Segmentation mask (bilinear-filtered for smooth edges).
|
||||
float mask = saturate( MaskTex.Sample( BilinearSamp, uv ).r );
|
||||
|
||||
// Camera: crop-to-fill UV mapping + gamma correction.
|
||||
float2 camUV = CropOffset + uv * CropScale;
|
||||
float4 cam = CameraTex.Sample( BilinearSamp, camUV );
|
||||
cam.rgb = pow( max( cam.rgb, 0.001 ), Gamma );
|
||||
|
||||
// Blur: already gamma-corrected from CPU downsample.
|
||||
float4 blur = BlurTex.Sample( BilinearSamp, uv );
|
||||
|
||||
// Blend: mask=1 → camera (foreground), mask=0 → blur (background).
|
||||
float3 result = lerp( blur.rgb, cam.rgb, mask );
|
||||
return float4( result, 1.0 );
|
||||
}
|
||||
else
|
||||
{
|
||||
// No segmentation mask — just display the processing buffer.
|
||||
float4 blur = BlurTex.Sample( BilinearSamp, uv );
|
||||
return float4( blur.rgb, 1.0 );
|
||||
}
|
||||
}
|
||||
616
src/modules/ZoomIt/ZoomIt/WebcamCompositePS.h
Normal file
616
src/modules/ZoomIt/ZoomIt/WebcamCompositePS.h
Normal file
@@ -0,0 +1,616 @@
|
||||
#if 0
|
||||
//
|
||||
// Generated by Microsoft (R) HLSL Shader Compiler 10.1
|
||||
//
|
||||
//
|
||||
// Buffer Definitions:
|
||||
//
|
||||
// cbuffer CompositeConstants
|
||||
// {
|
||||
//
|
||||
// float2 CropOffset; // Offset: 0 Size: 8
|
||||
// float2 CropScale; // Offset: 8 Size: 8
|
||||
// float Gamma; // Offset: 16 Size: 4
|
||||
// float CornerRadius; // Offset: 20 Size: 4
|
||||
// float OutputW; // Offset: 24 Size: 4
|
||||
// float OutputH; // Offset: 28 Size: 4
|
||||
// uint ShapeType; // Offset: 32 Size: 4
|
||||
// uint HasMask; // Offset: 36 Size: 4
|
||||
// float2 Pad; // Offset: 40 Size: 8 [unused]
|
||||
//
|
||||
// }
|
||||
//
|
||||
//
|
||||
// Resource Bindings:
|
||||
//
|
||||
// Name Type Format Dim HLSL Bind Count
|
||||
// ------------------------------ ---------- ------- ----------- -------------- ------
|
||||
// BilinearSamp sampler NA NA s0 1
|
||||
// CameraTex texture float4 2d t0 1
|
||||
// BlurTex texture float4 2d t1 1
|
||||
// MaskTex texture float4 2d t2 1
|
||||
// CompositeConstants cbuffer NA NA cb0 1
|
||||
//
|
||||
//
|
||||
//
|
||||
// Input signature:
|
||||
//
|
||||
// Name Index Mask Register SysValue Format Used
|
||||
// -------------------- ----- ------ -------- -------- ------- ------
|
||||
// SV_POSITION 0 xyzw 0 POS float
|
||||
// TEXCOORD 0 xy 1 NONE float xy
|
||||
//
|
||||
//
|
||||
// Output signature:
|
||||
//
|
||||
// Name Index Mask Register SysValue Format Used
|
||||
// -------------------- ----- ------ -------- -------- ------- ------
|
||||
// SV_TARGET 0 xyzw 0 TARGET float xyzw
|
||||
//
|
||||
ps_5_0
|
||||
dcl_globalFlags refactoringAllowed
|
||||
dcl_constantbuffer CB0[3], immediateIndexed
|
||||
dcl_sampler s0, mode_default
|
||||
dcl_resource_texture2d (float,float,float,float) t0
|
||||
dcl_resource_texture2d (float,float,float,float) t1
|
||||
dcl_resource_texture2d (float,float,float,float) t2
|
||||
dcl_input_ps linear v1.xy
|
||||
dcl_output o0.xyzw
|
||||
dcl_temps 4
|
||||
mul r0.xy, v1.xyxx, cb0[1].zwzz
|
||||
ieq r0.z, cb0[2].x, l(3)
|
||||
if_nz r0.z
|
||||
mul r0.zw, cb0[1].zzzw, l(0.000000, 0.000000, 0.500000, 0.500000)
|
||||
min r1.x, r0.w, r0.z
|
||||
mad r0.zw, v1.xxxy, cb0[1].zzzw, -r0.zzzw
|
||||
div r0.zw, r0.zzzw, r1.xxxx
|
||||
mul r0.zw, r0.zzzw, r0.zzzw
|
||||
add r0.z, r0.w, r0.z
|
||||
lt r0.z, l(1.000000), r0.z
|
||||
if_nz r0.z
|
||||
mov o0.xyzw, l(0,0,0,0)
|
||||
ret
|
||||
endif
|
||||
else
|
||||
uge r0.z, cb0[2].x, l(1)
|
||||
if_nz r0.z
|
||||
lt r0.zw, r0.yyyx, cb0[1].yyyy
|
||||
and r1.x, r0.z, r0.w
|
||||
add r2.xy, -cb0[1].yyyy, cb0[1].zwzz
|
||||
lt r0.xy, r2.xyxx, r0.xyxx
|
||||
and r0.zw, r0.zzzw, r0.xxxy
|
||||
and r3.z, r0.y, r0.x
|
||||
and r3.xy, r2.xyxx, r3.zzzz
|
||||
mov r2.z, cb0[1].y
|
||||
mov r2.w, l(-1)
|
||||
movc r0.xyw, r0.wwww, r2.zyzw, r3.xyxz
|
||||
movc r0.xyz, r0.zzzz, r2.xzwx, r0.xywx
|
||||
movc r0.xyz, r1.xxxx, r2.zzwz, r0.xyzx
|
||||
if_nz r0.z
|
||||
mad r0.xy, v1.xyxx, cb0[1].zwzz, -r0.xyxx
|
||||
mul r0.xy, r0.xyxx, r0.xyxx
|
||||
add r0.x, r0.y, r0.x
|
||||
mul r0.y, cb0[1].y, cb0[1].y
|
||||
lt r0.x, r0.y, r0.x
|
||||
if_nz r0.x
|
||||
mov o0.xyzw, l(0,0,0,0)
|
||||
ret
|
||||
endif
|
||||
endif
|
||||
endif
|
||||
endif
|
||||
if_nz cb0[2].y
|
||||
sample_indexable(texture2d)(float,float,float,float) r0.x, v1.xyxx, t2.xyzw, s0
|
||||
mov_sat r0.x, r0.x
|
||||
mad r0.yz, v1.xxyx, cb0[0].zzwz, cb0[0].xxyx
|
||||
sample_indexable(texture2d)(float,float,float,float) r0.yzw, r0.yzyy, t0.wxyz, s0
|
||||
max r0.yzw, r0.yyzw, l(0.000000, 0.001000, 0.001000, 0.001000)
|
||||
log r0.yzw, r0.yyzw
|
||||
mul r0.yzw, r0.yyzw, cb0[1].xxxx
|
||||
exp r0.yzw, r0.yyzw
|
||||
sample_indexable(texture2d)(float,float,float,float) r1.xyz, v1.xyxx, t1.xyzw, s0
|
||||
add r0.yzw, r0.yyzw, -r1.xxyz
|
||||
mad o0.xyz, r0.xxxx, r0.yzwy, r1.xyzx
|
||||
mov o0.w, l(1.000000)
|
||||
ret
|
||||
else
|
||||
sample_indexable(texture2d)(float,float,float,float) r0.xyz, v1.xyxx, t1.xyzw, s0
|
||||
mov o0.xyz, r0.xyzx
|
||||
mov o0.w, l(1.000000)
|
||||
ret
|
||||
endif
|
||||
ret
|
||||
// Approximately 63 instruction slots used
|
||||
#endif
|
||||
|
||||
const BYTE g_WebcamCompositePS[] =
|
||||
{
|
||||
68, 88, 66, 67, 78, 221,
|
||||
134, 205, 221, 100, 55, 97,
|
||||
108, 36, 219, 137, 244, 189,
|
||||
76, 224, 1, 0, 0, 0,
|
||||
108, 11, 0, 0, 5, 0,
|
||||
0, 0, 52, 0, 0, 0,
|
||||
208, 3, 0, 0, 40, 4,
|
||||
0, 0, 92, 4, 0, 0,
|
||||
208, 10, 0, 0, 82, 68,
|
||||
69, 70, 148, 3, 0, 0,
|
||||
1, 0, 0, 0, 24, 1,
|
||||
0, 0, 5, 0, 0, 0,
|
||||
60, 0, 0, 0, 0, 5,
|
||||
255, 255, 0, 1, 0, 0,
|
||||
108, 3, 0, 0, 82, 68,
|
||||
49, 49, 60, 0, 0, 0,
|
||||
24, 0, 0, 0, 32, 0,
|
||||
0, 0, 40, 0, 0, 0,
|
||||
36, 0, 0, 0, 12, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
220, 0, 0, 0, 3, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
1, 0, 0, 0, 1, 0,
|
||||
0, 0, 233, 0, 0, 0,
|
||||
2, 0, 0, 0, 5, 0,
|
||||
0, 0, 4, 0, 0, 0,
|
||||
255, 255, 255, 255, 0, 0,
|
||||
0, 0, 1, 0, 0, 0,
|
||||
13, 0, 0, 0, 243, 0,
|
||||
0, 0, 2, 0, 0, 0,
|
||||
5, 0, 0, 0, 4, 0,
|
||||
0, 0, 255, 255, 255, 255,
|
||||
1, 0, 0, 0, 1, 0,
|
||||
0, 0, 13, 0, 0, 0,
|
||||
251, 0, 0, 0, 2, 0,
|
||||
0, 0, 5, 0, 0, 0,
|
||||
4, 0, 0, 0, 255, 255,
|
||||
255, 255, 2, 0, 0, 0,
|
||||
1, 0, 0, 0, 13, 0,
|
||||
0, 0, 3, 1, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 1, 0, 0, 0,
|
||||
1, 0, 0, 0, 66, 105,
|
||||
108, 105, 110, 101, 97, 114,
|
||||
83, 97, 109, 112, 0, 67,
|
||||
97, 109, 101, 114, 97, 84,
|
||||
101, 120, 0, 66, 108, 117,
|
||||
114, 84, 101, 120, 0, 77,
|
||||
97, 115, 107, 84, 101, 120,
|
||||
0, 67, 111, 109, 112, 111,
|
||||
115, 105, 116, 101, 67, 111,
|
||||
110, 115, 116, 97, 110, 116,
|
||||
115, 0, 171, 171, 3, 1,
|
||||
0, 0, 9, 0, 0, 0,
|
||||
48, 1, 0, 0, 48, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 152, 2,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
8, 0, 0, 0, 2, 0,
|
||||
0, 0, 172, 2, 0, 0,
|
||||
0, 0, 0, 0, 255, 255,
|
||||
255, 255, 0, 0, 0, 0,
|
||||
255, 255, 255, 255, 0, 0,
|
||||
0, 0, 208, 2, 0, 0,
|
||||
8, 0, 0, 0, 8, 0,
|
||||
0, 0, 2, 0, 0, 0,
|
||||
172, 2, 0, 0, 0, 0,
|
||||
0, 0, 255, 255, 255, 255,
|
||||
0, 0, 0, 0, 255, 255,
|
||||
255, 255, 0, 0, 0, 0,
|
||||
218, 2, 0, 0, 16, 0,
|
||||
0, 0, 4, 0, 0, 0,
|
||||
2, 0, 0, 0, 232, 2,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
255, 255, 255, 255, 0, 0,
|
||||
0, 0, 255, 255, 255, 255,
|
||||
0, 0, 0, 0, 12, 3,
|
||||
0, 0, 20, 0, 0, 0,
|
||||
4, 0, 0, 0, 2, 0,
|
||||
0, 0, 232, 2, 0, 0,
|
||||
0, 0, 0, 0, 255, 255,
|
||||
255, 255, 0, 0, 0, 0,
|
||||
255, 255, 255, 255, 0, 0,
|
||||
0, 0, 25, 3, 0, 0,
|
||||
24, 0, 0, 0, 4, 0,
|
||||
0, 0, 2, 0, 0, 0,
|
||||
232, 2, 0, 0, 0, 0,
|
||||
0, 0, 255, 255, 255, 255,
|
||||
0, 0, 0, 0, 255, 255,
|
||||
255, 255, 0, 0, 0, 0,
|
||||
33, 3, 0, 0, 28, 0,
|
||||
0, 0, 4, 0, 0, 0,
|
||||
2, 0, 0, 0, 232, 2,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
255, 255, 255, 255, 0, 0,
|
||||
0, 0, 255, 255, 255, 255,
|
||||
0, 0, 0, 0, 41, 3,
|
||||
0, 0, 32, 0, 0, 0,
|
||||
4, 0, 0, 0, 2, 0,
|
||||
0, 0, 60, 3, 0, 0,
|
||||
0, 0, 0, 0, 255, 255,
|
||||
255, 255, 0, 0, 0, 0,
|
||||
255, 255, 255, 255, 0, 0,
|
||||
0, 0, 96, 3, 0, 0,
|
||||
36, 0, 0, 0, 4, 0,
|
||||
0, 0, 2, 0, 0, 0,
|
||||
60, 3, 0, 0, 0, 0,
|
||||
0, 0, 255, 255, 255, 255,
|
||||
0, 0, 0, 0, 255, 255,
|
||||
255, 255, 0, 0, 0, 0,
|
||||
104, 3, 0, 0, 40, 0,
|
||||
0, 0, 8, 0, 0, 0,
|
||||
0, 0, 0, 0, 172, 2,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
255, 255, 255, 255, 0, 0,
|
||||
0, 0, 255, 255, 255, 255,
|
||||
0, 0, 0, 0, 67, 114,
|
||||
111, 112, 79, 102, 102, 115,
|
||||
101, 116, 0, 102, 108, 111,
|
||||
97, 116, 50, 0, 171, 171,
|
||||
1, 0, 3, 0, 1, 0,
|
||||
2, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 163, 2, 0, 0,
|
||||
67, 114, 111, 112, 83, 99,
|
||||
97, 108, 101, 0, 71, 97,
|
||||
109, 109, 97, 0, 102, 108,
|
||||
111, 97, 116, 0, 171, 171,
|
||||
0, 0, 3, 0, 1, 0,
|
||||
1, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 224, 2, 0, 0,
|
||||
67, 111, 114, 110, 101, 114,
|
||||
82, 97, 100, 105, 117, 115,
|
||||
0, 79, 117, 116, 112, 117,
|
||||
116, 87, 0, 79, 117, 116,
|
||||
112, 117, 116, 72, 0, 83,
|
||||
104, 97, 112, 101, 84, 121,
|
||||
112, 101, 0, 100, 119, 111,
|
||||
114, 100, 0, 171, 171, 171,
|
||||
0, 0, 19, 0, 1, 0,
|
||||
1, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 51, 3, 0, 0,
|
||||
72, 97, 115, 77, 97, 115,
|
||||
107, 0, 80, 97, 100, 0,
|
||||
77, 105, 99, 114, 111, 115,
|
||||
111, 102, 116, 32, 40, 82,
|
||||
41, 32, 72, 76, 83, 76,
|
||||
32, 83, 104, 97, 100, 101,
|
||||
114, 32, 67, 111, 109, 112,
|
||||
105, 108, 101, 114, 32, 49,
|
||||
48, 46, 49, 0, 73, 83,
|
||||
71, 78, 80, 0, 0, 0,
|
||||
2, 0, 0, 0, 8, 0,
|
||||
0, 0, 56, 0, 0, 0,
|
||||
0, 0, 0, 0, 1, 0,
|
||||
0, 0, 3, 0, 0, 0,
|
||||
0, 0, 0, 0, 15, 0,
|
||||
0, 0, 68, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 3, 0, 0, 0,
|
||||
1, 0, 0, 0, 3, 3,
|
||||
0, 0, 83, 86, 95, 80,
|
||||
79, 83, 73, 84, 73, 79,
|
||||
78, 0, 84, 69, 88, 67,
|
||||
79, 79, 82, 68, 0, 171,
|
||||
171, 171, 79, 83, 71, 78,
|
||||
44, 0, 0, 0, 1, 0,
|
||||
0, 0, 8, 0, 0, 0,
|
||||
32, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
3, 0, 0, 0, 0, 0,
|
||||
0, 0, 15, 0, 0, 0,
|
||||
83, 86, 95, 84, 65, 82,
|
||||
71, 69, 84, 0, 171, 171,
|
||||
83, 72, 69, 88, 108, 6,
|
||||
0, 0, 80, 0, 0, 0,
|
||||
155, 1, 0, 0, 106, 8,
|
||||
0, 1, 89, 0, 0, 4,
|
||||
70, 142, 32, 0, 0, 0,
|
||||
0, 0, 3, 0, 0, 0,
|
||||
90, 0, 0, 3, 0, 96,
|
||||
16, 0, 0, 0, 0, 0,
|
||||
88, 24, 0, 4, 0, 112,
|
||||
16, 0, 0, 0, 0, 0,
|
||||
85, 85, 0, 0, 88, 24,
|
||||
0, 4, 0, 112, 16, 0,
|
||||
1, 0, 0, 0, 85, 85,
|
||||
0, 0, 88, 24, 0, 4,
|
||||
0, 112, 16, 0, 2, 0,
|
||||
0, 0, 85, 85, 0, 0,
|
||||
98, 16, 0, 3, 50, 16,
|
||||
16, 0, 1, 0, 0, 0,
|
||||
101, 0, 0, 3, 242, 32,
|
||||
16, 0, 0, 0, 0, 0,
|
||||
104, 0, 0, 2, 4, 0,
|
||||
0, 0, 56, 0, 0, 8,
|
||||
50, 0, 16, 0, 0, 0,
|
||||
0, 0, 70, 16, 16, 0,
|
||||
1, 0, 0, 0, 230, 138,
|
||||
32, 0, 0, 0, 0, 0,
|
||||
1, 0, 0, 0, 32, 0,
|
||||
0, 8, 66, 0, 16, 0,
|
||||
0, 0, 0, 0, 10, 128,
|
||||
32, 0, 0, 0, 0, 0,
|
||||
2, 0, 0, 0, 1, 64,
|
||||
0, 0, 3, 0, 0, 0,
|
||||
31, 0, 4, 3, 42, 0,
|
||||
16, 0, 0, 0, 0, 0,
|
||||
56, 0, 0, 11, 194, 0,
|
||||
16, 0, 0, 0, 0, 0,
|
||||
166, 142, 32, 0, 0, 0,
|
||||
0, 0, 1, 0, 0, 0,
|
||||
2, 64, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 63, 0, 0,
|
||||
0, 63, 51, 0, 0, 7,
|
||||
18, 0, 16, 0, 1, 0,
|
||||
0, 0, 58, 0, 16, 0,
|
||||
0, 0, 0, 0, 42, 0,
|
||||
16, 0, 0, 0, 0, 0,
|
||||
50, 0, 0, 11, 194, 0,
|
||||
16, 0, 0, 0, 0, 0,
|
||||
6, 20, 16, 0, 1, 0,
|
||||
0, 0, 166, 142, 32, 0,
|
||||
0, 0, 0, 0, 1, 0,
|
||||
0, 0, 166, 14, 16, 128,
|
||||
65, 0, 0, 0, 0, 0,
|
||||
0, 0, 14, 0, 0, 7,
|
||||
194, 0, 16, 0, 0, 0,
|
||||
0, 0, 166, 14, 16, 0,
|
||||
0, 0, 0, 0, 6, 0,
|
||||
16, 0, 1, 0, 0, 0,
|
||||
56, 0, 0, 7, 194, 0,
|
||||
16, 0, 0, 0, 0, 0,
|
||||
166, 14, 16, 0, 0, 0,
|
||||
0, 0, 166, 14, 16, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 7, 66, 0, 16, 0,
|
||||
0, 0, 0, 0, 58, 0,
|
||||
16, 0, 0, 0, 0, 0,
|
||||
42, 0, 16, 0, 0, 0,
|
||||
0, 0, 49, 0, 0, 7,
|
||||
66, 0, 16, 0, 0, 0,
|
||||
0, 0, 1, 64, 0, 0,
|
||||
0, 0, 128, 63, 42, 0,
|
||||
16, 0, 0, 0, 0, 0,
|
||||
31, 0, 4, 3, 42, 0,
|
||||
16, 0, 0, 0, 0, 0,
|
||||
54, 0, 0, 8, 242, 32,
|
||||
16, 0, 0, 0, 0, 0,
|
||||
2, 64, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 62, 0, 0, 1,
|
||||
21, 0, 0, 1, 18, 0,
|
||||
0, 1, 80, 0, 0, 8,
|
||||
66, 0, 16, 0, 0, 0,
|
||||
0, 0, 10, 128, 32, 0,
|
||||
0, 0, 0, 0, 2, 0,
|
||||
0, 0, 1, 64, 0, 0,
|
||||
1, 0, 0, 0, 31, 0,
|
||||
4, 3, 42, 0, 16, 0,
|
||||
0, 0, 0, 0, 49, 0,
|
||||
0, 8, 194, 0, 16, 0,
|
||||
0, 0, 0, 0, 86, 1,
|
||||
16, 0, 0, 0, 0, 0,
|
||||
86, 133, 32, 0, 0, 0,
|
||||
0, 0, 1, 0, 0, 0,
|
||||
1, 0, 0, 7, 18, 0,
|
||||
16, 0, 1, 0, 0, 0,
|
||||
42, 0, 16, 0, 0, 0,
|
||||
0, 0, 58, 0, 16, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 10, 50, 0, 16, 0,
|
||||
2, 0, 0, 0, 86, 133,
|
||||
32, 128, 65, 0, 0, 0,
|
||||
0, 0, 0, 0, 1, 0,
|
||||
0, 0, 230, 138, 32, 0,
|
||||
0, 0, 0, 0, 1, 0,
|
||||
0, 0, 49, 0, 0, 7,
|
||||
50, 0, 16, 0, 0, 0,
|
||||
0, 0, 70, 0, 16, 0,
|
||||
2, 0, 0, 0, 70, 0,
|
||||
16, 0, 0, 0, 0, 0,
|
||||
1, 0, 0, 7, 194, 0,
|
||||
16, 0, 0, 0, 0, 0,
|
||||
166, 14, 16, 0, 0, 0,
|
||||
0, 0, 6, 4, 16, 0,
|
||||
0, 0, 0, 0, 1, 0,
|
||||
0, 7, 66, 0, 16, 0,
|
||||
3, 0, 0, 0, 26, 0,
|
||||
16, 0, 0, 0, 0, 0,
|
||||
10, 0, 16, 0, 0, 0,
|
||||
0, 0, 1, 0, 0, 7,
|
||||
50, 0, 16, 0, 3, 0,
|
||||
0, 0, 70, 0, 16, 0,
|
||||
2, 0, 0, 0, 166, 10,
|
||||
16, 0, 3, 0, 0, 0,
|
||||
54, 0, 0, 6, 66, 0,
|
||||
16, 0, 2, 0, 0, 0,
|
||||
26, 128, 32, 0, 0, 0,
|
||||
0, 0, 1, 0, 0, 0,
|
||||
54, 0, 0, 5, 130, 0,
|
||||
16, 0, 2, 0, 0, 0,
|
||||
1, 64, 0, 0, 255, 255,
|
||||
255, 255, 55, 0, 0, 9,
|
||||
178, 0, 16, 0, 0, 0,
|
||||
0, 0, 246, 15, 16, 0,
|
||||
0, 0, 0, 0, 102, 14,
|
||||
16, 0, 2, 0, 0, 0,
|
||||
70, 8, 16, 0, 3, 0,
|
||||
0, 0, 55, 0, 0, 9,
|
||||
114, 0, 16, 0, 0, 0,
|
||||
0, 0, 166, 10, 16, 0,
|
||||
0, 0, 0, 0, 134, 3,
|
||||
16, 0, 2, 0, 0, 0,
|
||||
70, 3, 16, 0, 0, 0,
|
||||
0, 0, 55, 0, 0, 9,
|
||||
114, 0, 16, 0, 0, 0,
|
||||
0, 0, 6, 0, 16, 0,
|
||||
1, 0, 0, 0, 166, 11,
|
||||
16, 0, 2, 0, 0, 0,
|
||||
70, 2, 16, 0, 0, 0,
|
||||
0, 0, 31, 0, 4, 3,
|
||||
42, 0, 16, 0, 0, 0,
|
||||
0, 0, 50, 0, 0, 11,
|
||||
50, 0, 16, 0, 0, 0,
|
||||
0, 0, 70, 16, 16, 0,
|
||||
1, 0, 0, 0, 230, 138,
|
||||
32, 0, 0, 0, 0, 0,
|
||||
1, 0, 0, 0, 70, 0,
|
||||
16, 128, 65, 0, 0, 0,
|
||||
0, 0, 0, 0, 56, 0,
|
||||
0, 7, 50, 0, 16, 0,
|
||||
0, 0, 0, 0, 70, 0,
|
||||
16, 0, 0, 0, 0, 0,
|
||||
70, 0, 16, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 7,
|
||||
18, 0, 16, 0, 0, 0,
|
||||
0, 0, 26, 0, 16, 0,
|
||||
0, 0, 0, 0, 10, 0,
|
||||
16, 0, 0, 0, 0, 0,
|
||||
56, 0, 0, 9, 34, 0,
|
||||
16, 0, 0, 0, 0, 0,
|
||||
26, 128, 32, 0, 0, 0,
|
||||
0, 0, 1, 0, 0, 0,
|
||||
26, 128, 32, 0, 0, 0,
|
||||
0, 0, 1, 0, 0, 0,
|
||||
49, 0, 0, 7, 18, 0,
|
||||
16, 0, 0, 0, 0, 0,
|
||||
26, 0, 16, 0, 0, 0,
|
||||
0, 0, 10, 0, 16, 0,
|
||||
0, 0, 0, 0, 31, 0,
|
||||
4, 3, 10, 0, 16, 0,
|
||||
0, 0, 0, 0, 54, 0,
|
||||
0, 8, 242, 32, 16, 0,
|
||||
0, 0, 0, 0, 2, 64,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
62, 0, 0, 1, 21, 0,
|
||||
0, 1, 21, 0, 0, 1,
|
||||
21, 0, 0, 1, 21, 0,
|
||||
0, 1, 31, 0, 4, 4,
|
||||
26, 128, 32, 0, 0, 0,
|
||||
0, 0, 2, 0, 0, 0,
|
||||
69, 0, 0, 139, 194, 0,
|
||||
0, 128, 67, 85, 21, 0,
|
||||
18, 0, 16, 0, 0, 0,
|
||||
0, 0, 70, 16, 16, 0,
|
||||
1, 0, 0, 0, 70, 126,
|
||||
16, 0, 2, 0, 0, 0,
|
||||
0, 96, 16, 0, 0, 0,
|
||||
0, 0, 54, 32, 0, 5,
|
||||
18, 0, 16, 0, 0, 0,
|
||||
0, 0, 10, 0, 16, 0,
|
||||
0, 0, 0, 0, 50, 0,
|
||||
0, 11, 98, 0, 16, 0,
|
||||
0, 0, 0, 0, 6, 17,
|
||||
16, 0, 1, 0, 0, 0,
|
||||
166, 139, 32, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
6, 129, 32, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
69, 0, 0, 139, 194, 0,
|
||||
0, 128, 67, 85, 21, 0,
|
||||
226, 0, 16, 0, 0, 0,
|
||||
0, 0, 150, 5, 16, 0,
|
||||
0, 0, 0, 0, 54, 121,
|
||||
16, 0, 0, 0, 0, 0,
|
||||
0, 96, 16, 0, 0, 0,
|
||||
0, 0, 52, 0, 0, 10,
|
||||
226, 0, 16, 0, 0, 0,
|
||||
0, 0, 86, 14, 16, 0,
|
||||
0, 0, 0, 0, 2, 64,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
111, 18, 131, 58, 111, 18,
|
||||
131, 58, 111, 18, 131, 58,
|
||||
47, 0, 0, 5, 226, 0,
|
||||
16, 0, 0, 0, 0, 0,
|
||||
86, 14, 16, 0, 0, 0,
|
||||
0, 0, 56, 0, 0, 8,
|
||||
226, 0, 16, 0, 0, 0,
|
||||
0, 0, 86, 14, 16, 0,
|
||||
0, 0, 0, 0, 6, 128,
|
||||
32, 0, 0, 0, 0, 0,
|
||||
1, 0, 0, 0, 25, 0,
|
||||
0, 5, 226, 0, 16, 0,
|
||||
0, 0, 0, 0, 86, 14,
|
||||
16, 0, 0, 0, 0, 0,
|
||||
69, 0, 0, 139, 194, 0,
|
||||
0, 128, 67, 85, 21, 0,
|
||||
114, 0, 16, 0, 1, 0,
|
||||
0, 0, 70, 16, 16, 0,
|
||||
1, 0, 0, 0, 70, 126,
|
||||
16, 0, 1, 0, 0, 0,
|
||||
0, 96, 16, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 8,
|
||||
226, 0, 16, 0, 0, 0,
|
||||
0, 0, 86, 14, 16, 0,
|
||||
0, 0, 0, 0, 6, 9,
|
||||
16, 128, 65, 0, 0, 0,
|
||||
1, 0, 0, 0, 50, 0,
|
||||
0, 9, 114, 32, 16, 0,
|
||||
0, 0, 0, 0, 6, 0,
|
||||
16, 0, 0, 0, 0, 0,
|
||||
150, 7, 16, 0, 0, 0,
|
||||
0, 0, 70, 2, 16, 0,
|
||||
1, 0, 0, 0, 54, 0,
|
||||
0, 5, 130, 32, 16, 0,
|
||||
0, 0, 0, 0, 1, 64,
|
||||
0, 0, 0, 0, 128, 63,
|
||||
62, 0, 0, 1, 18, 0,
|
||||
0, 1, 69, 0, 0, 139,
|
||||
194, 0, 0, 128, 67, 85,
|
||||
21, 0, 114, 0, 16, 0,
|
||||
0, 0, 0, 0, 70, 16,
|
||||
16, 0, 1, 0, 0, 0,
|
||||
70, 126, 16, 0, 1, 0,
|
||||
0, 0, 0, 96, 16, 0,
|
||||
0, 0, 0, 0, 54, 0,
|
||||
0, 5, 114, 32, 16, 0,
|
||||
0, 0, 0, 0, 70, 2,
|
||||
16, 0, 0, 0, 0, 0,
|
||||
54, 0, 0, 5, 130, 32,
|
||||
16, 0, 0, 0, 0, 0,
|
||||
1, 64, 0, 0, 0, 0,
|
||||
128, 63, 62, 0, 0, 1,
|
||||
21, 0, 0, 1, 62, 0,
|
||||
0, 1, 83, 84, 65, 84,
|
||||
148, 0, 0, 0, 63, 0,
|
||||
0, 0, 4, 0, 0, 0,
|
||||
0, 0, 0, 0, 2, 0,
|
||||
0, 0, 23, 0, 0, 0,
|
||||
1, 0, 0, 0, 5, 0,
|
||||
0, 0, 7, 0, 0, 0,
|
||||
5, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
4, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 8, 0, 0, 0,
|
||||
3, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0
|
||||
};
|
||||
162
src/modules/ZoomIt/ZoomIt/WebcamCompositeVS.h
Normal file
162
src/modules/ZoomIt/ZoomIt/WebcamCompositeVS.h
Normal file
@@ -0,0 +1,162 @@
|
||||
#if 0
|
||||
//
|
||||
// Generated by Microsoft (R) HLSL Shader Compiler 10.1
|
||||
//
|
||||
//
|
||||
//
|
||||
// Input signature:
|
||||
//
|
||||
// Name Index Mask Register SysValue Format Used
|
||||
// -------------------- ----- ------ -------- -------- ------- ------
|
||||
// SV_VertexID 0 x 0 VERTID uint x
|
||||
//
|
||||
//
|
||||
// Output signature:
|
||||
//
|
||||
// Name Index Mask Register SysValue Format Used
|
||||
// -------------------- ----- ------ -------- -------- ------- ------
|
||||
// SV_POSITION 0 xyzw 0 POS float xyzw
|
||||
// TEXCOORD 0 xy 1 NONE float xy
|
||||
//
|
||||
vs_5_0
|
||||
dcl_globalFlags refactoringAllowed
|
||||
dcl_input_sgv v0.x, vertex_id
|
||||
dcl_output_siv o0.xyzw, position
|
||||
dcl_output o1.xy
|
||||
dcl_temps 1
|
||||
bfi r0.x, l(1), l(1), v0.x, l(0)
|
||||
and r0.z, v0.x, l(2)
|
||||
utof r0.xy, r0.xzxx
|
||||
mad o0.xy, r0.xyxx, l(2.000000, -2.000000, 0.000000, 0.000000), l(-1.000000, 1.000000, 0.000000, 0.000000)
|
||||
mov o1.xy, r0.xyxx
|
||||
mov o0.zw, l(0,0,0,1.000000)
|
||||
ret
|
||||
// Approximately 7 instruction slots used
|
||||
#endif
|
||||
|
||||
const BYTE g_WebcamCompositeVS[] =
|
||||
{
|
||||
68, 88, 66, 67, 94, 195,
|
||||
253, 40, 165, 172, 45, 84,
|
||||
30, 136, 47, 40, 247, 112,
|
||||
58, 27, 1, 0, 0, 0,
|
||||
224, 2, 0, 0, 5, 0,
|
||||
0, 0, 52, 0, 0, 0,
|
||||
160, 0, 0, 0, 212, 0,
|
||||
0, 0, 44, 1, 0, 0,
|
||||
68, 2, 0, 0, 82, 68,
|
||||
69, 70, 100, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
60, 0, 0, 0, 0, 5,
|
||||
254, 255, 0, 1, 0, 0,
|
||||
60, 0, 0, 0, 82, 68,
|
||||
49, 49, 60, 0, 0, 0,
|
||||
24, 0, 0, 0, 32, 0,
|
||||
0, 0, 40, 0, 0, 0,
|
||||
36, 0, 0, 0, 12, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
77, 105, 99, 114, 111, 115,
|
||||
111, 102, 116, 32, 40, 82,
|
||||
41, 32, 72, 76, 83, 76,
|
||||
32, 83, 104, 97, 100, 101,
|
||||
114, 32, 67, 111, 109, 112,
|
||||
105, 108, 101, 114, 32, 49,
|
||||
48, 46, 49, 0, 73, 83,
|
||||
71, 78, 44, 0, 0, 0,
|
||||
1, 0, 0, 0, 8, 0,
|
||||
0, 0, 32, 0, 0, 0,
|
||||
0, 0, 0, 0, 6, 0,
|
||||
0, 0, 1, 0, 0, 0,
|
||||
0, 0, 0, 0, 1, 1,
|
||||
0, 0, 83, 86, 95, 86,
|
||||
101, 114, 116, 101, 120, 73,
|
||||
68, 0, 79, 83, 71, 78,
|
||||
80, 0, 0, 0, 2, 0,
|
||||
0, 0, 8, 0, 0, 0,
|
||||
56, 0, 0, 0, 0, 0,
|
||||
0, 0, 1, 0, 0, 0,
|
||||
3, 0, 0, 0, 0, 0,
|
||||
0, 0, 15, 0, 0, 0,
|
||||
68, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
3, 0, 0, 0, 1, 0,
|
||||
0, 0, 3, 12, 0, 0,
|
||||
83, 86, 95, 80, 79, 83,
|
||||
73, 84, 73, 79, 78, 0,
|
||||
84, 69, 88, 67, 79, 79,
|
||||
82, 68, 0, 171, 171, 171,
|
||||
83, 72, 69, 88, 16, 1,
|
||||
0, 0, 80, 0, 1, 0,
|
||||
68, 0, 0, 0, 106, 8,
|
||||
0, 1, 96, 0, 0, 4,
|
||||
18, 16, 16, 0, 0, 0,
|
||||
0, 0, 6, 0, 0, 0,
|
||||
103, 0, 0, 4, 242, 32,
|
||||
16, 0, 0, 0, 0, 0,
|
||||
1, 0, 0, 0, 101, 0,
|
||||
0, 3, 50, 32, 16, 0,
|
||||
1, 0, 0, 0, 104, 0,
|
||||
0, 2, 1, 0, 0, 0,
|
||||
140, 0, 0, 11, 18, 0,
|
||||
16, 0, 0, 0, 0, 0,
|
||||
1, 64, 0, 0, 1, 0,
|
||||
0, 0, 1, 64, 0, 0,
|
||||
1, 0, 0, 0, 10, 16,
|
||||
16, 0, 0, 0, 0, 0,
|
||||
1, 64, 0, 0, 0, 0,
|
||||
0, 0, 1, 0, 0, 7,
|
||||
66, 0, 16, 0, 0, 0,
|
||||
0, 0, 10, 16, 16, 0,
|
||||
0, 0, 0, 0, 1, 64,
|
||||
0, 0, 2, 0, 0, 0,
|
||||
86, 0, 0, 5, 50, 0,
|
||||
16, 0, 0, 0, 0, 0,
|
||||
134, 0, 16, 0, 0, 0,
|
||||
0, 0, 50, 0, 0, 15,
|
||||
50, 32, 16, 0, 0, 0,
|
||||
0, 0, 70, 0, 16, 0,
|
||||
0, 0, 0, 0, 2, 64,
|
||||
0, 0, 0, 0, 0, 64,
|
||||
0, 0, 0, 192, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
2, 64, 0, 0, 0, 0,
|
||||
128, 191, 0, 0, 128, 63,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 54, 0, 0, 5,
|
||||
50, 32, 16, 0, 1, 0,
|
||||
0, 0, 70, 0, 16, 0,
|
||||
0, 0, 0, 0, 54, 0,
|
||||
0, 8, 194, 32, 16, 0,
|
||||
0, 0, 0, 0, 2, 64,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 128, 63,
|
||||
62, 0, 0, 1, 83, 84,
|
||||
65, 84, 148, 0, 0, 0,
|
||||
7, 0, 0, 0, 1, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
3, 0, 0, 0, 1, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
1, 0, 0, 0, 1, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 2, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
1, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0
|
||||
};
|
||||
@@ -121,7 +121,7 @@ FONT 8, "MS Shell Dlg", 0, 0, 0x0
|
||||
BEGIN
|
||||
DEFPUSHBUTTON "OK",IDOK,184,308,50,14
|
||||
PUSHBUTTON "Cancel",IDCANCEL,241,308,50,14
|
||||
LTEXT "ZoomIt v12.0",IDC_VERSION,42,7,73,10
|
||||
LTEXT "ZoomIt v12.1",IDC_VERSION,42,7,73,10
|
||||
LTEXT "Copyright \251 2006-2026 Mark Russinovich",IDC_COPYRIGHT,42,17,251,8
|
||||
CONTROL "<a HREF=""https://www.sysinternals.com"">Sysinternals - www.sysinternals.com</a>",IDC_LINK,
|
||||
"SysLink",WS_TABSTOP,42,26,150,9
|
||||
@@ -267,33 +267,52 @@ RECORD DIALOGEX 0, 0, 263, 228
|
||||
STYLE DS_SETFONT | DS_FIXEDSYS | DS_CONTROL | WS_CHILD | WS_SYSMENU
|
||||
FONT 8, "MS Shell Dlg", 400, 0, 0x1
|
||||
BEGIN
|
||||
CONTROL "",IDC_RECORD_HOTKEY,"msctls_hotkey32",WS_BORDER | WS_TABSTOP,61,71,80,12
|
||||
LTEXT "Record Toggle:",IDC_STATIC,7,73,54,8
|
||||
CONTROL "",IDC_RECORD_HOTKEY,"msctls_hotkey32",WS_BORDER | WS_TABSTOP,65,59,80,12
|
||||
RTEXT "Record Toggle:",IDC_STATIC,7,62,54,8
|
||||
LTEXT "Record video of the unzoomed live screen or a static zoomed session by entering the recording hot key and finish the recording by entering it again. ",IDC_STATIC,7,7,248,22
|
||||
LTEXT "Scaling:",IDC_STATIC,30,90,26,8
|
||||
COMBOBOX IDC_RECORD_SCALING,61,89,26,30,CBS_DROPDOWNLIST | CBS_AUTOHSCROLL | CBS_OEMCONVERT | CBS_SORT | WS_VSCROLL | WS_TABSTOP
|
||||
CONTROL "16:9:",IDC_RECORD_ASPECT_RATIO,"Button",BS_AUTOCHECKBOX | BS_LEFTTEXT | WS_TABSTOP,93,90,30,10
|
||||
LTEXT "Format:",IDC_STATIC,30,108,26,8
|
||||
COMBOBOX IDC_RECORD_FORMAT,61,106,60,30,CBS_DROPDOWNLIST | CBS_AUTOHSCROLL | CBS_OEMCONVERT | WS_VSCROLL | WS_TABSTOP
|
||||
LTEXT "Frame Rate:",IDC_STATIC,135,90,44,8,NOT WS_VISIBLE
|
||||
COMBOBOX IDC_RECORD_FRAME_RATE,182,89,42,30,CBS_DROPDOWNLIST | CBS_AUTOHSCROLL | CBS_OEMCONVERT | CBS_SORT | NOT WS_VISIBLE | WS_VSCROLL | WS_TABSTOP
|
||||
LTEXT "To crop the portion of the screen that will be recorded, enter the hotkey with the Shift key in the opposite mode. ",IDC_STATIC,7,29,245,18
|
||||
LTEXT "To record a specific window, enter the hotkey with the Alt key in the opposite mode.",IDC_STATIC,7,48,251,19
|
||||
CONTROL "Capture &system audio",IDC_CAPTURE_SYSTEM_AUDIO,"Button",BS_AUTOCHECKBOX | BS_LEFTTEXT | WS_TABSTOP,7,124,86,10
|
||||
CONTROL "&Capture audio input:",IDC_CAPTURE_AUDIO,"Button",BS_AUTOCHECKBOX | BS_LEFTTEXT | WS_TABSTOP,125,124,83,10
|
||||
COMBOBOX IDC_MICROPHONE,81,137,128,30,CBS_DROPDOWNLIST | WS_VSCROLL | WS_TABSTOP
|
||||
LTEXT "Microphone:",IDC_MICROPHONE_LABEL,34,139,47,8
|
||||
RTEXT "Scaling:",IDC_STATIC,36,79,26,8
|
||||
COMBOBOX IDC_RECORD_SCALING,65,78,26,30,CBS_DROPDOWNLIST | CBS_AUTOHSCROLL | CBS_OEMCONVERT | CBS_SORT | WS_VSCROLL | WS_TABSTOP
|
||||
CONTROL "16:9:",IDC_RECORD_ASPECT_RATIO,"Button",BS_AUTOCHECKBOX | BS_LEFTTEXT | WS_TABSTOP,94,79,30,10
|
||||
RTEXT "Format:",IDC_STATIC,36,97,26,8
|
||||
COMBOBOX IDC_RECORD_FORMAT,65,96,60,30,CBS_DROPDOWNLIST | CBS_AUTOHSCROLL | CBS_OEMCONVERT | WS_VSCROLL | WS_TABSTOP
|
||||
LTEXT "Frame Rate:",IDC_STATIC,134,79,44,8,NOT WS_VISIBLE
|
||||
COMBOBOX IDC_RECORD_FRAME_RATE,177,78,42,30,CBS_DROPDOWNLIST | CBS_AUTOHSCROLL | CBS_OEMCONVERT | CBS_SORT | NOT WS_VISIBLE | WS_VSCROLL | WS_TABSTOP
|
||||
LTEXT "To crop the portion of the screen that will be recorded, enter the hotkey with the Shift key in the opposite mode. ",IDC_STATIC,7,7,245,18
|
||||
LTEXT "To record a specific window, enter the hotkey with the Alt key in the opposite mode.",IDC_STATIC,7,38,251,19
|
||||
CONTROL "Capture &system audio:",IDC_CAPTURE_SYSTEM_AUDIO,"Button",BS_AUTOCHECKBOX | BS_LEFTTEXT | BS_RIGHT | WS_TABSTOP,7,115,90,10
|
||||
CONTROL "&Capture audio input:",IDC_CAPTURE_AUDIO,"Button",BS_AUTOCHECKBOX | BS_LEFTTEXT | BS_RIGHT | WS_TABSTOP,11,133,86,10
|
||||
CONTROL "Mono:",IDC_MIC_MONO_MIX,"Button",BS_AUTOCHECKBOX | BS_LEFTTEXT | BS_RIGHT | WS_TABSTOP,108,147,48,10
|
||||
COMBOBOX IDC_MICROPHONE,52,162,208,30,CBS_DROPDOWNLIST | WS_VSCROLL | WS_TABSTOP
|
||||
LTEXT "Microphone:",IDC_MICROPHONE_LABEL,7,163,42,8
|
||||
CONTROL "&Noise cancellation:",IDC_NOISE_CANCELLATION,"Button",BS_AUTOCHECKBOX | BS_LEFTTEXT | BS_RIGHT | WS_TABSTOP,23,147,74,10
|
||||
CONTROL "Show &webcam overlay (Ctrl+C toggles)",IDC_WEBCAM_OVERLAY,
|
||||
"Button",BS_AUTOCHECKBOX | BS_LEFTTEXT | WS_TABSTOP,7,153,148,10
|
||||
LTEXT "Camera:",IDC_WEBCAM_DEVICE_LABEL,46,167,28,8
|
||||
COMBOBOX IDC_WEBCAM_DEVICE,82,165,127,30,CBS_DROPDOWNLIST | WS_VSCROLL | WS_TABSTOP
|
||||
LTEXT "Position:",IDC_WEBCAM_POSITION_LABEL,33,185,32,8
|
||||
COMBOBOX IDC_WEBCAM_POSITION,64,183,55,30,CBS_DROPDOWNLIST | WS_VSCROLL | WS_TABSTOP
|
||||
LTEXT "Size:",IDC_WEBCAM_SIZE_LABEL,137,185,20,8
|
||||
COMBOBOX IDC_WEBCAM_SIZE,159,183,50,30,CBS_DROPDOWNLIST | WS_VSCROLL | WS_TABSTOP
|
||||
LTEXT "Shape:",IDC_WEBCAM_SHAPE_LABEL,33,201,24,8
|
||||
COMBOBOX IDC_WEBCAM_SHAPE,64,199,80,30,CBS_DROPDOWNLIST | WS_VSCROLL | WS_TABSTOP
|
||||
PUSHBUTTON "&Trim",IDC_TRIM_FILE,207,207,53,14
|
||||
"Button",BS_AUTOCHECKBOX | BS_LEFTTEXT | WS_TABSTOP,7,181,148,10
|
||||
PUSHBUTTON "Webcam S&ettings...",IDC_WEBCAM_SETTINGS,192,180,68,14
|
||||
PUSHBUTTON "&Trim",IDC_TRIM_FILE,207,209,53,14
|
||||
END
|
||||
|
||||
WEBCAM_SETTINGS DIALOGEX 0, 0, 220, 163
|
||||
STYLE DS_SETFONT | DS_MODALFRAME | DS_FIXEDSYS | DS_CENTER | WS_POPUP | WS_CAPTION | WS_SYSMENU
|
||||
CAPTION "Webcam Settings"
|
||||
FONT 8, "MS Shell Dlg", 400, 0, 0x1
|
||||
BEGIN
|
||||
LTEXT "Camera:",IDC_WEBCAM_DEVICE_LABEL,14,10,28,8
|
||||
COMBOBOX IDC_WEBCAM_DEVICE,50,8,158,30,CBS_DROPDOWNLIST | WS_VSCROLL | WS_TABSTOP
|
||||
LTEXT "Position:",IDC_WEBCAM_POSITION_LABEL,14,28,32,8
|
||||
COMBOBOX IDC_WEBCAM_POSITION,50,26,65,30,CBS_DROPDOWNLIST | WS_VSCROLL | WS_TABSTOP
|
||||
LTEXT "Size:",IDC_WEBCAM_SIZE_LABEL,125,28,20,8
|
||||
COMBOBOX IDC_WEBCAM_SIZE,148,26,60,30,CBS_DROPDOWNLIST | WS_VSCROLL | WS_TABSTOP
|
||||
LTEXT "Shape:",IDC_WEBCAM_SHAPE_LABEL,14,46,24,8
|
||||
COMBOBOX IDC_WEBCAM_SHAPE,50,44,80,30,CBS_DROPDOWNLIST | WS_VSCROLL | WS_TABSTOP
|
||||
LTEXT "Background:",IDC_WEBCAM_BG_LABEL,14,64,44,8
|
||||
COMBOBOX IDC_WEBCAM_BG_MODE,60,62,55,30,CBS_DROPDOWNLIST | WS_VSCROLL | WS_TABSTOP
|
||||
EDITTEXT IDC_WEBCAM_BG_IMAGE,117,62,83,12,ES_AUTOHSCROLL | ES_READONLY
|
||||
PUSHBUTTON "...",IDC_WEBCAM_BG_BROWSE,202,62,14,12
|
||||
LTEXT "Brightness:",IDC_WEBCAM_BRIGHTNESS_LABEL,14,82,44,8
|
||||
CONTROL "",IDC_WEBCAM_BRIGHTNESS_SLIDER,"msctls_trackbar32",TBS_AUTOTICKS | TBS_NOTICKS | TBS_TOOLTIPS | WS_TABSTOP,60,80,148,15
|
||||
LTEXT "Uses MediaPipe SelfieSegmentation (Apache 2.0)",IDC_THIRDPARTY_NOTICES,14,102,200,8
|
||||
DEFPUSHBUTTON "OK",IDOK,108,142,50,14
|
||||
PUSHBUTTON "Cancel",IDCANCEL,162,142,50,14
|
||||
END
|
||||
|
||||
SNIP DIALOGEX 0, 0, 260, 80
|
||||
|
||||
@@ -90,7 +90,7 @@
|
||||
<AdditionalIncludeDirectories>$(MSBuildThisFileDirectory)..\..\..\common\version;$(MSBuildThisFileDirectory)PowerToys;$(InterPlatformDir)</AdditionalIncludeDirectories>
|
||||
</ResourceCompile>
|
||||
<Link>
|
||||
<AdditionalDependencies>Shlwapi.lib;comctl32.lib;odbc32.lib;odbccp32.lib;version.lib;Winmm.lib;gdiplus.lib;Msimg32.lib;Wtsapi32.lib;%(AdditionalDependencies)</AdditionalDependencies>
|
||||
<AdditionalDependencies>Shlwapi.lib;comctl32.lib;odbc32.lib;odbccp32.lib;version.lib;Winmm.lib;gdiplus.lib;Msimg32.lib;Wtsapi32.lib;windowscodecs.lib;%(AdditionalDependencies)</AdditionalDependencies>
|
||||
<GenerateDebugInformation>true</GenerateDebugInformation>
|
||||
<SubSystem>Windows</SubSystem>
|
||||
<RandomizedBaseAddress>true</RandomizedBaseAddress>
|
||||
@@ -112,7 +112,7 @@
|
||||
<AdditionalIncludeDirectories>$(MSBuildThisFileDirectory)..\..\..\common\version;$(MSBuildThisFileDirectory)PowerToys;$(MSBuildThisFileDirectory)..\ZoomItBreak\$(Platform)\$(Configuration)\</AdditionalIncludeDirectories>
|
||||
</ResourceCompile>
|
||||
<Link>
|
||||
<AdditionalDependencies>Shlwapi.lib;comctl32.lib;odbc32.lib;odbccp32.lib;version.lib;Winmm.lib;gdiplus.lib;Msimg32.lib;Wtsapi32.lib;%(AdditionalDependencies)</AdditionalDependencies>
|
||||
<AdditionalDependencies>Shlwapi.lib;comctl32.lib;odbc32.lib;odbccp32.lib;version.lib;Winmm.lib;gdiplus.lib;Msimg32.lib;Wtsapi32.lib;windowscodecs.lib;%(AdditionalDependencies)</AdditionalDependencies>
|
||||
<GenerateDebugInformation>true</GenerateDebugInformation>
|
||||
<SubSystem>Windows</SubSystem>
|
||||
<RandomizedBaseAddress>true</RandomizedBaseAddress>
|
||||
@@ -135,7 +135,7 @@
|
||||
<AdditionalIncludeDirectories>$(MSBuildThisFileDirectory)..\..\..\common\version;$(MSBuildThisFileDirectory)PowerToys;$(MSBuildThisFileDirectory)..\ZoomItBreak\$(Platform)\$(Configuration)\</AdditionalIncludeDirectories>
|
||||
</ResourceCompile>
|
||||
<Link>
|
||||
<AdditionalDependencies>Shlwapi.lib;comctl32.lib;odbc32.lib;odbccp32.lib;version.lib;Winmm.lib;gdiplus.lib;Msimg32.lib;Wtsapi32.lib;%(AdditionalDependencies)</AdditionalDependencies>
|
||||
<AdditionalDependencies>Shlwapi.lib;comctl32.lib;odbc32.lib;odbccp32.lib;version.lib;Winmm.lib;gdiplus.lib;Msimg32.lib;Wtsapi32.lib;windowscodecs.lib;%(AdditionalDependencies)</AdditionalDependencies>
|
||||
<GenerateDebugInformation>true</GenerateDebugInformation>
|
||||
<SubSystem>Windows</SubSystem>
|
||||
<FixedBaseAddress>
|
||||
@@ -156,7 +156,7 @@
|
||||
<AdditionalIncludeDirectories>$(MSBuildThisFileDirectory)..\..\..\common\version;$(MSBuildThisFileDirectory)PowerToys;$(InterPlatformDir)</AdditionalIncludeDirectories>
|
||||
</ResourceCompile>
|
||||
<Link>
|
||||
<AdditionalDependencies>Shlwapi.lib;comctl32.lib;odbc32.lib;odbccp32.lib;version.lib;Winmm.lib;gdiplus.lib;Msimg32.lib;Wtsapi32.lib;%(AdditionalDependencies)</AdditionalDependencies>
|
||||
<AdditionalDependencies>Shlwapi.lib;comctl32.lib;odbc32.lib;odbccp32.lib;version.lib;Winmm.lib;gdiplus.lib;Msimg32.lib;Wtsapi32.lib;windowscodecs.lib;%(AdditionalDependencies)</AdditionalDependencies>
|
||||
<GenerateDebugInformation>true</GenerateDebugInformation>
|
||||
<SubSystem>Windows</SubSystem>
|
||||
<RandomizedBaseAddress>false</RandomizedBaseAddress>
|
||||
@@ -177,7 +177,7 @@
|
||||
<AdditionalIncludeDirectories>$(MSBuildThisFileDirectory)..\..\..\common\version;$(MSBuildThisFileDirectory)PowerToys;$(MSBuildThisFileDirectory)..\ZoomItBreak\$(Platform)\$(Configuration)\</AdditionalIncludeDirectories>
|
||||
</ResourceCompile>
|
||||
<Link>
|
||||
<AdditionalDependencies>Shlwapi.lib;comctl32.lib;odbc32.lib;odbccp32.lib;version.lib;Winmm.lib;gdiplus.lib;Msimg32.lib;Wtsapi32.lib;%(AdditionalDependencies)</AdditionalDependencies>
|
||||
<AdditionalDependencies>Shlwapi.lib;comctl32.lib;odbc32.lib;odbccp32.lib;version.lib;Winmm.lib;gdiplus.lib;Msimg32.lib;Wtsapi32.lib;windowscodecs.lib;%(AdditionalDependencies)</AdditionalDependencies>
|
||||
<UACUIAccess>true</UACUIAccess>
|
||||
<GenerateDebugInformation>true</GenerateDebugInformation>
|
||||
<SubSystem>Windows</SubSystem>
|
||||
@@ -199,7 +199,7 @@
|
||||
<AdditionalIncludeDirectories>$(MSBuildThisFileDirectory)..\..\..\common\version;$(MSBuildThisFileDirectory)PowerToys;$(MSBuildThisFileDirectory)..\ZoomItBreak\$(Platform)\$(Configuration)\</AdditionalIncludeDirectories>
|
||||
</ResourceCompile>
|
||||
<Link>
|
||||
<AdditionalDependencies>Shlwapi.lib;comctl32.lib;odbc32.lib;odbccp32.lib;version.lib;Winmm.lib;gdiplus.lib;Msimg32.lib;Wtsapi32.lib;%(AdditionalDependencies)</AdditionalDependencies>
|
||||
<AdditionalDependencies>Shlwapi.lib;comctl32.lib;odbc32.lib;odbccp32.lib;version.lib;Winmm.lib;gdiplus.lib;Msimg32.lib;Wtsapi32.lib;windowscodecs.lib;%(AdditionalDependencies)</AdditionalDependencies>
|
||||
<UACUIAccess>true</UACUIAccess>
|
||||
<GenerateDebugInformation>true</GenerateDebugInformation>
|
||||
<SubSystem>Windows</SubSystem>
|
||||
@@ -257,6 +257,7 @@
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Release|x64'">NotUsing</PrecompiledHeader>
|
||||
</ClCompile>
|
||||
<ClCompile Include="GifRecordingSession.cpp" />
|
||||
<ClCompile Include="NoiseSuppressor.cpp" />
|
||||
<ClCompile Include="PanoramaCapture.cpp">
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">Use</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Release|ARM64'">Use</PrecompiledHeader>
|
||||
@@ -266,6 +267,128 @@
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Debug|x64'">Use</PrecompiledHeader>
|
||||
</ClCompile>
|
||||
<ClCompile Include="pch.cpp" />
|
||||
<ClCompile Include="rnnoise\celt_lpc.c">
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">NotUsing</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">NotUsing</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Debug|ARM64'">NotUsing</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Release|ARM64'">NotUsing</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Debug|x64'">NotUsing</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Release|x64'">NotUsing</PrecompiledHeader>
|
||||
<DisableSpecificWarnings Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">26451;4100;4091;4245</DisableSpecificWarnings>
|
||||
<DisableSpecificWarnings Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">26451;4100;4091;4245</DisableSpecificWarnings>
|
||||
<DisableSpecificWarnings Condition="'$(Configuration)|$(Platform)'=='Debug|ARM64'">26451;4100;4091;4245</DisableSpecificWarnings>
|
||||
<DisableSpecificWarnings Condition="'$(Configuration)|$(Platform)'=='Release|ARM64'">26451;4100;4091;4245</DisableSpecificWarnings>
|
||||
<DisableSpecificWarnings Condition="'$(Configuration)|$(Platform)'=='Debug|x64'">26451;4100;4091;4245</DisableSpecificWarnings>
|
||||
<DisableSpecificWarnings Condition="'$(Configuration)|$(Platform)'=='Release|x64'">26451;4100;4091;4245</DisableSpecificWarnings>
|
||||
</ClCompile>
|
||||
<ClCompile Include="rnnoise\denoise.c">
|
||||
<DisableSpecificWarnings Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">26451;4100;4091;4244;4245;4305</DisableSpecificWarnings>
|
||||
<DisableSpecificWarnings Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">26451;4100;4091;4244;4245;4305</DisableSpecificWarnings>
|
||||
<DisableSpecificWarnings Condition="'$(Configuration)|$(Platform)'=='Debug|ARM64'">26451;4100;4091;4244;4245;4305</DisableSpecificWarnings>
|
||||
<DisableSpecificWarnings Condition="'$(Configuration)|$(Platform)'=='Release|ARM64'">26451;4100;4091;4244;4245;4305</DisableSpecificWarnings>
|
||||
<DisableSpecificWarnings Condition="'$(Configuration)|$(Platform)'=='Debug|x64'">26451;4100;4091;4244;4245;4305</DisableSpecificWarnings>
|
||||
<DisableSpecificWarnings Condition="'$(Configuration)|$(Platform)'=='Release|x64'">26451;4100;4091;4244;4245;4305</DisableSpecificWarnings>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">NotUsing</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">NotUsing</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Debug|ARM64'">NotUsing</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Release|ARM64'">NotUsing</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Debug|x64'">NotUsing</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Release|x64'">NotUsing</PrecompiledHeader>
|
||||
</ClCompile>
|
||||
<ClCompile Include="rnnoise\kiss_fft.c">
|
||||
<DisableSpecificWarnings Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">26451;4100;4091;4244;4245</DisableSpecificWarnings>
|
||||
<DisableSpecificWarnings Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">26451;4100;4091;4244;4245</DisableSpecificWarnings>
|
||||
<DisableSpecificWarnings Condition="'$(Configuration)|$(Platform)'=='Debug|ARM64'">26451;4100;4091;4244;4245</DisableSpecificWarnings>
|
||||
<DisableSpecificWarnings Condition="'$(Configuration)|$(Platform)'=='Release|ARM64'">26451;4100;4091;4244;4245</DisableSpecificWarnings>
|
||||
<DisableSpecificWarnings Condition="'$(Configuration)|$(Platform)'=='Debug|x64'">26451;4100;4091;4244;4245</DisableSpecificWarnings>
|
||||
<DisableSpecificWarnings Condition="'$(Configuration)|$(Platform)'=='Release|x64'">26451;4100;4091;4244;4245</DisableSpecificWarnings>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">NotUsing</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">NotUsing</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Debug|ARM64'">NotUsing</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Release|ARM64'">NotUsing</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Debug|x64'">NotUsing</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Release|x64'">NotUsing</PrecompiledHeader>
|
||||
</ClCompile>
|
||||
<ClCompile Include="rnnoise\nnet.c">
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">NotUsing</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">NotUsing</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Debug|ARM64'">NotUsing</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Release|ARM64'">NotUsing</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Debug|x64'">NotUsing</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Release|x64'">NotUsing</PrecompiledHeader>
|
||||
<DisableSpecificWarnings Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">26451;4100;4091;4244;4245</DisableSpecificWarnings>
|
||||
<DisableSpecificWarnings Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">26451;4100;4091;4244;4245</DisableSpecificWarnings>
|
||||
<DisableSpecificWarnings Condition="'$(Configuration)|$(Platform)'=='Debug|ARM64'">26451;4100;4091;4244;4245</DisableSpecificWarnings>
|
||||
<DisableSpecificWarnings Condition="'$(Configuration)|$(Platform)'=='Release|ARM64'">26451;4100;4091;4244;4245</DisableSpecificWarnings>
|
||||
<DisableSpecificWarnings Condition="'$(Configuration)|$(Platform)'=='Debug|x64'">26451;4100;4091;4244;4245</DisableSpecificWarnings>
|
||||
<DisableSpecificWarnings Condition="'$(Configuration)|$(Platform)'=='Release|x64'">26451;4100;4091;4244;4245</DisableSpecificWarnings>
|
||||
</ClCompile>
|
||||
<ClCompile Include="rnnoise\nnet_default.c">
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">NotUsing</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">NotUsing</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Debug|ARM64'">NotUsing</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Release|ARM64'">NotUsing</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Debug|x64'">NotUsing</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Release|x64'">NotUsing</PrecompiledHeader>
|
||||
<DisableSpecificWarnings Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">26451;4100;4091;4244;4245</DisableSpecificWarnings>
|
||||
<DisableSpecificWarnings Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">26451;4100;4091;4244;4245</DisableSpecificWarnings>
|
||||
<DisableSpecificWarnings Condition="'$(Configuration)|$(Platform)'=='Debug|ARM64'">26451;4100;4091;4244;4245</DisableSpecificWarnings>
|
||||
<DisableSpecificWarnings Condition="'$(Configuration)|$(Platform)'=='Release|ARM64'">26451;4100;4091;4244;4245</DisableSpecificWarnings>
|
||||
<DisableSpecificWarnings Condition="'$(Configuration)|$(Platform)'=='Debug|x64'">26451;4100;4091;4244;4245</DisableSpecificWarnings>
|
||||
<DisableSpecificWarnings Condition="'$(Configuration)|$(Platform)'=='Release|x64'">26451;4100;4091;4244;4245</DisableSpecificWarnings>
|
||||
</ClCompile>
|
||||
<ClCompile Include="rnnoise\parse_lpcnet_weights.c">
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">NotUsing</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">NotUsing</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Debug|ARM64'">NotUsing</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Release|ARM64'">NotUsing</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Debug|x64'">NotUsing</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Release|x64'">NotUsing</PrecompiledHeader>
|
||||
</ClCompile>
|
||||
<ClCompile Include="rnnoise\pitch.c">
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">NotUsing</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">NotUsing</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Debug|ARM64'">NotUsing</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Release|ARM64'">NotUsing</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Debug|x64'">NotUsing</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Release|x64'">NotUsing</PrecompiledHeader>
|
||||
<DisableSpecificWarnings Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">26451;4100;4091;4244;4245</DisableSpecificWarnings>
|
||||
<DisableSpecificWarnings Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">26451;4100;4091;4244;4245</DisableSpecificWarnings>
|
||||
<DisableSpecificWarnings Condition="'$(Configuration)|$(Platform)'=='Debug|ARM64'">26451;4100;4091;4244;4245</DisableSpecificWarnings>
|
||||
<DisableSpecificWarnings Condition="'$(Configuration)|$(Platform)'=='Release|ARM64'">26451;4100;4091;4244;4245</DisableSpecificWarnings>
|
||||
<DisableSpecificWarnings Condition="'$(Configuration)|$(Platform)'=='Debug|x64'">26451;4100;4091;4244;4245</DisableSpecificWarnings>
|
||||
<DisableSpecificWarnings Condition="'$(Configuration)|$(Platform)'=='Release|x64'">26451;4100;4091;4244;4245</DisableSpecificWarnings>
|
||||
</ClCompile>
|
||||
<ClCompile Include="rnnoise\rnn.c">
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">NotUsing</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">NotUsing</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Debug|ARM64'">NotUsing</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Release|ARM64'">NotUsing</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Debug|x64'">NotUsing</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Release|x64'">NotUsing</PrecompiledHeader>
|
||||
</ClCompile>
|
||||
<ClCompile Include="rnnoise\rnnoise_data_little.c">
|
||||
<DisableSpecificWarnings Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">26451;4100;4091;4245;4305</DisableSpecificWarnings>
|
||||
<DisableSpecificWarnings Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">26451;4100;4091;4245;4305</DisableSpecificWarnings>
|
||||
<DisableSpecificWarnings Condition="'$(Configuration)|$(Platform)'=='Debug|ARM64'">26451;4100;4091;4245;4305</DisableSpecificWarnings>
|
||||
<DisableSpecificWarnings Condition="'$(Configuration)|$(Platform)'=='Release|ARM64'">26451;4100;4091;4245;4305</DisableSpecificWarnings>
|
||||
<DisableSpecificWarnings Condition="'$(Configuration)|$(Platform)'=='Debug|x64'">26451;4100;4091;4245;4305</DisableSpecificWarnings>
|
||||
<DisableSpecificWarnings Condition="'$(Configuration)|$(Platform)'=='Release|x64'">26451;4100;4091;4245;4305</DisableSpecificWarnings>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">NotUsing</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">NotUsing</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Debug|ARM64'">NotUsing</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Release|ARM64'">NotUsing</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Debug|x64'">NotUsing</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Release|x64'">NotUsing</PrecompiledHeader>
|
||||
</ClCompile>
|
||||
<ClCompile Include="rnnoise\rnnoise_tables.c">
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">NotUsing</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">NotUsing</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Debug|ARM64'">NotUsing</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Release|ARM64'">NotUsing</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Debug|x64'">NotUsing</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Release|x64'">NotUsing</PrecompiledHeader>
|
||||
</ClCompile>
|
||||
<ClCompile Include="SelectRectangle.cpp">
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">Use</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Release|ARM64'">Use</PrecompiledHeader>
|
||||
@@ -314,6 +437,14 @@
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Debug|x64'">Use</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Release|x64'">Use</PrecompiledHeader>
|
||||
</ClCompile>
|
||||
<ClCompile Include="BackgroundBlur.cpp">
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">Use</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">Use</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Debug|ARM64'">Use</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Release|ARM64'">Use</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Debug|x64'">Use</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Release|x64'">Use</PrecompiledHeader>
|
||||
</ClCompile>
|
||||
<ClCompile Include="WebcamPreviewWindow.cpp">
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Debug|Win32'">Use</PrecompiledHeader>
|
||||
<PrecompiledHeader Condition="'$(Configuration)|$(Platform)'=='Release|Win32'">Use</PrecompiledHeader>
|
||||
@@ -338,16 +469,21 @@
|
||||
<ClInclude Include="$(MSBuildThisFileDirectory)..\..\..\common\sysinternals\Eula\Eula.h" />
|
||||
<ClInclude Include="$(MSBuildThisFileDirectory)..\ZoomItModuleInterface\Trace.h" />
|
||||
<ClInclude Include="GifRecordingSession.h" />
|
||||
<ClInclude Include="NoiseSuppressor.h" />
|
||||
<ClInclude Include="PanoramaCapture.h" />
|
||||
<ClInclude Include="pch.h" />
|
||||
<ClInclude Include="Registry.h" />
|
||||
<ClInclude Include="resource.h" />
|
||||
<ClInclude Include="rnnoise\nnet_arch.h" />
|
||||
<ClInclude Include="rnnoise\rnnoise.h" />
|
||||
<ClInclude Include="rnnoise\vec.h" />
|
||||
<ClInclude Include="SelectRectangle.h" />
|
||||
<ClInclude Include="Utility.h" />
|
||||
<ClInclude Include="DemoType.h" />
|
||||
<ClInclude Include="VersionHelper.h" />
|
||||
<ClInclude Include="VideoRecordingSession.h" />
|
||||
<ClInclude Include="WebcamCapture.h" />
|
||||
<ClInclude Include="BackgroundBlur.h" />
|
||||
<ClInclude Include="WebcamPreviewWindow.h" />
|
||||
<ClInclude Include="ZoomIt.h" />
|
||||
<ClInclude Include="ZoomItSettings.h" />
|
||||
|
||||
@@ -13,6 +13,12 @@
|
||||
<UniqueIdentifier>{e1fa606f-a2e6-40c8-8779-8ca1813d9f01}</UniqueIdentifier>
|
||||
<Extensions>ico;cur;bmp;dlg;rc2;rct;bin;rgs;gif;jpg;jpeg;jpe</Extensions>
|
||||
</Filter>
|
||||
<Filter Include="Source Files\rnnoise">
|
||||
<UniqueIdentifier>{c1cc2820-c3a4-413d-b5d8-0d2034de2474}</UniqueIdentifier>
|
||||
</Filter>
|
||||
<Filter Include="Header Files\rnnoise">
|
||||
<UniqueIdentifier>{7795b908-e073-46ed-b209-c07ac0324adb}</UniqueIdentifier>
|
||||
</Filter>
|
||||
</ItemGroup>
|
||||
<ItemGroup>
|
||||
<ClCompile Include="Zoomit.cpp">
|
||||
@@ -66,6 +72,48 @@
|
||||
<ClCompile Include="..\ZoomItBreak\BreakTimer.cpp">
|
||||
<Filter>Source Files</Filter>
|
||||
</ClCompile>
|
||||
<ClCompile Include="BackgroundBlur.cpp">
|
||||
<Filter>Source Files</Filter>
|
||||
</ClCompile>
|
||||
<ClCompile Include="WebcamCapture.cpp">
|
||||
<Filter>Source Files</Filter>
|
||||
</ClCompile>
|
||||
<ClCompile Include="WebcamPreviewWindow.cpp">
|
||||
<Filter>Source Files</Filter>
|
||||
</ClCompile>
|
||||
<ClCompile Include="rnnoise\celt_lpc.c">
|
||||
<Filter>Source Files\rnnoise</Filter>
|
||||
</ClCompile>
|
||||
<ClCompile Include="rnnoise\denoise.c">
|
||||
<Filter>Source Files\rnnoise</Filter>
|
||||
</ClCompile>
|
||||
<ClCompile Include="rnnoise\kiss_fft.c">
|
||||
<Filter>Source Files\rnnoise</Filter>
|
||||
</ClCompile>
|
||||
<ClCompile Include="rnnoise\nnet.c">
|
||||
<Filter>Source Files\rnnoise</Filter>
|
||||
</ClCompile>
|
||||
<ClCompile Include="rnnoise\nnet_default.c">
|
||||
<Filter>Source Files\rnnoise</Filter>
|
||||
</ClCompile>
|
||||
<ClCompile Include="rnnoise\parse_lpcnet_weights.c">
|
||||
<Filter>Source Files\rnnoise</Filter>
|
||||
</ClCompile>
|
||||
<ClCompile Include="rnnoise\pitch.c">
|
||||
<Filter>Source Files\rnnoise</Filter>
|
||||
</ClCompile>
|
||||
<ClCompile Include="rnnoise\rnn.c">
|
||||
<Filter>Source Files\rnnoise</Filter>
|
||||
</ClCompile>
|
||||
<ClCompile Include="rnnoise\rnnoise_data_little.c">
|
||||
<Filter>Source Files\rnnoise</Filter>
|
||||
</ClCompile>
|
||||
<ClCompile Include="rnnoise\rnnoise_tables.c">
|
||||
<Filter>Source Files\rnnoise</Filter>
|
||||
</ClCompile>
|
||||
<ClCompile Include="NoiseSuppressor.cpp">
|
||||
<Filter>Source Files</Filter>
|
||||
</ClCompile>
|
||||
</ItemGroup>
|
||||
<ItemGroup>
|
||||
<ClInclude Include="Registry.h">
|
||||
@@ -119,6 +167,27 @@
|
||||
<ClInclude Include="..\ZoomItBreak\BreakTimer.h">
|
||||
<Filter>Header Files</Filter>
|
||||
</ClInclude>
|
||||
<ClInclude Include="BackgroundBlur.h">
|
||||
<Filter>Header Files</Filter>
|
||||
</ClInclude>
|
||||
<ClInclude Include="WebcamCapture.h">
|
||||
<Filter>Header Files</Filter>
|
||||
</ClInclude>
|
||||
<ClInclude Include="WebcamPreviewWindow.h">
|
||||
<Filter>Header Files</Filter>
|
||||
</ClInclude>
|
||||
<ClInclude Include="rnnoise\vec.h">
|
||||
<Filter>Header Files\rnnoise</Filter>
|
||||
</ClInclude>
|
||||
<ClInclude Include="rnnoise\nnet_arch.h">
|
||||
<Filter>Header Files\rnnoise</Filter>
|
||||
</ClInclude>
|
||||
<ClInclude Include="NoiseSuppressor.h">
|
||||
<Filter>Header Files</Filter>
|
||||
</ClInclude>
|
||||
<ClInclude Include="rnnoise\rnnoise.h">
|
||||
<Filter>Header Files\rnnoise</Filter>
|
||||
</ClInclude>
|
||||
</ItemGroup>
|
||||
<ItemGroup>
|
||||
<Image Include="appicon.ico">
|
||||
@@ -150,5 +219,6 @@
|
||||
</ItemGroup>
|
||||
<ItemGroup>
|
||||
<Natvis Include="$(MSBuildThisFileDirectory)..\..\natvis\wil.natvis" />
|
||||
<Natvis Include="$(MSBuildThisFileDirectory)..\..\natvis\wil.natstepfilter" />
|
||||
</ItemGroup>
|
||||
</Project>
|
||||
@@ -56,12 +56,16 @@ RecordingFormat g_RecordingFormat = RecordingFormat::MP4;
|
||||
BOOLEAN g_CaptureSystemAudio = TRUE;
|
||||
BOOLEAN g_CaptureAudio = FALSE;
|
||||
BOOLEAN g_MicMonoMix = FALSE;
|
||||
BOOLEAN g_NoiseCancellation = TRUE;
|
||||
TCHAR g_MicrophoneDeviceId[MAX_PATH] = {0};
|
||||
BOOLEAN g_WebcamOverlay = FALSE;
|
||||
DWORD g_WebcamPosition = 3; // 0=TL, 1=TR, 2=BL, 3=BR
|
||||
DWORD g_WebcamSize = 1; // 0=Small(15%), 1=Medium(25%), 2=Large(33%), 3=XLarge(50%)
|
||||
DWORD g_WebcamShape = 0; // 0=Square, 1=RoundedRect, 2=RoundedSquare, 3=Circle
|
||||
TCHAR g_WebcamDeviceSymLink[MAX_PATH] = {0};
|
||||
DWORD g_WebcamBackgroundMode = 0; // 0=None, 1=Blur, 2=Image
|
||||
TCHAR g_WebcamBackgroundImage[MAX_PATH] = {0};
|
||||
DWORD g_WebcamBrightness = 50; // 0=dark, 50=neutral, 100=bright
|
||||
BOOLEAN g_RecordAspectRatio = FALSE; // Lock region selection to 16:9
|
||||
TCHAR g_RecordingSaveLocationBuffer[MAX_PATH] = {0};
|
||||
TCHAR g_ScreenshotSaveLocationBuffer[MAX_PATH] = {0};
|
||||
@@ -116,11 +120,15 @@ REG_SETTING RegSettings[] = {
|
||||
{ L"CaptureSystemAudio", SETTING_TYPE_BOOLEAN, 0, &g_CaptureSystemAudio, static_cast<DOUBLE>(g_CaptureSystemAudio) },
|
||||
{ L"MicMonoMix", SETTING_TYPE_BOOLEAN, 0, &g_MicMonoMix, static_cast<DOUBLE>(g_MicMonoMix) },
|
||||
{ L"MicrophoneDeviceId", SETTING_TYPE_STRING, sizeof(g_MicrophoneDeviceId), g_MicrophoneDeviceId, static_cast<DOUBLE>(0) },
|
||||
{ L"NoiseCancellation", SETTING_TYPE_BOOLEAN, 0, &g_NoiseCancellation, static_cast<DOUBLE>(g_NoiseCancellation) },
|
||||
{ L"WebcamOverlay", SETTING_TYPE_BOOLEAN, 0, &g_WebcamOverlay, static_cast<DOUBLE>(g_WebcamOverlay) },
|
||||
{ L"WebcamPosition", SETTING_TYPE_DWORD, 0, &g_WebcamPosition, static_cast<DOUBLE>(g_WebcamPosition) },
|
||||
{ L"WebcamSize", SETTING_TYPE_DWORD, 0, &g_WebcamSize, static_cast<DOUBLE>(g_WebcamSize) },
|
||||
{ L"WebcamShape", SETTING_TYPE_DWORD, 0, &g_WebcamShape, static_cast<DOUBLE>(g_WebcamShape) },
|
||||
{ L"WebcamDeviceSymLink", SETTING_TYPE_STRING, sizeof(g_WebcamDeviceSymLink), g_WebcamDeviceSymLink, static_cast<DOUBLE>(0) },
|
||||
{ L"WebcamBackgroundMode", SETTING_TYPE_DWORD, 0, &g_WebcamBackgroundMode, static_cast<DOUBLE>(g_WebcamBackgroundMode) },
|
||||
{ L"WebcamBackgroundImage", SETTING_TYPE_STRING, sizeof(g_WebcamBackgroundImage), g_WebcamBackgroundImage, static_cast<DOUBLE>(0) },
|
||||
{ L"WebcamBrightness", SETTING_TYPE_DWORD, 0, &g_WebcamBrightness, static_cast<DOUBLE>(g_WebcamBrightness) },
|
||||
{ L"RecordAspectRatio", SETTING_TYPE_BOOLEAN, 0, &g_RecordAspectRatio, static_cast<DOUBLE>(g_RecordAspectRatio) },
|
||||
{ L"RecordingSaveLocation", SETTING_TYPE_STRING, sizeof(g_RecordingSaveLocationBuffer), g_RecordingSaveLocationBuffer, static_cast<DOUBLE>(0) },
|
||||
{ L"ScreenshotSaveLocation", SETTING_TYPE_STRING, sizeof(g_ScreenshotSaveLocationBuffer), g_ScreenshotSaveLocationBuffer, static_cast<DOUBLE>(0) },
|
||||
|
||||
@@ -21,7 +21,6 @@
|
||||
#include "PanoramaCapture.h"
|
||||
#include <wtsapi32.h>
|
||||
#include <tlhelp32.h>
|
||||
#include <limits>
|
||||
#include <vector>
|
||||
|
||||
#ifdef __ZOOMIT_POWERTOYS__
|
||||
@@ -725,7 +724,7 @@ RunningOnWin64(
|
||||
// this executable.
|
||||
//
|
||||
//--------------------------------------------------------------------
|
||||
BOOLEAN ExtractImageResource( PTCHAR ResourceName, PTCHAR TargetFile )
|
||||
BOOLEAN ExtractImageResource( PCTSTR ResourceName, PTCHAR TargetFile )
|
||||
{
|
||||
HRSRC hResource;
|
||||
HGLOBAL hImageResource;
|
||||
@@ -2835,6 +2834,323 @@ INT_PTR CALLBACK AdvancedBreakProc( HWND hDlg, UINT message, WPARAM wParam, LPAR
|
||||
}
|
||||
|
||||
|
||||
//----------------------------------------------------------------------------
|
||||
//
|
||||
// WebcamSettingsProc
|
||||
//
|
||||
//----------------------------------------------------------------------------
|
||||
INT_PTR CALLBACK WebcamSettingsProc( HWND hDlg, UINT message, WPARAM wParam, LPARAM lParam )
|
||||
{
|
||||
static std::vector<std::pair<std::wstring, std::wstring>> webcams;
|
||||
static UINT currentDpi = DPI_BASELINE;
|
||||
static HWND s_hBrightnessTooltip = nullptr;
|
||||
|
||||
switch( message ) {
|
||||
case WM_INITDIALOG:
|
||||
{
|
||||
HICON hIcon = LoadIcon( g_hInstance, L"APPICON" );
|
||||
if( hIcon )
|
||||
{
|
||||
SendMessage( hDlg, WM_SETICON, ICON_BIG, reinterpret_cast<LPARAM>(hIcon) );
|
||||
SendMessage( hDlg, WM_SETICON, ICON_SMALL, reinterpret_cast<LPARAM>(hIcon) );
|
||||
}
|
||||
|
||||
// Enumerate webcam devices
|
||||
webcams.clear();
|
||||
{
|
||||
MFStartup( MF_VERSION, MFSTARTUP_LITE );
|
||||
IMFAttributes* pAttributes = nullptr;
|
||||
if( SUCCEEDED( MFCreateAttributes( &pAttributes, 1 ) ) )
|
||||
{
|
||||
pAttributes->SetGUID( MF_DEVSOURCE_ATTRIBUTE_SOURCE_TYPE, MF_DEVSOURCE_ATTRIBUTE_SOURCE_TYPE_VIDCAP_GUID );
|
||||
IMFActivate** ppDevices = nullptr;
|
||||
UINT32 count = 0;
|
||||
if( SUCCEEDED( MFEnumDeviceSources( pAttributes, &ppDevices, &count ) ) )
|
||||
{
|
||||
for( UINT32 i = 0; i < count; i++ )
|
||||
{
|
||||
LPWSTR symLink = nullptr, friendlyName = nullptr;
|
||||
UINT32 nameLen = 0;
|
||||
if( SUCCEEDED( ppDevices[i]->GetAllocatedString( MF_DEVSOURCE_ATTRIBUTE_SOURCE_TYPE_VIDCAP_SYMBOLIC_LINK, &symLink, &nameLen ) ) &&
|
||||
SUCCEEDED( ppDevices[i]->GetAllocatedString( MF_DEVSOURCE_ATTRIBUTE_FRIENDLY_NAME, &friendlyName, &nameLen ) ) )
|
||||
{
|
||||
webcams.emplace_back( symLink, friendlyName );
|
||||
}
|
||||
if( symLink ) CoTaskMemFree( symLink );
|
||||
if( friendlyName ) CoTaskMemFree( friendlyName );
|
||||
ppDevices[i]->Release();
|
||||
}
|
||||
CoTaskMemFree( ppDevices );
|
||||
}
|
||||
pAttributes->Release();
|
||||
}
|
||||
MFShutdown();
|
||||
}
|
||||
|
||||
// Populate camera combo
|
||||
SendMessage( GetDlgItem( hDlg, IDC_WEBCAM_DEVICE ), CB_ADDSTRING, 0, reinterpret_cast<LPARAM>(L"Default") );
|
||||
{
|
||||
int selection = 0;
|
||||
for( size_t i = 0; i < webcams.size(); i++ )
|
||||
{
|
||||
SendMessage( GetDlgItem( hDlg, IDC_WEBCAM_DEVICE ), CB_ADDSTRING, 0, reinterpret_cast<LPARAM>(webcams[i].second.c_str()) );
|
||||
if( selection == 0 && wcscmp( webcams[i].first.c_str(), g_WebcamDeviceSymLink ) == 0 )
|
||||
selection = static_cast<int>(i + 1);
|
||||
}
|
||||
SendMessage( GetDlgItem( hDlg, IDC_WEBCAM_DEVICE ), CB_SETCURSEL, selection, 0 );
|
||||
}
|
||||
|
||||
// Populate position combo
|
||||
{
|
||||
const wchar_t* positions[] = { L"Top-left", L"Top-right", L"Bottom-left", L"Bottom-right" };
|
||||
for( int i = 0; i < 4; i++ )
|
||||
SendMessage( GetDlgItem( hDlg, IDC_WEBCAM_POSITION ), CB_ADDSTRING, 0, reinterpret_cast<LPARAM>(positions[i]) );
|
||||
SendMessage( GetDlgItem( hDlg, IDC_WEBCAM_POSITION ), CB_SETCURSEL, g_WebcamPosition, 0 );
|
||||
}
|
||||
|
||||
// Populate size combo
|
||||
{
|
||||
const wchar_t* sizes[] = { L"Small", L"Medium", L"Large", L"X-Large", L"Full screen" };
|
||||
for( int i = 0; i < 5; i++ )
|
||||
SendMessage( GetDlgItem( hDlg, IDC_WEBCAM_SIZE ), CB_ADDSTRING, 0, reinterpret_cast<LPARAM>(sizes[i]) );
|
||||
SendMessage( GetDlgItem( hDlg, IDC_WEBCAM_SIZE ), CB_SETCURSEL, g_WebcamSize, 0 );
|
||||
}
|
||||
|
||||
// Populate shape combo
|
||||
{
|
||||
const wchar_t* shapes[] = { L"Rectangle", L"Rounded Rectangle", L"Rounded Square", L"Circle" };
|
||||
for( int i = 0; i < 4; i++ )
|
||||
SendMessage( GetDlgItem( hDlg, IDC_WEBCAM_SHAPE ), CB_ADDSTRING, 0, reinterpret_cast<LPARAM>(shapes[i]) );
|
||||
SendMessage( GetDlgItem( hDlg, IDC_WEBCAM_SHAPE ), CB_SETCURSEL, g_WebcamShape, 0 );
|
||||
}
|
||||
|
||||
// Populate background mode combo
|
||||
{
|
||||
HWND hBgMode = GetDlgItem( hDlg, IDC_WEBCAM_BG_MODE );
|
||||
SendMessage( hBgMode, CB_ADDSTRING, 0, reinterpret_cast<LPARAM>(L"None") );
|
||||
SendMessage( hBgMode, CB_ADDSTRING, 0, reinterpret_cast<LPARAM>(L"Blur") );
|
||||
SendMessage( hBgMode, CB_ADDSTRING, 0, reinterpret_cast<LPARAM>(L"Image") );
|
||||
SendMessage( hBgMode, CB_SETCURSEL, g_WebcamBackgroundMode, 0 );
|
||||
SetDlgItemText( hDlg, IDC_WEBCAM_BG_IMAGE, g_WebcamBackgroundImage );
|
||||
}
|
||||
|
||||
// Initialize brightness slider
|
||||
{
|
||||
HWND hSlider = GetDlgItem( hDlg, IDC_WEBCAM_BRIGHTNESS_SLIDER );
|
||||
SendMessage( hSlider, TBM_SETRANGE, FALSE, MAKELONG( 0, 100 ) );
|
||||
SendMessage( hSlider, TBM_SETPOS, TRUE, g_WebcamBrightness );
|
||||
|
||||
// Place the built-in drag tooltip (TBS_TOOLTIPS) beneath the slider.
|
||||
SendMessage( hSlider, TBM_SETTIPSIDE, TBTS_BOTTOM, 0 );
|
||||
|
||||
// Hover tooltip that displays the current brightness when not dragging.
|
||||
// TTF_SUBCLASS lets the tooltip detect hover itself; LPSTR_TEXTCALLBACK
|
||||
// makes us supply the current value via TTN_GETDISPINFO.
|
||||
s_hBrightnessTooltip = CreateWindowEx(
|
||||
WS_EX_TOPMOST, TOOLTIPS_CLASS, nullptr,
|
||||
WS_POPUP | TTS_ALWAYSTIP | TTS_NOPREFIX,
|
||||
CW_USEDEFAULT, CW_USEDEFAULT, CW_USEDEFAULT, CW_USEDEFAULT,
|
||||
hDlg, nullptr, g_hInstance, nullptr );
|
||||
if( s_hBrightnessTooltip )
|
||||
{
|
||||
TOOLINFO ti = { sizeof(TOOLINFO) };
|
||||
ti.uFlags = TTF_IDISHWND | TTF_SUBCLASS;
|
||||
ti.hwnd = hDlg;
|
||||
ti.uId = reinterpret_cast<UINT_PTR>(hSlider);
|
||||
ti.lpszText = LPSTR_TEXTCALLBACK;
|
||||
SendMessage( s_hBrightnessTooltip, TTM_ADDTOOL, 0, reinterpret_cast<LPARAM>(&ti) );
|
||||
}
|
||||
}
|
||||
|
||||
// Set initial enabled state
|
||||
{
|
||||
bool isFullScreen = g_WebcamSize == 4;
|
||||
EnableWindow( GetDlgItem( hDlg, IDC_WEBCAM_SHAPE_LABEL ), !isFullScreen );
|
||||
EnableWindow( GetDlgItem( hDlg, IDC_WEBCAM_SHAPE ), !isFullScreen );
|
||||
bool isImageMode = g_WebcamBackgroundMode == 2;
|
||||
EnableWindow( GetDlgItem( hDlg, IDC_WEBCAM_BG_IMAGE ), isImageMode );
|
||||
EnableWindow( GetDlgItem( hDlg, IDC_WEBCAM_BG_BROWSE ), isImageMode );
|
||||
}
|
||||
|
||||
// DPI scaling
|
||||
currentDpi = GetDpiForWindowHelper( hDlg );
|
||||
if( currentDpi != DPI_BASELINE )
|
||||
ScaleDialogForDpi( hDlg, currentDpi, DPI_BASELINE );
|
||||
|
||||
ApplyDarkModeToDialog( hDlg );
|
||||
return TRUE;
|
||||
}
|
||||
|
||||
case WM_DPICHANGED:
|
||||
HandleDialogDpiChange( hDlg, wParam, lParam, currentDpi );
|
||||
return TRUE;
|
||||
|
||||
case WM_ERASEBKGND:
|
||||
if( IsDarkModeEnabled() )
|
||||
{
|
||||
HDC hdc = reinterpret_cast<HDC>(wParam);
|
||||
RECT rc;
|
||||
GetClientRect( hDlg, &rc );
|
||||
FillRect( hdc, &rc, GetDarkModeBrush() );
|
||||
return TRUE;
|
||||
}
|
||||
break;
|
||||
|
||||
case WM_CTLCOLORDLG:
|
||||
case WM_CTLCOLORSTATIC:
|
||||
case WM_CTLCOLORBTN:
|
||||
case WM_CTLCOLOREDIT:
|
||||
case WM_CTLCOLORLISTBOX:
|
||||
{
|
||||
HDC hdc = reinterpret_cast<HDC>(wParam);
|
||||
HWND hCtrl = reinterpret_cast<HWND>(lParam);
|
||||
HBRUSH hBrush = HandleDarkModeCtlColor( hdc, hCtrl, message );
|
||||
if( hBrush )
|
||||
return reinterpret_cast<INT_PTR>(hBrush);
|
||||
break;
|
||||
}
|
||||
|
||||
case WM_NOTIFY:
|
||||
{
|
||||
LPNMHDR pnmh = reinterpret_cast<LPNMHDR>(lParam);
|
||||
if( s_hBrightnessTooltip && pnmh->hwndFrom == s_hBrightnessTooltip )
|
||||
{
|
||||
if( pnmh->code == TTN_GETDISPINFO )
|
||||
{
|
||||
LPNMTTDISPINFO pdi = reinterpret_cast<LPNMTTDISPINFO>(lParam);
|
||||
static wchar_t buf[16];
|
||||
int pos = static_cast<int>(SendMessage(
|
||||
GetDlgItem( hDlg, IDC_WEBCAM_BRIGHTNESS_SLIDER ), TBM_GETPOS, 0, 0 ));
|
||||
_snwprintf_s( buf, _countof(buf), _TRUNCATE, L"%d", pos );
|
||||
pdi->lpszText = buf;
|
||||
return TRUE;
|
||||
}
|
||||
if( pnmh->code == TTN_SHOW )
|
||||
{
|
||||
// Reposition the tooltip directly beneath the slider, centered on the cursor.
|
||||
HWND hSlider = GetDlgItem( hDlg, IDC_WEBCAM_BRIGHTNESS_SLIDER );
|
||||
RECT rcSlider, rcTip;
|
||||
GetWindowRect( hSlider, &rcSlider );
|
||||
GetWindowRect( s_hBrightnessTooltip, &rcTip );
|
||||
POINT pt;
|
||||
GetCursorPos( &pt );
|
||||
int tipW = rcTip.right - rcTip.left;
|
||||
SetWindowPos( s_hBrightnessTooltip, HWND_TOPMOST,
|
||||
pt.x - tipW / 2, rcSlider.bottom + 4, 0, 0,
|
||||
SWP_NOSIZE | SWP_NOACTIVATE );
|
||||
return TRUE;
|
||||
}
|
||||
}
|
||||
break;
|
||||
}
|
||||
|
||||
case WM_DESTROY:
|
||||
if( s_hBrightnessTooltip )
|
||||
{
|
||||
DestroyWindow( s_hBrightnessTooltip );
|
||||
s_hBrightnessTooltip = nullptr;
|
||||
}
|
||||
break;
|
||||
|
||||
case WM_COMMAND:
|
||||
// Handle size combo change — disable shape when Full screen
|
||||
if( HIWORD(wParam) == CBN_SELCHANGE && LOWORD(wParam) == IDC_WEBCAM_SIZE )
|
||||
{
|
||||
bool isFullScreen = SendMessage( GetDlgItem( hDlg, IDC_WEBCAM_SIZE ), CB_GETCURSEL, 0, 0 ) == 4;
|
||||
EnableWindow( GetDlgItem( hDlg, IDC_WEBCAM_SHAPE_LABEL ), !isFullScreen );
|
||||
EnableWindow( GetDlgItem( hDlg, IDC_WEBCAM_SHAPE ), !isFullScreen );
|
||||
}
|
||||
// Handle background mode combo change — show/hide browse controls
|
||||
if( HIWORD(wParam) == CBN_SELCHANGE && LOWORD(wParam) == IDC_WEBCAM_BG_MODE )
|
||||
{
|
||||
bool isImageMode = SendMessage( GetDlgItem( hDlg, IDC_WEBCAM_BG_MODE ), CB_GETCURSEL, 0, 0 ) == 2;
|
||||
EnableWindow( GetDlgItem( hDlg, IDC_WEBCAM_BG_IMAGE ), isImageMode );
|
||||
EnableWindow( GetDlgItem( hDlg, IDC_WEBCAM_BG_BROWSE ), isImageMode );
|
||||
}
|
||||
|
||||
switch( LOWORD(wParam) ) {
|
||||
case IDC_WEBCAM_BG_BROWSE:
|
||||
{
|
||||
auto openDialog = wil::CoCreateInstance<IFileOpenDialog>( CLSID_FileOpenDialog );
|
||||
|
||||
FILEOPENDIALOGOPTIONS options;
|
||||
if( SUCCEEDED( openDialog->GetOptions( &options ) ) )
|
||||
openDialog->SetOptions( options | FOS_FORCEFILESYSTEM );
|
||||
|
||||
COMDLG_FILTERSPEC fileTypes[] = {
|
||||
{ L"Bitmap Files (*.bmp;*.dib)", L"*.bmp;*.dib" },
|
||||
{ L"PNG (*.png)", L"*.png" },
|
||||
{ L"JPEG (*.jpg;*.jpeg;*.jpe;*.jfif)", L"*.jpg;*.jpeg;*.jpe;*.jfif" },
|
||||
{ L"GIF (*.gif)", L"*.gif" },
|
||||
{ L"All Picture Files", L"*.bmp;*.dib;*.png;*.jpg;*.jpeg;*.jpe;*.jfif;*.gif" },
|
||||
{ L"All Files", L"*.*" }
|
||||
};
|
||||
openDialog->SetFileTypes( _countof( fileTypes ), fileTypes );
|
||||
openDialog->SetFileTypeIndex( 5 );
|
||||
openDialog->SetTitle( L"ZoomIt: Specify Background Image..." );
|
||||
|
||||
TCHAR bgFilePath[MAX_PATH], bgInitDir[MAX_PATH];
|
||||
GetDlgItemText( hDlg, IDC_WEBCAM_BG_IMAGE, bgFilePath, _countof( bgFilePath ) );
|
||||
if( _tcsrchr( bgFilePath, '\\' ) )
|
||||
{
|
||||
_tcscpy( bgInitDir, bgFilePath );
|
||||
*( _tcsrchr( bgInitDir, '\\' ) + 1 ) = 0;
|
||||
}
|
||||
else
|
||||
{
|
||||
_tcscpy( bgFilePath, L"%USERPROFILE%\\Pictures" );
|
||||
ExpandEnvironmentStrings( bgFilePath, bgInitDir, _countof( bgInitDir ) );
|
||||
}
|
||||
wil::com_ptr<IShellItem> folderItem;
|
||||
if( SUCCEEDED( SHCreateItemFromParsingName( bgInitDir, nullptr, IID_PPV_ARGS( &folderItem ) ) ) )
|
||||
openDialog->SetFolder( folderItem.get() );
|
||||
|
||||
OpenSaveDialogEvents* pEvents = new OpenSaveDialogEvents(false);
|
||||
DWORD dwCookie = 0;
|
||||
openDialog->Advise( pEvents, &dwCookie );
|
||||
|
||||
if( SUCCEEDED( openDialog->Show( hDlg ) ) )
|
||||
{
|
||||
wil::com_ptr<IShellItem> resultItem;
|
||||
if( SUCCEEDED( openDialog->GetResult( &resultItem ) ) )
|
||||
{
|
||||
wil::unique_cotaskmem_string pathStr;
|
||||
if( SUCCEEDED( resultItem->GetDisplayName( SIGDN_FILESYSPATH, &pathStr ) ) )
|
||||
SetDlgItemText( hDlg, IDC_WEBCAM_BG_IMAGE, pathStr.get() );
|
||||
}
|
||||
}
|
||||
|
||||
openDialog->Unadvise( dwCookie );
|
||||
pEvents->Release();
|
||||
break;
|
||||
}
|
||||
|
||||
case IDOK:
|
||||
g_WebcamPosition = static_cast<DWORD>(SendMessage( GetDlgItem( hDlg, IDC_WEBCAM_POSITION ), CB_GETCURSEL, 0, 0 ));
|
||||
g_WebcamSize = static_cast<DWORD>(SendMessage( GetDlgItem( hDlg, IDC_WEBCAM_SIZE ), CB_GETCURSEL, 0, 0 ));
|
||||
g_WebcamShape = static_cast<DWORD>(SendMessage( GetDlgItem( hDlg, IDC_WEBCAM_SHAPE ), CB_GETCURSEL, 0, 0 ));
|
||||
g_WebcamBackgroundMode = static_cast<DWORD>(SendMessage( GetDlgItem( hDlg, IDC_WEBCAM_BG_MODE ), CB_GETCURSEL, 0, 0 ));
|
||||
GetDlgItemText( hDlg, IDC_WEBCAM_BG_IMAGE, g_WebcamBackgroundImage, MAX_PATH );
|
||||
g_WebcamBrightness = static_cast<DWORD>(SendMessage( GetDlgItem( hDlg, IDC_WEBCAM_BRIGHTNESS_SLIDER ), TBM_GETPOS, 0, 0 ));
|
||||
{
|
||||
int wcIndex = static_cast<int>(SendMessage( GetDlgItem( hDlg, IDC_WEBCAM_DEVICE ), CB_GETCURSEL, 0, 0 ));
|
||||
_tcscpy( g_WebcamDeviceSymLink, wcIndex == 0 ? L"" : webcams[static_cast<size_t>(wcIndex) - 1].first.c_str() );
|
||||
}
|
||||
reg.WriteRegSettings( RegSettings );
|
||||
EndDialog( hDlg, 0 );
|
||||
break;
|
||||
|
||||
case IDCANCEL:
|
||||
EndDialog( hDlg, 0 );
|
||||
return TRUE;
|
||||
}
|
||||
break;
|
||||
}
|
||||
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
|
||||
//----------------------------------------------------------------------------
|
||||
//
|
||||
// OptionsTabProc
|
||||
@@ -2972,52 +3288,33 @@ INT_PTR CALLBACK OptionsTabProc( HWND hDlg, UINT message,
|
||||
// Enable/disable audio controls based on selection (GIF has no audio)
|
||||
EnableWindow(GetDlgItem(hDlg, IDC_CAPTURE_SYSTEM_AUDIO), !isGifSelected);
|
||||
EnableWindow(GetDlgItem(hDlg, IDC_CAPTURE_AUDIO), !isGifSelected);
|
||||
EnableWindow(GetDlgItem(hDlg, IDC_NOISE_CANCELLATION), !isGifSelected);
|
||||
EnableWindow(GetDlgItem(hDlg, IDC_MIC_MONO_MIX), !isGifSelected);
|
||||
EnableWindow(GetDlgItem(hDlg, IDC_MICROPHONE_LABEL), !isGifSelected);
|
||||
EnableWindow(GetDlgItem(hDlg, IDC_MICROPHONE), !isGifSelected);
|
||||
|
||||
// Enable/disable webcam controls (webcam overlay is MP4-only).
|
||||
// Also keep everything disabled if no webcam is present.
|
||||
bool hasWebcam = SendMessage(GetDlgItem(hDlg, IDC_WEBCAM_DEVICE), CB_GETCOUNT, 0, 0) > 1;
|
||||
bool webcamEnabled = hasWebcam && !isGifSelected && IsDlgButtonChecked(hDlg, IDC_WEBCAM_OVERLAY) == BST_CHECKED;
|
||||
EnableWindow(GetDlgItem(hDlg, IDC_WEBCAM_OVERLAY), hasWebcam && !isGifSelected);
|
||||
EnableWindow(GetDlgItem(hDlg, IDC_WEBCAM_DEVICE_LABEL), webcamEnabled);
|
||||
EnableWindow(GetDlgItem(hDlg, IDC_WEBCAM_DEVICE), webcamEnabled);
|
||||
EnableWindow(GetDlgItem(hDlg, IDC_WEBCAM_POSITION_LABEL), webcamEnabled);
|
||||
EnableWindow(GetDlgItem(hDlg, IDC_WEBCAM_POSITION), webcamEnabled);
|
||||
EnableWindow(GetDlgItem(hDlg, IDC_WEBCAM_SIZE_LABEL), webcamEnabled);
|
||||
EnableWindow(GetDlgItem(hDlg, IDC_WEBCAM_SIZE), webcamEnabled);
|
||||
{
|
||||
bool isFullScreen = webcamEnabled && SendMessage(GetDlgItem(hDlg, IDC_WEBCAM_SIZE), CB_GETCURSEL, 0, 0) == 4;
|
||||
EnableWindow(GetDlgItem(hDlg, IDC_WEBCAM_SHAPE_LABEL), webcamEnabled && !isFullScreen);
|
||||
EnableWindow(GetDlgItem(hDlg, IDC_WEBCAM_SHAPE), webcamEnabled && !isFullScreen);
|
||||
}
|
||||
// Enable/disable webcam overlay and settings button (webcam overlay is MP4-only).
|
||||
EnableWindow(GetDlgItem(hDlg, IDC_WEBCAM_OVERLAY), !isGifSelected);
|
||||
EnableWindow(GetDlgItem(hDlg, IDC_WEBCAM_SETTINGS), !isGifSelected && IsDlgButtonChecked(hDlg, IDC_WEBCAM_OVERLAY) == BST_CHECKED);
|
||||
}
|
||||
}
|
||||
if (HIWORD(wParam) == BN_CLICKED && LOWORD(wParam) == IDC_WEBCAM_OVERLAY) {
|
||||
bool isGif = (g_RecordingFormat == RecordingFormat::GIF);
|
||||
bool webcamEnabled = !isGif && IsDlgButtonChecked(hDlg, IDC_WEBCAM_OVERLAY) == BST_CHECKED;
|
||||
EnableWindow(GetDlgItem(hDlg, IDC_WEBCAM_DEVICE_LABEL), webcamEnabled);
|
||||
EnableWindow(GetDlgItem(hDlg, IDC_WEBCAM_DEVICE), webcamEnabled);
|
||||
EnableWindow(GetDlgItem(hDlg, IDC_WEBCAM_POSITION_LABEL), webcamEnabled);
|
||||
EnableWindow(GetDlgItem(hDlg, IDC_WEBCAM_POSITION), webcamEnabled);
|
||||
EnableWindow(GetDlgItem(hDlg, IDC_WEBCAM_SIZE_LABEL), webcamEnabled);
|
||||
EnableWindow(GetDlgItem(hDlg, IDC_WEBCAM_SIZE), webcamEnabled);
|
||||
{
|
||||
bool isFullScreen = webcamEnabled && SendMessage(GetDlgItem(hDlg, IDC_WEBCAM_SIZE), CB_GETCURSEL, 0, 0) == 4;
|
||||
EnableWindow(GetDlgItem(hDlg, IDC_WEBCAM_SHAPE_LABEL), webcamEnabled && !isFullScreen);
|
||||
EnableWindow(GetDlgItem(hDlg, IDC_WEBCAM_SHAPE), webcamEnabled && !isFullScreen);
|
||||
}
|
||||
bool webcamEnabled = IsDlgButtonChecked(hDlg, IDC_WEBCAM_OVERLAY) == BST_CHECKED;
|
||||
EnableWindow(GetDlgItem(hDlg, IDC_WEBCAM_SETTINGS), webcamEnabled);
|
||||
}
|
||||
// Handle webcam size combo change — disable shape when Full screen selected
|
||||
if (HIWORD(wParam) == CBN_SELCHANGE && LOWORD(wParam) == IDC_WEBCAM_SIZE) {
|
||||
bool isGif = (g_RecordingFormat == RecordingFormat::GIF);
|
||||
bool webcamEnabled = !isGif && IsDlgButtonChecked(hDlg, IDC_WEBCAM_OVERLAY) == BST_CHECKED;
|
||||
bool isFullScreen = SendMessage(GetDlgItem(hDlg, IDC_WEBCAM_SIZE), CB_GETCURSEL, 0, 0) == 4;
|
||||
EnableWindow(GetDlgItem(hDlg, IDC_WEBCAM_SHAPE_LABEL), webcamEnabled && !isFullScreen);
|
||||
EnableWindow(GetDlgItem(hDlg, IDC_WEBCAM_SHAPE), webcamEnabled && !isFullScreen);
|
||||
if (HIWORD(wParam) == BN_CLICKED && LOWORD(wParam) == IDC_CAPTURE_AUDIO) {
|
||||
bool micEnabled = IsDlgButtonChecked(hDlg, IDC_CAPTURE_AUDIO) == BST_CHECKED;
|
||||
EnableWindow(GetDlgItem(hDlg, IDC_NOISE_CANCELLATION), micEnabled);
|
||||
EnableWindow(GetDlgItem(hDlg, IDC_MIC_MONO_MIX), micEnabled);
|
||||
EnableWindow(GetDlgItem(hDlg, IDC_MICROPHONE_LABEL), micEnabled);
|
||||
EnableWindow(GetDlgItem(hDlg, IDC_MICROPHONE), micEnabled);
|
||||
|
||||
}
|
||||
|
||||
switch ( LOWORD( wParam )) {
|
||||
case IDC_WEBCAM_SETTINGS:
|
||||
DialogBox( g_hInstance, L"WEBCAM_SETTINGS", hDlg, WebcamSettingsProc );
|
||||
break;
|
||||
case IDC_ADVANCED_BREAK:
|
||||
DialogBox( g_hInstance, L"ADVANCED_BREAK", hDlg, AdvancedBreakProc );
|
||||
break;
|
||||
@@ -4577,7 +4874,6 @@ INT_PTR CALLBACK OptionsProc( HWND hDlg, UINT message,
|
||||
DWORD newDrawToggleKey, newDrawToggleMod, newBreakToggleMod, newDemoTypeToggleMod, newRecordToggleMod, newSnipToggleMod, newSnipPanoramaToggleMod, newSnipOcrToggleMod;
|
||||
DWORD newLiveZoomToggleKey, newLiveZoomToggleMod;
|
||||
static std::vector<std::pair<std::wstring, std::wstring>> microphones;
|
||||
static std::vector<std::pair<std::wstring, std::wstring>> webcams;
|
||||
|
||||
auto CleanupFonts = [&]()
|
||||
{
|
||||
@@ -4849,6 +5145,9 @@ INT_PTR CALLBACK OptionsProc( HWND hDlg, UINT message,
|
||||
CheckDlgButton( g_OptionsTabs[RECORD_PAGE].hPage, IDC_MIC_MONO_MIX,
|
||||
g_MicMonoMix ? BST_CHECKED: BST_UNCHECKED );
|
||||
|
||||
CheckDlgButton( g_OptionsTabs[RECORD_PAGE].hPage, IDC_NOISE_CANCELLATION,
|
||||
g_NoiseCancellation ? BST_CHECKED: BST_UNCHECKED );
|
||||
|
||||
CheckDlgButton( g_OptionsTabs[RECORD_PAGE].hPage, IDC_RECORD_ASPECT_RATIO,
|
||||
g_RecordAspectRatio ? BST_CHECKED: BST_UNCHECKED );
|
||||
|
||||
@@ -4924,101 +5223,19 @@ INT_PTR CALLBACK OptionsProc( HWND hDlg, UINT message,
|
||||
bool isGifSelected = (g_RecordingFormat == RecordingFormat::GIF);
|
||||
EnableWindow(GetDlgItem(g_OptionsTabs[RECORD_PAGE].hPage, IDC_CAPTURE_SYSTEM_AUDIO), !isGifSelected);
|
||||
EnableWindow(GetDlgItem(g_OptionsTabs[RECORD_PAGE].hPage, IDC_CAPTURE_AUDIO), !isGifSelected);
|
||||
EnableWindow(GetDlgItem(g_OptionsTabs[RECORD_PAGE].hPage, IDC_MICROPHONE_LABEL), !isGifSelected);
|
||||
EnableWindow(GetDlgItem(g_OptionsTabs[RECORD_PAGE].hPage, IDC_MICROPHONE), !isGifSelected);
|
||||
EnableWindow(GetDlgItem(g_OptionsTabs[RECORD_PAGE].hPage, IDC_NOISE_CANCELLATION), !isGifSelected && g_CaptureAudio);
|
||||
EnableWindow(GetDlgItem(g_OptionsTabs[RECORD_PAGE].hPage, IDC_MIC_MONO_MIX), !isGifSelected && g_CaptureAudio);
|
||||
EnableWindow(GetDlgItem(g_OptionsTabs[RECORD_PAGE].hPage, IDC_MICROPHONE_LABEL), !isGifSelected && g_CaptureAudio);
|
||||
EnableWindow(GetDlgItem(g_OptionsTabs[RECORD_PAGE].hPage, IDC_MICROPHONE), !isGifSelected && g_CaptureAudio);
|
||||
|
||||
// Webcam overlay controls
|
||||
CheckDlgButton( g_OptionsTabs[RECORD_PAGE].hPage, IDC_WEBCAM_OVERLAY,
|
||||
g_WebcamOverlay ? BST_CHECKED : BST_UNCHECKED );
|
||||
|
||||
// Enumerate webcam devices
|
||||
webcams.clear();
|
||||
// Set initial enabled state for webcam overlay and settings button
|
||||
{
|
||||
MFStartup( MF_VERSION, MFSTARTUP_LITE );
|
||||
IMFAttributes* pAttributes = nullptr;
|
||||
if( SUCCEEDED( MFCreateAttributes( &pAttributes, 1 ) ) )
|
||||
{
|
||||
pAttributes->SetGUID( MF_DEVSOURCE_ATTRIBUTE_SOURCE_TYPE, MF_DEVSOURCE_ATTRIBUTE_SOURCE_TYPE_VIDCAP_GUID );
|
||||
IMFActivate** ppDevices = nullptr;
|
||||
UINT32 count = 0;
|
||||
if( SUCCEEDED( MFEnumDeviceSources( pAttributes, &ppDevices, &count ) ) )
|
||||
{
|
||||
for( UINT32 i = 0; i < count; i++ )
|
||||
{
|
||||
WCHAR* symLink = nullptr;
|
||||
UINT32 symLinkLen = 0;
|
||||
WCHAR* friendlyName = nullptr;
|
||||
UINT32 nameLen = 0;
|
||||
if( SUCCEEDED( ppDevices[i]->GetAllocatedString( MF_DEVSOURCE_ATTRIBUTE_SOURCE_TYPE_VIDCAP_SYMBOLIC_LINK, &symLink, &symLinkLen ) ) &&
|
||||
SUCCEEDED( ppDevices[i]->GetAllocatedString( MF_DEVSOURCE_ATTRIBUTE_FRIENDLY_NAME, &friendlyName, &nameLen ) ) )
|
||||
{
|
||||
webcams.emplace_back( symLink, friendlyName );
|
||||
}
|
||||
if( symLink ) CoTaskMemFree( symLink );
|
||||
if( friendlyName ) CoTaskMemFree( friendlyName );
|
||||
ppDevices[i]->Release();
|
||||
}
|
||||
CoTaskMemFree( ppDevices );
|
||||
}
|
||||
pAttributes->Release();
|
||||
}
|
||||
MFShutdown();
|
||||
}
|
||||
|
||||
// Add webcam devices to combo box
|
||||
SendMessage( GetDlgItem( g_OptionsTabs[RECORD_PAGE].hPage, IDC_WEBCAM_DEVICE ), static_cast<UINT>(CB_ADDSTRING), static_cast<WPARAM>(0), reinterpret_cast<LPARAM>(L"Default") );
|
||||
selection = 0;
|
||||
for( size_t i = 0; i < webcams.size(); i++ )
|
||||
{
|
||||
SendMessage( GetDlgItem( g_OptionsTabs[RECORD_PAGE].hPage, IDC_WEBCAM_DEVICE ), static_cast<UINT>(CB_ADDSTRING), static_cast<WPARAM>(0), reinterpret_cast<LPARAM>(webcams[i].second.c_str()) );
|
||||
if( selection == 0 && wcscmp( webcams[i].first.c_str(), g_WebcamDeviceSymLink ) == 0 )
|
||||
{
|
||||
selection = i + 1;
|
||||
}
|
||||
}
|
||||
SendMessage( GetDlgItem( g_OptionsTabs[RECORD_PAGE].hPage, IDC_WEBCAM_DEVICE ), CB_SETCURSEL, static_cast<WPARAM>(selection), static_cast<LPARAM>(0) );
|
||||
|
||||
// Webcam position combo
|
||||
{
|
||||
const wchar_t* positions[] = { L"Top left", L"Top right", L"Bottom left", L"Bottom right" };
|
||||
for( int i = 0; i < 4; i++ )
|
||||
SendMessage( GetDlgItem( g_OptionsTabs[RECORD_PAGE].hPage, IDC_WEBCAM_POSITION ), static_cast<UINT>(CB_ADDSTRING), static_cast<WPARAM>(0), reinterpret_cast<LPARAM>(positions[i]) );
|
||||
SendMessage( GetDlgItem( g_OptionsTabs[RECORD_PAGE].hPage, IDC_WEBCAM_POSITION ), CB_SETCURSEL, static_cast<WPARAM>(g_WebcamPosition), static_cast<LPARAM>(0) );
|
||||
}
|
||||
|
||||
// Webcam size combo
|
||||
{
|
||||
const wchar_t* sizes[] = { L"Small", L"Medium", L"Large", L"X-Large", L"Full screen" };
|
||||
for( int i = 0; i < 5; i++ )
|
||||
SendMessage( GetDlgItem( g_OptionsTabs[RECORD_PAGE].hPage, IDC_WEBCAM_SIZE ), static_cast<UINT>(CB_ADDSTRING), static_cast<WPARAM>(0), reinterpret_cast<LPARAM>(sizes[i]) );
|
||||
SendMessage( GetDlgItem( g_OptionsTabs[RECORD_PAGE].hPage, IDC_WEBCAM_SIZE ), CB_SETCURSEL, static_cast<WPARAM>(g_WebcamSize), static_cast<LPARAM>(0) );
|
||||
}
|
||||
|
||||
// Webcam shape combo
|
||||
{
|
||||
const wchar_t* shapes[] = { L"Rectangle", L"Rounded rectangle", L"Rounded square", L"Circle" };
|
||||
for( int i = 0; i < 4; i++ )
|
||||
SendMessage( GetDlgItem( g_OptionsTabs[RECORD_PAGE].hPage, IDC_WEBCAM_SHAPE ), static_cast<UINT>(CB_ADDSTRING), static_cast<WPARAM>(0), reinterpret_cast<LPARAM>(shapes[i]) );
|
||||
SendMessage( GetDlgItem( g_OptionsTabs[RECORD_PAGE].hPage, IDC_WEBCAM_SHAPE ), CB_SETCURSEL, static_cast<WPARAM>(g_WebcamShape), static_cast<LPARAM>(0) );
|
||||
}
|
||||
|
||||
// Set initial enabled state for webcam controls.
|
||||
// Disable everything if no webcam is detected on the system.
|
||||
{
|
||||
bool hasWebcam = !webcams.empty();
|
||||
bool webcamEnabled = hasWebcam && !isGifSelected && g_WebcamOverlay;
|
||||
EnableWindow(GetDlgItem(g_OptionsTabs[RECORD_PAGE].hPage, IDC_WEBCAM_OVERLAY), hasWebcam && !isGifSelected);
|
||||
EnableWindow(GetDlgItem(g_OptionsTabs[RECORD_PAGE].hPage, IDC_WEBCAM_DEVICE_LABEL), webcamEnabled);
|
||||
EnableWindow(GetDlgItem(g_OptionsTabs[RECORD_PAGE].hPage, IDC_WEBCAM_DEVICE), webcamEnabled);
|
||||
EnableWindow(GetDlgItem(g_OptionsTabs[RECORD_PAGE].hPage, IDC_WEBCAM_POSITION_LABEL), webcamEnabled);
|
||||
EnableWindow(GetDlgItem(g_OptionsTabs[RECORD_PAGE].hPage, IDC_WEBCAM_POSITION), webcamEnabled);
|
||||
EnableWindow(GetDlgItem(g_OptionsTabs[RECORD_PAGE].hPage, IDC_WEBCAM_SIZE_LABEL), webcamEnabled);
|
||||
EnableWindow(GetDlgItem(g_OptionsTabs[RECORD_PAGE].hPage, IDC_WEBCAM_SIZE), webcamEnabled);
|
||||
{
|
||||
bool isFullScreen = webcamEnabled && g_WebcamSize == 4;
|
||||
EnableWindow(GetDlgItem(g_OptionsTabs[RECORD_PAGE].hPage, IDC_WEBCAM_SHAPE_LABEL), webcamEnabled && !isFullScreen);
|
||||
EnableWindow(GetDlgItem(g_OptionsTabs[RECORD_PAGE].hPage, IDC_WEBCAM_SHAPE), webcamEnabled && !isFullScreen);
|
||||
}
|
||||
EnableWindow(GetDlgItem(g_OptionsTabs[RECORD_PAGE].hPage, IDC_WEBCAM_OVERLAY), !isGifSelected);
|
||||
EnableWindow(GetDlgItem(g_OptionsTabs[RECORD_PAGE].hPage, IDC_WEBCAM_SETTINGS), !isGifSelected && g_WebcamOverlay);
|
||||
}
|
||||
|
||||
if( GetFileAttributes( g_DemoTypeFile ) == -1 )
|
||||
@@ -5372,6 +5589,7 @@ INT_PTR CALLBACK OptionsProc( HWND hDlg, UINT message,
|
||||
g_CaptureSystemAudio = IsDlgButtonChecked(g_OptionsTabs[RECORD_PAGE].hPage, IDC_CAPTURE_SYSTEM_AUDIO) == BST_CHECKED;
|
||||
g_CaptureAudio = IsDlgButtonChecked(g_OptionsTabs[RECORD_PAGE].hPage, IDC_CAPTURE_AUDIO) == BST_CHECKED;
|
||||
g_MicMonoMix = IsDlgButtonChecked(g_OptionsTabs[RECORD_PAGE].hPage, IDC_MIC_MONO_MIX) == BST_CHECKED;
|
||||
g_NoiseCancellation = IsDlgButtonChecked(g_OptionsTabs[RECORD_PAGE].hPage, IDC_NOISE_CANCELLATION) == BST_CHECKED;
|
||||
g_RecordAspectRatio = IsDlgButtonChecked(g_OptionsTabs[RECORD_PAGE].hPage, IDC_RECORD_ASPECT_RATIO) == BST_CHECKED;
|
||||
GetDlgItemText( g_OptionsTabs[BREAK_PAGE].hPage, IDC_TIMER, text, 3 );
|
||||
text[2] = 0;
|
||||
@@ -5385,15 +5603,8 @@ INT_PTR CALLBACK OptionsProc( HWND hDlg, UINT message,
|
||||
int index = static_cast<int>(SendMessage( GetDlgItem( g_OptionsTabs[RECORD_PAGE].hPage, IDC_MICROPHONE ), static_cast<UINT>(CB_GETCURSEL), static_cast<WPARAM>(0), static_cast<LPARAM>(0) ));
|
||||
_tcscpy( g_MicrophoneDeviceId, index == 0 ? L"" : microphones[static_cast<size_t>(index) - 1].first.c_str() );
|
||||
|
||||
// Get the webcam settings
|
||||
// Get the webcam overlay setting (other webcam settings are saved in WebcamSettingsProc)
|
||||
g_WebcamOverlay = IsDlgButtonChecked(g_OptionsTabs[RECORD_PAGE].hPage, IDC_WEBCAM_OVERLAY) == BST_CHECKED;
|
||||
g_WebcamPosition = static_cast<DWORD>(SendMessage(GetDlgItem(g_OptionsTabs[RECORD_PAGE].hPage, IDC_WEBCAM_POSITION), CB_GETCURSEL, 0, 0));
|
||||
g_WebcamSize = static_cast<DWORD>(SendMessage(GetDlgItem(g_OptionsTabs[RECORD_PAGE].hPage, IDC_WEBCAM_SIZE), CB_GETCURSEL, 0, 0));
|
||||
g_WebcamShape = static_cast<DWORD>(SendMessage(GetDlgItem(g_OptionsTabs[RECORD_PAGE].hPage, IDC_WEBCAM_SHAPE), CB_GETCURSEL, 0, 0));
|
||||
{
|
||||
int wcIndex = static_cast<int>(SendMessage(GetDlgItem(g_OptionsTabs[RECORD_PAGE].hPage, IDC_WEBCAM_DEVICE), CB_GETCURSEL, 0, 0));
|
||||
_tcscpy( g_WebcamDeviceSymLink, wcIndex == 0 ? L"" : webcams[static_cast<size_t>(wcIndex) - 1].first.c_str() );
|
||||
}
|
||||
|
||||
if( newToggleKey && !RegisterHotKey( GetParent( hDlg ), ZOOM_HOTKEY, newToggleMod, newToggleKey & 0xFF )) {
|
||||
|
||||
@@ -5465,18 +5676,15 @@ INT_PTR CALLBACK OptionsProc( HWND hDlg, UINT message,
|
||||
break;
|
||||
|
||||
}
|
||||
else if( newRecordToggleKey ) {
|
||||
UINT cropMod = newRecordToggleMod ^ MOD_SHIFT;
|
||||
UINT windowMod = newRecordToggleMod ^ MOD_ALT;
|
||||
if (!RegisterHotKey(GetParent(hDlg), RECORD_HOTKEY, newRecordToggleMod | MOD_NOREPEAT, newRecordToggleKey & 0xFF) ||
|
||||
(cropMod != 0 && !RegisterHotKey(GetParent(hDlg), RECORD_CROP_HOTKEY, cropMod | MOD_NOREPEAT, newRecordToggleKey & 0xFF)) ||
|
||||
(windowMod != 0 && !RegisterHotKey(GetParent(hDlg), RECORD_WINDOW_HOTKEY, windowMod | MOD_NOREPEAT, newRecordToggleKey & 0xFF))) {
|
||||
else if( UINT cropMod = newRecordToggleMod ^ MOD_SHIFT, windowMod = newRecordToggleMod ^ MOD_ALT; newRecordToggleKey &&
|
||||
(!RegisterHotKey(GetParent(hDlg), RECORD_HOTKEY, newRecordToggleMod | MOD_NOREPEAT, newRecordToggleKey & 0xFF) ||
|
||||
(cropMod != 0 && !RegisterHotKey(GetParent(hDlg), RECORD_CROP_HOTKEY, cropMod | MOD_NOREPEAT, newRecordToggleKey & 0xFF)) ||
|
||||
(windowMod != 0 && !RegisterHotKey(GetParent(hDlg), RECORD_WINDOW_HOTKEY, windowMod | MOD_NOREPEAT, newRecordToggleKey & 0xFF)))) {
|
||||
|
||||
MessageBox(hDlg, L"The specified record hotkey is already in use.\nSelect a different record hotkey.",
|
||||
APPNAME, MB_ICONERROR);
|
||||
UnregisterAllHotkeys(GetParent(hDlg));
|
||||
break;
|
||||
}
|
||||
MessageBox(hDlg, L"The specified record hotkey is already in use.\nSelect a different record hotkey.",
|
||||
APPNAME, MB_ICONERROR);
|
||||
UnregisterAllHotkeys(GetParent(hDlg));
|
||||
break;
|
||||
} else {
|
||||
|
||||
g_BreakTimeout = newTimeout;
|
||||
@@ -6706,6 +6914,22 @@ winrt::fire_and_forget StartRecordingAsync( HWND hWnd, LPRECT rcCrop, HWND hWndR
|
||||
// Capture the UI thread context so we can resume on it for the save dialog
|
||||
winrt::apartment_context uiThread;
|
||||
|
||||
// Start audio initialization as early as possible. AudioGraph creation
|
||||
// and microphone device opening take ~1400 ms. By starting here, the
|
||||
// init runs in the background during D3D device creation, capture-item
|
||||
// creation, file I/O, and the entire VideoRecordingSession constructor
|
||||
// (webcam probe + warmup), giving it ~1400 ms of overlap — enough to
|
||||
// finish before StartAsync even needs the result.
|
||||
std::unique_ptr<AudioSampleGenerator> audioGenerator;
|
||||
winrt::Windows::Foundation::IAsyncAction audioInitAction{ nullptr };
|
||||
if ((g_RecordingFormat != RecordingFormat::GIF) && (g_CaptureAudio || g_CaptureSystemAudio))
|
||||
{
|
||||
audioGenerator = std::make_unique<AudioSampleGenerator>(
|
||||
g_CaptureAudio, g_CaptureSystemAudio, g_MicMonoMix, g_NoiseCancellation );
|
||||
audioInitAction = audioGenerator->InitializeAsync();
|
||||
_diagLog( L"audio InitializeAsync started (background)" );
|
||||
}
|
||||
|
||||
auto tempFolderPath = std::filesystem::temp_directory_path().wstring();
|
||||
auto tempFolder = co_await winrt::StorageFolder::GetFolderFromPathAsync( tempFolderPath );
|
||||
auto appFolder = co_await tempFolder.CreateFolderAsync( L"ZoomIt", winrt::CreationCollisionOption::OpenIfExists );
|
||||
@@ -6794,9 +7018,8 @@ winrt::fire_and_forget StartRecordingAsync( HWND hWnd, LPRECT rcCrop, HWND hWndR
|
||||
item,
|
||||
*rcCrop,
|
||||
g_RecordFrameRate,
|
||||
g_CaptureAudio,
|
||||
g_CaptureSystemAudio,
|
||||
g_MicMonoMix,
|
||||
std::move(audioGenerator),
|
||||
audioInitAction,
|
||||
stream );
|
||||
_diagLog( L"VideoRecordingSession::Create returned" );
|
||||
|
||||
|
||||
@@ -25,4 +25,7 @@ RCZOOMITSCR BINRES MOVEABLE PURE "ZoomItBreak64.scr"
|
||||
RCZOOMITSCR BINRES MOVEABLE PURE "ZoomItBreak64a.scr"
|
||||
#endif
|
||||
|
||||
// Embed the selfie segmentation ONNX model (platform-independent).
|
||||
RCSEGMODEL BINRES MOVEABLE PURE "selfie_segmentation.onnx"
|
||||
|
||||
CREATEPROCESS_MANIFEST_RESOURCE_ID RT_MANIFEST "ZoomIt.exe.manifest"
|
||||
|
||||
@@ -127,6 +127,16 @@
|
||||
#define IDC_WEBCAM_SHAPE_LABEL 1121
|
||||
#define IDC_TRIM_APPEND 1122
|
||||
#define IDC_RECORD_ASPECT_RATIO 1123
|
||||
#define IDC_WEBCAM_BACKGROUND_BLUR 1124
|
||||
#define IDC_WEBCAM_BG_LABEL 1125
|
||||
#define IDC_WEBCAM_BG_MODE 1126
|
||||
#define IDC_WEBCAM_BG_IMAGE 1127
|
||||
#define IDC_WEBCAM_BG_BROWSE 1128
|
||||
#define IDC_THIRDPARTY_NOTICES 1129
|
||||
#define IDC_WEBCAM_SETTINGS 1130
|
||||
#define IDC_WEBCAM_BRIGHTNESS_LABEL 1131
|
||||
#define IDC_WEBCAM_BRIGHTNESS_SLIDER 1132
|
||||
#define IDC_NOISE_CANCELLATION 1133
|
||||
#define IDC_SAVE 40002
|
||||
#define IDC_COPY 40004
|
||||
#define IDC_RECORD 40006
|
||||
@@ -142,7 +152,7 @@
|
||||
#ifndef APSTUDIO_READONLY_SYMBOLS
|
||||
#define _APS_NEXT_RESOURCE_VALUE 120
|
||||
#define _APS_NEXT_COMMAND_VALUE 40015
|
||||
#define _APS_NEXT_CONTROL_VALUE 1124
|
||||
#define _APS_NEXT_CONTROL_VALUE 1134
|
||||
#define _APS_NEXT_SYMED_VALUE 101
|
||||
#endif
|
||||
#endif
|
||||
|
||||
32
src/modules/ZoomIt/ZoomIt/rnnoise/COPYING
Normal file
32
src/modules/ZoomIt/ZoomIt/rnnoise/COPYING
Normal file
@@ -0,0 +1,32 @@
|
||||
Copyright (c) 2007-2017, 2024 Jean-Marc Valin
|
||||
Copyright (c) 2023 Amazon
|
||||
Copyright (c) 2017, Mozilla
|
||||
Copyright (c) 2005-2017, Xiph.Org Foundation
|
||||
Copyright (c) 2003-2004, Mark Borgerding
|
||||
|
||||
Redistribution and use in source and binary forms, with or without
|
||||
modification, are permitted provided that the following conditions
|
||||
are met:
|
||||
|
||||
- Redistributions of source code must retain the above copyright
|
||||
notice, this list of conditions and the following disclaimer.
|
||||
|
||||
- Redistributions in binary form must reproduce the above copyright
|
||||
notice, this list of conditions and the following disclaimer in the
|
||||
documentation and/or other materials provided with the distribution.
|
||||
|
||||
- Neither the name of the Xiph.Org Foundation nor the names of its
|
||||
contributors may be used to endorse or promote products derived from
|
||||
this software without specific prior written permission.
|
||||
|
||||
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||||
``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
||||
LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
||||
A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION
|
||||
OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
||||
SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
||||
LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
||||
DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
||||
THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
||||
(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
||||
OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
182
src/modules/ZoomIt/ZoomIt/rnnoise/_kiss_fft_guts.h
Normal file
182
src/modules/ZoomIt/ZoomIt/rnnoise/_kiss_fft_guts.h
Normal file
@@ -0,0 +1,182 @@
|
||||
/*Copyright (c) 2003-2004, Mark Borgerding
|
||||
|
||||
All rights reserved.
|
||||
|
||||
Redistribution and use in source and binary forms, with or without
|
||||
modification, are permitted provided that the following conditions are met:
|
||||
|
||||
* Redistributions of source code must retain the above copyright notice,
|
||||
this list of conditions and the following disclaimer.
|
||||
* Redistributions in binary form must reproduce the above copyright notice,
|
||||
this list of conditions and the following disclaimer in the
|
||||
documentation and/or other materials provided with the distribution.
|
||||
|
||||
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
||||
AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
||||
IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
||||
ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
|
||||
LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
||||
CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
||||
SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
||||
INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
||||
CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
||||
ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
||||
POSSIBILITY OF SUCH DAMAGE.*/
|
||||
|
||||
#ifndef KISS_FFT_GUTS_H
|
||||
#define KISS_FFT_GUTS_H
|
||||
|
||||
#define MIN(a,b) ((a)<(b) ? (a):(b))
|
||||
#define MAX(a,b) ((a)>(b) ? (a):(b))
|
||||
|
||||
/* kiss_fft.h
|
||||
defines kiss_fft_scalar as either short or a float type
|
||||
and defines
|
||||
typedef struct { kiss_fft_scalar r; kiss_fft_scalar i; }kiss_fft_cpx; */
|
||||
#include "kiss_fft.h"
|
||||
|
||||
/*
|
||||
Explanation of macros dealing with complex math:
|
||||
|
||||
C_MUL(m,a,b) : m = a*b
|
||||
C_FIXDIV( c , div ) : if a fixed point impl., c /= div. noop otherwise
|
||||
C_SUB( res, a,b) : res = a - b
|
||||
C_SUBFROM( res , a) : res -= a
|
||||
C_ADDTO( res , a) : res += a
|
||||
* */
|
||||
#ifdef FIXED_POINT
|
||||
#include "arch.h"
|
||||
|
||||
|
||||
#define SAMP_MAX 2147483647
|
||||
#define TWID_MAX 32767
|
||||
#define TRIG_UPSCALE 1
|
||||
|
||||
#define SAMP_MIN -SAMP_MAX
|
||||
|
||||
|
||||
# define S_MUL(a,b) MULT16_32_Q15(b, a)
|
||||
|
||||
# define C_MUL(m,a,b) \
|
||||
do{ (m).r = SUB32_ovflw(S_MUL((a).r,(b).r) , S_MUL((a).i,(b).i)); \
|
||||
(m).i = ADD32_ovflw(S_MUL((a).r,(b).i) , S_MUL((a).i,(b).r)); }while(0)
|
||||
|
||||
# define C_MULC(m,a,b) \
|
||||
do{ (m).r = ADD32_ovflw(S_MUL((a).r,(b).r) , S_MUL((a).i,(b).i)); \
|
||||
(m).i = SUB32_ovflw(S_MUL((a).i,(b).r) , S_MUL((a).r,(b).i)); }while(0)
|
||||
|
||||
# define C_MULBYSCALAR( c, s ) \
|
||||
do{ (c).r = S_MUL( (c).r , s ) ;\
|
||||
(c).i = S_MUL( (c).i , s ) ; }while(0)
|
||||
|
||||
# define DIVSCALAR(x,k) \
|
||||
(x) = S_MUL( x, (TWID_MAX-((k)>>1))/(k)+1 )
|
||||
|
||||
# define C_FIXDIV(c,div) \
|
||||
do { DIVSCALAR( (c).r , div); \
|
||||
DIVSCALAR( (c).i , div); }while (0)
|
||||
|
||||
#define C_ADD( res, a,b)\
|
||||
do {(res).r=ADD32_ovflw((a).r,(b).r); (res).i=ADD32_ovflw((a).i,(b).i); \
|
||||
}while(0)
|
||||
#define C_SUB( res, a,b)\
|
||||
do {(res).r=SUB32_ovflw((a).r,(b).r); (res).i=SUB32_ovflw((a).i,(b).i); \
|
||||
}while(0)
|
||||
#define C_ADDTO( res , a)\
|
||||
do {(res).r = ADD32_ovflw((res).r, (a).r); (res).i = ADD32_ovflw((res).i,(a).i);\
|
||||
}while(0)
|
||||
|
||||
#define C_SUBFROM( res , a)\
|
||||
do {(res).r = ADD32_ovflw((res).r,(a).r); (res).i = SUB32_ovflw((res).i,(a).i); \
|
||||
}while(0)
|
||||
|
||||
#if defined(OPUS_ARM_INLINE_ASM)
|
||||
#include "arm/kiss_fft_armv4.h"
|
||||
#endif
|
||||
|
||||
#if defined(OPUS_ARM_INLINE_EDSP)
|
||||
#include "arm/kiss_fft_armv5e.h"
|
||||
#endif
|
||||
#if defined(MIPSr1_ASM)
|
||||
#include "mips/kiss_fft_mipsr1.h"
|
||||
#endif
|
||||
|
||||
#else /* not FIXED_POINT*/
|
||||
|
||||
# define S_MUL(a,b) ( (a)*(b) )
|
||||
#define C_MUL(m,a,b) \
|
||||
do{ (m).r = (a).r*(b).r - (a).i*(b).i;\
|
||||
(m).i = (a).r*(b).i + (a).i*(b).r; }while(0)
|
||||
#define C_MULC(m,a,b) \
|
||||
do{ (m).r = (a).r*(b).r + (a).i*(b).i;\
|
||||
(m).i = (a).i*(b).r - (a).r*(b).i; }while(0)
|
||||
|
||||
#define C_MUL4(m,a,b) C_MUL(m,a,b)
|
||||
|
||||
# define C_FIXDIV(c,div) /* NOOP */
|
||||
# define C_MULBYSCALAR( c, s ) \
|
||||
do{ (c).r *= (s);\
|
||||
(c).i *= (s); }while(0)
|
||||
#endif
|
||||
|
||||
#ifndef CHECK_OVERFLOW_OP
|
||||
# define CHECK_OVERFLOW_OP(a,op,b) /* noop */
|
||||
#endif
|
||||
|
||||
#ifndef C_ADD
|
||||
#define C_ADD( res, a,b)\
|
||||
do { \
|
||||
CHECK_OVERFLOW_OP((a).r,+,(b).r)\
|
||||
CHECK_OVERFLOW_OP((a).i,+,(b).i)\
|
||||
(res).r=(a).r+(b).r; (res).i=(a).i+(b).i; \
|
||||
}while(0)
|
||||
#define C_SUB( res, a,b)\
|
||||
do { \
|
||||
CHECK_OVERFLOW_OP((a).r,-,(b).r)\
|
||||
CHECK_OVERFLOW_OP((a).i,-,(b).i)\
|
||||
(res).r=(a).r-(b).r; (res).i=(a).i-(b).i; \
|
||||
}while(0)
|
||||
#define C_ADDTO( res , a)\
|
||||
do { \
|
||||
CHECK_OVERFLOW_OP((res).r,+,(a).r)\
|
||||
CHECK_OVERFLOW_OP((res).i,+,(a).i)\
|
||||
(res).r += (a).r; (res).i += (a).i;\
|
||||
}while(0)
|
||||
|
||||
#define C_SUBFROM( res , a)\
|
||||
do {\
|
||||
CHECK_OVERFLOW_OP((res).r,-,(a).r)\
|
||||
CHECK_OVERFLOW_OP((res).i,-,(a).i)\
|
||||
(res).r -= (a).r; (res).i -= (a).i; \
|
||||
}while(0)
|
||||
#endif /* C_ADD defined */
|
||||
|
||||
#ifdef FIXED_POINT
|
||||
/*# define KISS_FFT_COS(phase) TRIG_UPSCALE*floor(MIN(32767,MAX(-32767,.5+32768 * cos (phase))))
|
||||
# define KISS_FFT_SIN(phase) TRIG_UPSCALE*floor(MIN(32767,MAX(-32767,.5+32768 * sin (phase))))*/
|
||||
# define KISS_FFT_COS(phase) floor(.5+TWID_MAX*cos (phase))
|
||||
# define KISS_FFT_SIN(phase) floor(.5+TWID_MAX*sin (phase))
|
||||
# define HALF_OF(x) ((x)>>1)
|
||||
#elif defined(USE_SIMD)
|
||||
# define KISS_FFT_COS(phase) _mm_set1_ps( cos(phase) )
|
||||
# define KISS_FFT_SIN(phase) _mm_set1_ps( sin(phase) )
|
||||
# define HALF_OF(x) ((x)*_mm_set1_ps(.5f))
|
||||
#else
|
||||
# define KISS_FFT_COS(phase) (kiss_fft_scalar) cos(phase)
|
||||
# define KISS_FFT_SIN(phase) (kiss_fft_scalar) sin(phase)
|
||||
# define HALF_OF(x) ((x)*.5f)
|
||||
#endif
|
||||
|
||||
#define kf_cexp(x,phase) \
|
||||
do{ \
|
||||
(x)->r = KISS_FFT_COS(phase);\
|
||||
(x)->i = KISS_FFT_SIN(phase);\
|
||||
}while(0)
|
||||
|
||||
#define kf_cexp2(x,phase) \
|
||||
do{ \
|
||||
(x)->r = TRIG_UPSCALE*celt_cos_norm((phase));\
|
||||
(x)->i = TRIG_UPSCALE*celt_cos_norm((phase)-32768);\
|
||||
}while(0)
|
||||
|
||||
#endif /* KISS_FFT_GUTS_H */
|
||||
261
src/modules/ZoomIt/ZoomIt/rnnoise/arch.h
Normal file
261
src/modules/ZoomIt/ZoomIt/rnnoise/arch.h
Normal file
@@ -0,0 +1,261 @@
|
||||
/* Copyright (c) 2003-2008 Jean-Marc Valin
|
||||
Copyright (c) 2007-2008 CSIRO
|
||||
Copyright (c) 2007-2009 Xiph.Org Foundation
|
||||
Written by Jean-Marc Valin */
|
||||
/**
|
||||
@file arch.h
|
||||
@brief Various architecture definitions for CELT
|
||||
*/
|
||||
/*
|
||||
Redistribution and use in source and binary forms, with or without
|
||||
modification, are permitted provided that the following conditions
|
||||
are met:
|
||||
|
||||
- Redistributions of source code must retain the above copyright
|
||||
notice, this list of conditions and the following disclaimer.
|
||||
|
||||
- Redistributions in binary form must reproduce the above copyright
|
||||
notice, this list of conditions and the following disclaimer in the
|
||||
documentation and/or other materials provided with the distribution.
|
||||
|
||||
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||||
``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
||||
LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
||||
A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER
|
||||
OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
|
||||
EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
|
||||
PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
|
||||
PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
|
||||
LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
|
||||
NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
||||
SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
*/
|
||||
|
||||
#ifndef ARCH_H
|
||||
#define ARCH_H
|
||||
|
||||
#include "opus_types.h"
|
||||
#include "common.h"
|
||||
|
||||
# if !defined(__GNUC_PREREQ)
|
||||
# if defined(__GNUC__)&&defined(__GNUC_MINOR__)
|
||||
# define __GNUC_PREREQ(_maj,_min) \
|
||||
((__GNUC__<<16)+__GNUC_MINOR__>=((_maj)<<16)+(_min))
|
||||
# else
|
||||
# define __GNUC_PREREQ(_maj,_min) 0
|
||||
# endif
|
||||
# endif
|
||||
|
||||
#define CELT_SIG_SCALE 32768.f
|
||||
|
||||
#define celt_fatal(str) _celt_fatal(str, __FILE__, __LINE__);
|
||||
#ifdef ENABLE_ASSERTIONS
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#ifdef __GNUC__
|
||||
__attribute__((noreturn))
|
||||
#endif
|
||||
static OPUS_INLINE void _celt_fatal(const char *str, const char *file, int line)
|
||||
{
|
||||
fprintf (stderr, "Fatal (internal) error in %s, line %d: %s\n", file, line, str);
|
||||
abort();
|
||||
}
|
||||
#define celt_assert(cond) {if (!(cond)) {celt_fatal("assertion failed: " #cond);}}
|
||||
#define celt_assert2(cond, message) {if (!(cond)) {celt_fatal("assertion failed: " #cond "\n" message);}}
|
||||
#else
|
||||
#define celt_assert(cond)
|
||||
#define celt_assert2(cond, message)
|
||||
#endif
|
||||
|
||||
#define IMUL32(a,b) ((a)*(b))
|
||||
|
||||
#define MIN16(a,b) ((a) < (b) ? (a) : (b)) /**< Minimum 16-bit value. */
|
||||
#define MAX16(a,b) ((a) > (b) ? (a) : (b)) /**< Maximum 16-bit value. */
|
||||
#define MIN32(a,b) ((a) < (b) ? (a) : (b)) /**< Minimum 32-bit value. */
|
||||
#define MAX32(a,b) ((a) > (b) ? (a) : (b)) /**< Maximum 32-bit value. */
|
||||
#define IMIN(a,b) ((a) < (b) ? (a) : (b)) /**< Minimum int value. */
|
||||
#define IMAX(a,b) ((a) > (b) ? (a) : (b)) /**< Maximum int value. */
|
||||
#define UADD32(a,b) ((a)+(b))
|
||||
#define USUB32(a,b) ((a)-(b))
|
||||
|
||||
/* Set this if opus_int64 is a native type of the CPU. */
|
||||
/* Assume that all LP64 architectures have fast 64-bit types; also x86_64
|
||||
(which can be ILP32 for x32) and Win64 (which is LLP64). */
|
||||
#if defined(__x86_64__) || defined(__LP64__) || defined(_WIN64)
|
||||
#define OPUS_FAST_INT64 1
|
||||
#else
|
||||
#define OPUS_FAST_INT64 0
|
||||
#endif
|
||||
|
||||
#define PRINT_MIPS(file)
|
||||
|
||||
#ifdef FIXED_POINT
|
||||
|
||||
typedef opus_int16 opus_val16;
|
||||
typedef opus_int32 opus_val32;
|
||||
typedef opus_int64 opus_val64;
|
||||
|
||||
typedef opus_val32 celt_sig;
|
||||
typedef opus_val16 celt_norm;
|
||||
typedef opus_val32 celt_ener;
|
||||
|
||||
#define Q15ONE 32767
|
||||
|
||||
#define SIG_SHIFT 12
|
||||
/* Safe saturation value for 32-bit signals. Should be less than
|
||||
2^31*(1-0.85) to avoid blowing up on DC at deemphasis.*/
|
||||
#define SIG_SAT (300000000)
|
||||
|
||||
#define NORM_SCALING 16384
|
||||
|
||||
#define DB_SHIFT 10
|
||||
|
||||
#define EPSILON 1
|
||||
#define VERY_SMALL 0
|
||||
#define VERY_LARGE16 ((opus_val16)32767)
|
||||
#define Q15_ONE ((opus_val16)32767)
|
||||
|
||||
#define SCALEIN(a) (a)
|
||||
#define SCALEOUT(a) (a)
|
||||
|
||||
#define ABS16(x) ((x) < 0 ? (-(x)) : (x))
|
||||
#define ABS32(x) ((x) < 0 ? (-(x)) : (x))
|
||||
|
||||
static OPUS_INLINE opus_int16 SAT16(opus_int32 x) {
|
||||
return x > 32767 ? 32767 : x < -32768 ? -32768 : (opus_int16)x;
|
||||
}
|
||||
|
||||
#ifdef FIXED_DEBUG
|
||||
#include "fixed_debug.h"
|
||||
#else
|
||||
|
||||
#include "fixed_generic.h"
|
||||
|
||||
#ifdef OPUS_ARM_PRESUME_AARCH64_NEON_INTR
|
||||
#include "arm/fixed_arm64.h"
|
||||
#elif OPUS_ARM_INLINE_EDSP
|
||||
#include "arm/fixed_armv5e.h"
|
||||
#elif defined (OPUS_ARM_INLINE_ASM)
|
||||
#include "arm/fixed_armv4.h"
|
||||
#elif defined (BFIN_ASM)
|
||||
#include "fixed_bfin.h"
|
||||
#elif defined (TI_C5X_ASM)
|
||||
#include "fixed_c5x.h"
|
||||
#elif defined (TI_C6X_ASM)
|
||||
#include "fixed_c6x.h"
|
||||
#endif
|
||||
|
||||
#endif
|
||||
|
||||
#else /* FIXED_POINT */
|
||||
|
||||
typedef float opus_val16;
|
||||
typedef float opus_val32;
|
||||
typedef float opus_val64;
|
||||
|
||||
typedef float celt_sig;
|
||||
typedef float celt_norm;
|
||||
typedef float celt_ener;
|
||||
|
||||
#ifdef FLOAT_APPROX
|
||||
/* This code should reliably detect NaN/inf even when -ffast-math is used.
|
||||
Assumes IEEE 754 format. */
|
||||
static OPUS_INLINE int celt_isnan(float x)
|
||||
{
|
||||
union {float f; opus_uint32 i;} in;
|
||||
in.f = x;
|
||||
return ((in.i>>23)&0xFF)==0xFF && (in.i&0x007FFFFF)!=0;
|
||||
}
|
||||
#else
|
||||
#ifdef __FAST_MATH__
|
||||
#error Cannot build libopus with -ffast-math unless FLOAT_APPROX is defined. This could result in crashes on extreme (e.g. NaN) input
|
||||
#endif
|
||||
#define celt_isnan(x) ((x)!=(x))
|
||||
#endif
|
||||
|
||||
#define Q15ONE 1.0f
|
||||
|
||||
#define NORM_SCALING 1.f
|
||||
|
||||
#define EPSILON 1e-15f
|
||||
#define VERY_SMALL 1e-30f
|
||||
#define VERY_LARGE16 1e15f
|
||||
#define Q15_ONE ((opus_val16)1.f)
|
||||
|
||||
/* This appears to be the same speed as C99's fabsf() but it's more portable. */
|
||||
#define ABS16(x) ((float)fabs(x))
|
||||
#define ABS32(x) ((float)fabs(x))
|
||||
|
||||
#define QCONST16(x,bits) (x)
|
||||
#define QCONST32(x,bits) (x)
|
||||
|
||||
#define NEG16(x) (-(x))
|
||||
#define NEG32(x) (-(x))
|
||||
#define NEG32_ovflw(x) (-(x))
|
||||
#define EXTRACT16(x) (x)
|
||||
#define EXTEND32(x) (x)
|
||||
#define SHR16(a,shift) (a)
|
||||
#define SHL16(a,shift) (a)
|
||||
#define SHR32(a,shift) (a)
|
||||
#define SHL32(a,shift) (a)
|
||||
#define PSHR32(a,shift) (a)
|
||||
#define VSHR32(a,shift) (a)
|
||||
|
||||
#define PSHR(a,shift) (a)
|
||||
#define SHR(a,shift) (a)
|
||||
#define SHL(a,shift) (a)
|
||||
#define SATURATE(x,a) (x)
|
||||
#define SATURATE16(x) (x)
|
||||
|
||||
#define ROUND16(a,shift) (a)
|
||||
#define SROUND16(a,shift) (a)
|
||||
#define HALF16(x) (.5f*(x))
|
||||
#define HALF32(x) (.5f*(x))
|
||||
|
||||
#define ADD16(a,b) ((a)+(b))
|
||||
#define SUB16(a,b) ((a)-(b))
|
||||
#define ADD32(a,b) ((a)+(b))
|
||||
#define SUB32(a,b) ((a)-(b))
|
||||
#define ADD32_ovflw(a,b) ((a)+(b))
|
||||
#define SUB32_ovflw(a,b) ((a)-(b))
|
||||
#define MULT16_16_16(a,b) ((a)*(b))
|
||||
#define MULT16_16(a,b) ((opus_val32)(a)*(opus_val32)(b))
|
||||
#define MAC16_16(c,a,b) ((c)+(opus_val32)(a)*(opus_val32)(b))
|
||||
|
||||
#define MULT16_32_Q15(a,b) ((a)*(b))
|
||||
#define MULT16_32_Q16(a,b) ((a)*(b))
|
||||
|
||||
#define MULT32_32_Q31(a,b) ((a)*(b))
|
||||
|
||||
#define MAC16_32_Q15(c,a,b) ((c)+(a)*(b))
|
||||
#define MAC16_32_Q16(c,a,b) ((c)+(a)*(b))
|
||||
|
||||
#define MULT16_16_Q11_32(a,b) ((a)*(b))
|
||||
#define MULT16_16_Q11(a,b) ((a)*(b))
|
||||
#define MULT16_16_Q13(a,b) ((a)*(b))
|
||||
#define MULT16_16_Q14(a,b) ((a)*(b))
|
||||
#define MULT16_16_Q15(a,b) ((a)*(b))
|
||||
#define MULT16_16_P15(a,b) ((a)*(b))
|
||||
#define MULT16_16_P13(a,b) ((a)*(b))
|
||||
#define MULT16_16_P14(a,b) ((a)*(b))
|
||||
#define MULT16_32_P16(a,b) ((a)*(b))
|
||||
|
||||
#define DIV32_16(a,b) (((opus_val32)(a))/(opus_val16)(b))
|
||||
#define DIV32(a,b) (((opus_val32)(a))/(opus_val32)(b))
|
||||
|
||||
#define SCALEIN(a) ((a)*CELT_SIG_SCALE)
|
||||
#define SCALEOUT(a) ((a)*(1/CELT_SIG_SCALE))
|
||||
|
||||
#define SIG2WORD16(x) (x)
|
||||
|
||||
#endif /* !FIXED_POINT */
|
||||
|
||||
#ifndef GLOBAL_STACK_SIZE
|
||||
#ifdef FIXED_POINT
|
||||
#define GLOBAL_STACK_SIZE 120000
|
||||
#else
|
||||
#define GLOBAL_STACK_SIZE 120000
|
||||
#endif
|
||||
#endif
|
||||
|
||||
#endif /* ARCH_H */
|
||||
174
src/modules/ZoomIt/ZoomIt/rnnoise/celt_lpc.c
Normal file
174
src/modules/ZoomIt/ZoomIt/rnnoise/celt_lpc.c
Normal file
@@ -0,0 +1,174 @@
|
||||
/* Copyright (c) 2009-2010 Xiph.Org Foundation
|
||||
Written by Jean-Marc Valin */
|
||||
/*
|
||||
Redistribution and use in source and binary forms, with or without
|
||||
modification, are permitted provided that the following conditions
|
||||
are met:
|
||||
|
||||
- Redistributions of source code must retain the above copyright
|
||||
notice, this list of conditions and the following disclaimer.
|
||||
|
||||
- Redistributions in binary form must reproduce the above copyright
|
||||
notice, this list of conditions and the following disclaimer in the
|
||||
documentation and/or other materials provided with the distribution.
|
||||
|
||||
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||||
``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
||||
LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
||||
A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER
|
||||
OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
|
||||
EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
|
||||
PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
|
||||
PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
|
||||
LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
|
||||
NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
||||
SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
*/
|
||||
|
||||
#ifdef HAVE_CONFIG_H
|
||||
#include "config.h"
|
||||
#endif
|
||||
|
||||
#include "celt_lpc.h"
|
||||
#include "arch.h"
|
||||
#include "common.h"
|
||||
#include "pitch.h"
|
||||
#include "denoise.h"
|
||||
|
||||
void rnn_lpc(
|
||||
opus_val16 *_lpc, /* out: [0...p-1] LPC coefficients */
|
||||
const opus_val32 *ac, /* in: [0...p] autocorrelation values */
|
||||
int p
|
||||
)
|
||||
{
|
||||
int i, j;
|
||||
opus_val32 r;
|
||||
opus_val32 error = ac[0];
|
||||
#ifdef FIXED_POINT
|
||||
opus_val32 lpc[LPC_ORDER];
|
||||
#else
|
||||
float *lpc = _lpc;
|
||||
#endif
|
||||
|
||||
RNN_CLEAR(lpc, p);
|
||||
if (ac[0] != 0)
|
||||
{
|
||||
for (i = 0; i < p; i++) {
|
||||
/* Sum up this iteration's reflection coefficient */
|
||||
opus_val32 rr = 0;
|
||||
for (j = 0; j < i; j++)
|
||||
rr += MULT32_32_Q31(lpc[j],ac[i - j]);
|
||||
rr += SHR32(ac[i + 1],3);
|
||||
r = -SHL32(rr,3)/error;
|
||||
/* Update LPC coefficients and total error */
|
||||
lpc[i] = SHR32(r,3);
|
||||
for (j = 0; j < (i+1)>>1; j++)
|
||||
{
|
||||
opus_val32 tmp1, tmp2;
|
||||
tmp1 = lpc[j];
|
||||
tmp2 = lpc[i-1-j];
|
||||
lpc[j] = tmp1 + MULT32_32_Q31(r,tmp2);
|
||||
lpc[i-1-j] = tmp2 + MULT32_32_Q31(r,tmp1);
|
||||
}
|
||||
|
||||
error = error - MULT32_32_Q31(MULT32_32_Q31(r,r),error);
|
||||
/* Bail out once we get 30 dB gain */
|
||||
#ifdef FIXED_POINT
|
||||
if (error<SHR32(ac[0],10))
|
||||
break;
|
||||
#else
|
||||
if (error<.001f*ac[0])
|
||||
break;
|
||||
#endif
|
||||
}
|
||||
}
|
||||
#ifdef FIXED_POINT
|
||||
for (i=0;i<p;i++)
|
||||
_lpc[i] = ROUND16(lpc[i],16);
|
||||
#endif
|
||||
}
|
||||
|
||||
|
||||
int rnn_autocorr(
|
||||
const opus_val16 *x, /* in: [0...n-1] samples x */
|
||||
opus_val32 *ac, /* out: [0...lag-1] ac values */
|
||||
const opus_val16 *window,
|
||||
int overlap,
|
||||
int lag,
|
||||
int n)
|
||||
{
|
||||
opus_val32 d;
|
||||
int i, k;
|
||||
int fastN=n-lag;
|
||||
int shift;
|
||||
const opus_val16 *xptr;
|
||||
opus_val16 xx[PITCH_BUF_SIZE/2];
|
||||
celt_assert(n>0);
|
||||
celt_assert(n<=PITCH_BUF_SIZE/2)
|
||||
celt_assert(overlap>=0);
|
||||
if (overlap == 0)
|
||||
{
|
||||
xptr = x;
|
||||
} else {
|
||||
for (i=0;i<n;i++)
|
||||
xx[i] = x[i];
|
||||
for (i=0;i<overlap;i++)
|
||||
{
|
||||
xx[i] = MULT16_16_Q15(x[i],window[i]);
|
||||
xx[n-i-1] = MULT16_16_Q15(x[n-i-1],window[i]);
|
||||
}
|
||||
xptr = xx;
|
||||
}
|
||||
shift=0;
|
||||
#ifdef FIXED_POINT
|
||||
{
|
||||
opus_val32 ac0;
|
||||
ac0 = 1+(n<<7);
|
||||
if (n&1) ac0 += SHR32(MULT16_16(xptr[0],xptr[0]),9);
|
||||
for(i=(n&1);i<n;i+=2)
|
||||
{
|
||||
ac0 += SHR32(MULT16_16(xptr[i],xptr[i]),9);
|
||||
ac0 += SHR32(MULT16_16(xptr[i+1],xptr[i+1]),9);
|
||||
}
|
||||
|
||||
shift = celt_ilog2(ac0)-30+10;
|
||||
shift = (shift)/2;
|
||||
if (shift>0)
|
||||
{
|
||||
for(i=0;i<n;i++)
|
||||
xx[i] = PSHR32(xptr[i], shift);
|
||||
xptr = xx;
|
||||
} else
|
||||
shift = 0;
|
||||
}
|
||||
#endif
|
||||
rnn_pitch_xcorr(xptr, xptr, ac, fastN, lag+1);
|
||||
for (k=0;k<=lag;k++)
|
||||
{
|
||||
for (i = k+fastN, d = 0; i < n; i++)
|
||||
d = MAC16_16(d, xptr[i], xptr[i-k]);
|
||||
ac[k] += d;
|
||||
}
|
||||
#ifdef FIXED_POINT
|
||||
shift = 2*shift;
|
||||
if (shift<=0)
|
||||
ac[0] += SHL32((opus_int32)1, -shift);
|
||||
if (ac[0] < 268435456)
|
||||
{
|
||||
int shift2 = 29 - EC_ILOG(ac[0]);
|
||||
for (i=0;i<=lag;i++)
|
||||
ac[i] = SHL32(ac[i], shift2);
|
||||
shift -= shift2;
|
||||
} else if (ac[0] >= 536870912)
|
||||
{
|
||||
int shift2=1;
|
||||
if (ac[0] >= 1073741824)
|
||||
shift2++;
|
||||
for (i=0;i<=lag;i++)
|
||||
ac[i] = SHR32(ac[i], shift2);
|
||||
shift += shift2;
|
||||
}
|
||||
#endif
|
||||
|
||||
return shift;
|
||||
}
|
||||
45
src/modules/ZoomIt/ZoomIt/rnnoise/celt_lpc.h
Normal file
45
src/modules/ZoomIt/ZoomIt/rnnoise/celt_lpc.h
Normal file
@@ -0,0 +1,45 @@
|
||||
/* Copyright (c) 2009-2010 Xiph.Org Foundation
|
||||
Written by Jean-Marc Valin */
|
||||
/*
|
||||
Redistribution and use in source and binary forms, with or without
|
||||
modification, are permitted provided that the following conditions
|
||||
are met:
|
||||
|
||||
- Redistributions of source code must retain the above copyright
|
||||
notice, this list of conditions and the following disclaimer.
|
||||
|
||||
- Redistributions in binary form must reproduce the above copyright
|
||||
notice, this list of conditions and the following disclaimer in the
|
||||
documentation and/or other materials provided with the distribution.
|
||||
|
||||
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||||
``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
||||
LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
||||
A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER
|
||||
OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
|
||||
EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
|
||||
PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
|
||||
PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
|
||||
LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
|
||||
NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
||||
SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
*/
|
||||
|
||||
#ifndef PLC_H
|
||||
#define PLC_H
|
||||
|
||||
#include "arch.h"
|
||||
#include "common.h"
|
||||
|
||||
#if defined(OPUS_X86_MAY_HAVE_SSE4_1)
|
||||
#include "x86/celt_lpc_sse.h"
|
||||
#endif
|
||||
|
||||
#define LPC_ORDER 24
|
||||
|
||||
void rnn_lpc(opus_val16 *_lpc, const opus_val32 *ac, int p);
|
||||
|
||||
int rnn_autocorr(const opus_val16 *x, opus_val32 *ac,
|
||||
const opus_val16 *window, int overlap, int lag, int n);
|
||||
|
||||
#endif /* PLC_H */
|
||||
56
src/modules/ZoomIt/ZoomIt/rnnoise/common.h
Normal file
56
src/modules/ZoomIt/ZoomIt/rnnoise/common.h
Normal file
@@ -0,0 +1,56 @@
|
||||
|
||||
|
||||
#ifndef COMMON_H
|
||||
#define COMMON_H
|
||||
|
||||
#include "stdlib.h"
|
||||
#include "string.h"
|
||||
|
||||
#define RNN_INLINE inline
|
||||
#define OPUS_INLINE inline
|
||||
|
||||
|
||||
/** RNNoise wrapper for malloc(). To do your own dynamic allocation, all you need t
|
||||
o do is replace this function and rnnoise_free */
|
||||
#ifndef OVERRIDE_RNNOISE_ALLOC
|
||||
static RNN_INLINE void *rnnoise_alloc (size_t size)
|
||||
{
|
||||
return malloc(size);
|
||||
}
|
||||
#endif
|
||||
|
||||
/** RNNoise wrapper for free(). To do your own dynamic allocation, all you need to do is replace this function and rnnoise_alloc */
|
||||
#ifndef OVERRIDE_RNNOISE_FREE
|
||||
static RNN_INLINE void rnnoise_free (void *ptr)
|
||||
{
|
||||
free(ptr);
|
||||
}
|
||||
#endif
|
||||
|
||||
/** Copy n elements from src to dst. The 0* term provides compile-time type checking */
|
||||
#ifndef OVERRIDE_RNN_COPY
|
||||
#define RNN_COPY(dst, src, n) (memcpy((dst), (src), (n)*sizeof(*(dst)) + 0*((dst)-(src)) ))
|
||||
#endif
|
||||
|
||||
/** Copy n elements from src to dst, allowing overlapping regions. The 0* term
|
||||
provides compile-time type checking */
|
||||
#ifndef OVERRIDE_RNN_MOVE
|
||||
#define RNN_MOVE(dst, src, n) (memmove((dst), (src), (n)*sizeof(*(dst)) + 0*((dst)-(src)) ))
|
||||
#endif
|
||||
|
||||
/** Set n elements of dst to zero */
|
||||
#ifndef OVERRIDE_RNN_CLEAR
|
||||
#define RNN_CLEAR(dst, n) (memset((dst), 0, (n)*sizeof(*(dst))))
|
||||
#endif
|
||||
|
||||
# if !defined(OPUS_GNUC_PREREQ)
|
||||
# if defined(__GNUC__)&&defined(__GNUC_MINOR__)
|
||||
# define OPUS_GNUC_PREREQ(_maj,_min) \
|
||||
((__GNUC__<<16)+__GNUC_MINOR__>=((_maj)<<16)+(_min))
|
||||
# else
|
||||
# define OPUS_GNUC_PREREQ(_maj,_min) 0
|
||||
# endif
|
||||
# endif
|
||||
|
||||
|
||||
#endif
|
||||
53
src/modules/ZoomIt/ZoomIt/rnnoise/cpu_support.h
Normal file
53
src/modules/ZoomIt/ZoomIt/rnnoise/cpu_support.h
Normal file
@@ -0,0 +1,53 @@
|
||||
/* Copyright (c) 2010 Xiph.Org Foundation
|
||||
* Copyright (c) 2013 Parrot */
|
||||
/*
|
||||
Redistribution and use in source and binary forms, with or without
|
||||
modification, are permitted provided that the following conditions
|
||||
are met:
|
||||
|
||||
- Redistributions of source code must retain the above copyright
|
||||
notice, this list of conditions and the following disclaimer.
|
||||
|
||||
- Redistributions in binary form must reproduce the above copyright
|
||||
notice, this list of conditions and the following disclaimer in the
|
||||
documentation and/or other materials provided with the distribution.
|
||||
|
||||
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||||
``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
||||
LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
||||
A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER
|
||||
OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
|
||||
EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
|
||||
PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
|
||||
PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
|
||||
LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
|
||||
NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
||||
SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
*/
|
||||
|
||||
#ifndef CPU_SUPPORT_H
|
||||
#define CPU_SUPPORT_H
|
||||
|
||||
#include "opus_types.h"
|
||||
#include "common.h"
|
||||
|
||||
#ifdef RNN_ENABLE_X86_RTCD
|
||||
|
||||
#include "x86/x86cpu.h"
|
||||
/* We currently support 5 x86 variants:
|
||||
* arch[0] -> sse2
|
||||
* arch[1] -> sse4.1
|
||||
* arch[2] -> avx2
|
||||
*/
|
||||
#define OPUS_ARCHMASK 3
|
||||
int rnn_select_arch(void);
|
||||
|
||||
#else
|
||||
#define OPUS_ARCHMASK 0
|
||||
|
||||
static OPUS_INLINE int rnn_select_arch(void)
|
||||
{
|
||||
return 0;
|
||||
}
|
||||
#endif
|
||||
#endif
|
||||
505
src/modules/ZoomIt/ZoomIt/rnnoise/denoise.c
Normal file
505
src/modules/ZoomIt/ZoomIt/rnnoise/denoise.c
Normal file
@@ -0,0 +1,505 @@
|
||||
/* Copyright (c) 2024 Jean-Marc Valin
|
||||
* Copyright (c) 2018 Gregor Richards
|
||||
* Copyright (c) 2017 Mozilla */
|
||||
/*
|
||||
Redistribution and use in source and binary forms, with or without
|
||||
modification, are permitted provided that the following conditions
|
||||
are met:
|
||||
|
||||
- Redistributions of source code must retain the above copyright
|
||||
notice, this list of conditions and the following disclaimer.
|
||||
|
||||
- Redistributions in binary form must reproduce the above copyright
|
||||
notice, this list of conditions and the following disclaimer in the
|
||||
documentation and/or other materials provided with the distribution.
|
||||
|
||||
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||||
``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
||||
LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
||||
A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR
|
||||
CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
|
||||
EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
|
||||
PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
|
||||
PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
|
||||
LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
|
||||
NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
||||
SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
*/
|
||||
|
||||
#ifdef HAVE_CONFIG_H
|
||||
#include "config.h"
|
||||
#endif
|
||||
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
#include <stdio.h>
|
||||
#include "kiss_fft.h"
|
||||
#include "common.h"
|
||||
#include "denoise.h"
|
||||
#include <math.h>
|
||||
#include "rnnoise.h"
|
||||
#include "pitch.h"
|
||||
#include "arch.h"
|
||||
#include "rnn.h"
|
||||
#include "cpu_support.h"
|
||||
|
||||
#define SQUARE(x) ((x)*(x))
|
||||
|
||||
|
||||
#ifndef TRAINING
|
||||
#define TRAINING 0
|
||||
#endif
|
||||
|
||||
|
||||
/* ERB bandwidths going in reverse from 20 kHz and then replacing the 700 and 800
|
||||
with just 750 because having 32 bands is convenient for the DNN.
|
||||
B(1)=400;
|
||||
for k=2:35
|
||||
B(k) = B(k-1) - max(2, round(24.7*(4.37*B(k-1)/20+1)/50));
|
||||
end
|
||||
printf("%d, ", B(end:-1:1));
|
||||
printf("\n")
|
||||
*/
|
||||
const int eband20ms[NB_BANDS+2] = {
|
||||
/*0 100 200 300 400 500 600 750 900 1.1 1.2 1.4 1.6 1.8 2.1 2.4 2.7 3.0 3.4 3.9 4.4 4.9 5.5 6.2 7.0 7.9 8.8 9.9 11.2 12.6 14.1 15.9 17.8 20.0*/
|
||||
0, 2, 4, 6, 8, 10, 12, 15, 18, 21, 24, 28, 32, 36, 41, 47, 53, 60, 68, 77, 87, 98, 110, 124, 140, 157, 176, 198, 223, 251, 282, 317, 356, 400};
|
||||
|
||||
|
||||
struct DenoiseState {
|
||||
RNNoise model;
|
||||
#if !TRAINING
|
||||
int arch;
|
||||
#endif
|
||||
float analysis_mem[FRAME_SIZE];
|
||||
int memid;
|
||||
float synthesis_mem[FRAME_SIZE];
|
||||
float pitch_buf[PITCH_BUF_SIZE];
|
||||
float pitch_enh_buf[PITCH_BUF_SIZE];
|
||||
float last_gain;
|
||||
int last_period;
|
||||
float mem_hp_x[2];
|
||||
float lastg[NB_BANDS];
|
||||
RNNState rnn;
|
||||
kiss_fft_cpx delayed_X[FREQ_SIZE];
|
||||
kiss_fft_cpx delayed_P[FREQ_SIZE];
|
||||
float delayed_Ex[NB_BANDS], delayed_Ep[NB_BANDS];
|
||||
float delayed_Exp[NB_BANDS];
|
||||
|
||||
};
|
||||
|
||||
static void compute_band_energy(float *bandE, const kiss_fft_cpx *X) {
|
||||
int i;
|
||||
float sum[NB_BANDS+2] = {0};
|
||||
for (i=0;i<NB_BANDS+1;i++)
|
||||
{
|
||||
int j;
|
||||
int band_size;
|
||||
band_size = eband20ms[i+1]-eband20ms[i];
|
||||
for (j=0;j<band_size;j++) {
|
||||
float tmp;
|
||||
float frac = (float)j/band_size;
|
||||
tmp = SQUARE(X[eband20ms[i] + j].r);
|
||||
tmp += SQUARE(X[eband20ms[i] + j].i);
|
||||
sum[i] += (1-frac)*tmp;
|
||||
sum[i+1] += frac*tmp;
|
||||
}
|
||||
}
|
||||
sum[1] = (sum[0]+sum[1])*2/3;
|
||||
sum[NB_BANDS] = (sum[NB_BANDS]+sum[NB_BANDS+1])*2/3;
|
||||
for (i=0;i<NB_BANDS;i++)
|
||||
{
|
||||
bandE[i] = sum[i+1];
|
||||
}
|
||||
}
|
||||
|
||||
static void compute_band_corr(float *bandE, const kiss_fft_cpx *X, const kiss_fft_cpx *P) {
|
||||
int i;
|
||||
float sum[NB_BANDS+2] = {0};
|
||||
for (i=0;i<NB_BANDS+1;i++)
|
||||
{
|
||||
int j;
|
||||
int band_size;
|
||||
band_size = eband20ms[i+1]-eband20ms[i];
|
||||
for (j=0;j<band_size;j++) {
|
||||
float tmp;
|
||||
float frac = (float)j/band_size;
|
||||
tmp = X[eband20ms[i] + j].r * P[eband20ms[i] + j].r;
|
||||
tmp += X[eband20ms[i] + j].i * P[eband20ms[i] + j].i;
|
||||
sum[i] += (1-frac)*tmp;
|
||||
sum[i+1] += frac*tmp;
|
||||
}
|
||||
}
|
||||
sum[1] = (sum[0]+sum[1])*2/3;
|
||||
sum[NB_BANDS] = (sum[NB_BANDS]+sum[NB_BANDS+1])*2/3;
|
||||
for (i=0;i<NB_BANDS;i++)
|
||||
{
|
||||
bandE[i] = sum[i+1];
|
||||
}
|
||||
}
|
||||
|
||||
static void interp_band_gain(float *g, const float *bandE) {
|
||||
int i,j;
|
||||
memset(g, 0, FREQ_SIZE);
|
||||
for (i=1;i<NB_BANDS;i++)
|
||||
{
|
||||
int band_size;
|
||||
band_size = eband20ms[i+1]-eband20ms[i];
|
||||
for (j=0;j<band_size;j++) {
|
||||
float frac = (float)j/band_size;
|
||||
g[eband20ms[i] + j] = (1-frac)*bandE[i-1] + frac*bandE[i];
|
||||
}
|
||||
}
|
||||
for (j=0;j<eband20ms[1];j++) g[j] = bandE[0];
|
||||
for (j=eband20ms[NB_BANDS];j<eband20ms[NB_BANDS+1];j++) g[j] = bandE[NB_BANDS-1];
|
||||
}
|
||||
|
||||
extern const float rnn_dct_table[];
|
||||
extern const kiss_fft_state rnn_kfft;
|
||||
extern const float rnn_half_window[];
|
||||
|
||||
static void dct(float *out, const float *in) {
|
||||
int i;
|
||||
for (i=0;i<NB_BANDS;i++) {
|
||||
int j;
|
||||
float sum = 0;
|
||||
for (j=0;j<NB_BANDS;j++) {
|
||||
sum += in[j] * rnn_dct_table[j*NB_BANDS + i];
|
||||
}
|
||||
out[i] = sum*sqrt(2./22);
|
||||
}
|
||||
}
|
||||
|
||||
#if 0
|
||||
static void idct(float *out, const float *in) {
|
||||
int i;
|
||||
for (i=0;i<NB_BANDS;i++) {
|
||||
int j;
|
||||
float sum = 0;
|
||||
for (j=0;j<NB_BANDS;j++) {
|
||||
sum += in[j] * rnn_dct_table[i*NB_BANDS + j];
|
||||
}
|
||||
out[i] = sum*sqrt(2./22);
|
||||
}
|
||||
}
|
||||
#endif
|
||||
|
||||
static void forward_transform(kiss_fft_cpx *out, const float *in) {
|
||||
int i;
|
||||
kiss_fft_cpx x[WINDOW_SIZE];
|
||||
kiss_fft_cpx y[WINDOW_SIZE];
|
||||
for (i=0;i<WINDOW_SIZE;i++) {
|
||||
x[i].r = in[i];
|
||||
x[i].i = 0;
|
||||
}
|
||||
rnn_fft(&rnn_kfft, x, y, 0);
|
||||
for (i=0;i<FREQ_SIZE;i++) {
|
||||
out[i] = y[i];
|
||||
}
|
||||
}
|
||||
|
||||
static void inverse_transform(float *out, const kiss_fft_cpx *in) {
|
||||
int i;
|
||||
kiss_fft_cpx x[WINDOW_SIZE];
|
||||
kiss_fft_cpx y[WINDOW_SIZE];
|
||||
for (i=0;i<FREQ_SIZE;i++) {
|
||||
x[i] = in[i];
|
||||
}
|
||||
for (;i<WINDOW_SIZE;i++) {
|
||||
x[i].r = x[WINDOW_SIZE - i].r;
|
||||
x[i].i = -x[WINDOW_SIZE - i].i;
|
||||
}
|
||||
rnn_fft(&rnn_kfft, x, y, 0);
|
||||
/* output in reverse order for IFFT. */
|
||||
out[0] = WINDOW_SIZE*y[0].r;
|
||||
for (i=1;i<WINDOW_SIZE;i++) {
|
||||
out[i] = WINDOW_SIZE*y[WINDOW_SIZE - i].r;
|
||||
}
|
||||
}
|
||||
|
||||
static void apply_window(float *x) {
|
||||
int i;
|
||||
for (i=0;i<FRAME_SIZE;i++) {
|
||||
x[i] *= rnn_half_window[i];
|
||||
x[WINDOW_SIZE - 1 - i] *= rnn_half_window[i];
|
||||
}
|
||||
}
|
||||
|
||||
struct RNNModel {
|
||||
/* Set either blob or const_blob. */
|
||||
const void *const_blob;
|
||||
void *blob;
|
||||
int blob_len;
|
||||
FILE *file;
|
||||
};
|
||||
|
||||
RNNModel *rnnoise_model_from_buffer(const void *ptr, int len) {
|
||||
RNNModel *model;
|
||||
model = malloc(sizeof(*model));
|
||||
model->blob = NULL;
|
||||
model->const_blob = ptr;
|
||||
model->blob_len = len;
|
||||
return model;
|
||||
}
|
||||
|
||||
RNNModel *rnnoise_model_from_filename(const char *filename) {
|
||||
RNNModel *model;
|
||||
FILE *f = fopen(filename, "rb");
|
||||
model = rnnoise_model_from_file(f);
|
||||
model->file = f;
|
||||
return model;
|
||||
}
|
||||
|
||||
RNNModel *rnnoise_model_from_file(FILE *f) {
|
||||
RNNModel *model;
|
||||
model = malloc(sizeof(*model));
|
||||
model->file = NULL;
|
||||
|
||||
fseek(f, 0, SEEK_END);
|
||||
model->blob_len = ftell(f);
|
||||
fseek(f, 0, SEEK_SET);
|
||||
|
||||
model->const_blob = NULL;
|
||||
model->blob = malloc(model->blob_len);
|
||||
if (fread(model->blob, model->blob_len, 1, f) != 1)
|
||||
{
|
||||
rnnoise_model_free(model);
|
||||
return NULL;
|
||||
}
|
||||
return model;
|
||||
}
|
||||
|
||||
void rnnoise_model_free(RNNModel *model) {
|
||||
if (model->file != NULL) fclose(model->file);
|
||||
if (model->blob != NULL) free(model->blob);
|
||||
free(model);
|
||||
}
|
||||
|
||||
int rnnoise_get_size(void) {
|
||||
return sizeof(DenoiseState);
|
||||
}
|
||||
|
||||
int rnnoise_get_frame_size(void) {
|
||||
return FRAME_SIZE;
|
||||
}
|
||||
|
||||
int rnnoise_init(DenoiseState *st, RNNModel *model) {
|
||||
memset(st, 0, sizeof(*st));
|
||||
#if !TRAINING
|
||||
if (model != NULL) {
|
||||
WeightArray *list;
|
||||
int ret = 1;
|
||||
parse_weights(&list, model->blob ? model->blob : model->const_blob, model->blob_len);
|
||||
if (list != NULL) {
|
||||
ret = init_rnnoise(&st->model, list);
|
||||
opus_free(list);
|
||||
}
|
||||
if (ret != 0) return -1;
|
||||
}
|
||||
#ifndef USE_WEIGHTS_FILE
|
||||
else {
|
||||
int ret = init_rnnoise(&st->model, rnnoise_arrays);
|
||||
if (ret != 0) return -1;
|
||||
}
|
||||
#endif
|
||||
st->arch = rnn_select_arch();
|
||||
#else
|
||||
(void)model;
|
||||
#endif
|
||||
return 0;
|
||||
}
|
||||
|
||||
DenoiseState *rnnoise_create(RNNModel *model) {
|
||||
int ret;
|
||||
DenoiseState *st;
|
||||
st = malloc(rnnoise_get_size());
|
||||
ret = rnnoise_init(st, model);
|
||||
if (ret != 0) {
|
||||
free(st);
|
||||
return NULL;
|
||||
}
|
||||
return st;
|
||||
}
|
||||
|
||||
void rnnoise_destroy(DenoiseState *st) {
|
||||
free(st);
|
||||
}
|
||||
|
||||
#if TRAINING
|
||||
extern int lowpass;
|
||||
extern int band_lp;
|
||||
#endif
|
||||
|
||||
void rnn_frame_analysis(DenoiseState *st, kiss_fft_cpx *X, float *Ex, const float *in) {
|
||||
int i;
|
||||
float x[WINDOW_SIZE];
|
||||
RNN_COPY(x, st->analysis_mem, FRAME_SIZE);
|
||||
for (i=0;i<FRAME_SIZE;i++) x[FRAME_SIZE + i] = in[i];
|
||||
RNN_COPY(st->analysis_mem, in, FRAME_SIZE);
|
||||
apply_window(x);
|
||||
forward_transform(X, x);
|
||||
#if TRAINING
|
||||
for (i=lowpass;i<FREQ_SIZE;i++)
|
||||
X[i].r = X[i].i = 0;
|
||||
#endif
|
||||
compute_band_energy(Ex, X);
|
||||
}
|
||||
|
||||
int rnn_compute_frame_features(DenoiseState *st, kiss_fft_cpx *X, kiss_fft_cpx *P,
|
||||
float *Ex, float *Ep, float *Exp, float *features, const float *in) {
|
||||
int i;
|
||||
float E = 0;
|
||||
float Ly[NB_BANDS];
|
||||
float p[WINDOW_SIZE];
|
||||
float pitch_buf[PITCH_BUF_SIZE>>1];
|
||||
int pitch_index;
|
||||
float gain;
|
||||
float *(pre[1]);
|
||||
float follow, logMax;
|
||||
rnn_frame_analysis(st, X, Ex, in);
|
||||
RNN_MOVE(st->pitch_buf, &st->pitch_buf[FRAME_SIZE], PITCH_BUF_SIZE-FRAME_SIZE);
|
||||
RNN_COPY(&st->pitch_buf[PITCH_BUF_SIZE-FRAME_SIZE], in, FRAME_SIZE);
|
||||
pre[0] = &st->pitch_buf[0];
|
||||
rnn_pitch_downsample(pre, pitch_buf, PITCH_BUF_SIZE, 1);
|
||||
rnn_pitch_search(pitch_buf+(PITCH_MAX_PERIOD>>1), pitch_buf, PITCH_FRAME_SIZE,
|
||||
PITCH_MAX_PERIOD-3*PITCH_MIN_PERIOD, &pitch_index);
|
||||
pitch_index = PITCH_MAX_PERIOD-pitch_index;
|
||||
|
||||
gain = rnn_remove_doubling(pitch_buf, PITCH_MAX_PERIOD, PITCH_MIN_PERIOD,
|
||||
PITCH_FRAME_SIZE, &pitch_index, st->last_period, st->last_gain);
|
||||
st->last_period = pitch_index;
|
||||
st->last_gain = gain;
|
||||
for (i=0;i<WINDOW_SIZE;i++)
|
||||
p[i] = st->pitch_buf[PITCH_BUF_SIZE-WINDOW_SIZE-pitch_index+i];
|
||||
apply_window(p);
|
||||
forward_transform(P, p);
|
||||
compute_band_energy(Ep, P);
|
||||
compute_band_corr(Exp, X, P);
|
||||
for (i=0;i<NB_BANDS;i++) Exp[i] = Exp[i]/sqrt(.001+Ex[i]*Ep[i]);
|
||||
dct(&features[NB_BANDS], Exp);
|
||||
features[2*NB_BANDS] = .01*(pitch_index-300);
|
||||
logMax = -2;
|
||||
follow = -2;
|
||||
for (i=0;i<NB_BANDS;i++) {
|
||||
Ly[i] = log10(1e-2+Ex[i]);
|
||||
Ly[i] = MAX16(logMax-7, MAX16(follow-1.5, Ly[i]));
|
||||
logMax = MAX16(logMax, Ly[i]);
|
||||
follow = MAX16(follow-1.5, Ly[i]);
|
||||
E += Ex[i];
|
||||
}
|
||||
if (!TRAINING && E < 0.04) {
|
||||
/* If there's no audio, avoid messing up the state. */
|
||||
RNN_CLEAR(features, NB_FEATURES);
|
||||
return 1;
|
||||
}
|
||||
dct(features, Ly);
|
||||
features[0] -= 12;
|
||||
features[1] -= 4;
|
||||
return TRAINING && E < 0.1;
|
||||
}
|
||||
|
||||
static void frame_synthesis(DenoiseState *st, float *out, const kiss_fft_cpx *y) {
|
||||
float x[WINDOW_SIZE];
|
||||
int i;
|
||||
inverse_transform(x, y);
|
||||
apply_window(x);
|
||||
for (i=0;i<FRAME_SIZE;i++) out[i] = x[i] + st->synthesis_mem[i];
|
||||
RNN_COPY(st->synthesis_mem, &x[FRAME_SIZE], FRAME_SIZE);
|
||||
}
|
||||
|
||||
void rnn_biquad(float *y, float mem[2], const float *x, const float *b, const float *a, int N) {
|
||||
int i;
|
||||
for (i=0;i<N;i++) {
|
||||
float xi, yi;
|
||||
xi = x[i];
|
||||
yi = x[i] + mem[0];
|
||||
mem[0] = mem[1] + (b[0]*(double)xi - a[0]*(double)yi);
|
||||
mem[1] = (b[1]*(double)xi - a[1]*(double)yi);
|
||||
y[i] = yi;
|
||||
}
|
||||
}
|
||||
|
||||
void rnn_pitch_filter(kiss_fft_cpx *X, const kiss_fft_cpx *P, const float *Ex, const float *Ep,
|
||||
const float *Exp, const float *g) {
|
||||
int i;
|
||||
float r[NB_BANDS];
|
||||
float rf[FREQ_SIZE] = {0};
|
||||
float newE[NB_BANDS];
|
||||
float norm[NB_BANDS];
|
||||
float normf[FREQ_SIZE]={0};
|
||||
for (i=0;i<NB_BANDS;i++) {
|
||||
#if 0
|
||||
if (Exp[i]>g[i]) r[i] = 1;
|
||||
else r[i] = Exp[i]*(1-g[i])/(.001 + g[i]*(1-Exp[i]));
|
||||
r[i] = MIN16(1, MAX16(0, r[i]));
|
||||
#else
|
||||
if (Exp[i]>g[i]) r[i] = 1;
|
||||
else r[i] = SQUARE(Exp[i])*(1-SQUARE(g[i]))/(.001 + SQUARE(g[i])*(1-SQUARE(Exp[i])));
|
||||
r[i] = sqrt(MIN16(1, MAX16(0, r[i])));
|
||||
#endif
|
||||
r[i] *= sqrt(Ex[i]/(1e-8+Ep[i]));
|
||||
}
|
||||
interp_band_gain(rf, r);
|
||||
for (i=0;i<FREQ_SIZE;i++) {
|
||||
X[i].r += rf[i]*P[i].r;
|
||||
X[i].i += rf[i]*P[i].i;
|
||||
}
|
||||
compute_band_energy(newE, X);
|
||||
for (i=0;i<NB_BANDS;i++) {
|
||||
norm[i] = sqrt(Ex[i]/(1e-8+newE[i]));
|
||||
}
|
||||
interp_band_gain(normf, norm);
|
||||
for (i=0;i<FREQ_SIZE;i++) {
|
||||
X[i].r *= normf[i];
|
||||
X[i].i *= normf[i];
|
||||
}
|
||||
}
|
||||
|
||||
float rnnoise_process_frame(DenoiseState *st, float *out, const float *in) {
|
||||
int i;
|
||||
kiss_fft_cpx X[FREQ_SIZE];
|
||||
kiss_fft_cpx P[FREQ_SIZE];
|
||||
float x[FRAME_SIZE];
|
||||
float Ex[NB_BANDS], Ep[NB_BANDS];
|
||||
float Exp[NB_BANDS];
|
||||
float features[NB_FEATURES];
|
||||
float g[NB_BANDS];
|
||||
float gf[FREQ_SIZE]={1};
|
||||
float vad_prob = 0;
|
||||
int silence;
|
||||
static const float a_hp[2] = {-1.99599, 0.99600};
|
||||
static const float b_hp[2] = {-2, 1};
|
||||
rnn_biquad(x, st->mem_hp_x, in, b_hp, a_hp, FRAME_SIZE);
|
||||
silence = rnn_compute_frame_features(st, X, P, Ex, Ep, Exp, features, x);
|
||||
|
||||
if (!silence) {
|
||||
#if !TRAINING
|
||||
compute_rnn(&st->model, &st->rnn, g, &vad_prob, features, st->arch);
|
||||
#endif
|
||||
rnn_pitch_filter(st->delayed_X, st->delayed_P, st->delayed_Ex, st->delayed_Ep, st->delayed_Exp, g);
|
||||
for (i=0;i<NB_BANDS;i++) {
|
||||
float alpha = .6f;
|
||||
/* Cap the decay at 0.6 per frame, corresponding to an RT60 of 135 ms.
|
||||
That avoids unnaturally quick attenuation. */
|
||||
g[i] = MAX16(g[i], alpha*st->lastg[i]);
|
||||
/* Compensate for energy change across frame when computing the threshold gain.
|
||||
Avoids leaking noise when energy increases (e.g. transient noise). */
|
||||
st->lastg[i] = MIN16(1.f, g[i]*(st->delayed_Ex[i]+1e-3)/(Ex[i]+1e-3));
|
||||
}
|
||||
interp_band_gain(gf, g);
|
||||
#if 1
|
||||
for (i=0;i<FREQ_SIZE;i++) {
|
||||
st->delayed_X[i].r *= gf[i];
|
||||
st->delayed_X[i].i *= gf[i];
|
||||
}
|
||||
#endif
|
||||
}
|
||||
frame_synthesis(st, out, st->delayed_X);
|
||||
|
||||
RNN_COPY(st->delayed_X, X, FREQ_SIZE);
|
||||
RNN_COPY(st->delayed_P, P, FREQ_SIZE);
|
||||
RNN_COPY(st->delayed_Ex, Ex, NB_BANDS);
|
||||
RNN_COPY(st->delayed_Ep, Ep, NB_BANDS);
|
||||
RNN_COPY(st->delayed_Exp, Exp, NB_BANDS);
|
||||
return vad_prob;
|
||||
}
|
||||
|
||||
56
src/modules/ZoomIt/ZoomIt/rnnoise/denoise.h
Normal file
56
src/modules/ZoomIt/ZoomIt/rnnoise/denoise.h
Normal file
@@ -0,0 +1,56 @@
|
||||
/* Copyright (c) 2017 Mozilla */
|
||||
/*
|
||||
Redistribution and use in source and binary forms, with or without
|
||||
modification, are permitted provided that the following conditions
|
||||
are met:
|
||||
|
||||
- Redistributions of source code must retain the above copyright
|
||||
notice, this list of conditions and the following disclaimer.
|
||||
|
||||
- Redistributions in binary form must reproduce the above copyright
|
||||
notice, this list of conditions and the following disclaimer in the
|
||||
documentation and/or other materials provided with the distribution.
|
||||
|
||||
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||||
``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
||||
LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
||||
A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR
|
||||
CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
|
||||
EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
|
||||
PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
|
||||
PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
|
||||
LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
|
||||
NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
||||
SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
*/
|
||||
|
||||
#include "rnnoise.h"
|
||||
#include "kiss_fft.h"
|
||||
#include "nnet.h"
|
||||
|
||||
#define FRAME_SIZE 480
|
||||
#define WINDOW_SIZE (2*FRAME_SIZE)
|
||||
#define FREQ_SIZE (FRAME_SIZE + 1)
|
||||
#define NB_BANDS 32
|
||||
#define NB_FEATURES (2*NB_BANDS+1)
|
||||
|
||||
|
||||
#define PITCH_MIN_PERIOD 60
|
||||
#define PITCH_MAX_PERIOD 768
|
||||
#define PITCH_FRAME_SIZE 960
|
||||
#define PITCH_BUF_SIZE (PITCH_MAX_PERIOD+PITCH_FRAME_SIZE)
|
||||
|
||||
extern const WeightArray rnnoise_arrays[];
|
||||
|
||||
extern const int eband20ms[];
|
||||
|
||||
|
||||
void rnn_biquad(float *y, float mem[2], const float *x, const float *b, const float *a, int N);
|
||||
|
||||
void rnn_pitch_filter(kiss_fft_cpx *X, const kiss_fft_cpx *P, const float *Ex, const float *Ep,
|
||||
const float *Exp, const float *g);
|
||||
|
||||
void rnn_frame_analysis(DenoiseState *st, kiss_fft_cpx *X, float *Ex, const float *in);
|
||||
|
||||
int rnn_compute_frame_features(DenoiseState *st, kiss_fft_cpx *X, kiss_fft_cpx *P,
|
||||
float *Ex, float *Ep, float *Exp, float *features, const float *in);
|
||||
601
src/modules/ZoomIt/ZoomIt/rnnoise/kiss_fft.c
Normal file
601
src/modules/ZoomIt/ZoomIt/rnnoise/kiss_fft.c
Normal file
@@ -0,0 +1,601 @@
|
||||
/*Copyright (c) 2003-2004, Mark Borgerding
|
||||
Lots of modifications by Jean-Marc Valin
|
||||
Copyright (c) 2005-2007, Xiph.Org Foundation
|
||||
Copyright (c) 2008, Xiph.Org Foundation, CSIRO
|
||||
|
||||
All rights reserved.
|
||||
|
||||
Redistribution and use in source and binary forms, with or without
|
||||
modification, are permitted provided that the following conditions are met:
|
||||
|
||||
* Redistributions of source code must retain the above copyright notice,
|
||||
this list of conditions and the following disclaimer.
|
||||
* Redistributions in binary form must reproduce the above copyright notice,
|
||||
this list of conditions and the following disclaimer in the
|
||||
documentation and/or other materials provided with the distribution.
|
||||
|
||||
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
||||
AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
||||
IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
||||
ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
|
||||
LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
||||
CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
||||
SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
||||
INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
||||
CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
||||
ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
||||
POSSIBILITY OF SUCH DAMAGE.*/
|
||||
|
||||
/* This code is originally from Mark Borgerding's KISS-FFT but has been
|
||||
heavily modified to better suit Opus */
|
||||
|
||||
#ifndef SKIP_CONFIG_H
|
||||
# ifdef HAVE_CONFIG_H
|
||||
# include "config.h"
|
||||
# endif
|
||||
#endif
|
||||
|
||||
#include "_kiss_fft_guts.h"
|
||||
#define CUSTOM_MODES
|
||||
|
||||
/* The guts header contains all the multiplication and addition macros that are defined for
|
||||
complex numbers. It also declares the kf_ internal functions.
|
||||
*/
|
||||
|
||||
static void kf_bfly2(
|
||||
kiss_fft_cpx * Fout,
|
||||
int m,
|
||||
int N
|
||||
)
|
||||
{
|
||||
kiss_fft_cpx * Fout2;
|
||||
int i;
|
||||
(void)m;
|
||||
#ifdef CUSTOM_MODES
|
||||
if (m==1)
|
||||
{
|
||||
celt_assert(m==1);
|
||||
for (i=0;i<N;i++)
|
||||
{
|
||||
kiss_fft_cpx t;
|
||||
Fout2 = Fout + 1;
|
||||
t = *Fout2;
|
||||
C_SUB( *Fout2 , *Fout , t );
|
||||
C_ADDTO( *Fout , t );
|
||||
Fout += 2;
|
||||
}
|
||||
} else
|
||||
#endif
|
||||
{
|
||||
opus_val16 tw;
|
||||
tw = QCONST16(0.7071067812f, 15);
|
||||
/* We know that m==4 here because the radix-2 is just after a radix-4 */
|
||||
celt_assert(m==4);
|
||||
for (i=0;i<N;i++)
|
||||
{
|
||||
kiss_fft_cpx t;
|
||||
Fout2 = Fout + 4;
|
||||
t = Fout2[0];
|
||||
C_SUB( Fout2[0] , Fout[0] , t );
|
||||
C_ADDTO( Fout[0] , t );
|
||||
|
||||
t.r = S_MUL(ADD32_ovflw(Fout2[1].r, Fout2[1].i), tw);
|
||||
t.i = S_MUL(SUB32_ovflw(Fout2[1].i, Fout2[1].r), tw);
|
||||
C_SUB( Fout2[1] , Fout[1] , t );
|
||||
C_ADDTO( Fout[1] , t );
|
||||
|
||||
t.r = Fout2[2].i;
|
||||
t.i = -Fout2[2].r;
|
||||
C_SUB( Fout2[2] , Fout[2] , t );
|
||||
C_ADDTO( Fout[2] , t );
|
||||
|
||||
t.r = S_MUL(SUB32_ovflw(Fout2[3].i, Fout2[3].r), tw);
|
||||
t.i = S_MUL(NEG32_ovflw(ADD32_ovflw(Fout2[3].i, Fout2[3].r)), tw);
|
||||
C_SUB( Fout2[3] , Fout[3] , t );
|
||||
C_ADDTO( Fout[3] , t );
|
||||
Fout += 8;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
static void kf_bfly4(
|
||||
kiss_fft_cpx * Fout,
|
||||
const size_t fstride,
|
||||
const kiss_fft_state *st,
|
||||
int m,
|
||||
int N,
|
||||
int mm
|
||||
)
|
||||
{
|
||||
int i;
|
||||
|
||||
if (m==1)
|
||||
{
|
||||
/* Degenerate case where all the twiddles are 1. */
|
||||
for (i=0;i<N;i++)
|
||||
{
|
||||
kiss_fft_cpx scratch0, scratch1;
|
||||
|
||||
C_SUB( scratch0 , *Fout, Fout[2] );
|
||||
C_ADDTO(*Fout, Fout[2]);
|
||||
C_ADD( scratch1 , Fout[1] , Fout[3] );
|
||||
C_SUB( Fout[2], *Fout, scratch1 );
|
||||
C_ADDTO( *Fout , scratch1 );
|
||||
C_SUB( scratch1 , Fout[1] , Fout[3] );
|
||||
|
||||
Fout[1].r = ADD32_ovflw(scratch0.r, scratch1.i);
|
||||
Fout[1].i = SUB32_ovflw(scratch0.i, scratch1.r);
|
||||
Fout[3].r = SUB32_ovflw(scratch0.r, scratch1.i);
|
||||
Fout[3].i = ADD32_ovflw(scratch0.i, scratch1.r);
|
||||
Fout+=4;
|
||||
}
|
||||
} else {
|
||||
int j;
|
||||
kiss_fft_cpx scratch[6];
|
||||
const kiss_twiddle_cpx *tw1,*tw2,*tw3;
|
||||
const int m2=2*m;
|
||||
const int m3=3*m;
|
||||
kiss_fft_cpx * Fout_beg = Fout;
|
||||
for (i=0;i<N;i++)
|
||||
{
|
||||
Fout = Fout_beg + i*mm;
|
||||
tw3 = tw2 = tw1 = st->twiddles;
|
||||
/* m is guaranteed to be a multiple of 4. */
|
||||
for (j=0;j<m;j++)
|
||||
{
|
||||
C_MUL(scratch[0],Fout[m] , *tw1 );
|
||||
C_MUL(scratch[1],Fout[m2] , *tw2 );
|
||||
C_MUL(scratch[2],Fout[m3] , *tw3 );
|
||||
|
||||
C_SUB( scratch[5] , *Fout, scratch[1] );
|
||||
C_ADDTO(*Fout, scratch[1]);
|
||||
C_ADD( scratch[3] , scratch[0] , scratch[2] );
|
||||
C_SUB( scratch[4] , scratch[0] , scratch[2] );
|
||||
C_SUB( Fout[m2], *Fout, scratch[3] );
|
||||
tw1 += fstride;
|
||||
tw2 += fstride*2;
|
||||
tw3 += fstride*3;
|
||||
C_ADDTO( *Fout , scratch[3] );
|
||||
|
||||
Fout[m].r = ADD32_ovflw(scratch[5].r, scratch[4].i);
|
||||
Fout[m].i = SUB32_ovflw(scratch[5].i, scratch[4].r);
|
||||
Fout[m3].r = SUB32_ovflw(scratch[5].r, scratch[4].i);
|
||||
Fout[m3].i = ADD32_ovflw(scratch[5].i, scratch[4].r);
|
||||
++Fout;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
#ifndef RADIX_TWO_ONLY
|
||||
|
||||
static void kf_bfly3(
|
||||
kiss_fft_cpx * Fout,
|
||||
const size_t fstride,
|
||||
const kiss_fft_state *st,
|
||||
int m,
|
||||
int N,
|
||||
int mm
|
||||
)
|
||||
{
|
||||
int i;
|
||||
size_t k;
|
||||
const size_t m2 = 2*m;
|
||||
const kiss_twiddle_cpx *tw1,*tw2;
|
||||
kiss_fft_cpx scratch[5];
|
||||
kiss_twiddle_cpx epi3;
|
||||
|
||||
kiss_fft_cpx * Fout_beg = Fout;
|
||||
#ifdef FIXED_POINT
|
||||
/*epi3.r = -16384;*/ /* Unused */
|
||||
epi3.i = -28378;
|
||||
#else
|
||||
epi3 = st->twiddles[fstride*m];
|
||||
#endif
|
||||
for (i=0;i<N;i++)
|
||||
{
|
||||
Fout = Fout_beg + i*mm;
|
||||
tw1=tw2=st->twiddles;
|
||||
/* For non-custom modes, m is guaranteed to be a multiple of 4. */
|
||||
k=m;
|
||||
do {
|
||||
|
||||
C_MUL(scratch[1],Fout[m] , *tw1);
|
||||
C_MUL(scratch[2],Fout[m2] , *tw2);
|
||||
|
||||
C_ADD(scratch[3],scratch[1],scratch[2]);
|
||||
C_SUB(scratch[0],scratch[1],scratch[2]);
|
||||
tw1 += fstride;
|
||||
tw2 += fstride*2;
|
||||
|
||||
Fout[m].r = SUB32_ovflw(Fout->r, HALF_OF(scratch[3].r));
|
||||
Fout[m].i = SUB32_ovflw(Fout->i, HALF_OF(scratch[3].i));
|
||||
|
||||
C_MULBYSCALAR( scratch[0] , epi3.i );
|
||||
|
||||
C_ADDTO(*Fout,scratch[3]);
|
||||
|
||||
Fout[m2].r = ADD32_ovflw(Fout[m].r, scratch[0].i);
|
||||
Fout[m2].i = SUB32_ovflw(Fout[m].i, scratch[0].r);
|
||||
|
||||
Fout[m].r = SUB32_ovflw(Fout[m].r, scratch[0].i);
|
||||
Fout[m].i = ADD32_ovflw(Fout[m].i, scratch[0].r);
|
||||
|
||||
++Fout;
|
||||
} while(--k);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
#ifndef OVERRIDE_kf_bfly5
|
||||
static void kf_bfly5(
|
||||
kiss_fft_cpx * Fout,
|
||||
const size_t fstride,
|
||||
const kiss_fft_state *st,
|
||||
int m,
|
||||
int N,
|
||||
int mm
|
||||
)
|
||||
{
|
||||
kiss_fft_cpx *Fout0,*Fout1,*Fout2,*Fout3,*Fout4;
|
||||
int i, u;
|
||||
kiss_fft_cpx scratch[13];
|
||||
const kiss_twiddle_cpx *tw;
|
||||
kiss_twiddle_cpx ya,yb;
|
||||
kiss_fft_cpx * Fout_beg = Fout;
|
||||
|
||||
#ifdef FIXED_POINT
|
||||
ya.r = 10126;
|
||||
ya.i = -31164;
|
||||
yb.r = -26510;
|
||||
yb.i = -19261;
|
||||
#else
|
||||
ya = st->twiddles[fstride*m];
|
||||
yb = st->twiddles[fstride*2*m];
|
||||
#endif
|
||||
tw=st->twiddles;
|
||||
|
||||
for (i=0;i<N;i++)
|
||||
{
|
||||
Fout = Fout_beg + i*mm;
|
||||
Fout0=Fout;
|
||||
Fout1=Fout0+m;
|
||||
Fout2=Fout0+2*m;
|
||||
Fout3=Fout0+3*m;
|
||||
Fout4=Fout0+4*m;
|
||||
|
||||
/* For non-custom modes, m is guaranteed to be a multiple of 4. */
|
||||
for ( u=0; u<m; ++u ) {
|
||||
scratch[0] = *Fout0;
|
||||
|
||||
C_MUL(scratch[1] ,*Fout1, tw[u*fstride]);
|
||||
C_MUL(scratch[2] ,*Fout2, tw[2*u*fstride]);
|
||||
C_MUL(scratch[3] ,*Fout3, tw[3*u*fstride]);
|
||||
C_MUL(scratch[4] ,*Fout4, tw[4*u*fstride]);
|
||||
|
||||
C_ADD( scratch[7],scratch[1],scratch[4]);
|
||||
C_SUB( scratch[10],scratch[1],scratch[4]);
|
||||
C_ADD( scratch[8],scratch[2],scratch[3]);
|
||||
C_SUB( scratch[9],scratch[2],scratch[3]);
|
||||
|
||||
Fout0->r = ADD32_ovflw(Fout0->r, ADD32_ovflw(scratch[7].r, scratch[8].r));
|
||||
Fout0->i = ADD32_ovflw(Fout0->i, ADD32_ovflw(scratch[7].i, scratch[8].i));
|
||||
|
||||
scratch[5].r = ADD32_ovflw(scratch[0].r, ADD32_ovflw(S_MUL(scratch[7].r,ya.r), S_MUL(scratch[8].r,yb.r)));
|
||||
scratch[5].i = ADD32_ovflw(scratch[0].i, ADD32_ovflw(S_MUL(scratch[7].i,ya.r), S_MUL(scratch[8].i,yb.r)));
|
||||
|
||||
scratch[6].r = ADD32_ovflw(S_MUL(scratch[10].i,ya.i), S_MUL(scratch[9].i,yb.i));
|
||||
scratch[6].i = NEG32_ovflw(ADD32_ovflw(S_MUL(scratch[10].r,ya.i), S_MUL(scratch[9].r,yb.i)));
|
||||
|
||||
C_SUB(*Fout1,scratch[5],scratch[6]);
|
||||
C_ADD(*Fout4,scratch[5],scratch[6]);
|
||||
|
||||
scratch[11].r = ADD32_ovflw(scratch[0].r, ADD32_ovflw(S_MUL(scratch[7].r,yb.r), S_MUL(scratch[8].r,ya.r)));
|
||||
scratch[11].i = ADD32_ovflw(scratch[0].i, ADD32_ovflw(S_MUL(scratch[7].i,yb.r), S_MUL(scratch[8].i,ya.r)));
|
||||
scratch[12].r = SUB32_ovflw(S_MUL(scratch[9].i,ya.i), S_MUL(scratch[10].i,yb.i));
|
||||
scratch[12].i = SUB32_ovflw(S_MUL(scratch[10].r,yb.i), S_MUL(scratch[9].r,ya.i));
|
||||
|
||||
C_ADD(*Fout2,scratch[11],scratch[12]);
|
||||
C_SUB(*Fout3,scratch[11],scratch[12]);
|
||||
|
||||
++Fout0;++Fout1;++Fout2;++Fout3;++Fout4;
|
||||
}
|
||||
}
|
||||
}
|
||||
#endif /* OVERRIDE_kf_bfly5 */
|
||||
|
||||
|
||||
#endif
|
||||
|
||||
|
||||
#ifdef CUSTOM_MODES
|
||||
|
||||
static
|
||||
void compute_bitrev_table(
|
||||
int Fout,
|
||||
opus_int32 *f,
|
||||
const size_t fstride,
|
||||
int in_stride,
|
||||
opus_int16 * factors,
|
||||
const kiss_fft_state *st
|
||||
)
|
||||
{
|
||||
const int p=*factors++; /* the radix */
|
||||
const int m=*factors++; /* stage's fft length/p */
|
||||
|
||||
/*printf ("fft %d %d %d %d %d %d\n", p*m, m, p, s2, fstride*in_stride, N);*/
|
||||
if (m==1)
|
||||
{
|
||||
int j;
|
||||
for (j=0;j<p;j++)
|
||||
{
|
||||
*f = Fout+j;
|
||||
f += fstride*in_stride;
|
||||
}
|
||||
} else {
|
||||
int j;
|
||||
for (j=0;j<p;j++)
|
||||
{
|
||||
compute_bitrev_table( Fout , f, fstride*p, in_stride, factors,st);
|
||||
f += fstride*in_stride;
|
||||
Fout += m;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/* facbuf is populated by p1,m1,p2,m2, ...
|
||||
where
|
||||
p[i] * m[i] = m[i-1]
|
||||
m0 = n */
|
||||
static
|
||||
int kf_factor(int n,opus_int16 * facbuf)
|
||||
{
|
||||
int p=4;
|
||||
int i;
|
||||
int stages=0;
|
||||
int nbak = n;
|
||||
|
||||
/*factor out powers of 4, powers of 2, then any remaining primes */
|
||||
do {
|
||||
while (n % p) {
|
||||
switch (p) {
|
||||
case 4: p = 2; break;
|
||||
case 2: p = 3; break;
|
||||
default: p += 2; break;
|
||||
}
|
||||
if (p>32000 || (opus_int32)p*(opus_int32)p > n)
|
||||
p = n; /* no more factors, skip to end */
|
||||
}
|
||||
n /= p;
|
||||
#ifdef RADIX_TWO_ONLY
|
||||
if (p!=2 && p != 4)
|
||||
#else
|
||||
if (p>5)
|
||||
#endif
|
||||
{
|
||||
return 0;
|
||||
}
|
||||
facbuf[2*stages] = p;
|
||||
if (p==2 && stages > 1)
|
||||
{
|
||||
facbuf[2*stages] = 4;
|
||||
facbuf[2] = 2;
|
||||
}
|
||||
stages++;
|
||||
} while (n > 1);
|
||||
n = nbak;
|
||||
/* Reverse the order to get the radix 4 at the end, so we can use the
|
||||
fast degenerate case. It turns out that reversing the order also
|
||||
improves the noise behaviour. */
|
||||
for (i=0;i<stages/2;i++)
|
||||
{
|
||||
int tmp;
|
||||
tmp = facbuf[2*i];
|
||||
facbuf[2*i] = facbuf[2*(stages-i-1)];
|
||||
facbuf[2*(stages-i-1)] = tmp;
|
||||
}
|
||||
for (i=0;i<stages;i++)
|
||||
{
|
||||
n /= facbuf[2*i];
|
||||
facbuf[2*i+1] = n;
|
||||
}
|
||||
return 1;
|
||||
}
|
||||
|
||||
static void compute_twiddles(kiss_twiddle_cpx *twiddles, int nfft)
|
||||
{
|
||||
int i;
|
||||
#ifdef FIXED_POINT
|
||||
for (i=0;i<nfft;++i) {
|
||||
opus_val32 phase = -i;
|
||||
kf_cexp2(twiddles+i, DIV32(SHL32(phase,17),nfft));
|
||||
}
|
||||
#else
|
||||
for (i=0;i<nfft;++i) {
|
||||
const double pi=3.14159265358979323846264338327;
|
||||
double phase = ( -2*pi /nfft ) * i;
|
||||
kf_cexp(twiddles+i, phase );
|
||||
}
|
||||
#endif
|
||||
}
|
||||
|
||||
int rnn_fft_alloc_arch_c(kiss_fft_state *st) {
|
||||
(void)st;
|
||||
return 0;
|
||||
}
|
||||
|
||||
/*
|
||||
*
|
||||
* Allocates all necessary storage space for the fft and ifft.
|
||||
* The return value is a contiguous block of memory. As such,
|
||||
* It can be freed with free().
|
||||
* */
|
||||
kiss_fft_state *rnn_fft_alloc_twiddles(int nfft,void * mem,size_t * lenmem,
|
||||
const kiss_fft_state *base, int arch)
|
||||
{
|
||||
kiss_fft_state *st=NULL;
|
||||
size_t memneeded = sizeof(struct kiss_fft_state); /* twiddle factors*/
|
||||
|
||||
if ( lenmem==NULL ) {
|
||||
st = ( kiss_fft_state*)KISS_FFT_MALLOC( memneeded );
|
||||
}else{
|
||||
if (mem != NULL && *lenmem >= memneeded)
|
||||
st = (kiss_fft_state*)mem;
|
||||
*lenmem = memneeded;
|
||||
}
|
||||
if (st) {
|
||||
opus_int32 *bitrev;
|
||||
kiss_twiddle_cpx *twiddles;
|
||||
|
||||
st->nfft=nfft;
|
||||
#ifdef FIXED_POINT
|
||||
st->scale_shift = celt_ilog2(st->nfft);
|
||||
if (st->nfft == 1<<st->scale_shift)
|
||||
st->scale = Q15ONE;
|
||||
else
|
||||
st->scale = (1073741824+st->nfft/2)/st->nfft>>(15-st->scale_shift);
|
||||
#else
|
||||
st->scale = 1.f/nfft;
|
||||
#endif
|
||||
if (base != NULL)
|
||||
{
|
||||
st->twiddles = base->twiddles;
|
||||
st->shift = 0;
|
||||
while (st->shift < 32 && nfft<<st->shift != base->nfft)
|
||||
st->shift++;
|
||||
if (st->shift>=32)
|
||||
goto fail;
|
||||
} else {
|
||||
st->twiddles = twiddles = (kiss_twiddle_cpx*)KISS_FFT_MALLOC(sizeof(kiss_twiddle_cpx)*nfft);
|
||||
compute_twiddles(twiddles, nfft);
|
||||
st->shift = -1;
|
||||
}
|
||||
if (!kf_factor(nfft,st->factors))
|
||||
{
|
||||
goto fail;
|
||||
}
|
||||
|
||||
/* bitrev */
|
||||
st->bitrev = bitrev = (opus_int32*)KISS_FFT_MALLOC(sizeof(opus_int32)*nfft);
|
||||
if (st->bitrev==NULL)
|
||||
goto fail;
|
||||
compute_bitrev_table(0, bitrev, 1,1, st->factors,st);
|
||||
|
||||
/* Initialize architecture specific fft parameters */
|
||||
if (rnn_fft_alloc_arch(st, arch))
|
||||
goto fail;
|
||||
}
|
||||
return st;
|
||||
fail:
|
||||
rnn_fft_free(st, arch);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
kiss_fft_state *rnn_fft_alloc(int nfft,void * mem,size_t * lenmem, int arch)
|
||||
{
|
||||
return rnn_fft_alloc_twiddles(nfft, mem, lenmem, NULL, arch);
|
||||
}
|
||||
|
||||
void rnn_fft_free_arch_c(kiss_fft_state *st) {
|
||||
(void)st;
|
||||
}
|
||||
|
||||
void rnn_fft_free(const kiss_fft_state *cfg, int arch)
|
||||
{
|
||||
if (cfg)
|
||||
{
|
||||
rnn_fft_free_arch((kiss_fft_state *)cfg, arch);
|
||||
opus_free((opus_int32*)cfg->bitrev);
|
||||
if (cfg->shift < 0)
|
||||
opus_free((kiss_twiddle_cpx*)cfg->twiddles);
|
||||
opus_free((kiss_fft_state*)cfg);
|
||||
}
|
||||
}
|
||||
|
||||
#endif /* CUSTOM_MODES */
|
||||
|
||||
void rnn_fft_impl(const kiss_fft_state *st,kiss_fft_cpx *fout)
|
||||
{
|
||||
int m2, m;
|
||||
int p;
|
||||
int L;
|
||||
int fstride[MAXFACTORS];
|
||||
int i;
|
||||
int shift;
|
||||
|
||||
/* st->shift can be -1 */
|
||||
shift = st->shift>0 ? st->shift : 0;
|
||||
|
||||
fstride[0] = 1;
|
||||
L=0;
|
||||
do {
|
||||
p = st->factors[2*L];
|
||||
m = st->factors[2*L+1];
|
||||
fstride[L+1] = fstride[L]*p;
|
||||
L++;
|
||||
} while(m!=1);
|
||||
m = st->factors[2*L-1];
|
||||
for (i=L-1;i>=0;i--)
|
||||
{
|
||||
if (i!=0)
|
||||
m2 = st->factors[2*i-1];
|
||||
else
|
||||
m2 = 1;
|
||||
switch (st->factors[2*i])
|
||||
{
|
||||
case 2:
|
||||
kf_bfly2(fout, m, fstride[i]);
|
||||
break;
|
||||
case 4:
|
||||
kf_bfly4(fout,fstride[i]<<shift,st,m, fstride[i], m2);
|
||||
break;
|
||||
#ifndef RADIX_TWO_ONLY
|
||||
case 3:
|
||||
kf_bfly3(fout,fstride[i]<<shift,st,m, fstride[i], m2);
|
||||
break;
|
||||
case 5:
|
||||
kf_bfly5(fout,fstride[i]<<shift,st,m, fstride[i], m2);
|
||||
break;
|
||||
#endif
|
||||
}
|
||||
m = m2;
|
||||
}
|
||||
}
|
||||
|
||||
void rnn_fft_c(const kiss_fft_state *st,const kiss_fft_cpx *fin,kiss_fft_cpx *fout)
|
||||
{
|
||||
int i;
|
||||
opus_val16 scale;
|
||||
#ifdef FIXED_POINT
|
||||
/* Allows us to scale with MULT16_32_Q16(), which is faster than
|
||||
MULT16_32_Q15() on ARM. */
|
||||
int scale_shift = st->scale_shift-1;
|
||||
#endif
|
||||
scale = st->scale;
|
||||
|
||||
celt_assert2 (fin != fout, "In-place FFT not supported");
|
||||
/* Bit-reverse the input */
|
||||
for (i=0;i<st->nfft;i++)
|
||||
{
|
||||
kiss_fft_cpx x = fin[i];
|
||||
fout[st->bitrev[i]].r = SHR32(MULT16_32_Q16(scale, x.r), scale_shift);
|
||||
fout[st->bitrev[i]].i = SHR32(MULT16_32_Q16(scale, x.i), scale_shift);
|
||||
}
|
||||
rnn_fft_impl(st, fout);
|
||||
}
|
||||
|
||||
|
||||
void rnn_ifft_c(const kiss_fft_state *st,const kiss_fft_cpx *fin,kiss_fft_cpx *fout)
|
||||
{
|
||||
int i;
|
||||
celt_assert2 (fin != fout, "In-place FFT not supported");
|
||||
/* Bit-reverse the input */
|
||||
for (i=0;i<st->nfft;i++)
|
||||
fout[st->bitrev[i]] = fin[i];
|
||||
for (i=0;i<st->nfft;i++)
|
||||
fout[i].i = -fout[i].i;
|
||||
rnn_fft_impl(st, fout);
|
||||
for (i=0;i<st->nfft;i++)
|
||||
fout[i].i = -fout[i].i;
|
||||
}
|
||||
203
src/modules/ZoomIt/ZoomIt/rnnoise/kiss_fft.h
Normal file
203
src/modules/ZoomIt/ZoomIt/rnnoise/kiss_fft.h
Normal file
@@ -0,0 +1,203 @@
|
||||
/*Copyright (c) 2003-2004, Mark Borgerding
|
||||
Lots of modifications by Jean-Marc Valin
|
||||
Copyright (c) 2005-2007, Xiph.Org Foundation
|
||||
Copyright (c) 2008, Xiph.Org Foundation, CSIRO
|
||||
|
||||
All rights reserved.
|
||||
|
||||
Redistribution and use in source and binary forms, with or without
|
||||
modification, are permitted provided that the following conditions are met:
|
||||
|
||||
* Redistributions of source code must retain the above copyright notice,
|
||||
this list of conditions and the following disclaimer.
|
||||
* Redistributions in binary form must reproduce the above copyright notice,
|
||||
this list of conditions and the following disclaimer in the
|
||||
documentation and/or other materials provided with the distribution.
|
||||
|
||||
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
||||
AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
||||
IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
||||
ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
|
||||
LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
||||
CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
||||
SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
||||
INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
||||
CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
||||
ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
||||
POSSIBILITY OF SUCH DAMAGE.*/
|
||||
|
||||
#ifndef KISS_FFT_H
|
||||
#define KISS_FFT_H
|
||||
|
||||
#include <stdlib.h>
|
||||
#include <math.h>
|
||||
#include "arch.h"
|
||||
|
||||
#include <stdlib.h>
|
||||
#define opus_alloc(x) malloc(x)
|
||||
#define opus_free(x) free(x)
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
#ifdef USE_SIMD
|
||||
# include <xmmintrin.h>
|
||||
# define kiss_fft_scalar __m128
|
||||
#define KISS_FFT_MALLOC(nbytes) memalign(16,nbytes)
|
||||
#else
|
||||
#define KISS_FFT_MALLOC opus_alloc
|
||||
#endif
|
||||
|
||||
#ifdef FIXED_POINT
|
||||
#include "arch.h"
|
||||
|
||||
# define kiss_fft_scalar opus_int32
|
||||
# define kiss_twiddle_scalar opus_int16
|
||||
|
||||
|
||||
#else
|
||||
# ifndef kiss_fft_scalar
|
||||
/* default is float */
|
||||
# define kiss_fft_scalar float
|
||||
# define kiss_twiddle_scalar float
|
||||
# define KF_SUFFIX _celt_single
|
||||
# endif
|
||||
#endif
|
||||
|
||||
typedef struct {
|
||||
kiss_fft_scalar r;
|
||||
kiss_fft_scalar i;
|
||||
}kiss_fft_cpx;
|
||||
|
||||
typedef struct {
|
||||
kiss_twiddle_scalar r;
|
||||
kiss_twiddle_scalar i;
|
||||
}kiss_twiddle_cpx;
|
||||
|
||||
#define MAXFACTORS 8
|
||||
/* e.g. an fft of length 128 has 4 factors
|
||||
as far as kissfft is concerned
|
||||
4*4*4*2
|
||||
*/
|
||||
|
||||
typedef struct arch_fft_state{
|
||||
int is_supported;
|
||||
void *priv;
|
||||
} arch_fft_state;
|
||||
|
||||
typedef struct kiss_fft_state{
|
||||
int nfft;
|
||||
opus_val16 scale;
|
||||
#ifdef FIXED_POINT
|
||||
int scale_shift;
|
||||
#endif
|
||||
int shift;
|
||||
opus_int16 factors[2*MAXFACTORS];
|
||||
const opus_int32 *bitrev;
|
||||
const kiss_twiddle_cpx *twiddles;
|
||||
arch_fft_state *arch_fft;
|
||||
} kiss_fft_state;
|
||||
|
||||
#if defined(HAVE_ARM_NE10)
|
||||
#include "arm/fft_arm.h"
|
||||
#endif
|
||||
|
||||
/*typedef struct kiss_fft_state* kiss_fft_cfg;*/
|
||||
|
||||
/**
|
||||
* opus_fft_alloc
|
||||
*
|
||||
* Initialize a FFT (or IFFT) algorithm's cfg/state buffer.
|
||||
*
|
||||
* typical usage: kiss_fft_cfg mycfg=opus_fft_alloc(1024,0,NULL,NULL);
|
||||
*
|
||||
* The return value from fft_alloc is a cfg buffer used internally
|
||||
* by the fft routine or NULL.
|
||||
*
|
||||
* If lenmem is NULL, then opus_fft_alloc will allocate a cfg buffer using malloc.
|
||||
* The returned value should be free()d when done to avoid memory leaks.
|
||||
*
|
||||
* The state can be placed in a user supplied buffer 'mem':
|
||||
* If lenmem is not NULL and mem is not NULL and *lenmem is large enough,
|
||||
* then the function places the cfg in mem and the size used in *lenmem
|
||||
* and returns mem.
|
||||
*
|
||||
* If lenmem is not NULL and ( mem is NULL or *lenmem is not large enough),
|
||||
* then the function returns NULL and places the minimum cfg
|
||||
* buffer size in *lenmem.
|
||||
* */
|
||||
|
||||
kiss_fft_state *rnn_fft_alloc_twiddles(int nfft,void * mem,size_t * lenmem, const kiss_fft_state *base, int arch);
|
||||
|
||||
kiss_fft_state *rnn_fft_alloc(int nfft,void * mem,size_t * lenmem, int arch);
|
||||
|
||||
/**
|
||||
* opus_fft(cfg,in_out_buf)
|
||||
*
|
||||
* Perform an FFT on a complex input buffer.
|
||||
* for a forward FFT,
|
||||
* fin should be f[0] , f[1] , ... ,f[nfft-1]
|
||||
* fout will be F[0] , F[1] , ... ,F[nfft-1]
|
||||
* Note that each element is complex and can be accessed like
|
||||
f[k].r and f[k].i
|
||||
* */
|
||||
void rnn_fft_c(const kiss_fft_state *cfg,const kiss_fft_cpx *fin,kiss_fft_cpx *fout);
|
||||
void rnn_ifft_c(const kiss_fft_state *cfg,const kiss_fft_cpx *fin,kiss_fft_cpx *fout);
|
||||
|
||||
void rnn_fft_impl(const kiss_fft_state *st,kiss_fft_cpx *fout);
|
||||
void rnn_ifft_impl(const kiss_fft_state *st,kiss_fft_cpx *fout);
|
||||
|
||||
void rnn_fft_free(const kiss_fft_state *cfg, int arch);
|
||||
|
||||
|
||||
void rnn_fft_free_arch_c(kiss_fft_state *st);
|
||||
int rnn_fft_alloc_arch_c(kiss_fft_state *st);
|
||||
|
||||
#if !defined(OVERRIDE_OPUS_FFT)
|
||||
/* Is run-time CPU detection enabled on this platform? */
|
||||
#if defined(OPUS_HAVE_RTCD) && (defined(HAVE_ARM_NE10))
|
||||
|
||||
extern int (*const OPUS_FFT_ALLOC_ARCH_IMPL[OPUS_ARCHMASK+1])(
|
||||
kiss_fft_state *st);
|
||||
|
||||
#define opus_fft_alloc_arch(_st, arch) \
|
||||
((*OPUS_FFT_ALLOC_ARCH_IMPL[(arch)&OPUS_ARCHMASK])(_st))
|
||||
|
||||
extern void (*const OPUS_FFT_FREE_ARCH_IMPL[OPUS_ARCHMASK+1])(
|
||||
kiss_fft_state *st);
|
||||
#define opus_fft_free_arch(_st, arch) \
|
||||
((*OPUS_FFT_FREE_ARCH_IMPL[(arch)&OPUS_ARCHMASK])(_st))
|
||||
|
||||
extern void (*const OPUS_FFT[OPUS_ARCHMASK+1])(const kiss_fft_state *cfg,
|
||||
const kiss_fft_cpx *fin, kiss_fft_cpx *fout);
|
||||
#define opus_fft(_cfg, _fin, _fout, arch) \
|
||||
((*OPUS_FFT[(arch)&OPUS_ARCHMASK])(_cfg, _fin, _fout))
|
||||
|
||||
extern void (*const OPUS_IFFT[OPUS_ARCHMASK+1])(const kiss_fft_state *cfg,
|
||||
const kiss_fft_cpx *fin, kiss_fft_cpx *fout);
|
||||
#define opus_ifft(_cfg, _fin, _fout, arch) \
|
||||
((*OPUS_IFFT[(arch)&OPUS_ARCHMASK])(_cfg, _fin, _fout))
|
||||
|
||||
#else /* else for if defined(OPUS_HAVE_RTCD) && (defined(HAVE_ARM_NE10)) */
|
||||
|
||||
#define rnn_fft_alloc_arch(_st, arch) \
|
||||
((void)(arch), rnn_fft_alloc_arch_c(_st))
|
||||
|
||||
#define rnn_fft_free_arch(_st, arch) \
|
||||
((void)(arch), rnn_fft_free_arch_c(_st))
|
||||
|
||||
#define rnn_fft(_cfg, _fin, _fout, arch) \
|
||||
((void)(arch), rnn_fft_c(_cfg, _fin, _fout))
|
||||
|
||||
#define rnn_ifft(_cfg, _fin, _fout, arch) \
|
||||
((void)(arch), rnn_ifft_c(_cfg, _fin, _fout))
|
||||
|
||||
#endif /* end if defined(OPUS_HAVE_RTCD) && (defined(HAVE_ARM_NE10)) */
|
||||
#endif /* end if !defined(OVERRIDE_OPUS_FFT) */
|
||||
|
||||
#ifdef __cplusplus
|
||||
}
|
||||
#endif
|
||||
|
||||
#endif
|
||||
123
src/modules/ZoomIt/ZoomIt/rnnoise/nnet.c
Normal file
123
src/modules/ZoomIt/ZoomIt/rnnoise/nnet.c
Normal file
@@ -0,0 +1,123 @@
|
||||
/* Copyright (c) 2018 Mozilla
|
||||
2008-2011 Octasic Inc.
|
||||
2012-2017 Jean-Marc Valin */
|
||||
/*
|
||||
Redistribution and use in source and binary forms, with or without
|
||||
modification, are permitted provided that the following conditions
|
||||
are met:
|
||||
|
||||
- Redistributions of source code must retain the above copyright
|
||||
notice, this list of conditions and the following disclaimer.
|
||||
|
||||
- Redistributions in binary form must reproduce the above copyright
|
||||
notice, this list of conditions and the following disclaimer in the
|
||||
documentation and/or other materials provided with the distribution.
|
||||
|
||||
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||||
``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
||||
LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
||||
A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR
|
||||
CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
|
||||
EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
|
||||
PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
|
||||
PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
|
||||
LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
|
||||
NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
||||
SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
*/
|
||||
|
||||
#ifdef HAVE_CONFIG_H
|
||||
#include "config.h"
|
||||
#endif
|
||||
|
||||
#include <stdlib.h>
|
||||
#include <math.h>
|
||||
#include "opus_types.h"
|
||||
#include "arch.h"
|
||||
#include "nnet.h"
|
||||
#include "common.h"
|
||||
#include "vec.h"
|
||||
|
||||
#ifdef ENABLE_OSCE
|
||||
#include "osce.h"
|
||||
#endif
|
||||
|
||||
#ifdef NO_OPTIMIZATIONS
|
||||
#if defined(_MSC_VER)
|
||||
#pragma message ("Compiling without any vectorization. This code will be very slow")
|
||||
#else
|
||||
#warning Compiling without any vectorization. This code will be very slow
|
||||
#endif
|
||||
#endif
|
||||
|
||||
|
||||
#define SOFTMAX_HACK
|
||||
|
||||
|
||||
void compute_generic_dense(const LinearLayer *layer, float *output, const float *input, int activation, int arch)
|
||||
{
|
||||
compute_linear(layer, output, input, arch);
|
||||
compute_activation(output, output, layer->nb_outputs, activation, arch);
|
||||
}
|
||||
|
||||
#define MAX_RNN_NEURONS_ALL 1024
|
||||
|
||||
void compute_generic_gru(const LinearLayer *input_weights, const LinearLayer *recurrent_weights, float *state, const float *in, int arch)
|
||||
{
|
||||
int i;
|
||||
int N;
|
||||
float zrh[3*MAX_RNN_NEURONS_ALL];
|
||||
float recur[3*MAX_RNN_NEURONS_ALL];
|
||||
float *z;
|
||||
float *r;
|
||||
float *h;
|
||||
celt_assert(3*recurrent_weights->nb_inputs == recurrent_weights->nb_outputs);
|
||||
celt_assert(input_weights->nb_outputs == recurrent_weights->nb_outputs);
|
||||
N = recurrent_weights->nb_inputs;
|
||||
z = zrh;
|
||||
r = &zrh[N];
|
||||
h = &zrh[2*N];
|
||||
celt_assert(recurrent_weights->nb_outputs <= 3*MAX_RNN_NEURONS_ALL);
|
||||
celt_assert(in != state);
|
||||
compute_linear(input_weights, zrh, in, arch);
|
||||
compute_linear(recurrent_weights, recur, state, arch);
|
||||
for (i=0;i<2*N;i++)
|
||||
zrh[i] += recur[i];
|
||||
compute_activation(zrh, zrh, 2*N, ACTIVATION_SIGMOID, arch);
|
||||
for (i=0;i<N;i++)
|
||||
h[i] += recur[2*N+i]*r[i];
|
||||
compute_activation(h, h, N, ACTIVATION_TANH, arch);
|
||||
for (i=0;i<N;i++)
|
||||
h[i] = z[i]*state[i] + (1-z[i])*h[i];
|
||||
for (i=0;i<N;i++)
|
||||
state[i] = h[i];
|
||||
}
|
||||
|
||||
void compute_glu(const LinearLayer *layer, float *output, const float *input, int arch)
|
||||
{
|
||||
int i;
|
||||
float act2[MAX_INPUTS];
|
||||
celt_assert(layer->nb_inputs == layer->nb_outputs);
|
||||
compute_linear(layer, act2, input, arch);
|
||||
compute_activation(act2, act2, layer->nb_outputs, ACTIVATION_SIGMOID, arch);
|
||||
if (input == output) {
|
||||
/* Give a vectorization hint to the compiler for the in-place case. */
|
||||
for (i=0;i<layer->nb_outputs;i++) output[i] = output[i]*act2[i];
|
||||
} else {
|
||||
for (i=0;i<layer->nb_outputs;i++) output[i] = input[i]*act2[i];
|
||||
}
|
||||
}
|
||||
|
||||
#define MAX_CONV_INPUTS_ALL 1024
|
||||
|
||||
void compute_generic_conv1d(const LinearLayer *layer, float *output, float *mem, const float *input, int input_size, int activation, int arch)
|
||||
{
|
||||
float tmp[MAX_CONV_INPUTS_ALL];
|
||||
celt_assert(input != output);
|
||||
celt_assert(layer->nb_inputs <= MAX_CONV_INPUTS_ALL);
|
||||
if (layer->nb_inputs!=input_size) RNN_COPY(tmp, mem, layer->nb_inputs-input_size);
|
||||
RNN_COPY(&tmp[layer->nb_inputs-input_size], input, input_size);
|
||||
compute_linear(layer, output, tmp, arch);
|
||||
compute_activation(output, output, layer->nb_outputs, activation, arch);
|
||||
if (layer->nb_inputs!=input_size) RNN_COPY(mem, &tmp[input_size], layer->nb_inputs-input_size);
|
||||
}
|
||||
169
src/modules/ZoomIt/ZoomIt/rnnoise/nnet.h
Normal file
169
src/modules/ZoomIt/ZoomIt/rnnoise/nnet.h
Normal file
@@ -0,0 +1,169 @@
|
||||
/* Copyright (c) 2018 Mozilla
|
||||
Copyright (c) 2017 Jean-Marc Valin */
|
||||
/*
|
||||
Redistribution and use in source and binary forms, with or without
|
||||
modification, are permitted provided that the following conditions
|
||||
are met:
|
||||
|
||||
- Redistributions of source code must retain the above copyright
|
||||
notice, this list of conditions and the following disclaimer.
|
||||
|
||||
- Redistributions in binary form must reproduce the above copyright
|
||||
notice, this list of conditions and the following disclaimer in the
|
||||
documentation and/or other materials provided with the distribution.
|
||||
|
||||
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||||
``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
||||
LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
||||
A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR
|
||||
CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
|
||||
EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
|
||||
PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
|
||||
PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
|
||||
LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
|
||||
NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
||||
SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
*/
|
||||
|
||||
#ifndef NNET_H_
|
||||
#define NNET_H_
|
||||
|
||||
#include <stddef.h>
|
||||
#include "opus_types.h"
|
||||
|
||||
#define ACTIVATION_LINEAR 0
|
||||
#define ACTIVATION_SIGMOID 1
|
||||
#define ACTIVATION_TANH 2
|
||||
#define ACTIVATION_RELU 3
|
||||
#define ACTIVATION_SOFTMAX 4
|
||||
#define ACTIVATION_SWISH 5
|
||||
|
||||
#define WEIGHT_BLOB_VERSION 0
|
||||
#define WEIGHT_BLOCK_SIZE 64
|
||||
typedef struct {
|
||||
const char *name;
|
||||
int type;
|
||||
int size;
|
||||
const void *data;
|
||||
} WeightArray;
|
||||
|
||||
#define WEIGHT_TYPE_float 0
|
||||
#define WEIGHT_TYPE_int 1
|
||||
#define WEIGHT_TYPE_qweight 2
|
||||
#define WEIGHT_TYPE_int8 3
|
||||
|
||||
typedef struct {
|
||||
char head[4];
|
||||
int version;
|
||||
int type;
|
||||
int size;
|
||||
int block_size;
|
||||
char name[44];
|
||||
} WeightHead;
|
||||
|
||||
/* Generic sparse affine transformation. */
|
||||
typedef struct {
|
||||
const float *bias;
|
||||
const float *subias;
|
||||
const opus_int8 *weights;
|
||||
const float *float_weights;
|
||||
const int *weights_idx;
|
||||
const float *diag;
|
||||
const float *scale;
|
||||
int nb_inputs;
|
||||
int nb_outputs;
|
||||
} LinearLayer;
|
||||
|
||||
/* Generic sparse affine transformation. */
|
||||
typedef struct {
|
||||
const float *bias;
|
||||
const float *float_weights;
|
||||
int in_channels;
|
||||
int out_channels;
|
||||
int ktime;
|
||||
int kheight;
|
||||
} Conv2dLayer;
|
||||
|
||||
|
||||
/* Changes some symbol names to add the rnn_ prefix so we don't get conflicts with Opus. */
|
||||
#define linear_init rnn_linear_init
|
||||
#define conv2d_init rnn_conv2d_init
|
||||
#define compute_generic_dense rnn_compute_generic_dense
|
||||
#define compute_generic_gru rnn_compute_generic_gru
|
||||
#define compute_generic_conv1d rnn_compute_generic_conv1d
|
||||
#define compute_glu rnn_compute_glu
|
||||
|
||||
#define parse_weights rnn_parse_weights
|
||||
|
||||
#define compute_linear_c rnn_compute_linear_c
|
||||
#define compute_activation_c rnn_compute_activation_c
|
||||
#define compute_conv2d_c rnn_compute_conv2d_c
|
||||
#define compute_linear_sse4_1 rnn_compute_linear_sse4_1
|
||||
#define compute_activation_sse4_1 rnn_compute_activation_sse4_1
|
||||
#define compute_conv2d_sse4_1 rnn_compute_conv2d_sse4_1
|
||||
#define compute_linear_avx2 rnn_compute_linear_avx2
|
||||
#define compute_activation_avx2 rnn_compute_activation_avx2
|
||||
#define compute_conv2d_avx2 rnn_compute_conv2d_avx2
|
||||
|
||||
|
||||
void compute_generic_dense(const LinearLayer *layer, float *output, const float *input, int activation, int arch);
|
||||
void compute_generic_gru(const LinearLayer *input_weights, const LinearLayer *recurrent_weights, float *state, const float *in, int arch);
|
||||
void compute_generic_conv1d(const LinearLayer *layer, float *output, float *mem, const float *input, int input_size, int activation, int arch);
|
||||
void compute_glu(const LinearLayer *layer, float *output, const float *input, int arch);
|
||||
|
||||
|
||||
int parse_weights(WeightArray **list, const void *data, int len);
|
||||
|
||||
|
||||
|
||||
int linear_init(LinearLayer *layer, const WeightArray *arrays,
|
||||
const char *bias,
|
||||
const char *subias,
|
||||
const char *weights,
|
||||
const char *float_weights,
|
||||
const char *weights_idx,
|
||||
const char *diag,
|
||||
const char *scale,
|
||||
int nb_inputs,
|
||||
int nb_outputs);
|
||||
|
||||
int conv2d_init(Conv2dLayer *layer, const WeightArray *arrays,
|
||||
const char *bias,
|
||||
const char *float_weights,
|
||||
int in_channels,
|
||||
int out_channels,
|
||||
int ktime,
|
||||
int kheight);
|
||||
|
||||
|
||||
void compute_linear_c(const LinearLayer *linear, float *out, const float *in);
|
||||
void compute_activation_c(float *output, const float *input, int N, int activation);
|
||||
void compute_conv2d_c(const Conv2dLayer *conv, float *out, float *mem, const float *in, int height, int hstride, int activation);
|
||||
|
||||
#ifdef RNN_ENABLE_X86_RTCD
|
||||
#include "x86/dnn_x86.h"
|
||||
#endif
|
||||
|
||||
#ifndef OVERRIDE_COMPUTE_LINEAR
|
||||
#define compute_linear(linear, out, in, arch) ((void)(arch),compute_linear_c(linear, out, in))
|
||||
#endif
|
||||
|
||||
#ifndef OVERRIDE_COMPUTE_ACTIVATION
|
||||
#define compute_activation(output, input, N, activation, arch) ((void)(arch),compute_activation_c(output, input, N, activation))
|
||||
#endif
|
||||
|
||||
#ifndef OVERRIDE_COMPUTE_CONV2D
|
||||
#define compute_conv2d(conv, out, mem, in, height, hstride, activation, arch) ((void)(arch),compute_conv2d_c(conv, out, mem, in, height, hstride, activation))
|
||||
#endif
|
||||
|
||||
#if defined(__x86_64__) && !defined(RNN_ENABLE_X86_RTCD) && !defined(__AVX2__)
|
||||
#if defined(_MSC_VER)
|
||||
#pragma message ("Only SSE and SSE2 are available. On newer machines, enable SSSE3/AVX/AVX2 to get better performance")
|
||||
#else
|
||||
#warning "Only SSE and SSE2 are available. On newer machines, enable SSSE3/AVX/AVX2 using -march= to get better performance"
|
||||
#endif
|
||||
#endif
|
||||
|
||||
|
||||
|
||||
#endif /* NNET_H_ */
|
||||
257
src/modules/ZoomIt/ZoomIt/rnnoise/nnet_arch.h
Normal file
257
src/modules/ZoomIt/ZoomIt/rnnoise/nnet_arch.h
Normal file
@@ -0,0 +1,257 @@
|
||||
/* Copyright (c) 2018-2019 Mozilla
|
||||
2023 Amazon */
|
||||
/*
|
||||
Redistribution and use in source and binary forms, with or without
|
||||
modification, are permitted provided that the following conditions
|
||||
are met:
|
||||
|
||||
- Redistributions of source code must retain the above copyright
|
||||
notice, this list of conditions and the following disclaimer.
|
||||
|
||||
- Redistributions in binary form must reproduce the above copyright
|
||||
notice, this list of conditions and the following disclaimer in the
|
||||
documentation and/or other materials provided with the distribution.
|
||||
|
||||
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||||
``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
||||
LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
||||
A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR
|
||||
CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
|
||||
EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
|
||||
PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
|
||||
PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
|
||||
LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
|
||||
NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
||||
SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
*/
|
||||
|
||||
#ifndef NNET_ARCH_H
|
||||
#define NNET_ARCH_H
|
||||
|
||||
#include "nnet.h"
|
||||
#include "arch.h"
|
||||
#include "common.h"
|
||||
#include "vec.h"
|
||||
|
||||
#define CAT_SUFFIX2(a,b) a ## b
|
||||
#define CAT_SUFFIX(a,b) CAT_SUFFIX2(a, b)
|
||||
|
||||
#define RTCD_SUF(name) CAT_SUFFIX(name, RTCD_ARCH)
|
||||
|
||||
# if !defined(OPUS_GNUC_PREREQ)
|
||||
# if defined(__GNUC__)&&defined(__GNUC_MINOR__)
|
||||
# define OPUS_GNUC_PREREQ(_maj,_min) \
|
||||
((__GNUC__<<16)+__GNUC_MINOR__>=((_maj)<<16)+(_min))
|
||||
# else
|
||||
# define OPUS_GNUC_PREREQ(_maj,_min) 0
|
||||
# endif
|
||||
# endif
|
||||
|
||||
|
||||
/* Force vectorization on for DNN code because some of the loops rely on
|
||||
compiler vectorization rather than explicitly using intrinsics. */
|
||||
#if OPUS_GNUC_PREREQ(5,1)
|
||||
#define GCC_POP_OPTIONS
|
||||
#pragma GCC push_options
|
||||
#pragma GCC optimize("tree-vectorize")
|
||||
#endif
|
||||
|
||||
|
||||
#define MAX_ACTIVATIONS (4096)
|
||||
|
||||
static OPUS_INLINE void vec_swish(float *y, const float *x, int N)
|
||||
{
|
||||
int i;
|
||||
float tmp[MAX_ACTIVATIONS];
|
||||
celt_assert(N <= MAX_ACTIVATIONS);
|
||||
vec_sigmoid(tmp, x, N);
|
||||
for (i=0;i<N;i++)
|
||||
y[i] = x[i]*tmp[i];
|
||||
}
|
||||
|
||||
static OPUS_INLINE float relu(float x)
|
||||
{
|
||||
return x < 0 ? 0 : x;
|
||||
}
|
||||
|
||||
/*#define HIGH_ACCURACY */
|
||||
|
||||
void RTCD_SUF(compute_activation_)(float *output, const float *input, int N, int activation)
|
||||
{
|
||||
int i;
|
||||
if (activation == ACTIVATION_SIGMOID) {
|
||||
#ifdef HIGH_ACCURACY
|
||||
for (int n=0; n<N; n++)
|
||||
{
|
||||
output[n] = 1.f / (1 + exp(-input[n]));
|
||||
}
|
||||
#else
|
||||
vec_sigmoid(output, input, N);
|
||||
#endif
|
||||
} else if (activation == ACTIVATION_TANH) {
|
||||
#ifdef HIGH_ACCURACY
|
||||
for (int n=0; n<N; n++)
|
||||
{
|
||||
output[n] = tanh(input[n]);
|
||||
}
|
||||
#else
|
||||
vec_tanh(output, input, N);
|
||||
#endif
|
||||
} else if (activation == ACTIVATION_SWISH) {
|
||||
vec_swish(output, input, N);
|
||||
} else if (activation == ACTIVATION_RELU) {
|
||||
for (i=0;i<N;i++)
|
||||
output[i] = relu(input[i]);
|
||||
} else if (activation == ACTIVATION_SOFTMAX) {
|
||||
#ifdef SOFTMAX_HACK
|
||||
RNN_COPY(output, input, N);
|
||||
/*for (i=0;i<N;i++)
|
||||
output[i] = input[i];*/
|
||||
#else
|
||||
float sum = 0;
|
||||
softmax(output, input, N);
|
||||
for (i=0;i<N;i++) {
|
||||
sum += output[i];
|
||||
}
|
||||
sum = 1.f/(sum+1e-30);
|
||||
for (i=0;i<N;i++)
|
||||
output[i] = sum*output[i];
|
||||
#endif
|
||||
} else {
|
||||
celt_assert(activation == ACTIVATION_LINEAR);
|
||||
if (input != output) {
|
||||
for (i=0;i<N;i++)
|
||||
output[i] = input[i];
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
void RTCD_SUF(compute_linear_) (const LinearLayer *linear, float *out, const float *in)
|
||||
{
|
||||
int i, M, N;
|
||||
const float *bias;
|
||||
celt_assert(in != out);
|
||||
bias = linear->bias;
|
||||
M = linear->nb_inputs;
|
||||
N = linear->nb_outputs;
|
||||
if (linear->float_weights != NULL) {
|
||||
if (linear->weights_idx != NULL) sparse_sgemv8x4(out, linear->float_weights, linear->weights_idx, N, in);
|
||||
else sgemv(out, linear->float_weights, N, M, N, in);
|
||||
} else if (linear->weights != NULL) {
|
||||
if (linear->weights_idx != NULL) sparse_cgemv8x4(out, linear->weights, linear->weights_idx, linear->scale, N, M, in);
|
||||
else cgemv8x4(out, linear->weights, linear->scale, N, M, in);
|
||||
/* Only use SU biases on for integer matrices on SU archs. */
|
||||
#ifdef USE_SU_BIAS
|
||||
bias = linear->subias;
|
||||
#endif
|
||||
}
|
||||
else RNN_CLEAR(out, N);
|
||||
if (bias != NULL) {
|
||||
for (i=0;i<N;i++) out[i] += bias[i];
|
||||
}
|
||||
if (linear->diag) {
|
||||
/* Diag is only used for GRU recurrent weights. */
|
||||
celt_assert(3*M == N);
|
||||
for (i=0;i<M;i++) {
|
||||
out[i] += linear->diag[i]*in[i];
|
||||
out[i+M] += linear->diag[i+M]*in[i];
|
||||
out[i+2*M] += linear->diag[i+2*M]*in[i];
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/* Computes non-padded convolution for input [ ksize1 x in_channels x (len2+ksize2) ],
|
||||
kernel [ out_channels x in_channels x ksize1 x ksize2 ],
|
||||
storing the output as [ out_channels x len2 ].
|
||||
We assume that the output dimension along the ksize1 axis is 1,
|
||||
i.e. processing one frame at a time. */
|
||||
static void conv2d_float(float *out, const float *weights, int in_channels, int out_channels, int ktime, int kheight, const float *in, int height, int hstride)
|
||||
{
|
||||
int i;
|
||||
int in_stride;
|
||||
in_stride = height+kheight-1;
|
||||
for (i=0;i<out_channels;i++) {
|
||||
int m;
|
||||
RNN_CLEAR(&out[i*hstride], height);
|
||||
for (m=0;m<in_channels;m++) {
|
||||
int t;
|
||||
for (t=0;t<ktime;t++) {
|
||||
int h;
|
||||
for (h=0;h<kheight;h++) {
|
||||
int j;
|
||||
for (j=0;j<height;j++) {
|
||||
out[i*hstride + j] += weights[i*in_channels*ktime*kheight + m*ktime*kheight + t*kheight + h] *
|
||||
in[t*in_channels*in_stride + m*in_stride + j + h];
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/* There's no intrinsics in this function (or the one above) because the gcc (and hopefully other compiler) auto-vectorizer is smart enough to
|
||||
produce the right code by itself based on the compile flags. */
|
||||
static void conv2d_3x3_float(float *out, const float *weights, int in_channels, int out_channels, const float *in, int height, int hstride)
|
||||
{
|
||||
int i;
|
||||
int in_stride;
|
||||
int kheight, ktime;
|
||||
kheight = ktime = 3;
|
||||
in_stride = height+kheight-1;
|
||||
for (i=0;i<out_channels;i++) {
|
||||
int m;
|
||||
RNN_CLEAR(&out[i*hstride], height);
|
||||
for (m=0;m<in_channels;m++) {
|
||||
int j;
|
||||
for (j=0;j<height;j++) {
|
||||
/* Unrolled version of previous function -- compiler will figure out the indexing simplifications. */
|
||||
out[i*hstride + j] += weights[i*in_channels*ktime*kheight + m*ktime*kheight + 0*kheight + 0]*in[0*in_channels*in_stride + m*in_stride + j + 0]
|
||||
+ weights[i*in_channels*ktime*kheight + m*ktime*kheight + 0*kheight + 1]*in[0*in_channels*in_stride + m*in_stride + j + 1]
|
||||
+ weights[i*in_channels*ktime*kheight + m*ktime*kheight + 0*kheight + 2]*in[0*in_channels*in_stride + m*in_stride + j + 2]
|
||||
+ weights[i*in_channels*ktime*kheight + m*ktime*kheight + 1*kheight + 0]*in[1*in_channels*in_stride + m*in_stride + j + 0]
|
||||
+ weights[i*in_channels*ktime*kheight + m*ktime*kheight + 1*kheight + 1]*in[1*in_channels*in_stride + m*in_stride + j + 1]
|
||||
+ weights[i*in_channels*ktime*kheight + m*ktime*kheight + 1*kheight + 2]*in[1*in_channels*in_stride + m*in_stride + j + 2]
|
||||
+ weights[i*in_channels*ktime*kheight + m*ktime*kheight + 2*kheight + 0]*in[2*in_channels*in_stride + m*in_stride + j + 0]
|
||||
+ weights[i*in_channels*ktime*kheight + m*ktime*kheight + 2*kheight + 1]*in[2*in_channels*in_stride + m*in_stride + j + 1]
|
||||
+ weights[i*in_channels*ktime*kheight + m*ktime*kheight + 2*kheight + 2]*in[2*in_channels*in_stride + m*in_stride + j + 2];
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#define MAX_CONV2D_INPUTS 8192
|
||||
|
||||
void RTCD_SUF(compute_conv2d_)(const Conv2dLayer *conv, float *out, float *mem, const float *in, int height, int hstride, int activation)
|
||||
{
|
||||
int i;
|
||||
const float *bias;
|
||||
float in_buf[MAX_CONV2D_INPUTS];
|
||||
int time_stride;
|
||||
celt_assert(in != out);
|
||||
time_stride = conv->in_channels*(height+conv->kheight-1);
|
||||
celt_assert(conv->ktime*time_stride <= MAX_CONV2D_INPUTS);
|
||||
RNN_COPY(in_buf, mem, (conv->ktime-1)*time_stride);
|
||||
RNN_COPY(&in_buf[(conv->ktime-1)*time_stride], in, time_stride);
|
||||
RNN_COPY(mem, &in_buf[time_stride], (conv->ktime-1)*time_stride);
|
||||
bias = conv->bias;
|
||||
if (conv->kheight == 3 && conv->ktime == 3)
|
||||
conv2d_3x3_float(out, conv->float_weights, conv->in_channels, conv->out_channels, in_buf, height, hstride);
|
||||
else
|
||||
conv2d_float(out, conv->float_weights, conv->in_channels, conv->out_channels, conv->ktime, conv->kheight, in_buf, height, hstride);
|
||||
if (bias != NULL) {
|
||||
for (i=0;i<conv->out_channels;i++) {
|
||||
int j;
|
||||
for (j=0;j<height;j++) out[i*hstride+j] += bias[i];
|
||||
}
|
||||
}
|
||||
for (i=0;i<conv->out_channels;i++) {
|
||||
RTCD_SUF(compute_activation_)(&out[i*hstride], &out[i*hstride], height, activation);
|
||||
}
|
||||
}
|
||||
|
||||
#ifdef GCC_POP_OPTIONS
|
||||
#pragma GCC pop_options
|
||||
#endif
|
||||
|
||||
#endif
|
||||
35
src/modules/ZoomIt/ZoomIt/rnnoise/nnet_default.c
Normal file
35
src/modules/ZoomIt/ZoomIt/rnnoise/nnet_default.c
Normal file
@@ -0,0 +1,35 @@
|
||||
/* Copyright (c) 2018-2019 Mozilla
|
||||
2023 Amazon */
|
||||
/*
|
||||
Redistribution and use in source and binary forms, with or without
|
||||
modification, are permitted provided that the following conditions
|
||||
are met:
|
||||
|
||||
- Redistributions of source code must retain the above copyright
|
||||
notice, this list of conditions and the following disclaimer.
|
||||
|
||||
- Redistributions in binary form must reproduce the above copyright
|
||||
notice, this list of conditions and the following disclaimer in the
|
||||
documentation and/or other materials provided with the distribution.
|
||||
|
||||
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||||
``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
||||
LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
||||
A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR
|
||||
CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
|
||||
EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
|
||||
PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
|
||||
PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
|
||||
LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
|
||||
NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
||||
SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
*/
|
||||
|
||||
#ifdef HAVE_CONFIG_H
|
||||
#include "config.h"
|
||||
#endif
|
||||
|
||||
|
||||
#define RTCD_ARCH c
|
||||
|
||||
#include "nnet_arch.h"
|
||||
159
src/modules/ZoomIt/ZoomIt/rnnoise/opus_types.h
Normal file
159
src/modules/ZoomIt/ZoomIt/rnnoise/opus_types.h
Normal file
@@ -0,0 +1,159 @@
|
||||
/* (C) COPYRIGHT 1994-2002 Xiph.Org Foundation */
|
||||
/* Modified by Jean-Marc Valin */
|
||||
/*
|
||||
Redistribution and use in source and binary forms, with or without
|
||||
modification, are permitted provided that the following conditions
|
||||
are met:
|
||||
|
||||
- Redistributions of source code must retain the above copyright
|
||||
notice, this list of conditions and the following disclaimer.
|
||||
|
||||
- Redistributions in binary form must reproduce the above copyright
|
||||
notice, this list of conditions and the following disclaimer in the
|
||||
documentation and/or other materials provided with the distribution.
|
||||
|
||||
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||||
``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
||||
LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
||||
A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER
|
||||
OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
|
||||
EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
|
||||
PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
|
||||
PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
|
||||
LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
|
||||
NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
||||
SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
*/
|
||||
/* opus_types.h based on ogg_types.h from libogg */
|
||||
|
||||
/**
|
||||
@file opus_types.h
|
||||
@brief Opus reference implementation types
|
||||
*/
|
||||
#ifndef OPUS_TYPES_H
|
||||
#define OPUS_TYPES_H
|
||||
|
||||
/* Use the real stdint.h if it's there (taken from Paul Hsieh's pstdint.h) */
|
||||
#if (defined(__STDC__) && __STDC__ && defined(__STDC_VERSION__) && __STDC_VERSION__ >= 199901L) || (defined(__GNUC__) && (defined(_STDINT_H) || defined(_STDINT_H_)) || defined (HAVE_STDINT_H))
|
||||
#include <stdint.h>
|
||||
|
||||
typedef int16_t opus_int16;
|
||||
typedef uint16_t opus_uint16;
|
||||
typedef int32_t opus_int32;
|
||||
typedef uint32_t opus_uint32;
|
||||
#elif defined(_WIN32)
|
||||
|
||||
# if defined(__CYGWIN__)
|
||||
# include <_G_config.h>
|
||||
typedef _G_int32_t opus_int32;
|
||||
typedef _G_uint32_t opus_uint32;
|
||||
typedef _G_int16 opus_int16;
|
||||
typedef _G_uint16 opus_uint16;
|
||||
# elif defined(__MINGW32__)
|
||||
typedef short opus_int16;
|
||||
typedef unsigned short opus_uint16;
|
||||
typedef int opus_int32;
|
||||
typedef unsigned int opus_uint32;
|
||||
# elif defined(__MWERKS__)
|
||||
typedef int opus_int32;
|
||||
typedef unsigned int opus_uint32;
|
||||
typedef short opus_int16;
|
||||
typedef unsigned short opus_uint16;
|
||||
# else
|
||||
/* MSVC/Borland */
|
||||
typedef __int32 opus_int32;
|
||||
typedef unsigned __int32 opus_uint32;
|
||||
typedef __int16 opus_int16;
|
||||
typedef unsigned __int16 opus_uint16;
|
||||
# endif
|
||||
|
||||
#elif defined(__MACOS__)
|
||||
|
||||
# include <sys/types.h>
|
||||
typedef SInt16 opus_int16;
|
||||
typedef UInt16 opus_uint16;
|
||||
typedef SInt32 opus_int32;
|
||||
typedef UInt32 opus_uint32;
|
||||
|
||||
#elif (defined(__APPLE__) && defined(__MACH__)) /* MacOS X Framework build */
|
||||
|
||||
# include <sys/types.h>
|
||||
typedef int16_t opus_int16;
|
||||
typedef u_int16_t opus_uint16;
|
||||
typedef int32_t opus_int32;
|
||||
typedef u_int32_t opus_uint32;
|
||||
|
||||
#elif defined(__BEOS__)
|
||||
|
||||
/* Be */
|
||||
# include <inttypes.h>
|
||||
typedef int16 opus_int16;
|
||||
typedef u_int16 opus_uint16;
|
||||
typedef int32_t opus_int32;
|
||||
typedef u_int32_t opus_uint32;
|
||||
|
||||
#elif defined (__EMX__)
|
||||
|
||||
/* OS/2 GCC */
|
||||
typedef short opus_int16;
|
||||
typedef unsigned short opus_uint16;
|
||||
typedef int opus_int32;
|
||||
typedef unsigned int opus_uint32;
|
||||
|
||||
#elif defined (DJGPP)
|
||||
|
||||
/* DJGPP */
|
||||
typedef short opus_int16;
|
||||
typedef unsigned short opus_uint16;
|
||||
typedef int opus_int32;
|
||||
typedef unsigned int opus_uint32;
|
||||
|
||||
#elif defined(R5900)
|
||||
|
||||
/* PS2 EE */
|
||||
typedef int opus_int32;
|
||||
typedef unsigned opus_uint32;
|
||||
typedef short opus_int16;
|
||||
typedef unsigned short opus_uint16;
|
||||
|
||||
#elif defined(__SYMBIAN32__)
|
||||
|
||||
/* Symbian GCC */
|
||||
typedef signed short opus_int16;
|
||||
typedef unsigned short opus_uint16;
|
||||
typedef signed int opus_int32;
|
||||
typedef unsigned int opus_uint32;
|
||||
|
||||
#elif defined(CONFIG_TI_C54X) || defined (CONFIG_TI_C55X)
|
||||
|
||||
typedef short opus_int16;
|
||||
typedef unsigned short opus_uint16;
|
||||
typedef long opus_int32;
|
||||
typedef unsigned long opus_uint32;
|
||||
|
||||
#elif defined(CONFIG_TI_C6X)
|
||||
|
||||
typedef short opus_int16;
|
||||
typedef unsigned short opus_uint16;
|
||||
typedef int opus_int32;
|
||||
typedef unsigned int opus_uint32;
|
||||
|
||||
#else
|
||||
|
||||
/* Give up, take a reasonable guess */
|
||||
typedef short opus_int16;
|
||||
typedef unsigned short opus_uint16;
|
||||
typedef int opus_int32;
|
||||
typedef unsigned int opus_uint32;
|
||||
|
||||
#endif
|
||||
|
||||
#define opus_int int /* used for counters etc; at least 16 bits */
|
||||
#define opus_int64 long long
|
||||
#define opus_int8 signed char
|
||||
|
||||
#define opus_uint unsigned int /* used for counters etc; at least 16 bits */
|
||||
#define opus_uint64 unsigned long long
|
||||
#define opus_uint8 unsigned char
|
||||
|
||||
#endif /* OPUS_TYPES_H */
|
||||
237
src/modules/ZoomIt/ZoomIt/rnnoise/parse_lpcnet_weights.c
Normal file
237
src/modules/ZoomIt/ZoomIt/rnnoise/parse_lpcnet_weights.c
Normal file
@@ -0,0 +1,237 @@
|
||||
/* Copyright (c) 2023 Amazon */
|
||||
/*
|
||||
Redistribution and use in source and binary forms, with or without
|
||||
modification, are permitted provided that the following conditions
|
||||
are met:
|
||||
|
||||
- Redistributions of source code must retain the above copyright
|
||||
notice, this list of conditions and the following disclaimer.
|
||||
|
||||
- Redistributions in binary form must reproduce the above copyright
|
||||
notice, this list of conditions and the following disclaimer in the
|
||||
documentation and/or other materials provided with the distribution.
|
||||
|
||||
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||||
``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
||||
LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
||||
A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR
|
||||
CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
|
||||
EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
|
||||
PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
|
||||
PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
|
||||
LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
|
||||
NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
||||
SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
*/
|
||||
|
||||
#ifdef HAVE_CONFIG_H
|
||||
#include "config.h"
|
||||
#endif
|
||||
|
||||
#include <string.h>
|
||||
#include <stdlib.h>
|
||||
#include "nnet.h"
|
||||
|
||||
#define SPARSE_BLOCK_SIZE 32
|
||||
|
||||
static int parse_record(const void **data, int *len, WeightArray *array) {
|
||||
WeightHead *h = (WeightHead *)*data;
|
||||
if (*len < WEIGHT_BLOCK_SIZE) return -1;
|
||||
if (h->block_size < h->size) return -1;
|
||||
if (h->block_size > *len-WEIGHT_BLOCK_SIZE) return -1;
|
||||
if (h->name[sizeof(h->name)-1] != 0) return -1;
|
||||
if (h->size < 0) return -1;
|
||||
array->name = h->name;
|
||||
array->type = h->type;
|
||||
array->size = h->size;
|
||||
array->data = (void*)((unsigned char*)(*data)+WEIGHT_BLOCK_SIZE);
|
||||
|
||||
*data = (void*)((unsigned char*)*data + h->block_size+WEIGHT_BLOCK_SIZE);
|
||||
*len -= h->block_size+WEIGHT_BLOCK_SIZE;
|
||||
return array->size;
|
||||
}
|
||||
|
||||
int parse_weights(WeightArray **list, const void *data, int len)
|
||||
{
|
||||
int nb_arrays=0;
|
||||
int capacity=20;
|
||||
*list = calloc(capacity*sizeof(WeightArray), 1);
|
||||
while (len > 0) {
|
||||
int ret;
|
||||
WeightArray array = {NULL, 0, 0, 0};
|
||||
ret = parse_record(&data, &len, &array);
|
||||
if (ret > 0) {
|
||||
if (nb_arrays+1 >= capacity) {
|
||||
/* Make sure there's room for the ending NULL element too. */
|
||||
capacity = capacity*3/2;
|
||||
*list = realloc(*list, capacity*sizeof(WeightArray));
|
||||
}
|
||||
(*list)[nb_arrays++] = array;
|
||||
} else {
|
||||
free(*list);
|
||||
*list = NULL;
|
||||
return -1;
|
||||
}
|
||||
}
|
||||
(*list)[nb_arrays].name=NULL;
|
||||
return nb_arrays;
|
||||
}
|
||||
|
||||
static const void *find_array_entry(const WeightArray *arrays, const char *name) {
|
||||
while (arrays->name && strcmp(arrays->name, name) != 0) arrays++;
|
||||
return arrays;
|
||||
}
|
||||
|
||||
static const void *find_array_check(const WeightArray *arrays, const char *name, int size) {
|
||||
const WeightArray *a = find_array_entry(arrays, name);
|
||||
if (a->name && a->size == size) return a->data;
|
||||
else return NULL;
|
||||
}
|
||||
|
||||
static const void *opt_array_check(const WeightArray *arrays, const char *name, int size, int *error) {
|
||||
const WeightArray *a = find_array_entry(arrays, name);
|
||||
*error = (a->name != NULL && a->size != size);
|
||||
if (a->name && a->size == size) return a->data;
|
||||
else return NULL;
|
||||
}
|
||||
|
||||
static const void *find_idx_check(const WeightArray *arrays, const char *name, int nb_in, int nb_out, int *total_blocks) {
|
||||
int remain;
|
||||
const int *idx;
|
||||
const WeightArray *a = find_array_entry(arrays, name);
|
||||
*total_blocks = 0;
|
||||
if (a == NULL) return NULL;
|
||||
idx = a->data;
|
||||
remain = a->size/sizeof(int);
|
||||
while (remain > 0) {
|
||||
int nb_blocks;
|
||||
int i;
|
||||
nb_blocks = *idx++;
|
||||
if (remain < nb_blocks+1) return NULL;
|
||||
for (i=0;i<nb_blocks;i++) {
|
||||
int pos = *idx++;
|
||||
if (pos+3 >= nb_in || (pos&0x3)) return NULL;
|
||||
}
|
||||
nb_out -= 8;
|
||||
remain -= nb_blocks+1;
|
||||
*total_blocks += nb_blocks;
|
||||
}
|
||||
if (nb_out != 0) return NULL;
|
||||
return a->data;
|
||||
}
|
||||
|
||||
int linear_init(LinearLayer *layer, const WeightArray *arrays,
|
||||
const char *bias,
|
||||
const char *subias,
|
||||
const char *weights,
|
||||
const char *float_weights,
|
||||
const char *weights_idx,
|
||||
const char *diag,
|
||||
const char *scale,
|
||||
int nb_inputs,
|
||||
int nb_outputs)
|
||||
{
|
||||
int err;
|
||||
layer->bias = NULL;
|
||||
layer->subias = NULL;
|
||||
layer->weights = NULL;
|
||||
layer->float_weights = NULL;
|
||||
layer->weights_idx = NULL;
|
||||
layer->diag = NULL;
|
||||
layer->scale = NULL;
|
||||
if (bias != NULL) {
|
||||
if ((layer->bias = find_array_check(arrays, bias, nb_outputs*sizeof(layer->bias[0]))) == NULL) return 1;
|
||||
}
|
||||
if (subias != NULL) {
|
||||
if ((layer->subias = find_array_check(arrays, subias, nb_outputs*sizeof(layer->subias[0]))) == NULL) return 1;
|
||||
}
|
||||
if (weights_idx != NULL) {
|
||||
int total_blocks;
|
||||
if ((layer->weights_idx = find_idx_check(arrays, weights_idx, nb_inputs, nb_outputs, &total_blocks)) == NULL) return 1;
|
||||
if (weights != NULL) {
|
||||
if ((layer->weights = find_array_check(arrays, weights, SPARSE_BLOCK_SIZE*total_blocks*sizeof(layer->weights[0]))) == NULL) return 1;
|
||||
}
|
||||
if (float_weights != NULL) {
|
||||
layer->float_weights = opt_array_check(arrays, float_weights, SPARSE_BLOCK_SIZE*total_blocks*sizeof(layer->float_weights[0]), &err);
|
||||
if (err) return 1;
|
||||
}
|
||||
} else {
|
||||
if (weights != NULL) {
|
||||
if ((layer->weights = find_array_check(arrays, weights, nb_inputs*nb_outputs*sizeof(layer->weights[0]))) == NULL) return 1;
|
||||
}
|
||||
if (float_weights != NULL) {
|
||||
layer->float_weights = opt_array_check(arrays, float_weights, nb_inputs*nb_outputs*sizeof(layer->float_weights[0]), &err);
|
||||
if (err) return 1;
|
||||
}
|
||||
}
|
||||
if (diag != NULL) {
|
||||
if ((layer->diag = find_array_check(arrays, diag, nb_outputs*sizeof(layer->diag[0]))) == NULL) return 1;
|
||||
}
|
||||
if (weights != NULL) {
|
||||
if ((layer->scale = find_array_check(arrays, scale, nb_outputs*sizeof(layer->scale[0]))) == NULL) return 1;
|
||||
}
|
||||
layer->nb_inputs = nb_inputs;
|
||||
layer->nb_outputs = nb_outputs;
|
||||
return 0;
|
||||
}
|
||||
|
||||
int conv2d_init(Conv2dLayer *layer, const WeightArray *arrays,
|
||||
const char *bias,
|
||||
const char *float_weights,
|
||||
int in_channels,
|
||||
int out_channels,
|
||||
int ktime,
|
||||
int kheight)
|
||||
{
|
||||
int err;
|
||||
layer->bias = NULL;
|
||||
layer->float_weights = NULL;
|
||||
if (bias != NULL) {
|
||||
if ((layer->bias = find_array_check(arrays, bias, out_channels*sizeof(layer->bias[0]))) == NULL) return 1;
|
||||
}
|
||||
if (float_weights != NULL) {
|
||||
layer->float_weights = opt_array_check(arrays, float_weights, in_channels*out_channels*ktime*kheight*sizeof(layer->float_weights[0]), &err);
|
||||
if (err) return 1;
|
||||
}
|
||||
layer->in_channels = in_channels;
|
||||
layer->out_channels = out_channels;
|
||||
layer->ktime = ktime;
|
||||
layer->kheight = kheight;
|
||||
return 0;
|
||||
}
|
||||
|
||||
|
||||
|
||||
#if 0
|
||||
#include <fcntl.h>
|
||||
#include <sys/mman.h>
|
||||
#include <unistd.h>
|
||||
#include <sys/stat.h>
|
||||
#include <stdio.h>
|
||||
|
||||
int main()
|
||||
{
|
||||
int fd;
|
||||
void *data;
|
||||
int len;
|
||||
int nb_arrays;
|
||||
int i;
|
||||
WeightArray *list;
|
||||
struct stat st;
|
||||
const char *filename = "weights_blob.bin";
|
||||
stat(filename, &st);
|
||||
len = st.st_size;
|
||||
fd = open(filename, O_RDONLY);
|
||||
data = mmap(NULL, len, PROT_READ, MAP_SHARED, fd, 0);
|
||||
printf("size is %d\n", len);
|
||||
nb_arrays = parse_weights(&list, data, len);
|
||||
for (i=0;i<nb_arrays;i++) {
|
||||
printf("found %s: size %d\n", list[i].name, list[i].size);
|
||||
}
|
||||
printf("%p\n", list[i].name);
|
||||
free(list);
|
||||
munmap(data, len);
|
||||
close(fd);
|
||||
return 0;
|
||||
}
|
||||
#endif
|
||||
528
src/modules/ZoomIt/ZoomIt/rnnoise/pitch.c
Normal file
528
src/modules/ZoomIt/ZoomIt/rnnoise/pitch.c
Normal file
@@ -0,0 +1,528 @@
|
||||
/* Copyright (c) 2007-2008 CSIRO
|
||||
Copyright (c) 2007-2009 Xiph.Org Foundation
|
||||
Written by Jean-Marc Valin */
|
||||
/**
|
||||
@file pitch.c
|
||||
@brief Pitch analysis
|
||||
*/
|
||||
|
||||
/*
|
||||
Redistribution and use in source and binary forms, with or without
|
||||
modification, are permitted provided that the following conditions
|
||||
are met:
|
||||
|
||||
- Redistributions of source code must retain the above copyright
|
||||
notice, this list of conditions and the following disclaimer.
|
||||
|
||||
- Redistributions in binary form must reproduce the above copyright
|
||||
notice, this list of conditions and the following disclaimer in the
|
||||
documentation and/or other materials provided with the distribution.
|
||||
|
||||
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||||
``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
||||
LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
||||
A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER
|
||||
OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
|
||||
EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
|
||||
PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
|
||||
PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
|
||||
LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
|
||||
NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
||||
SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
*/
|
||||
|
||||
#ifdef HAVE_CONFIG_H
|
||||
#include "config.h"
|
||||
#endif
|
||||
|
||||
#include "pitch.h"
|
||||
#include "common.h"
|
||||
#include "denoise.h"
|
||||
#include "celt_lpc.h"
|
||||
#include "math.h"
|
||||
|
||||
static void find_best_pitch(opus_val32 *xcorr, opus_val16 *y, int len,
|
||||
int max_pitch, int *best_pitch
|
||||
#ifdef FIXED_POINT
|
||||
, int yshift, opus_val32 maxcorr
|
||||
#endif
|
||||
)
|
||||
{
|
||||
int i, j;
|
||||
opus_val32 Syy=1;
|
||||
opus_val16 best_num[2];
|
||||
opus_val32 best_den[2];
|
||||
#ifdef FIXED_POINT
|
||||
int xshift;
|
||||
|
||||
xshift = celt_ilog2(maxcorr)-14;
|
||||
#endif
|
||||
|
||||
best_num[0] = -1;
|
||||
best_num[1] = -1;
|
||||
best_den[0] = 0;
|
||||
best_den[1] = 0;
|
||||
best_pitch[0] = 0;
|
||||
best_pitch[1] = 1;
|
||||
for (j=0;j<len;j++)
|
||||
Syy = ADD32(Syy, SHR32(MULT16_16(y[j],y[j]), yshift));
|
||||
for (i=0;i<max_pitch;i++)
|
||||
{
|
||||
if (xcorr[i]>0)
|
||||
{
|
||||
opus_val16 num;
|
||||
opus_val32 xcorr16;
|
||||
xcorr16 = EXTRACT16(VSHR32(xcorr[i], xshift));
|
||||
#ifndef FIXED_POINT
|
||||
/* Considering the range of xcorr16, this should avoid both underflows
|
||||
and overflows (inf) when squaring xcorr16 */
|
||||
xcorr16 *= 1e-12f;
|
||||
#endif
|
||||
num = MULT16_16_Q15(xcorr16,xcorr16);
|
||||
if (MULT16_32_Q15(num,best_den[1]) > MULT16_32_Q15(best_num[1],Syy))
|
||||
{
|
||||
if (MULT16_32_Q15(num,best_den[0]) > MULT16_32_Q15(best_num[0],Syy))
|
||||
{
|
||||
best_num[1] = best_num[0];
|
||||
best_den[1] = best_den[0];
|
||||
best_pitch[1] = best_pitch[0];
|
||||
best_num[0] = num;
|
||||
best_den[0] = Syy;
|
||||
best_pitch[0] = i;
|
||||
} else {
|
||||
best_num[1] = num;
|
||||
best_den[1] = Syy;
|
||||
best_pitch[1] = i;
|
||||
}
|
||||
}
|
||||
}
|
||||
Syy += SHR32(MULT16_16(y[i+len],y[i+len]),yshift) - SHR32(MULT16_16(y[i],y[i]),yshift);
|
||||
Syy = MAX32(1, Syy);
|
||||
}
|
||||
}
|
||||
|
||||
static void celt_fir5(const opus_val16 *x,
|
||||
const opus_val16 *num,
|
||||
opus_val16 *y,
|
||||
int N,
|
||||
opus_val16 *mem)
|
||||
{
|
||||
int i;
|
||||
opus_val16 num0, num1, num2, num3, num4;
|
||||
opus_val32 mem0, mem1, mem2, mem3, mem4;
|
||||
num0=num[0];
|
||||
num1=num[1];
|
||||
num2=num[2];
|
||||
num3=num[3];
|
||||
num4=num[4];
|
||||
mem0=mem[0];
|
||||
mem1=mem[1];
|
||||
mem2=mem[2];
|
||||
mem3=mem[3];
|
||||
mem4=mem[4];
|
||||
for (i=0;i<N;i++)
|
||||
{
|
||||
opus_val32 sum = SHL32(EXTEND32(x[i]), SIG_SHIFT);
|
||||
sum = MAC16_16(sum,num0,mem0);
|
||||
sum = MAC16_16(sum,num1,mem1);
|
||||
sum = MAC16_16(sum,num2,mem2);
|
||||
sum = MAC16_16(sum,num3,mem3);
|
||||
sum = MAC16_16(sum,num4,mem4);
|
||||
mem4 = mem3;
|
||||
mem3 = mem2;
|
||||
mem2 = mem1;
|
||||
mem1 = mem0;
|
||||
mem0 = x[i];
|
||||
y[i] = ROUND16(sum, SIG_SHIFT);
|
||||
}
|
||||
mem[0]=mem0;
|
||||
mem[1]=mem1;
|
||||
mem[2]=mem2;
|
||||
mem[3]=mem3;
|
||||
mem[4]=mem4;
|
||||
}
|
||||
|
||||
|
||||
void rnn_pitch_downsample(celt_sig *x[], opus_val16 *x_lp,
|
||||
int len, int C)
|
||||
{
|
||||
int i;
|
||||
opus_val32 ac[5];
|
||||
opus_val16 tmp=Q15ONE;
|
||||
opus_val16 lpc[4], mem[5]={0,0,0,0,0};
|
||||
opus_val16 lpc2[5];
|
||||
opus_val16 c1 = QCONST16(.8f,15);
|
||||
#ifdef FIXED_POINT
|
||||
int shift;
|
||||
opus_val32 maxabs = celt_maxabs32(x[0], len);
|
||||
if (C==2)
|
||||
{
|
||||
opus_val32 maxabs_1 = celt_maxabs32(x[1], len);
|
||||
maxabs = MAX32(maxabs, maxabs_1);
|
||||
}
|
||||
if (maxabs<1)
|
||||
maxabs=1;
|
||||
shift = celt_ilog2(maxabs)-10;
|
||||
if (shift<0)
|
||||
shift=0;
|
||||
if (C==2)
|
||||
shift++;
|
||||
#endif
|
||||
for (i=1;i<len>>1;i++)
|
||||
x_lp[i] = SHR32(HALF32(HALF32(x[0][(2*i-1)]+x[0][(2*i+1)])+x[0][2*i]), shift);
|
||||
x_lp[0] = SHR32(HALF32(HALF32(x[0][1])+x[0][0]), shift);
|
||||
if (C==2)
|
||||
{
|
||||
for (i=1;i<len>>1;i++)
|
||||
x_lp[i] += SHR32(HALF32(HALF32(x[1][(2*i-1)]+x[1][(2*i+1)])+x[1][2*i]), shift);
|
||||
x_lp[0] += SHR32(HALF32(HALF32(x[1][1])+x[1][0]), shift);
|
||||
}
|
||||
|
||||
rnn_autocorr(x_lp, ac, NULL, 0,
|
||||
4, len>>1);
|
||||
|
||||
/* Noise floor -40 dB */
|
||||
#ifdef FIXED_POINT
|
||||
ac[0] += SHR32(ac[0],13);
|
||||
#else
|
||||
ac[0] *= 1.0001f;
|
||||
#endif
|
||||
/* Lag windowing */
|
||||
for (i=1;i<=4;i++)
|
||||
{
|
||||
/*ac[i] *= exp(-.5*(2*M_PI*.002*i)*(2*M_PI*.002*i));*/
|
||||
#ifdef FIXED_POINT
|
||||
ac[i] -= MULT16_32_Q15(2*i*i, ac[i]);
|
||||
#else
|
||||
ac[i] -= ac[i]*(.008f*i)*(.008f*i);
|
||||
#endif
|
||||
}
|
||||
|
||||
rnn_lpc(lpc, ac, 4);
|
||||
for (i=0;i<4;i++)
|
||||
{
|
||||
tmp = MULT16_16_Q15(QCONST16(.9f,15), tmp);
|
||||
lpc[i] = MULT16_16_Q15(lpc[i], tmp);
|
||||
}
|
||||
/* Add a zero */
|
||||
lpc2[0] = lpc[0] + QCONST16(.8f,SIG_SHIFT);
|
||||
lpc2[1] = lpc[1] + MULT16_16_Q15(c1,lpc[0]);
|
||||
lpc2[2] = lpc[2] + MULT16_16_Q15(c1,lpc[1]);
|
||||
lpc2[3] = lpc[3] + MULT16_16_Q15(c1,lpc[2]);
|
||||
lpc2[4] = MULT16_16_Q15(c1,lpc[3]);
|
||||
celt_fir5(x_lp, lpc2, x_lp, len>>1, mem);
|
||||
}
|
||||
|
||||
void rnn_pitch_xcorr(const opus_val16 *_x, const opus_val16 *_y,
|
||||
opus_val32 *xcorr, int len, int max_pitch)
|
||||
{
|
||||
|
||||
#if 0 /* This is a simple version of the pitch correlation that should work
|
||||
well on DSPs like Blackfin and TI C5x/C6x */
|
||||
int i, j;
|
||||
#ifdef FIXED_POINT
|
||||
opus_val32 maxcorr=1;
|
||||
#endif
|
||||
for (i=0;i<max_pitch;i++)
|
||||
{
|
||||
opus_val32 sum = 0;
|
||||
for (j=0;j<len;j++)
|
||||
sum = MAC16_16(sum, _x[j], _y[i+j]);
|
||||
xcorr[i] = sum;
|
||||
#ifdef FIXED_POINT
|
||||
maxcorr = MAX32(maxcorr, sum);
|
||||
#endif
|
||||
}
|
||||
#ifdef FIXED_POINT
|
||||
return maxcorr;
|
||||
#endif
|
||||
|
||||
#else /* Unrolled version of the pitch correlation -- runs faster on x86 and ARM */
|
||||
int i;
|
||||
/*The EDSP version requires that max_pitch is at least 1, and that _x is
|
||||
32-bit aligned.
|
||||
Since it's hard to put asserts in assembly, put them here.*/
|
||||
#ifdef FIXED_POINT
|
||||
opus_val32 maxcorr=1;
|
||||
#endif
|
||||
celt_assert(max_pitch>0);
|
||||
celt_assert((((unsigned char *)_x-(unsigned char *)NULL)&3)==0);
|
||||
for (i=0;i<max_pitch-3;i+=4)
|
||||
{
|
||||
opus_val32 sum[4]={0,0,0,0};
|
||||
xcorr_kernel(_x, _y+i, sum, len);
|
||||
xcorr[i]=sum[0];
|
||||
xcorr[i+1]=sum[1];
|
||||
xcorr[i+2]=sum[2];
|
||||
xcorr[i+3]=sum[3];
|
||||
#ifdef FIXED_POINT
|
||||
sum[0] = MAX32(sum[0], sum[1]);
|
||||
sum[2] = MAX32(sum[2], sum[3]);
|
||||
sum[0] = MAX32(sum[0], sum[2]);
|
||||
maxcorr = MAX32(maxcorr, sum[0]);
|
||||
#endif
|
||||
}
|
||||
/* In case max_pitch isn't a multiple of 4, do non-unrolled version. */
|
||||
for (;i<max_pitch;i++)
|
||||
{
|
||||
opus_val32 sum;
|
||||
sum = celt_inner_prod(_x, _y+i, len);
|
||||
xcorr[i] = sum;
|
||||
#ifdef FIXED_POINT
|
||||
maxcorr = MAX32(maxcorr, sum);
|
||||
#endif
|
||||
}
|
||||
#ifdef FIXED_POINT
|
||||
return maxcorr;
|
||||
#endif
|
||||
#endif
|
||||
}
|
||||
|
||||
void rnn_pitch_search(const opus_val16 *x_lp, opus_val16 *y,
|
||||
int len, int max_pitch, int *pitch)
|
||||
{
|
||||
int i, j;
|
||||
int lag;
|
||||
int best_pitch[2]={0,0};
|
||||
#ifdef FIXED_POINT
|
||||
opus_val32 maxcorr;
|
||||
opus_val32 xmax, ymax;
|
||||
int shift=0;
|
||||
#endif
|
||||
int offset;
|
||||
opus_val16 x_lp4[PITCH_FRAME_SIZE>>2];
|
||||
opus_val16 y_lp4[(PITCH_FRAME_SIZE+PITCH_MAX_PERIOD)>>2];
|
||||
opus_val32 xcorr[PITCH_MAX_PERIOD>>1];
|
||||
|
||||
celt_assert(len <= PITCH_FRAME_SIZE);
|
||||
celt_assert(max_pitch <= PITCH_MAX_PERIOD);
|
||||
celt_assert(len>0);
|
||||
celt_assert(max_pitch>0);
|
||||
lag = len+max_pitch;
|
||||
|
||||
|
||||
/* Downsample by 2 again */
|
||||
for (j=0;j<len>>2;j++)
|
||||
x_lp4[j] = x_lp[2*j];
|
||||
for (j=0;j<lag>>2;j++)
|
||||
y_lp4[j] = y[2*j];
|
||||
|
||||
#ifdef FIXED_POINT
|
||||
xmax = celt_maxabs16(x_lp4, len>>2);
|
||||
ymax = celt_maxabs16(y_lp4, lag>>2);
|
||||
shift = celt_ilog2(MAX32(1, MAX32(xmax, ymax)))-11;
|
||||
if (shift>0)
|
||||
{
|
||||
for (j=0;j<len>>2;j++)
|
||||
x_lp4[j] = SHR16(x_lp4[j], shift);
|
||||
for (j=0;j<lag>>2;j++)
|
||||
y_lp4[j] = SHR16(y_lp4[j], shift);
|
||||
/* Use double the shift for a MAC */
|
||||
shift *= 2;
|
||||
} else {
|
||||
shift = 0;
|
||||
}
|
||||
#endif
|
||||
|
||||
/* Coarse search with 4x decimation */
|
||||
|
||||
#ifdef FIXED_POINT
|
||||
maxcorr =
|
||||
#endif
|
||||
rnn_pitch_xcorr(x_lp4, y_lp4, xcorr, len>>2, max_pitch>>2);
|
||||
|
||||
find_best_pitch(xcorr, y_lp4, len>>2, max_pitch>>2, best_pitch
|
||||
#ifdef FIXED_POINT
|
||||
, 0, maxcorr
|
||||
#endif
|
||||
);
|
||||
|
||||
/* Finer search with 2x decimation */
|
||||
#ifdef FIXED_POINT
|
||||
maxcorr=1;
|
||||
#endif
|
||||
for (i=0;i<max_pitch>>1;i++)
|
||||
{
|
||||
opus_val32 sum;
|
||||
xcorr[i] = 0;
|
||||
if (abs(i-2*best_pitch[0])>2 && abs(i-2*best_pitch[1])>2)
|
||||
continue;
|
||||
#ifdef FIXED_POINT
|
||||
sum = 0;
|
||||
for (j=0;j<len>>1;j++)
|
||||
sum += SHR32(MULT16_16(x_lp[j],y[i+j]), shift);
|
||||
#else
|
||||
sum = celt_inner_prod(x_lp, y+i, len>>1);
|
||||
#endif
|
||||
xcorr[i] = MAX32(-1, sum);
|
||||
#ifdef FIXED_POINT
|
||||
maxcorr = MAX32(maxcorr, sum);
|
||||
#endif
|
||||
}
|
||||
find_best_pitch(xcorr, y, len>>1, max_pitch>>1, best_pitch
|
||||
#ifdef FIXED_POINT
|
||||
, shift+1, maxcorr
|
||||
#endif
|
||||
);
|
||||
|
||||
/* Refine by pseudo-interpolation */
|
||||
if (best_pitch[0]>0 && best_pitch[0]<(max_pitch>>1)-1)
|
||||
{
|
||||
opus_val32 a, b, c;
|
||||
a = xcorr[best_pitch[0]-1];
|
||||
b = xcorr[best_pitch[0]];
|
||||
c = xcorr[best_pitch[0]+1];
|
||||
if ((c-a) > MULT16_32_Q15(QCONST16(.7f,15),b-a))
|
||||
offset = 1;
|
||||
else if ((a-c) > MULT16_32_Q15(QCONST16(.7f,15),b-c))
|
||||
offset = -1;
|
||||
else
|
||||
offset = 0;
|
||||
} else {
|
||||
offset = 0;
|
||||
}
|
||||
*pitch = 2*best_pitch[0]-offset;
|
||||
}
|
||||
|
||||
#ifdef FIXED_POINT
|
||||
static opus_val16 compute_pitch_gain(opus_val32 xy, opus_val32 xx, opus_val32 yy)
|
||||
{
|
||||
opus_val32 x2y2;
|
||||
int sx, sy, shift;
|
||||
opus_val32 g;
|
||||
opus_val16 den;
|
||||
if (xy == 0 || xx == 0 || yy == 0)
|
||||
return 0;
|
||||
sx = celt_ilog2(xx)-14;
|
||||
sy = celt_ilog2(yy)-14;
|
||||
shift = sx + sy;
|
||||
x2y2 = SHR32(MULT16_16(VSHR32(xx, sx), VSHR32(yy, sy)), 14);
|
||||
if (shift & 1) {
|
||||
if (x2y2 < 32768)
|
||||
{
|
||||
x2y2 <<= 1;
|
||||
shift--;
|
||||
} else {
|
||||
x2y2 >>= 1;
|
||||
shift++;
|
||||
}
|
||||
}
|
||||
den = celt_rsqrt_norm(x2y2);
|
||||
g = MULT16_32_Q15(den, xy);
|
||||
g = VSHR32(g, (shift>>1)-1);
|
||||
return EXTRACT16(MIN32(g, Q15ONE));
|
||||
}
|
||||
#else
|
||||
static opus_val16 compute_pitch_gain(opus_val32 xy, opus_val32 xx, opus_val32 yy)
|
||||
{
|
||||
return xy/sqrt(1+xx*yy);
|
||||
}
|
||||
#endif
|
||||
|
||||
static const int second_check[16] = {0, 0, 3, 2, 3, 2, 5, 2, 3, 2, 3, 2, 5, 2, 3, 2};
|
||||
opus_val16 rnn_remove_doubling(opus_val16 *x, int maxperiod, int minperiod,
|
||||
int N, int *T0_, int prev_period, opus_val16 prev_gain)
|
||||
{
|
||||
int k, i, T, T0;
|
||||
opus_val16 g, g0;
|
||||
opus_val16 pg;
|
||||
opus_val32 xy,xx,yy,xy2;
|
||||
opus_val32 xcorr[3];
|
||||
opus_val32 best_xy, best_yy;
|
||||
int offset;
|
||||
int minperiod0;
|
||||
opus_val32 yy_lookup[PITCH_MAX_PERIOD+1];
|
||||
|
||||
celt_assert(maxperiod <= PITCH_MAX_PERIOD);
|
||||
|
||||
minperiod0 = minperiod;
|
||||
maxperiod /= 2;
|
||||
minperiod /= 2;
|
||||
*T0_ /= 2;
|
||||
prev_period /= 2;
|
||||
N /= 2;
|
||||
x += maxperiod;
|
||||
if (*T0_>=maxperiod)
|
||||
*T0_=maxperiod-1;
|
||||
|
||||
T = T0 = *T0_;
|
||||
dual_inner_prod(x, x, x-T0, N, &xx, &xy);
|
||||
yy_lookup[0] = xx;
|
||||
yy=xx;
|
||||
for (i=1;i<=maxperiod;i++)
|
||||
{
|
||||
yy = yy+MULT16_16(x[-i],x[-i])-MULT16_16(x[N-i],x[N-i]);
|
||||
yy_lookup[i] = MAX32(0, yy);
|
||||
}
|
||||
yy = yy_lookup[T0];
|
||||
best_xy = xy;
|
||||
best_yy = yy;
|
||||
g = g0 = compute_pitch_gain(xy, xx, yy);
|
||||
/* Look for any pitch at T/k */
|
||||
for (k=2;k<=15;k++)
|
||||
{
|
||||
int T1, T1b;
|
||||
opus_val16 g1;
|
||||
opus_val16 cont=0;
|
||||
opus_val16 thresh;
|
||||
T1 = (2*T0+k)/(2*k);
|
||||
if (T1 < minperiod)
|
||||
break;
|
||||
/* Look for another strong correlation at T1b */
|
||||
if (k==2)
|
||||
{
|
||||
if (T1+T0>maxperiod)
|
||||
T1b = T0;
|
||||
else
|
||||
T1b = T0+T1;
|
||||
} else
|
||||
{
|
||||
T1b = (2*second_check[k]*T0+k)/(2*k);
|
||||
}
|
||||
dual_inner_prod(x, &x[-T1], &x[-T1b], N, &xy, &xy2);
|
||||
xy = HALF32(xy + xy2);
|
||||
yy = HALF32(yy_lookup[T1] + yy_lookup[T1b]);
|
||||
g1 = compute_pitch_gain(xy, xx, yy);
|
||||
if (abs(T1-prev_period)<=1)
|
||||
cont = prev_gain;
|
||||
else if (abs(T1-prev_period)<=2 && 5*k*k < T0)
|
||||
cont = HALF16(prev_gain);
|
||||
else
|
||||
cont = 0;
|
||||
thresh = MAX16(QCONST16(.3f,15), MULT16_16_Q15(QCONST16(.7f,15),g0)-cont);
|
||||
/* Bias against very high pitch (very short period) to avoid false-positives
|
||||
due to short-term correlation */
|
||||
if (T1<3*minperiod)
|
||||
thresh = MAX16(QCONST16(.4f,15), MULT16_16_Q15(QCONST16(.85f,15),g0)-cont);
|
||||
else if (T1<2*minperiod)
|
||||
thresh = MAX16(QCONST16(.5f,15), MULT16_16_Q15(QCONST16(.9f,15),g0)-cont);
|
||||
if (g1 > thresh)
|
||||
{
|
||||
best_xy = xy;
|
||||
best_yy = yy;
|
||||
T = T1;
|
||||
g = g1;
|
||||
}
|
||||
}
|
||||
best_xy = MAX32(0, best_xy);
|
||||
if (best_yy <= best_xy)
|
||||
pg = Q15ONE;
|
||||
else
|
||||
pg = best_xy/(best_yy+1);
|
||||
|
||||
for (k=0;k<3;k++)
|
||||
xcorr[k] = celt_inner_prod(x, x-(T+k-1), N);
|
||||
if ((xcorr[2]-xcorr[0]) > MULT16_32_Q15(QCONST16(.7f,15),xcorr[1]-xcorr[0]))
|
||||
offset = 1;
|
||||
else if ((xcorr[0]-xcorr[2]) > MULT16_32_Q15(QCONST16(.7f,15),xcorr[1]-xcorr[2]))
|
||||
offset = -1;
|
||||
else
|
||||
offset = 0;
|
||||
if (pg > g)
|
||||
pg = g;
|
||||
*T0_ = 2*T+offset;
|
||||
|
||||
if (*T0_<minperiod0)
|
||||
*T0_=minperiod0;
|
||||
return pg;
|
||||
}
|
||||
147
src/modules/ZoomIt/ZoomIt/rnnoise/pitch.h
Normal file
147
src/modules/ZoomIt/ZoomIt/rnnoise/pitch.h
Normal file
@@ -0,0 +1,147 @@
|
||||
/* Copyright (c) 2007-2008 CSIRO
|
||||
Copyright (c) 2007-2009 Xiph.Org Foundation
|
||||
Written by Jean-Marc Valin */
|
||||
/**
|
||||
@file pitch.h
|
||||
@brief Pitch analysis
|
||||
*/
|
||||
|
||||
/*
|
||||
Redistribution and use in source and binary forms, with or without
|
||||
modification, are permitted provided that the following conditions
|
||||
are met:
|
||||
|
||||
- Redistributions of source code must retain the above copyright
|
||||
notice, this list of conditions and the following disclaimer.
|
||||
|
||||
- Redistributions in binary form must reproduce the above copyright
|
||||
notice, this list of conditions and the following disclaimer in the
|
||||
documentation and/or other materials provided with the distribution.
|
||||
|
||||
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||||
``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
||||
LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
||||
A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER
|
||||
OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
|
||||
EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
|
||||
PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
|
||||
PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
|
||||
LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
|
||||
NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
||||
SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
*/
|
||||
|
||||
#ifndef PITCH_H
|
||||
#define PITCH_H
|
||||
|
||||
#include "arch.h"
|
||||
|
||||
void rnn_pitch_downsample(celt_sig *x[], opus_val16 *x_lp,
|
||||
int len, int C);
|
||||
|
||||
void rnn_pitch_search(const opus_val16 *x_lp, opus_val16 *y,
|
||||
int len, int max_pitch, int *pitch);
|
||||
|
||||
opus_val16 rnn_remove_doubling(opus_val16 *x, int maxperiod, int minperiod,
|
||||
int N, int *T0, int prev_period, opus_val16 prev_gain);
|
||||
|
||||
|
||||
/* OPT: This is the kernel you really want to optimize. It gets used a lot
|
||||
by the prefilter and by the PLC. */
|
||||
static OPUS_INLINE void xcorr_kernel(const opus_val16 * x, const opus_val16 * y, opus_val32 sum[4], int len)
|
||||
{
|
||||
int j;
|
||||
opus_val16 y_0, y_1, y_2, y_3;
|
||||
celt_assert(len>=3);
|
||||
y_3=0; /* gcc doesn't realize that y_3 can't be used uninitialized */
|
||||
y_0=*y++;
|
||||
y_1=*y++;
|
||||
y_2=*y++;
|
||||
for (j=0;j<len-3;j+=4)
|
||||
{
|
||||
opus_val16 tmp;
|
||||
tmp = *x++;
|
||||
y_3=*y++;
|
||||
sum[0] = MAC16_16(sum[0],tmp,y_0);
|
||||
sum[1] = MAC16_16(sum[1],tmp,y_1);
|
||||
sum[2] = MAC16_16(sum[2],tmp,y_2);
|
||||
sum[3] = MAC16_16(sum[3],tmp,y_3);
|
||||
tmp=*x++;
|
||||
y_0=*y++;
|
||||
sum[0] = MAC16_16(sum[0],tmp,y_1);
|
||||
sum[1] = MAC16_16(sum[1],tmp,y_2);
|
||||
sum[2] = MAC16_16(sum[2],tmp,y_3);
|
||||
sum[3] = MAC16_16(sum[3],tmp,y_0);
|
||||
tmp=*x++;
|
||||
y_1=*y++;
|
||||
sum[0] = MAC16_16(sum[0],tmp,y_2);
|
||||
sum[1] = MAC16_16(sum[1],tmp,y_3);
|
||||
sum[2] = MAC16_16(sum[2],tmp,y_0);
|
||||
sum[3] = MAC16_16(sum[3],tmp,y_1);
|
||||
tmp=*x++;
|
||||
y_2=*y++;
|
||||
sum[0] = MAC16_16(sum[0],tmp,y_3);
|
||||
sum[1] = MAC16_16(sum[1],tmp,y_0);
|
||||
sum[2] = MAC16_16(sum[2],tmp,y_1);
|
||||
sum[3] = MAC16_16(sum[3],tmp,y_2);
|
||||
}
|
||||
if (j++<len)
|
||||
{
|
||||
opus_val16 tmp = *x++;
|
||||
y_3=*y++;
|
||||
sum[0] = MAC16_16(sum[0],tmp,y_0);
|
||||
sum[1] = MAC16_16(sum[1],tmp,y_1);
|
||||
sum[2] = MAC16_16(sum[2],tmp,y_2);
|
||||
sum[3] = MAC16_16(sum[3],tmp,y_3);
|
||||
}
|
||||
if (j++<len)
|
||||
{
|
||||
opus_val16 tmp=*x++;
|
||||
y_0=*y++;
|
||||
sum[0] = MAC16_16(sum[0],tmp,y_1);
|
||||
sum[1] = MAC16_16(sum[1],tmp,y_2);
|
||||
sum[2] = MAC16_16(sum[2],tmp,y_3);
|
||||
sum[3] = MAC16_16(sum[3],tmp,y_0);
|
||||
}
|
||||
if (j<len)
|
||||
{
|
||||
opus_val16 tmp=*x++;
|
||||
y_1=*y++;
|
||||
sum[0] = MAC16_16(sum[0],tmp,y_2);
|
||||
sum[1] = MAC16_16(sum[1],tmp,y_3);
|
||||
sum[2] = MAC16_16(sum[2],tmp,y_0);
|
||||
sum[3] = MAC16_16(sum[3],tmp,y_1);
|
||||
}
|
||||
}
|
||||
|
||||
static OPUS_INLINE void dual_inner_prod(const opus_val16 *x, const opus_val16 *y01, const opus_val16 *y02,
|
||||
int N, opus_val32 *xy1, opus_val32 *xy2)
|
||||
{
|
||||
int i;
|
||||
opus_val32 xy01=0;
|
||||
opus_val32 xy02=0;
|
||||
for (i=0;i<N;i++)
|
||||
{
|
||||
xy01 = MAC16_16(xy01, x[i], y01[i]);
|
||||
xy02 = MAC16_16(xy02, x[i], y02[i]);
|
||||
}
|
||||
*xy1 = xy01;
|
||||
*xy2 = xy02;
|
||||
}
|
||||
|
||||
/*We make sure a C version is always available for cases where the overhead of
|
||||
vectorization and passing around an arch flag aren't worth it.*/
|
||||
static OPUS_INLINE opus_val32 celt_inner_prod(const opus_val16 *x,
|
||||
const opus_val16 *y, int N)
|
||||
{
|
||||
int i;
|
||||
opus_val32 xy=0;
|
||||
for (i=0;i<N;i++)
|
||||
xy = MAC16_16(xy, x[i], y[i]);
|
||||
return xy;
|
||||
}
|
||||
|
||||
void rnn_pitch_xcorr(const opus_val16 *_x, const opus_val16 *_y,
|
||||
opus_val32 *xcorr, int len, int max_pitch);
|
||||
|
||||
#endif
|
||||
60
src/modules/ZoomIt/ZoomIt/rnnoise/rnn.c
Normal file
60
src/modules/ZoomIt/ZoomIt/rnnoise/rnn.c
Normal file
@@ -0,0 +1,60 @@
|
||||
/* Copyright (c) 2008-2011 Octasic Inc.
|
||||
2012-2017 Jean-Marc Valin */
|
||||
/*
|
||||
Redistribution and use in source and binary forms, with or without
|
||||
modification, are permitted provided that the following conditions
|
||||
are met:
|
||||
|
||||
- Redistributions of source code must retain the above copyright
|
||||
notice, this list of conditions and the following disclaimer.
|
||||
|
||||
- Redistributions in binary form must reproduce the above copyright
|
||||
notice, this list of conditions and the following disclaimer in the
|
||||
documentation and/or other materials provided with the distribution.
|
||||
|
||||
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||||
``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
||||
LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
||||
A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR
|
||||
CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
|
||||
EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
|
||||
PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
|
||||
PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
|
||||
LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
|
||||
NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
||||
SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
*/
|
||||
|
||||
#ifdef HAVE_CONFIG_H
|
||||
#include "config.h"
|
||||
#endif
|
||||
|
||||
#include <math.h>
|
||||
#include "opus_types.h"
|
||||
#include "common.h"
|
||||
#include "arch.h"
|
||||
#include "rnn.h"
|
||||
#include "rnnoise_data.h"
|
||||
#include <stdio.h>
|
||||
|
||||
|
||||
#define INPUT_SIZE 42
|
||||
|
||||
|
||||
void compute_rnn(const RNNoise *model, RNNState *rnn, float *gains, float *vad, const float *input, int arch) {
|
||||
float tmp[MAX_NEURONS];
|
||||
float cat[CONV2_OUT_SIZE + GRU1_OUT_SIZE + GRU2_OUT_SIZE + GRU3_OUT_SIZE];
|
||||
/*for (int i=0;i<INPUT_SIZE;i++) printf("%f ", input[i]);printf("\n");*/
|
||||
compute_generic_conv1d(&model->conv1, tmp, rnn->conv1_state, input, CONV1_IN_SIZE, ACTIVATION_TANH, arch);
|
||||
compute_generic_conv1d(&model->conv2, cat, rnn->conv2_state, tmp, CONV2_IN_SIZE, ACTIVATION_TANH, arch);
|
||||
compute_generic_gru(&model->gru1_input, &model->gru1_recurrent, rnn->gru1_state, cat, arch);
|
||||
compute_generic_gru(&model->gru2_input, &model->gru2_recurrent, rnn->gru2_state, rnn->gru1_state, arch);
|
||||
compute_generic_gru(&model->gru3_input, &model->gru3_recurrent, rnn->gru3_state, rnn->gru2_state, arch);
|
||||
RNN_COPY(&cat[CONV2_OUT_SIZE], rnn->gru1_state, GRU1_OUT_SIZE);
|
||||
RNN_COPY(&cat[CONV2_OUT_SIZE+GRU1_OUT_SIZE], rnn->gru2_state, GRU2_OUT_SIZE);
|
||||
RNN_COPY(&cat[CONV2_OUT_SIZE+GRU1_OUT_SIZE+GRU2_OUT_SIZE], rnn->gru3_state, GRU3_OUT_SIZE);
|
||||
compute_generic_dense(&model->dense_out, gains, cat, ACTIVATION_SIGMOID, arch);
|
||||
compute_generic_dense(&model->vad_dense, vad, cat, ACTIVATION_SIGMOID, arch);
|
||||
/*for (int i=0;i<22;i++) printf("%f ", gains[i]);printf("\n");*/
|
||||
/*printf("%f\n", *vad);*/
|
||||
}
|
||||
49
src/modules/ZoomIt/ZoomIt/rnnoise/rnn.h
Normal file
49
src/modules/ZoomIt/ZoomIt/rnnoise/rnn.h
Normal file
@@ -0,0 +1,49 @@
|
||||
/* Copyright (c) 2017 Jean-Marc Valin */
|
||||
/*
|
||||
Redistribution and use in source and binary forms, with or without
|
||||
modification, are permitted provided that the following conditions
|
||||
are met:
|
||||
|
||||
- Redistributions of source code must retain the above copyright
|
||||
notice, this list of conditions and the following disclaimer.
|
||||
|
||||
- Redistributions in binary form must reproduce the above copyright
|
||||
notice, this list of conditions and the following disclaimer in the
|
||||
documentation and/or other materials provided with the distribution.
|
||||
|
||||
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||||
``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
||||
LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
||||
A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR
|
||||
CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
|
||||
EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
|
||||
PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
|
||||
PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
|
||||
LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
|
||||
NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
||||
SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
*/
|
||||
|
||||
#ifndef RNN_H_
|
||||
#define RNN_H_
|
||||
|
||||
#include "rnnoise.h"
|
||||
#include "rnnoise_data.h"
|
||||
|
||||
#include "opus_types.h"
|
||||
|
||||
#define WEIGHTS_SCALE (1.f/256)
|
||||
|
||||
#define MAX_NEURONS 1024
|
||||
|
||||
|
||||
typedef struct {
|
||||
float conv1_state[CONV1_STATE_SIZE];
|
||||
float conv2_state[CONV2_STATE_SIZE];
|
||||
float gru1_state[GRU1_STATE_SIZE];
|
||||
float gru2_state[GRU2_STATE_SIZE];
|
||||
float gru3_state[GRU3_STATE_SIZE];
|
||||
} RNNState;
|
||||
void compute_rnn(const RNNoise *model, RNNState *rnn, float *gains, float *vad, const float *input, int arch);
|
||||
|
||||
#endif /* RNN_H_ */
|
||||
131
src/modules/ZoomIt/ZoomIt/rnnoise/rnnoise.h
Normal file
131
src/modules/ZoomIt/ZoomIt/rnnoise/rnnoise.h
Normal file
@@ -0,0 +1,131 @@
|
||||
/* Copyright (c) 2018 Gregor Richards
|
||||
* Copyright (c) 2017 Mozilla */
|
||||
/*
|
||||
Redistribution and use in source and binary forms, with or without
|
||||
modification, are permitted provided that the following conditions
|
||||
are met:
|
||||
|
||||
- Redistributions of source code must retain the above copyright
|
||||
notice, this list of conditions and the following disclaimer.
|
||||
|
||||
- Redistributions in binary form must reproduce the above copyright
|
||||
notice, this list of conditions and the following disclaimer in the
|
||||
documentation and/or other materials provided with the distribution.
|
||||
|
||||
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||||
``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
||||
LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
||||
A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR
|
||||
CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
|
||||
EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
|
||||
PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
|
||||
PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
|
||||
LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
|
||||
NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
||||
SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
*/
|
||||
|
||||
#ifndef RNNOISE_H
|
||||
#define RNNOISE_H 1
|
||||
|
||||
#include <stdio.h>
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
#ifndef RNNOISE_EXPORT
|
||||
# if defined(WIN32)
|
||||
# if defined(RNNOISE_BUILD) && defined(DLL_EXPORT)
|
||||
# define RNNOISE_EXPORT __declspec(dllexport)
|
||||
# else
|
||||
# define RNNOISE_EXPORT
|
||||
# endif
|
||||
# elif defined(__GNUC__) && defined(RNNOISE_BUILD)
|
||||
# define RNNOISE_EXPORT __attribute__ ((visibility ("default")))
|
||||
# else
|
||||
# define RNNOISE_EXPORT
|
||||
# endif
|
||||
#endif
|
||||
|
||||
typedef struct DenoiseState DenoiseState;
|
||||
typedef struct RNNModel RNNModel;
|
||||
|
||||
/**
|
||||
* Return the size of DenoiseState
|
||||
*/
|
||||
RNNOISE_EXPORT int rnnoise_get_size(void);
|
||||
|
||||
/**
|
||||
* Return the number of samples processed by rnnoise_process_frame at a time
|
||||
*/
|
||||
RNNOISE_EXPORT int rnnoise_get_frame_size(void);
|
||||
|
||||
/**
|
||||
* Initializes a pre-allocated DenoiseState
|
||||
*
|
||||
* If model is NULL the default model is used.
|
||||
*
|
||||
* See: rnnoise_create() and rnnoise_model_from_file()
|
||||
*/
|
||||
RNNOISE_EXPORT int rnnoise_init(DenoiseState *st, RNNModel *model);
|
||||
|
||||
/**
|
||||
* Allocate and initialize a DenoiseState
|
||||
*
|
||||
* If model is NULL the default model is used.
|
||||
*
|
||||
* The returned pointer MUST be freed with rnnoise_destroy().
|
||||
*/
|
||||
RNNOISE_EXPORT DenoiseState *rnnoise_create(RNNModel *model);
|
||||
|
||||
/**
|
||||
* Free a DenoiseState produced by rnnoise_create.
|
||||
*
|
||||
* The optional custom model must be freed by rnnoise_model_free() after.
|
||||
*/
|
||||
RNNOISE_EXPORT void rnnoise_destroy(DenoiseState *st);
|
||||
|
||||
/**
|
||||
* Denoise a frame of samples
|
||||
*
|
||||
* in and out must be at least rnnoise_get_frame_size() large.
|
||||
*/
|
||||
RNNOISE_EXPORT float rnnoise_process_frame(DenoiseState *st, float *out, const float *in);
|
||||
|
||||
/**
|
||||
* Load a model from a memory buffer
|
||||
*
|
||||
* It must be deallocated with rnnoise_model_free() and the buffer must remain
|
||||
* valid until after the returned object is destroyed.
|
||||
*/
|
||||
RNNOISE_EXPORT RNNModel *rnnoise_model_from_buffer(const void *ptr, int len);
|
||||
|
||||
|
||||
/**
|
||||
* Load a model from a file
|
||||
*
|
||||
* It must be deallocated with rnnoise_model_free() and the file must not be
|
||||
* closed until the returned object is destroyed.
|
||||
*/
|
||||
RNNOISE_EXPORT RNNModel *rnnoise_model_from_file(FILE *f);
|
||||
|
||||
/**
|
||||
* Load a model from a file name
|
||||
*
|
||||
* It must be deallocated with rnnoise_model_free()
|
||||
*/
|
||||
RNNOISE_EXPORT RNNModel *rnnoise_model_from_filename(const char *filename);
|
||||
|
||||
/**
|
||||
* Free a custom model
|
||||
*
|
||||
* It must be called after all the DenoiseStates referring to it are freed.
|
||||
*/
|
||||
RNNOISE_EXPORT void rnnoise_model_free(RNNModel *model);
|
||||
|
||||
#ifdef __cplusplus
|
||||
}
|
||||
#endif
|
||||
|
||||
#endif
|
||||
55
src/modules/ZoomIt/ZoomIt/rnnoise/rnnoise_data.h
Normal file
55
src/modules/ZoomIt/ZoomIt/rnnoise/rnnoise_data.h
Normal file
@@ -0,0 +1,55 @@
|
||||
|
||||
#ifndef RNNOISE_DATA_H
|
||||
#define RNNOISE_DATA_H
|
||||
|
||||
#include "nnet.h"
|
||||
|
||||
|
||||
#define CONV1_OUT_SIZE 128
|
||||
|
||||
#define CONV1_IN_SIZE 65
|
||||
|
||||
#define CONV1_STATE_SIZE (65 * (2))
|
||||
|
||||
#define CONV1_DELAY 1
|
||||
|
||||
#define CONV2_OUT_SIZE 384
|
||||
|
||||
#define CONV2_IN_SIZE 128
|
||||
|
||||
#define CONV2_STATE_SIZE (128 * (2))
|
||||
|
||||
#define CONV2_DELAY 1
|
||||
|
||||
#define GRU1_OUT_SIZE 384
|
||||
|
||||
#define GRU1_STATE_SIZE 384
|
||||
|
||||
#define GRU2_OUT_SIZE 384
|
||||
|
||||
#define GRU2_STATE_SIZE 384
|
||||
|
||||
#define GRU3_OUT_SIZE 384
|
||||
|
||||
#define GRU3_STATE_SIZE 384
|
||||
|
||||
#define DENSE_OUT_OUT_SIZE 32
|
||||
|
||||
#define VAD_DENSE_OUT_SIZE 1
|
||||
|
||||
typedef struct {
|
||||
LinearLayer conv1;
|
||||
LinearLayer conv2;
|
||||
LinearLayer gru1_input;
|
||||
LinearLayer gru1_recurrent;
|
||||
LinearLayer gru2_input;
|
||||
LinearLayer gru2_recurrent;
|
||||
LinearLayer gru3_input;
|
||||
LinearLayer gru3_recurrent;
|
||||
LinearLayer dense_out;
|
||||
LinearLayer vad_dense;
|
||||
} RNNoise;
|
||||
|
||||
int init_rnnoise(RNNoise *model, const WeightArray *arrays);
|
||||
|
||||
#endif /* RNNOISE_DATA_H */
|
||||
274950
src/modules/ZoomIt/ZoomIt/rnnoise/rnnoise_data_little.c
Normal file
274950
src/modules/ZoomIt/ZoomIt/rnnoise/rnnoise_data_little.c
Normal file
File diff suppressed because it is too large
Load Diff
874
src/modules/ZoomIt/ZoomIt/rnnoise/rnnoise_tables.c
Normal file
874
src/modules/ZoomIt/ZoomIt/rnnoise/rnnoise_tables.c
Normal file
@@ -0,0 +1,874 @@
|
||||
/* The contents of this file was automatically generated by dump_rnnoise_tables.c*/
|
||||
|
||||
#ifdef HAVE_CONFIG_H
|
||||
#include "config.h"
|
||||
#endif
|
||||
#include "kiss_fft.h"
|
||||
|
||||
static const arch_fft_state arch_fft = {0, NULL};
|
||||
|
||||
static const opus_int32 fft_bitrev[960] = {
|
||||
0, 192, 384, 576, 768, 64, 256, 448, 640, 832, 128, 320, 512, 704, 896,
|
||||
16, 208, 400, 592, 784, 80, 272, 464, 656, 848, 144, 336, 528, 720, 912,
|
||||
32, 224, 416, 608, 800, 96, 288, 480, 672, 864, 160, 352, 544, 736, 928,
|
||||
48, 240, 432, 624, 816, 112, 304, 496, 688, 880, 176, 368, 560, 752, 944,
|
||||
4, 196, 388, 580, 772, 68, 260, 452, 644, 836, 132, 324, 516, 708, 900,
|
||||
20, 212, 404, 596, 788, 84, 276, 468, 660, 852, 148, 340, 532, 724, 916,
|
||||
36, 228, 420, 612, 804, 100, 292, 484, 676, 868, 164, 356, 548, 740, 932,
|
||||
52, 244, 436, 628, 820, 116, 308, 500, 692, 884, 180, 372, 564, 756, 948,
|
||||
8, 200, 392, 584, 776, 72, 264, 456, 648, 840, 136, 328, 520, 712, 904,
|
||||
24, 216, 408, 600, 792, 88, 280, 472, 664, 856, 152, 344, 536, 728, 920,
|
||||
40, 232, 424, 616, 808, 104, 296, 488, 680, 872, 168, 360, 552, 744, 936,
|
||||
56, 248, 440, 632, 824, 120, 312, 504, 696, 888, 184, 376, 568, 760, 952,
|
||||
12, 204, 396, 588, 780, 76, 268, 460, 652, 844, 140, 332, 524, 716, 908,
|
||||
28, 220, 412, 604, 796, 92, 284, 476, 668, 860, 156, 348, 540, 732, 924,
|
||||
44, 236, 428, 620, 812, 108, 300, 492, 684, 876, 172, 364, 556, 748, 940,
|
||||
60, 252, 444, 636, 828, 124, 316, 508, 700, 892, 188, 380, 572, 764, 956,
|
||||
1, 193, 385, 577, 769, 65, 257, 449, 641, 833, 129, 321, 513, 705, 897,
|
||||
17, 209, 401, 593, 785, 81, 273, 465, 657, 849, 145, 337, 529, 721, 913,
|
||||
33, 225, 417, 609, 801, 97, 289, 481, 673, 865, 161, 353, 545, 737, 929,
|
||||
49, 241, 433, 625, 817, 113, 305, 497, 689, 881, 177, 369, 561, 753, 945,
|
||||
5, 197, 389, 581, 773, 69, 261, 453, 645, 837, 133, 325, 517, 709, 901,
|
||||
21, 213, 405, 597, 789, 85, 277, 469, 661, 853, 149, 341, 533, 725, 917,
|
||||
37, 229, 421, 613, 805, 101, 293, 485, 677, 869, 165, 357, 549, 741, 933,
|
||||
53, 245, 437, 629, 821, 117, 309, 501, 693, 885, 181, 373, 565, 757, 949,
|
||||
9, 201, 393, 585, 777, 73, 265, 457, 649, 841, 137, 329, 521, 713, 905,
|
||||
25, 217, 409, 601, 793, 89, 281, 473, 665, 857, 153, 345, 537, 729, 921,
|
||||
41, 233, 425, 617, 809, 105, 297, 489, 681, 873, 169, 361, 553, 745, 937,
|
||||
57, 249, 441, 633, 825, 121, 313, 505, 697, 889, 185, 377, 569, 761, 953,
|
||||
13, 205, 397, 589, 781, 77, 269, 461, 653, 845, 141, 333, 525, 717, 909,
|
||||
29, 221, 413, 605, 797, 93, 285, 477, 669, 861, 157, 349, 541, 733, 925,
|
||||
45, 237, 429, 621, 813, 109, 301, 493, 685, 877, 173, 365, 557, 749, 941,
|
||||
61, 253, 445, 637, 829, 125, 317, 509, 701, 893, 189, 381, 573, 765, 957,
|
||||
2, 194, 386, 578, 770, 66, 258, 450, 642, 834, 130, 322, 514, 706, 898,
|
||||
18, 210, 402, 594, 786, 82, 274, 466, 658, 850, 146, 338, 530, 722, 914,
|
||||
34, 226, 418, 610, 802, 98, 290, 482, 674, 866, 162, 354, 546, 738, 930,
|
||||
50, 242, 434, 626, 818, 114, 306, 498, 690, 882, 178, 370, 562, 754, 946,
|
||||
6, 198, 390, 582, 774, 70, 262, 454, 646, 838, 134, 326, 518, 710, 902,
|
||||
22, 214, 406, 598, 790, 86, 278, 470, 662, 854, 150, 342, 534, 726, 918,
|
||||
38, 230, 422, 614, 806, 102, 294, 486, 678, 870, 166, 358, 550, 742, 934,
|
||||
54, 246, 438, 630, 822, 118, 310, 502, 694, 886, 182, 374, 566, 758, 950,
|
||||
10, 202, 394, 586, 778, 74, 266, 458, 650, 842, 138, 330, 522, 714, 906,
|
||||
26, 218, 410, 602, 794, 90, 282, 474, 666, 858, 154, 346, 538, 730, 922,
|
||||
42, 234, 426, 618, 810, 106, 298, 490, 682, 874, 170, 362, 554, 746, 938,
|
||||
58, 250, 442, 634, 826, 122, 314, 506, 698, 890, 186, 378, 570, 762, 954,
|
||||
14, 206, 398, 590, 782, 78, 270, 462, 654, 846, 142, 334, 526, 718, 910,
|
||||
30, 222, 414, 606, 798, 94, 286, 478, 670, 862, 158, 350, 542, 734, 926,
|
||||
46, 238, 430, 622, 814, 110, 302, 494, 686, 878, 174, 366, 558, 750, 942,
|
||||
62, 254, 446, 638, 830, 126, 318, 510, 702, 894, 190, 382, 574, 766, 958,
|
||||
3, 195, 387, 579, 771, 67, 259, 451, 643, 835, 131, 323, 515, 707, 899,
|
||||
19, 211, 403, 595, 787, 83, 275, 467, 659, 851, 147, 339, 531, 723, 915,
|
||||
35, 227, 419, 611, 803, 99, 291, 483, 675, 867, 163, 355, 547, 739, 931,
|
||||
51, 243, 435, 627, 819, 115, 307, 499, 691, 883, 179, 371, 563, 755, 947,
|
||||
7, 199, 391, 583, 775, 71, 263, 455, 647, 839, 135, 327, 519, 711, 903,
|
||||
23, 215, 407, 599, 791, 87, 279, 471, 663, 855, 151, 343, 535, 727, 919,
|
||||
39, 231, 423, 615, 807, 103, 295, 487, 679, 871, 167, 359, 551, 743, 935,
|
||||
55, 247, 439, 631, 823, 119, 311, 503, 695, 887, 183, 375, 567, 759, 951,
|
||||
11, 203, 395, 587, 779, 75, 267, 459, 651, 843, 139, 331, 523, 715, 907,
|
||||
27, 219, 411, 603, 795, 91, 283, 475, 667, 859, 155, 347, 539, 731, 923,
|
||||
43, 235, 427, 619, 811, 107, 299, 491, 683, 875, 171, 363, 555, 747, 939,
|
||||
59, 251, 443, 635, 827, 123, 315, 507, 699, 891, 187, 379, 571, 763, 955,
|
||||
15, 207, 399, 591, 783, 79, 271, 463, 655, 847, 143, 335, 527, 719, 911,
|
||||
31, 223, 415, 607, 799, 95, 287, 479, 671, 863, 159, 351, 543, 735, 927,
|
||||
47, 239, 431, 623, 815, 111, 303, 495, 687, 879, 175, 367, 559, 751, 943,
|
||||
63, 255, 447, 639, 831, 127, 319, 511, 703, 895, 191, 383, 575, 767, 959,
|
||||
};
|
||||
|
||||
static const kiss_twiddle_cpx fft_twiddles[960] = {
|
||||
{1.00000000f, -0.00000000f}, {0.999978602f, -0.00654493785f},
|
||||
{0.999914348f, -0.0130895954f}, {0.999807239f, -0.0196336918f},
|
||||
{0.999657333f, -0.0261769481f}, {0.999464571f, -0.0327190831f},
|
||||
{0.999229014f, -0.0392598175f}, {0.998950660f, -0.0457988679f},
|
||||
{0.998629510f, -0.0523359552f}, {0.998265624f, -0.0588708036f},
|
||||
{0.997858942f, -0.0654031262f}, {0.997409463f, -0.0719326511f},
|
||||
{0.996917307f, -0.0784590989f}, {0.996382475f, -0.0849821791f},
|
||||
{0.995804906f, -0.0915016159f}, {0.995184720f, -0.0980171412f},
|
||||
{0.994521916f, -0.104528464f}, {0.993816435f, -0.111035310f},
|
||||
{0.993068457f, -0.117537394f}, {0.992277920f, -0.124034449f},
|
||||
{0.991444886f, -0.130526185f}, {0.990569353f, -0.137012348f},
|
||||
{0.989651382f, -0.143492624f}, {0.988691032f, -0.149966761f},
|
||||
{0.987688363f, -0.156434461f}, {0.986643314f, -0.162895471f},
|
||||
{0.985556066f, -0.169349506f}, {0.984426558f, -0.175796285f},
|
||||
{0.983254910f, -0.182235524f}, {0.982041121f, -0.188666970f},
|
||||
{0.980785251f, -0.195090324f}, {0.979487419f, -0.201505318f},
|
||||
{0.978147626f, -0.207911685f}, {0.976765871f, -0.214309156f},
|
||||
{0.975342333f, -0.220697433f}, {0.973876953f, -0.227076262f},
|
||||
{0.972369909f, -0.233445361f}, {0.970821202f, -0.239804462f},
|
||||
{0.969230890f, -0.246153295f}, {0.967599094f, -0.252491564f},
|
||||
{0.965925813f, -0.258819044f}, {0.964211166f, -0.265135437f},
|
||||
{0.962455213f, -0.271440446f}, {0.960658073f, -0.277733833f},
|
||||
{0.958819747f, -0.284015357f}, {0.956940353f, -0.290284663f},
|
||||
{0.955019951f, -0.296541572f}, {0.953058660f, -0.302785784f},
|
||||
{0.951056540f, -0.309017003f}, {0.949013650f, -0.315234989f},
|
||||
{0.946930110f, -0.321439475f}, {0.944806039f, -0.327630192f},
|
||||
{0.942641497f, -0.333806872f}, {0.940436542f, -0.339969248f},
|
||||
{0.938191354f, -0.346117049f}, {0.935905933f, -0.352250040f},
|
||||
{0.933580399f, -0.358367950f}, {0.931214929f, -0.364470512f},
|
||||
{0.928809524f, -0.370557427f}, {0.926364362f, -0.376628488f},
|
||||
{0.923879504f, -0.382683426f}, {0.921355128f, -0.388721973f},
|
||||
{0.918791234f, -0.394743860f}, {0.916187942f, -0.400748819f},
|
||||
{0.913545430f, -0.406736642f}, {0.910863817f, -0.412707031f},
|
||||
{0.908143163f, -0.418659747f}, {0.905383646f, -0.424594522f},
|
||||
{0.902585268f, -0.430511087f}, {0.899748266f, -0.436409235f},
|
||||
{0.896872759f, -0.442288697f}, {0.893958807f, -0.448149204f},
|
||||
{0.891006529f, -0.453990489f}, {0.888016105f, -0.459812373f},
|
||||
{0.884987652f, -0.465614527f}, {0.881921291f, -0.471396744f},
|
||||
{0.878817141f, -0.477158755f}, {0.875675321f, -0.482900351f},
|
||||
{0.872496009f, -0.488621235f}, {0.869279325f, -0.494321197f},
|
||||
{0.866025388f, -0.500000000f}, {0.862734377f, -0.505657375f},
|
||||
{0.859406412f, -0.511293113f}, {0.856041610f, -0.516906917f},
|
||||
{0.852640152f, -0.522498548f}, {0.849202156f, -0.528067827f},
|
||||
{0.845727801f, -0.533614516f}, {0.842217207f, -0.539138317f},
|
||||
{0.838670552f, -0.544639051f}, {0.835087955f, -0.550116420f},
|
||||
{0.831469595f, -0.555570245f}, {0.827815652f, -0.561000228f},
|
||||
{0.824126184f, -0.566406250f}, {0.820401430f, -0.571787953f},
|
||||
{0.816641569f, -0.577145219f}, {0.812846661f, -0.582477689f},
|
||||
{0.809017003f, -0.587785244f}, {0.805152655f, -0.593067646f},
|
||||
{0.801253796f, -0.598324597f}, {0.797320664f, -0.603555918f},
|
||||
{0.793353319f, -0.608761430f}, {0.789352059f, -0.613940835f},
|
||||
{0.785316944f, -0.619093955f}, {0.781248152f, -0.624220550f},
|
||||
{0.777145982f, -0.629320383f}, {0.773010433f, -0.634393275f},
|
||||
{0.768841803f, -0.639438987f}, {0.764640272f, -0.644457340f},
|
||||
{0.760405958f, -0.649448037f}, {0.756139100f, -0.654410958f},
|
||||
{0.751839817f, -0.659345806f}, {0.747508347f, -0.664252460f},
|
||||
{0.743144810f, -0.669130623f}, {0.738749504f, -0.673980117f},
|
||||
{0.734322488f, -0.678800762f}, {0.729864061f, -0.683592319f},
|
||||
{0.725374401f, -0.688354552f}, {0.720853567f, -0.693087339f},
|
||||
{0.716301918f, -0.697790444f}, {0.711719632f, -0.702463686f},
|
||||
{0.707106769f, -0.707106769f}, {0.702463686f, -0.711719632f},
|
||||
{0.697790444f, -0.716301918f}, {0.693087339f, -0.720853567f},
|
||||
{0.688354552f, -0.725374401f}, {0.683592319f, -0.729864061f},
|
||||
{0.678800762f, -0.734322488f}, {0.673980117f, -0.738749504f},
|
||||
{0.669130623f, -0.743144810f}, {0.664252460f, -0.747508347f},
|
||||
{0.659345806f, -0.751839817f}, {0.654410958f, -0.756139100f},
|
||||
{0.649448037f, -0.760405958f}, {0.644457340f, -0.764640272f},
|
||||
{0.639438987f, -0.768841803f}, {0.634393275f, -0.773010433f},
|
||||
{0.629320383f, -0.777145982f}, {0.624220550f, -0.781248152f},
|
||||
{0.619093955f, -0.785316944f}, {0.613940835f, -0.789352059f},
|
||||
{0.608761430f, -0.793353319f}, {0.603555918f, -0.797320664f},
|
||||
{0.598324597f, -0.801253796f}, {0.593067646f, -0.805152655f},
|
||||
{0.587785244f, -0.809017003f}, {0.582477689f, -0.812846661f},
|
||||
{0.577145219f, -0.816641569f}, {0.571787953f, -0.820401430f},
|
||||
{0.566406250f, -0.824126184f}, {0.561000228f, -0.827815652f},
|
||||
{0.555570245f, -0.831469595f}, {0.550116420f, -0.835087955f},
|
||||
{0.544639051f, -0.838670552f}, {0.539138317f, -0.842217207f},
|
||||
{0.533614516f, -0.845727801f}, {0.528067827f, -0.849202156f},
|
||||
{0.522498548f, -0.852640152f}, {0.516906917f, -0.856041610f},
|
||||
{0.511293113f, -0.859406412f}, {0.505657375f, -0.862734377f},
|
||||
{0.500000000f, -0.866025388f}, {0.494321197f, -0.869279325f},
|
||||
{0.488621235f, -0.872496009f}, {0.482900351f, -0.875675321f},
|
||||
{0.477158755f, -0.878817141f}, {0.471396744f, -0.881921291f},
|
||||
{0.465614527f, -0.884987652f}, {0.459812373f, -0.888016105f},
|
||||
{0.453990489f, -0.891006529f}, {0.448149204f, -0.893958807f},
|
||||
{0.442288697f, -0.896872759f}, {0.436409235f, -0.899748266f},
|
||||
{0.430511087f, -0.902585268f}, {0.424594522f, -0.905383646f},
|
||||
{0.418659747f, -0.908143163f}, {0.412707031f, -0.910863817f},
|
||||
{0.406736642f, -0.913545430f}, {0.400748819f, -0.916187942f},
|
||||
{0.394743860f, -0.918791234f}, {0.388721973f, -0.921355128f},
|
||||
{0.382683426f, -0.923879504f}, {0.376628488f, -0.926364362f},
|
||||
{0.370557427f, -0.928809524f}, {0.364470512f, -0.931214929f},
|
||||
{0.358367950f, -0.933580399f}, {0.352250040f, -0.935905933f},
|
||||
{0.346117049f, -0.938191354f}, {0.339969248f, -0.940436542f},
|
||||
{0.333806872f, -0.942641497f}, {0.327630192f, -0.944806039f},
|
||||
{0.321439475f, -0.946930110f}, {0.315234989f, -0.949013650f},
|
||||
{0.309017003f, -0.951056540f}, {0.302785784f, -0.953058660f},
|
||||
{0.296541572f, -0.955019951f}, {0.290284663f, -0.956940353f},
|
||||
{0.284015357f, -0.958819747f}, {0.277733833f, -0.960658073f},
|
||||
{0.271440446f, -0.962455213f}, {0.265135437f, -0.964211166f},
|
||||
{0.258819044f, -0.965925813f}, {0.252491564f, -0.967599094f},
|
||||
{0.246153295f, -0.969230890f}, {0.239804462f, -0.970821202f},
|
||||
{0.233445361f, -0.972369909f}, {0.227076262f, -0.973876953f},
|
||||
{0.220697433f, -0.975342333f}, {0.214309156f, -0.976765871f},
|
||||
{0.207911685f, -0.978147626f}, {0.201505318f, -0.979487419f},
|
||||
{0.195090324f, -0.980785251f}, {0.188666970f, -0.982041121f},
|
||||
{0.182235524f, -0.983254910f}, {0.175796285f, -0.984426558f},
|
||||
{0.169349506f, -0.985556066f}, {0.162895471f, -0.986643314f},
|
||||
{0.156434461f, -0.987688363f}, {0.149966761f, -0.988691032f},
|
||||
{0.143492624f, -0.989651382f}, {0.137012348f, -0.990569353f},
|
||||
{0.130526185f, -0.991444886f}, {0.124034449f, -0.992277920f},
|
||||
{0.117537394f, -0.993068457f}, {0.111035310f, -0.993816435f},
|
||||
{0.104528464f, -0.994521916f}, {0.0980171412f, -0.995184720f},
|
||||
{0.0915016159f, -0.995804906f}, {0.0849821791f, -0.996382475f},
|
||||
{0.0784590989f, -0.996917307f}, {0.0719326511f, -0.997409463f},
|
||||
{0.0654031262f, -0.997858942f}, {0.0588708036f, -0.998265624f},
|
||||
{0.0523359552f, -0.998629510f}, {0.0457988679f, -0.998950660f},
|
||||
{0.0392598175f, -0.999229014f}, {0.0327190831f, -0.999464571f},
|
||||
{0.0261769481f, -0.999657333f}, {0.0196336918f, -0.999807239f},
|
||||
{0.0130895954f, -0.999914348f}, {0.00654493785f, -0.999978602f},
|
||||
{6.12323426e-17f, -1.00000000f}, {-0.00654493785f, -0.999978602f},
|
||||
{-0.0130895954f, -0.999914348f}, {-0.0196336918f, -0.999807239f},
|
||||
{-0.0261769481f, -0.999657333f}, {-0.0327190831f, -0.999464571f},
|
||||
{-0.0392598175f, -0.999229014f}, {-0.0457988679f, -0.998950660f},
|
||||
{-0.0523359552f, -0.998629510f}, {-0.0588708036f, -0.998265624f},
|
||||
{-0.0654031262f, -0.997858942f}, {-0.0719326511f, -0.997409463f},
|
||||
{-0.0784590989f, -0.996917307f}, {-0.0849821791f, -0.996382475f},
|
||||
{-0.0915016159f, -0.995804906f}, {-0.0980171412f, -0.995184720f},
|
||||
{-0.104528464f, -0.994521916f}, {-0.111035310f, -0.993816435f},
|
||||
{-0.117537394f, -0.993068457f}, {-0.124034449f, -0.992277920f},
|
||||
{-0.130526185f, -0.991444886f}, {-0.137012348f, -0.990569353f},
|
||||
{-0.143492624f, -0.989651382f}, {-0.149966761f, -0.988691032f},
|
||||
{-0.156434461f, -0.987688363f}, {-0.162895471f, -0.986643314f},
|
||||
{-0.169349506f, -0.985556066f}, {-0.175796285f, -0.984426558f},
|
||||
{-0.182235524f, -0.983254910f}, {-0.188666970f, -0.982041121f},
|
||||
{-0.195090324f, -0.980785251f}, {-0.201505318f, -0.979487419f},
|
||||
{-0.207911685f, -0.978147626f}, {-0.214309156f, -0.976765871f},
|
||||
{-0.220697433f, -0.975342333f}, {-0.227076262f, -0.973876953f},
|
||||
{-0.233445361f, -0.972369909f}, {-0.239804462f, -0.970821202f},
|
||||
{-0.246153295f, -0.969230890f}, {-0.252491564f, -0.967599094f},
|
||||
{-0.258819044f, -0.965925813f}, {-0.265135437f, -0.964211166f},
|
||||
{-0.271440446f, -0.962455213f}, {-0.277733833f, -0.960658073f},
|
||||
{-0.284015357f, -0.958819747f}, {-0.290284663f, -0.956940353f},
|
||||
{-0.296541572f, -0.955019951f}, {-0.302785784f, -0.953058660f},
|
||||
{-0.309017003f, -0.951056540f}, {-0.315234989f, -0.949013650f},
|
||||
{-0.321439475f, -0.946930110f}, {-0.327630192f, -0.944806039f},
|
||||
{-0.333806872f, -0.942641497f}, {-0.339969248f, -0.940436542f},
|
||||
{-0.346117049f, -0.938191354f}, {-0.352250040f, -0.935905933f},
|
||||
{-0.358367950f, -0.933580399f}, {-0.364470512f, -0.931214929f},
|
||||
{-0.370557427f, -0.928809524f}, {-0.376628488f, -0.926364362f},
|
||||
{-0.382683426f, -0.923879504f}, {-0.388721973f, -0.921355128f},
|
||||
{-0.394743860f, -0.918791234f}, {-0.400748819f, -0.916187942f},
|
||||
{-0.406736642f, -0.913545430f}, {-0.412707031f, -0.910863817f},
|
||||
{-0.418659747f, -0.908143163f}, {-0.424594522f, -0.905383646f},
|
||||
{-0.430511087f, -0.902585268f}, {-0.436409235f, -0.899748266f},
|
||||
{-0.442288697f, -0.896872759f}, {-0.448149204f, -0.893958807f},
|
||||
{-0.453990489f, -0.891006529f}, {-0.459812373f, -0.888016105f},
|
||||
{-0.465614527f, -0.884987652f}, {-0.471396744f, -0.881921291f},
|
||||
{-0.477158755f, -0.878817141f}, {-0.482900351f, -0.875675321f},
|
||||
{-0.488621235f, -0.872496009f}, {-0.494321197f, -0.869279325f},
|
||||
{-0.500000000f, -0.866025388f}, {-0.505657375f, -0.862734377f},
|
||||
{-0.511293113f, -0.859406412f}, {-0.516906917f, -0.856041610f},
|
||||
{-0.522498548f, -0.852640152f}, {-0.528067827f, -0.849202156f},
|
||||
{-0.533614516f, -0.845727801f}, {-0.539138317f, -0.842217207f},
|
||||
{-0.544639051f, -0.838670552f}, {-0.550116420f, -0.835087955f},
|
||||
{-0.555570245f, -0.831469595f}, {-0.561000228f, -0.827815652f},
|
||||
{-0.566406250f, -0.824126184f}, {-0.571787953f, -0.820401430f},
|
||||
{-0.577145219f, -0.816641569f}, {-0.582477689f, -0.812846661f},
|
||||
{-0.587785244f, -0.809017003f}, {-0.593067646f, -0.805152655f},
|
||||
{-0.598324597f, -0.801253796f}, {-0.603555918f, -0.797320664f},
|
||||
{-0.608761430f, -0.793353319f}, {-0.613940835f, -0.789352059f},
|
||||
{-0.619093955f, -0.785316944f}, {-0.624220550f, -0.781248152f},
|
||||
{-0.629320383f, -0.777145982f}, {-0.634393275f, -0.773010433f},
|
||||
{-0.639438987f, -0.768841803f}, {-0.644457340f, -0.764640272f},
|
||||
{-0.649448037f, -0.760405958f}, {-0.654410958f, -0.756139100f},
|
||||
{-0.659345806f, -0.751839817f}, {-0.664252460f, -0.747508347f},
|
||||
{-0.669130623f, -0.743144810f}, {-0.673980117f, -0.738749504f},
|
||||
{-0.678800762f, -0.734322488f}, {-0.683592319f, -0.729864061f},
|
||||
{-0.688354552f, -0.725374401f}, {-0.693087339f, -0.720853567f},
|
||||
{-0.697790444f, -0.716301918f}, {-0.702463686f, -0.711719632f},
|
||||
{-0.707106769f, -0.707106769f}, {-0.711719632f, -0.702463686f},
|
||||
{-0.716301918f, -0.697790444f}, {-0.720853567f, -0.693087339f},
|
||||
{-0.725374401f, -0.688354552f}, {-0.729864061f, -0.683592319f},
|
||||
{-0.734322488f, -0.678800762f}, {-0.738749504f, -0.673980117f},
|
||||
{-0.743144810f, -0.669130623f}, {-0.747508347f, -0.664252460f},
|
||||
{-0.751839817f, -0.659345806f}, {-0.756139100f, -0.654410958f},
|
||||
{-0.760405958f, -0.649448037f}, {-0.764640272f, -0.644457340f},
|
||||
{-0.768841803f, -0.639438987f}, {-0.773010433f, -0.634393275f},
|
||||
{-0.777145982f, -0.629320383f}, {-0.781248152f, -0.624220550f},
|
||||
{-0.785316944f, -0.619093955f}, {-0.789352059f, -0.613940835f},
|
||||
{-0.793353319f, -0.608761430f}, {-0.797320664f, -0.603555918f},
|
||||
{-0.801253796f, -0.598324597f}, {-0.805152655f, -0.593067646f},
|
||||
{-0.809017003f, -0.587785244f}, {-0.812846661f, -0.582477689f},
|
||||
{-0.816641569f, -0.577145219f}, {-0.820401430f, -0.571787953f},
|
||||
{-0.824126184f, -0.566406250f}, {-0.827815652f, -0.561000228f},
|
||||
{-0.831469595f, -0.555570245f}, {-0.835087955f, -0.550116420f},
|
||||
{-0.838670552f, -0.544639051f}, {-0.842217207f, -0.539138317f},
|
||||
{-0.845727801f, -0.533614516f}, {-0.849202156f, -0.528067827f},
|
||||
{-0.852640152f, -0.522498548f}, {-0.856041610f, -0.516906917f},
|
||||
{-0.859406412f, -0.511293113f}, {-0.862734377f, -0.505657375f},
|
||||
{-0.866025388f, -0.500000000f}, {-0.869279325f, -0.494321197f},
|
||||
{-0.872496009f, -0.488621235f}, {-0.875675321f, -0.482900351f},
|
||||
{-0.878817141f, -0.477158755f}, {-0.881921291f, -0.471396744f},
|
||||
{-0.884987652f, -0.465614527f}, {-0.888016105f, -0.459812373f},
|
||||
{-0.891006529f, -0.453990489f}, {-0.893958807f, -0.448149204f},
|
||||
{-0.896872759f, -0.442288697f}, {-0.899748266f, -0.436409235f},
|
||||
{-0.902585268f, -0.430511087f}, {-0.905383646f, -0.424594522f},
|
||||
{-0.908143163f, -0.418659747f}, {-0.910863817f, -0.412707031f},
|
||||
{-0.913545430f, -0.406736642f}, {-0.916187942f, -0.400748819f},
|
||||
{-0.918791234f, -0.394743860f}, {-0.921355128f, -0.388721973f},
|
||||
{-0.923879504f, -0.382683426f}, {-0.926364362f, -0.376628488f},
|
||||
{-0.928809524f, -0.370557427f}, {-0.931214929f, -0.364470512f},
|
||||
{-0.933580399f, -0.358367950f}, {-0.935905933f, -0.352250040f},
|
||||
{-0.938191354f, -0.346117049f}, {-0.940436542f, -0.339969248f},
|
||||
{-0.942641497f, -0.333806872f}, {-0.944806039f, -0.327630192f},
|
||||
{-0.946930110f, -0.321439475f}, {-0.949013650f, -0.315234989f},
|
||||
{-0.951056540f, -0.309017003f}, {-0.953058660f, -0.302785784f},
|
||||
{-0.955019951f, -0.296541572f}, {-0.956940353f, -0.290284663f},
|
||||
{-0.958819747f, -0.284015357f}, {-0.960658073f, -0.277733833f},
|
||||
{-0.962455213f, -0.271440446f}, {-0.964211166f, -0.265135437f},
|
||||
{-0.965925813f, -0.258819044f}, {-0.967599094f, -0.252491564f},
|
||||
{-0.969230890f, -0.246153295f}, {-0.970821202f, -0.239804462f},
|
||||
{-0.972369909f, -0.233445361f}, {-0.973876953f, -0.227076262f},
|
||||
{-0.975342333f, -0.220697433f}, {-0.976765871f, -0.214309156f},
|
||||
{-0.978147626f, -0.207911685f}, {-0.979487419f, -0.201505318f},
|
||||
{-0.980785251f, -0.195090324f}, {-0.982041121f, -0.188666970f},
|
||||
{-0.983254910f, -0.182235524f}, {-0.984426558f, -0.175796285f},
|
||||
{-0.985556066f, -0.169349506f}, {-0.986643314f, -0.162895471f},
|
||||
{-0.987688363f, -0.156434461f}, {-0.988691032f, -0.149966761f},
|
||||
{-0.989651382f, -0.143492624f}, {-0.990569353f, -0.137012348f},
|
||||
{-0.991444886f, -0.130526185f}, {-0.992277920f, -0.124034449f},
|
||||
{-0.993068457f, -0.117537394f}, {-0.993816435f, -0.111035310f},
|
||||
{-0.994521916f, -0.104528464f}, {-0.995184720f, -0.0980171412f},
|
||||
{-0.995804906f, -0.0915016159f}, {-0.996382475f, -0.0849821791f},
|
||||
{-0.996917307f, -0.0784590989f}, {-0.997409463f, -0.0719326511f},
|
||||
{-0.997858942f, -0.0654031262f}, {-0.998265624f, -0.0588708036f},
|
||||
{-0.998629510f, -0.0523359552f}, {-0.998950660f, -0.0457988679f},
|
||||
{-0.999229014f, -0.0392598175f}, {-0.999464571f, -0.0327190831f},
|
||||
{-0.999657333f, -0.0261769481f}, {-0.999807239f, -0.0196336918f},
|
||||
{-0.999914348f, -0.0130895954f}, {-0.999978602f, -0.00654493785f},
|
||||
{-1.00000000f, -1.22464685e-16f}, {-0.999978602f, 0.00654493785f},
|
||||
{-0.999914348f, 0.0130895954f}, {-0.999807239f, 0.0196336918f},
|
||||
{-0.999657333f, 0.0261769481f}, {-0.999464571f, 0.0327190831f},
|
||||
{-0.999229014f, 0.0392598175f}, {-0.998950660f, 0.0457988679f},
|
||||
{-0.998629510f, 0.0523359552f}, {-0.998265624f, 0.0588708036f},
|
||||
{-0.997858942f, 0.0654031262f}, {-0.997409463f, 0.0719326511f},
|
||||
{-0.996917307f, 0.0784590989f}, {-0.996382475f, 0.0849821791f},
|
||||
{-0.995804906f, 0.0915016159f}, {-0.995184720f, 0.0980171412f},
|
||||
{-0.994521916f, 0.104528464f}, {-0.993816435f, 0.111035310f},
|
||||
{-0.993068457f, 0.117537394f}, {-0.992277920f, 0.124034449f},
|
||||
{-0.991444886f, 0.130526185f}, {-0.990569353f, 0.137012348f},
|
||||
{-0.989651382f, 0.143492624f}, {-0.988691032f, 0.149966761f},
|
||||
{-0.987688363f, 0.156434461f}, {-0.986643314f, 0.162895471f},
|
||||
{-0.985556066f, 0.169349506f}, {-0.984426558f, 0.175796285f},
|
||||
{-0.983254910f, 0.182235524f}, {-0.982041121f, 0.188666970f},
|
||||
{-0.980785251f, 0.195090324f}, {-0.979487419f, 0.201505318f},
|
||||
{-0.978147626f, 0.207911685f}, {-0.976765871f, 0.214309156f},
|
||||
{-0.975342333f, 0.220697433f}, {-0.973876953f, 0.227076262f},
|
||||
{-0.972369909f, 0.233445361f}, {-0.970821202f, 0.239804462f},
|
||||
{-0.969230890f, 0.246153295f}, {-0.967599094f, 0.252491564f},
|
||||
{-0.965925813f, 0.258819044f}, {-0.964211166f, 0.265135437f},
|
||||
{-0.962455213f, 0.271440446f}, {-0.960658073f, 0.277733833f},
|
||||
{-0.958819747f, 0.284015357f}, {-0.956940353f, 0.290284663f},
|
||||
{-0.955019951f, 0.296541572f}, {-0.953058660f, 0.302785784f},
|
||||
{-0.951056540f, 0.309017003f}, {-0.949013650f, 0.315234989f},
|
||||
{-0.946930110f, 0.321439475f}, {-0.944806039f, 0.327630192f},
|
||||
{-0.942641497f, 0.333806872f}, {-0.940436542f, 0.339969248f},
|
||||
{-0.938191354f, 0.346117049f}, {-0.935905933f, 0.352250040f},
|
||||
{-0.933580399f, 0.358367950f}, {-0.931214929f, 0.364470512f},
|
||||
{-0.928809524f, 0.370557427f}, {-0.926364362f, 0.376628488f},
|
||||
{-0.923879504f, 0.382683426f}, {-0.921355128f, 0.388721973f},
|
||||
{-0.918791234f, 0.394743860f}, {-0.916187942f, 0.400748819f},
|
||||
{-0.913545430f, 0.406736642f}, {-0.910863817f, 0.412707031f},
|
||||
{-0.908143163f, 0.418659747f}, {-0.905383646f, 0.424594522f},
|
||||
{-0.902585268f, 0.430511087f}, {-0.899748266f, 0.436409235f},
|
||||
{-0.896872759f, 0.442288697f}, {-0.893958807f, 0.448149204f},
|
||||
{-0.891006529f, 0.453990489f}, {-0.888016105f, 0.459812373f},
|
||||
{-0.884987652f, 0.465614527f}, {-0.881921291f, 0.471396744f},
|
||||
{-0.878817141f, 0.477158755f}, {-0.875675321f, 0.482900351f},
|
||||
{-0.872496009f, 0.488621235f}, {-0.869279325f, 0.494321197f},
|
||||
{-0.866025388f, 0.500000000f}, {-0.862734377f, 0.505657375f},
|
||||
{-0.859406412f, 0.511293113f}, {-0.856041610f, 0.516906917f},
|
||||
{-0.852640152f, 0.522498548f}, {-0.849202156f, 0.528067827f},
|
||||
{-0.845727801f, 0.533614516f}, {-0.842217207f, 0.539138317f},
|
||||
{-0.838670552f, 0.544639051f}, {-0.835087955f, 0.550116420f},
|
||||
{-0.831469595f, 0.555570245f}, {-0.827815652f, 0.561000228f},
|
||||
{-0.824126184f, 0.566406250f}, {-0.820401430f, 0.571787953f},
|
||||
{-0.816641569f, 0.577145219f}, {-0.812846661f, 0.582477689f},
|
||||
{-0.809017003f, 0.587785244f}, {-0.805152655f, 0.593067646f},
|
||||
{-0.801253796f, 0.598324597f}, {-0.797320664f, 0.603555918f},
|
||||
{-0.793353319f, 0.608761430f}, {-0.789352059f, 0.613940835f},
|
||||
{-0.785316944f, 0.619093955f}, {-0.781248152f, 0.624220550f},
|
||||
{-0.777145982f, 0.629320383f}, {-0.773010433f, 0.634393275f},
|
||||
{-0.768841803f, 0.639438987f}, {-0.764640272f, 0.644457340f},
|
||||
{-0.760405958f, 0.649448037f}, {-0.756139100f, 0.654410958f},
|
||||
{-0.751839817f, 0.659345806f}, {-0.747508347f, 0.664252460f},
|
||||
{-0.743144810f, 0.669130623f}, {-0.738749504f, 0.673980117f},
|
||||
{-0.734322488f, 0.678800762f}, {-0.729864061f, 0.683592319f},
|
||||
{-0.725374401f, 0.688354552f}, {-0.720853567f, 0.693087339f},
|
||||
{-0.716301918f, 0.697790444f}, {-0.711719632f, 0.702463686f},
|
||||
{-0.707106769f, 0.707106769f}, {-0.702463686f, 0.711719632f},
|
||||
{-0.697790444f, 0.716301918f}, {-0.693087339f, 0.720853567f},
|
||||
{-0.688354552f, 0.725374401f}, {-0.683592319f, 0.729864061f},
|
||||
{-0.678800762f, 0.734322488f}, {-0.673980117f, 0.738749504f},
|
||||
{-0.669130623f, 0.743144810f}, {-0.664252460f, 0.747508347f},
|
||||
{-0.659345806f, 0.751839817f}, {-0.654410958f, 0.756139100f},
|
||||
{-0.649448037f, 0.760405958f}, {-0.644457340f, 0.764640272f},
|
||||
{-0.639438987f, 0.768841803f}, {-0.634393275f, 0.773010433f},
|
||||
{-0.629320383f, 0.777145982f}, {-0.624220550f, 0.781248152f},
|
||||
{-0.619093955f, 0.785316944f}, {-0.613940835f, 0.789352059f},
|
||||
{-0.608761430f, 0.793353319f}, {-0.603555918f, 0.797320664f},
|
||||
{-0.598324597f, 0.801253796f}, {-0.593067646f, 0.805152655f},
|
||||
{-0.587785244f, 0.809017003f}, {-0.582477689f, 0.812846661f},
|
||||
{-0.577145219f, 0.816641569f}, {-0.571787953f, 0.820401430f},
|
||||
{-0.566406250f, 0.824126184f}, {-0.561000228f, 0.827815652f},
|
||||
{-0.555570245f, 0.831469595f}, {-0.550116420f, 0.835087955f},
|
||||
{-0.544639051f, 0.838670552f}, {-0.539138317f, 0.842217207f},
|
||||
{-0.533614516f, 0.845727801f}, {-0.528067827f, 0.849202156f},
|
||||
{-0.522498548f, 0.852640152f}, {-0.516906917f, 0.856041610f},
|
||||
{-0.511293113f, 0.859406412f}, {-0.505657375f, 0.862734377f},
|
||||
{-0.500000000f, 0.866025388f}, {-0.494321197f, 0.869279325f},
|
||||
{-0.488621235f, 0.872496009f}, {-0.482900351f, 0.875675321f},
|
||||
{-0.477158755f, 0.878817141f}, {-0.471396744f, 0.881921291f},
|
||||
{-0.465614527f, 0.884987652f}, {-0.459812373f, 0.888016105f},
|
||||
{-0.453990489f, 0.891006529f}, {-0.448149204f, 0.893958807f},
|
||||
{-0.442288697f, 0.896872759f}, {-0.436409235f, 0.899748266f},
|
||||
{-0.430511087f, 0.902585268f}, {-0.424594522f, 0.905383646f},
|
||||
{-0.418659747f, 0.908143163f}, {-0.412707031f, 0.910863817f},
|
||||
{-0.406736642f, 0.913545430f}, {-0.400748819f, 0.916187942f},
|
||||
{-0.394743860f, 0.918791234f}, {-0.388721973f, 0.921355128f},
|
||||
{-0.382683426f, 0.923879504f}, {-0.376628488f, 0.926364362f},
|
||||
{-0.370557427f, 0.928809524f}, {-0.364470512f, 0.931214929f},
|
||||
{-0.358367950f, 0.933580399f}, {-0.352250040f, 0.935905933f},
|
||||
{-0.346117049f, 0.938191354f}, {-0.339969248f, 0.940436542f},
|
||||
{-0.333806872f, 0.942641497f}, {-0.327630192f, 0.944806039f},
|
||||
{-0.321439475f, 0.946930110f}, {-0.315234989f, 0.949013650f},
|
||||
{-0.309017003f, 0.951056540f}, {-0.302785784f, 0.953058660f},
|
||||
{-0.296541572f, 0.955019951f}, {-0.290284663f, 0.956940353f},
|
||||
{-0.284015357f, 0.958819747f}, {-0.277733833f, 0.960658073f},
|
||||
{-0.271440446f, 0.962455213f}, {-0.265135437f, 0.964211166f},
|
||||
{-0.258819044f, 0.965925813f}, {-0.252491564f, 0.967599094f},
|
||||
{-0.246153295f, 0.969230890f}, {-0.239804462f, 0.970821202f},
|
||||
{-0.233445361f, 0.972369909f}, {-0.227076262f, 0.973876953f},
|
||||
{-0.220697433f, 0.975342333f}, {-0.214309156f, 0.976765871f},
|
||||
{-0.207911685f, 0.978147626f}, {-0.201505318f, 0.979487419f},
|
||||
{-0.195090324f, 0.980785251f}, {-0.188666970f, 0.982041121f},
|
||||
{-0.182235524f, 0.983254910f}, {-0.175796285f, 0.984426558f},
|
||||
{-0.169349506f, 0.985556066f}, {-0.162895471f, 0.986643314f},
|
||||
{-0.156434461f, 0.987688363f}, {-0.149966761f, 0.988691032f},
|
||||
{-0.143492624f, 0.989651382f}, {-0.137012348f, 0.990569353f},
|
||||
{-0.130526185f, 0.991444886f}, {-0.124034449f, 0.992277920f},
|
||||
{-0.117537394f, 0.993068457f}, {-0.111035310f, 0.993816435f},
|
||||
{-0.104528464f, 0.994521916f}, {-0.0980171412f, 0.995184720f},
|
||||
{-0.0915016159f, 0.995804906f}, {-0.0849821791f, 0.996382475f},
|
||||
{-0.0784590989f, 0.996917307f}, {-0.0719326511f, 0.997409463f},
|
||||
{-0.0654031262f, 0.997858942f}, {-0.0588708036f, 0.998265624f},
|
||||
{-0.0523359552f, 0.998629510f}, {-0.0457988679f, 0.998950660f},
|
||||
{-0.0392598175f, 0.999229014f}, {-0.0327190831f, 0.999464571f},
|
||||
{-0.0261769481f, 0.999657333f}, {-0.0196336918f, 0.999807239f},
|
||||
{-0.0130895954f, 0.999914348f}, {-0.00654493785f, 0.999978602f},
|
||||
{-1.83697015e-16f, 1.00000000f}, {0.00654493785f, 0.999978602f},
|
||||
{0.0130895954f, 0.999914348f}, {0.0196336918f, 0.999807239f},
|
||||
{0.0261769481f, 0.999657333f}, {0.0327190831f, 0.999464571f},
|
||||
{0.0392598175f, 0.999229014f}, {0.0457988679f, 0.998950660f},
|
||||
{0.0523359552f, 0.998629510f}, {0.0588708036f, 0.998265624f},
|
||||
{0.0654031262f, 0.997858942f}, {0.0719326511f, 0.997409463f},
|
||||
{0.0784590989f, 0.996917307f}, {0.0849821791f, 0.996382475f},
|
||||
{0.0915016159f, 0.995804906f}, {0.0980171412f, 0.995184720f},
|
||||
{0.104528464f, 0.994521916f}, {0.111035310f, 0.993816435f},
|
||||
{0.117537394f, 0.993068457f}, {0.124034449f, 0.992277920f},
|
||||
{0.130526185f, 0.991444886f}, {0.137012348f, 0.990569353f},
|
||||
{0.143492624f, 0.989651382f}, {0.149966761f, 0.988691032f},
|
||||
{0.156434461f, 0.987688363f}, {0.162895471f, 0.986643314f},
|
||||
{0.169349506f, 0.985556066f}, {0.175796285f, 0.984426558f},
|
||||
{0.182235524f, 0.983254910f}, {0.188666970f, 0.982041121f},
|
||||
{0.195090324f, 0.980785251f}, {0.201505318f, 0.979487419f},
|
||||
{0.207911685f, 0.978147626f}, {0.214309156f, 0.976765871f},
|
||||
{0.220697433f, 0.975342333f}, {0.227076262f, 0.973876953f},
|
||||
{0.233445361f, 0.972369909f}, {0.239804462f, 0.970821202f},
|
||||
{0.246153295f, 0.969230890f}, {0.252491564f, 0.967599094f},
|
||||
{0.258819044f, 0.965925813f}, {0.265135437f, 0.964211166f},
|
||||
{0.271440446f, 0.962455213f}, {0.277733833f, 0.960658073f},
|
||||
{0.284015357f, 0.958819747f}, {0.290284663f, 0.956940353f},
|
||||
{0.296541572f, 0.955019951f}, {0.302785784f, 0.953058660f},
|
||||
{0.309017003f, 0.951056540f}, {0.315234989f, 0.949013650f},
|
||||
{0.321439475f, 0.946930110f}, {0.327630192f, 0.944806039f},
|
||||
{0.333806872f, 0.942641497f}, {0.339969248f, 0.940436542f},
|
||||
{0.346117049f, 0.938191354f}, {0.352250040f, 0.935905933f},
|
||||
{0.358367950f, 0.933580399f}, {0.364470512f, 0.931214929f},
|
||||
{0.370557427f, 0.928809524f}, {0.376628488f, 0.926364362f},
|
||||
{0.382683426f, 0.923879504f}, {0.388721973f, 0.921355128f},
|
||||
{0.394743860f, 0.918791234f}, {0.400748819f, 0.916187942f},
|
||||
{0.406736642f, 0.913545430f}, {0.412707031f, 0.910863817f},
|
||||
{0.418659747f, 0.908143163f}, {0.424594522f, 0.905383646f},
|
||||
{0.430511087f, 0.902585268f}, {0.436409235f, 0.899748266f},
|
||||
{0.442288697f, 0.896872759f}, {0.448149204f, 0.893958807f},
|
||||
{0.453990489f, 0.891006529f}, {0.459812373f, 0.888016105f},
|
||||
{0.465614527f, 0.884987652f}, {0.471396744f, 0.881921291f},
|
||||
{0.477158755f, 0.878817141f}, {0.482900351f, 0.875675321f},
|
||||
{0.488621235f, 0.872496009f}, {0.494321197f, 0.869279325f},
|
||||
{0.500000000f, 0.866025388f}, {0.505657375f, 0.862734377f},
|
||||
{0.511293113f, 0.859406412f}, {0.516906917f, 0.856041610f},
|
||||
{0.522498548f, 0.852640152f}, {0.528067827f, 0.849202156f},
|
||||
{0.533614516f, 0.845727801f}, {0.539138317f, 0.842217207f},
|
||||
{0.544639051f, 0.838670552f}, {0.550116420f, 0.835087955f},
|
||||
{0.555570245f, 0.831469595f}, {0.561000228f, 0.827815652f},
|
||||
{0.566406250f, 0.824126184f}, {0.571787953f, 0.820401430f},
|
||||
{0.577145219f, 0.816641569f}, {0.582477689f, 0.812846661f},
|
||||
{0.587785244f, 0.809017003f}, {0.593067646f, 0.805152655f},
|
||||
{0.598324597f, 0.801253796f}, {0.603555918f, 0.797320664f},
|
||||
{0.608761430f, 0.793353319f}, {0.613940835f, 0.789352059f},
|
||||
{0.619093955f, 0.785316944f}, {0.624220550f, 0.781248152f},
|
||||
{0.629320383f, 0.777145982f}, {0.634393275f, 0.773010433f},
|
||||
{0.639438987f, 0.768841803f}, {0.644457340f, 0.764640272f},
|
||||
{0.649448037f, 0.760405958f}, {0.654410958f, 0.756139100f},
|
||||
{0.659345806f, 0.751839817f}, {0.664252460f, 0.747508347f},
|
||||
{0.669130623f, 0.743144810f}, {0.673980117f, 0.738749504f},
|
||||
{0.678800762f, 0.734322488f}, {0.683592319f, 0.729864061f},
|
||||
{0.688354552f, 0.725374401f}, {0.693087339f, 0.720853567f},
|
||||
{0.697790444f, 0.716301918f}, {0.702463686f, 0.711719632f},
|
||||
{0.707106769f, 0.707106769f}, {0.711719632f, 0.702463686f},
|
||||
{0.716301918f, 0.697790444f}, {0.720853567f, 0.693087339f},
|
||||
{0.725374401f, 0.688354552f}, {0.729864061f, 0.683592319f},
|
||||
{0.734322488f, 0.678800762f}, {0.738749504f, 0.673980117f},
|
||||
{0.743144810f, 0.669130623f}, {0.747508347f, 0.664252460f},
|
||||
{0.751839817f, 0.659345806f}, {0.756139100f, 0.654410958f},
|
||||
{0.760405958f, 0.649448037f}, {0.764640272f, 0.644457340f},
|
||||
{0.768841803f, 0.639438987f}, {0.773010433f, 0.634393275f},
|
||||
{0.777145982f, 0.629320383f}, {0.781248152f, 0.624220550f},
|
||||
{0.785316944f, 0.619093955f}, {0.789352059f, 0.613940835f},
|
||||
{0.793353319f, 0.608761430f}, {0.797320664f, 0.603555918f},
|
||||
{0.801253796f, 0.598324597f}, {0.805152655f, 0.593067646f},
|
||||
{0.809017003f, 0.587785244f}, {0.812846661f, 0.582477689f},
|
||||
{0.816641569f, 0.577145219f}, {0.820401430f, 0.571787953f},
|
||||
{0.824126184f, 0.566406250f}, {0.827815652f, 0.561000228f},
|
||||
{0.831469595f, 0.555570245f}, {0.835087955f, 0.550116420f},
|
||||
{0.838670552f, 0.544639051f}, {0.842217207f, 0.539138317f},
|
||||
{0.845727801f, 0.533614516f}, {0.849202156f, 0.528067827f},
|
||||
{0.852640152f, 0.522498548f}, {0.856041610f, 0.516906917f},
|
||||
{0.859406412f, 0.511293113f}, {0.862734377f, 0.505657375f},
|
||||
{0.866025388f, 0.500000000f}, {0.869279325f, 0.494321197f},
|
||||
{0.872496009f, 0.488621235f}, {0.875675321f, 0.482900351f},
|
||||
{0.878817141f, 0.477158755f}, {0.881921291f, 0.471396744f},
|
||||
{0.884987652f, 0.465614527f}, {0.888016105f, 0.459812373f},
|
||||
{0.891006529f, 0.453990489f}, {0.893958807f, 0.448149204f},
|
||||
{0.896872759f, 0.442288697f}, {0.899748266f, 0.436409235f},
|
||||
{0.902585268f, 0.430511087f}, {0.905383646f, 0.424594522f},
|
||||
{0.908143163f, 0.418659747f}, {0.910863817f, 0.412707031f},
|
||||
{0.913545430f, 0.406736642f}, {0.916187942f, 0.400748819f},
|
||||
{0.918791234f, 0.394743860f}, {0.921355128f, 0.388721973f},
|
||||
{0.923879504f, 0.382683426f}, {0.926364362f, 0.376628488f},
|
||||
{0.928809524f, 0.370557427f}, {0.931214929f, 0.364470512f},
|
||||
{0.933580399f, 0.358367950f}, {0.935905933f, 0.352250040f},
|
||||
{0.938191354f, 0.346117049f}, {0.940436542f, 0.339969248f},
|
||||
{0.942641497f, 0.333806872f}, {0.944806039f, 0.327630192f},
|
||||
{0.946930110f, 0.321439475f}, {0.949013650f, 0.315234989f},
|
||||
{0.951056540f, 0.309017003f}, {0.953058660f, 0.302785784f},
|
||||
{0.955019951f, 0.296541572f}, {0.956940353f, 0.290284663f},
|
||||
{0.958819747f, 0.284015357f}, {0.960658073f, 0.277733833f},
|
||||
{0.962455213f, 0.271440446f}, {0.964211166f, 0.265135437f},
|
||||
{0.965925813f, 0.258819044f}, {0.967599094f, 0.252491564f},
|
||||
{0.969230890f, 0.246153295f}, {0.970821202f, 0.239804462f},
|
||||
{0.972369909f, 0.233445361f}, {0.973876953f, 0.227076262f},
|
||||
{0.975342333f, 0.220697433f}, {0.976765871f, 0.214309156f},
|
||||
{0.978147626f, 0.207911685f}, {0.979487419f, 0.201505318f},
|
||||
{0.980785251f, 0.195090324f}, {0.982041121f, 0.188666970f},
|
||||
{0.983254910f, 0.182235524f}, {0.984426558f, 0.175796285f},
|
||||
{0.985556066f, 0.169349506f}, {0.986643314f, 0.162895471f},
|
||||
{0.987688363f, 0.156434461f}, {0.988691032f, 0.149966761f},
|
||||
{0.989651382f, 0.143492624f}, {0.990569353f, 0.137012348f},
|
||||
{0.991444886f, 0.130526185f}, {0.992277920f, 0.124034449f},
|
||||
{0.993068457f, 0.117537394f}, {0.993816435f, 0.111035310f},
|
||||
{0.994521916f, 0.104528464f}, {0.995184720f, 0.0980171412f},
|
||||
{0.995804906f, 0.0915016159f}, {0.996382475f, 0.0849821791f},
|
||||
{0.996917307f, 0.0784590989f}, {0.997409463f, 0.0719326511f},
|
||||
{0.997858942f, 0.0654031262f}, {0.998265624f, 0.0588708036f},
|
||||
{0.998629510f, 0.0523359552f}, {0.998950660f, 0.0457988679f},
|
||||
{0.999229014f, 0.0392598175f}, {0.999464571f, 0.0327190831f},
|
||||
{0.999657333f, 0.0261769481f}, {0.999807239f, 0.0196336918f},
|
||||
{0.999914348f, 0.0130895954f}, {0.999978602f, 0.00654493785f},
|
||||
};
|
||||
|
||||
const kiss_fft_state rnn_kfft = {
|
||||
960, /* nfft */
|
||||
0.0010416667f, /* scale */
|
||||
-1, /* shift */
|
||||
{5, 192, 3, 64, 4, 16, 4, 4, 4, 1, 0, 0, 0, 0, 0, 0, }, /* factors */
|
||||
fft_bitrev, /* bitrev*/
|
||||
fft_twiddles, /* twiddles*/
|
||||
(arch_fft_state *)&arch_fft, /* arch_fft*/
|
||||
};
|
||||
|
||||
const float rnn_half_window[] = {
|
||||
4.20549168e-06f, 3.78491532e-05f, 0.000105135041f, 0.000206060256f, 0.000340620492f,
|
||||
0.000508809986f, 0.000710621476f, 0.000946046319f, 0.00121507444f, 0.00151769421f,
|
||||
0.00185389258f, 0.00222365512f, 0.00262696599f, 0.00306380726f, 0.00353416055f,
|
||||
0.00403800514f, 0.00457531959f, 0.00514607970f, 0.00575026125f, 0.00638783723f,
|
||||
0.00705878017f, 0.00776306028f, 0.00850064680f, 0.00927150715f, 0.0100756064f,
|
||||
0.0109129101f, 0.0117833801f, 0.0126869772f, 0.0136236614f, 0.0145933898f,
|
||||
0.0155961197f, 0.0166318044f, 0.0177003983f, 0.0188018531f, 0.0199361145f,
|
||||
0.0211031344f, 0.0223028567f, 0.0235352255f, 0.0248001851f, 0.0260976739f,
|
||||
0.0274276342f, 0.0287899990f, 0.0301847085f, 0.0316116922f, 0.0330708846f,
|
||||
0.0345622115f, 0.0360856056f, 0.0376409888f, 0.0392282903f, 0.0408474281f,
|
||||
0.0424983241f, 0.0441808924f, 0.0458950549f, 0.0476407260f, 0.0494178124f,
|
||||
0.0512262285f, 0.0530658774f, 0.0549366735f, 0.0568385124f, 0.0587713011f,
|
||||
0.0607349351f, 0.0627293140f, 0.0647543296f, 0.0668098852f, 0.0688958541f,
|
||||
0.0710121393f, 0.0731586292f, 0.0753351897f, 0.0775417164f, 0.0797780901f,
|
||||
0.0820441842f, 0.0843398646f, 0.0866650119f, 0.0890194997f, 0.0914031938f,
|
||||
0.0938159525f, 0.0962576419f, 0.0987281203f, 0.101227246f, 0.103754878f,
|
||||
0.106310867f, 0.108895063f, 0.111507311f, 0.114147455f, 0.116815343f,
|
||||
0.119510807f, 0.122233689f, 0.124983832f, 0.127761051f, 0.130565181f,
|
||||
0.133396059f, 0.136253506f, 0.139137328f, 0.142047361f, 0.144983411f,
|
||||
0.147945285f, 0.150932819f, 0.153945804f, 0.156984031f, 0.160047337f,
|
||||
0.163135484f, 0.166248307f, 0.169385567f, 0.172547072f, 0.175732598f,
|
||||
0.178941950f, 0.182174906f, 0.185431242f, 0.188710734f, 0.192013159f,
|
||||
0.195338294f, 0.198685899f, 0.202055752f, 0.205447599f, 0.208861232f,
|
||||
0.212296382f, 0.215752810f, 0.219230279f, 0.222728521f, 0.226247311f,
|
||||
0.229786381f, 0.233345464f, 0.236924306f, 0.240522653f, 0.244140238f,
|
||||
0.247776777f, 0.251432031f, 0.255105674f, 0.258797467f, 0.262507141f,
|
||||
0.266234398f, 0.269978970f, 0.273740560f, 0.277518868f, 0.281313598f,
|
||||
0.285124481f, 0.288951218f, 0.292793512f, 0.296651065f, 0.300523549f,
|
||||
0.304410696f, 0.308312178f, 0.312227666f, 0.316156894f, 0.320099503f,
|
||||
0.324055225f, 0.328023702f, 0.332004637f, 0.335997701f, 0.340002567f,
|
||||
0.344018906f, 0.348046392f, 0.352084726f, 0.356133521f, 0.360192508f,
|
||||
0.364261299f, 0.368339598f, 0.372427016f, 0.376523286f, 0.380627990f,
|
||||
0.384740859f, 0.388861477f, 0.392989576f, 0.397124738f, 0.401266664f,
|
||||
0.405414969f, 0.409569323f, 0.413729399f, 0.417894781f, 0.422065198f,
|
||||
0.426240236f, 0.430419534f, 0.434602767f, 0.438789606f, 0.442979604f,
|
||||
0.447172493f, 0.451367885f, 0.455565393f, 0.459764689f, 0.463965416f,
|
||||
0.468167186f, 0.472369671f, 0.476572484f, 0.480775267f, 0.484977663f,
|
||||
0.489179343f, 0.493379891f, 0.497579008f, 0.501776278f, 0.505971372f,
|
||||
0.510163903f, 0.514353573f, 0.518539906f, 0.522722721f, 0.526901484f,
|
||||
0.531075954f, 0.535245717f, 0.539410412f, 0.543569744f, 0.547723293f,
|
||||
0.551870763f, 0.556011736f, 0.560145974f, 0.564273000f, 0.568392515f,
|
||||
0.572504222f, 0.576607704f, 0.580702662f, 0.584788740f, 0.588865638f,
|
||||
0.592932940f, 0.596990347f, 0.601037502f, 0.605074167f, 0.609099925f,
|
||||
0.613114417f, 0.617117405f, 0.621108532f, 0.625087440f, 0.629053831f,
|
||||
0.633007407f, 0.636947870f, 0.640874863f, 0.644788086f, 0.648687243f,
|
||||
0.652572036f, 0.656442165f, 0.660297334f, 0.664137185f, 0.667961538f,
|
||||
0.671769977f, 0.675562322f, 0.679338276f, 0.683097482f, 0.686839759f,
|
||||
0.690564752f, 0.694272280f, 0.697961986f, 0.701633692f, 0.705287039f,
|
||||
0.708921850f, 0.712537885f, 0.716134787f, 0.719712436f, 0.723270535f,
|
||||
0.726808906f, 0.730327189f, 0.733825266f, 0.737302899f, 0.740759790f,
|
||||
0.744195819f, 0.747610688f, 0.751004279f, 0.754376352f, 0.757726669f,
|
||||
0.761055112f, 0.764361382f, 0.767645359f, 0.770906866f, 0.774145722f,
|
||||
0.777361751f, 0.780554771f, 0.783724606f, 0.786871076f, 0.789994121f,
|
||||
0.793093503f, 0.796169102f, 0.799220800f, 0.802248418f, 0.805251837f,
|
||||
0.808230937f, 0.811185598f, 0.814115703f, 0.817021132f, 0.819901764f,
|
||||
0.822757542f, 0.825588286f, 0.828393936f, 0.831174433f, 0.833929658f,
|
||||
0.836659551f, 0.839363992f, 0.842042983f, 0.844696403f, 0.847324252f,
|
||||
0.849926353f, 0.852502763f, 0.855053425f, 0.857578218f, 0.860077202f,
|
||||
0.862550259f, 0.864997447f, 0.867418647f, 0.869813919f, 0.872183204f,
|
||||
0.874526560f, 0.876843870f, 0.879135191f, 0.881400526f, 0.883639932f,
|
||||
0.885853291f, 0.888040781f, 0.890202343f, 0.892337978f, 0.894447744f,
|
||||
0.896531701f, 0.898589849f, 0.900622249f, 0.902628958f, 0.904610038f,
|
||||
0.906565487f, 0.908495426f, 0.910399914f, 0.912279010f, 0.914132774f,
|
||||
0.915961266f, 0.917764664f, 0.919542909f, 0.921296239f, 0.923024654f,
|
||||
0.924728215f, 0.926407158f, 0.928061485f, 0.929691315f, 0.931296766f,
|
||||
0.932878017f, 0.934435070f, 0.935968161f, 0.937477291f, 0.938962698f,
|
||||
0.940424502f, 0.941862822f, 0.943277776f, 0.944669485f, 0.946038187f,
|
||||
0.947383940f, 0.948706925f, 0.950007319f, 0.951285243f, 0.952540874f,
|
||||
0.953774393f, 0.954985917f, 0.956175685f, 0.957343817f, 0.958490491f,
|
||||
0.959615886f, 0.960720181f, 0.961803555f, 0.962866247f, 0.963908315f,
|
||||
0.964930058f, 0.965931594f, 0.966913164f, 0.967874944f, 0.968817174f,
|
||||
0.969739914f, 0.970643520f, 0.971528113f, 0.972393870f, 0.973241091f,
|
||||
0.974069893f, 0.974880517f, 0.975673139f, 0.976447999f, 0.977205336f,
|
||||
0.977945268f, 0.978668094f, 0.979374051f, 0.980063200f, 0.980735898f,
|
||||
0.981392324f, 0.982032716f, 0.982657254f, 0.983266115f, 0.983859658f,
|
||||
0.984437943f, 0.985001266f, 0.985549867f, 0.986083925f, 0.986603677f,
|
||||
0.987109363f, 0.987601161f, 0.988079309f, 0.988544047f, 0.988995552f,
|
||||
0.989434063f, 0.989859879f, 0.990273118f, 0.990674019f, 0.991062820f,
|
||||
0.991439700f, 0.991804957f, 0.992158771f, 0.992501318f, 0.992832899f,
|
||||
0.993153632f, 0.993463814f, 0.993763626f, 0.994053245f, 0.994332969f,
|
||||
0.994602919f, 0.994863331f, 0.995114446f, 0.995356441f, 0.995589554f,
|
||||
0.995813966f, 0.996029854f, 0.996237516f, 0.996437073f, 0.996628702f,
|
||||
0.996812642f, 0.996989131f, 0.997158289f, 0.997320294f, 0.997475445f,
|
||||
0.997623861f, 0.997765720f, 0.997901261f, 0.998030603f, 0.998153925f,
|
||||
0.998271465f, 0.998383403f, 0.998489857f, 0.998591006f, 0.998687088f,
|
||||
0.998778164f, 0.998864532f, 0.998946249f, 0.999023557f, 0.999096513f,
|
||||
0.999165416f, 0.999230266f, 0.999291301f, 0.999348700f, 0.999402523f,
|
||||
0.999453008f, 0.999500215f, 0.999544322f, 0.999585509f, 0.999623775f,
|
||||
0.999659419f, 0.999692440f, 0.999723017f, 0.999751270f, 0.999777317f,
|
||||
0.999801278f, 0.999823213f, 0.999843359f, 0.999861658f, 0.999878347f,
|
||||
0.999893486f, 0.999907196f, 0.999919534f, 0.999930561f, 0.999940455f,
|
||||
0.999949217f, 0.999957025f, 0.999963880f, 0.999969840f, 0.999975085f,
|
||||
0.999979615f, 0.999983490f, 0.999986768f, 0.999989510f, 0.999991834f,
|
||||
0.999993742f, 0.999995291f, 0.999996543f, 0.999997556f, 0.999998271f,
|
||||
0.999998868f, 0.999999285f, 0.999999523f, 0.999999762f, 0.999999881f,
|
||||
0.999999940f, 1.00000000f, 1.00000000f, 1.00000000f, 1.00000000f,
|
||||
};
|
||||
|
||||
const float rnn_dct_table[] = {
|
||||
0.707106769f, 0.998795450f, 0.995184720f, 0.989176512f, 0.980785251f,
|
||||
0.970031261f, 0.956940353f, 0.941544056f, 0.923879504f, 0.903989315f,
|
||||
0.881921291f, 0.857728601f, 0.831469595f, 0.803207517f, 0.773010433f,
|
||||
0.740951121f, 0.707106769f, 0.671558976f, 0.634393275f, 0.595699310f,
|
||||
0.555570245f, 0.514102757f, 0.471396744f, 0.427555084f, 0.382683426f,
|
||||
0.336889863f, 0.290284663f, 0.242980182f, 0.195090324f, 0.146730468f,
|
||||
0.0980171412f, 0.0490676761f, 0.707106769f, 0.989176512f, 0.956940353f,
|
||||
0.903989315f, 0.831469595f, 0.740951121f, 0.634393275f, 0.514102757f,
|
||||
0.382683426f, 0.242980182f, 0.0980171412f, -0.0490676761f, -0.195090324f,
|
||||
-0.336889863f, -0.471396744f, -0.595699310f, -0.707106769f, -0.803207517f,
|
||||
-0.881921291f, -0.941544056f, -0.980785251f, -0.998795450f, -0.995184720f,
|
||||
-0.970031261f, -0.923879504f, -0.857728601f, -0.773010433f, -0.671558976f,
|
||||
-0.555570245f, -0.427555084f, -0.290284663f, -0.146730468f, 0.707106769f,
|
||||
0.970031261f, 0.881921291f, 0.740951121f, 0.555570245f, 0.336889863f,
|
||||
0.0980171412f, -0.146730468f, -0.382683426f, -0.595699310f, -0.773010433f,
|
||||
-0.903989315f, -0.980785251f, -0.998795450f, -0.956940353f, -0.857728601f,
|
||||
-0.707106769f, -0.514102757f, -0.290284663f, -0.0490676761f, 0.195090324f,
|
||||
0.427555084f, 0.634393275f, 0.803207517f, 0.923879504f, 0.989176512f,
|
||||
0.995184720f, 0.941544056f, 0.831469595f, 0.671558976f, 0.471396744f,
|
||||
0.242980182f, 0.707106769f, 0.941544056f, 0.773010433f, 0.514102757f,
|
||||
0.195090324f, -0.146730468f, -0.471396744f, -0.740951121f, -0.923879504f,
|
||||
-0.998795450f, -0.956940353f, -0.803207517f, -0.555570245f, -0.242980182f,
|
||||
0.0980171412f, 0.427555084f, 0.707106769f, 0.903989315f, 0.995184720f,
|
||||
0.970031261f, 0.831469595f, 0.595699310f, 0.290284663f, -0.0490676761f,
|
||||
-0.382683426f, -0.671558976f, -0.881921291f, -0.989176512f, -0.980785251f,
|
||||
-0.857728601f, -0.634393275f, -0.336889863f, 0.707106769f, 0.903989315f,
|
||||
0.634393275f, 0.242980182f, -0.195090324f, -0.595699310f, -0.881921291f,
|
||||
-0.998795450f, -0.923879504f, -0.671558976f, -0.290284663f, 0.146730468f,
|
||||
0.555570245f, 0.857728601f, 0.995184720f, 0.941544056f, 0.707106769f,
|
||||
0.336889863f, -0.0980171412f, -0.514102757f, -0.831469595f, -0.989176512f,
|
||||
-0.956940353f, -0.740951121f, -0.382683426f, 0.0490676761f, 0.471396744f,
|
||||
0.803207517f, 0.980785251f, 0.970031261f, 0.773010433f, 0.427555084f,
|
||||
0.707106769f, 0.857728601f, 0.471396744f, -0.0490676761f, -0.555570245f,
|
||||
-0.903989315f, -0.995184720f, -0.803207517f, -0.382683426f, 0.146730468f,
|
||||
0.634393275f, 0.941544056f, 0.980785251f, 0.740951121f, 0.290284663f,
|
||||
-0.242980182f, -0.707106769f, -0.970031261f, -0.956940353f, -0.671558976f,
|
||||
-0.195090324f, 0.336889863f, 0.773010433f, 0.989176512f, 0.923879504f,
|
||||
0.595699310f, 0.0980171412f, -0.427555084f, -0.831469595f, -0.998795450f,
|
||||
-0.881921291f, -0.514102757f, 0.707106769f, 0.803207517f, 0.290284663f,
|
||||
-0.336889863f, -0.831469595f, -0.998795450f, -0.773010433f, -0.242980182f,
|
||||
0.382683426f, 0.857728601f, 0.995184720f, 0.740951121f, 0.195090324f,
|
||||
-0.427555084f, -0.881921291f, -0.989176512f, -0.707106769f, -0.146730468f,
|
||||
0.471396744f, 0.903989315f, 0.980785251f, 0.671558976f, 0.0980171412f,
|
||||
-0.514102757f, -0.923879504f, -0.970031261f, -0.634393275f, -0.0490676761f,
|
||||
0.555570245f, 0.941544056f, 0.956940353f, 0.595699310f, 0.707106769f,
|
||||
0.740951121f, 0.0980171412f, -0.595699310f, -0.980785251f, -0.857728601f,
|
||||
-0.290284663f, 0.427555084f, 0.923879504f, 0.941544056f, 0.471396744f,
|
||||
-0.242980182f, -0.831469595f, -0.989176512f, -0.634393275f, 0.0490676761f,
|
||||
0.707106769f, 0.998795450f, 0.773010433f, 0.146730468f, -0.555570245f,
|
||||
-0.970031261f, -0.881921291f, -0.336889863f, 0.382683426f, 0.903989315f,
|
||||
0.956940353f, 0.514102757f, -0.195090324f, -0.803207517f, -0.995184720f,
|
||||
-0.671558976f, 0.707106769f, 0.671558976f, -0.0980171412f, -0.803207517f,
|
||||
-0.980785251f, -0.514102757f, 0.290284663f, 0.903989315f, 0.923879504f,
|
||||
0.336889863f, -0.471396744f, -0.970031261f, -0.831469595f, -0.146730468f,
|
||||
0.634393275f, 0.998795450f, 0.707106769f, -0.0490676761f, -0.773010433f,
|
||||
-0.989176512f, -0.555570245f, 0.242980182f, 0.881921291f, 0.941544056f,
|
||||
0.382683426f, -0.427555084f, -0.956940353f, -0.857728601f, -0.195090324f,
|
||||
0.595699310f, 0.995184720f, 0.740951121f, 0.707106769f, 0.595699310f,
|
||||
-0.290284663f, -0.941544056f, -0.831469595f, -0.0490676761f, 0.773010433f,
|
||||
0.970031261f, 0.382683426f, -0.514102757f, -0.995184720f, -0.671558976f,
|
||||
0.195090324f, 0.903989315f, 0.881921291f, 0.146730468f, -0.707106769f,
|
||||
-0.989176512f, -0.471396744f, 0.427555084f, 0.980785251f, 0.740951121f,
|
||||
-0.0980171412f, -0.857728601f, -0.923879504f, -0.242980182f, 0.634393275f,
|
||||
0.998795450f, 0.555570245f, -0.336889863f, -0.956940353f, -0.803207517f,
|
||||
0.707106769f, 0.514102757f, -0.471396744f, -0.998795450f, -0.555570245f,
|
||||
0.427555084f, 0.995184720f, 0.595699310f, -0.382683426f, -0.989176512f,
|
||||
-0.634393275f, 0.336889863f, 0.980785251f, 0.671558976f, -0.290284663f,
|
||||
-0.970031261f, -0.707106769f, 0.242980182f, 0.956940353f, 0.740951121f,
|
||||
-0.195090324f, -0.941544056f, -0.773010433f, 0.146730468f, 0.923879504f,
|
||||
0.803207517f, -0.0980171412f, -0.903989315f, -0.831469595f, 0.0490676761f,
|
||||
0.881921291f, 0.857728601f, 0.707106769f, 0.427555084f, -0.634393275f,
|
||||
-0.970031261f, -0.195090324f, 0.803207517f, 0.881921291f, -0.0490676761f,
|
||||
-0.923879504f, -0.740951121f, 0.290284663f, 0.989176512f, 0.555570245f,
|
||||
-0.514102757f, -0.995184720f, -0.336889863f, 0.707106769f, 0.941544056f,
|
||||
0.0980171412f, -0.857728601f, -0.831469595f, 0.146730468f, 0.956940353f,
|
||||
0.671558976f, -0.382683426f, -0.998795450f, -0.471396744f, 0.595699310f,
|
||||
0.980785251f, 0.242980182f, -0.773010433f, -0.903989315f, 0.707106769f,
|
||||
0.336889863f, -0.773010433f, -0.857728601f, 0.195090324f, 0.989176512f,
|
||||
0.471396744f, -0.671558976f, -0.923879504f, 0.0490676761f, 0.956940353f,
|
||||
0.595699310f, -0.555570245f, -0.970031261f, -0.0980171412f, 0.903989315f,
|
||||
0.707106769f, -0.427555084f, -0.995184720f, -0.242980182f, 0.831469595f,
|
||||
0.803207517f, -0.290284663f, -0.998795450f, -0.382683426f, 0.740951121f,
|
||||
0.881921291f, -0.146730468f, -0.980785251f, -0.514102757f, 0.634393275f,
|
||||
0.941544056f, 0.707106769f, 0.242980182f, -0.881921291f, -0.671558976f,
|
||||
0.555570245f, 0.941544056f, -0.0980171412f, -0.989176512f, -0.382683426f,
|
||||
0.803207517f, 0.773010433f, -0.427555084f, -0.980785251f, -0.0490676761f,
|
||||
0.956940353f, 0.514102757f, -0.707106769f, -0.857728601f, 0.290284663f,
|
||||
0.998795450f, 0.195090324f, -0.903989315f, -0.634393275f, 0.595699310f,
|
||||
0.923879504f, -0.146730468f, -0.995184720f, -0.336889863f, 0.831469595f,
|
||||
0.740951121f, -0.471396744f, -0.970031261f, 0.707106769f, 0.146730468f,
|
||||
-0.956940353f, -0.427555084f, 0.831469595f, 0.671558976f, -0.634393275f,
|
||||
-0.857728601f, 0.382683426f, 0.970031261f, -0.0980171412f, -0.998795450f,
|
||||
-0.195090324f, 0.941544056f, 0.471396744f, -0.803207517f, -0.707106769f,
|
||||
0.595699310f, 0.881921291f, -0.336889863f, -0.980785251f, 0.0490676761f,
|
||||
0.995184720f, 0.242980182f, -0.923879504f, -0.514102757f, 0.773010433f,
|
||||
0.740951121f, -0.555570245f, -0.903989315f, 0.290284663f, 0.989176512f,
|
||||
0.707106769f, 0.0490676761f, -0.995184720f, -0.146730468f, 0.980785251f,
|
||||
0.242980182f, -0.956940353f, -0.336889863f, 0.923879504f, 0.427555084f,
|
||||
-0.881921291f, -0.514102757f, 0.831469595f, 0.595699310f, -0.773010433f,
|
||||
-0.671558976f, 0.707106769f, 0.740951121f, -0.634393275f, -0.803207517f,
|
||||
0.555570245f, 0.857728601f, -0.471396744f, -0.903989315f, 0.382683426f,
|
||||
0.941544056f, -0.290284663f, -0.970031261f, 0.195090324f, 0.989176512f,
|
||||
-0.0980171412f, -0.998795450f, 0.707106769f, -0.0490676761f, -0.995184720f,
|
||||
0.146730468f, 0.980785251f, -0.242980182f, -0.956940353f, 0.336889863f,
|
||||
0.923879504f, -0.427555084f, -0.881921291f, 0.514102757f, 0.831469595f,
|
||||
-0.595699310f, -0.773010433f, 0.671558976f, 0.707106769f, -0.740951121f,
|
||||
-0.634393275f, 0.803207517f, 0.555570245f, -0.857728601f, -0.471396744f,
|
||||
0.903989315f, 0.382683426f, -0.941544056f, -0.290284663f, 0.970031261f,
|
||||
0.195090324f, -0.989176512f, -0.0980171412f, 0.998795450f, 0.707106769f,
|
||||
-0.146730468f, -0.956940353f, 0.427555084f, 0.831469595f, -0.671558976f,
|
||||
-0.634393275f, 0.857728601f, 0.382683426f, -0.970031261f, -0.0980171412f,
|
||||
0.998795450f, -0.195090324f, -0.941544056f, 0.471396744f, 0.803207517f,
|
||||
-0.707106769f, -0.595699310f, 0.881921291f, 0.336889863f, -0.980785251f,
|
||||
-0.0490676761f, 0.995184720f, -0.242980182f, -0.923879504f, 0.514102757f,
|
||||
0.773010433f, -0.740951121f, -0.555570245f, 0.903989315f, 0.290284663f,
|
||||
-0.989176512f, 0.707106769f, -0.242980182f, -0.881921291f, 0.671558976f,
|
||||
0.555570245f, -0.941544056f, -0.0980171412f, 0.989176512f, -0.382683426f,
|
||||
-0.803207517f, 0.773010433f, 0.427555084f, -0.980785251f, 0.0490676761f,
|
||||
0.956940353f, -0.514102757f, -0.707106769f, 0.857728601f, 0.290284663f,
|
||||
-0.998795450f, 0.195090324f, 0.903989315f, -0.634393275f, -0.595699310f,
|
||||
0.923879504f, 0.146730468f, -0.995184720f, 0.336889863f, 0.831469595f,
|
||||
-0.740951121f, -0.471396744f, 0.970031261f, 0.707106769f, -0.336889863f,
|
||||
-0.773010433f, 0.857728601f, 0.195090324f, -0.989176512f, 0.471396744f,
|
||||
0.671558976f, -0.923879504f, -0.0490676761f, 0.956940353f, -0.595699310f,
|
||||
-0.555570245f, 0.970031261f, -0.0980171412f, -0.903989315f, 0.707106769f,
|
||||
0.427555084f, -0.995184720f, 0.242980182f, 0.831469595f, -0.803207517f,
|
||||
-0.290284663f, 0.998795450f, -0.382683426f, -0.740951121f, 0.881921291f,
|
||||
0.146730468f, -0.980785251f, 0.514102757f, 0.634393275f, -0.941544056f,
|
||||
0.707106769f, -0.427555084f, -0.634393275f, 0.970031261f, -0.195090324f,
|
||||
-0.803207517f, 0.881921291f, 0.0490676761f, -0.923879504f, 0.740951121f,
|
||||
0.290284663f, -0.989176512f, 0.555570245f, 0.514102757f, -0.995184720f,
|
||||
0.336889863f, 0.707106769f, -0.941544056f, 0.0980171412f, 0.857728601f,
|
||||
-0.831469595f, -0.146730468f, 0.956940353f, -0.671558976f, -0.382683426f,
|
||||
0.998795450f, -0.471396744f, -0.595699310f, 0.980785251f, -0.242980182f,
|
||||
-0.773010433f, 0.903989315f, 0.707106769f, -0.514102757f, -0.471396744f,
|
||||
0.998795450f, -0.555570245f, -0.427555084f, 0.995184720f, -0.595699310f,
|
||||
-0.382683426f, 0.989176512f, -0.634393275f, -0.336889863f, 0.980785251f,
|
||||
-0.671558976f, -0.290284663f, 0.970031261f, -0.707106769f, -0.242980182f,
|
||||
0.956940353f, -0.740951121f, -0.195090324f, 0.941544056f, -0.773010433f,
|
||||
-0.146730468f, 0.923879504f, -0.803207517f, -0.0980171412f, 0.903989315f,
|
||||
-0.831469595f, -0.0490676761f, 0.881921291f, -0.857728601f, 0.707106769f,
|
||||
-0.595699310f, -0.290284663f, 0.941544056f, -0.831469595f, 0.0490676761f,
|
||||
0.773010433f, -0.970031261f, 0.382683426f, 0.514102757f, -0.995184720f,
|
||||
0.671558976f, 0.195090324f, -0.903989315f, 0.881921291f, -0.146730468f,
|
||||
-0.707106769f, 0.989176512f, -0.471396744f, -0.427555084f, 0.980785251f,
|
||||
-0.740951121f, -0.0980171412f, 0.857728601f, -0.923879504f, 0.242980182f,
|
||||
0.634393275f, -0.998795450f, 0.555570245f, 0.336889863f, -0.956940353f,
|
||||
0.803207517f, 0.707106769f, -0.671558976f, -0.0980171412f, 0.803207517f,
|
||||
-0.980785251f, 0.514102757f, 0.290284663f, -0.903989315f, 0.923879504f,
|
||||
-0.336889863f, -0.471396744f, 0.970031261f, -0.831469595f, 0.146730468f,
|
||||
0.634393275f, -0.998795450f, 0.707106769f, 0.0490676761f, -0.773010433f,
|
||||
0.989176512f, -0.555570245f, -0.242980182f, 0.881921291f, -0.941544056f,
|
||||
0.382683426f, 0.427555084f, -0.956940353f, 0.857728601f, -0.195090324f,
|
||||
-0.595699310f, 0.995184720f, -0.740951121f, 0.707106769f, -0.740951121f,
|
||||
0.0980171412f, 0.595699310f, -0.980785251f, 0.857728601f, -0.290284663f,
|
||||
-0.427555084f, 0.923879504f, -0.941544056f, 0.471396744f, 0.242980182f,
|
||||
-0.831469595f, 0.989176512f, -0.634393275f, -0.0490676761f, 0.707106769f,
|
||||
-0.998795450f, 0.773010433f, -0.146730468f, -0.555570245f, 0.970031261f,
|
||||
-0.881921291f, 0.336889863f, 0.382683426f, -0.903989315f, 0.956940353f,
|
||||
-0.514102757f, -0.195090324f, 0.803207517f, -0.995184720f, 0.671558976f,
|
||||
0.707106769f, -0.803207517f, 0.290284663f, 0.336889863f, -0.831469595f,
|
||||
0.998795450f, -0.773010433f, 0.242980182f, 0.382683426f, -0.857728601f,
|
||||
0.995184720f, -0.740951121f, 0.195090324f, 0.427555084f, -0.881921291f,
|
||||
0.989176512f, -0.707106769f, 0.146730468f, 0.471396744f, -0.903989315f,
|
||||
0.980785251f, -0.671558976f, 0.0980171412f, 0.514102757f, -0.923879504f,
|
||||
0.970031261f, -0.634393275f, 0.0490676761f, 0.555570245f, -0.941544056f,
|
||||
0.956940353f, -0.595699310f, 0.707106769f, -0.857728601f, 0.471396744f,
|
||||
0.0490676761f, -0.555570245f, 0.903989315f, -0.995184720f, 0.803207517f,
|
||||
-0.382683426f, -0.146730468f, 0.634393275f, -0.941544056f, 0.980785251f,
|
||||
-0.740951121f, 0.290284663f, 0.242980182f, -0.707106769f, 0.970031261f,
|
||||
-0.956940353f, 0.671558976f, -0.195090324f, -0.336889863f, 0.773010433f,
|
||||
-0.989176512f, 0.923879504f, -0.595699310f, 0.0980171412f, 0.427555084f,
|
||||
-0.831469595f, 0.998795450f, -0.881921291f, 0.514102757f, 0.707106769f,
|
||||
-0.903989315f, 0.634393275f, -0.242980182f, -0.195090324f, 0.595699310f,
|
||||
-0.881921291f, 0.998795450f, -0.923879504f, 0.671558976f, -0.290284663f,
|
||||
-0.146730468f, 0.555570245f, -0.857728601f, 0.995184720f, -0.941544056f,
|
||||
0.707106769f, -0.336889863f, -0.0980171412f, 0.514102757f, -0.831469595f,
|
||||
0.989176512f, -0.956940353f, 0.740951121f, -0.382683426f, -0.0490676761f,
|
||||
0.471396744f, -0.803207517f, 0.980785251f, -0.970031261f, 0.773010433f,
|
||||
-0.427555084f, 0.707106769f, -0.941544056f, 0.773010433f, -0.514102757f,
|
||||
0.195090324f, 0.146730468f, -0.471396744f, 0.740951121f, -0.923879504f,
|
||||
0.998795450f, -0.956940353f, 0.803207517f, -0.555570245f, 0.242980182f,
|
||||
0.0980171412f, -0.427555084f, 0.707106769f, -0.903989315f, 0.995184720f,
|
||||
-0.970031261f, 0.831469595f, -0.595699310f, 0.290284663f, 0.0490676761f,
|
||||
-0.382683426f, 0.671558976f, -0.881921291f, 0.989176512f, -0.980785251f,
|
||||
0.857728601f, -0.634393275f, 0.336889863f, 0.707106769f, -0.970031261f,
|
||||
0.881921291f, -0.740951121f, 0.555570245f, -0.336889863f, 0.0980171412f,
|
||||
0.146730468f, -0.382683426f, 0.595699310f, -0.773010433f, 0.903989315f,
|
||||
-0.980785251f, 0.998795450f, -0.956940353f, 0.857728601f, -0.707106769f,
|
||||
0.514102757f, -0.290284663f, 0.0490676761f, 0.195090324f, -0.427555084f,
|
||||
0.634393275f, -0.803207517f, 0.923879504f, -0.989176512f, 0.995184720f,
|
||||
-0.941544056f, 0.831469595f, -0.671558976f, 0.471396744f, -0.242980182f,
|
||||
0.707106769f, -0.989176512f, 0.956940353f, -0.903989315f, 0.831469595f,
|
||||
-0.740951121f, 0.634393275f, -0.514102757f, 0.382683426f, -0.242980182f,
|
||||
0.0980171412f, 0.0490676761f, -0.195090324f, 0.336889863f, -0.471396744f,
|
||||
0.595699310f, -0.707106769f, 0.803207517f, -0.881921291f, 0.941544056f,
|
||||
-0.980785251f, 0.998795450f, -0.995184720f, 0.970031261f, -0.923879504f,
|
||||
0.857728601f, -0.773010433f, 0.671558976f, -0.555570245f, 0.427555084f,
|
||||
-0.290284663f, 0.146730468f, 0.707106769f, -0.998795450f, 0.995184720f,
|
||||
-0.989176512f, 0.980785251f, -0.970031261f, 0.956940353f, -0.941544056f,
|
||||
0.923879504f, -0.903989315f, 0.881921291f, -0.857728601f, 0.831469595f,
|
||||
-0.803207517f, 0.773010433f, -0.740951121f, 0.707106769f, -0.671558976f,
|
||||
0.634393275f, -0.595699310f, 0.555570245f, -0.514102757f, 0.471396744f,
|
||||
-0.427555084f, 0.382683426f, -0.336889863f, 0.290284663f, -0.242980182f,
|
||||
0.195090324f, -0.146730468f, 0.0980171412f, -0.0490676761f, };
|
||||
388
src/modules/ZoomIt/ZoomIt/rnnoise/vec.h
Normal file
388
src/modules/ZoomIt/ZoomIt/rnnoise/vec.h
Normal file
@@ -0,0 +1,388 @@
|
||||
/* Copyright (c) 2018 Mozilla
|
||||
2008-2011 Octasic Inc.
|
||||
2012-2017 Jean-Marc Valin */
|
||||
/*
|
||||
Redistribution and use in source and binary forms, with or without
|
||||
modification, are permitted provided that the following conditions
|
||||
are met:
|
||||
|
||||
- Redistributions of source code must retain the above copyright
|
||||
notice, this list of conditions and the following disclaimer.
|
||||
|
||||
- Redistributions in binary form must reproduce the above copyright
|
||||
notice, this list of conditions and the following disclaimer in the
|
||||
documentation and/or other materials provided with the distribution.
|
||||
|
||||
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||||
``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
||||
LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
||||
A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR
|
||||
CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
|
||||
EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
|
||||
PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
|
||||
PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
|
||||
LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
|
||||
NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
||||
SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
*/
|
||||
|
||||
#ifndef VEC_H
|
||||
#define VEC_H
|
||||
|
||||
#include "opus_types.h"
|
||||
#include "common.h"
|
||||
#include <math.h>
|
||||
#include "arch.h"
|
||||
#include "x86/x86_arch_macros.h"
|
||||
|
||||
|
||||
#if defined(__AVX__) || defined(__SSE2__)
|
||||
#include "vec_avx.h"
|
||||
#elif (defined(__ARM_NEON__) || defined(__ARM_NEON)) && !defined(DISABLE_NEON)
|
||||
#include "vec_neon.h"
|
||||
#else
|
||||
|
||||
#define MAX_INPUTS (2048)
|
||||
|
||||
#define NO_OPTIMIZATIONS
|
||||
|
||||
static inline void sgemv16x1(float *out, const float *weights, int rows, int cols, int col_stride, const float *x)
|
||||
{
|
||||
int i, j;
|
||||
RNN_CLEAR(out, rows);
|
||||
for (i=0;i<rows;i+=16)
|
||||
{
|
||||
for (j=0;j<cols;j++)
|
||||
{
|
||||
const float * restrict w;
|
||||
float * restrict y;
|
||||
float xj;
|
||||
w = &weights[j*col_stride + i];
|
||||
xj = x[j];
|
||||
y = &out[i];
|
||||
y[0] += w[0]*xj;
|
||||
y[1] += w[1]*xj;
|
||||
y[2] += w[2]*xj;
|
||||
y[3] += w[3]*xj;
|
||||
y[4] += w[4]*xj;
|
||||
y[5] += w[5]*xj;
|
||||
y[6] += w[6]*xj;
|
||||
y[7] += w[7]*xj;
|
||||
y[8] += w[8]*xj;
|
||||
y[9] += w[9]*xj;
|
||||
y[10] += w[10]*xj;
|
||||
y[11] += w[11]*xj;
|
||||
y[12] += w[12]*xj;
|
||||
y[13] += w[13]*xj;
|
||||
y[14] += w[14]*xj;
|
||||
y[15] += w[15]*xj;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
static inline void sgemv8x1(float *out, const float *weights, int rows, int cols, int col_stride, const float *x)
|
||||
{
|
||||
int i, j;
|
||||
RNN_CLEAR(out, rows);
|
||||
for (i=0;i<rows;i+=8)
|
||||
{
|
||||
for (j=0;j<cols;j++)
|
||||
{
|
||||
const float * restrict w;
|
||||
float * restrict y;
|
||||
float xj;
|
||||
w = &weights[j*col_stride + i];
|
||||
xj = x[j];
|
||||
y = &out[i];
|
||||
y[0] += w[0]*xj;
|
||||
y[1] += w[1]*xj;
|
||||
y[2] += w[2]*xj;
|
||||
y[3] += w[3]*xj;
|
||||
y[4] += w[4]*xj;
|
||||
y[5] += w[5]*xj;
|
||||
y[6] += w[6]*xj;
|
||||
y[7] += w[7]*xj;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
static inline void sgemv(float *out, const float *weights, int rows, int cols, int col_stride, const float *x)
|
||||
{
|
||||
if ((rows&0xf) == 0) sgemv16x1(out, weights, rows, cols, col_stride, x);
|
||||
else if ((rows&0x7) == 0) sgemv8x1(out, weights, rows, cols, col_stride, x);
|
||||
else {
|
||||
int i, j;
|
||||
for (i=0;i<rows;i++)
|
||||
{
|
||||
out[i] = 0;
|
||||
for (j=0;j<cols;j++) out[i] += weights[j*col_stride + i]*x[j];
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
static inline void sparse_sgemv8x4(float *out, const float *w, const int *idx, int rows, const float *x)
|
||||
{
|
||||
int i, j;
|
||||
RNN_CLEAR(out, rows);
|
||||
for (i=0;i<rows;i+=8)
|
||||
{
|
||||
int cols;
|
||||
cols = *idx++;
|
||||
for (j=0;j<cols;j++)
|
||||
{
|
||||
int pos;
|
||||
float * restrict y;
|
||||
float xj0, xj1, xj2, xj3;
|
||||
pos = (*idx++);
|
||||
xj0 = x[pos+0];
|
||||
xj1 = x[pos+1];
|
||||
xj2 = x[pos+2];
|
||||
xj3 = x[pos+3];
|
||||
y = &out[i];
|
||||
y[0] += w[0]*xj0;
|
||||
y[1] += w[1]*xj0;
|
||||
y[2] += w[2]*xj0;
|
||||
y[3] += w[3]*xj0;
|
||||
y[4] += w[4]*xj0;
|
||||
y[5] += w[5]*xj0;
|
||||
y[6] += w[6]*xj0;
|
||||
y[7] += w[7]*xj0;
|
||||
|
||||
y[0] += w[8]*xj1;
|
||||
y[1] += w[9]*xj1;
|
||||
y[2] += w[10]*xj1;
|
||||
y[3] += w[11]*xj1;
|
||||
y[4] += w[12]*xj1;
|
||||
y[5] += w[13]*xj1;
|
||||
y[6] += w[14]*xj1;
|
||||
y[7] += w[15]*xj1;
|
||||
|
||||
y[0] += w[16]*xj2;
|
||||
y[1] += w[17]*xj2;
|
||||
y[2] += w[18]*xj2;
|
||||
y[3] += w[19]*xj2;
|
||||
y[4] += w[20]*xj2;
|
||||
y[5] += w[21]*xj2;
|
||||
y[6] += w[22]*xj2;
|
||||
y[7] += w[23]*xj2;
|
||||
|
||||
y[0] += w[24]*xj3;
|
||||
y[1] += w[25]*xj3;
|
||||
y[2] += w[26]*xj3;
|
||||
y[3] += w[27]*xj3;
|
||||
y[4] += w[28]*xj3;
|
||||
y[5] += w[29]*xj3;
|
||||
y[6] += w[30]*xj3;
|
||||
y[7] += w[31]*xj3;
|
||||
w += 32;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#ifdef USE_SU_BIAS
|
||||
static inline void sparse_cgemv8x4(float *out, const opus_int8 *w, const int *idx, const float *scale, int rows, int cols, const float *_x)
|
||||
{
|
||||
int i, j;
|
||||
unsigned char x[MAX_INPUTS];
|
||||
for (i=0;i<rows;i++) out[i] = 0;
|
||||
for (i=0;i<cols;i++) x[i] = 127+floor(.5+127*_x[i]);
|
||||
for (i=0;i<rows;i+=8)
|
||||
{
|
||||
int colblocks;
|
||||
colblocks = *idx++;
|
||||
for (j=0;j<colblocks;j++)
|
||||
{
|
||||
int pos;
|
||||
float * restrict y;
|
||||
int xj0, xj1, xj2, xj3;
|
||||
pos = (*idx++);
|
||||
xj0 = x[pos+0];
|
||||
xj1 = x[pos+1];
|
||||
xj2 = x[pos+2];
|
||||
xj3 = x[pos+3];
|
||||
y = &out[i];
|
||||
y[0] += (w[0]*xj0+w[1]*xj1+w[2]*xj2+w[3]*xj3);
|
||||
y[1] += (w[4]*xj0+w[5]*xj1+w[6]*xj2+w[7]*xj3);
|
||||
y[2] += (w[8]*xj0+w[9]*xj1+w[10]*xj2+w[11]*xj3);
|
||||
y[3] += (w[12]*xj0+w[13]*xj1+w[14]*xj2+w[15]*xj3);
|
||||
y[4] += (w[16]*xj0+w[17]*xj1+w[18]*xj2+w[19]*xj3);
|
||||
y[5] += (w[20]*xj0+w[21]*xj1+w[22]*xj2+w[23]*xj3);
|
||||
y[6] += (w[24]*xj0+w[25]*xj1+w[26]*xj2+w[27]*xj3);
|
||||
y[7] += (w[28]*xj0+w[29]*xj1+w[30]*xj2+w[31]*xj3);
|
||||
w += 32;
|
||||
}
|
||||
}
|
||||
for (i=0;i<rows;i++) out[i] *= scale[i];
|
||||
}
|
||||
static inline void cgemv8x4(float *out, const opus_int8 *w, const float *scale, int rows, int cols, const float *_x)
|
||||
{
|
||||
int i, j;
|
||||
unsigned char x[MAX_INPUTS];
|
||||
for (i=0;i<rows;i++) out[i] = 0;
|
||||
for (i=0;i<cols;i++) x[i] = 127+(int)floor(.5+127*_x[i]);
|
||||
for (i=0;i<rows;i+=8)
|
||||
{
|
||||
for (j=0;j<cols;j+=4)
|
||||
{
|
||||
float *y;
|
||||
float xj0, xj1, xj2, xj3;
|
||||
xj0 = x[j+0];
|
||||
xj1 = x[j+1];
|
||||
xj2 = x[j+2];
|
||||
xj3 = x[j+3];
|
||||
y = &out[i];
|
||||
y[0] += (w[0]*xj0+w[1]*xj1+w[2]*xj2+w[3]*xj3);
|
||||
y[1] += (w[4]*xj0+w[5]*xj1+w[6]*xj2+w[7]*xj3);
|
||||
y[2] += (w[8]*xj0+w[9]*xj1+w[10]*xj2+w[11]*xj3);
|
||||
y[3] += (w[12]*xj0+w[13]*xj1+w[14]*xj2+w[15]*xj3);
|
||||
y[4] += (w[16]*xj0+w[17]*xj1+w[18]*xj2+w[19]*xj3);
|
||||
y[5] += (w[20]*xj0+w[21]*xj1+w[22]*xj2+w[23]*xj3);
|
||||
y[6] += (w[24]*xj0+w[25]*xj1+w[26]*xj2+w[27]*xj3);
|
||||
y[7] += (w[28]*xj0+w[29]*xj1+w[30]*xj2+w[31]*xj3);
|
||||
w += 32;
|
||||
}
|
||||
}
|
||||
for (i=0;i<rows;i++) out[i] *= scale[i];
|
||||
}
|
||||
#else
|
||||
static inline void sparse_cgemv8x4(float *out, const opus_int8 *w, const int *idx, const float *scale, int rows, int cols, const float *_x)
|
||||
{
|
||||
int i, j;
|
||||
opus_int8 x[MAX_INPUTS];
|
||||
for (i=0;i<rows;i++) out[i] = 0;
|
||||
for (i=0;i<cols;i++) x[i] = (int)floor(.5+127*_x[i]);
|
||||
for (i=0;i<rows;i+=8)
|
||||
{
|
||||
int colblocks;
|
||||
colblocks = *idx++;
|
||||
for (j=0;j<colblocks;j++)
|
||||
{
|
||||
int pos;
|
||||
float * restrict y;
|
||||
int xj0, xj1, xj2, xj3;
|
||||
pos = (*idx++);
|
||||
xj0 = x[pos+0];
|
||||
xj1 = x[pos+1];
|
||||
xj2 = x[pos+2];
|
||||
xj3 = x[pos+3];
|
||||
y = &out[i];
|
||||
y[0] += (w[0]*xj0+w[1]*xj1+w[2]*xj2+w[3]*xj3);
|
||||
y[1] += (w[4]*xj0+w[5]*xj1+w[6]*xj2+w[7]*xj3);
|
||||
y[2] += (w[8]*xj0+w[9]*xj1+w[10]*xj2+w[11]*xj3);
|
||||
y[3] += (w[12]*xj0+w[13]*xj1+w[14]*xj2+w[15]*xj3);
|
||||
y[4] += (w[16]*xj0+w[17]*xj1+w[18]*xj2+w[19]*xj3);
|
||||
y[5] += (w[20]*xj0+w[21]*xj1+w[22]*xj2+w[23]*xj3);
|
||||
y[6] += (w[24]*xj0+w[25]*xj1+w[26]*xj2+w[27]*xj3);
|
||||
y[7] += (w[28]*xj0+w[29]*xj1+w[30]*xj2+w[31]*xj3);
|
||||
w += 32;
|
||||
}
|
||||
}
|
||||
for (i=0;i<rows;i++) out[i] *= scale[i];
|
||||
}
|
||||
static inline void cgemv8x4(float *out, const opus_int8 *w, const float *scale, int rows, int cols, const float *_x)
|
||||
{
|
||||
int i, j;
|
||||
opus_int8 x[MAX_INPUTS];
|
||||
for (i=0;i<rows;i++) out[i] = 0;
|
||||
for (i=0;i<cols;i++) x[i] = (int)floor(.5+127*_x[i]);
|
||||
for (i=0;i<rows;i+=8)
|
||||
{
|
||||
for (j=0;j<cols;j+=4)
|
||||
{
|
||||
float *y;
|
||||
float xj0, xj1, xj2, xj3;
|
||||
xj0 = x[j+0];
|
||||
xj1 = x[j+1];
|
||||
xj2 = x[j+2];
|
||||
xj3 = x[j+3];
|
||||
y = &out[i];
|
||||
y[0] += (w[0]*xj0+w[1]*xj1+w[2]*xj2+w[3]*xj3);
|
||||
y[1] += (w[4]*xj0+w[5]*xj1+w[6]*xj2+w[7]*xj3);
|
||||
y[2] += (w[8]*xj0+w[9]*xj1+w[10]*xj2+w[11]*xj3);
|
||||
y[3] += (w[12]*xj0+w[13]*xj1+w[14]*xj2+w[15]*xj3);
|
||||
y[4] += (w[16]*xj0+w[17]*xj1+w[18]*xj2+w[19]*xj3);
|
||||
y[5] += (w[20]*xj0+w[21]*xj1+w[22]*xj2+w[23]*xj3);
|
||||
y[6] += (w[24]*xj0+w[25]*xj1+w[26]*xj2+w[27]*xj3);
|
||||
y[7] += (w[28]*xj0+w[29]*xj1+w[30]*xj2+w[31]*xj3);
|
||||
w += 32;
|
||||
}
|
||||
}
|
||||
for (i=0;i<rows;i++) out[i] *= scale[i];
|
||||
}
|
||||
#endif
|
||||
|
||||
/* No AVX2/FMA support */
|
||||
#ifndef LPCNET_TEST
|
||||
static inline float lpcnet_exp2(float x)
|
||||
{
|
||||
int integer;
|
||||
float frac;
|
||||
union {
|
||||
float f;
|
||||
opus_uint32 i;
|
||||
} res;
|
||||
integer = floor(x);
|
||||
if (integer < -50)
|
||||
return 0;
|
||||
frac = x-integer;
|
||||
/* K0 = 1, K1 = log(2), K2 = 3-4*log(2), K3 = 3*log(2) - 2 */
|
||||
res.f = 0.99992522f + frac * (0.69583354f
|
||||
+ frac * (0.22606716f + 0.078024523f*frac));
|
||||
res.i = (res.i + (integer<<23)) & 0x7fffffff;
|
||||
return res.f;
|
||||
}
|
||||
#define lpcnet_exp(x) lpcnet_exp2((x)*1.44269504f)
|
||||
|
||||
#define fmadd(a, b, c) ((a)*(b)+(c))
|
||||
static OPUS_INLINE float tanh_approx(float x)
|
||||
{
|
||||
const float N0 = 952.52801514f;
|
||||
const float N1 = 96.39235687f;
|
||||
const float N2 = 0.60863042f;
|
||||
const float D0 = 952.72399902f;
|
||||
const float D1 = 413.36801147f;
|
||||
const float D2 = 11.88600922f;
|
||||
float X2, num, den;
|
||||
X2 = x*x;
|
||||
num = fmadd(fmadd(N2, X2, N1), X2, N0);
|
||||
den = fmadd(fmadd(D2, X2, D1), X2, D0);
|
||||
num = num*x/den;
|
||||
return MAX32(-1.f, MIN32(1.f, num));
|
||||
}
|
||||
|
||||
static inline float sigmoid_approx(float x)
|
||||
{
|
||||
return .5f + .5f*tanh_approx(.5f*x);
|
||||
}
|
||||
|
||||
static inline void softmax(float *y, const float *x, int N)
|
||||
{
|
||||
int i;
|
||||
for (i=0;i<N;i++)
|
||||
y[i] = lpcnet_exp(x[i]);
|
||||
}
|
||||
|
||||
static inline void vec_tanh(float *y, const float *x, int N)
|
||||
{
|
||||
int i;
|
||||
for (i=0;i<N;i++)
|
||||
{
|
||||
y[i] = tanh_approx(x[i]);
|
||||
}
|
||||
}
|
||||
|
||||
static inline void vec_sigmoid(float *y, const float *x, int N)
|
||||
{
|
||||
int i;
|
||||
for (i=0;i<N;i++)
|
||||
{
|
||||
y[i] = sigmoid_approx(x[i]);
|
||||
}
|
||||
}
|
||||
#endif
|
||||
|
||||
#define SCALE (128.f*127.f)
|
||||
#define SCALE_1 (1.f/128.f/127.f)
|
||||
|
||||
#endif /*no optimizations*/
|
||||
#endif /*VEC_H*/
|
||||
85
src/modules/ZoomIt/ZoomIt/rnnoise/x86/dnn_x86.h
Normal file
85
src/modules/ZoomIt/ZoomIt/rnnoise/x86/dnn_x86.h
Normal file
@@ -0,0 +1,85 @@
|
||||
/* Copyright (c) 2011-2019 Mozilla
|
||||
2023 Amazon */
|
||||
/*
|
||||
Redistribution and use in source and binary forms, with or without
|
||||
modification, are permitted provided that the following conditions
|
||||
are met:
|
||||
|
||||
- Redistributions of source code must retain the above copyright
|
||||
notice, this list of conditions and the following disclaimer.
|
||||
|
||||
- Redistributions in binary form must reproduce the above copyright
|
||||
notice, this list of conditions and the following disclaimer in the
|
||||
documentation and/or other materials provided with the distribution.
|
||||
|
||||
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||||
``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
||||
LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
||||
A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR
|
||||
CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
|
||||
EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
|
||||
PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
|
||||
PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
|
||||
LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
|
||||
NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
||||
SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
*/
|
||||
|
||||
#ifndef DNN_X86_H
|
||||
#define DNN_X86_H
|
||||
|
||||
#include "cpu_support.h"
|
||||
#include "opus_types.h"
|
||||
|
||||
void compute_linear_sse4_1(const LinearLayer *linear, float *out, const float *in);
|
||||
void compute_activation_sse4_1(float *output, const float *input, int N, int activation);
|
||||
void compute_conv2d_sse4_1(const Conv2dLayer *conv, float *out, float *mem, const float *in, int height, int hstride, int activation);
|
||||
|
||||
void compute_linear_avx2(const LinearLayer *linear, float *out, const float *in);
|
||||
void compute_activation_avx2(float *output, const float *input, int N, int activation);
|
||||
void compute_conv2d_avx2(const Conv2dLayer *conv, float *out, float *mem, const float *in, int height, int hstride, int activation);
|
||||
|
||||
|
||||
|
||||
#ifdef RNN_ENABLE_X86_RTCD
|
||||
|
||||
extern void (*const RNN_COMPUTE_LINEAR_IMPL[OPUS_ARCHMASK + 1])(
|
||||
const LinearLayer *linear,
|
||||
float *out,
|
||||
const float *in
|
||||
);
|
||||
#define OVERRIDE_COMPUTE_LINEAR
|
||||
#define compute_linear(linear, out, in, arch) \
|
||||
((*RNN_COMPUTE_LINEAR_IMPL[(arch) & OPUS_ARCHMASK])(linear, out, in))
|
||||
|
||||
|
||||
extern void (*const RNN_COMPUTE_ACTIVATION_IMPL[OPUS_ARCHMASK + 1])(
|
||||
float *output,
|
||||
const float *input,
|
||||
int N,
|
||||
int activation
|
||||
);
|
||||
#define OVERRIDE_COMPUTE_ACTIVATION
|
||||
#define compute_activation(output, input, N, activation, arch) \
|
||||
((*RNN_COMPUTE_ACTIVATION_IMPL[(arch) & OPUS_ARCHMASK])(output, input, N, activation))
|
||||
|
||||
|
||||
extern void (*const RNN_COMPUTE_CONV2D_IMPL[OPUS_ARCHMASK + 1])(
|
||||
const Conv2dLayer *conv,
|
||||
float *out,
|
||||
float *mem,
|
||||
const float *in,
|
||||
int height,
|
||||
int hstride,
|
||||
int activation
|
||||
);
|
||||
#define OVERRIDE_COMPUTE_CONV2D
|
||||
#define compute_conv2d(conv, out, mem, in, height, hstride, activation, arch) \
|
||||
((*RNN_COMPUTE_CONV2D_IMPL[(arch) & OPUS_ARCHMASK])(conv, out, mem, in, height, hstride, activation))
|
||||
|
||||
|
||||
#endif
|
||||
|
||||
|
||||
|
||||
#endif /* DNN_X86_H */
|
||||
40
src/modules/ZoomIt/ZoomIt/rnnoise/x86/nnet_avx2.c
Normal file
40
src/modules/ZoomIt/ZoomIt/rnnoise/x86/nnet_avx2.c
Normal file
@@ -0,0 +1,40 @@
|
||||
/* Copyright (c) 2018-2019 Mozilla
|
||||
2023 Amazon */
|
||||
/*
|
||||
Redistribution and use in source and binary forms, with or without
|
||||
modification, are permitted provided that the following conditions
|
||||
are met:
|
||||
|
||||
- Redistributions of source code must retain the above copyright
|
||||
notice, this list of conditions and the following disclaimer.
|
||||
|
||||
- Redistributions in binary form must reproduce the above copyright
|
||||
notice, this list of conditions and the following disclaimer in the
|
||||
documentation and/or other materials provided with the distribution.
|
||||
|
||||
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||||
``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
||||
LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
||||
A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR
|
||||
CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
|
||||
EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
|
||||
PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
|
||||
PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
|
||||
LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
|
||||
NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
||||
SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
*/
|
||||
|
||||
#ifdef HAVE_CONFIG_H
|
||||
#include "config.h"
|
||||
#endif
|
||||
|
||||
#include "x86/x86_arch_macros.h"
|
||||
|
||||
#ifndef __AVX2__
|
||||
#error nnet_avx2.c is being compiled without AVX2 enabled
|
||||
#endif
|
||||
|
||||
#define RTCD_ARCH avx2
|
||||
|
||||
#include "nnet_arch.h"
|
||||
40
src/modules/ZoomIt/ZoomIt/rnnoise/x86/nnet_sse4_1.c
Normal file
40
src/modules/ZoomIt/ZoomIt/rnnoise/x86/nnet_sse4_1.c
Normal file
@@ -0,0 +1,40 @@
|
||||
/* Copyright (c) 2018-2019 Mozilla
|
||||
2023 Amazon */
|
||||
/*
|
||||
Redistribution and use in source and binary forms, with or without
|
||||
modification, are permitted provided that the following conditions
|
||||
are met:
|
||||
|
||||
- Redistributions of source code must retain the above copyright
|
||||
notice, this list of conditions and the following disclaimer.
|
||||
|
||||
- Redistributions in binary form must reproduce the above copyright
|
||||
notice, this list of conditions and the following disclaimer in the
|
||||
documentation and/or other materials provided with the distribution.
|
||||
|
||||
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||||
``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
||||
LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
||||
A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR
|
||||
CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
|
||||
EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
|
||||
PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
|
||||
PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
|
||||
LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
|
||||
NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
||||
SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
*/
|
||||
|
||||
#ifdef HAVE_CONFIG_H
|
||||
#include "config.h"
|
||||
#endif
|
||||
|
||||
#include "x86/x86_arch_macros.h"
|
||||
|
||||
#ifndef __SSE4_1__
|
||||
#error nnet_sse4_1.c is being compiled without SSE4.1 enabled
|
||||
#endif
|
||||
|
||||
#define RTCD_ARCH sse4_1
|
||||
|
||||
#include "nnet_arch.h"
|
||||
47
src/modules/ZoomIt/ZoomIt/rnnoise/x86/x86_arch_macros.h
Normal file
47
src/modules/ZoomIt/ZoomIt/rnnoise/x86/x86_arch_macros.h
Normal file
@@ -0,0 +1,47 @@
|
||||
/* Copyright (c) 2023 Amazon */
|
||||
/*
|
||||
Redistribution and use in source and binary forms, with or without
|
||||
modification, are permitted provided that the following conditions
|
||||
are met:
|
||||
|
||||
- Redistributions of source code must retain the above copyright
|
||||
notice, this list of conditions and the following disclaimer.
|
||||
|
||||
- Redistributions in binary form must reproduce the above copyright
|
||||
notice, this list of conditions and the following disclaimer in the
|
||||
documentation and/or other materials provided with the distribution.
|
||||
|
||||
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||||
``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
||||
LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
||||
A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR
|
||||
CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
|
||||
EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
|
||||
PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
|
||||
PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
|
||||
LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
|
||||
NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
||||
SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
*/
|
||||
|
||||
#ifdef _MSC_VER
|
||||
|
||||
# ifdef OPUS_X86_MAY_HAVE_SSE
|
||||
# ifndef __SSE__
|
||||
# define __SSE__
|
||||
# endif
|
||||
# endif
|
||||
|
||||
# ifdef OPUS_X86_MAY_HAVE_SSE2
|
||||
# ifndef __SSE2__
|
||||
# define __SSE2__
|
||||
# endif
|
||||
# endif
|
||||
|
||||
# ifdef OPUS_X86_MAY_HAVE_SSE4_1
|
||||
# ifndef __SSE4_1__
|
||||
# define __SSE4_1__
|
||||
# endif
|
||||
# endif
|
||||
|
||||
#endif
|
||||
74
src/modules/ZoomIt/ZoomIt/rnnoise/x86/x86_dnn_map.c
Normal file
74
src/modules/ZoomIt/ZoomIt/rnnoise/x86/x86_dnn_map.c
Normal file
@@ -0,0 +1,74 @@
|
||||
/* Copyright (c) 2018-2019 Mozilla
|
||||
2023 Amazon */
|
||||
/*
|
||||
Redistribution and use in source and binary forms, with or without
|
||||
modification, are permitted provided that the following conditions
|
||||
are met:
|
||||
|
||||
- Redistributions of source code must retain the above copyright
|
||||
notice, this list of conditions and the following disclaimer.
|
||||
|
||||
- Redistributions in binary form must reproduce the above copyright
|
||||
notice, this list of conditions and the following disclaimer in the
|
||||
documentation and/or other materials provided with the distribution.
|
||||
|
||||
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||||
``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
||||
LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
||||
A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR
|
||||
CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
|
||||
EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
|
||||
PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
|
||||
PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
|
||||
LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
|
||||
NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
||||
SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
*/
|
||||
|
||||
#ifdef HAVE_CONFIG_H
|
||||
#include "config.h"
|
||||
#endif
|
||||
|
||||
#include "x86/x86cpu.h"
|
||||
#include "nnet.h"
|
||||
|
||||
#ifdef RNN_ENABLE_X86_RTCD
|
||||
|
||||
|
||||
void (*const RNN_COMPUTE_LINEAR_IMPL[OPUS_ARCHMASK + 1])(
|
||||
const LinearLayer *linear,
|
||||
float *out,
|
||||
const float *in
|
||||
) = {
|
||||
compute_linear_c, /* non-sse */
|
||||
MAY_HAVE_SSE4_1(compute_linear), /* sse4.1 */
|
||||
MAY_HAVE_AVX2(compute_linear) /* avx */
|
||||
};
|
||||
|
||||
void (*const RNN_COMPUTE_ACTIVATION_IMPL[OPUS_ARCHMASK + 1])(
|
||||
float *output,
|
||||
const float *input,
|
||||
int N,
|
||||
int activation
|
||||
) = {
|
||||
compute_activation_c, /* non-sse */
|
||||
MAY_HAVE_SSE4_1(compute_activation), /* sse4.1 */
|
||||
MAY_HAVE_AVX2(compute_activation) /* avx */
|
||||
};
|
||||
|
||||
void (*const RNN_COMPUTE_CONV2D_IMPL[OPUS_ARCHMASK + 1])(
|
||||
const Conv2dLayer *conv,
|
||||
float *out,
|
||||
float *mem,
|
||||
const float *in,
|
||||
int height,
|
||||
int hstride,
|
||||
int activation
|
||||
) = {
|
||||
compute_conv2d_c, /* non-sse */
|
||||
MAY_HAVE_SSE4_1(compute_conv2d), /* sse4.1 */
|
||||
MAY_HAVE_AVX2(compute_conv2d) /* avx */
|
||||
};
|
||||
|
||||
|
||||
#endif
|
||||
166
src/modules/ZoomIt/ZoomIt/rnnoise/x86/x86cpu.c
Normal file
166
src/modules/ZoomIt/ZoomIt/rnnoise/x86/x86cpu.c
Normal file
@@ -0,0 +1,166 @@
|
||||
/* Copyright (c) 2014, Cisco Systems, INC
|
||||
Written by XiangMingZhu WeiZhou MinPeng YanWang
|
||||
|
||||
Redistribution and use in source and binary forms, with or without
|
||||
modification, are permitted provided that the following conditions
|
||||
are met:
|
||||
|
||||
- Redistributions of source code must retain the above copyright
|
||||
notice, this list of conditions and the following disclaimer.
|
||||
|
||||
- Redistributions in binary form must reproduce the above copyright
|
||||
notice, this list of conditions and the following disclaimer in the
|
||||
documentation and/or other materials provided with the distribution.
|
||||
|
||||
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||||
``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
||||
LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
||||
A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER
|
||||
OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
|
||||
EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
|
||||
PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
|
||||
PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
|
||||
LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
|
||||
NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
||||
SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
*/
|
||||
|
||||
#ifdef HAVE_CONFIG_H
|
||||
#include "config.h"
|
||||
#endif
|
||||
|
||||
#include "cpu_support.h"
|
||||
#include "pitch.h"
|
||||
#include "x86cpu.h"
|
||||
|
||||
#ifdef RNN_ENABLE_X86_RTCD
|
||||
|
||||
#if defined(_MSC_VER)
|
||||
|
||||
#include <intrin.h>
|
||||
static _inline void cpuid(unsigned int CPUInfo[4], unsigned int InfoType)
|
||||
{
|
||||
__cpuid((int*)CPUInfo, InfoType);
|
||||
}
|
||||
|
||||
#else
|
||||
|
||||
#if defined(CPU_INFO_BY_C)
|
||||
#include <cpuid.h>
|
||||
#endif
|
||||
|
||||
static void cpuid(unsigned int CPUInfo[4], unsigned int InfoType)
|
||||
{
|
||||
#if defined(CPU_INFO_BY_ASM)
|
||||
#if defined(__i386__) && defined(__PIC__)
|
||||
/* %ebx is PIC register in 32-bit, so mustn't clobber it. */
|
||||
__asm__ __volatile__ (
|
||||
"xchg %%ebx, %1\n"
|
||||
"cpuid\n"
|
||||
"xchg %%ebx, %1\n":
|
||||
"=a" (CPUInfo[0]),
|
||||
"=r" (CPUInfo[1]),
|
||||
"=c" (CPUInfo[2]),
|
||||
"=d" (CPUInfo[3]) :
|
||||
/* We clear ECX to avoid a valgrind false-positive prior to v3.17.0. */
|
||||
"0" (InfoType), "2" (0)
|
||||
);
|
||||
#else
|
||||
__asm__ __volatile__ (
|
||||
"cpuid":
|
||||
"=a" (CPUInfo[0]),
|
||||
"=b" (CPUInfo[1]),
|
||||
"=c" (CPUInfo[2]),
|
||||
"=d" (CPUInfo[3]) :
|
||||
/* We clear ECX to avoid a valgrind false-positive prior to v3.17.0. */
|
||||
"0" (InfoType), "2" (0)
|
||||
);
|
||||
#endif
|
||||
#elif defined(CPU_INFO_BY_C)
|
||||
/* We use __get_cpuid_count to clear ECX to avoid a valgrind false-positive
|
||||
prior to v3.17.0.*/
|
||||
if (!__get_cpuid_count(InfoType, 0, &(CPUInfo[0]), &(CPUInfo[1]), &(CPUInfo[2]), &(CPUInfo[3]))) {
|
||||
/* Our function cannot fail, but __get_cpuid{_count} can.
|
||||
Returning all zeroes will effectively disable all SIMD, which is
|
||||
what we want on CPUs that don't support CPUID. */
|
||||
CPUInfo[3] = CPUInfo[2] = CPUInfo[1] = CPUInfo[0] = 0;
|
||||
}
|
||||
#else
|
||||
# error "Configured to use x86 RTCD, but no CPU detection method available. " \
|
||||
"Reconfigure with --disable-rtcd (or send patches)."
|
||||
#endif
|
||||
}
|
||||
|
||||
#endif
|
||||
|
||||
typedef struct CPU_Feature{
|
||||
/* SIMD: 128-bit */
|
||||
int HW_SSE;
|
||||
int HW_SSE2;
|
||||
int HW_SSE41;
|
||||
/* SIMD: 256-bit */
|
||||
int HW_AVX2;
|
||||
} CPU_Feature;
|
||||
|
||||
static void rnn_cpu_feature_check(CPU_Feature *cpu_feature)
|
||||
{
|
||||
unsigned int info[4];
|
||||
unsigned int nIds = 0;
|
||||
|
||||
cpuid(info, 0);
|
||||
nIds = info[0];
|
||||
|
||||
if (nIds >= 1){
|
||||
cpuid(info, 1);
|
||||
cpu_feature->HW_SSE = (info[3] & (1 << 25)) != 0;
|
||||
cpu_feature->HW_SSE2 = (info[3] & (1 << 26)) != 0;
|
||||
cpu_feature->HW_SSE41 = (info[2] & (1 << 19)) != 0;
|
||||
cpu_feature->HW_AVX2 = (info[2] & (1 << 28)) != 0 && (info[2] & (1 << 12)) != 0;
|
||||
if (cpu_feature->HW_AVX2 && nIds >= 7) {
|
||||
cpuid(info, 7);
|
||||
cpu_feature->HW_AVX2 = cpu_feature->HW_AVX2 && (info[1] & (1 << 5)) != 0;
|
||||
} else {
|
||||
cpu_feature->HW_AVX2 = 0;
|
||||
}
|
||||
}
|
||||
else {
|
||||
cpu_feature->HW_SSE = 0;
|
||||
cpu_feature->HW_SSE2 = 0;
|
||||
cpu_feature->HW_SSE41 = 0;
|
||||
cpu_feature->HW_AVX2 = 0;
|
||||
}
|
||||
}
|
||||
|
||||
static int rnn_select_arch_impl(void)
|
||||
{
|
||||
CPU_Feature cpu_feature;
|
||||
int arch;
|
||||
|
||||
rnn_cpu_feature_check(&cpu_feature);
|
||||
|
||||
arch = 0;
|
||||
if (!cpu_feature.HW_SSE41)
|
||||
{
|
||||
return arch;
|
||||
}
|
||||
arch++;
|
||||
|
||||
if (!cpu_feature.HW_AVX2)
|
||||
{
|
||||
return arch;
|
||||
}
|
||||
arch++;
|
||||
|
||||
return arch;
|
||||
}
|
||||
|
||||
int rnn_select_arch(void) {
|
||||
int arch = rnn_select_arch_impl();
|
||||
#ifdef FUZZING
|
||||
/* Randomly downgrade the architecture. */
|
||||
arch = rand()%(arch+1);
|
||||
#endif
|
||||
return arch;
|
||||
}
|
||||
|
||||
#endif
|
||||
88
src/modules/ZoomIt/ZoomIt/rnnoise/x86/x86cpu.h
Normal file
88
src/modules/ZoomIt/ZoomIt/rnnoise/x86/x86cpu.h
Normal file
@@ -0,0 +1,88 @@
|
||||
/* Copyright (c) 2014, Cisco Systems, INC
|
||||
Written by XiangMingZhu WeiZhou MinPeng YanWang
|
||||
|
||||
Redistribution and use in source and binary forms, with or without
|
||||
modification, are permitted provided that the following conditions
|
||||
are met:
|
||||
|
||||
- Redistributions of source code must retain the above copyright
|
||||
notice, this list of conditions and the following disclaimer.
|
||||
|
||||
- Redistributions in binary form must reproduce the above copyright
|
||||
notice, this list of conditions and the following disclaimer in the
|
||||
documentation and/or other materials provided with the distribution.
|
||||
|
||||
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||||
``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
||||
LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
||||
A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER
|
||||
OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
|
||||
EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
|
||||
PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
|
||||
PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
|
||||
LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
|
||||
NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
||||
SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
*/
|
||||
|
||||
#if !defined(X86CPU_H)
|
||||
# define X86CPU_H
|
||||
|
||||
# define MAY_HAVE_SSE4_1(name) name ## _sse4_1
|
||||
|
||||
# define MAY_HAVE_AVX2(name) name ## _avx2
|
||||
|
||||
# ifdef RNN_ENABLE_X86_RTCD
|
||||
int opus_select_arch(void);
|
||||
# endif
|
||||
|
||||
# if defined(__SSE2__)
|
||||
# include "common.h"
|
||||
|
||||
/*MOVD should not impose any alignment restrictions, but the C standard does,
|
||||
and UBSan will report errors if we actually make unaligned accesses.
|
||||
Use this to work around those restrictions (which should hopefully all get
|
||||
optimized to a single MOVD instruction).
|
||||
GCC implemented _mm_loadu_si32() since GCC 11; HOWEVER, there is a bug!
|
||||
https://gcc.gnu.org/bugzilla/show_bug.cgi?id=99754
|
||||
LLVM implemented _mm_loadu_si32() since Clang 8.0, however the
|
||||
__clang_major__ version number macro is unreliable, as vendors
|
||||
(specifically, Apple) will use different numbering schemes than upstream.
|
||||
Clang's advice is "use feature detection", but they do not provide feature
|
||||
detection support for specific SIMD functions.
|
||||
We follow the approach from the SIMDe project and instead detect unrelated
|
||||
features that should be available in the version we want (see
|
||||
<https://github.com/simd-everywhere/simde/blob/master/simde/simde-detect-clang.h>).*/
|
||||
# if defined(__clang__)
|
||||
# if __has_warning("-Wextra-semi-stmt") || \
|
||||
__has_builtin(__builtin_rotateleft32)
|
||||
# define OPUS_CLANG_8 (1)
|
||||
# endif
|
||||
# endif
|
||||
# if !defined(_MSC_VER) && !OPUS_GNUC_PREREQ(11,3) && !defined(OPUS_CLANG_8)
|
||||
# include <string.h>
|
||||
# include <emmintrin.h>
|
||||
|
||||
# ifdef _mm_loadu_si32
|
||||
# undef _mm_loadu_si32
|
||||
# endif
|
||||
# define _mm_loadu_si32 WORKAROUND_mm_loadu_si32
|
||||
static inline __m128i WORKAROUND_mm_loadu_si32(void const* mem_addr) {
|
||||
int val;
|
||||
memcpy(&val, mem_addr, sizeof(val));
|
||||
return _mm_cvtsi32_si128(val);
|
||||
}
|
||||
# elif defined(_MSC_VER)
|
||||
/* MSVC needs this for _mm_loadu_si32 */
|
||||
# include <immintrin.h>
|
||||
# endif
|
||||
|
||||
# define OP_CVTEPI8_EPI32_M32(x) \
|
||||
(_mm_cvtepi8_epi32(_mm_loadu_si32(x)))
|
||||
|
||||
# define OP_CVTEPI16_EPI32_M64(x) \
|
||||
(_mm_cvtepi16_epi32(_mm_loadl_epi64((__m128i *)(void*)(x))))
|
||||
|
||||
# endif
|
||||
|
||||
#endif
|
||||
BIN
src/modules/ZoomIt/ZoomIt/selfie_segmentation.onnx
Normal file
BIN
src/modules/ZoomIt/ZoomIt/selfie_segmentation.onnx
Normal file
Binary file not shown.
@@ -17,4 +17,6 @@ internal sealed class Icons
|
||||
internal static IconInfo TimeDateIcon { get; } = new IconInfo("\uEC92");
|
||||
|
||||
internal static IconInfo ErrorIcon { get; } = IconHelpers.FromRelativePaths("Microsoft.CmdPal.Ext.TimeDate\\Assets\\Warning.light.png", "Microsoft.CmdPal.Ext.TimeDate\\Assets\\Warning.dark.png");
|
||||
|
||||
internal static IconInfo NotificationCenterIcon { get; } = new IconInfo("\uEA8F");
|
||||
}
|
||||
|
||||
@@ -959,5 +959,23 @@ namespace Microsoft.CmdPal.Ext.TimeDate {
|
||||
return ResourceManager.GetString("timedate_copy_time_command_name", resourceCulture);
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Looks up a localized string similar to Open notification center.
|
||||
/// </summary>
|
||||
public static string timedate_show_notification_center_command_name {
|
||||
get {
|
||||
return ResourceManager.GetString("timedate_show_notification_center_command_name", resourceCulture);
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Looks up a localized string similar to Notification center.
|
||||
/// </summary>
|
||||
public static string timedate_notification_center_band_title {
|
||||
get {
|
||||
return ResourceManager.GetString("timedate_notification_center_band_title", resourceCulture);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@@ -444,4 +444,12 @@
|
||||
<value>Copy date</value>
|
||||
<comment>Name of a command to copy the current date to the clipboard</comment>
|
||||
</data>
|
||||
<data name="timedate_show_notification_center_command_name" xml:space="preserve">
|
||||
<value>Open notification center</value>
|
||||
<comment>Name of a command to open the Windows notification center (Action Center)</comment>
|
||||
</data>
|
||||
<data name="timedate_notification_center_band_title" xml:space="preserve">
|
||||
<value>Notification center</value>
|
||||
<comment>Title for the notification center dock band</comment>
|
||||
</data>
|
||||
</root>
|
||||
@@ -21,6 +21,7 @@ public sealed partial class TimeDateCommandsProvider : CommandProvider
|
||||
private readonly FallbackTimeDateItem _fallbackTimeDateItem = new(_settingsManager);
|
||||
|
||||
private readonly ListItem _bandItem;
|
||||
private readonly ListItem _notificationCenterBandItem;
|
||||
|
||||
public TimeDateCommandsProvider()
|
||||
{
|
||||
@@ -37,6 +38,7 @@ public sealed partial class TimeDateCommandsProvider : CommandProvider
|
||||
Settings = _settingsManager.Settings;
|
||||
|
||||
_bandItem = new NowDockBand();
|
||||
_notificationCenterBandItem = new NotificationCenterDockBand();
|
||||
}
|
||||
|
||||
private string GetTranslatedPluginDescription()
|
||||
@@ -54,7 +56,7 @@ public sealed partial class TimeDateCommandsProvider : CommandProvider
|
||||
|
||||
public override ICommandItem[] GetDockBands()
|
||||
{
|
||||
var wrappedBand = new WrappedDockItem(
|
||||
var clockBand = new WrappedDockItem(
|
||||
[_bandItem],
|
||||
"com.microsoft.cmdpal.timedate.dockBand",
|
||||
Resources.Microsoft_plugin_timedate_dock_band_title)
|
||||
@@ -62,7 +64,15 @@ public sealed partial class TimeDateCommandsProvider : CommandProvider
|
||||
Icon = Icons.TimeDateExtIcon,
|
||||
};
|
||||
|
||||
return new ICommandItem[] { wrappedBand };
|
||||
var notificationBand = new WrappedDockItem(
|
||||
[_notificationCenterBandItem],
|
||||
"com.microsoft.cmdpal.timedate.notificationCenterBand",
|
||||
Resources.timedate_notification_center_band_title)
|
||||
{
|
||||
Icon = Icons.NotificationCenterIcon,
|
||||
};
|
||||
|
||||
return new ICommandItem[] { clockBand, notificationBand };
|
||||
}
|
||||
}
|
||||
#pragma warning disable SA1402 // File may only contain a single type
|
||||
@@ -75,7 +85,8 @@ internal sealed partial class NowDockBand : ListItem
|
||||
|
||||
public NowDockBand()
|
||||
{
|
||||
Command = new NoOpCommand() { Id = "com.microsoft.cmdpal.timedate.dockBand" };
|
||||
// Open Notification Center on click
|
||||
Command = new OpenUrlCommand("ms-actioncenter:") { Id = "com.microsoft.cmdpal.timedate.dockBand", Name = Resources.timedate_show_notification_center_command_name, Result = CommandResult.Dismiss(), Icon = null };
|
||||
_copyTimeCommand = new CopyTextCommand(string.Empty) { Name = Resources.timedate_copy_time_command_name };
|
||||
_copyDateCommand = new CopyTextCommand(string.Empty) { Name = Resources.timedate_copy_date_command_name };
|
||||
MoreCommands = [
|
||||
@@ -135,4 +146,20 @@ internal sealed partial class NowDockBand : ListItem
|
||||
_copyTimeCommand.Text = timeString;
|
||||
}
|
||||
}
|
||||
|
||||
internal sealed partial class NotificationCenterDockBand : ListItem
|
||||
{
|
||||
public NotificationCenterDockBand()
|
||||
{
|
||||
Icon = Icons.NotificationCenterIcon; // Notification bell
|
||||
Title = Resources.timedate_notification_center_band_title;
|
||||
Command = new OpenUrlCommand("ms-actioncenter:")
|
||||
{
|
||||
Id = "com.microsoft.cmdpal.timedate.notificationCenterBand",
|
||||
Name = Resources.timedate_show_notification_center_command_name,
|
||||
Result = CommandResult.Dismiss(),
|
||||
};
|
||||
}
|
||||
}
|
||||
|
||||
#pragma warning restore SA1402 // File may only contain a single type
|
||||
|
||||
@@ -6,6 +6,8 @@
|
||||
#include <string>
|
||||
#include <memory>
|
||||
|
||||
#include <common/utils/json.h>
|
||||
|
||||
#include <common/utils/logger_helper.h>
|
||||
#include <keyboardmanager/KeyboardManagerEditor/KeyboardManagerEditor.h>
|
||||
#include <keyboardmanager/KeyboardManagerEditorLibrary/EditorHelpers.h>
|
||||
@@ -41,6 +43,35 @@ extern "C"
|
||||
return buffer;
|
||||
}
|
||||
|
||||
// Populates the template metadata fields on a marshaled mapping. Always sets both pointers
|
||||
// (the editor frees every field), using empty strings for non-template / non-RunProgram entries.
|
||||
// Note: kept void-returning so it is valid inside the surrounding extern "C" block (a C-linkage
|
||||
// function may not return a C++ type such as std::wstring).
|
||||
void SetTemplateMetadata(ShortcutMapping* mapping, const Shortcut& targetShortcut)
|
||||
{
|
||||
mapping->templateId = AllocateAndCopyString(targetShortcut.templateId);
|
||||
|
||||
if (targetShortcut.templateParameters.empty())
|
||||
{
|
||||
mapping->templateParametersJson = AllocateAndCopyString(L"");
|
||||
return;
|
||||
}
|
||||
|
||||
json::JsonObject paramsObj;
|
||||
for (auto const& [k, v] : targetShortcut.templateParameters)
|
||||
{
|
||||
paramsObj.SetNamedValue(k, json::JsonValue::CreateStringValue(v));
|
||||
}
|
||||
|
||||
mapping->templateParametersJson = AllocateAndCopyString(paramsObj.Stringify().c_str());
|
||||
}
|
||||
|
||||
void SetEmptyTemplateMetadata(ShortcutMapping* mapping)
|
||||
{
|
||||
mapping->templateId = AllocateAndCopyString(L"");
|
||||
mapping->templateParametersJson = AllocateAndCopyString(L"");
|
||||
}
|
||||
|
||||
int GetSingleKeyRemapCount(void* config)
|
||||
{
|
||||
auto mapping = static_cast<MappingConfiguration*>(config);
|
||||
@@ -382,6 +413,17 @@ bool GetShortcutRemapByType(void* config, int operationType, int index, Shortcut
|
||||
mapping->uriToOpen = AllocateAndCopyString(L"");
|
||||
}
|
||||
|
||||
// Carry template metadata back to the editor so a template mapping re-opens as RunTemplate.
|
||||
// Only RunProgram shortcuts ever carry it; every other entry gets empty (but allocated) fields.
|
||||
if (targetShortcutUnion.index() == 1)
|
||||
{
|
||||
SetTemplateMetadata(mapping, std::get<Shortcut>(targetShortcutUnion));
|
||||
}
|
||||
else
|
||||
{
|
||||
SetEmptyTemplateMetadata(mapping);
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
@@ -483,6 +525,17 @@ bool GetShortcutRemapByType(void* config, int operationType, int index, Shortcut
|
||||
mapping->uriToOpen = AllocateAndCopyString(L"");
|
||||
}
|
||||
|
||||
// Carry template metadata back to the editor so a template mapping re-opens as RunTemplate.
|
||||
// Only RunProgram shortcuts ever carry it; every other entry gets empty (but allocated) fields.
|
||||
if (targetShortcutUnion.index() == 1)
|
||||
{
|
||||
SetTemplateMetadata(mapping, std::get<Shortcut>(targetShortcutUnion));
|
||||
}
|
||||
else
|
||||
{
|
||||
SetEmptyTemplateMetadata(mapping);
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
@@ -527,13 +580,15 @@ bool GetShortcutRemapByType(void* config, int operationType, int index, Shortcut
|
||||
const wchar_t* originalKeys,
|
||||
const wchar_t* targetKeys,
|
||||
const wchar_t* targetApp,
|
||||
int operationType,
|
||||
int operationType,
|
||||
const wchar_t* appPathOrUri,
|
||||
const wchar_t* args,
|
||||
const wchar_t* startDirectory,
|
||||
int elevation,
|
||||
int ifRunningAction,
|
||||
int visibility)
|
||||
int visibility,
|
||||
const wchar_t* templateId,
|
||||
const wchar_t* templateParametersJson)
|
||||
{
|
||||
auto mappingConfig = static_cast<MappingConfiguration*>(config);
|
||||
|
||||
@@ -558,6 +613,28 @@ bool GetShortcutRemapByType(void* config, int operationType, int index, Shortcut
|
||||
std::get<Shortcut>(targetShortcut).alreadyRunningAction = static_cast<Shortcut::ProgramAlreadyRunningAction>(ifRunningAction);
|
||||
std::get<Shortcut>(targetShortcut).startWindowType = static_cast<Shortcut::StartWindowType>(visibility);
|
||||
std::get<Shortcut>(targetShortcut).operationType = static_cast<Shortcut::OperationType>(operationType);
|
||||
|
||||
// Optional template metadata — only set when provided.
|
||||
if (templateId && templateId[0] != L'\0')
|
||||
{
|
||||
std::get<Shortcut>(targetShortcut).templateId = std::wstring(templateId);
|
||||
}
|
||||
if (templateParametersJson && templateParametersJson[0] != L'\0')
|
||||
{
|
||||
json::JsonObject paramsObj;
|
||||
if (json::JsonObject::TryParse(templateParametersJson, paramsObj))
|
||||
{
|
||||
for (auto const& kv : paramsObj)
|
||||
{
|
||||
if (kv.Value().ValueType() == json::JsonValueType::String)
|
||||
{
|
||||
std::get<Shortcut>(targetShortcut).templateParameters.emplace(
|
||||
std::wstring(kv.Key()),
|
||||
std::wstring(kv.Value().GetString()));
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
break;
|
||||
case 2:
|
||||
targetShortcut = Shortcut(targetKeys);
|
||||
|
||||
@@ -33,6 +33,8 @@ struct ShortcutMapping
|
||||
wchar_t* programPath;
|
||||
wchar_t* programArgs;
|
||||
wchar_t* uriToOpen;
|
||||
wchar_t* templateId;
|
||||
wchar_t* templateParametersJson;
|
||||
};
|
||||
|
||||
extern "C"
|
||||
@@ -69,7 +71,9 @@ extern "C"
|
||||
const wchar_t* startDirectory = nullptr,
|
||||
int elevation = 0,
|
||||
int ifRunningAction = 0,
|
||||
int visibility = 0);
|
||||
int visibility = 0,
|
||||
const wchar_t* templateId = nullptr,
|
||||
const wchar_t* templateParametersJson = nullptr);
|
||||
|
||||
__declspec(dllexport) void GetKeyDisplayName(int keyCode, wchar_t* keyName, int maxCount);
|
||||
__declspec(dllexport) int GetKeyCodeFromName(const wchar_t* keyName);
|
||||
|
||||
@@ -0,0 +1,125 @@
|
||||
// Copyright (c) Microsoft Corporation
|
||||
// The Microsoft Corporation licenses this file to you under the MIT license.
|
||||
// See the LICENSE file in the project root for more information.
|
||||
|
||||
using System.Linq;
|
||||
using System.Text.Json;
|
||||
|
||||
using KeyboardManagerEditorUI.Templates;
|
||||
using Microsoft.VisualStudio.TestTools.UnitTesting;
|
||||
|
||||
namespace KeyboardManagerEditorUI.UnitTests
|
||||
{
|
||||
/// <summary>
|
||||
/// Validates that the catalog data model deserializes the shipped powertoyscli.json schema
|
||||
/// shape correctly. (Reflection-based deserialization mirrors the source-generated context.)
|
||||
/// </summary>
|
||||
[TestClass]
|
||||
public class CommandTemplateCatalogModelTests
|
||||
{
|
||||
private static readonly JsonSerializerOptions Options = new()
|
||||
{
|
||||
PropertyNamingPolicy = JsonNamingPolicy.CamelCase,
|
||||
};
|
||||
|
||||
private const string SeedCatalog = @"
|
||||
{
|
||||
""schemaVersion"": 1,
|
||||
""modules"": [
|
||||
{
|
||||
""id"": ""settings"",
|
||||
""displayResourceKey"": ""TemplateModule_Settings"",
|
||||
""iconGlyph"": """",
|
||||
""commands"": [
|
||||
{
|
||||
""id"": ""settings.openMain"",
|
||||
""displayResourceKey"": ""TemplateCmd_Settings_OpenMain"",
|
||||
""executable"": ""%LOCALAPPDATA%\\PowerToys\\PowerToys.exe"",
|
||||
""argsTemplate"": ""--open-settings"",
|
||||
""parameters"": []
|
||||
},
|
||||
{
|
||||
""id"": ""settings.openModule"",
|
||||
""displayResourceKey"": ""TemplateCmd_Settings_OpenModule"",
|
||||
""executable"": ""%LOCALAPPDATA%\\PowerToys\\PowerToys.exe"",
|
||||
""argsTemplate"": ""--open-settings={module}"",
|
||||
""parameters"": [
|
||||
{
|
||||
""name"": ""module"",
|
||||
""labelResourceKey"": ""TemplateParam_Module"",
|
||||
""type"": ""Combo"",
|
||||
""required"": true,
|
||||
""choices"": [
|
||||
{ ""value"": ""ColorPicker"", ""displayResourceKey"": ""Module_ColorPicker"" }
|
||||
]
|
||||
}
|
||||
]
|
||||
}
|
||||
]
|
||||
}
|
||||
]
|
||||
}";
|
||||
|
||||
[TestMethod]
|
||||
public void Deserialize_SeedCatalog_MapsAllFields()
|
||||
{
|
||||
var catalog = JsonSerializer.Deserialize<PowerToysCliCatalog>(SeedCatalog, Options);
|
||||
|
||||
Assert.IsNotNull(catalog);
|
||||
Assert.AreEqual(1, catalog!.SchemaVersion);
|
||||
Assert.AreEqual(1, catalog.Modules.Count);
|
||||
|
||||
var module = catalog.Modules[0];
|
||||
Assert.AreEqual("settings", module.Id);
|
||||
Assert.AreEqual("TemplateModule_Settings", module.DisplayResourceKey);
|
||||
Assert.AreEqual(2, module.Commands.Count);
|
||||
|
||||
var openMain = module.Commands.Single(c => c.Id == "settings.openMain");
|
||||
Assert.AreEqual("--open-settings", openMain.ArgsTemplate);
|
||||
Assert.AreEqual(0, openMain.Parameters.Count);
|
||||
|
||||
var openModule = module.Commands.Single(c => c.Id == "settings.openModule");
|
||||
Assert.AreEqual("--open-settings={module}", openModule.ArgsTemplate);
|
||||
Assert.AreEqual(1, openModule.Parameters.Count);
|
||||
|
||||
var param = openModule.Parameters[0];
|
||||
Assert.AreEqual("module", param.Name);
|
||||
Assert.AreEqual("Combo", param.Type);
|
||||
Assert.IsTrue(param.Required);
|
||||
Assert.IsNotNull(param.Choices);
|
||||
Assert.AreEqual("ColorPicker", param.Choices![0].Value);
|
||||
Assert.AreEqual("Module_ColorPicker", param.Choices[0].DisplayResourceKey);
|
||||
}
|
||||
|
||||
[TestMethod]
|
||||
public void Deserialize_ForwardCompatibleVersion_StillParses()
|
||||
{
|
||||
// A newer, additive schemaVersion with an unknown field must deserialize (unknown fields ignored).
|
||||
const string newer = @"
|
||||
{
|
||||
""schemaVersion"": 2,
|
||||
""futureField"": ""ignored"",
|
||||
""modules"": [
|
||||
{ ""id"": ""m"", ""displayResourceKey"": ""k"", ""commands"": [] }
|
||||
]
|
||||
}";
|
||||
|
||||
var catalog = JsonSerializer.Deserialize<PowerToysCliCatalog>(newer, Options);
|
||||
|
||||
Assert.IsNotNull(catalog);
|
||||
Assert.AreEqual(2, catalog!.SchemaVersion);
|
||||
Assert.AreEqual(1, catalog.Modules.Count);
|
||||
}
|
||||
|
||||
[TestMethod]
|
||||
public void Deserialize_OptionalChoices_DefaultToNull()
|
||||
{
|
||||
var param = JsonSerializer.Deserialize<TemplateParameter>(
|
||||
@"{ ""name"": ""p"", ""type"": ""Text"" }", Options);
|
||||
|
||||
Assert.IsNotNull(param);
|
||||
Assert.AreEqual("p", param!.Name);
|
||||
Assert.IsNull(param.Choices);
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,32 @@
|
||||
<Project Sdk="Microsoft.NET.Sdk">
|
||||
<!-- Look at Directory.Build.props in root for common stuff as well -->
|
||||
<Import Project="$(RepoRoot)src\Common.Dotnet.CsWinRT.props" />
|
||||
|
||||
<PropertyGroup>
|
||||
<SelfContained>true</SelfContained>
|
||||
<RuntimeIdentifier Condition="'$(Platform)' == 'x64'">win-x64</RuntimeIdentifier>
|
||||
<RuntimeIdentifier Condition="'$(Platform)' == 'ARM64'">win-arm64</RuntimeIdentifier>
|
||||
<AppendRuntimeIdentifierToOutputPath>false</AppendRuntimeIdentifierToOutputPath>
|
||||
<IsPackable>false</IsPackable>
|
||||
<OutputType>Exe</OutputType>
|
||||
<Nullable>enable</Nullable>
|
||||
<OutputPath>$(RepoRoot)$(Configuration)\$(Platform)\tests\KeyboardManagerEditorUITests\</OutputPath>
|
||||
</PropertyGroup>
|
||||
|
||||
<ItemGroup>
|
||||
<PackageReference Include="MSTest" />
|
||||
</ItemGroup>
|
||||
|
||||
<!--
|
||||
Source-link the pure (WinUI-free) template logic so it can be unit-tested without referencing
|
||||
the KeyboardManagerEditorUI WinExe app. These files have no UI/WinRT dependencies.
|
||||
-->
|
||||
<ItemGroup>
|
||||
<Compile Include="..\KeyboardManagerEditorUI\Templates\TemplateResolver.cs" Link="Templates\TemplateResolver.cs" />
|
||||
<Compile Include="..\KeyboardManagerEditorUI\Templates\CommandTemplate.cs" Link="Templates\CommandTemplate.cs" />
|
||||
<Compile Include="..\KeyboardManagerEditorUI\Templates\CommandTemplateModule.cs" Link="Templates\CommandTemplateModule.cs" />
|
||||
<Compile Include="..\KeyboardManagerEditorUI\Templates\PowerToysCliCatalog.cs" Link="Templates\PowerToysCliCatalog.cs" />
|
||||
<Compile Include="..\KeyboardManagerEditorUI\Templates\TemplateParameter.cs" Link="Templates\TemplateParameter.cs" />
|
||||
<Compile Include="..\KeyboardManagerEditorUI\Templates\TemplateChoice.cs" Link="Templates\TemplateChoice.cs" />
|
||||
</ItemGroup>
|
||||
</Project>
|
||||
@@ -0,0 +1,145 @@
|
||||
// Copyright (c) Microsoft Corporation
|
||||
// The Microsoft Corporation licenses this file to you under the MIT license.
|
||||
// See the LICENSE file in the project root for more information.
|
||||
|
||||
using System.Collections.Generic;
|
||||
|
||||
using KeyboardManagerEditorUI.Templates;
|
||||
using Microsoft.VisualStudio.TestTools.UnitTesting;
|
||||
|
||||
namespace KeyboardManagerEditorUI.UnitTests
|
||||
{
|
||||
[TestClass]
|
||||
public class TemplateResolverTests
|
||||
{
|
||||
private static CommandTemplate Template(string args, params TemplateParameter[] parameters)
|
||||
=> new()
|
||||
{
|
||||
Id = "test.cmd",
|
||||
Executable = "%LOCALAPPDATA%\\PowerToys\\PowerToys.exe",
|
||||
ArgsTemplate = args,
|
||||
Parameters = new List<TemplateParameter>(parameters),
|
||||
};
|
||||
|
||||
private static TemplateParameter Param(string name, string type = "Text", bool required = true)
|
||||
=> new() { Name = name, Type = type, Required = required };
|
||||
|
||||
[TestMethod]
|
||||
public void Resolve_SubstitutesPresentValue()
|
||||
{
|
||||
var result = TemplateResolver.Resolve(
|
||||
Template("--open-settings={module}", Param("module")),
|
||||
new Dictionary<string, string> { ["module"] = "ColorPicker" });
|
||||
|
||||
Assert.AreEqual("%LOCALAPPDATA%\\PowerToys\\PowerToys.exe", result.Executable);
|
||||
Assert.AreEqual("--open-settings=ColorPicker", result.Args);
|
||||
}
|
||||
|
||||
[TestMethod]
|
||||
public void Resolve_MissingValue_SubstitutesEmpty()
|
||||
{
|
||||
var result = TemplateResolver.Resolve(
|
||||
Template("--open-settings={module}", Param("module")),
|
||||
new Dictionary<string, string>());
|
||||
|
||||
Assert.AreEqual("--open-settings=", result.Args);
|
||||
}
|
||||
|
||||
[TestMethod]
|
||||
public void Resolve_NullValues_SubstitutesEmpty()
|
||||
{
|
||||
var result = TemplateResolver.Resolve(
|
||||
Template("--open-settings={module}", Param("module")),
|
||||
values: null);
|
||||
|
||||
Assert.AreEqual("--open-settings=", result.Args);
|
||||
}
|
||||
|
||||
[TestMethod]
|
||||
public void Resolve_NoParameters_ReturnsTemplateVerbatim()
|
||||
{
|
||||
var result = TemplateResolver.Resolve(Template("--open-settings"), null);
|
||||
Assert.AreEqual("--open-settings", result.Args);
|
||||
}
|
||||
|
||||
[TestMethod]
|
||||
public void Resolve_UnknownPlaceholder_LeftUntouched()
|
||||
{
|
||||
// {other} is not a declared parameter, so it must be preserved literally.
|
||||
var result = TemplateResolver.Resolve(
|
||||
Template("--a={module} --b={other}", Param("module")),
|
||||
new Dictionary<string, string> { ["module"] = "X" });
|
||||
|
||||
Assert.AreEqual("--a=X --b={other}", result.Args);
|
||||
}
|
||||
|
||||
[TestMethod]
|
||||
public void Resolve_IsSinglePass_DoesNotReSubstituteInjectedPlaceholder()
|
||||
{
|
||||
// If the first parameter's value contains "{second}", the second pass must NOT replace it.
|
||||
var result = TemplateResolver.Resolve(
|
||||
Template("{first}-{second}", Param("first"), Param("second")),
|
||||
new Dictionary<string, string>
|
||||
{
|
||||
["first"] = "{second}",
|
||||
["second"] = "INJECTED",
|
||||
});
|
||||
|
||||
Assert.AreEqual("{second}-INJECTED", result.Args);
|
||||
}
|
||||
|
||||
[TestMethod]
|
||||
public void Resolve_NoSubstringCollisionBetweenSimilarNames()
|
||||
{
|
||||
var result = TemplateResolver.Resolve(
|
||||
Template("{module}|{moduleVersion}", Param("module"), Param("moduleVersion")),
|
||||
new Dictionary<string, string> { ["module"] = "A", ["moduleVersion"] = "B" });
|
||||
|
||||
Assert.AreEqual("A|B", result.Args);
|
||||
}
|
||||
|
||||
[TestMethod]
|
||||
public void Resolve_ValueWithSpace_IsQuoted()
|
||||
{
|
||||
var result = TemplateResolver.Resolve(
|
||||
Template("--path={p}", Param("p")),
|
||||
new Dictionary<string, string> { ["p"] = "C:\\Program Files\\App" });
|
||||
|
||||
Assert.AreEqual("--path=\"C:\\Program Files\\App\"", result.Args);
|
||||
}
|
||||
|
||||
[TestMethod]
|
||||
public void Resolve_SimpleValue_NotQuoted()
|
||||
{
|
||||
var result = TemplateResolver.Resolve(
|
||||
Template("--x={p}", Param("p")),
|
||||
new Dictionary<string, string> { ["p"] = "Plain" });
|
||||
|
||||
Assert.AreEqual("--x=Plain", result.Args);
|
||||
}
|
||||
|
||||
[TestMethod]
|
||||
public void Resolve_ValueWithQuote_IsEscaped()
|
||||
{
|
||||
var result = TemplateResolver.Resolve(
|
||||
Template("--x={p}", Param("p")),
|
||||
new Dictionary<string, string> { ["p"] = "a\"b c" });
|
||||
|
||||
// Embedded quote is backslash-escaped and the whole value wrapped in quotes.
|
||||
Assert.AreEqual("--x=\"a\\\"b c\"", result.Args);
|
||||
}
|
||||
|
||||
[TestMethod]
|
||||
public void QuoteArgumentIfNeeded_TrailingBackslashesBeforeClosingQuote_AreDoubled()
|
||||
{
|
||||
// "a b\\" must become "a b\\\\" so the backslashes don't escape the closing quote.
|
||||
Assert.AreEqual("\"a b\\\\\"", TemplateResolver.QuoteArgumentIfNeeded("a b\\"));
|
||||
}
|
||||
|
||||
[TestMethod]
|
||||
public void QuoteArgumentIfNeeded_Empty_ReturnsEmpty()
|
||||
{
|
||||
Assert.AreEqual(string.Empty, TemplateResolver.QuoteArgumentIfNeeded(string.Empty));
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,87 @@
|
||||
<?xml version="1.0" encoding="utf-8" ?>
|
||||
<UserControl
|
||||
x:Class="KeyboardManagerEditorUI.Controls.CommandTemplatePickerControl"
|
||||
xmlns="http://schemas.microsoft.com/winfx/2006/xaml/presentation"
|
||||
xmlns:x="http://schemas.microsoft.com/winfx/2006/xaml"
|
||||
xmlns:d="http://schemas.microsoft.com/expression/blend/2008"
|
||||
xmlns:local="using:KeyboardManagerEditorUI.Controls"
|
||||
xmlns:mc="http://schemas.openxmlformats.org/markup-compatibility/2006"
|
||||
xmlns:vm="using:KeyboardManagerEditorUI.ViewModels"
|
||||
mc:Ignorable="d">
|
||||
|
||||
<UserControl.Resources>
|
||||
<DataTemplate x:Key="TextParamTemplate" x:DataType="vm:TemplateParameterViewModel">
|
||||
<TextBox
|
||||
Header="{x:Bind Label}"
|
||||
Text="{x:Bind Value, Mode=TwoWay, UpdateSourceTrigger=PropertyChanged}" />
|
||||
</DataTemplate>
|
||||
|
||||
<DataTemplate x:Key="ComboParamTemplate" x:DataType="vm:TemplateParameterViewModel">
|
||||
<ComboBox
|
||||
HorizontalAlignment="Stretch"
|
||||
Header="{x:Bind Label}"
|
||||
ItemsSource="{x:Bind Choices}"
|
||||
SelectedItem="{x:Bind SelectedChoice, Mode=TwoWay}"
|
||||
DisplayMemberPath="DisplayText" />
|
||||
</DataTemplate>
|
||||
|
||||
<local:TemplateParameterSelector
|
||||
x:Key="ParamSelector"
|
||||
ComboTemplate="{StaticResource ComboParamTemplate}"
|
||||
TextTemplate="{StaticResource TextParamTemplate}" />
|
||||
</UserControl.Resources>
|
||||
|
||||
<StackPanel Orientation="Vertical" Spacing="12">
|
||||
<TextBlock
|
||||
x:Name="SelectionDescriptionText"
|
||||
FontStyle="Italic"
|
||||
Foreground="{ThemeResource TextFillColorSecondaryBrush}"
|
||||
Text="{x:Bind ViewModel.SelectionDescription, Mode=OneWay}" />
|
||||
|
||||
<Rectangle
|
||||
Height="1"
|
||||
HorizontalAlignment="Stretch"
|
||||
Fill="{ThemeResource DividerStrokeColorDefaultBrush}" />
|
||||
|
||||
<ItemsControl
|
||||
ItemsSource="{x:Bind ViewModel.CurrentParameters, Mode=OneWay}"
|
||||
ItemTemplateSelector="{StaticResource ParamSelector}">
|
||||
<ItemsControl.ItemsPanel>
|
||||
<ItemsPanelTemplate>
|
||||
<StackPanel Orientation="Vertical" Spacing="12" />
|
||||
</ItemsPanelTemplate>
|
||||
</ItemsControl.ItemsPanel>
|
||||
</ItemsControl>
|
||||
|
||||
<Rectangle
|
||||
Height="1"
|
||||
HorizontalAlignment="Stretch"
|
||||
Fill="{ThemeResource DividerStrokeColorDefaultBrush}" />
|
||||
|
||||
<TextBlock x:Uid="TemplatePreviewLabel" FontWeight="SemiBold" />
|
||||
|
||||
<TextBlock
|
||||
FontFamily="Consolas"
|
||||
Foreground="{ThemeResource TextFillColorPrimaryBrush}"
|
||||
IsTextSelectionEnabled="True"
|
||||
Text="{x:Bind ViewModel.ResolvedCommandLine, Mode=OneWay}"
|
||||
TextWrapping="Wrap" />
|
||||
|
||||
<InfoBar
|
||||
x:Name="MissingTemplateInfoBar"
|
||||
x:Uid="MissingTemplateInfoBar"
|
||||
IsClosable="False"
|
||||
IsOpen="False"
|
||||
Severity="Warning">
|
||||
<StackPanel
|
||||
Margin="0,0,0,8"
|
||||
Orientation="Horizontal"
|
||||
Spacing="8">
|
||||
<Button
|
||||
x:Name="MissingTemplateKeepButton"
|
||||
x:Uid="TemplateMissingKeepButton"
|
||||
Click="MissingTemplateKeepButton_Click" />
|
||||
</StackPanel>
|
||||
</InfoBar>
|
||||
</StackPanel>
|
||||
</UserControl>
|
||||
@@ -0,0 +1,112 @@
|
||||
// Copyright (c) Microsoft Corporation
|
||||
// The Microsoft Corporation licenses this file to you under the MIT license.
|
||||
// See the LICENSE file in the project root for more information.
|
||||
|
||||
using System;
|
||||
using System.Collections.Generic;
|
||||
|
||||
using KeyboardManagerEditorUI.Helpers;
|
||||
using KeyboardManagerEditorUI.Templates;
|
||||
using KeyboardManagerEditorUI.ViewModels;
|
||||
using Microsoft.UI.Xaml;
|
||||
using Microsoft.UI.Xaml.Controls;
|
||||
|
||||
namespace KeyboardManagerEditorUI.Controls
|
||||
{
|
||||
public sealed partial class CommandTemplatePickerControl : UserControl
|
||||
{
|
||||
public CommandTemplatePickerControl()
|
||||
{
|
||||
InitializeComponent();
|
||||
ViewModel = new CommandTemplatePickerViewModel();
|
||||
|
||||
// Re-raise SelectionChanged when parameter validity could have changed so the host
|
||||
// (UnifiedMappingControl/MainPage) re-evaluates Save-button enablement, not just on
|
||||
// the initial template pick.
|
||||
ViewModel.PropertyChanged += (_, e) =>
|
||||
{
|
||||
if (e.PropertyName == nameof(CommandTemplatePickerViewModel.IsAllValid))
|
||||
{
|
||||
SelectionChanged?.Invoke(this, EventArgs.Empty);
|
||||
}
|
||||
};
|
||||
}
|
||||
|
||||
public CommandTemplatePickerViewModel ViewModel { get; }
|
||||
|
||||
public event EventHandler? SelectionChanged;
|
||||
|
||||
/// <summary>
|
||||
/// Gets a value indicating whether a template is selected and all its required
|
||||
/// parameters have values. Drives Save-button enablement for the RunTemplate action.
|
||||
/// </summary>
|
||||
public bool IsTemplateInputValid =>
|
||||
ViewModel.SelectedTemplate is not null && ViewModel.IsAllValid;
|
||||
|
||||
public event EventHandler? MissingTemplateKeepRequested;
|
||||
|
||||
public TemplateResolver.Resolved? ResolveCurrent()
|
||||
{
|
||||
if (ViewModel.SelectedTemplate is null)
|
||||
{
|
||||
return null;
|
||||
}
|
||||
|
||||
return TemplateResolver.Resolve(
|
||||
ViewModel.SelectedTemplate,
|
||||
ViewModel.CollectParameterValues());
|
||||
}
|
||||
|
||||
public string? CurrentTemplateId => ViewModel.SelectedTemplate?.Id;
|
||||
|
||||
public Dictionary<string, string> CurrentParameterValues => ViewModel.CollectParameterValues();
|
||||
|
||||
public void LoadExisting(string templateId, IReadOnlyDictionary<string, string>? values)
|
||||
{
|
||||
try
|
||||
{
|
||||
ViewModel.LoadExisting(templateId, values);
|
||||
MissingTemplateInfoBar.IsOpen = false;
|
||||
}
|
||||
catch (InvalidOperationException)
|
||||
{
|
||||
ShowMissingTemplateInfoBar();
|
||||
}
|
||||
}
|
||||
|
||||
public void Reset()
|
||||
{
|
||||
ViewModel.Clear();
|
||||
MissingTemplateInfoBar.IsOpen = false;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Selects a command template by id (driven by the host action menu) and notifies listeners.
|
||||
/// </summary>
|
||||
public void SelectCommand(string templateId)
|
||||
{
|
||||
ViewModel.SelectTemplate(templateId);
|
||||
MissingTemplateInfoBar.IsOpen = false;
|
||||
SelectionChanged?.Invoke(this, EventArgs.Empty);
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Gets the display name of the currently selected command (empty when none selected).
|
||||
/// </summary>
|
||||
public string CurrentCommandDisplay =>
|
||||
ViewModel.SelectedTemplate is { } t
|
||||
? ResourceHelper.GetString(t.DisplayResourceKey)
|
||||
: string.Empty;
|
||||
|
||||
private void ShowMissingTemplateInfoBar()
|
||||
{
|
||||
MissingTemplateInfoBar.IsOpen = true;
|
||||
}
|
||||
|
||||
private void MissingTemplateKeepButton_Click(object sender, RoutedEventArgs e)
|
||||
{
|
||||
MissingTemplateInfoBar.IsOpen = false;
|
||||
MissingTemplateKeepRequested?.Invoke(this, EventArgs.Empty);
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,31 @@
|
||||
// Copyright (c) Microsoft Corporation
|
||||
// The Microsoft Corporation licenses this file to you under the MIT license.
|
||||
// See the LICENSE file in the project root for more information.
|
||||
|
||||
using KeyboardManagerEditorUI.ViewModels;
|
||||
using Microsoft.UI.Xaml;
|
||||
using Microsoft.UI.Xaml.Controls;
|
||||
|
||||
namespace KeyboardManagerEditorUI.Controls
|
||||
{
|
||||
public sealed partial class TemplateParameterSelector : DataTemplateSelector
|
||||
{
|
||||
public DataTemplate? TextTemplate { get; set; }
|
||||
|
||||
public DataTemplate? ComboTemplate { get; set; }
|
||||
|
||||
protected override DataTemplate SelectTemplateCore(object item, DependencyObject container)
|
||||
{
|
||||
if (item is TemplateParameterViewModel vm)
|
||||
{
|
||||
return vm.Type switch
|
||||
{
|
||||
"Combo" => ComboTemplate ?? TextTemplate!,
|
||||
_ => TextTemplate!,
|
||||
};
|
||||
}
|
||||
|
||||
return TextTemplate!;
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -175,52 +175,67 @@
|
||||
Orientation="Vertical"
|
||||
Spacing="8">
|
||||
<TextBlock x:Uid="ActionLabel" FontWeight="SemiBold" />
|
||||
<ComboBox
|
||||
x:Name="ActionTypeComboBox"
|
||||
<DropDownButton
|
||||
x:Name="ActionTypeButton"
|
||||
HorizontalAlignment="Stretch"
|
||||
SelectionChanged="ActionTypeComboBox_SelectionChanged">
|
||||
<ComboBoxItem
|
||||
x:Uid="ActionType_KeyOrShortcut"
|
||||
IsSelected="True"
|
||||
Tag="KeyOrShortcut">
|
||||
<StackPanel Orientation="Horizontal" Spacing="8">
|
||||
<FontIcon FontSize="14" Glyph="" />
|
||||
<TextBlock x:Uid="ActionType_KeyOrShortcut_Text" />
|
||||
</StackPanel>
|
||||
</ComboBoxItem>
|
||||
<ComboBoxItem x:Uid="ActionType_Text" Tag="Text">
|
||||
<StackPanel Orientation="Horizontal" Spacing="8">
|
||||
<FontIcon FontSize="14" Glyph="" />
|
||||
<TextBlock x:Uid="ActionType_Text_Text" />
|
||||
</StackPanel>
|
||||
</ComboBoxItem>
|
||||
<ComboBoxItem x:Uid="ActionType_OpenUrl" Tag="OpenUrl">
|
||||
<StackPanel Orientation="Horizontal" Spacing="8">
|
||||
<FontIcon FontSize="14" Glyph="" />
|
||||
<TextBlock x:Uid="ActionType_OpenUrl_Text" />
|
||||
</StackPanel>
|
||||
</ComboBoxItem>
|
||||
<ComboBoxItem x:Uid="ActionType_OpenApp" Tag="OpenApp">
|
||||
<StackPanel Orientation="Horizontal" Spacing="8">
|
||||
<FontIcon FontSize="14" Glyph="" />
|
||||
<TextBlock x:Uid="ActionType_OpenApp_Text" />
|
||||
</StackPanel>
|
||||
</ComboBoxItem>
|
||||
<ComboBoxItem x:Uid="ActionType_Disable" Tag="Disable">
|
||||
<StackPanel Orientation="Horizontal" Spacing="8">
|
||||
<FontIcon FontSize="14" Glyph="" />
|
||||
<TextBlock x:Uid="ActionType_Disable_Text" />
|
||||
</StackPanel>
|
||||
</ComboBoxItem>
|
||||
<!--
|
||||
<ComboBoxItem x:Uid="ActionType_MouseClick" Tag="MouseClick">
|
||||
<StackPanel Orientation="Horizontal" Spacing="8">
|
||||
<FontIcon FontSize="14" Glyph="" />
|
||||
<TextBlock x:Uid="ActionType_MouseClick_Text" />
|
||||
</StackPanel>
|
||||
</ComboBoxItem>
|
||||
-->
|
||||
</ComboBox>
|
||||
HorizontalContentAlignment="Left">
|
||||
<DropDownButton.Flyout>
|
||||
<MenuFlyout Placement="Bottom">
|
||||
<MenuFlyoutItem
|
||||
x:Name="ActionItem_KeyOrShortcut"
|
||||
x:Uid="ActionType_KeyOrShortcut_Text"
|
||||
Click="OnActionTypeClick"
|
||||
Tag="KeyOrShortcut">
|
||||
<MenuFlyoutItem.Icon>
|
||||
<FontIcon FontSize="14" Glyph="" />
|
||||
</MenuFlyoutItem.Icon>
|
||||
</MenuFlyoutItem>
|
||||
<MenuFlyoutItem
|
||||
x:Name="ActionItem_Text"
|
||||
x:Uid="ActionType_Text_Text"
|
||||
Click="OnActionTypeClick"
|
||||
Tag="Text">
|
||||
<MenuFlyoutItem.Icon>
|
||||
<FontIcon FontSize="14" Glyph="" />
|
||||
</MenuFlyoutItem.Icon>
|
||||
</MenuFlyoutItem>
|
||||
<MenuFlyoutItem
|
||||
x:Name="ActionItem_OpenUrl"
|
||||
x:Uid="ActionType_OpenUrl_Text"
|
||||
Click="OnActionTypeClick"
|
||||
Tag="OpenUrl">
|
||||
<MenuFlyoutItem.Icon>
|
||||
<FontIcon FontSize="14" Glyph="" />
|
||||
</MenuFlyoutItem.Icon>
|
||||
</MenuFlyoutItem>
|
||||
<MenuFlyoutItem
|
||||
x:Name="ActionItem_OpenApp"
|
||||
x:Uid="ActionType_OpenApp_Text"
|
||||
Click="OnActionTypeClick"
|
||||
Tag="OpenApp">
|
||||
<MenuFlyoutItem.Icon>
|
||||
<FontIcon FontSize="14" Glyph="" />
|
||||
</MenuFlyoutItem.Icon>
|
||||
</MenuFlyoutItem>
|
||||
<MenuFlyoutItem
|
||||
x:Name="ActionItem_Disable"
|
||||
x:Uid="ActionType_Disable_Text"
|
||||
Click="OnActionTypeClick"
|
||||
Tag="Disable">
|
||||
<MenuFlyoutItem.Icon>
|
||||
<FontIcon FontSize="14" Glyph="" />
|
||||
</MenuFlyoutItem.Icon>
|
||||
</MenuFlyoutItem>
|
||||
<MenuFlyoutSubItem
|
||||
x:Name="RunPtCommandSubItem"
|
||||
x:Uid="ActionType_RunTemplate_Text">
|
||||
<MenuFlyoutSubItem.Icon>
|
||||
<FontIcon FontSize="14" Glyph="" />
|
||||
</MenuFlyoutSubItem.Icon>
|
||||
</MenuFlyoutSubItem>
|
||||
</MenuFlyout>
|
||||
</DropDownButton.Flyout>
|
||||
</DropDownButton>
|
||||
<Rectangle
|
||||
Height="1"
|
||||
Margin="0,12,0,12"
|
||||
@@ -229,7 +244,7 @@
|
||||
<tkcontrols:SwitchPresenter
|
||||
x:Name="ActionSwitchPresenter"
|
||||
TargetType="x:String"
|
||||
Value="{Binding SelectedItem.Tag, ElementName=ActionTypeComboBox}">
|
||||
Value="KeyOrShortcut">
|
||||
<!-- Key or Shortcut Action -->
|
||||
<tkcontrols:Case Value="KeyOrShortcut">
|
||||
<ToggleButton
|
||||
@@ -398,6 +413,12 @@
|
||||
Foreground="{ThemeResource TextFillColorSecondaryBrush}"
|
||||
TextWrapping="Wrap" />
|
||||
</tkcontrols:Case>
|
||||
<!-- Run From Template Action -->
|
||||
<tkcontrols:Case Value="RunTemplate">
|
||||
<local:CommandTemplatePickerControl
|
||||
x:Name="TemplatePicker"
|
||||
MissingTemplateKeepRequested="TemplatePicker_MissingTemplateKeepRequested" />
|
||||
</tkcontrols:Case>
|
||||
</tkcontrols:SwitchPresenter>
|
||||
</StackPanel>
|
||||
<!-- Validation InfoBar spanning all columns -->
|
||||
|
||||
@@ -8,6 +8,7 @@ using System.Collections.ObjectModel;
|
||||
using System.Linq;
|
||||
using KeyboardManagerEditorUI.Helpers;
|
||||
using KeyboardManagerEditorUI.Interop;
|
||||
using KeyboardManagerEditorUI.Templates;
|
||||
using Microsoft.UI.Xaml;
|
||||
using Microsoft.UI.Xaml.Controls;
|
||||
using Windows.Storage;
|
||||
@@ -44,6 +45,13 @@ namespace KeyboardManagerEditorUI.Controls
|
||||
private bool _urlPathDirty;
|
||||
private bool _programPathDirty;
|
||||
|
||||
private string _currentActionTag = "KeyOrShortcut";
|
||||
|
||||
// Resolved program path/args captured when loading a template mapping, so the
|
||||
// missing-template "Keep as plain command" path can preserve the command.
|
||||
private string _templateFallbackProgramPath = string.Empty;
|
||||
private string _templateFallbackProgramArgs = string.Empty;
|
||||
|
||||
public bool AllowChords { get; set; } = true;
|
||||
|
||||
#endregion
|
||||
@@ -79,6 +87,7 @@ namespace KeyboardManagerEditorUI.Controls
|
||||
OpenApp,
|
||||
MouseClick,
|
||||
Disable,
|
||||
RunTemplate,
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
@@ -119,26 +128,16 @@ namespace KeyboardManagerEditorUI.Controls
|
||||
/// <summary>
|
||||
/// Gets the current action type.
|
||||
/// </summary>
|
||||
public ActionType CurrentActionType
|
||||
public ActionType CurrentActionType => _currentActionTag switch
|
||||
{
|
||||
get
|
||||
{
|
||||
if (ActionTypeComboBox?.SelectedItem is ComboBoxItem item)
|
||||
{
|
||||
return item.Tag?.ToString() switch
|
||||
{
|
||||
"Text" => ActionType.Text,
|
||||
"OpenUrl" => ActionType.OpenUrl,
|
||||
"OpenApp" => ActionType.OpenApp,
|
||||
"MouseClick" => ActionType.MouseClick,
|
||||
"Disable" => ActionType.Disable,
|
||||
_ => ActionType.KeyOrShortcut,
|
||||
};
|
||||
}
|
||||
|
||||
return ActionType.KeyOrShortcut;
|
||||
}
|
||||
}
|
||||
"Text" => ActionType.Text,
|
||||
"OpenUrl" => ActionType.OpenUrl,
|
||||
"OpenApp" => ActionType.OpenApp,
|
||||
"MouseClick" => ActionType.MouseClick,
|
||||
"Disable" => ActionType.Disable,
|
||||
"RunTemplate" => ActionType.RunTemplate,
|
||||
_ => ActionType.KeyOrShortcut,
|
||||
};
|
||||
|
||||
#endregion
|
||||
|
||||
@@ -163,6 +162,8 @@ namespace KeyboardManagerEditorUI.Controls
|
||||
RaiseValidationStateChanged();
|
||||
};
|
||||
|
||||
BuildRunPtCommandMenu();
|
||||
|
||||
this.Unloaded += UnifiedMappingControl_Unloaded;
|
||||
}
|
||||
|
||||
@@ -172,24 +173,49 @@ namespace KeyboardManagerEditorUI.Controls
|
||||
|
||||
private void UserControl_Loaded(object sender, RoutedEventArgs e)
|
||||
{
|
||||
// Set up event handlers for app-specific checkbox
|
||||
// Set up event handlers for app-specific checkbox.
|
||||
// Detach first so re-entering the visual tree (dialog reopened) does not stack handlers.
|
||||
AppSpecificCheckBox.Checked -= AppSpecificCheckBox_Changed;
|
||||
AppSpecificCheckBox.Unchecked -= AppSpecificCheckBox_Changed;
|
||||
AppSpecificCheckBox.Checked += AppSpecificCheckBox_Changed;
|
||||
AppSpecificCheckBox.Unchecked += AppSpecificCheckBox_Changed;
|
||||
|
||||
// Wire template picker selection/validity changes so validation re-runs when the user
|
||||
// picks a template or edits its parameters. Guarded against duplicate subscriptions.
|
||||
if (TemplatePicker != null)
|
||||
{
|
||||
TemplatePicker.SelectionChanged -= TemplatePicker_SelectionChanged;
|
||||
TemplatePicker.SelectionChanged += TemplatePicker_SelectionChanged;
|
||||
}
|
||||
|
||||
// Activate keyboard hook for the trigger input
|
||||
if (TriggerKeyToggleBtn.IsChecked == true)
|
||||
{
|
||||
_currentInputMode = KeyInputMode.OriginalKeys;
|
||||
KeyboardHookHelper.Instance.ActivateHook(this);
|
||||
}
|
||||
|
||||
// Initialize the action button label here (not in the constructor) so the flyout
|
||||
// items' localized Text is guaranteed populated before it is read.
|
||||
UpdateActionButtonContent(_currentActionTag);
|
||||
}
|
||||
|
||||
private void UnifiedMappingControl_Unloaded(object sender, RoutedEventArgs e)
|
||||
{
|
||||
// Detach handlers wired in UserControl_Loaded so they don't accumulate across reopens.
|
||||
AppSpecificCheckBox.Checked -= AppSpecificCheckBox_Changed;
|
||||
AppSpecificCheckBox.Unchecked -= AppSpecificCheckBox_Changed;
|
||||
if (TemplatePicker != null)
|
||||
{
|
||||
TemplatePicker.SelectionChanged -= TemplatePicker_SelectionChanged;
|
||||
}
|
||||
|
||||
Reset();
|
||||
CleanupKeyboardHook();
|
||||
}
|
||||
|
||||
private void TemplatePicker_SelectionChanged(object? sender, EventArgs e) => RaiseValidationStateChanged();
|
||||
|
||||
#endregion
|
||||
|
||||
#region Trigger Type Handling
|
||||
@@ -242,29 +268,128 @@ namespace KeyboardManagerEditorUI.Controls
|
||||
|
||||
#region Action Type Handling
|
||||
|
||||
private void ActionTypeComboBox_SelectionChanged(object sender, SelectionChangedEventArgs e)
|
||||
private void OnActionTypeClick(object sender, RoutedEventArgs e)
|
||||
{
|
||||
if (ActionTypeComboBox?.SelectedItem is ComboBoxItem item)
|
||||
if (sender is FrameworkElement fe && fe.Tag is string tag)
|
||||
{
|
||||
string? tag = item.Tag?.ToString();
|
||||
SetActionTag(tag);
|
||||
}
|
||||
}
|
||||
|
||||
// Cleanup keyboard hook when switching away from key/shortcut
|
||||
if (tag != "KeyOrShortcut")
|
||||
private void OnCommandClick(object sender, RoutedEventArgs e)
|
||||
{
|
||||
if (sender is MenuFlyoutItem item && item.Tag is string templateId)
|
||||
{
|
||||
// Select the template first so the action-type switch validates against the picked
|
||||
// command (not the previously-selected one), avoiding a transient stale Save state.
|
||||
TemplatePicker?.SelectCommand(templateId);
|
||||
SetActionTag("RunTemplate");
|
||||
UpdateActionButtonContent("RunTemplate", item.Text);
|
||||
}
|
||||
}
|
||||
|
||||
private void SetActionTag(string tag)
|
||||
{
|
||||
_currentActionTag = tag;
|
||||
|
||||
if (ActionSwitchPresenter != null)
|
||||
{
|
||||
ActionSwitchPresenter.Value = tag;
|
||||
}
|
||||
|
||||
// Cleanup keyboard hook when switching away from key/shortcut
|
||||
if (tag != "KeyOrShortcut")
|
||||
{
|
||||
if (_currentInputMode == KeyInputMode.RemappedKeys)
|
||||
{
|
||||
if (_currentInputMode == KeyInputMode.RemappedKeys)
|
||||
{
|
||||
CleanupKeyboardHook();
|
||||
}
|
||||
CleanupKeyboardHook();
|
||||
}
|
||||
|
||||
if (ActionKeyToggleBtn?.IsChecked == true)
|
||||
{
|
||||
ActionKeyToggleBtn.IsChecked = false;
|
||||
}
|
||||
if (ActionKeyToggleBtn?.IsChecked == true)
|
||||
{
|
||||
ActionKeyToggleBtn.IsChecked = false;
|
||||
}
|
||||
}
|
||||
|
||||
HideValidationMessage();
|
||||
RaiseValidationStateChanged();
|
||||
UpdateActionButtonContent(tag);
|
||||
}
|
||||
|
||||
private void UpdateActionButtonContent(string tag, string? commandDisplay = null)
|
||||
{
|
||||
if (ActionTypeButton == null)
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
string cmd = commandDisplay ?? TemplatePicker?.CurrentCommandDisplay ?? string.Empty;
|
||||
|
||||
string text = tag switch
|
||||
{
|
||||
"Text" => ActionItem_Text.Text,
|
||||
"OpenUrl" => ActionItem_OpenUrl.Text,
|
||||
"OpenApp" => ActionItem_OpenApp.Text,
|
||||
"Disable" => ActionItem_Disable.Text,
|
||||
"RunTemplate" => string.IsNullOrEmpty(cmd)
|
||||
? RunPtCommandSubItem.Text
|
||||
: $"{RunPtCommandSubItem.Text}: {cmd}",
|
||||
_ => ActionItem_KeyOrShortcut.Text,
|
||||
};
|
||||
|
||||
ActionTypeButton.Content = text;
|
||||
}
|
||||
|
||||
private void BuildRunPtCommandMenu()
|
||||
{
|
||||
// A malformed/missing catalog must not crash construction of the whole mapping editor.
|
||||
// Degrade gracefully: log and leave the "Run PowerToys command" submenu empty/disabled.
|
||||
try
|
||||
{
|
||||
foreach (var module in CommandTemplateCatalog.Instance.Data.Modules)
|
||||
{
|
||||
var moduleSub = new MenuFlyoutSubItem
|
||||
{
|
||||
Text = ResourceHelper.GetString(module.DisplayResourceKey),
|
||||
};
|
||||
|
||||
if (!string.IsNullOrEmpty(module.IconGlyph))
|
||||
{
|
||||
moduleSub.Icon = new FontIcon { Glyph = module.IconGlyph };
|
||||
}
|
||||
|
||||
foreach (var cmd in module.Commands)
|
||||
{
|
||||
var item = new MenuFlyoutItem
|
||||
{
|
||||
Text = ResourceHelper.GetString(cmd.DisplayResourceKey),
|
||||
Tag = cmd.Id,
|
||||
};
|
||||
item.Click += OnCommandClick;
|
||||
moduleSub.Items.Add(item);
|
||||
}
|
||||
|
||||
RunPtCommandSubItem.Items.Add(moduleSub);
|
||||
}
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
ManagedCommon.Logger.LogError($"Failed to build PowerToys command template menu: {ex.Message}");
|
||||
RunPtCommandSubItem.IsEnabled = false;
|
||||
}
|
||||
}
|
||||
|
||||
private void TemplatePicker_MissingTemplateKeepRequested(object sender, EventArgs e)
|
||||
{
|
||||
// The user chose to keep the resolved command but discard the template association.
|
||||
// Switch to OpenApp first so the program-path inputs are realized, then populate them
|
||||
// with the previously-resolved command so the mapping is preserved (not blanked).
|
||||
SetActionType(ActionType.OpenApp);
|
||||
SetProgramPath(_templateFallbackProgramPath);
|
||||
SetProgramArgs(_templateFallbackProgramArgs);
|
||||
|
||||
// Clear the picker so it doesn't retain stale template state.
|
||||
TemplatePicker?.Reset();
|
||||
}
|
||||
|
||||
private void ActionKeyToggleBtn_Checked(object sender, RoutedEventArgs e)
|
||||
@@ -798,6 +923,29 @@ namespace KeyboardManagerEditorUI.Controls
|
||||
/// </summary>
|
||||
public ProgramAlreadyRunningAction GetIfRunningAction() => (ProgramAlreadyRunningAction)(IfRunningComboBox?.SelectedIndex ?? 0);
|
||||
|
||||
/// <summary>
|
||||
/// Gets the resolved template executable (for RunTemplate action type).
|
||||
/// Returns null if no template is selected or the template cannot be resolved.
|
||||
/// </summary>
|
||||
public string? GetResolvedTemplateExecutable() => TemplatePicker?.ResolveCurrent()?.Executable;
|
||||
|
||||
/// <summary>
|
||||
/// Gets the resolved template arguments (for RunTemplate action type).
|
||||
/// Returns null if no template is selected or the template cannot be resolved.
|
||||
/// </summary>
|
||||
public string? GetResolvedTemplateArgs() => TemplatePicker?.ResolveCurrent()?.Args;
|
||||
|
||||
/// <summary>
|
||||
/// Gets the template id of the currently selected template (for RunTemplate action type).
|
||||
/// Returns null if no template is selected.
|
||||
/// </summary>
|
||||
public string? GetCurrentTemplateId() => TemplatePicker?.CurrentTemplateId;
|
||||
|
||||
/// <summary>
|
||||
/// Gets the current template parameter values (for RunTemplate action type).
|
||||
/// </summary>
|
||||
public Dictionary<string, string> GetCurrentTemplateParameterValues() => TemplatePicker?.CurrentParameterValues ?? new Dictionary<string, string>();
|
||||
|
||||
#endregion
|
||||
|
||||
#region Public API - Validation
|
||||
@@ -820,6 +968,7 @@ namespace KeyboardManagerEditorUI.Controls
|
||||
ActionType.OpenUrl => !string.IsNullOrWhiteSpace(UrlPathInput?.Text),
|
||||
ActionType.OpenApp => !string.IsNullOrWhiteSpace(ProgramPathInput?.Text),
|
||||
ActionType.Disable => true,
|
||||
ActionType.RunTemplate => TemplatePicker?.IsTemplateInputValid == true,
|
||||
_ => false,
|
||||
};
|
||||
}
|
||||
@@ -865,11 +1014,6 @@ namespace KeyboardManagerEditorUI.Controls
|
||||
/// </summary>
|
||||
public void SetActionType(ActionType actionType)
|
||||
{
|
||||
if (ActionTypeComboBox == null)
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
string tag = actionType switch
|
||||
{
|
||||
ActionType.Text => "Text",
|
||||
@@ -877,17 +1021,11 @@ namespace KeyboardManagerEditorUI.Controls
|
||||
ActionType.OpenApp => "OpenApp",
|
||||
ActionType.Disable => "Disable",
|
||||
ActionType.MouseClick => "MouseClick",
|
||||
ActionType.RunTemplate => "RunTemplate",
|
||||
_ => "KeyOrShortcut",
|
||||
};
|
||||
|
||||
foreach (var item in ActionTypeComboBox.Items)
|
||||
{
|
||||
if (item is ComboBoxItem comboBoxItem && comboBoxItem.Tag is string itemTag && itemTag == tag)
|
||||
{
|
||||
ActionTypeComboBox.SelectedItem = comboBoxItem;
|
||||
return;
|
||||
}
|
||||
}
|
||||
SetActionTag(tag);
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
@@ -978,6 +1116,26 @@ namespace KeyboardManagerEditorUI.Controls
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Loads an existing template mapping into the picker (for RunTemplate action type).
|
||||
/// Switches the action type to RunTemplate and calls LoadExisting on the picker.
|
||||
/// </summary>
|
||||
public void SetRunTemplate(
|
||||
string templateId,
|
||||
IReadOnlyDictionary<string, string>? parameterValues,
|
||||
string fallbackProgramPath = "",
|
||||
string fallbackProgramArgs = "")
|
||||
{
|
||||
// Remember the already-resolved command so "Keep as plain command" can preserve it
|
||||
// if the stored templateId is no longer in the catalog.
|
||||
_templateFallbackProgramPath = fallbackProgramPath ?? string.Empty;
|
||||
_templateFallbackProgramArgs = fallbackProgramArgs ?? string.Empty;
|
||||
|
||||
SetActionType(ActionType.RunTemplate);
|
||||
TemplatePicker?.LoadExisting(templateId, parameterValues);
|
||||
UpdateActionButtonContent("RunTemplate");
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Sets whether the mapping is app-specific.
|
||||
/// </summary>
|
||||
@@ -1121,10 +1279,7 @@ namespace KeyboardManagerEditorUI.Controls
|
||||
TriggerTypeComboBox.SelectedIndex = 0;
|
||||
}
|
||||
|
||||
if (ActionTypeComboBox != null)
|
||||
{
|
||||
ActionTypeComboBox.SelectedIndex = 0;
|
||||
}
|
||||
SetActionTag("KeyOrShortcut");
|
||||
|
||||
if (MouseTriggerComboBox != null)
|
||||
{
|
||||
@@ -1186,6 +1341,13 @@ namespace KeyboardManagerEditorUI.Controls
|
||||
VisibilityComboBox.SelectedIndex = 0;
|
||||
}
|
||||
|
||||
// Reset template picker
|
||||
TemplatePicker?.Reset();
|
||||
|
||||
// Clear the cached missing-template fallback command so it can't leak into the next open.
|
||||
_templateFallbackProgramPath = string.Empty;
|
||||
_templateFallbackProgramArgs = string.Empty;
|
||||
|
||||
HideValidationMessage();
|
||||
}
|
||||
|
||||
|
||||
@@ -83,7 +83,9 @@ namespace KeyboardManagerEditorUI.Interop
|
||||
[MarshalAs(UnmanagedType.LPWStr)] string? startDirectory = null,
|
||||
int elevation = 0,
|
||||
int ifRunningAction = 0,
|
||||
int visibility = 0);
|
||||
int visibility = 0,
|
||||
[MarshalAs(UnmanagedType.LPWStr)] string? templateId = null,
|
||||
[MarshalAs(UnmanagedType.LPWStr)] string? templateParametersJson = null);
|
||||
|
||||
// Delete Mapping Functions
|
||||
[DllImport(DllName, CallingConvention = Convention)]
|
||||
@@ -165,6 +167,8 @@ namespace KeyboardManagerEditorUI.Interop
|
||||
public IntPtr ProgramPath;
|
||||
public IntPtr ProgramArgs;
|
||||
public IntPtr UriToOpen;
|
||||
public IntPtr TemplateId;
|
||||
public IntPtr TemplateParametersJson;
|
||||
}
|
||||
|
||||
[StructLayout(LayoutKind.Sequential, CharSet = CharSet.Unicode)]
|
||||
|
||||
@@ -7,7 +7,9 @@ using System.Collections.Generic;
|
||||
using System.Linq;
|
||||
using System.Runtime.InteropServices;
|
||||
using System.Text;
|
||||
using System.Text.Json;
|
||||
using System.Threading.Tasks;
|
||||
using KeyboardManagerEditorUI.Templates;
|
||||
using ManagedCommon;
|
||||
|
||||
namespace KeyboardManagerEditorUI.Interop
|
||||
@@ -61,6 +63,7 @@ namespace KeyboardManagerEditorUI.Interop
|
||||
var mapping = default(ShortcutMapping);
|
||||
if (KeyboardManagerInterop.GetShortcutRemap(_configHandle, i, ref mapping))
|
||||
{
|
||||
var (templateId, templateParameters) = ReadTemplateFields(ref mapping);
|
||||
result.Add(new ShortcutKeyMapping
|
||||
{
|
||||
OriginalKeys = KeyboardManagerInterop.GetStringAndFree(mapping.OriginalKeys),
|
||||
@@ -71,6 +74,8 @@ namespace KeyboardManagerEditorUI.Interop
|
||||
ProgramPath = KeyboardManagerInterop.GetStringAndFree(mapping.ProgramPath),
|
||||
ProgramArgs = KeyboardManagerInterop.GetStringAndFree(mapping.ProgramArgs),
|
||||
UriToOpen = KeyboardManagerInterop.GetStringAndFree(mapping.UriToOpen),
|
||||
TemplateId = templateId,
|
||||
TemplateParameters = templateParameters,
|
||||
});
|
||||
}
|
||||
}
|
||||
@@ -78,6 +83,33 @@ namespace KeyboardManagerEditorUI.Interop
|
||||
return result;
|
||||
}
|
||||
|
||||
// Reads (and frees) the template metadata strings returned by the native layer.
|
||||
// Must be called for every mapping so the native-allocated strings are always freed.
|
||||
private static (string? TemplateId, Dictionary<string, string>? TemplateParameters) ReadTemplateFields(ref ShortcutMapping mapping)
|
||||
{
|
||||
string templateId = KeyboardManagerInterop.GetStringAndFree(mapping.TemplateId);
|
||||
string templateParametersJson = KeyboardManagerInterop.GetStringAndFree(mapping.TemplateParametersJson);
|
||||
|
||||
Dictionary<string, string>? parameters = null;
|
||||
if (!string.IsNullOrEmpty(templateParametersJson))
|
||||
{
|
||||
try
|
||||
{
|
||||
parameters = JsonSerializer.Deserialize(
|
||||
templateParametersJson,
|
||||
CommandTemplateJsonContext.Default.DictionaryStringString);
|
||||
}
|
||||
catch (Exception)
|
||||
{
|
||||
// Malformed on-disk metadata must never break config load — the two template
|
||||
// strings are already freed above; swallow so the caller still frees the rest.
|
||||
parameters = null;
|
||||
}
|
||||
}
|
||||
|
||||
return (string.IsNullOrEmpty(templateId) ? null : templateId, parameters);
|
||||
}
|
||||
|
||||
public List<ShortcutKeyMapping> GetShortcutMappingsByType(ShortcutOperationType operationType)
|
||||
{
|
||||
var result = new List<ShortcutKeyMapping>();
|
||||
@@ -88,6 +120,7 @@ namespace KeyboardManagerEditorUI.Interop
|
||||
var mapping = default(ShortcutMapping);
|
||||
if (KeyboardManagerInterop.GetShortcutRemapByType(_configHandle, (int)operationType, i, ref mapping))
|
||||
{
|
||||
var (templateId, templateParameters) = ReadTemplateFields(ref mapping);
|
||||
result.Add(new ShortcutKeyMapping
|
||||
{
|
||||
OriginalKeys = KeyboardManagerInterop.GetStringAndFree(mapping.OriginalKeys),
|
||||
@@ -98,6 +131,8 @@ namespace KeyboardManagerEditorUI.Interop
|
||||
ProgramPath = KeyboardManagerInterop.GetStringAndFree(mapping.ProgramPath),
|
||||
ProgramArgs = KeyboardManagerInterop.GetStringAndFree(mapping.ProgramArgs),
|
||||
UriToOpen = KeyboardManagerInterop.GetStringAndFree(mapping.UriToOpen),
|
||||
TemplateId = templateId,
|
||||
TemplateParameters = templateParameters,
|
||||
});
|
||||
}
|
||||
}
|
||||
@@ -221,6 +256,14 @@ namespace KeyboardManagerEditorUI.Interop
|
||||
|
||||
if (shortcutKeyMapping.OperationType == ShortcutOperationType.RunProgram)
|
||||
{
|
||||
string? templateParametersJson = null;
|
||||
if (shortcutKeyMapping.TemplateParameters is not null && shortcutKeyMapping.TemplateParameters.Count > 0)
|
||||
{
|
||||
templateParametersJson = JsonSerializer.Serialize(
|
||||
shortcutKeyMapping.TemplateParameters,
|
||||
CommandTemplateJsonContext.Default.DictionaryStringString);
|
||||
}
|
||||
|
||||
return KeyboardManagerInterop.AddShortcutRemap(
|
||||
_configHandle,
|
||||
shortcutKeyMapping.OriginalKeys,
|
||||
@@ -232,7 +275,9 @@ namespace KeyboardManagerEditorUI.Interop
|
||||
string.IsNullOrEmpty(shortcutKeyMapping.StartInDirectory) ? null : shortcutKeyMapping.StartInDirectory,
|
||||
(int)shortcutKeyMapping.Elevation,
|
||||
(int)shortcutKeyMapping.IfRunningAction,
|
||||
(int)shortcutKeyMapping.Visibility);
|
||||
(int)shortcutKeyMapping.Visibility,
|
||||
shortcutKeyMapping.TemplateId,
|
||||
templateParametersJson);
|
||||
}
|
||||
else if (shortcutKeyMapping.OperationType == ShortcutOperationType.OpenUri)
|
||||
{
|
||||
|
||||
@@ -36,6 +36,18 @@ namespace KeyboardManagerEditorUI.Interop
|
||||
|
||||
public string UriToOpen { get; set; } = string.Empty;
|
||||
|
||||
/// <summary>
|
||||
/// When non-null, indicates the mapping was created from a command template.
|
||||
/// Used to re-open the template picker on edit.
|
||||
/// </summary>
|
||||
public string? TemplateId { get; set; }
|
||||
|
||||
/// <summary>
|
||||
/// Parameter values captured at save time for a template-based mapping.
|
||||
/// Null when <see cref="TemplateId"/> is null.
|
||||
/// </summary>
|
||||
public Dictionary<string, string>? TemplateParameters { get; set; }
|
||||
|
||||
public enum ElevationLevel
|
||||
{
|
||||
NonElevated = 0,
|
||||
|
||||
@@ -76,6 +76,11 @@
|
||||
<CopyToOutputDirectory>Always</CopyToOutputDirectory>
|
||||
</Content>
|
||||
</ItemGroup>
|
||||
<ItemGroup>
|
||||
<EmbeddedResource Include="Templates\powertoyscli.json">
|
||||
<LogicalName>KeyboardManagerEditorUI.Templates.powertoyscli.json</LogicalName>
|
||||
</EmbeddedResource>
|
||||
</ItemGroup>
|
||||
<ItemGroup>
|
||||
<Page Update="Styles\Colors.xaml">
|
||||
<Generator>MSBuild:Compile</Generator>
|
||||
|
||||
@@ -260,13 +260,16 @@ namespace KeyboardManagerEditorUI.Pages
|
||||
|
||||
UnifiedMappingControl.Reset();
|
||||
UnifiedMappingControl.SetTriggerKeys(programShortcut.Shortcut.ToList());
|
||||
UnifiedMappingControl.SetActionType(UnifiedMappingControl.ActionType.OpenApp);
|
||||
UnifiedMappingControl.SetProgramPath(programShortcut.AppToRun);
|
||||
UnifiedMappingControl.SetProgramArgs(programShortcut.Args);
|
||||
|
||||
// Check if this program shortcut was originally created from a command template.
|
||||
string? templateId = null;
|
||||
Dictionary<string, string>? templateParameters = null;
|
||||
if (!string.IsNullOrEmpty(programShortcut.Id) &&
|
||||
SettingsManager.EditorSettings.ShortcutSettingsDictionary.TryGetValue(programShortcut.Id, out var settings))
|
||||
{
|
||||
templateId = settings.Shortcut.TemplateId;
|
||||
templateParameters = settings.Shortcut.TemplateParameters;
|
||||
|
||||
var mapping = settings.Shortcut;
|
||||
UnifiedMappingControl.SetStartInDirectory(mapping.StartInDirectory);
|
||||
UnifiedMappingControl.SetElevationLevel(mapping.Elevation);
|
||||
@@ -274,6 +277,20 @@ namespace KeyboardManagerEditorUI.Pages
|
||||
UnifiedMappingControl.SetIfRunningAction(mapping.IfRunningAction);
|
||||
}
|
||||
|
||||
if (!string.IsNullOrEmpty(templateId))
|
||||
{
|
||||
// Restore as RunTemplate: the picker handles missing-template degradation internally.
|
||||
// Pass the resolved command so "Keep as plain command" can preserve it if the
|
||||
// stored templateId is no longer present in the catalog.
|
||||
UnifiedMappingControl.SetRunTemplate(templateId, templateParameters, programShortcut.AppToRun, programShortcut.Args);
|
||||
}
|
||||
else
|
||||
{
|
||||
UnifiedMappingControl.SetActionType(UnifiedMappingControl.ActionType.OpenApp);
|
||||
UnifiedMappingControl.SetProgramPath(programShortcut.AppToRun);
|
||||
UnifiedMappingControl.SetProgramArgs(programShortcut.Args);
|
||||
}
|
||||
|
||||
UnifiedMappingControl.SetAppSpecific(!programShortcut.IsAllApps, programShortcut.AppName);
|
||||
RemappingDialog.Title = ResourceHelper.GetString("RemappingDialog_TitleEdit");
|
||||
await ShowRemappingDialog();
|
||||
@@ -394,6 +411,7 @@ namespace KeyboardManagerEditorUI.Pages
|
||||
UnifiedMappingControl.ActionType.OpenUrl => SaveUrlMapping(triggerKeys),
|
||||
UnifiedMappingControl.ActionType.OpenApp => SaveProgramMapping(triggerKeys),
|
||||
UnifiedMappingControl.ActionType.Disable => SaveDisableMapping(triggerKeys),
|
||||
UnifiedMappingControl.ActionType.RunTemplate => SaveRunTemplateMapping(triggerKeys),
|
||||
UnifiedMappingControl.ActionType.MouseClick => throw new NotImplementedException("Mouse click remapping is not yet supported."),
|
||||
_ => false,
|
||||
};
|
||||
@@ -439,6 +457,8 @@ namespace KeyboardManagerEditorUI.Pages
|
||||
triggerKeys, UnifiedMappingControl.GetProgramPath(), isAppSpecific, appName, _mappingService!, _isEditMode),
|
||||
UnifiedMappingControl.ActionType.Disable => ValidationHelper.ValidateDisableMapping(
|
||||
triggerKeys, isAppSpecific, appName, _mappingService!, _isEditMode, editingRemapping),
|
||||
UnifiedMappingControl.ActionType.RunTemplate => ValidationHelper.ValidateAppMapping(
|
||||
triggerKeys, UnifiedMappingControl.GetResolvedTemplateExecutable() ?? string.Empty, isAppSpecific, appName, _mappingService!, _isEditMode),
|
||||
_ => ValidationErrorType.NoError,
|
||||
};
|
||||
}
|
||||
@@ -683,6 +703,51 @@ namespace KeyboardManagerEditorUI.Pages
|
||||
return saved;
|
||||
}
|
||||
|
||||
private bool SaveRunTemplateMapping(List<string> triggerKeys)
|
||||
{
|
||||
string? programPath = UnifiedMappingControl.GetResolvedTemplateExecutable();
|
||||
if (string.IsNullOrEmpty(programPath))
|
||||
{
|
||||
// No template selected or unresolvable — validation should have caught this.
|
||||
return false;
|
||||
}
|
||||
|
||||
// Retarget the template's default per-user path to a machine-wide install when needed,
|
||||
// so the saved mapping launches regardless of how PowerToys was installed.
|
||||
programPath = Templates.PowerToysInstallResolver.ResolveExecutable(programPath);
|
||||
|
||||
string originalKeysString = string.Join(";", triggerKeys.Select(k => _mappingService!.GetKeyCodeFromName(k).ToString(CultureInfo.InvariantCulture)));
|
||||
|
||||
// Preserve the run-options that were loaded onto the control when editing an existing
|
||||
// template mapping so re-saving does not silently reset them to defaults.
|
||||
var templateParameters = UnifiedMappingControl.GetCurrentTemplateParameterValues();
|
||||
|
||||
var shortcutKeyMapping = new ShortcutKeyMapping
|
||||
{
|
||||
OperationType = ShortcutOperationType.RunProgram,
|
||||
OriginalKeys = originalKeysString,
|
||||
TargetKeys = originalKeysString,
|
||||
ProgramPath = programPath,
|
||||
ProgramArgs = UnifiedMappingControl.GetResolvedTemplateArgs() ?? string.Empty,
|
||||
StartInDirectory = UnifiedMappingControl.GetStartInDirectory(),
|
||||
IfRunningAction = UnifiedMappingControl.GetIfRunningAction(),
|
||||
Visibility = UnifiedMappingControl.GetVisibility(),
|
||||
Elevation = UnifiedMappingControl.GetElevationLevel(),
|
||||
TargetApp = UnifiedMappingControl.GetIsAppSpecific() ? UnifiedMappingControl.GetAppName() : string.Empty,
|
||||
TemplateId = UnifiedMappingControl.GetCurrentTemplateId(),
|
||||
TemplateParameters = templateParameters.Count > 0 ? templateParameters : null,
|
||||
};
|
||||
|
||||
bool saved = _mappingService!.AddShortcutMapping(shortcutKeyMapping);
|
||||
if (saved)
|
||||
{
|
||||
_mappingService.SaveSettings();
|
||||
SettingsManager.AddShortcutKeyMappingToSettings(shortcutKeyMapping);
|
||||
}
|
||||
|
||||
return saved;
|
||||
}
|
||||
|
||||
#endregion
|
||||
|
||||
#region Delete Handlers
|
||||
|
||||
@@ -501,4 +501,52 @@
|
||||
<data name="CheckServiceBtn.Content" xml:space="preserve">
|
||||
<value>Check service status</value>
|
||||
</data>
|
||||
<data name="ActionType_RunTemplate_Text.Text" xml:space="preserve">
|
||||
<value>Run PowerToys Command</value>
|
||||
</data>
|
||||
<data name="TemplatePreviewLabel.Text" xml:space="preserve">
|
||||
<value>Preview</value>
|
||||
</data>
|
||||
<data name="MissingTemplateInfoBar.Title" xml:space="preserve">
|
||||
<value>Template no longer available</value>
|
||||
</data>
|
||||
<data name="MissingTemplateInfoBar.Message" xml:space="preserve">
|
||||
<value>The template originally used for this mapping is no longer in the catalog.</value>
|
||||
</data>
|
||||
<data name="TemplateMissingKeepButton.Content" xml:space="preserve">
|
||||
<value>Keep as plain command</value>
|
||||
</data>
|
||||
<data name="TemplateModule_Settings" xml:space="preserve">
|
||||
<value>Settings</value>
|
||||
</data>
|
||||
<data name="TemplateCmd_Settings_OpenMain" xml:space="preserve">
|
||||
<value>Open Settings</value>
|
||||
</data>
|
||||
<data name="TemplateCmd_Settings_OpenModule" xml:space="preserve">
|
||||
<value>Open Settings for module...</value>
|
||||
</data>
|
||||
<data name="TemplateParam_Module" xml:space="preserve">
|
||||
<value>Module</value>
|
||||
</data>
|
||||
<data name="Module_ColorPicker" xml:space="preserve">
|
||||
<value>Color Picker</value>
|
||||
</data>
|
||||
<data name="Module_FancyZones" xml:space="preserve">
|
||||
<value>FancyZones</value>
|
||||
</data>
|
||||
<data name="Module_KeyboardManager" xml:space="preserve">
|
||||
<value>Keyboard Manager</value>
|
||||
</data>
|
||||
<data name="Module_PowerLauncher" xml:space="preserve">
|
||||
<value>PowerToys Run</value>
|
||||
</data>
|
||||
<data name="Module_Hosts" xml:space="preserve">
|
||||
<value>Hosts File Editor</value>
|
||||
</data>
|
||||
<data name="Module_RegistryPreview" xml:space="preserve">
|
||||
<value>Registry Preview</value>
|
||||
</data>
|
||||
<data name="Module_ZoomIt" xml:space="preserve">
|
||||
<value>ZoomIt</value>
|
||||
</data>
|
||||
</root>
|
||||
@@ -0,0 +1,21 @@
|
||||
// Copyright (c) Microsoft Corporation
|
||||
// The Microsoft Corporation licenses this file to you under the MIT license.
|
||||
// See the LICENSE file in the project root for more information.
|
||||
|
||||
using System.Collections.Generic;
|
||||
|
||||
namespace KeyboardManagerEditorUI.Templates
|
||||
{
|
||||
public sealed class CommandTemplate
|
||||
{
|
||||
public string Id { get; init; } = string.Empty;
|
||||
|
||||
public string DisplayResourceKey { get; init; } = string.Empty;
|
||||
|
||||
public string Executable { get; init; } = string.Empty;
|
||||
|
||||
public string ArgsTemplate { get; init; } = string.Empty;
|
||||
|
||||
public List<TemplateParameter> Parameters { get; init; } = new();
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,65 @@
|
||||
// Copyright (c) Microsoft Corporation
|
||||
// The Microsoft Corporation licenses this file to you under the MIT license.
|
||||
// See the LICENSE file in the project root for more information.
|
||||
|
||||
using System;
|
||||
using System.Text.Json;
|
||||
|
||||
namespace KeyboardManagerEditorUI.Templates
|
||||
{
|
||||
public sealed class CommandTemplateCatalog
|
||||
{
|
||||
private const string ResourceName = "KeyboardManagerEditorUI.Templates.powertoyscli.json";
|
||||
private const int SupportedSchemaVersion = 1;
|
||||
|
||||
private static readonly Lazy<CommandTemplateCatalog> _instance = new(() => Load());
|
||||
|
||||
public static CommandTemplateCatalog Instance => _instance.Value;
|
||||
|
||||
public PowerToysCliCatalog Data { get; }
|
||||
|
||||
private CommandTemplateCatalog(PowerToysCliCatalog data)
|
||||
{
|
||||
Data = data;
|
||||
}
|
||||
|
||||
private static CommandTemplateCatalog Load()
|
||||
{
|
||||
var assembly = typeof(CommandTemplateCatalog).Assembly;
|
||||
using var stream = assembly.GetManifestResourceStream(ResourceName)
|
||||
?? throw new InvalidOperationException(
|
||||
$"Embedded resource '{ResourceName}' not found. " +
|
||||
"Check KeyboardManagerEditorUI.csproj <EmbeddedResource> entry.");
|
||||
|
||||
var data = JsonSerializer.Deserialize(
|
||||
stream,
|
||||
CommandTemplateJsonContext.Default.PowerToysCliCatalog)
|
||||
?? throw new InvalidOperationException(
|
||||
$"Failed to deserialize '{ResourceName}' — JsonSerializer returned null.");
|
||||
|
||||
if (data.SchemaVersion < 1)
|
||||
{
|
||||
throw new InvalidOperationException(
|
||||
$"Invalid powertoyscli.json schemaVersion={data.SchemaVersion}; " +
|
||||
$"expected >= 1.");
|
||||
}
|
||||
|
||||
if (data.SchemaVersion > SupportedSchemaVersion)
|
||||
{
|
||||
// Newer catalogs are read best-effort: unknown additive fields are ignored by
|
||||
// the deserializer, so a forward-compatible bump must not break older binaries.
|
||||
ManagedCommon.Logger.LogWarning(
|
||||
$"powertoyscli.json schemaVersion={data.SchemaVersion} is newer than " +
|
||||
$"supported {SupportedSchemaVersion}; reading best-effort.");
|
||||
}
|
||||
|
||||
if (data.Modules.Count == 0)
|
||||
{
|
||||
throw new InvalidOperationException(
|
||||
"powertoyscli.json has zero modules — at least one module is required.");
|
||||
}
|
||||
|
||||
return new CommandTemplateCatalog(data);
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,20 @@
|
||||
// Copyright (c) Microsoft Corporation
|
||||
// The Microsoft Corporation licenses this file to you under the MIT license.
|
||||
// See the LICENSE file in the project root for more information.
|
||||
|
||||
using System.Collections.Generic;
|
||||
using System.Text.Json;
|
||||
using System.Text.Json.Serialization;
|
||||
|
||||
namespace KeyboardManagerEditorUI.Templates
|
||||
{
|
||||
[JsonSerializable(typeof(PowerToysCliCatalog))]
|
||||
[JsonSerializable(typeof(Dictionary<string, string>))]
|
||||
[JsonSourceGenerationOptions(
|
||||
PropertyNamingPolicy = JsonKnownNamingPolicy.CamelCase,
|
||||
WriteIndented = false,
|
||||
ReadCommentHandling = JsonCommentHandling.Skip)]
|
||||
internal sealed partial class CommandTemplateJsonContext : JsonSerializerContext
|
||||
{
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,19 @@
|
||||
// Copyright (c) Microsoft Corporation
|
||||
// The Microsoft Corporation licenses this file to you under the MIT license.
|
||||
// See the LICENSE file in the project root for more information.
|
||||
|
||||
using System.Collections.Generic;
|
||||
|
||||
namespace KeyboardManagerEditorUI.Templates
|
||||
{
|
||||
public sealed class CommandTemplateModule
|
||||
{
|
||||
public string Id { get; init; } = string.Empty;
|
||||
|
||||
public string DisplayResourceKey { get; init; } = string.Empty;
|
||||
|
||||
public string? IconGlyph { get; init; }
|
||||
|
||||
public List<CommandTemplate> Commands { get; init; } = new();
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,15 @@
|
||||
// Copyright (c) Microsoft Corporation
|
||||
// The Microsoft Corporation licenses this file to you under the MIT license.
|
||||
// See the LICENSE file in the project root for more information.
|
||||
|
||||
using System.Collections.Generic;
|
||||
|
||||
namespace KeyboardManagerEditorUI.Templates
|
||||
{
|
||||
public sealed class PowerToysCliCatalog
|
||||
{
|
||||
public int SchemaVersion { get; init; }
|
||||
|
||||
public List<CommandTemplateModule> Modules { get; init; } = new();
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,65 @@
|
||||
// Copyright (c) Microsoft Corporation
|
||||
// The Microsoft Corporation licenses this file to you under the MIT license.
|
||||
// See the LICENSE file in the project root for more information.
|
||||
|
||||
using System;
|
||||
using System.IO;
|
||||
|
||||
namespace KeyboardManagerEditorUI.Templates
|
||||
{
|
||||
/// <summary>
|
||||
/// Resolves the on-disk location of a PowerToys executable referenced by a command template.
|
||||
/// Templates ship a per-user (<c>%LOCALAPPDATA%</c>) path by default; this retargets to a
|
||||
/// machine-wide (<c>%ProgramFiles%</c>) install when the per-user path is absent so the saved
|
||||
/// mapping works regardless of install scope.
|
||||
/// </summary>
|
||||
internal static class PowerToysInstallResolver
|
||||
{
|
||||
// Candidate install locations, in preference order. Env-var form is preserved in the saved
|
||||
// value so the engine expands it at trigger time (and it survives a reinstall in place).
|
||||
private static readonly string[] Candidates =
|
||||
{
|
||||
@"%LOCALAPPDATA%\PowerToys\PowerToys.exe",
|
||||
|
||||
// The editor is always a 64-bit process (x64/ARM64; no x86 build), so %ProgramFiles%
|
||||
// already resolves to the native 64-bit Program Files where a machine-wide install lives.
|
||||
@"%ProgramFiles%\PowerToys\PowerToys.exe",
|
||||
};
|
||||
|
||||
/// <summary>
|
||||
/// Returns an executable path that exists on disk. If <paramref name="executable"/> already
|
||||
/// resolves to an existing file it is returned unchanged. For a nonexistent
|
||||
/// <c>PowerToys.exe</c> path, known install locations are probed. If none exist the original
|
||||
/// value is returned (the engine will surface a "program not found" error at trigger time).
|
||||
/// </summary>
|
||||
public static string ResolveExecutable(string executable)
|
||||
{
|
||||
if (string.IsNullOrEmpty(executable))
|
||||
{
|
||||
return executable;
|
||||
}
|
||||
|
||||
if (File.Exists(Environment.ExpandEnvironmentVariables(executable)))
|
||||
{
|
||||
return executable;
|
||||
}
|
||||
|
||||
// Only retarget the known PowerToys.exe; leave arbitrary executables untouched.
|
||||
string fileName = Path.GetFileName(Environment.ExpandEnvironmentVariables(executable));
|
||||
if (!string.Equals(fileName, "PowerToys.exe", StringComparison.OrdinalIgnoreCase))
|
||||
{
|
||||
return executable;
|
||||
}
|
||||
|
||||
foreach (var candidate in Candidates)
|
||||
{
|
||||
if (File.Exists(Environment.ExpandEnvironmentVariables(candidate)))
|
||||
{
|
||||
return candidate;
|
||||
}
|
||||
}
|
||||
|
||||
return executable;
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,13 @@
|
||||
// Copyright (c) Microsoft Corporation
|
||||
// The Microsoft Corporation licenses this file to you under the MIT license.
|
||||
// See the LICENSE file in the project root for more information.
|
||||
|
||||
namespace KeyboardManagerEditorUI.Templates
|
||||
{
|
||||
public sealed class TemplateChoice
|
||||
{
|
||||
public string Value { get; init; } = string.Empty;
|
||||
|
||||
public string DisplayResourceKey { get; init; } = string.Empty;
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,21 @@
|
||||
// Copyright (c) Microsoft Corporation
|
||||
// The Microsoft Corporation licenses this file to you under the MIT license.
|
||||
// See the LICENSE file in the project root for more information.
|
||||
|
||||
using System.Collections.Generic;
|
||||
|
||||
namespace KeyboardManagerEditorUI.Templates
|
||||
{
|
||||
public sealed class TemplateParameter
|
||||
{
|
||||
public string Name { get; init; } = string.Empty;
|
||||
|
||||
public string LabelResourceKey { get; init; } = string.Empty;
|
||||
|
||||
public string Type { get; init; } = "Text";
|
||||
|
||||
public bool Required { get; init; } = true;
|
||||
|
||||
public List<TemplateChoice>? Choices { get; init; }
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,93 @@
|
||||
// Copyright (c) Microsoft Corporation
|
||||
// The Microsoft Corporation licenses this file to you under the MIT license.
|
||||
// See the LICENSE file in the project root for more information.
|
||||
|
||||
using System.Collections.Generic;
|
||||
using System.Text;
|
||||
using System.Text.RegularExpressions;
|
||||
|
||||
namespace KeyboardManagerEditorUI.Templates
|
||||
{
|
||||
public static class TemplateResolver
|
||||
{
|
||||
public readonly record struct Resolved(string Executable, string Args);
|
||||
|
||||
// Matches {paramName} placeholders. Resolution is single-pass over the original
|
||||
// template so a substituted value can never be re-interpreted as another placeholder
|
||||
// (prevents substitution-injection once free-text parameters are introduced).
|
||||
private static readonly Regex PlaceholderRegex = new(@"\{(\w+)\}");
|
||||
|
||||
// Characters that force argument quoting per CommandLineToArgvW parsing rules.
|
||||
private static readonly char[] QuoteTriggers = { ' ', '\t', '\n', '\v', '"' };
|
||||
|
||||
public static Resolved Resolve(
|
||||
CommandTemplate template,
|
||||
IReadOnlyDictionary<string, string>? values)
|
||||
{
|
||||
var argsTemplate = template.ArgsTemplate ?? string.Empty;
|
||||
|
||||
// Pre-compute the (quoted-if-needed) replacement for every declared parameter.
|
||||
var substitutions = new Dictionary<string, string>();
|
||||
foreach (var p in template.Parameters)
|
||||
{
|
||||
string raw = string.Empty;
|
||||
if (values is not null && values.TryGetValue(p.Name, out var v))
|
||||
{
|
||||
raw = v ?? string.Empty;
|
||||
}
|
||||
|
||||
substitutions[p.Name] = QuoteArgumentIfNeeded(raw);
|
||||
}
|
||||
|
||||
// Single pass: each {name} is replaced exactly once against the original template.
|
||||
// Unknown placeholders are left untouched (matches prior behavior).
|
||||
string args = PlaceholderRegex.Replace(argsTemplate, m =>
|
||||
substitutions.TryGetValue(m.Groups[1].Value, out var replacement)
|
||||
? replacement
|
||||
: m.Value);
|
||||
|
||||
return new Resolved(template.Executable ?? string.Empty, args);
|
||||
}
|
||||
|
||||
// Quotes a value for a Windows command line (CommandLineToArgvW rules) only when it
|
||||
// contains whitespace or a quote, so simple values (e.g. fixed combo choices) and
|
||||
// empty values pass through unchanged.
|
||||
internal static string QuoteArgumentIfNeeded(string value)
|
||||
{
|
||||
if (value.Length == 0 || value.IndexOfAny(QuoteTriggers) < 0)
|
||||
{
|
||||
return value;
|
||||
}
|
||||
|
||||
var sb = new StringBuilder();
|
||||
sb.Append('"');
|
||||
|
||||
int backslashes = 0;
|
||||
foreach (char c in value)
|
||||
{
|
||||
if (c == '\\')
|
||||
{
|
||||
backslashes++;
|
||||
}
|
||||
else if (c == '"')
|
||||
{
|
||||
// Escape the run of backslashes preceding a quote, then the quote itself.
|
||||
sb.Append('\\', (backslashes * 2) + 1);
|
||||
backslashes = 0;
|
||||
sb.Append('"');
|
||||
}
|
||||
else
|
||||
{
|
||||
sb.Append('\\', backslashes);
|
||||
backslashes = 0;
|
||||
sb.Append(c);
|
||||
}
|
||||
}
|
||||
|
||||
// Escape trailing backslashes so they don't escape the closing quote.
|
||||
sb.Append('\\', backslashes * 2);
|
||||
sb.Append('"');
|
||||
return sb.ToString();
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,42 @@
|
||||
{
|
||||
"schemaVersion": 1,
|
||||
"modules": [
|
||||
{
|
||||
"id": "settings",
|
||||
"displayResourceKey": "TemplateModule_Settings",
|
||||
"iconGlyph": "",
|
||||
"commands": [
|
||||
{
|
||||
"id": "settings.openMain",
|
||||
"displayResourceKey": "TemplateCmd_Settings_OpenMain",
|
||||
"executable": "%LOCALAPPDATA%\\PowerToys\\PowerToys.exe",
|
||||
"argsTemplate": "--open-settings",
|
||||
"parameters": []
|
||||
},
|
||||
{
|
||||
"id": "settings.openModule",
|
||||
"displayResourceKey": "TemplateCmd_Settings_OpenModule",
|
||||
"executable": "%LOCALAPPDATA%\\PowerToys\\PowerToys.exe",
|
||||
"argsTemplate": "--open-settings={module}",
|
||||
"parameters": [
|
||||
{
|
||||
"name": "module",
|
||||
"labelResourceKey": "TemplateParam_Module",
|
||||
"type": "Combo",
|
||||
"required": true,
|
||||
"choices": [
|
||||
{ "value": "ColorPicker", "displayResourceKey": "Module_ColorPicker" },
|
||||
{ "value": "FancyZones", "displayResourceKey": "Module_FancyZones" },
|
||||
{ "value": "KeyboardManager", "displayResourceKey": "Module_KeyboardManager" },
|
||||
{ "value": "PowerLauncher", "displayResourceKey": "Module_PowerLauncher" },
|
||||
{ "value": "Hosts", "displayResourceKey": "Module_Hosts" },
|
||||
{ "value": "RegistryPreview", "displayResourceKey": "Module_RegistryPreview" },
|
||||
{ "value": "ZoomIt", "displayResourceKey": "Module_ZoomIt" }
|
||||
]
|
||||
}
|
||||
]
|
||||
}
|
||||
]
|
||||
}
|
||||
]
|
||||
}
|
||||
Some files were not shown because too many files have changed in this diff Show More
Reference in New Issue
Block a user