From 9cecaca74323a1c6a5609438bc47b035eab5f525 Mon Sep 17 00:00:00 2001
From: weil <weil@cybryda.org>
Date: Mon, 15 Jul 2024 19:14:45 +0200
Subject: [PATCH] Major cleanup of AudioCapture class

---
 src/AudioCapture.cpp | 97 ++++++++++++++++----------------------------
 src/AudioCapture.h   | 14 ++++---
 2 files changed, 44 insertions(+), 67 deletions(-)

diff --git a/src/AudioCapture.cpp b/src/AudioCapture.cpp
index c79459c..204ac85 100644
--- a/src/AudioCapture.cpp
+++ b/src/AudioCapture.cpp
@@ -1,29 +1,18 @@
 #include "AudioCapture.h"
 
-#include <windows.h>
-#include <mmdeviceapi.h>
-#include <audioclient.h>
-#include <iostream>
-
 #include <godot_cpp/variant/utility_functions.hpp>
 
 namespace godot {
 
-
 void AudioCapture::_bind_methods()
 {
-  ClassDB::bind_method(D_METHOD("process"), &AudioCapture::process);
-  ClassDB::bind_method(D_METHOD("start"), &AudioCapture::start);
   ClassDB::bind_method(D_METHOD("get_frame"), &AudioCapture::getFrame);
+  ClassDB::bind_method(D_METHOD("get_gain"), &AudioCapture::setGain);
 }
 
 AudioCapture::AudioCapture()
 {
-  UtilityFunctions::print("BEDZIE RUCHANES");
-
   HRESULT hr;
-  IMMDeviceEnumerator* pEnumerator = nullptr;
-  IMMDevice* pDevice = nullptr;
 
   hr = CoInitialize(nullptr);
   if (FAILED(hr)) {
@@ -31,19 +20,19 @@ AudioCapture::AudioCapture()
     return;
   }
 
-  hr = CoCreateInstance(__uuidof(MMDeviceEnumerator), nullptr, CLSCTX_ALL, IID_PPV_ARGS(&pEnumerator));
+  hr = CoCreateInstance(__uuidof(MMDeviceEnumerator), nullptr, CLSCTX_ALL, IID_PPV_ARGS(&m_deviceEnumerator));
   if (FAILED(hr)) {
     UtilityFunctions::printerr("CoCreateInstance failed:", hr);
     return;
   }
 
-  hr = pEnumerator->GetDefaultAudioEndpoint(eRender, eConsole, &pDevice);
+  hr = m_deviceEnumerator->GetDefaultAudioEndpoint(eRender, eConsole, &m_device);
   if (FAILED(hr)) {
     UtilityFunctions::printerr("GetDefaultAudioEndpoint failed:", hr);
     return;
   }
 
-  hr = pDevice->Activate(__uuidof(IAudioClient), CLSCTX_ALL, nullptr, (void**)&m_audioClient);
+  hr = m_device->Activate(__uuidof(IAudioClient), CLSCTX_ALL, nullptr, (void**)&m_audioClient);
   if (FAILED(hr)) {
     UtilityFunctions::printerr("Device activate failed:", hr);
     return;
@@ -68,13 +57,38 @@ AudioCapture::AudioCapture()
     return;
   }
 
-  UtilityFunctions::print("Gites majonez");
+  hr = m_audioClient->Start();
+  if (FAILED(hr)) {
+    UtilityFunctions::printerr("Start failed:", hr);
+    return;
+  };
 }
 
-Array AudioCapture::getFrame() const
+AudioCapture::~AudioCapture()
 {
-  // TODO: common pre-resized buffer?
-  Array out{};
+  if (m_captureClient) {
+    m_captureClient->Release();
+  }
+
+  if (m_audioClient) {
+    m_audioClient->Release();
+  }
+
+  if (m_device) {
+    m_device->Release();
+  }
+
+  if (m_deviceEnumerator) {
+    m_deviceEnumerator->Release();
+  }
+
+  CoUninitialize();
+}
+
+
+PackedVector2Array AudioCapture::getFrame() const
+{
+  PackedVector2Array out{};
 
   UINT32 len = 0;
   UINT32 frames = 0;
@@ -91,12 +105,9 @@ Array AudioCapture::getFrame() const
       for (UINT32 i = 0; i < frames; ++i) {
         float leftChannel = pfData[i * 2];
         float rightChannel = pfData[i * 2 + 1];
-        out.push_back(Vector2(leftChannel, rightChannel));
+        out.push_back(Vector2(leftChannel * m_gain, rightChannel * m_gain));
       }
     }
-    else {
-      //out.push_back(Vector2(0, 0));
-    }
 
     m_captureClient->ReleaseBuffer(frames);
     m_captureClient->GetNextPacketSize(&len);
@@ -105,45 +116,9 @@ Array AudioCapture::getFrame() const
   return out;
 }
 
-void AudioCapture::process()
+void AudioCapture::setGain(float gain)
 {
-  UINT32 len = 0;
-  UINT32 frames = 0;
-  DWORD flags = 0;
-
-  m_captureClient->GetNextPacketSize(&len);
-
-  while (len > 0) {
-    BYTE* pData = nullptr;
-    m_captureClient->GetBuffer(&pData, &frames, &flags, NULL, NULL);
-
-    if (pData && frames > 0) {
-      float* pfData = (float*)pData;
-      for (UINT32 i = 0; i < frames; ++i) {
-        float leftChannel = pfData[i * 2];
-        float rightChannel = pfData[i * 2 + 1];
-        m_audioPlayback->push_frame(Vector2(leftChannel, rightChannel));
-      }
-    }
-    else {
-      m_audioPlayback->push_frame(Vector2(0, 0));
-    }
-
-    m_captureClient->ReleaseBuffer(frames);
-    m_captureClient->GetNextPacketSize(&len);
-  }
-
+  m_gain = gain;
 }
 
-void AudioCapture::start(Ref<AudioStreamGeneratorPlayback> playback)
-{
-  m_audioPlayback = playback;
-
-  const auto hr = m_audioClient->Start();
-  if (FAILED(hr)) {
-    UtilityFunctions::printerr("Start failed:", hr);
-    return;
-  };
 }
-
-}
\ No newline at end of file
diff --git a/src/AudioCapture.h b/src/AudioCapture.h
index 940f589..cbdae31 100644
--- a/src/AudioCapture.h
+++ b/src/AudioCapture.h
@@ -20,20 +20,22 @@ protected:
 
 public:
 	AudioCapture();
+	~AudioCapture();
 
-	Array getFrame() const;
+	PackedVector2Array getFrame() const;
 
-	void process();
-  void start(Ref<AudioStreamGeneratorPlayback> playback);
+	void setGain(float gain);
 
 private:
-	IAudioClient* m_audioClient = nullptr;
-	IAudioCaptureClient* m_captureClient = nullptr;
+	IMMDeviceEnumerator* m_deviceEnumerator{nullptr};
+	IMMDevice* m_device{nullptr};
+	IAudioClient* m_audioClient{nullptr};
+	IAudioCaptureClient* m_captureClient{nullptr};
 
   Ref<AudioStreamGenerator> m_audioStream;
   Ref<AudioStreamGeneratorPlayback> m_audioPlayback;
 
-
+	float m_gain{0.9};
 };
 
 }