Skip to content

Commit

Permalink
feat: Refactor audio playback functions
Browse files Browse the repository at this point in the history
  • Loading branch information
@alvarorichard
alvarorichard committed Jul 7, 2024
1 parent 7e2f911 commit 64a3078
Showing 1 changed file with 150 additions and 135 deletions.
285 changes: 150 additions & 135 deletions src/main.c
View file
Original file line number Diff line number Diff line change
@@ -1,5 +1,3 @@


#include <math.h>
#include <ncurses.h>
#include <stdbool.h>
Expand All @@ -20,162 +18,179 @@ bool is_paused = false; // Variable to track pause state
DuckWaveSoundData dw_sdata; // Declare dw_sdata in the global scope

// Function to draw bar graphs and playback time
void draw_bars_and_time(float *data, int count, float current_time,float total_time)
void draw_bars_and_time(float* data, int count, float current_time, float total_time)
{
clear();
int max_height = LINES - 3;

for (int i = 0; i < count; i++) {
smoothed_data[i] = SMOOTHING_FACTOR * smoothed_data[i] +
(1.0 - SMOOTHING_FACTOR) * data[i];
int height = (int)(smoothed_data[i] * max_height);
int color_pair = 1; // Solid color (Cyan)
attron(COLOR_PAIR(color_pair));

for (int j = 0; j < height; j++) {
mvprintw(max_height - j, i * 2,
"#"); // Adjust column width to avoid overlap
}
clear();
int max_height = LINES - 3;

for (int i = 0; i < count; i++)
{
smoothed_data[i] = SMOOTHING_FACTOR * smoothed_data[i] + (1.0 - SMOOTHING_FACTOR) * data[i];
int height = (int)(smoothed_data[i] * max_height);
int color_pair = 1; // Solid color (Cyan)
attron(COLOR_PAIR(color_pair));

for (int j = 0; j < height; j++)
{
mvprintw(max_height - j, i * 2, "#"); // Adjust column width to avoid overlap
}

attroff(COLOR_PAIR(color_pair));
}
attroff(COLOR_PAIR(color_pair));
}

char time_str[128];
char total_time_str[128];
generate_timestamp(current_time, time_str);
generate_timestamp(total_time, total_time_str);
mvprintw(LINES - 1, 0, "Time: %s / %s", time_str, total_time_str);
char time_str[128];
char total_time_str[128];
generate_timestamp(current_time, time_str);
generate_timestamp(total_time, total_time_str);
mvprintw(LINES - 1, 0, "Time: %s / %s", time_str, total_time_str);

refresh();
refresh();
}

