1
/* $Id: wav_player.c 4122 2012-05-14 11:04:46Z bennylp $ */
3
* Copyright (C) 2008-2011 Teluu Inc. (http://www.teluu.com)
4
* Copyright (C) 2003-2008 Benny Prijono <benny@prijono.org>
6
* This program is free software; you can redistribute it and/or modify
7
* it under the terms of the GNU General Public License as published by
8
* the Free Software Foundation; either version 2 of the License, or
9
* (at your option) any later version.
11
* This program is distributed in the hope that it will be useful,
12
* but WITHOUT ANY WARRANTY; without even the implied warranty of
13
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14
* GNU General Public License for more details.
16
* You should have received a copy of the GNU General Public License
17
* along with this program; if not, write to the Free Software
18
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
20
#include <pjmedia/wav_port.h>
21
#include <pjmedia/alaw_ulaw.h>
22
#include <pjmedia/errno.h>
23
#include <pjmedia/wave.h>
24
#include <pj/assert.h>
25
#include <pj/file_access.h>
26
#include <pj/file_io.h>
29
#include <pj/string.h>
32
#define THIS_FILE "wav_player.c"
35
#define SIGNATURE PJMEDIA_SIG_PORT_WAV_PLAYER
36
#define BITS_PER_SAMPLE 16
39
# define TRACE_(x) PJ_LOG(4,x)
44
#if defined(PJ_IS_BIG_ENDIAN) && PJ_IS_BIG_ENDIAN!=0
45
static void samples_to_host(pj_int16_t *samples, unsigned count)
48
for (i=0; i<count; ++i) {
49
samples[i] = pj_swap16(samples[i]);
53
# define samples_to_host(samples,count)
56
struct file_reader_port
60
pjmedia_wave_fmt_tag fmt_tag;
61
pj_uint16_t bytes_per_sample;
75
pj_status_t (*cb)(pjmedia_port*, void*);
79
static pj_status_t file_get_frame(pjmedia_port *this_port,
80
pjmedia_frame *frame);
81
static pj_status_t file_on_destroy(pjmedia_port *this_port);
83
static struct file_reader_port *create_file_port(pj_pool_t *pool)
85
const pj_str_t name = pj_str("file");
86
struct file_reader_port *port;
88
port = PJ_POOL_ZALLOC_T(pool, struct file_reader_port);
92
/* Put in default values.
93
* These will be overriden once the file is read.
95
pjmedia_port_info_init(&port->base.info, &name, SIGNATURE,
98
port->base.get_frame = &file_get_frame;
99
port->base.on_destroy = &file_on_destroy;
108
static pj_status_t fill_buffer(struct file_reader_port *fport)
110
pj_ssize_t size_left = fport->bufsize;
111
unsigned size_to_read;
115
fport->eofpos = NULL;
117
while (size_left > 0) {
119
/* Calculate how many bytes to read in this run. */
120
size = size_to_read = size_left;
121
status = pj_file_read(fport->fd,
122
&fport->buf[fport->bufsize-size_left],
124
if (status != PJ_SUCCESS)
127
/* Should return more appropriate error code here.. */
128
return PJ_ECANCELLED;
131
if (size > (pj_ssize_t)fport->data_left) {
132
/* We passed the end of the data chunk,
133
* only count the portion read from the data chunk.
135
size = (pj_ssize_t)fport->data_left;
139
fport->data_left -= size;
142
/* If size is less than size_to_read, it indicates that we've
143
* encountered EOF. Rewind the file.
145
if (size < (pj_ssize_t)size_to_read) {
146
fport->eof = PJ_TRUE;
147
fport->eofpos = fport->buf + fport->bufsize - size_left;
149
if (fport->options & PJMEDIA_FILE_NO_LOOP) {
150
/* Zero remaining buffer */
151
if (fport->fmt_tag == PJMEDIA_WAVE_FMT_TAG_PCM) {
152
pj_bzero(fport->eofpos, size_left);
153
} else if (fport->fmt_tag == PJMEDIA_WAVE_FMT_TAG_ULAW) {
154
int val = pjmedia_linear2ulaw(0);
155
pj_memset(fport->eofpos, val, size_left);
156
} else if (fport->fmt_tag == PJMEDIA_WAVE_FMT_TAG_ALAW) {
157
int val = pjmedia_linear2alaw(0);
158
pj_memset(fport->eofpos, val, size_left);
164
fport->fpos = fport->start_data;
165
pj_file_setpos( fport->fd, fport->fpos, PJ_SEEK_SET);
166
fport->data_left = fport->data_len;
170
/* Convert samples to host rep */
171
samples_to_host((pj_int16_t*)fport->buf,
172
fport->bufsize/fport->bytes_per_sample);
179
* Create WAVE player port.
181
PJ_DEF(pj_status_t) pjmedia_wav_player_port_create( pj_pool_t *pool,
182
const char *filename,
185
pj_ssize_t buff_size,
186
pjmedia_port **p_port )
188
pjmedia_wave_hdr wave_hdr;
189
pj_ssize_t size_to_read, size_read;
190
struct file_reader_port *fport;
191
pjmedia_audio_format_detail *ad;
194
unsigned samples_per_frame;
195
pj_status_t status = PJ_SUCCESS;
198
/* Check arguments. */
199
PJ_ASSERT_RETURN(pool && filename && p_port, PJ_EINVAL);
201
/* Check the file really exists. */
202
if (!pj_file_exists(filename)) {
206
/* Normalize ptime */
210
/* Normalize buff_size */
211
if (buff_size < 1) buff_size = PJMEDIA_FILE_PORT_BUFSIZE;
214
/* Create fport instance. */
215
fport = create_file_port(pool);
221
/* Get the file size. */
222
fport->fsize = pj_file_size(filename);
224
/* Size must be more than WAVE header size */
225
if (fport->fsize <= sizeof(pjmedia_wave_hdr)) {
226
return PJMEDIA_ENOTVALIDWAVE;
230
status = pj_file_open( pool, filename, PJ_O_RDONLY, &fport->fd);
231
if (status != PJ_SUCCESS)
234
/* Read the file header plus fmt header only. */
235
size_read = size_to_read = sizeof(wave_hdr) - 8;
236
status = pj_file_read( fport->fd, &wave_hdr, &size_read);
237
if (status != PJ_SUCCESS) {
238
pj_file_close(fport->fd);
241
if (size_read != size_to_read) {
242
pj_file_close(fport->fd);
243
return PJMEDIA_ENOTVALIDWAVE;
246
/* Normalize WAVE header fields values from little-endian to host
249
pjmedia_wave_hdr_file_to_host(&wave_hdr);
251
/* Validate WAVE file. */
252
if (wave_hdr.riff_hdr.riff != PJMEDIA_RIFF_TAG ||
253
wave_hdr.riff_hdr.wave != PJMEDIA_WAVE_TAG ||
254
wave_hdr.fmt_hdr.fmt != PJMEDIA_FMT_TAG)
256
pj_file_close(fport->fd);
258
"actual value|expected riff=%x|%x, wave=%x|%x fmt=%x|%x",
259
wave_hdr.riff_hdr.riff, PJMEDIA_RIFF_TAG,
260
wave_hdr.riff_hdr.wave, PJMEDIA_WAVE_TAG,
261
wave_hdr.fmt_hdr.fmt, PJMEDIA_FMT_TAG));
262
return PJMEDIA_ENOTVALIDWAVE;
265
/* Validate format and its attributes (i.e: bits per sample, block align) */
266
switch (wave_hdr.fmt_hdr.fmt_tag) {
267
case PJMEDIA_WAVE_FMT_TAG_PCM:
268
if (wave_hdr.fmt_hdr.bits_per_sample != 16 ||
269
wave_hdr.fmt_hdr.block_align != 2 * wave_hdr.fmt_hdr.nchan)
270
status = PJMEDIA_EWAVEUNSUPP;
273
case PJMEDIA_WAVE_FMT_TAG_ALAW:
274
case PJMEDIA_WAVE_FMT_TAG_ULAW:
275
if (wave_hdr.fmt_hdr.bits_per_sample != 8 ||
276
wave_hdr.fmt_hdr.block_align != wave_hdr.fmt_hdr.nchan)
277
status = PJMEDIA_ENOTVALIDWAVE;
281
status = PJMEDIA_EWAVEUNSUPP;
285
if (status != PJ_SUCCESS) {
286
pj_file_close(fport->fd);
290
fport->fmt_tag = (pjmedia_wave_fmt_tag)wave_hdr.fmt_hdr.fmt_tag;
291
fport->bytes_per_sample = (pj_uint16_t)
292
(wave_hdr.fmt_hdr.bits_per_sample / 8);
294
/* If length of fmt_header is greater than 16, skip the remaining
297
if (wave_hdr.fmt_hdr.len > 16) {
298
size_to_read = wave_hdr.fmt_hdr.len - 16;
299
status = pj_file_setpos(fport->fd, size_to_read, PJ_SEEK_CUR);
300
if (status != PJ_SUCCESS) {
301
pj_file_close(fport->fd);
306
/* Repeat reading the WAVE file until we have 'data' chunk */
308
pjmedia_wave_subchunk subchunk;
310
status = pj_file_read(fport->fd, &subchunk, &size_read);
311
if (status != PJ_SUCCESS || size_read != 8) {
312
pj_file_close(fport->fd);
313
return PJMEDIA_EWAVETOOSHORT;
316
/* Normalize endianness */
317
PJMEDIA_WAVE_NORMALIZE_SUBCHUNK(&subchunk);
319
/* Break if this is "data" chunk */
320
if (subchunk.id == PJMEDIA_DATA_TAG) {
321
wave_hdr.data_hdr.data = PJMEDIA_DATA_TAG;
322
wave_hdr.data_hdr.len = subchunk.len;
326
/* Otherwise skip the chunk contents */
327
size_to_read = subchunk.len;
328
status = pj_file_setpos(fport->fd, size_to_read, PJ_SEEK_CUR);
329
if (status != PJ_SUCCESS) {
330
pj_file_close(fport->fd);
335
/* Current file position now points to start of data */
336
status = pj_file_getpos(fport->fd, &pos);
337
fport->start_data = (unsigned)pos;
338
fport->data_len = wave_hdr.data_hdr.len;
339
fport->data_left = wave_hdr.data_hdr.len;
341
/* Validate length. */
342
if (wave_hdr.data_hdr.len > fport->fsize - fport->start_data) {
343
pj_file_close(fport->fd);
344
return PJMEDIA_EWAVEUNSUPP;
346
if (wave_hdr.data_hdr.len < ptime * wave_hdr.fmt_hdr.sample_rate *
347
wave_hdr.fmt_hdr.nchan / 1000)
349
pj_file_close(fport->fd);
350
return PJMEDIA_EWAVETOOSHORT;
353
/* It seems like we have a valid WAVE file. */
356
fport->options = options;
358
/* Update port info. */
359
ad = pjmedia_format_get_audio_format_detail(&fport->base.info.fmt, 1);
360
pj_strdup2(pool, &name, filename);
361
samples_per_frame = ptime * wave_hdr.fmt_hdr.sample_rate *
362
wave_hdr.fmt_hdr.nchan / 1000;
363
pjmedia_port_info_init(&fport->base.info, &name, SIGNATURE,
364
wave_hdr.fmt_hdr.sample_rate,
365
wave_hdr.fmt_hdr.nchan,
369
/* If file is shorter than buffer size, adjust buffer size to file
370
* size. Otherwise EOF callback will be called multiple times when
371
* fill_buffer() is called.
373
if (wave_hdr.data_hdr.len < (unsigned)buff_size)
374
buff_size = wave_hdr.data_hdr.len;
376
/* Create file buffer.
378
fport->bufsize = buff_size;
381
/* samples_per_frame must be smaller than bufsize (because get_frame()
382
* doesn't handle this case).
384
if (samples_per_frame * fport->bytes_per_sample >= fport->bufsize) {
385
pj_file_close(fport->fd);
390
fport->buf = (char*) pj_pool_alloc(pool, fport->bufsize);
392
pj_file_close(fport->fd);
396
fport->readpos = fport->buf;
398
/* Set initial position of the file. */
399
fport->fpos = fport->start_data;
401
/* Fill up the buffer. */
402
status = fill_buffer(fport);
403
if (status != PJ_SUCCESS) {
404
pj_file_close(fport->fd);
410
*p_port = &fport->base;
414
"File player '%.*s' created: samp.rate=%d, ch=%d, bufsize=%uKB, "
416
(int)fport->base.info.name.slen,
417
fport->base.info.name.ptr,
420
fport->bufsize / 1000,
421
(unsigned long)(fport->fsize / 1000)));
428
* Get the data length, in bytes.
430
PJ_DEF(pj_ssize_t) pjmedia_wav_player_get_len(pjmedia_port *port)
432
struct file_reader_port *fport;
436
PJ_ASSERT_RETURN(port, -PJ_EINVAL);
438
/* Check that this is really a player port */
439
PJ_ASSERT_RETURN(port->info.signature == SIGNATURE, -PJ_EINVALIDOP);
441
fport = (struct file_reader_port*) port;
443
size = (pj_ssize_t) fport->fsize;
444
return size - fport->start_data;
451
PJ_DEF(pj_status_t) pjmedia_wav_player_port_set_pos(pjmedia_port *port,
454
struct file_reader_port *fport;
458
PJ_ASSERT_RETURN(port, PJ_EINVAL);
460
/* Check that this is really a player port */
461
PJ_ASSERT_RETURN(port->info.signature == SIGNATURE, PJ_EINVALIDOP);
464
fport = (struct file_reader_port*) port;
466
/* Check that this offset does not pass the audio-data (in case of
467
* extra chunk after audio data chunk
469
PJ_ASSERT_RETURN(bytes < fport->data_len, PJ_EINVAL);
471
fport->fpos = fport->start_data + bytes;
472
fport->data_left = fport->data_len - bytes;
473
pj_file_setpos( fport->fd, fport->fpos, PJ_SEEK_SET);
475
fport->eof = PJ_FALSE;
476
status = fill_buffer(fport);
477
if (status != PJ_SUCCESS)
480
fport->readpos = fport->buf;
487
* Get the file play position of WAV player.
489
PJ_DEF(pj_ssize_t) pjmedia_wav_player_port_get_pos( pjmedia_port *port )
491
struct file_reader_port *fport;
492
pj_size_t payload_pos;
495
PJ_ASSERT_RETURN(port, -PJ_EINVAL);
497
/* Check that this is really a player port */
498
PJ_ASSERT_RETURN(port->info.signature == SIGNATURE, -PJ_EINVALIDOP);
500
fport = (struct file_reader_port*) port;
502
payload_pos = (pj_size_t)(fport->fpos - fport->start_data);
503
if (payload_pos >= fport->bufsize)
504
return payload_pos - fport->bufsize + (fport->readpos - fport->buf);
506
return (fport->readpos - fport->buf) % payload_pos;
512
* Register a callback to be called when the file reading has reached the
515
PJ_DEF(pj_status_t) pjmedia_wav_player_set_eof_cb( pjmedia_port *port,
517
pj_status_t (*cb)(pjmedia_port *port,
520
struct file_reader_port *fport;
523
PJ_ASSERT_RETURN(port, -PJ_EINVAL);
525
/* Check that this is really a player port */
526
PJ_ASSERT_RETURN(port->info.signature == SIGNATURE, -PJ_EINVALIDOP);
528
fport = (struct file_reader_port*) port;
530
fport->base.port_data.pdata = user_data;
538
* Get frame from file.
540
static pj_status_t file_get_frame(pjmedia_port *this_port,
541
pjmedia_frame *frame)
543
struct file_reader_port *fport = (struct file_reader_port*)this_port;
547
pj_assert(fport->base.info.signature == SIGNATURE);
548
pj_assert(frame->size <= fport->bufsize);
550
/* EOF is set and readpos already passed the eofpos */
551
if (fport->eof && fport->readpos >= fport->eofpos) {
552
pj_status_t status = PJ_SUCCESS;
554
PJ_LOG(5,(THIS_FILE, "File port %.*s EOF",
555
(int)fport->base.info.name.slen,
556
fport->base.info.name.ptr));
558
/* Call callback, if any */
560
status = (*fport->cb)(this_port, fport->base.port_data.pdata);
562
/* If callback returns non PJ_SUCCESS or 'no loop' is specified,
563
* return immediately (and don't try to access player port since
564
* it might have been destroyed by the callback).
566
if ((status != PJ_SUCCESS) || (fport->options & PJMEDIA_FILE_NO_LOOP)) {
567
frame->type = PJMEDIA_FRAME_TYPE_NONE;
572
PJ_LOG(5,(THIS_FILE, "File port %.*s rewinding..",
573
(int)fport->base.info.name.slen,
574
fport->base.info.name.ptr));
576
fport->eof = PJ_FALSE;
579
//pj_assert(frame->size == fport->base.info.bytes_per_frame);
580
if (fport->fmt_tag == PJMEDIA_WAVE_FMT_TAG_PCM) {
581
frame_size = frame->size;
582
//frame->size = frame_size;
584
/* Must be ULAW or ALAW */
585
pj_assert(fport->fmt_tag == PJMEDIA_WAVE_FMT_TAG_ULAW ||
586
fport->fmt_tag == PJMEDIA_WAVE_FMT_TAG_ALAW);
588
frame_size = frame->size >> 1;
589
frame->size = frame_size << 1;
592
/* Copy frame from buffer. */
593
frame->type = PJMEDIA_FRAME_TYPE_AUDIO;
594
frame->timestamp.u64 = 0;
596
if ((fport->readpos + frame_size) <= (fport->buf + fport->bufsize))
598
/* Read contiguous buffer. */
599
pj_memcpy(frame->buf, fport->readpos, frame_size);
601
/* Fill up the buffer if all has been read. */
602
fport->readpos += frame_size;
603
if (fport->readpos == fport->buf + fport->bufsize) {
604
fport->readpos = fport->buf;
606
status = fill_buffer(fport);
607
if (status != PJ_SUCCESS) {
608
frame->type = PJMEDIA_FRAME_TYPE_NONE;
610
fport->readpos = fport->buf + fport->bufsize;
618
* First stage: read until end of buffer.
620
endread = (fport->buf+fport->bufsize) - fport->readpos;
621
pj_memcpy(frame->buf, fport->readpos, endread);
623
/* End Of Buffer and EOF and NO LOOP */
624
if (fport->eof && (fport->options & PJMEDIA_FILE_NO_LOOP)) {
625
fport->readpos += endread;
627
if (fport->fmt_tag == PJMEDIA_WAVE_FMT_TAG_PCM) {
628
pj_bzero((char*)frame->buf + endread, frame_size - endread);
629
} else if (fport->fmt_tag == PJMEDIA_WAVE_FMT_TAG_ULAW) {
630
int val = pjmedia_linear2ulaw(0);
631
pj_memset((char*)frame->buf + endread, val,
632
frame_size - endread);
633
} else if (fport->fmt_tag == PJMEDIA_WAVE_FMT_TAG_ALAW) {
634
int val = pjmedia_linear2alaw(0);
635
pj_memset((char*)frame->buf + endread, val,
636
frame_size - endread);
642
/* Second stage: fill up buffer, and read from the start of buffer. */
643
status = fill_buffer(fport);
644
if (status != PJ_SUCCESS) {
645
frame->type = PJMEDIA_FRAME_TYPE_NONE;
647
fport->readpos = fport->buf + fport->bufsize;
651
pj_memcpy(((char*)frame->buf)+endread, fport->buf, frame_size-endread);
652
fport->readpos = fport->buf + (frame_size - endread);
655
if (fport->fmt_tag == PJMEDIA_WAVE_FMT_TAG_ULAW ||
656
fport->fmt_tag == PJMEDIA_WAVE_FMT_TAG_ALAW)
662
dst = (pj_uint16_t*)frame->buf + frame_size - 1;
663
src = (pj_uint8_t*)frame->buf + frame_size - 1;
665
if (fport->fmt_tag == PJMEDIA_WAVE_FMT_TAG_ULAW) {
666
for (i = 0; i < frame_size; ++i) {
667
*dst-- = (pj_uint16_t) pjmedia_ulaw2linear(*src--);
670
for (i = 0; i < frame_size; ++i) {
671
*dst-- = (pj_uint16_t) pjmedia_alaw2linear(*src--);
682
static pj_status_t file_on_destroy(pjmedia_port *this_port)
684
struct file_reader_port *fport = (struct file_reader_port*) this_port;
686
pj_assert(this_port->info.signature == SIGNATURE);
688
pj_file_close(fport->fd);