SoundLooper.hpp

#ifndef SOUND_LOOPER_HPP
#define SOUND_LOOPER_HPP

#include <QObject>
#include <bass.h>

#include "util.hpp"

class SoundLooper final : public QObject {
	Q_OBJECT
	Q_PROPERTY(bool running READ isRunning NOTIFY runningStateChanged)

	bool running{};
	HRECORD recording{};
	HSTREAM stream{};

	BASS_DX8_ECHO echoParams{};
	BASS_DX8_REVERB reverbParams{};

	HFX echoEffect{};
	HFX reverbEffect{};

	void setRunning(const bool b) {
		running = b;
		emit runningStateChanged(b);
		b ? emit started() : emit stopped();
	}

public:
	[[nodiscard]]
	Q_INVOKABLE bool isRunning() const {
		return running;
	}

	[[nodiscard]]
	Q_INVOKABLE QString getOutputDeviceName() const {
		BASS_DEVICEINFO deviceinfo;
		const auto device = BASS_GetDevice();
		BASS_GetDeviceInfo(device, &deviceinfo);
		return deviceinfo.name;
	}

	[[nodiscard]]
	Q_INVOKABLE QString getInputDeviceName() const {
		BASS_DEVICEINFO deviceinfo;
		const auto device = BASS_RecordGetDevice();
		BASS_RecordGetDeviceInfo(device, &deviceinfo);
		return deviceinfo.name;
	}

	static std::vector<EXTENDED_DEVICEINFO> getAllRecordingDevices() {
		std::vector<EXTENDED_DEVICEINFO> devices;
		EXTENDED_DEVICEINFO info{};

		for (int i = 0; BASS_RecordGetDeviceInfo(i, &info); i++) {
			info.id = i;
			devices.emplace_back(info);
		}

		return devices;
	}

	static std::vector<EXTENDED_DEVICEINFO> getAllOutputDevices() {
		std::vector<EXTENDED_DEVICEINFO> devices;
		EXTENDED_DEVICEINFO info{};

		for (int i = 1; BASS_GetDeviceInfo(i, &info); i++) {
			info.id = i;
			devices.emplace_back(info);
		}

		return devices;
	}

	static DWORD streamRecordProc(HSTREAM, void* buffer, const DWORD length, void* user) {
		const auto _this = static_cast<SoundLooper *>(user);
		return BASS_ChannelGetData(_this->recording, buffer, length);
	}

	Q_INVOKABLE void start() {
		if (not BASS_Init(-1, 96'000, BASS_DEVICE_DEFAULT, nullptr, nullptr))
			exit(-10);

		if (not BASS_RecordInit(0))
			exit(-11);

		recording = BASS_RecordStart(96'000, 1, 0, nullptr, nullptr);
		if (not recording)
			exit(-12);

		stream = BASS_StreamCreate(96'000, 1, 0, &streamRecordProc, this);
		if (not stream)
			exit(-13);

		BASS_ChannelSetAttribute(stream, BASS_ATTRIB_BUFFER, 0);
		BASS_ChannelStart(stream);
		setRunning(true);
	}

	Q_INVOKABLE void stop() {
		BASS_ChannelFree(stream);
		BASS_RecordFree();
		BASS_Free();
		echoEffect = 0;
		setRunning(false);
	}

	Q_INVOKABLE void toggle() {
		running ? stop() : start();
	}

	Q_INVOKABLE void echo_turnOn() {
		if (not running or echoEffect)
			return;

		echoEffect = BASS_ChannelSetFX(stream, BASS_FX_DX8_ECHO, 100);
		BASS_FXSetParameters(echoEffect, &echoParams);
	}

	Q_INVOKABLE void echo_turnOff() {
		if (not running or not echoEffect)
			return;

		BASS_ChannelRemoveFX(stream, echoEffect);
		echoEffect = 0;
	}

	Q_INVOKABLE void echo_setMix(const float value) {
		echoParams.fWetDryMix = value;
		BASS_FXSetParameters(echoEffect, &echoParams);
	}

	Q_INVOKABLE void echo_setFeedback(const float value) {
		echoParams.fFeedback = value;
		BASS_FXSetParameters(echoEffect, &echoParams);
	}

	Q_INVOKABLE void echo_setDelay(const float value) {
		echoParams.fLeftDelay = value;
		echoParams.fRightDelay = value;
		BASS_FXSetParameters(echoEffect, &echoParams);
	}

	Q_INVOKABLE void echo_setLeftDelay(const float value) {
		echoParams.fLeftDelay = value;
		BASS_FXSetParameters(echoEffect, &echoParams);
	}

	Q_INVOKABLE void echo_setRightDelay(const float value) {
		echoParams.fRightDelay = value;
		BASS_FXSetParameters(echoEffect, &echoParams);
	}

	Q_INVOKABLE void echo_setPanDelay(const bool value) {
		echoParams.lPanDelay = value;
		BASS_FXSetParameters(echoEffect, &echoParams);
	}

	Q_INVOKABLE void reverb_turnOn() {
		if (not running or reverbEffect)
			return;

		reverbEffect = BASS_ChannelSetFX(stream, BASS_FX_DX8_REVERB, 90);
		BASS_FXSetParameters(reverbEffect, &reverbParams);
	}

	Q_INVOKABLE void reverb_turnOff() {
		if (not running or not reverbEffect)
			return;

		BASS_ChannelRemoveFX(stream, reverbEffect);
		reverbEffect = 0;
	}

	Q_INVOKABLE void reverb_setGain(const float value) {
		reverbParams.fInGain = value;
		BASS_FXSetParameters(reverbEffect, &reverbParams);
	}

	Q_INVOKABLE void reverb_setMix(const float value) {
		reverbParams.fReverbMix = value;
		BASS_FXSetParameters(reverbEffect, &reverbParams);
	}

	Q_INVOKABLE void reverb_setTime(const float value) {
		reverbParams.fReverbTime = value;
		BASS_FXSetParameters(reverbEffect, &reverbParams);
	}

	Q_INVOKABLE void reverb_setRatio(const float value) {
		reverbParams.fHighFreqRTRatio = value;
		BASS_FXSetParameters(reverbEffect, &reverbParams);
	}

signals:
	void runningStateChanged(bool isRunning);
	void started();
	void stopped();
};

#endif