基于GEC210的语音识别控制

/* @file @brief record demo for linux @author taozhang9 @date 2016/05/27 */ #include <stdio.h> #include <stdlib.h> #include <fcntl.h> #include <alsa/asoundlib.h> #include <signal.h> #include <sys/stat.h> #include <pthread.h> #include "formats.h" #include "linuxrec.h" #define DBG_ON 1 #if DBG_ON #define dbg printf #else #define dbg #endif /* Do not change the sequence */ enum { RECORD_STATE_CREATED, /* Init */ RECORD_STATE_CLOSING, RECORD_STATE_READY, /* Opened */ RECORD_STATE_STOPPING, /* During Stop */ RECORD_STATE_RECORDING, /* Started */ }; #define SAMPLE_RATE 16000 #define SAMPLE_BIT_SIZE 16 #define FRAME_CNT 10 //#define BUF_COUNT 1 #define DEF_BUFF_TIME 500000 #define DEF_PERIOD_TIME 100000 #define DEFAULT_FORMAT \ {\ WAVE_FORMAT_PCM, \ 1, \ 16000, \ 32000, \ 2, \ 16, \ sizeof(WAVEFORMATEX) \ } #if 0 struct bufinfo { char *data; unsigned int bufsize; }; #endif static int show_xrun = 1; static int start_record_internal(snd_pcm_t *pcm) { return snd_pcm_start(pcm); } static int stop_record_internal(snd_pcm_t *pcm) { return snd_pcm_drop(pcm); } static int is_stopped_internal(struct recorder *rec) { snd_pcm_state_t state; state = snd_pcm_state((snd_pcm_t *)rec->wavein_hdl); switch (state) { case SND_PCM_STATE_RUNNING: case SND_PCM_STATE_DRAINING: return 0; default: break; } return 1; } static int format_ms_to_alsa(const WAVEFORMATEX * wavfmt, snd_pcm_format_t * format) { snd_pcm_format_t tmp; tmp = snd_pcm_build_linear_format(wavfmt->wBitsPerSample, wavfmt->wBitsPerSample, wavfmt->wBitsPerSample == 8 ? 1 : 0, 0); if ( tmp == SND_PCM_FORMAT_UNKNOWN ) return -EINVAL; *format = tmp; return 0; } /* set hardware and software params */ static int set_hwparams(struct recorder * rec, const WAVEFORMATEX *wavfmt, unsigned int buffertime, unsigned int periodtime) { snd_pcm_hw_params_t *params; int err; unsigned int rate; snd_pcm_format_t format; snd_pcm_uframes_t size; snd_pcm_t *handle = (snd_pcm_t *)rec->wavein_hdl; rec->buffer_time = buffertime; rec->period_time = periodtime; snd_pcm_hw_params_alloca(&params); err = snd_pcm_hw_params_any(handle, params); if (err < 0) { dbg("Broken configuration for this PCM"); return err; } err = snd_pcm_hw_params_set_access(handle, params, SND_PCM_ACCESS_RW_INTERLEAVED); if (err < 0) { dbg("Access type not available"); return err; } err = format_ms_to_alsa(wavfmt, &format); if (err) { dbg("Invalid format"); return - EINVAL; } err = snd_pcm_hw_params_set_format(handle, params, format); if (err < 0) { dbg("Sample format non available"); return err; } err = snd_pcm_hw_params_set_channels(handle, params, wavfmt->nChannels); if (err < 0) { dbg("Channels count non available"); return err; } rate = wavfmt->nSamplesPerSec; err = snd_pcm_hw_params_set_rate_near(handle, params, &rate, 0); if (err < 0) { dbg("Set rate failed"); return err; } if(rate != wavfmt->nSamplesPerSec) { dbg("Rate mismatch"); return -EINVAL; } if (rec->buffer_time == 0 || rec->period_time == 0) { err = snd_pcm_hw_params_get_buffer_time_max(params, &rec->buffer_time, 0); assert(err >= 0); if (rec->buffer_time > 500000) rec->buffer_time = 500000; rec->period_time = rec->buffer_time / 4; } err = snd_pcm_hw_params_set_period_time_near(handle, params, &rec->period_time, 0); if (err < 0) { dbg("set period time fail"); return err; } err = snd_pcm_hw_params_set_buffer_time_near(handle, params, &rec->buffer_time, 0); if (err < 0) { dbg("set buffer time failed"); return err; } err = snd_pcm_hw_params_get_period_size(params, &size, 0); if (err < 0) { dbg("get period size fail"); return err; } rec->period_frames = size; err = snd_pcm_hw_params_get_buffer_size(params, &size); if (size == rec->period_frames) { dbg("Can't use period equal to buffer size (%lu == %lu)", size, (unsigned long)rec->period_frames); return -EINVAL; } rec->buffer_frames = size; rec->bits_per_frame = wavfmt->wBitsPerSample; /* set to driver */ err = snd_pcm_hw_params(handle, params); if (err < 0) { dbg("Unable to install hw params:"); return err; } return 0; } static int set_swparams(struct recorder * rec) { int err; snd_pcm_sw_params_t *swparams; snd_pcm_t * handle = (snd_pcm_t*)(rec->wavein_hdl); /* sw para */ snd_pcm_sw_params_alloca(&swparams); err = snd_pcm_sw_params_current(handle, swparams); if (err < 0) { dbg("get current sw para fail"); return err; } err = snd_pcm_sw_params_set_avail_min(handle, swparams, rec->period_frames); if (err < 0) { dbg("set avail min failed"); return err; } /* set a value bigger than the buffer frames to prevent the auto start. * we use the snd_pcm_start to explicit start the pcm */ err = snd_pcm_sw_params_set_start_threshold(handle, swparams, rec->buffer_frames * 2); if (err < 0) { dbg("set start threshold fail"); return err; } if ( (err = snd_pcm_sw_params(handle, swparams)) < 0) { dbg("unable to install sw params:"); return err; } return 0; } static int set_params(struct recorder *rec, WAVEFORMATEX *fmt, unsigned int buffertime, unsigned int periodtime) { int err; WAVEFORMATEX defmt = DEFAULT_FORMAT; if (fmt == NULL) { fmt = &defmt; } err = set_hwparams(rec, fmt, buffertime, periodtime); if (err) return err; err = set_swparams(rec); if (err) return err; return 0; } /* * Underrun and suspend recovery */ static int xrun_recovery(snd_pcm_t *handle, int err) { if (err == -EPIPE) { /* over-run */ if (show_xrun) printf("!!!!!!overrun happend!!!!!!"); err = snd_pcm_prepare(handle); if (err < 0) { if (show_xrun) printf("Can't recovery from overrun," "prepare failed: %s\n", snd_strerror(err)); return err; } return 0; } else if (err == -ESTRPIPE) { while ((err = snd_pcm_resume(handle)) == -EAGAIN) usleep(200000); /* wait until the suspend flag is released */ if (err < 0) { err = snd_pcm_prepare(handle); if (err < 0) { if (show_xrun) printf("Can't recovery from suspend," "prepare failed: %s\n", snd_strerror(err)); return err; } } return 0; } return err; } static ssize_t pcm_read(struct recorder *rec, size_t rcount) { ssize_t r; size_t count = rcount; char *data; snd_pcm_t *handle = (snd_pcm_t *)rec->wavein_hdl; if(!handle) return -EINVAL; data = rec->audiobuf; while (count > 0) { r = snd_pcm_readi(handle, data, count); if (r == -EAGAIN || (r >= 0 && (size_t)r < count)) { snd_pcm_wait(handle, 100); } else if (r < 0) { if(xrun_recovery(handle, r) < 0) { return -1; } } if (r > 0) { count -= r; data += r * rec->bits_per_frame / 8; } } return rcount; } static void * record_thread_proc(void * para) { struct recorder * rec = (struct recorder *) para; size_t frames, bytes; sigset_t mask, oldmask; sigemptyset(&mask); sigaddset(&mask, SIGINT); sigaddset(&mask, SIGTERM); pthread_sigmask(SIG_BLOCK, &mask, &oldmask); while(1) { frames = rec->period_frames; bytes = frames * rec->bits_per_frame / 8; /* closing, exit the thread */ if (rec->state == RECORD_STATE_CLOSING) break; if(rec->state < RECORD_STATE_RECORDING) usleep(100000); if (pcm_read(rec, frames) != frames) { return NULL; } if (rec->on_data_ind) rec->on_data_ind(rec->audiobuf, bytes, rec->user_cb_para); } return rec; } static int create_record_thread(void * para, pthread_t * tidp) { int err; err = pthread_create(tidp, NULL, record_thread_proc, (void *)para); if (err != 0) return err; ret