// Modified callback to capture audio data
void duckwave_playback_callback_with_visual(ma_device *pDevice, void *pOutput, const void *pInput,unsigned int frameCount)
void duckwave_playback_callback_with_visual(ma_device* pDevice, void* pOutput, const void* pInput, unsigned int frameCount)
{
ma_decoder *pDecoder = pDevice->pUserData;
if (pDecoder == NULL)
return;

if (!is_paused) {
ma_decoder_read_pcm_frames(pDecoder, pOutput, frameCount, NULL);

// Process audio data for visualization
float *audio_data = (float *)pOutput;
float bar_data[NUM_BARS];
int samples_per_bar = frameCount / NUM_BARS;
for (int i = 0; i < NUM_BARS; i++) {
float sum = 0;
for (int j = 0; j < samples_per_bar; j++) {
sum += fabs(audio_data[i * samples_per_bar + j]);
}
bar_data[i] = sum / samples_per_bar;
}
ma_decoder* pDecoder = pDevice->pUserData;
if (pDecoder == NULL)
return;

if (!is_paused)
{
ma_decoder_read_pcm_frames(pDecoder, pOutput, frameCount, NULL);

// Process audio data for visualization
float* audio_data = (float*)pOutput;
float bar_data[NUM_BARS];
int samples_per_bar = frameCount / NUM_BARS;
for (int i = 0; i < NUM_BARS; i++)
{
float sum = 0;
for (int j = 0; j < samples_per_bar; j++)
{
sum += fabs(audio_data[i * samples_per_bar + j]);
}
bar_data[i] = sum / samples_per_bar;
}

song_cursor += (float)frameCount / pDevice->sampleRate;
draw_bars_and_time(bar_data, NUM_BARS, song_cursor, song_duration);
song_cursor += (float)frameCount / pDevice->sampleRate;
draw_bars_and_time(bar_data, NUM_BARS, song_cursor, song_duration);

// Check if the song has finished
if (song_cursor >= song_duration) {
song_finished = true;
ma_device_stop(pDevice); // Stop the device
// Check if the song has finished
if (song_cursor >= song_duration)
{
song_finished = true;
ma_device_stop(pDevice); // Stop the device
}
}
else
{
memset(pOutput, 0, frameCount * ma_get_bytes_per_frame(pDevice->playback.format, pDevice->playback.channels));
}
} else {
memset(pOutput, 0,frameCount * ma_get_bytes_per_frame(pDevice->playback.format,pDevice->playback.channels));
}

(void)pInput;
(void)pInput;
}

// Cleanup function to ensure ncurses is correctly terminated
void cleanup()
{

endwin();
ma_device_uninit(&dw_sdata.device);
fflush(stdout); // Clear the output buffer
{
endwin();
ma_device_uninit(&dw_sdata.device);
fflush(stdout); // Clear the output buffer
}

int main(int argc, char *argv[])
int main(int argc, char* argv[])
{
if (argc < 2) {
printf("Usage: %s <audio_file>\n", argv[0]);
return -1;
}

duckwave_init_file_decoder(&dw_sdata, argv[1]);
duckwave_init_device_playback(&dw_sdata);

// Get the duration of the song
ma_uint64 length_in_pcm_frames;
ma_result result = ma_decoder_get_length_in_pcm_frames(&dw_sdata.decoder,
&length_in_pcm_frames);
if (result != MA_SUCCESS) {
printf("Failed to get length of audio file.\n");
return -1;
}
song_duration =
(float)length_in_pcm_frames / dw_sdata.decoder.outputSampleRate;

// Replace the default callback with the visualization callback
dw_sdata.device_config.dataCallback = duckwave_playback_callback_with_visual;

initscr();
start_color();
cbreak();
noecho();
curs_set(FALSE);
timeout(0); // Allows getch() to be non-blocking

// Register the cleanup function
atexit(cleanup);

// Initialize color pairs
init_pair(1, COLOR_CYAN, COLOR_BLACK);

result = ma_device_init(NULL, &dw_sdata.device_config, &dw_sdata.device);
if (result != MA_SUCCESS) {
endwin(); // Ensure ncurses is terminated before exiting
printf("Failed to initialize playback device.\n");
return -1;
}

result = ma_device_start(&dw_sdata.device);
if (result != MA_SUCCESS) {
endwin(); // Ensure ncurses is terminated before exiting
printf("Failed to start playback device.\n");
ma_device_uninit(&dw_sdata.device);
return -1;
}

while (1) {
// Calculate timeout based on the remaining time of the song
int timeout_ms = (int)((song_duration - song_cursor) * 1000);
timeout(timeout_ms); // Set timeout for getch function

int ch = getch();
if (ch != ERR) {
if (ch == ' ') { // SPACE key pressed
is_paused = !is_paused;
if (is_paused) {
ma_device_stop(&dw_sdata.device);
} else {
ma_device_start(&dw_sdata.device);
}
} else {
break;
}
if (argc < 2)
{
printf("Usage: %s <audio_file>\n", argv[0]);
return -1;
}

duckwave_init_file_decoder(&dw_sdata, argv[1]);
duckwave_init_device_playback(&dw_sdata);

// Get the duration of the song
ma_uint64 length_in_pcm_frames;
ma_result result = ma_decoder_get_length_in_pcm_frames(&dw_sdata.decoder, &length_in_pcm_frames);
if (result != MA_SUCCESS)
{
printf("Failed to get length of audio file.\n");
return -1;
}
song_duration = (float)length_in_pcm_frames / dw_sdata.decoder.outputSampleRate;

// Replace the default callback with the visualization callback
dw_sdata.device_config.dataCallback = duckwave_playback_callback_with_visual;

initscr();
start_color();
cbreak();
noecho();
curs_set(FALSE);
timeout(0); // Allows getch() to be non-blocking

// Register the cleanup function
atexit(cleanup);

// Initialize color pairs
init_pair(1, COLOR_CYAN, COLOR_BLACK);

result = ma_device_init(NULL, &dw_sdata.device_config, &dw_sdata.device);
if (result != MA_SUCCESS)
{
endwin(); // Ensure ncurses is terminated before exiting
printf("Failed to initialize playback device.\n");
return -1;
}

if (song_finished) {
break;
result = ma_device_start(&dw_sdata.device);
if (result != MA_SUCCESS)
{
endwin(); // Ensure ncurses is terminated before exiting
printf("Failed to start playback device.\n");
ma_device_uninit(&dw_sdata.device);
return -1;
}
}

cleanup();
while (1)
{
// Calculate timeout based on the remaining time of the song
int timeout_ms = (int)((song_duration - song_cursor) * 1000);
timeout(timeout_ms); // Set timeout for getch function

int ch = getch();
if (ch != ERR)
{
if (ch == ' ')
{ // SPACE key pressed
is_paused = !is_paused;
if (is_paused)
{
ma_device_stop(&dw_sdata.device);
}
else
{
ma_device_start(&dw_sdata.device);
}
}
else
{
break;
}
}

if (song_finished) {
printf("Playback finished. Exiting...\n");
}
if (song_finished)
{
break;
}
}

cleanup();

if (song_finished)
{
printf("Playback finished. Exiting...\n");
}

return 0;
return 0;
}

0 comments on commit 64a3078

Please sign in to comment.