From 75d5c4ffe40a9660547e230bd2bb49a7edf8b995 Mon Sep 17 00:00:00 2001 From: Stone_Red <56473591+Stone-Red-Code@users.noreply.github.com> Date: Sat, 19 Feb 2022 18:57:52 +0100 Subject: [PATCH] Initial commit --- .gitattributes | 2 + .gitignore | 398 +++++++++++++++++++++++++++ App.xaml | 9 + App.xaml.cs | 15 + AssemblyInfo.cs | 10 + AudioProcessing/AudioBeatDetector.cs | 114 ++++++++ AudioProcessing/BeatChunk.cs | 11 + AudioProcessing/SampleAggregator.cs | 61 ++++ BeatDetector.csproj | 14 + LICENSE | 21 ++ MainWindow.xaml | 23 ++ MainWindow.xaml.cs | 69 +++++ README.md | 2 + 13 files changed, 749 insertions(+) create mode 100644 .gitattributes create mode 100644 .gitignore create mode 100644 App.xaml create mode 100644 App.xaml.cs create mode 100644 AssemblyInfo.cs create mode 100644 AudioProcessing/AudioBeatDetector.cs create mode 100644 AudioProcessing/BeatChunk.cs create mode 100644 AudioProcessing/SampleAggregator.cs create mode 100644 BeatDetector.csproj create mode 100644 LICENSE create mode 100644 MainWindow.xaml create mode 100644 MainWindow.xaml.cs create mode 100644 README.md diff --git a/.gitattributes b/.gitattributes new file mode 100644 index 0000000..dfe0770 --- /dev/null +++ b/.gitattributes @@ -0,0 +1,2 @@ +# Auto detect text files and perform LF normalization +* text=auto diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..426d76d --- /dev/null +++ b/.gitignore @@ -0,0 +1,398 @@ +## Ignore Visual Studio temporary files, build results, and +## files generated by popular Visual Studio add-ons. +## +## Get latest from https://github.com/github/gitignore/blob/main/VisualStudio.gitignore + +# User-specific files +*.rsuser +*.suo +*.user +*.userosscache +*.sln.docstates + +# User-specific files (MonoDevelop/Xamarin Studio) +*.userprefs + +# Mono auto generated files +mono_crash.* + +# Build results +[Dd]ebug/ +[Dd]ebugPublic/ +[Rr]elease/ +[Rr]eleases/ +x64/ +x86/ +[Ww][Ii][Nn]32/ +[Aa][Rr][Mm]/ +[Aa][Rr][Mm]64/ +bld/ +[Bb]in/ +[Oo]bj/ +[Ll]og/ +[Ll]ogs/ + +# Visual Studio 2015/2017 cache/options directory +.vs/ +# Uncomment if you have tasks that create the project's static files in wwwroot +#wwwroot/ + +# Visual Studio 2017 auto generated files +Generated\ Files/ + +# MSTest test Results +[Tt]est[Rr]esult*/ +[Bb]uild[Ll]og.* + +# NUnit +*.VisualState.xml +TestResult.xml +nunit-*.xml + +# Build Results of an ATL Project +[Dd]ebugPS/ +[Rr]eleasePS/ +dlldata.c + +# Benchmark Results +BenchmarkDotNet.Artifacts/ + +# .NET Core +project.lock.json +project.fragment.lock.json +artifacts/ + +# ASP.NET Scaffolding +ScaffoldingReadMe.txt + +# StyleCop +StyleCopReport.xml + +# Files built by Visual Studio +*_i.c +*_p.c +*_h.h +*.ilk +*.meta +*.obj +*.iobj +*.pch +*.pdb +*.ipdb +*.pgc +*.pgd +*.rsp +*.sbr +*.tlb +*.tli +*.tlh +*.tmp +*.tmp_proj +*_wpftmp.csproj +*.log +*.tlog +*.vspscc +*.vssscc +.builds +*.pidb +*.svclog +*.scc + +# Chutzpah Test files +_Chutzpah* + +# Visual C++ cache files +ipch/ +*.aps +*.ncb +*.opendb +*.opensdf +*.sdf +*.cachefile +*.VC.db +*.VC.VC.opendb + +# Visual Studio profiler +*.psess +*.vsp +*.vspx +*.sap + +# Visual Studio Trace Files +*.e2e + +# TFS 2012 Local Workspace +$tf/ + +# Guidance Automation Toolkit +*.gpState + +# ReSharper is a .NET coding add-in +_ReSharper*/ +*.[Rr]e[Ss]harper +*.DotSettings.user + +# TeamCity is a build add-in +_TeamCity* + +# DotCover is a Code Coverage Tool +*.dotCover + +# AxoCover is a Code Coverage Tool +.axoCover/* +!.axoCover/settings.json + +# Coverlet is a free, cross platform Code Coverage Tool +coverage*.json +coverage*.xml +coverage*.info + +# Visual Studio code coverage results +*.coverage +*.coveragexml + +# NCrunch +_NCrunch_* +.*crunch*.local.xml +nCrunchTemp_* + +# MightyMoose +*.mm.* +AutoTest.Net/ + +# Web workbench (sass) +.sass-cache/ + +# Installshield output folder +[Ee]xpress/ + +# DocProject is a documentation generator add-in +DocProject/buildhelp/ +DocProject/Help/*.HxT +DocProject/Help/*.HxC +DocProject/Help/*.hhc +DocProject/Help/*.hhk +DocProject/Help/*.hhp +DocProject/Help/Html2 +DocProject/Help/html + +# Click-Once directory +publish/ + +# Publish Web Output +*.[Pp]ublish.xml +*.azurePubxml +# Note: Comment the next line if you want to checkin your web deploy settings, +# but database connection strings (with potential passwords) will be unencrypted +*.pubxml +*.publishproj + +# Microsoft Azure Web App publish settings. Comment the next line if you want to +# checkin your Azure Web App publish settings, but sensitive information contained +# in these scripts will be unencrypted +PublishScripts/ + +# NuGet Packages +*.nupkg +# NuGet Symbol Packages +*.snupkg +# The packages folder can be ignored because of Package Restore +**/[Pp]ackages/* +# except build/, which is used as an MSBuild target. +!**/[Pp]ackages/build/ +# Uncomment if necessary however generally it will be regenerated when needed +#!**/[Pp]ackages/repositories.config +# NuGet v3's project.json files produces more ignorable files +*.nuget.props +*.nuget.targets + +# Microsoft Azure Build Output +csx/ +*.build.csdef + +# Microsoft Azure Emulator +ecf/ +rcf/ + +# Windows Store app package directories and files +AppPackages/ +BundleArtifacts/ +Package.StoreAssociation.xml +_pkginfo.txt +*.appx +*.appxbundle +*.appxupload + +# Visual Studio cache files +# files ending in .cache can be ignored +*.[Cc]ache +# but keep track of directories ending in .cache +!?*.[Cc]ache/ + +# Others +ClientBin/ +~$* +*~ +*.dbmdl +*.dbproj.schemaview +*.jfm +*.pfx +*.publishsettings +orleans.codegen.cs + +# Including strong name files can present a security risk +# (https://github.com/github/gitignore/pull/2483#issue-259490424) +#*.snk + +# Since there are multiple workflows, uncomment next line to ignore bower_components +# (https://github.com/github/gitignore/pull/1529#issuecomment-104372622) +#bower_components/ + +# RIA/Silverlight projects +Generated_Code/ + +# Backup & report files from converting an old project file +# to a newer Visual Studio version. Backup files are not needed, +# because we have git ;-) +_UpgradeReport_Files/ +Backup*/ +UpgradeLog*.XML +UpgradeLog*.htm +ServiceFabricBackup/ +*.rptproj.bak + +# SQL Server files +*.mdf +*.ldf +*.ndf + +# Business Intelligence projects +*.rdl.data +*.bim.layout +*.bim_*.settings +*.rptproj.rsuser +*- [Bb]ackup.rdl +*- [Bb]ackup ([0-9]).rdl +*- [Bb]ackup ([0-9][0-9]).rdl + +# Microsoft Fakes +FakesAssemblies/ + +# GhostDoc plugin setting file +*.GhostDoc.xml + +# Node.js Tools for Visual Studio +.ntvs_analysis.dat +node_modules/ + +# Visual Studio 6 build log +*.plg + +# Visual Studio 6 workspace options file +*.opt + +# Visual Studio 6 auto-generated workspace file (contains which files were open etc.) +*.vbw + +# Visual Studio 6 auto-generated project file (contains which files were open etc.) +*.vbp + +# Visual Studio 6 workspace and project file (working project files containing files to include in project) +*.dsw +*.dsp + +# Visual Studio 6 technical files +*.ncb +*.aps + +# Visual Studio LightSwitch build output +**/*.HTMLClient/GeneratedArtifacts +**/*.DesktopClient/GeneratedArtifacts +**/*.DesktopClient/ModelManifest.xml +**/*.Server/GeneratedArtifacts +**/*.Server/ModelManifest.xml +_Pvt_Extensions + +# Paket dependency manager +.paket/paket.exe +paket-files/ + +# FAKE - F# Make +.fake/ + +# CodeRush personal settings +.cr/personal + +# Python Tools for Visual Studio (PTVS) +__pycache__/ +*.pyc + +# Cake - Uncomment if you are using it +# tools/** +# !tools/packages.config + +# Tabs Studio +*.tss + +# Telerik's JustMock configuration file +*.jmconfig + +# BizTalk build output +*.btp.cs +*.btm.cs +*.odx.cs +*.xsd.cs + +# OpenCover UI analysis results +OpenCover/ + +# Azure Stream Analytics local run output +ASALocalRun/ + +# MSBuild Binary and Structured Log +*.binlog + +# NVidia Nsight GPU debugger configuration file +*.nvuser + +# MFractors (Xamarin productivity tool) working folder +.mfractor/ + +# Local History for Visual Studio +.localhistory/ + +# Visual Studio History (VSHistory) files +.vshistory/ + +# BeatPulse healthcheck temp database +healthchecksdb + +# Backup folder for Package Reference Convert tool in Visual Studio 2017 +MigrationBackup/ + +# Ionide (cross platform F# VS Code tools) working folder +.ionide/ + +# Fody - auto-generated XML schema +FodyWeavers.xsd + +# VS Code files for those working on multiple tools +.vscode/* +!.vscode/settings.json +!.vscode/tasks.json +!.vscode/launch.json +!.vscode/extensions.json +*.code-workspace + +# Local History for Visual Studio Code +.history/ + +# Windows Installer files from build outputs +*.cab +*.msi +*.msix +*.msm +*.msp + +# JetBrains Rider +*.sln.iml diff --git a/App.xaml b/App.xaml new file mode 100644 index 0000000..c5d420c --- /dev/null +++ b/App.xaml @@ -0,0 +1,9 @@ + + + + + diff --git a/App.xaml.cs b/App.xaml.cs new file mode 100644 index 0000000..c1b61ee --- /dev/null +++ b/App.xaml.cs @@ -0,0 +1,15 @@ +using System; +using System.Collections.Generic; +using System.Configuration; +using System.Data; +using System.Linq; +using System.Threading.Tasks; +using System.Windows; + +namespace BeatDetector; +/// +/// Interaction logic for App.xaml +/// +public partial class App : Application +{ +} diff --git a/AssemblyInfo.cs b/AssemblyInfo.cs new file mode 100644 index 0000000..8b5504e --- /dev/null +++ b/AssemblyInfo.cs @@ -0,0 +1,10 @@ +using System.Windows; + +[assembly: ThemeInfo( + ResourceDictionaryLocation.None, //where theme specific resource dictionaries are located + //(used if a resource is not found in the page, + // or application resource dictionaries) + ResourceDictionaryLocation.SourceAssembly //where the generic resource dictionary is located + //(used if a resource is not found in the page, + // app, or any theme specific resource dictionaries) +)] diff --git a/AudioProcessing/AudioBeatDetector.cs b/AudioProcessing/AudioBeatDetector.cs new file mode 100644 index 0000000..9a75a77 --- /dev/null +++ b/AudioProcessing/AudioBeatDetector.cs @@ -0,0 +1,114 @@ +using NAudio.Dsp; +using NAudio.Wave; + +using System; +using System.Linq; + +namespace BeatDetector.AudioProcessing; + +internal class AudioBeatDetector +{ + public event EventHandler? OnBeat; + + public event EventHandler? OnNoBeat; + + private readonly IWaveIn waveIn; + private readonly SampleAggregator sampleAggregator; + private readonly TimeSpan beatDetectionDelay; + private readonly int fftLength; + + private readonly int chunkCount; + private readonly BeatChunk[] beatChunks; + + public AudioBeatDetector(TimeSpan beatDetectionDelay, int fftLength, int chunkCount) + { + this.beatDetectionDelay = beatDetectionDelay; + this.chunkCount = chunkCount; + this.fftLength = fftLength; + + beatChunks = new BeatChunk[chunkCount]; + sampleAggregator = new SampleAggregator(fftLength); + + for (int i = 0; i < chunkCount; i++) + { + beatChunks[i] = new BeatChunk(); + beatChunks[i].StopWatch.Start(); + } + + sampleAggregator.FftCalculated += new EventHandler(FftCalculated); + sampleAggregator.PerformFFT = true; + + waveIn = new WasapiLoopbackCapture(); + waveIn.DataAvailable += OnDataAvailable; + + waveIn.StartRecording(); + } + + private void OnDataAvailable(object? sender, WaveInEventArgs e) + { + byte[] buffer = e.Buffer; + int bytesRecorded = e.BytesRecorded; + int bufferIncrement = waveIn.WaveFormat.BlockAlign; + + for (int index = 0; index < bytesRecorded; index += bufferIncrement) + { + float sample32 = BitConverter.ToSingle(buffer, index); + sampleAggregator.Add(sample32); + } + } + + private void FftCalculated(object? sender, FftEventArgs e) + { + Complex[][] chunks = e.Result.Chunk(fftLength / chunkCount).ToArray(); + + for (int i = 0; i < chunkCount; i++) + { + CalculateBeat(chunks[i], beatChunks[i], i); + } + } + + private void CalculateBeat(Complex[] values, BeatChunk beatChunk, int chunkIndex) + { + double energyLevel = 0; + foreach (Complex value in values) + { + energyLevel += Math.Pow(Math.Abs(value.Y), 2); + } + + double averageEnergyLevel = beatChunk.EnergyHistory.Count > 0 ? beatChunk.EnergyHistory.Average() : double.MaxValue; + + if (beatChunk.StopWatch.Elapsed >= beatDetectionDelay) + { + double difference = energyLevel - averageEnergyLevel; + + if (difference > beatChunk.AverageDifference / 2) + { + beatChunk.AverageDifference = (difference + beatChunk.AverageDifference) / 2d; + OnBeat?.Invoke(this, new BeatDetectorEventArgs(chunkIndex, difference)); + } + else + { + OnNoBeat?.Invoke(this, new BeatDetectorEventArgs(chunkIndex, 0)); + } + beatChunk.StopWatch.Restart(); + } + + beatChunk.EnergyHistory.Add(energyLevel); + if (beatChunk.EnergyHistory.Count > 43) + { + beatChunk.EnergyHistory.RemoveAt(0); + } + } +} + +public class BeatDetectorEventArgs : EventArgs +{ + public int ChunkIndex { get; } + public double DetectedValue { get; } + + public BeatDetectorEventArgs(int chunkIndex, double detectedValue) + { + ChunkIndex = chunkIndex; + DetectedValue = detectedValue; + } +} \ No newline at end of file diff --git a/AudioProcessing/BeatChunk.cs b/AudioProcessing/BeatChunk.cs new file mode 100644 index 0000000..7e2493c --- /dev/null +++ b/AudioProcessing/BeatChunk.cs @@ -0,0 +1,11 @@ +using System.Collections.Generic; +using System.Diagnostics; + +namespace BeatDetector.AudioProcessing; + +internal class BeatChunk +{ + public Stopwatch StopWatch { get; } = new Stopwatch(); + public List EnergyHistory { get; } = new List(); + public double AverageDifference { get; set; } +} \ No newline at end of file diff --git a/AudioProcessing/SampleAggregator.cs b/AudioProcessing/SampleAggregator.cs new file mode 100644 index 0000000..1634047 --- /dev/null +++ b/AudioProcessing/SampleAggregator.cs @@ -0,0 +1,61 @@ +using NAudio.Dsp; + +using System; + +namespace BeatDetector.AudioProcessing; + +internal class SampleAggregator +{ + // FFT + public event EventHandler? FftCalculated; + + public bool PerformFFT { get; set; } + private readonly Complex[] fftBuffer; + private readonly FftEventArgs fftArgs; + private int fftPos; + private readonly int fftLength; + private readonly int m; + + public SampleAggregator(int fftLength) + { + if (!IsPowerOfTwo(fftLength)) + { + throw new ArgumentException("FFT Length must be a power of two"); + } + m = (int)Math.Log(fftLength, 2.0); + this.fftLength = fftLength; + fftBuffer = new Complex[fftLength]; + fftArgs = new FftEventArgs(fftBuffer); + } + + private bool IsPowerOfTwo(int x) + { + return (x & (x - 1)) == 0; + } + + public void Add(float value) + { + if (PerformFFT && FftCalculated is not null) + { + fftBuffer[fftPos].X = (float)(value * FastFourierTransform.HammingWindow(fftPos, fftLength)); + fftBuffer[fftPos].Y = 0; // This is always zero with audio. + fftPos++; + if (fftPos >= fftLength) + { + fftPos = 0; + FastFourierTransform.FFT(true, m, fftBuffer); + FftCalculated(this, fftArgs); + } + } + } +} + +public class FftEventArgs : EventArgs +{ + public FftEventArgs(Complex[] result) + { + Result = result; + } + + public Complex[] Result { get; private set; } +} \ No newline at end of file diff --git a/BeatDetector.csproj b/BeatDetector.csproj new file mode 100644 index 0000000..2ecdb7d --- /dev/null +++ b/BeatDetector.csproj @@ -0,0 +1,14 @@ + + + + WinExe + net6.0-windows + enable + true + + + + + + + diff --git a/LICENSE b/LICENSE new file mode 100644 index 0000000..bb3bc82 --- /dev/null +++ b/LICENSE @@ -0,0 +1,21 @@ +MIT License + +Copyright (c) 2022 Stone_Red + +Permission is hereby granted, free of charge, to any person obtaining a copy +of this software and associated documentation files (the "Software"), to deal +in the Software without restriction, including without limitation the rights +to use, copy, modify, merge, publish, distribute, sublicense, and/or sell +copies of the Software, and to permit persons to whom the Software is +furnished to do so, subject to the following conditions: + +The above copyright notice and this permission notice shall be included in all +copies or substantial portions of the Software. + +THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR +IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, +FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE +AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER +LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, +OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE +SOFTWARE. diff --git a/MainWindow.xaml b/MainWindow.xaml new file mode 100644 index 0000000..1ca76e9 --- /dev/null +++ b/MainWindow.xaml @@ -0,0 +1,23 @@ + + + + + + + + + + + + + + + + \ No newline at end of file diff --git a/MainWindow.xaml.cs b/MainWindow.xaml.cs new file mode 100644 index 0000000..ba4a344 --- /dev/null +++ b/MainWindow.xaml.cs @@ -0,0 +1,69 @@ +using BeatDetector.AudioProcessing; + +using System; +using System.Collections.ObjectModel; +using System.ComponentModel; +using System.Runtime.CompilerServices; +using System.Windows; +using System.Windows.Media; +using System.Windows.Threading; + +namespace BeatDetector; + +/// +/// Interaction logic for MainWindow.xaml +/// +public partial class MainWindow : Window +{ + public ObservableCollection BeatLights { get; } = new ObservableCollection(); + + public MainWindow() + { + DataContext = this; + InitializeComponent(); + + for (int i = 0; i < 4; i++) + { + BeatLights.Add(new BeatLight()); + } + + AudioBeatDetector audioBeatDetector = new AudioBeatDetector(new TimeSpan(0, 0, 0, 0, 100), 1024 * 2, BeatLights.Count); + audioBeatDetector.OnBeat += AudioBeatDetector_OnBeat; + audioBeatDetector.OnNoBeat += AudioBeatDetector_OnNoBeat; + } + + private void AudioBeatDetector_OnBeat(object? sender, BeatDetectorEventArgs e) + { + Dispatcher.Invoke(() => BeatLights[e.ChunkIndex].Color = Brushes.Red); + } + + private void AudioBeatDetector_OnNoBeat(object? sender, BeatDetectorEventArgs e) + { + Dispatcher.Invoke(() => BeatLights[e.ChunkIndex].Color = Brushes.White); + } +} + +public class BeatLight : INotifyPropertyChanged +{ + public event PropertyChangedEventHandler? PropertyChanged; + + private Brush color = Brushes.White; + + public Brush Color + { + get => color; + set + { + if (value != color) + { + color = value; + NotifyPropertyChanged(); + } + } + } + + private void NotifyPropertyChanged([CallerMemberName] string propertyName = "") + { + PropertyChanged?.Invoke(this, new PropertyChangedEventArgs(propertyName)); + } +} \ No newline at end of file diff --git a/README.md b/README.md new file mode 100644 index 0000000..3383e49 --- /dev/null +++ b/README.md @@ -0,0 +1,2 @@ +# BeatDetector + Tries to detect beats in songs