| /* |
| * Copyright (c) 2018 The WebRTC project authors. All Rights Reserved. |
| * |
| * Use of this source code is governed by a BSD-style license |
| * that can be found in the LICENSE file in the root of the source |
| * tree. An additional intellectual property rights grant can be found |
| * in the file PATENTS. All contributing project authors may |
| * be found in the AUTHORS file in the root of the source tree. |
| */ |
| |
| #include "modules/audio_processing/agc2/rnn_vad/pitch_search_internal.h" |
| |
| #include <array> |
| #include <tuple> |
| |
| #include "modules/audio_processing/agc2/rnn_vad/test_utils.h" |
| // TODO(bugs.webrtc.org/8948): Add when the issue is fixed. |
| // #include "test/fpe_observer.h" |
| #include "test/gtest.h" |
| |
| namespace webrtc { |
| namespace rnn_vad { |
| namespace test { |
| namespace { |
| |
| // TODO(bugs.webrtc.org/9076): Move to resource file. |
| constexpr std::array<float, kBufSize24kHz> kPitchBufferData = { |
| -35.248100281f, -25.836528778f, 5.682674408f, 2.880297661f, -1.648161888f, |
| -4.094896793f, -3.500580072f, -0.896141529f, -2.989939451f, -4.608089447f, |
| -3.721750736f, -2.290785789f, -3.326566458f, -4.370154381f, -3.221047878f, |
| -4.049056530f, -2.846302271f, -1.805017233f, -1.547624588f, -0.809937477f, |
| -1.446955442f, -3.258146763f, -1.849959373f, 0.005283833f, -0.571619749f, |
| -0.630573988f, -0.162780523f, -2.699024916f, -0.856231451f, 2.748089552f, |
| 2.026614428f, -0.474685907f, -0.571918726f, 1.186420918f, 1.770769954f, |
| 2.017296791f, 1.154794335f, 1.082345366f, 1.954892635f, 2.249727726f, |
| 2.643483400f, 1.857815385f, 0.064472735f, 0.015978813f, 0.301099658f, |
| 0.478950322f, -0.669701457f, -0.654453993f, 1.338572979f, -0.493052602f, |
| -1.763812065f, 0.524392128f, 0.010438919f, -1.726593733f, -2.866710663f, |
| -2.065258503f, -3.010460854f, -3.994765282f, -4.102010250f, -3.135548830f, |
| -2.597487926f, -2.255330563f, -1.002008915f, 0.523116589f, 1.430158496f, |
| -1.655169368f, -2.263641357f, 0.766040802f, 1.166070461f, 0.002490997f, |
| 0.401043415f, -0.158550858f, -0.572042346f, 1.365390539f, -1.397871614f, |
| -2.020734787f, -1.979169965f, -1.025816441f, 0.012545407f, -1.042758584f, |
| -1.206598401f, -1.140330791f, -3.060853720f, -3.530077934f, -1.774474382f, |
| -1.342000484f, -3.171817064f, -2.489153862f, -1.593364000f, -2.552185535f, |
| -2.899760723f, -4.698278427f, -4.123534203f, -2.613421679f, -2.061793327f, |
| -4.113687515f, -3.174087524f, -2.367874622f, -4.523970604f, -4.250762939f, |
| -2.752931118f, -1.547106743f, -4.109455109f, -3.893044472f, -2.348384857f, |
| -3.194510698f, -3.502159357f, -2.785978794f, -1.981978416f, -3.279178143f, |
| -3.007923365f, -1.801304340f, -1.839247227f, -1.003675938f, -0.985928297f, |
| -1.647925615f, -2.166392088f, -1.947163343f, 0.488545895f, 1.567199469f, |
| -1.179960012f, -2.710370064f, -2.613196850f, -3.205850124f, -2.796218395f, |
| -0.715085745f, 1.406243801f, -0.779834270f, -2.075612307f, -0.922246933f, |
| -1.849850416f, 0.979040504f, 3.570628166f, 0.945924520f, -2.821768284f, |
| -6.262358189f, -6.154916763f, -0.567943573f, 2.386518955f, 1.673806906f, |
| -3.676584721f, -7.129202843f, -3.311969519f, 1.126702785f, 3.218248606f, |
| 1.600885630f, -1.709451079f, -6.822564125f, -6.011950970f, -0.671678543f, |
| 1.080205441f, -1.342422366f, -3.589303732f, -3.586701870f, -3.425134897f, |
| -1.078015327f, 2.556719542f, 0.469867468f, 0.139251709f, -0.118916273f, |
| -1.284181952f, 0.941113472f, 0.550188303f, -1.767568469f, -5.429461956f, |
| -5.065113068f, -2.111886740f, -3.606999397f, -2.410579205f, 1.013466120f, |
| 1.057218194f, 0.305267453f, 2.898609161f, 5.776575565f, 4.792305946f, |
| -0.863526106f, -2.439013481f, -0.825202525f, -2.297998428f, -0.520106375f, |
| -0.653605103f, -3.204111576f, -2.455038786f, -2.160304308f, 0.622359931f, |
| 3.803062916f, 4.340928555f, 2.390868664f, 1.645600080f, 0.405841053f, |
| -0.153203994f, 3.438643217f, 4.752261162f, 1.552502871f, 1.947945356f, |
| 0.856451511f, -0.606808305f, -1.223945618f, -1.845071912f, -0.204472303f, |
| 1.750840783f, 2.435559034f, -1.253612280f, -2.675215721f, 1.614801407f, |
| 3.002861023f, 1.743503809f, 3.409059286f, 4.303173542f, 2.441751957f, |
| 1.752274275f, 1.874113560f, 2.070837736f, 1.401355743f, -0.330647945f, |
| -0.664121151f, 1.196543574f, 1.506967187f, 0.985752344f, -1.265938520f, |
| -1.433794141f, 0.380195618f, 0.061504841f, 1.079771042f, 1.773771763f, |
| 3.226663589f, 4.170571804f, 4.220288277f, 3.619904041f, 2.316211224f, |
| 2.012817860f, 0.370972633f, 0.517094851f, 1.869508862f, 0.357770681f, |
| -2.991472483f, -3.216646433f, 0.232109070f, 1.803660274f, 2.928784370f, |
| 4.909455776f, 5.913621426f, 4.653719902f, 4.387111187f, 4.793289661f, |
| 4.744520187f, 5.214610100f, 3.996322632f, 2.619040728f, 0.758128643f, |
| -0.092789888f, 0.070066452f, 0.704165459f, 2.042234898f, 2.768569231f, |
| 3.340583324f, 3.212181091f, 2.748130322f, 3.077554941f, 2.189792156f, |
| 2.646749735f, 2.817450523f, 1.611892223f, 1.981805444f, -1.088236094f, |
| -2.187484741f, -0.654897690f, -0.900939941f, 0.148309708f, 1.498139143f, |
| -0.261296749f, -3.220157146f, -1.727450609f, 0.807144105f, -0.809251904f, |
| -2.361308336f, -1.421746969f, -0.793132067f, -0.313778281f, -0.641793191f, |
| -0.999286890f, 0.219423503f, 0.976444781f, 0.152786255f, -0.405437022f, |
| 0.120257735f, -0.392024517f, -0.019678771f, 1.492373466f, 0.926774263f, |
| 0.566291928f, 1.307234287f, 1.496955752f, 1.448441863f, 2.212901354f, |
| 1.314700723f, 0.213681281f, 1.011370897f, 1.827155828f, 0.250772655f, |
| -0.429592669f, 0.435638547f, 1.506532907f, 1.350761652f, -0.387142301f, |
| -1.770648122f, -2.690037489f, -1.788924456f, -2.023291588f, -2.354584694f, |
| -2.587521076f, -2.002159595f, -0.355855435f, 0.825611115f, 3.075081587f, |
| 2.687968254f, 0.074088633f, 0.439936757f, 1.214704275f, 2.670343399f, |
| 1.567362547f, -1.573154926f, -3.216549397f, -3.596383333f, -3.893716335f, |
| -2.456265688f, -4.313135624f, -5.783064842f, -5.344826221f, -3.484399319f, |
| -2.235594273f, -3.568959475f, -2.447141886f, -0.755384564f, -1.178364277f, |
| 1.034289122f, 1.746821165f, -1.159413576f, -2.569937706f, -1.742212296f, |
| -0.270784855f, 1.886857986f, 0.831889153f, 0.636521816f, -0.067433357f, |
| -0.256595969f, 0.907287478f, 1.575596929f, 0.393882513f, -0.510042071f, |
| 0.507258415f, 0.059408009f, 1.776192427f, 1.664948106f, -0.341539711f, |
| -0.072047889f, -0.795555651f, 0.704908550f, 2.127685547f, 1.486027241f, |
| 1.973046541f, 2.456688404f, 2.871328354f, 4.989626408f, 5.076294422f, |
| 4.262395859f, 3.622689009f, 3.241683960f, 4.222597599f, 3.575423479f, |
| 1.997965097f, 1.391216874f, 2.329971790f, 2.898612261f, 3.871258736f, |
| 2.857767582f, 2.960238218f, 3.047467470f, 2.790968180f, 2.183730364f, |
| 1.991029263f, 2.727865934f, 1.561259747f, 0.787606239f, 3.036532879f, |
| 2.430759192f, 1.475822210f, 2.307994127f, 1.857011318f, 1.538355589f, |
| 2.320549965f, 3.305005074f, 2.554165363f, 2.630100727f, 3.506094217f, |
| 4.454113483f, 2.894124269f, 4.061129570f, 4.425602436f, 3.218537807f, |
| 2.712452173f, 5.546891212f, 6.138017654f, 5.897895813f, 5.698192596f, |
| 4.096743584f, 2.661385298f, 3.646550655f, 4.626225948f, 5.025664330f, |
| 3.861543894f, 4.374861717f, 5.388185978f, 3.376737356f, 2.751175404f, |
| 3.299628258f, 2.025987387f, 1.094563961f, 0.128147125f, -4.321690559f, |
| -6.165239811f, -4.245608330f, -2.974690914f, -5.110438824f, -6.619713306f, |
| -6.594148636f, -7.972207069f, -8.034727097f, -7.296438217f, -6.822746754f, |
| -6.375267029f, -7.629575729f, -8.404177666f, -5.002337456f, -7.024040699f, |
| -7.799823761f, -5.423873901f, -4.861459732f, -2.772324085f, 0.002551556f, |
| -1.445306778f, -1.726813316f, 0.889497757f, 1.760663986f, 2.722227097f, |
| 4.755805969f, 4.188167572f, 1.547533512f, 2.444593906f, 1.612852097f, |
| -0.508655310f, 0.046535015f, 1.720140934f, 1.265070438f, 0.976964772f, |
| 2.446830273f, 6.308787823f, 7.798269272f, 5.347163200f, 3.540414810f, |
| 3.510186911f, 4.305843830f, 5.957427025f, 7.200410843f, 7.049768448f, |
| 7.179680824f, 8.508881569f, 9.094768524f, 12.307214737f, 14.215225220f, |
| 11.316717148f, 8.660657883f, 7.528784275f, 7.616339207f, 6.968524933f, |
| 4.246424198f, 0.214603424f, 0.449179649f, 1.695000648f, 0.110423088f, |
| -0.304885864f, -2.038585663f, -5.223299980f, -5.486608505f, -5.728059292f, |
| -4.866038799f, -2.678806305f, -3.464673519f, -3.407086372f, -2.490849733f, |
| -0.161162257f, 0.118952155f, 0.312392950f, -0.341049194f, 0.013419867f, |
| 3.722306252f, 3.901551247f, 1.781876802f, 2.446551561f, 3.659160852f, |
| 2.530288696f, 3.577404499f, 3.201550961f, 0.281389952f, -0.291333675f, |
| 1.386508465f, 2.181721210f, -2.802821159f, -1.531007886f, 1.608560324f, |
| -0.523656845f, -0.281057000f, 0.571323991f, 0.668095112f, -1.637194037f, |
| -2.756963253f, -1.340666890f, -2.180127621f, -1.874165773f, 0.660111070f, |
| 0.197176635f, 0.781580091f, 1.749967933f, 0.674724638f, -2.082683325f, |
| -3.159717083f, -2.898023844f, -4.691623211f, -5.614190102f, -6.157790661f, |
| -7.776132584f, -8.029224396f, -6.940879345f, -7.065263271f, -7.003522396f, |
| -5.691181183f, -7.872379780f, -7.614178658f, -5.778759003f, -4.605045319f, |
| -4.695390224f, -5.865473270f, -5.825413227f, -4.648111820f, -2.193091869f, |
| -0.172003269f, 1.482686043f, -0.915655136f, -2.626194954f, 1.852293015f, |
| 4.184171677f, 4.083235264f, 1.048256874f, -1.361350536f, 0.438748837f, |
| 1.716395378f, 2.916294813f, 2.639499664f, 0.059617281f, -1.883811951f, |
| 2.136622429f, 6.641947269f, 5.951328754f, 3.875293493f, 3.003573895f, |
| 2.687273264f, 4.843512535f, 6.420391560f, 6.014624596f, 3.444208860f, |
| 0.717782736f, 2.659932613f, 5.204012871f, 5.516477585f, 3.315031528f, |
| 0.454023123f, -0.026421070f, 0.802503586f, 2.606507778f, 1.679640770f, |
| -1.917723656f, -3.348850250f, -2.580049515f, -1.783200264f, -0.810425520f, |
| -0.374402523f, -3.705567360f, -5.367071629f, -4.344952106f, -0.968293428f, |
| 1.147591949f, -1.240655184f, -2.621209621f, -2.452539444f, -1.543132067f, |
| 0.422753096f, 1.026433110f, 0.858573675f, -0.695377707f, -0.242624998f, |
| 3.892488956f, 4.100893021f, 3.498974323f, 1.744507313f, -0.912925899f, |
| 0.929271877f, 3.531583786f, 4.938030243f, 4.081199646f, 0.061933577f, |
| -2.232783318f, -1.356980443f, 1.794556737f, 3.510458231f, 1.323192716f, |
| -0.505770206f, 2.126557350f, 2.507567406f, 2.232018232f, 1.872283101f, |
| 1.265762568f, 0.577634692f, 0.021484375f, 3.114191532f, 1.579384208f, |
| 0.930754900f, 0.308351398f, -0.425426602f, 3.359810352f, 2.437057972f, |
| 1.210662127f, 0.708607912f, -1.576705575f, 0.007833481f, -0.178357601f, |
| -0.880272985f, 0.078738928f, 0.339336634f, -0.763550043f, -1.669098496f, |
| -2.083987713f, -1.946106076f, -0.953974366f, -0.856883168f, -1.282670021f, |
| -1.551425457f, -2.249363184f, -2.555188894f, -1.254808664f, -1.368662596f, |
| -1.839509130f, -0.839046180f, -0.452676475f, 0.721064806f, 1.988085508f, |
| 0.456556678f, -0.255003691f, 0.384676337f, 1.075410485f, 0.617453933f, |
| 1.470067143f, 1.493275523f, 0.954153359f, 1.027234554f, -0.434967309f, |
| -0.694453120f, 0.477285773f, 0.436861426f, 1.486879349f, -0.158989906f, |
| 0.361879885f, 3.234876394f, 1.105287671f, -0.982552111f, 1.514200211f, |
| 0.821707547f, -1.142312169f, 1.845819831f, 3.934516191f, 2.251807690f, |
| 0.530044913f, -1.043874860f, -0.891365111f, -0.264675498f, 0.288083673f, |
| 0.606682122f, -1.132072091f, -3.530973911f, -2.005296707f, 0.335011721f, |
| -0.240332901f, -2.763209343f, -2.148519516f, -1.864180326f, -0.814615071f, |
| -1.589591861f, -2.455522776f, -0.756391644f, 0.689822078f, 0.171640277f, |
| -0.225937843f, 0.363246441f, 0.098157287f, -1.638891220f, -0.400456548f, |
| 1.076233864f, 2.288599968f, 2.716089964f, 1.585703373f, 0.846301913f, |
| 0.887506902f, -0.439320147f, -0.823126972f, 0.712436378f, 1.027045608f, |
| 0.360925227f, -2.289939404f, -1.035227180f, 0.931313038f, -0.133454978f, |
| 0.160856903f, 0.700653732f, 0.817580283f, -0.223383546f, 0.713623106f, |
| 1.327106714f, 1.558022618f, 1.346337557f, -0.661301017f, 0.707845926f, |
| 2.435726643f, 0.763329387f, 0.485213757f, 2.295393229f, 4.525130272f, |
| 2.354229450f, -0.043517172f, 1.635316610f, 1.651852608f, 1.240020633f, |
| 0.320237398f, -0.571269870f, -0.686546564f, -1.796948791f, -0.966899753f, |
| -0.404109240f, -1.295783877f, -2.058131218f, -2.279026985f, -2.183017731f, |
| -2.516988277f, -0.276667058f, -0.475267202f, -2.645681143f, -0.504431605f, |
| -1.031255722f, -3.401877880f, -1.075011969f, -0.667404234f, -2.419279575f, |
| -1.230643749f, 1.151491284f, 0.374734998f, -2.004124880f, -1.923788905f, |
| -0.767004371f, 0.512374282f, 2.254727125f, 1.373157024f, 0.633022547f, |
| 0.194831967f, 0.226476192f, 1.294842482f, 0.838023365f, 1.291390896f, |
| 0.128176212f, -1.109287858f, 0.166733295f, 0.847469866f, -0.662097514f, |
| -0.489783406f, 1.523754478f, 1.903803706f, -0.748670340f, 0.721136212f, |
| 1.627746105f, -0.731291413f, 0.646574259f, 1.722917080f, 0.372141778f, |
| -0.063563704f, 0.916404963f, 2.092662811f, 1.699481010f, 0.181074798f, |
| -1.361395121f, 0.581034362f, 1.451567292f, 0.526586652f, 1.206429839f, |
| -1.041464567f, -2.891606331f, 0.638695598f, 1.198848009f, -0.771047413f, |
| -1.074250221f, -0.500067651f, 0.308775485f, 0.552724898f, 1.083443999f, |
| 1.371356130f, 0.360372365f, 3.391613960f, 2.896605730f, 0.799045980f, |
| 0.922905385f, 3.240214348f, 4.740911484f, 2.945639610f, 2.544054747f, |
| 3.048654795f, 3.541822433f, 4.390746117f, 5.632675171f, 7.721554756f, |
| 6.390114784f, 5.962307930f, 5.873732567f, 5.625522137f, 4.857854843f, |
| 3.148367405f, 3.966898203f, 4.309705257f, 3.543770313f, 2.427399397f, |
| 0.324177742f, -1.809771061f, -2.191485405f, 0.006873131f, -0.876847267f, |
| -0.928904057f, 0.889565945f, -0.127671242f, -1.695463657f, -1.193793774f, |
| -1.452976227f, -3.406696558f, -2.564189196f, -2.136555195f, -2.374645710f, |
| -3.230790854f, -3.076714516f, -3.245117664f, -2.254387617f, -0.245034039f, |
| -1.072510719f, -1.887740970f, 0.431427240f, 1.132410765f, -1.015120149f, |
| -0.274977922f, -1.910447717f, -2.865208864f, -0.131696820f}; |
| |
| // TODO(bugs.webrtc.org/9076): Move to resource file. |
| constexpr std::array<float, 385> kPitchBufferFrameSquareEnergies = { |
| 5150.291992188f, 5150.894531250f, 5145.122558594f, 5148.914062500f, |
| 5152.802734375f, 5156.541015625f, 5163.048339844f, 5172.149414062f, |
| 5177.349121094f, 5184.365722656f, 5199.292480469f, 5202.612304688f, |
| 5197.510253906f, 5189.979492188f, 5183.533203125f, 5190.677734375f, |
| 5203.943359375f, 5207.876464844f, 5209.395019531f, 5225.451660156f, |
| 5249.794921875f, 5271.816894531f, 5280.045410156f, 5285.289062500f, |
| 5288.319335938f, 5289.758789062f, 5294.285644531f, 5289.979980469f, |
| 5287.337402344f, 5287.237792969f, 5281.462402344f, 5271.676269531f, |
| 5256.257324219f, 5240.524414062f, 5230.869628906f, 5207.531250000f, |
| 5176.040039062f, 5144.021484375f, 5109.295410156f, 5068.527832031f, |
| 5008.909667969f, 4977.587890625f, 4959.000976562f, 4950.016601562f, |
| 4940.795410156f, 4937.358398438f, 4935.286132812f, 4914.154296875f, |
| 4906.706542969f, 4906.924804688f, 4907.674804688f, 4899.855468750f, |
| 4894.340820312f, 4906.948242188f, 4910.065429688f, 4921.032714844f, |
| 4949.294433594f, 4982.643066406f, 5000.996093750f, 5005.875488281f, |
| 5020.441894531f, 5031.938964844f, 5041.877441406f, 5035.990722656f, |
| 5037.362792969f, 5043.038085938f, 5044.236328125f, 5042.322753906f, |
| 5041.990722656f, 5047.362304688f, 5056.785644531f, 5054.579101562f, |
| 5050.326171875f, 5053.495117188f, 5060.186523438f, 5065.591796875f, |
| 5066.717285156f, 5069.499511719f, 5076.201171875f, 5076.687011719f, |
| 5076.316894531f, 5077.581054688f, 5076.226074219f, 5074.094238281f, |
| 5074.039062500f, 5073.663574219f, 5076.283691406f, 5077.278808594f, |
| 5076.094238281f, 5077.806152344f, 5081.035644531f, 5082.431640625f, |
| 5082.995605469f, 5084.653320312f, 5084.936035156f, 5085.394042969f, |
| 5085.735351562f, 5080.651855469f, 5080.542968750f, 5079.969238281f, |
| 5076.432617188f, 5072.439453125f, 5073.252441406f, 5071.974609375f, |
| 5071.458496094f, 5066.017578125f, 5065.670898438f, 5065.144042969f, |
| 5055.592773438f, 5060.104980469f, 5060.505371094f, 5054.157226562f, |
| 5056.915039062f, 5067.208007812f, 5060.940917969f, 5058.419921875f, |
| 5053.248046875f, 5049.823730469f, 5048.573242188f, 5053.195312500f, |
| 5053.444335938f, 5054.143066406f, 5056.270019531f, 5063.881835938f, |
| 5070.784667969f, 5074.042480469f, 5080.785156250f, 5085.663085938f, |
| 5095.979003906f, 5101.596191406f, 5088.784667969f, 5087.686523438f, |
| 5087.946777344f, 5087.369140625f, 5081.445312500f, 5081.519042969f, |
| 5087.940917969f, 5102.099121094f, 5126.864257812f, 5147.613281250f, |
| 5170.079589844f, 5189.276367188f, 5210.265136719f, 5244.745117188f, |
| 5268.821777344f, 5277.381835938f, 5279.768066406f, 5278.750000000f, |
| 5283.853027344f, 5292.671386719f, 5291.744628906f, 5294.732421875f, |
| 5294.322265625f, 5294.267089844f, 5297.530761719f, 5302.179199219f, |
| 5312.768066406f, 5323.202148438f, 5335.357910156f, 5344.610839844f, |
| 5347.597167969f, 5346.077148438f, 5346.071289062f, 5346.083984375f, |
| 5348.088378906f, 5349.661621094f, 5350.157226562f, 5351.855957031f, |
| 5347.257812500f, 5345.171875000f, 5344.617675781f, 5343.106445312f, |
| 5342.778808594f, 5338.655761719f, 5341.668457031f, 5347.518066406f, |
| 5362.014160156f, 5361.167968750f, 5362.926269531f, 5371.575195312f, |
| 5374.099609375f, 5381.186523438f, 5381.963867188f, 5386.806152344f, |
| 5389.590820312f, 5384.562011719f, 5372.485839844f, 5370.576660156f, |
| 5369.640136719f, 5369.698242188f, 5371.199707031f, 5372.644531250f, |
| 5394.006835938f, 5395.366699219f, 5395.259277344f, 5395.398437500f, |
| 5395.895507812f, 5401.420898438f, 5420.036621094f, 5434.017578125f, |
| 5434.215820312f, 5437.827636719f, 5442.944335938f, 5450.980468750f, |
| 5449.246582031f, 5449.135742188f, 5453.259765625f, 5453.792968750f, |
| 5459.676757812f, 5460.213867188f, 5479.227539062f, 5512.076171875f, |
| 5520.272949219f, 5519.662109375f, 5517.395996094f, 5516.550292969f, |
| 5520.786621094f, 5527.268066406f, 5526.668457031f, 5549.916992188f, |
| 5577.750976562f, 5580.141113281f, 5579.533691406f, 5576.632324219f, |
| 5573.938476562f, 5571.166503906f, 5570.603027344f, 5570.708496094f, |
| 5577.238769531f, 5577.625976562f, 5589.325683594f, 5602.189941406f, |
| 5612.587402344f, 5613.887695312f, 5613.588867188f, 5608.100585938f, |
| 5632.956054688f, 5679.322265625f, 5682.149414062f, 5683.846191406f, |
| 5691.708496094f, 5683.279785156f, 5694.248535156f, 5744.740722656f, |
| 5756.655761719f, 5755.952148438f, 5756.665527344f, 5750.700195312f, |
| 5784.060546875f, 5823.021972656f, 5829.233398438f, 5817.804687500f, |
| 5827.333984375f, 5826.451171875f, 5824.887695312f, 5825.734375000f, |
| 5813.386230469f, 5789.609863281f, 5779.115234375f, 5778.762695312f, |
| 5785.748046875f, 5792.981933594f, 5787.567871094f, 5778.096679688f, |
| 5764.337402344f, 5766.734375000f, 5766.489746094f, 5769.543945312f, |
| 5773.183593750f, 5775.720703125f, 5774.311523438f, 5769.303710938f, |
| 5765.815917969f, 5767.521484375f, 5775.251953125f, 5785.067382812f, |
| 5770.117187500f, 5749.073242188f, 5747.606933594f, 5757.671875000f, |
| 5762.530273438f, 5774.506347656f, 5784.737304688f, 5775.916015625f, |
| 5779.816894531f, 5795.064453125f, 5808.736816406f, 5813.699707031f, |
| 5823.773925781f, 5840.490234375f, 5833.751953125f, 5810.150390625f, |
| 5800.072265625f, 5815.070800781f, 5822.964355469f, 5817.615234375f, |
| 5783.978027344f, 5748.952636719f, 5735.553710938f, 5730.132812500f, |
| 5724.260253906f, 5721.703613281f, 5695.653808594f, 5652.838867188f, |
| 5649.729980469f, 5647.268554688f, 5647.265136719f, 5641.350585938f, |
| 5636.762695312f, 5637.900390625f, 5639.662109375f, 5639.672851562f, |
| 5638.901367188f, 5622.253417969f, 5604.906738281f, 5601.475585938f, |
| 5595.938476562f, 5595.687011719f, 5598.612792969f, 5601.322753906f, |
| 5598.558593750f, 5577.227050781f, 5544.295410156f, 5514.978027344f, |
| 5499.678222656f, 5488.303222656f, 5471.735839844f, 5429.718261719f, |
| 5376.806640625f, 5348.682128906f, 5307.851074219f, 5260.914062500f, |
| 5212.738281250f, 5148.544921875f, 5091.187500000f, 5053.512207031f, |
| 5023.785156250f, 5002.202148438f, 4994.252441406f, 4984.498046875f, |
| 4980.251464844f, 4979.796875000f, 4976.738769531f, 4979.579589844f, |
| 4986.528320312f, 4991.153808594f, 4991.462890625f, 4987.881347656f, |
| 4987.417480469f, 4983.885742188f, 4984.341308594f, 4985.302734375f, |
| 4985.303710938f, 4985.449707031f, 4989.282226562f, 4994.246582031f, |
| 4992.635742188f, 4992.064453125f, 4987.331054688f, 4985.806152344f, |
| 4986.047851562f, 4985.968750000f, 4979.141113281f, 4976.958984375f, |
| 4972.650390625f, 4959.916503906f, 4956.325683594f, 4956.408691406f, |
| 4949.288085938f, 4951.827636719f, 4962.202636719f, 4981.184570312f, |
| 4992.152832031f, 4997.386230469f, 5011.211914062f, 5026.242187500f, |
| 5023.573730469f, 5012.373046875f, 5017.451171875f, 5010.541015625f, |
| 4980.446777344f, 4958.639648438f, 4963.649902344f, 5627.020507812f, |
| 6869.356445312f}; |
| |
| // TODO(bugs.webrtc.org/9076): Move to resource file. |
| constexpr std::array<float, 147> kPitchBufferAutoCorrCoeffs = { |
| -423.526794434f, -260.724456787f, -173.558380127f, -71.720344543f, |
| -1.149698257f, 71.451370239f, 71.455848694f, 149.755233765f, |
| 199.401885986f, 243.961334229f, 269.339721680f, 243.776992798f, |
| 294.753814697f, 209.465484619f, 139.224700928f, 131.474136353f, |
| 42.872886658f, -32.431114197f, -90.191261292f, -94.912338257f, |
| -172.627227783f, -138.089843750f, -89.236648560f, -69.348426819f, |
| 25.044368744f, 44.184486389f, 61.602676392f, 150.157394409f, |
| 185.254760742f, 233.352676392f, 296.255371094f, 292.464141846f, |
| 256.903472900f, 250.926574707f, 174.207122803f, 130.214172363f, |
| 65.655899048f, -68.448402405f, -147.239669800f, -230.553405762f, |
| -311.217895508f, -447.173889160f, -509.306060791f, -551.155822754f, |
| -580.678405762f, -658.902709961f, -697.141967773f, -751.233032227f, |
| -690.860351562f, -571.689575195f, -521.124572754f, -429.477294922f, |
| -375.685913086f, -277.387329102f, -154.100753784f, -105.723197937f, |
| 117.502632141f, 219.290512085f, 255.376770020f, 444.264831543f, |
| 470.727416992f, 460.139129639f, 494.179931641f, 389.801116943f, |
| 357.082763672f, 222.748138428f, 179.100601196f, -26.893497467f, |
| -85.033767700f, -223.577529907f, -247.136367798f, -223.011428833f, |
| -292.724914551f, -246.538131714f, -247.388458252f, -228.452484131f, |
| -30.476575851f, 4.652336121f, 64.730491638f, 156.081161499f, |
| 177.569305420f, 261.671569824f, 336.274414062f, 424.203369141f, |
| 564.190734863f, 608.841796875f, 671.252136230f, 712.249877930f, |
| 623.135498047f, 564.775695801f, 576.405639648f, 380.181854248f, |
| 306.687164307f, 180.344757080f, -41.317466736f, -183.548736572f, |
| -223.835021973f, -273.299652100f, -235.727813721f, -276.899627686f, |
| -302.224975586f, -349.227142334f, -370.935058594f, -364.022613525f, |
| -287.682952881f, -273.828704834f, -156.869720459f, -88.654510498f, |
| 14.299798012f, 137.048034668f, 260.182342529f, 423.380767822f, |
| 591.277282715f, 581.151306152f, 643.898864746f, 547.919006348f, |
| 355.534271240f, 238.222915649f, 4.463035583f, -193.763305664f, |
| -281.212432861f, -546.399353027f, -615.602600098f, -574.225891113f, |
| -726.701843262f, -564.840942383f, -588.488037109f, -651.052551270f, |
| -453.769104004f, -502.886627197f, -463.373016357f, -291.709564209f, |
| -288.857421875f, -152.114242554f, 105.401855469f, 211.479980469f, |
| 468.501983643f, 796.984985352f, 880.254089355f, 1114.614379883f, |
| 1219.664794922f, 1093.687377930f, 1125.042602539f, 1020.942382812f, |
| 794.315246582f, 772.126831055f, 447.410736084f}; |
| |
| constexpr std::array<size_t, 2> kTestPitchPeriods = { |
| 3 * kMinPitch48kHz / 2, (3 * kMinPitch48kHz + kMaxPitch48kHz) / 2, |
| }; |
| constexpr std::array<float, 2> kTestPitchGains = {0.35f, 0.75f}; |
| |
| } // namespace |
| |
| class ComputePitchGainThresholdTest |
| : public testing::Test, |
| public ::testing::WithParamInterface< |
| std::tuple<size_t, size_t, size_t, float, size_t, float, float>> {}; |
| |
| TEST_P(ComputePitchGainThresholdTest, BitExactness) { |
| const auto params = GetParam(); |
| const size_t candidate_pitch_period = std::get<0>(params); |
| const size_t pitch_period_ratio = std::get<1>(params); |
| const size_t initial_pitch_period = std::get<2>(params); |
| const float initial_pitch_gain = std::get<3>(params); |
| const size_t prev_pitch_period = std::get<4>(params); |
| const size_t prev_pitch_gain = std::get<5>(params); |
| const float threshold = std::get<6>(params); |
| |
| { |
| // TODO(bugs.webrtc.org/8948): Add when the issue is fixed. |
| // FloatingPointExceptionObserver fpe_observer; |
| |
| EXPECT_NEAR( |
| threshold, |
| ComputePitchGainThreshold(candidate_pitch_period, pitch_period_ratio, |
| initial_pitch_period, initial_pitch_gain, |
| prev_pitch_period, prev_pitch_gain), |
| 3e-6f); |
| } |
| } |
| |
| INSTANTIATE_TEST_CASE_P( |
| RnnVadTest, |
| ComputePitchGainThresholdTest, |
| ::testing::Values( |
| std::make_tuple(31, 7, 219, 0.45649201f, 199, 0.604747f, 0.40000001f), |
| std::make_tuple(113, |
| 2, |
| 226, |
| 0.20967799f, |
| 219, |
| 0.40392199f, |
| 0.30000001f), |
| std::make_tuple(63, 2, 126, 0.210788f, 364, 0.098519f, 0.40000001f), |
| std::make_tuple(30, 5, 152, 0.82356697f, 149, 0.55535901f, 0.700032f), |
| std::make_tuple(76, 2, 151, 0.79522997f, 151, 0.82356697f, 0.675946f), |
| std::make_tuple(31, 5, 153, 0.85069299f, 150, 0.79073799f, 0.72308898f), |
| std::make_tuple(78, 2, 156, 0.72750503f, 153, 0.85069299f, 0.618379f))); |
| |
| TEST(RnnVadTest, ComputeSlidingFrameSquareEnergiesBitExactness) { |
| std::array<float, kPitchBufferFrameSquareEnergies.size()> computed_output; |
| { |
| // TODO(bugs.webrtc.org/8948): Add when the issue is fixed. |
| // FloatingPointExceptionObserver fpe_observer; |
| |
| ComputeSlidingFrameSquareEnergies( |
| {kPitchBufferData.data(), kPitchBufferData.size()}, |
| {computed_output.data(), computed_output.size()}); |
| } |
| ExpectNearAbsolute({kPitchBufferFrameSquareEnergies.data(), |
| kPitchBufferFrameSquareEnergies.size()}, |
| {computed_output.data(), computed_output.size()}, 3e-2f); |
| } |
| |
| TEST(RnnVadTest, ComputePitchAutoCorrelationBitExactness) { |
| std::array<float, kBufSize12kHz> pitch_buf_decimated; |
| Decimate2x({kPitchBufferData.data(), kPitchBufferData.size()}, |
| {pitch_buf_decimated.data(), pitch_buf_decimated.size()}); |
| std::array<float, kPitchBufferAutoCorrCoeffs.size()> computed_output; |
| { |
| // TODO(bugs.webrtc.org/8948): Add when the issue is fixed. |
| // FloatingPointExceptionObserver fpe_observer; |
| |
| ComputePitchAutoCorrelation( |
| {pitch_buf_decimated.data(), pitch_buf_decimated.size()}, |
| kMaxPitch12kHz, {computed_output.data(), computed_output.size()}); |
| } |
| ExpectNearAbsolute( |
| {kPitchBufferAutoCorrCoeffs.data(), kPitchBufferAutoCorrCoeffs.size()}, |
| {computed_output.data(), computed_output.size()}, 3e-3f); |
| } |
| |
| TEST(RnnVadTest, FindBestPitchPeriodsBitExactness) { |
| std::array<float, kBufSize12kHz> pitch_buf_decimated; |
| Decimate2x({kPitchBufferData.data(), kPitchBufferData.size()}, |
| {pitch_buf_decimated.data(), pitch_buf_decimated.size()}); |
| std::array<size_t, 2> pitch_candidates_inv_lags; |
| { |
| // TODO(bugs.webrtc.org/8948): Add when the issue is fixed. |
| // FloatingPointExceptionObserver fpe_observer; |
| |
| pitch_candidates_inv_lags = FindBestPitchPeriods( |
| {kPitchBufferAutoCorrCoeffs}, {pitch_buf_decimated}, kMaxPitch12kHz); |
| } |
| const std::array<size_t, 2> expected_output = {140, 142}; |
| EXPECT_EQ(expected_output, pitch_candidates_inv_lags); |
| } |
| |
| TEST(RnnVadTest, RefinePitchPeriod48kHzBitExactness) { |
| std::array<float, kBufSize12kHz> pitch_buf_decimated; |
| Decimate2x({kPitchBufferData.data(), kPitchBufferData.size()}, |
| {pitch_buf_decimated.data(), pitch_buf_decimated.size()}); |
| size_t pitch_inv_lag; |
| { |
| // TODO(bugs.webrtc.org/8948): Add when the issue is fixed. |
| // FloatingPointExceptionObserver fpe_observer; |
| |
| const std::array<size_t, 2> pitch_candidates_inv_lags = {280, 284}; |
| pitch_inv_lag = RefinePitchPeriod48kHz( |
| {kPitchBufferData.data(), kPitchBufferData.size()}, |
| {pitch_candidates_inv_lags.data(), pitch_candidates_inv_lags.size()}); |
| } |
| EXPECT_EQ(560u, pitch_inv_lag); |
| } |
| |
| class CheckLowerPitchPeriodsAndComputePitchGainTest |
| : public testing::Test, |
| public ::testing::WithParamInterface< |
| std::tuple<size_t, size_t, float, size_t, float>> {}; |
| |
| TEST_P(CheckLowerPitchPeriodsAndComputePitchGainTest, BitExactness) { |
| const auto params = GetParam(); |
| const size_t initial_pitch_period = std::get<0>(params); |
| const size_t prev_pitch_period = std::get<1>(params); |
| const float prev_pitch_gain = std::get<2>(params); |
| const size_t expected_pitch_period = std::get<3>(params); |
| const float expected_pitch_gain = std::get<4>(params); |
| |
| { |
| // TODO(bugs.webrtc.org/8948): Add when the issue is fixed. |
| // FloatingPointExceptionObserver fpe_observer; |
| |
| const auto computed_output = CheckLowerPitchPeriodsAndComputePitchGain( |
| {kPitchBufferData.data(), kPitchBufferData.size()}, |
| initial_pitch_period, {prev_pitch_period, prev_pitch_gain}); |
| EXPECT_EQ(expected_pitch_period, computed_output.period); |
| EXPECT_NEAR(expected_pitch_gain, computed_output.gain, 1e-6f); |
| } |
| } |
| |
| INSTANTIATE_TEST_CASE_P(RnnVadTest, |
| CheckLowerPitchPeriodsAndComputePitchGainTest, |
| ::testing::Values(std::make_tuple(kTestPitchPeriods[0], |
| kTestPitchPeriods[0], |
| kTestPitchGains[0], |
| 91, |
| -0.0188608f), |
| std::make_tuple(kTestPitchPeriods[0], |
| kTestPitchPeriods[0], |
| kTestPitchGains[1], |
| 91, |
| -0.0188608f), |
| std::make_tuple(kTestPitchPeriods[0], |
| kTestPitchPeriods[1], |
| kTestPitchGains[0], |
| 91, |
| -0.0188608f), |
| std::make_tuple(kTestPitchPeriods[0], |
| kTestPitchPeriods[1], |
| kTestPitchGains[1], |
| 91, |
| -0.0188608f), |
| std::make_tuple(kTestPitchPeriods[1], |
| kTestPitchPeriods[0], |
| kTestPitchGains[0], |
| 475, |
| -0.0904344f), |
| std::make_tuple(kTestPitchPeriods[1], |
| kTestPitchPeriods[0], |
| kTestPitchGains[1], |
| 475, |
| -0.0904344f), |
| std::make_tuple(kTestPitchPeriods[1], |
| kTestPitchPeriods[1], |
| kTestPitchGains[0], |
| 475, |
| -0.0904344f), |
| std::make_tuple(kTestPitchPeriods[1], |
| kTestPitchPeriods[1], |
| kTestPitchGains[1], |
| 475, |
| -0.0904344f))); |
| |
| } // namespace test |
| } // namespace rnn_vad |
| } // namespace webrtc |
