This commit is contained in:
dec05eba 2020-03-29 09:10:24 +02:00
parent bcb79582c5
commit 7c8c97b1a8
3 changed files with 115 additions and 1 deletions

27
include/sound.hpp Normal file
View File

@ -0,0 +1,27 @@
#ifndef GPU_SCREEN_RECORDER_H
#define GPU_SCREEN_RECORDER_H
typedef struct {
void *handle;
char *buffer;
int buffer_size;
unsigned int frames;
} SoundDevice;
/*
Get a sound device by name, returning the device into the @device parameter.
The device should be closed with @sound_device_close after it has been used
to clean up internal resources.
Returns 0 on success, or a negative value on failure.
*/
int sound_device_get_by_name(SoundDevice *device, const char *name = "default", unsigned int num_channels = 1, unsigned int period_frame_size = 32);
int sound_device_close(SoundDevice *device);
/*
Returns the next chunk of audio into @buffer.
Returns the size of the buffer, or a negative value on failure.
*/
int sound_device_read_next_chunk(SoundDevice *device, char **buffer);
#endif /* GPU_SCREEN_RECORDER_H */

View File

@ -6,6 +6,7 @@ platforms = ["posix"]
[config] [config]
include_dirs = ["/opt/cuda/targets/x86_64-linux/include"] include_dirs = ["/opt/cuda/targets/x86_64-linux/include"]
libs = ["/usr/lib/libcuda.so"]
[dependencies] [dependencies]
ffnvcodec = ">=9" ffnvcodec = ">=9"
@ -20,4 +21,6 @@ xcomposite = ">=0.2"
xdamage = "1" xdamage = "1"
# TODO: Remove this dependency, this is needed right now for glfwMakeContextCurrent # TODO: Remove this dependency, this is needed right now for glfwMakeContextCurrent
glfw3 = "3" glfw3 = "3"
alsa = "1"

84
src/sound.cpp Normal file
View File

@ -0,0 +1,84 @@
#include "../include/sound.hpp"
#include <stdlib.h>
#include <stdio.h>
#define ALSA_PCM_NEW_HW_PARAMS_API
#include <alsa/asoundlib.h>
int sound_device_get_by_name(SoundDevice *device, const char *name, unsigned int num_channels, unsigned int period_frame_size) {
int rc;
snd_pcm_t *handle;
rc = snd_pcm_open(&handle, name, SND_PCM_STREAM_CAPTURE, 0);
if(rc < 0) {
fprintf(stderr, "unable to open pcm device 'default', reason: %s\n", snd_strerror(rc));
return rc;
}
snd_pcm_hw_params_t *params;
snd_pcm_hw_params_alloca(&params);
// Fill the params with default values
snd_pcm_hw_params_any(handle, params);
// Interleaved mode
snd_pcm_hw_params_set_access(handle, params, SND_PCM_ACCESS_RW_INTERLEAVED);
// Signed 16--bit little-endian format
snd_pcm_hw_params_set_format(handle, params, SND_PCM_FORMAT_S16_LE);
snd_pcm_hw_params_set_channels(handle, params, num_channels);
// 44100 bits/second samling rate (CD quality)
unsigned int val = 44100;
int dir;
snd_pcm_hw_params_set_rate_near(handle, params, &val, &dir);
snd_pcm_uframes_t frames = period_frame_size;
snd_pcm_hw_params_set_period_size_near(handle, params, &frames, &dir);
// Write the parmeters to the driver
rc = snd_pcm_hw_params(handle, params);
if(rc < 0) {
fprintf(stderr, "unable to set hw parameters, reason: %s\n", snd_strerror(rc));
snd_pcm_close(handle);
return rc;
}
// Use a buffer large enough to hold one period
snd_pcm_hw_params_get_period_size(params, &frames, &dir);
int buffer_size = frames * 2 * num_channels; // 2 bytes/sample, @num_channels channels
char *buffer = (char*)malloc(buffer_size);
if(!buffer) {
fprintf(stderr, "failed to allocate buffer for audio\n");
snd_pcm_close(handle);
return -1;
}
device->handle = (void*)handle;
device->buffer = buffer;
device->buffer_size = buffer_size;
device->frames = frames;
return 0;
}
int sound_device_close(SoundDevice *device) {
/* TODO: Is this also needed in @sound_device_get_by_name on failure? */
snd_pcm_drain((snd_pcm_t*)device->handle);
snd_pcm_close((snd_pcm_t*)device->handle);
free(device->buffer);
}
int sound_device_read_next_chunk(SoundDevice *device, char **buffer) {
int rc = snd_pcm_readi((snd_pcm_t*)device->handle, device->buffer, device->frames);
if (rc == -EPIPE) {
/* overrun */
fprintf(stderr, "overrun occured\n");
snd_pcm_prepare((snd_pcm_t*)device->handle);
return 0;
} else if(rc < 0) {
fprintf(stderr, "failed to read from sound device, reason: %s\n", snd_strerror(rc));
return rc;
} else if (rc != (int)device->frames) {
fprintf(stderr, "short read, read %d frames\n", rc);
}
*buffer = device->buffer;
return rc;
}