73 lines
1.5 KiB
C++
73 lines
1.5 KiB
C++
#pragma once
|
|
|
|
#include <BAN/Array.h>
|
|
#include <BAN/ByteSpan.h>
|
|
#include <BAN/CircularQueue.h>
|
|
#include <BAN/HashMap.h>
|
|
|
|
#include <LibAudio/Audio.h>
|
|
#include <LibAudio/Protocol.h>
|
|
|
|
struct AudioDevice
|
|
{
|
|
int fd;
|
|
uint32_t channels;
|
|
uint32_t sample_rate;
|
|
uint32_t total_pins;
|
|
uint32_t current_pin;
|
|
};
|
|
|
|
class AudioServer
|
|
{
|
|
BAN_NON_MOVABLE(AudioServer);
|
|
BAN_NON_COPYABLE(AudioServer);
|
|
|
|
public:
|
|
AudioServer(BAN::Vector<AudioDevice>&& audio_devices);
|
|
|
|
BAN::ErrorOr<void> on_new_client(int fd);
|
|
void on_client_disconnect(int fd);
|
|
bool on_client_packet(int fd, LibAudio::Packet);
|
|
|
|
uint64_t update();
|
|
|
|
private:
|
|
AudioDevice& device() { return m_audio_devices[m_current_audio_device]; }
|
|
|
|
private:
|
|
struct ClientInfo
|
|
{
|
|
LibAudio::AudioBuffer* buffer;
|
|
size_t queued_head { 0 };
|
|
|
|
size_t sample_frames_queued() const
|
|
{
|
|
return ((buffer->capacity + queued_head - buffer->tail) % buffer->capacity) / buffer->channels;
|
|
}
|
|
|
|
size_t sample_frames_available() const
|
|
{
|
|
return ((buffer->capacity + buffer->head - queued_head) % buffer->capacity) / buffer->channels;
|
|
}
|
|
};
|
|
|
|
using sample_t = LibAudio::AudioBuffer::sample_t;
|
|
|
|
private:
|
|
enum class AddOrRemove { Add, Remove };
|
|
|
|
void reset_kernel_buffer();
|
|
|
|
void send_samples();
|
|
|
|
private:
|
|
BAN::Vector<AudioDevice> m_audio_devices;
|
|
size_t m_current_audio_device { 0 };
|
|
|
|
size_t m_samples_sent { 0 };
|
|
BAN::Array<uint8_t, 4 * 1024> m_send_buffer;
|
|
BAN::CircularQueue<sample_t, 64 * 1024> m_samples;
|
|
|
|
BAN::HashMap<int, ClientInfo> m_audio_buffers;
|
|
};
|