From 969ba7919d2fccd7b52688b446f73d8248b5e1f4 Mon Sep 17 00:00:00 2001 From: Taylor Holberton Date: Sat, 19 Nov 2016 10:08:38 -0800 Subject: created cmd and ctx structures for tinyplay --- utils/tinyplay.c | 439 ++++++++++++++++++++++++++++++++++--------------------- 1 file changed, 270 insertions(+), 169 deletions(-) (limited to 'utils') diff --git a/utils/tinyplay.c b/utils/tinyplay.c index 1c4e23f..db1b611 100644 --- a/utils/tinyplay.c +++ b/utils/tinyplay.c @@ -33,6 +33,116 @@ #include #include +struct cmd { + const char *filename; + const char *filetype; + unsigned int card; + unsigned int device; + int flags; + struct pcm_config config; + unsigned int bits; +}; + +void cmd_init(struct cmd *cmd) +{ + cmd->filename = NULL; + cmd->filetype = NULL; + cmd->card = 0; + cmd->device = 0; + cmd->flags = PCM_OUT; + cmd->config.period_size = 1024; + cmd->config.period_count = 2; + cmd->config.channels = 2; + cmd->config.rate = 48000; + cmd->config.format = PCM_FORMAT_S16_LE; + cmd->config.silence_threshold = 1024 * 2; + cmd->config.stop_threshold = 1024 * 2; + cmd->config.start_threshold = 1024; + cmd->bits = 16; +} + +int cmd_parse_arg(struct cmd *cmd, int argc, const char **argv) +{ + if (argc < 1) { + return 0; + } + + if ((strcmp(argv[0], "-M") == 0) || (strcmp(argv[0], "--mmap") == 0)) { + cmd->flags |= PCM_MMAP; + return 1; + } + + if (argv[0][0] != '-') { + cmd->filename = argv[0]; + return 1; + } + + if (argc < 2) { + fprintf(stderr, "option '%s' is missing argument\n", argv[0]); + return -1; + } + + if ((strcmp(argv[0], "-D") == 0) || (strcmp(argv[0], "--card") == 0)) { + if (sscanf(argv[1], "%u", &cmd->card) != 1) { + fprintf(stderr, "failed parsing card number '%s'\n", argv[1]); + return -1; + } + } else if ((strcmp(argv[0], "-d") == 0) || (strcmp(argv[0], "--device") == 0)) { + if (sscanf(argv[1], "%u", &cmd->device) != 1) { + fprintf(stderr, "failed parsing device number '%s'\n", argv[1]); + return -1; + } + } else if ((strcmp(argv[0], "-p") == 0) || (strcmp(argv[0], "--period-size") == 0)) { + if (sscanf(argv[1], "%u", &cmd->config.period_size) != 1) { + fprintf(stderr, "failed parsing period size '%s'\n", argv[1]); + return -1; + } + } else if ((strcmp(argv[0], "-p") == 0) || (strcmp(argv[0], "--period-count") == 0)) { + if (sscanf(argv[1], "%u", &cmd->config.period_count) != 1) { + fprintf(stderr, "failed parsing period count '%s'\n", argv[1]); + return -1; + } + } else if ((strcmp(argv[0], "-c") == 0) || (strcmp(argv[0], "--channels") == 0)) { + if (sscanf(argv[1], "%u", &cmd->config.channels) != 1) { + fprintf(stderr, "failed parsing channel count '%s'\n", argv[1]); + return -1; + } + } else if ((strcmp(argv[0], "-r") == 0) || (strcmp(argv[0], "--rate") == 0)) { + if (sscanf(argv[1], "%u", &cmd->config.rate) != 1) { + fprintf(stderr, "failed parsing rate '%s'\n", argv[1]); + return -1; + } + } else if ((strcmp(argv[0], "-i") == 0) || (strcmp(argv[0], "--file-type") == 0)) { + cmd->filetype = argv[1]; + } else { + fprintf(stderr, "unknown option '%s'\n", argv[0]); + return -1; + } + return 2; +} + +int cmd_parse_args(struct cmd *cmd, int argc, const char **argv) +{ + int i = 0; + while (i < argc) { + int j = cmd_parse_arg(cmd, argc - i, &argv[i]); + if (j < 0){ + break; + } + i += j; + } + + if ((cmd->filename != NULL) + && (cmd->filetype == NULL)) { + cmd->filetype = strrchr(cmd->filename, '.'); + if (cmd->filetype != NULL) { + cmd->filetype++; + } + } + + return i; +} + #define ID_RIFF 0x46464952 #define ID_WAVE 0x45564157 #define ID_FMT 0x20746d66 @@ -58,135 +168,61 @@ struct chunk_fmt { uint16_t bits_per_sample; }; -static int close = 0; +struct ctx { + struct pcm *pcm; -void play_sample(FILE *file, unsigned int card, unsigned int device, unsigned int channels, - unsigned int rate, unsigned int bits, unsigned int period_size, - unsigned int period_count); + struct riff_wave_header wave_header; + struct chunk_header chunk_header; + struct chunk_fmt chunk_fmt; -void stream_close(int sig) -{ - /* allow the stream to be closed gracefully */ - signal(sig, SIG_IGN); - close = 1; -} + FILE *file; +}; -int main(int argc, char **argv) +int ctx_init(struct ctx* ctx, const struct cmd *cmd) { - FILE *file; - struct riff_wave_header riff_wave_header; - struct chunk_header chunk_header; - struct chunk_fmt chunk_fmt; - unsigned int device = 0; - unsigned int card = 0; - unsigned int period_size = 1024; - unsigned int period_count = 4; - unsigned int channels = 2; - unsigned int rate = 48000; - unsigned int bits = 16; - unsigned int is_raw = 0; /* Default wav file */ - char *filename; - int more_chunks = 1; + unsigned int bits = cmd->bits; + struct pcm_config config = cmd->config; - if (argc < 2) { - fprintf(stderr, "Usage1: %s file.wav [-D card] [-d device] [-p period_size]" - " [-n n_periods] \n", argv[0]); - fprintf(stderr, "Usage2: %s file.raw [-D card] [-d device] [-p period_size] " - "[-n n_periods] [-c channels] [-r rate] [-b bits] -i raw \n", argv[0]); - return 1; + if (cmd->filename == NULL) { + fprintf(stderr, "filename not specified\n"); + return -1; } - filename = argv[1]; - file = fopen(filename, "rb"); - if (!file) { - fprintf(stderr, "Unable to open file '%s'\n", filename); - return 1; + ctx->file = fopen(cmd->filename, "rb"); + if (ctx->file == NULL) { + fprintf(stderr, "failed to open '%s'\n", cmd->filename); + return -1; } - /* parse command line arguments */ - argv += 2; - while (*argv) { - if (strcmp(*argv, "-d") == 0) { - argv++; - if (*argv) - device = atoi(*argv); - } - if (strcmp(*argv, "-p") == 0) { - argv++; - if (*argv) - period_size = atoi(*argv); - } - if (strcmp(*argv, "-n") == 0) { - argv++; - if (*argv) - period_count = atoi(*argv); - } - if (strcmp(*argv, "-D") == 0) { - argv++; - if (*argv) - card = atoi(*argv); - } - if (strcmp(*argv, "-c") == 0) { - argv++; - if (*argv) - channels = atoi(*argv); - } - if (strcmp(*argv, "-r") == 0) { - argv++; - if (*argv) - rate = atoi(*argv); + if ((cmd->filetype != NULL) && (strcmp(cmd->filetype, "wav") == 0)) { + if (fread(&ctx->wave_header, sizeof(ctx->wave_header), 1, ctx->file) != 1){ + fprintf(stderr, "error: '%s' does not contain a riff/wave header\n", cmd->filename); + fclose(ctx->file); + return -1; } - if (strcmp(*argv, "-b") == 0) { - argv++; - if (*argv) - bits = atoi(*argv); - } - if (strcmp(*argv, "-i") == 0) { - argv++; - if (*argv) { - if (strcasecmp(*argv, "raw") == 0) { - is_raw = 1; - } else if (strcasecmp(*argv, "wav") == 0) { - is_raw = 0; - } else { - fprintf(stderr, "Error: file type '%s' not supported\n", *argv); - fclose(file); - return 1; - } - } - } - if (*argv) - argv++; - } - - if ( !is_raw ) { - if (fread(&riff_wave_header, sizeof(riff_wave_header), 1, file) != 1){ - fprintf(stderr, "Error: '%s' does not contain a riff/wave header\n", filename); - fclose(file); - return 1; - } - if ((riff_wave_header.riff_id != ID_RIFF) || - (riff_wave_header.wave_id != ID_WAVE)) { - fprintf(stderr, "Error: '%s' is not a riff/wave file\n", filename); - fclose(file); - return 1; + if ((ctx->wave_header.riff_id != ID_RIFF) || + (ctx->wave_header.wave_id != ID_WAVE)) { + fprintf(stderr, "error: '%s' is not a riff/wave file\n", cmd->filename); + fclose(ctx->file); + return -1; } + unsigned int more_chunks = 1; do { - if (fread(&chunk_header, sizeof(chunk_header), 1, file) != 1){ - fprintf(stderr, "Error: '%s' does not contain a data chunk\n", filename); - fclose(file); - return 1; + if (fread(&ctx->chunk_header, sizeof(ctx->chunk_header), 1, ctx->file) != 1){ + fprintf(stderr, "error: '%s' does not contain a data chunk\n", cmd->filename); + fclose(ctx->file); + return -1; } - switch (chunk_header.id) { + switch (ctx->chunk_header.id) { case ID_FMT: - if (fread(&chunk_fmt, sizeof(chunk_fmt), 1, file) != 1){ - fprintf(stderr, "Error: '%s' has incomplete format chunk\n", filename); - fclose(file); - return 1; + if (fread(&ctx->chunk_fmt, sizeof(ctx->chunk_fmt), 1, ctx->file) != 1){ + fprintf(stderr, "error: '%s' has incomplete format chunk\n", cmd->filename); + fclose(ctx->file); + return -1; } /* If the format header is larger, skip the rest */ - if (chunk_header.sz > sizeof(chunk_fmt)) - fseek(file, chunk_header.sz - sizeof(chunk_fmt), SEEK_CUR); + if (ctx->chunk_header.sz > sizeof(ctx->chunk_fmt)) + fseek(ctx->file, ctx->chunk_header.sz - sizeof(ctx->chunk_fmt), SEEK_CUR); break; case ID_DATA: /* Stop looking for chunks */ @@ -194,21 +230,120 @@ int main(int argc, char **argv) break; default: /* Unknown chunk, skip bytes */ - fseek(file, chunk_header.sz, SEEK_CUR); + fseek(ctx->file, ctx->chunk_header.sz, SEEK_CUR); } } while (more_chunks); - channels = chunk_fmt.num_channels; - rate = chunk_fmt.sample_rate; - bits = chunk_fmt.bits_per_sample; + config.channels = ctx->chunk_fmt.num_channels; + config.rate = ctx->chunk_fmt.sample_rate; + bits = ctx->chunk_fmt.bits_per_sample; } - play_sample(file, card, device, channels, rate, bits, period_size, period_count); + if (bits == 8) { + config.format = PCM_FORMAT_S8; + } else if (bits == 16) { + config.format = PCM_FORMAT_S16_LE; + } else if (bits == 24) { + config.format = PCM_FORMAT_S24_3LE; + } else if (bits == 32) { + config.format = PCM_FORMAT_S32_LE; + } else { + fprintf(stderr, "bit count '%u' not supported\n", bits); + fclose(ctx->file); + return -1; + } - fclose(file); + ctx->pcm = pcm_open(cmd->card, + cmd->device, + cmd->flags, + &config); + if (ctx->pcm == NULL) { + fprintf(stderr, "failed to allocate memory for pcm\n"); + fclose(ctx->file); + return -1; + } else if (!pcm_is_ready(ctx->pcm)) { + fprintf(stderr, "failed to open for pcm %u,%u\n", cmd->card, cmd->device); + fclose(ctx->file); + pcm_close(ctx->pcm); + return -1; + } return 0; } +void ctx_free(struct ctx *ctx) +{ + if (ctx == NULL) { + return; + } + if (ctx->pcm != NULL) { + pcm_close(ctx->pcm); + } + if (ctx->file != NULL) { + fclose(ctx->file); + } +} + +static int close = 0; + +int play_sample(struct ctx *ctx); + +void stream_close(int sig) +{ + /* allow the stream to be closed gracefully */ + signal(sig, SIG_IGN); + close = 1; +} + +void print_usage(const char *argv0) +{ + fprintf(stderr, "usage: %s file.wav [options]\n", argv0); + fprintf(stderr, "options:\n"); + fprintf(stderr, "-D | --card The device to receive the audio\n"); + fprintf(stderr, "-d | --device The card to receive the audio\n"); + fprintf(stderr, "-p | --period-size The size of the PCM's period\n"); + fprintf(stderr, "-n | --period-count The number of PCM periods\n"); + fprintf(stderr, "-i | --file-type The type of file to read (raw or wav)\n"); + fprintf(stderr, "-c | --channels The amount of channels per frame\n"); + fprintf(stderr, "-r | --rate The amount of frames per second\n"); + fprintf(stderr, "-b | --bits The number of bits in one sample\n"); + fprintf(stderr, "-M | --mmap Use memory mapped IO to play audio\n"); +} + +int main(int argc, const char **argv) +{ + struct cmd cmd; + struct ctx ctx; + + if (argc < 2) { + print_usage(argv[0]); + return EXIT_FAILURE; + } + + cmd_init(&cmd); + if (cmd_parse_args(&cmd, argc - 1, &argv[1]) < 0) { + return EXIT_FAILURE; + } + + if (ctx_init(&ctx, &cmd) < 0) { + return EXIT_FAILURE; + } + + /* TODO get parameters from context */ + printf("playing '%s': %u ch, %u hz, %u bit\n", + cmd.filename, + cmd.config.channels, + cmd.config.rate, + cmd.bits); + + if (play_sample(&ctx) < 0) { + ctx_free(&ctx); + return EXIT_FAILURE; + } + + ctx_free(&ctx); + return EXIT_SUCCESS; +} + int check_param(struct pcm_params *params, unsigned int param, unsigned int value, char *param_name, char *param_unit) { @@ -233,89 +368,55 @@ int check_param(struct pcm_params *params, unsigned int param, unsigned int valu return is_within_bounds; } -int sample_is_playable(unsigned int card, unsigned int device, unsigned int channels, - unsigned int rate, unsigned int bits, unsigned int period_size, - unsigned int period_count) +int sample_is_playable(const struct cmd *cmd) { struct pcm_params *params; int can_play; - params = pcm_params_get(card, device, PCM_OUT); + params = pcm_params_get(cmd->card, cmd->device, PCM_OUT); if (params == NULL) { - fprintf(stderr, "Unable to open PCM device %u.\n", device); + fprintf(stderr, "unable to open PCM %u,%u\n", cmd->card, cmd->device); return 0; } - can_play = check_param(params, PCM_PARAM_RATE, rate, "Sample rate", "Hz"); - can_play &= check_param(params, PCM_PARAM_CHANNELS, channels, "Sample", " channels"); - can_play &= check_param(params, PCM_PARAM_SAMPLE_BITS, bits, "Bitrate", " bits"); - can_play &= check_param(params, PCM_PARAM_PERIOD_SIZE, period_size, "Period size", "Hz"); - can_play &= check_param(params, PCM_PARAM_PERIODS, period_count, "Period count", "Hz"); + can_play = check_param(params, PCM_PARAM_RATE, cmd->config.rate, "sample rate", "hz"); + can_play &= check_param(params, PCM_PARAM_CHANNELS, cmd->config.channels, "sample", " channels"); + can_play &= check_param(params, PCM_PARAM_SAMPLE_BITS, cmd->bits, "bits", " bits"); + can_play &= check_param(params, PCM_PARAM_PERIOD_SIZE, cmd->config.period_size, "period size", ""); + can_play &= check_param(params, PCM_PARAM_PERIODS, cmd->config.period_count, "period count", ""); pcm_params_free(params); return can_play; } -void play_sample(FILE *file, unsigned int card, unsigned int device, unsigned int channels, - unsigned int rate, unsigned int bits, unsigned int period_size, - unsigned int period_count) +int play_sample(struct ctx *ctx) { - struct pcm_config config; - struct pcm *pcm; char *buffer; int size; int num_read; - memset(&config, 0, sizeof(config)); - config.channels = channels; - config.rate = rate; - config.period_size = period_size; - config.period_count = period_count; - if (bits == 32) - config.format = PCM_FORMAT_S32_LE; - else if (bits == 16) - config.format = PCM_FORMAT_S16_LE; - config.start_threshold = 0; - config.stop_threshold = 0; - config.silence_threshold = 0; - - if (!sample_is_playable(card, device, channels, rate, bits, period_size, period_count)) { - return; - } - - pcm = pcm_open(card, device, PCM_OUT, &config); - if (!pcm || !pcm_is_ready(pcm)) { - fprintf(stderr, "Unable to open PCM device %u (%s)\n", - device, pcm_get_error(pcm)); - return; - } - - size = pcm_frames_to_bytes(pcm, pcm_get_buffer_size(pcm)); + size = pcm_frames_to_bytes(ctx->pcm, pcm_get_buffer_size(ctx->pcm)); buffer = malloc(size); if (!buffer) { - fprintf(stderr, "Unable to allocate %d bytes\n", size); - free(buffer); - pcm_close(pcm); - return; + fprintf(stderr, "unable to allocate %d bytes\n", size); + return -1; } - printf("Playing sample: %u ch, %u hz, %u bit\n", channels, rate, bits); - /* catch ctrl-c to shutdown cleanly */ signal(SIGINT, stream_close); do { - num_read = fread(buffer, 1, size, file); + num_read = fread(buffer, 1, size, ctx->file); if (num_read > 0) { - if (pcm_write(pcm, buffer, num_read)) { - fprintf(stderr, "Error playing sample\n"); + if (pcm_write(ctx->pcm, buffer, num_read)) { + fprintf(stderr, "error playing sample\n"); break; } } } while (!close && num_read > 0); free(buffer); - pcm_close(pcm); + return 0; } -- cgit v1.2.3