Add native 16kHz format for wav file format.

(Closes issue #13657)


git-svn-id: http://svn.digium.com/svn/asterisk/trunk@148069 f38db490-d61c-443f-a65b-d21fe96a405b

1 files changed, 71 insertions, 12 deletions

diff --git a/formats/format_wav.c b/formats/format_wav.c
index 5cf39ce99..6bee6b4b4 100644
--- a/formats/format_wav.c
+++ b/formats/format_wav.c
@@ -39,6 +39,7 @@ ASTERISK_FILE_VERSION(__FILE__, "$Revision$")
 #define	WAV_BUF_SIZE	320
 
 struct wav_desc {	/* format-specific parameters */
+	int which;
 	int bytes;
 	int lasttimeout;
 	int maxlen;
@@ -70,7 +71,7 @@ struct wav_desc {	/* format-specific parameters */
 #endif
 
 
-static int check_header(FILE *f)
+static int check_header(FILE *f, int which)
 {
 	int type, size, formtype;
 	int fmt, hsize;
@@ -135,7 +136,10 @@ static int check_header(FILE *f)
 		ast_log(LOG_WARNING, "Read failed (freq)\n");
 		return -1;
 	}
-	if (ltohl(freq) != DEFAULT_SAMPLE_RATE) {
+	if (ltohl(freq) != DEFAULT_SAMPLE_RATE && which == 8000) {
+		ast_log(LOG_WARNING, "Unexpected freqency %d\n", ltohl(freq));
+		return -1;
+	} else if (ltohl(freq) != 16000 && which == 16000) {
 		ast_log(LOG_WARNING, "Unexpected freqency %d\n", ltohl(freq));
 		return -1;
 	}
@@ -239,7 +243,7 @@ static int update_header(FILE *f)
 	return 0;
 }
 
-static int write_header(FILE *f)
+static int write_header(FILE *f, int which)
 {
 	unsigned int hz=htoll(8000);
 	unsigned int bhz = htoll(16000);
@@ -249,6 +253,11 @@ static int write_header(FILE *f)
 	unsigned short bysam = htols(2);
 	unsigned short bisam = htols(16);
 	unsigned int size = htoll(0);
+	if (which == 16000) {
+		hz = htoll(16000);
+		bhz = htoll(32000);
+	}
+
 	/* Write a wav header, ignoring sizes which will be filled in later */
 	fseek(f,0,SEEK_SET);
 	if (fwrite("RIFF", 1, 4, f) != 4) {
@@ -308,22 +317,37 @@ static int wav_open(struct ast_filestream *s)
 	   if we did, it would go here.  We also might want to check
 	   and be sure it's a valid file.  */
 	struct wav_desc *tmp = (struct wav_desc *)s->_private;
-	if ((tmp->maxlen = check_header(s->f)) < 0)
+	if ((tmp->maxlen = check_header(s->f, tmp->which)) < 0)
 		return -1;
 	return 0;
 }
 
+static int wav16_open(struct ast_filestream *s)
+{
+	struct wav_desc *tmp = (struct wav_desc *)s->_private;
+	tmp->which = 16000;
+	return wav_open(s);
+}
+
 static int wav_rewrite(struct ast_filestream *s, const char *comment)
 {
 	/* We don't have any header to read or anything really, but
 	   if we did, it would go here.  We also might want to check
 	   and be sure it's a valid file.  */
+	struct wav_desc *tmp = (struct wav_desc *)s->_private;
 
-	if (write_header(s->f))
+	if (write_header(s->f, tmp->which))
 		return -1;
 	return 0;
 }
 
+static int wav16_rewrite(struct ast_filestream *s, const char *comment)
+{
+	struct wav_desc *tmp = (struct wav_desc *)s->_private;
+	tmp->which = 16000;
+	return wav_rewrite(s, comment);
+}
+
 static void wav_close(struct ast_filestream *s)
 {
 	char zero = 0;
@@ -351,6 +375,10 @@ static struct ast_frame *wav_read(struct ast_filestream *s, int *whennext)
 	/* Send a frame from the file to the appropriate channel */
 	struct wav_desc *fs = (struct wav_desc *)s->_private;
 
+	if (fs->which == 16000) {
+		bytes *= 2;
+	}
+
 	here = ftello(s->f);
 	if (fs->maxlen - here < bytes)		/* truncate if necessary */
 		bytes = fs->maxlen - here;
@@ -358,10 +386,10 @@ static struct ast_frame *wav_read(struct ast_filestream *s, int *whennext)
 		bytes = 0;
 /* 	ast_debug(1, "here: %d, maxlen: %d, bytes: %d\n", here, s->maxlen, bytes); */
 	s->fr.frametype = AST_FRAME_VOICE;
-	s->fr.subclass = AST_FORMAT_SLINEAR;
+	s->fr.subclass = fs->which == 16000 ? AST_FORMAT_SLINEAR16 : AST_FORMAT_SLINEAR;
 	s->fr.mallocd = 0;
 	AST_FRAME_SET_BUFFER(&s->fr, s->buf, AST_FRIENDLY_OFFSET, bytes);
-	
+
 	if ( (res = fread(s->fr.data.ptr, 1, s->fr.datalen, s->f)) <= 0 ) {
 		if (res)
 			ast_log(LOG_WARNING, "Short read (%d) (%s)!\n", res, strerror(errno));
@@ -373,7 +401,7 @@ static struct ast_frame *wav_read(struct ast_filestream *s, int *whennext)
 	tmp = (short *)(s->fr.data.ptr);
 #if __BYTE_ORDER == __BIG_ENDIAN
 	/* file format is little endian so we need to swap */
-	for( x = 0; x < samples; x++)
+	for( x = 0; x < samples * (fs->which == 16000 ? 2 : 1); x++)
 		tmp[x] = (tmp[x] << 8) | ((tmp[x] & 0xff00) >> 8);
 #endif
 
@@ -385,7 +413,7 @@ static int wav_write(struct ast_filestream *fs, struct ast_frame *f)
 {
 #if __BYTE_ORDER == __BIG_ENDIAN
 	int x;
-	short tmp[8000], *tmpi;
+	short tmp[16000], *tmpi;
 #endif
 	struct wav_desc *s = (struct wav_desc *)fs->_private;
 	int res;
@@ -394,9 +422,12 @@ static int wav_write(struct ast_filestream *fs, struct ast_frame *f)
 		ast_log(LOG_WARNING, "Asked to write non-voice frame!\n");
 		return -1;
 	}
-	if (f->subclass != AST_FORMAT_SLINEAR) {
+	if (f->subclass != AST_FORMAT_SLINEAR && s->which == 0) {
 		ast_log(LOG_WARNING, "Asked to write non-SLINEAR frame (%d)!\n", f->subclass);
 		return -1;
+	} else if (f->subclass != AST_FORMAT_SLINEAR16 && s->which == 16000) {
+		ast_log(LOG_WARNING, "Asked to write non-SLINEAR16 frame (%d)!\n", f->subclass);
+		return -1;
 	}
 	if (!f->datalen)
 		return -1;
@@ -421,9 +452,15 @@ static int wav_write(struct ast_filestream *fs, struct ast_frame *f)
 	}
 
 	s->bytes += f->datalen;
-		
+
 	return 0;
+}
 
+static int wav16_write(struct ast_filestream *fs, struct ast_frame *f)
+{
+	struct wav_desc *s = (struct wav_desc *)fs->_private;
+	s->which = 16000;
+	return wav_write(fs, f);
 }
 
 static int wav_seek(struct ast_filestream *fs, off_t sample_offset, int whence)
@@ -480,16 +517,38 @@ static const struct ast_format wav_f = {
 	.desc_size = sizeof(struct wav_desc),
 };
 
+static const struct ast_format Wav_f = {
+	.name = "wav16",
+	.exts = "Wav|wav16",
+	.format = AST_FORMAT_SLINEAR16,
+	.open =	wav16_open,
+	.rewrite = wav16_rewrite,
+	.write = wav16_write,
+	.seek = wav_seek,
+	.trunc = wav_trunc,
+	.tell =	wav_tell,
+	.read = wav_read,
+	.close = wav_close,
+	.buf_size = WAV_BUF_SIZE * 2 + AST_FRIENDLY_OFFSET,
+	.desc_size = sizeof(struct wav_desc),
+};
+
 static int load_module(void)
 {
 	if (ast_format_register(&wav_f))
 		return AST_MODULE_LOAD_FAILURE;
+	ast_format_register(&Wav_f);
 	return AST_MODULE_LOAD_SUCCESS;
 }
 
 static int unload_module(void)
 {
-	return ast_format_unregister(wav_f.name);
+	int res;
+	if (!(res = ast_format_unregister(Wav_f.name))) {
+		return ast_format_unregister(wav_f.name);
+	} else {
+		return res;
+	}
 }	
 
 AST_MODULE_INFO_STANDARD(ASTERISK_GPL_KEY, "Microsoft WAV format (8000Hz Signed Linear)");