root/modules/amr_dec/amr_in.c

DEFINITIONS

1. AMR_RegisterMimeTypes
2. AMR_GetESD
3. AMR_SetupObject
4. AMR_CanHandleURL
5. file_is_local
6. AMR_ConfigureFromFile
7. AMR_NetIO
8. AMR_DownloadFile
9. AMR_ConnectService
10. AMR_CloseService
11. AMR_GetServiceDesc
12. AMR_ConnectChannel
13. AMR_DisconnectChannel
14. AMR_ServiceCommand
15. AMR_ChannelGetSLP
16. AMR_ChannelReleaseSLP
17. NewAESReader
18. DeleteAESReader
19. QueryInterfaces
20. LoadInterface
21. ShutdownInterface

/*
 *                      GPAC - Multimedia Framework C SDK
 *
 *                      Authors: Jean Le Feuvre
 *                      Copyright (c) Telecom ParisTech 2000-2012
 *                                      All rights reserved
 *
 *  This file is part of GPAC / AMR&EVRC&SMV reader module
 *
 *  GPAC is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU Lesser General Public License as published by
 *  the Free Software Foundation; either version 2, or (at your option)
 *  any later version.
 *
 *  GPAC is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU Lesser General Public License for more details.
 *
 *  You should have received a copy of the GNU Lesser General Public
 *  License along with this library; see the file COPYING.  If not, write to
 *  the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA.
 *
 */

#include <gpac/modules/service.h>
#include <gpac/modules/codec.h>
#include <gpac/constants.h>
#include <gpac/isomedia.h>


enum
{
        TYPE_AMR = GF_4CC('s','a','m','r'),
        TYPE_AMR_WB = GF_4CC('s','a','w','b'),
        TYPE_EVRC = GF_4CC('e','v','r','c'),
        TYPE_SMV = GF_4CC('s','m','v',' ')
};

typedef struct
{
        GF_ClientService *service;

        Bool is_remote;
        u32 start_offset;
        u32 mtype, sample_rate, block_size;

        FILE *stream;
        u32 duration;

        Bool needs_connection;
        u32 pad_bytes;
        Bool done;
        u32 is_inline;
        LPNETCHANNEL ch;

        unsigned char *data;
        u32 data_size;
        GF_SLHeader sl_hdr;

        Double start_range, end_range;
        u32 current_time;
        /*file downloader*/
        GF_DownloadSession * dnload;

        //Bool is_live;
} AMR_Reader;

static u32 AMR_RegisterMimeTypes(const GF_InputService *plug)
{
        gf_service_register_mime(plug, "audio/amr", "amr awb", "AMR Speech Data");
        gf_service_register_mime(plug, "audio/evrc", "evc", "EVRC Speech Data");
        gf_service_register_mime(plug, "audio/smv", "smv", "SMV Speech Data");
        return 3;
}

static GF_ESD *AMR_GetESD(AMR_Reader *read)
{
        GF_BitStream *dsi;
        GF_ESD *esd;
        esd = gf_odf_desc_esd_new(0);
        esd->decoderConfig->streamType = GF_STREAM_AUDIO;
        esd->ESID = 1;
        esd->OCRESID = 0;
        esd->slConfig->timestampResolution = read->sample_rate;
        /*all packets are complete AUs*/
        esd->slConfig->useAccessUnitEndFlag = esd->slConfig->useAccessUnitStartFlag = 0;
        esd->slConfig->hasRandomAccessUnitsOnlyFlag = 1;

        if ((read->mtype==TYPE_AMR) || (read->mtype==TYPE_AMR_WB)) {
                esd->decoderConfig->objectTypeIndication = GPAC_OTI_MEDIA_GENERIC;
                dsi = gf_bs_new(NULL, 0, GF_BITSTREAM_WRITE);
                gf_bs_write_u32(dsi, read->mtype);
                gf_bs_write_u32(dsi, (read->mtype==TYPE_AMR) ? 8000 : 16000);
                gf_bs_write_u16(dsi, 1);
                gf_bs_write_u16(dsi, (read->mtype==TYPE_AMR) ? 160 : 320);
                gf_bs_write_u8(dsi, 16);
                gf_bs_write_u8(dsi, 1);
                gf_bs_get_content(dsi, & esd->decoderConfig->decoderSpecificInfo->data, & esd->decoderConfig->decoderSpecificInfo->dataLength);
                gf_bs_del(dsi);
        }
        else if (read->mtype==TYPE_EVRC) esd->decoderConfig->objectTypeIndication = GPAC_OTI_AUDIO_EVRC_VOICE;
        else if (read->mtype==TYPE_SMV) esd->decoderConfig->objectTypeIndication = GPAC_OTI_AUDIO_SMV_VOICE;
        return esd;
}

static void AMR_SetupObject(AMR_Reader *read)
{
        GF_ESD *esd;
        GF_ObjectDescriptor *od = (GF_ObjectDescriptor *) gf_odf_desc_new(GF_ODF_OD_TAG);
        od->objectDescriptorID = 1;
        esd = AMR_GetESD(read);
        esd->OCRESID = 0;
        gf_list_add(od->ESDescriptors, esd);
        gf_service_declare_media(read->service, (GF_Descriptor*)od, GF_FALSE);
}

static Bool AMR_CanHandleURL(GF_InputService *plug, const char *url)
{
        char *sExt;
        sExt = strrchr(url, '.');
        if (!sExt) return GF_FALSE;
        if (gf_service_check_mime_register(plug, "audio/amr", "amr awb", "AMR Speech Data", sExt)) return GF_TRUE;
        if (gf_service_check_mime_register(plug, "audio/evrc", "evc", "EVRC Speech Data", sExt)) return GF_TRUE;
        if (gf_service_check_mime_register(plug, "audio/smv", "smv", "SMV Speech Data", sExt)) return GF_TRUE;
        return GF_FALSE;
}

static Bool file_is_local(const char *url)
{
        if (!strnicmp(url, "file://", 7)) return GF_TRUE;
        if (strstr(url, "://")) return GF_FALSE;
        return GF_TRUE;
}


static Bool AMR_ConfigureFromFile(AMR_Reader *read)
{
        u32 i;
        char magic[20];

        if (!read->stream) return GF_FALSE;
        read->mtype = 0;
        read->start_offset = 6;
        read->sample_rate = 8000;
        read->block_size = 160;
        i = (u32) fread(magic, 1, 20, read->stream);
        if (i != 20) return GF_FALSE;

        if (!strnicmp(magic, "#!AMR\n", 6)) {
                fseek(read->stream, 6, SEEK_SET);
                read->mtype = TYPE_AMR;
        }
        else if (!strnicmp(magic, "#!EVRC\n", 7)) {
                fseek(read->stream, 7, SEEK_SET);
                read->start_offset = 7;
                read->mtype = TYPE_EVRC;
        }
        else if (!strnicmp(magic, "#!SMV\n", 6)) {
                fseek(read->stream, 6, SEEK_SET);
                read->mtype = TYPE_SMV;
        }
        else if (!strnicmp(magic, "#!AMR-WB\n", 9)) {
                read->mtype = TYPE_AMR_WB;
                read->start_offset = 9;
                read->sample_rate = 16000;
                read->block_size = 320;
                fseek(read->stream, 9, SEEK_SET);
        }
        else if (!strnicmp(magic, "#!AMR_MC1.0\n", 12)) return GF_FALSE;
        else if (!strnicmp(magic, "#!AMR-WB_MC1.0\n", 15)) return GF_FALSE;
        else return GF_FALSE;


        read->duration = 0;

        if (!read->is_remote) {
                u32 size=0;
                while (!feof(read->stream)) {
                        u8 ft = fgetc(read->stream);
                        switch (read->mtype) {
                        case TYPE_AMR:
                        case TYPE_AMR_WB:
                                ft = (ft >> 3) & 0x0F;
                                size = (read->mtype==TYPE_AMR_WB) ? GF_AMR_WB_FRAME_SIZE[ft] : GF_AMR_FRAME_SIZE[ft];
                                break;
                        default:
                                for (i=0; i<GF_SMV_EVRC_RATE_TO_SIZE_NB; i++) {
                                        if (GF_SMV_EVRC_RATE_TO_SIZE[2*i]==ft) {
                                                /*remove rate_type byte*/
                                                size = GF_SMV_EVRC_RATE_TO_SIZE[2*i+1] - 1;
                                                break;
                                        }
                                }
                                break;
                        }

                        if (size) fseek(read->stream, size, SEEK_CUR);
                        read->duration += read->block_size;
                }
        }
        fseek(read->stream, read->start_offset, SEEK_SET);
        return GF_TRUE;
}

static void AMR_NetIO(void *cbk, GF_NETIO_Parameter *param)
{
        GF_Err e;
        const char *szCache;
        u32 bytes_done;
        AMR_Reader *read = (AMR_Reader *) cbk;

        e = param->error;
        /*done*/
        if (param->msg_type==GF_NETIO_DATA_TRANSFERED) {
                if (read->stream) {
                        read->is_remote = GF_FALSE;
                        e = GF_EOS;
                } else {
                        return;
                }
        } else {
                /*handle service message*/
                gf_service_download_update_stats(read->dnload);
                if (param->msg_type!=GF_NETIO_DATA_EXCHANGE) return;
        }

        /*data fetching*/
        if (e >= GF_OK) {
                if (read->stream) return;

                /*open service*/
                szCache = gf_dm_sess_get_cache_name(read->dnload);
                if (!szCache) e = GF_IO_ERR;
                else {
                        read->stream = gf_fopen((char *) szCache, "rb");
                        if (!read->stream) e = GF_SERVICE_ERROR;
                        else {
                                /*if full file at once (in cache) parse duration*/
                                if (e==GF_EOS) read->is_remote = GF_FALSE;
                                e = GF_OK;
                                /*not enough data*/
                                if (!AMR_ConfigureFromFile(read)) {
                                        /*bad data...*/
                                        gf_dm_sess_get_stats(read->dnload, NULL, NULL, NULL, &bytes_done, NULL, NULL);
                                        if (bytes_done>10*1024) {
                                                e = GF_CORRUPTED_DATA;
                                        } else {
                                                gf_fclose(read->stream);
                                                read->stream = NULL;
                                                return;
                                        }
                                }
                        }
                }
        }
        /*OK confirm*/
        if (read->needs_connection) {
                read->needs_connection = GF_FALSE;
                gf_service_connect_ack(read->service, NULL, e);
                if (!e) AMR_SetupObject(read);
        }
}

static void AMR_DownloadFile(GF_InputService *plug, char *url)
{
        AMR_Reader *read = (AMR_Reader*) plug->priv;

        read->needs_connection = GF_TRUE;

        read->dnload = gf_service_download_new(read->service, url, 0, AMR_NetIO, read);
        if (!read->dnload) {
                read->needs_connection = GF_FALSE;
                gf_service_connect_ack(read->service, NULL, GF_NOT_SUPPORTED);
        } else {
                /*start our download (threaded)*/
                gf_dm_sess_process(read->dnload);
        }
        /*service confirm is done once fetched*/
}


static GF_Err AMR_ConnectService(GF_InputService *plug, GF_ClientService *serv, const char *url)
{
        char szURL[2048];
        char *ext;
        GF_Err reply;
        AMR_Reader *read = (AMR_Reader*)plug->priv;
        read->service = serv;

        if (read->dnload) gf_service_download_del(read->dnload);
        read->dnload = NULL;

        strcpy(szURL, url);
        ext = strrchr(szURL, '#');
        if (ext) ext[0] = 0;

        /*remote fetch*/
        read->is_remote = !file_is_local(szURL);
        if (read->is_remote) {
                AMR_DownloadFile(plug, (char *) szURL);
                return GF_OK;
        }

        reply = GF_OK;
        read->stream = gf_fopen(szURL, "rb");
        if (!read->stream) {
                reply = GF_URL_ERROR;
        } else if (!AMR_ConfigureFromFile(read)) {
                gf_fclose(read->stream);
                read->stream = NULL;
                reply = GF_NOT_SUPPORTED;
        }
        gf_service_connect_ack(serv, NULL, reply);
        if (!reply && read->is_inline) AMR_SetupObject(read);
        return GF_OK;
}

static GF_Err AMR_CloseService(GF_InputService *plug)
{
        AMR_Reader *read = (AMR_Reader*)plug->priv;
        if (read->stream) gf_fclose(read->stream);
        read->stream = NULL;
        if (read->dnload) gf_service_download_del(read->dnload);
        read->dnload = NULL;

        if (read->data) gf_free(read->data);
        read->data = NULL;
        gf_service_disconnect_ack(read->service, NULL, GF_OK);
        return GF_OK;
}

static GF_Descriptor *AMR_GetServiceDesc(GF_InputService *plug, u32 expect_type, const char *sub_url)
{
        AMR_Reader *read = (AMR_Reader*)plug->priv;
        /*since we don't handle multitrack in aac, we don't need to check sub_url, only use expected type*/

        /*override default*/
        if (expect_type==GF_MEDIA_OBJECT_UNDEF) expect_type=GF_MEDIA_OBJECT_AUDIO;

        /*audio object*/
        if (expect_type==GF_MEDIA_OBJECT_AUDIO) {
                GF_ObjectDescriptor *od = (GF_ObjectDescriptor *) gf_odf_desc_new(GF_ODF_OD_TAG);
                GF_ESD *esd = AMR_GetESD(read);
                od->objectDescriptorID = 1;
                gf_list_add(od->ESDescriptors, esd);
                return (GF_Descriptor *) od;
        }
        /*let player handle scene description*/
        read->is_inline = 1;
        return NULL;
}

static GF_Err AMR_ConnectChannel(GF_InputService *plug, LPNETCHANNEL channel, const char *url, Bool upstream)
{
        u32 ES_ID;
        GF_Err e;
        AMR_Reader *read = (AMR_Reader*)plug->priv;

        e = GF_SERVICE_ERROR;
        if (read->ch==channel) goto exit;

        e = GF_STREAM_NOT_FOUND;
        ES_ID=0;
        if (strstr(url, "ES_ID")) {
                sscanf(url, "ES_ID=%d", &ES_ID);
        }
        /*URL setup*/
        else if (!read->ch && AMR_CanHandleURL(plug, url)) ES_ID = 1;

        if (ES_ID==1) {
                read->ch = channel;
                e = GF_OK;
        }

exit:
        gf_service_connect_ack(read->service, channel, e);
        return e;
}

static GF_Err AMR_DisconnectChannel(GF_InputService *plug, LPNETCHANNEL channel)
{
        AMR_Reader *read = (AMR_Reader*)plug->priv;

        GF_Err e = GF_STREAM_NOT_FOUND;
        if (read->ch == channel) {
                read->ch = NULL;
                if (read->data) gf_free(read->data);
                read->data = NULL;
                e = GF_OK;
        }
        gf_service_disconnect_ack(read->service, channel, e);
        return GF_OK;
}

static GF_Err AMR_ServiceCommand(GF_InputService *plug, GF_NetworkCommand *com)
{
        AMR_Reader *read = (AMR_Reader*)plug->priv;

        if (!com->base.on_channel) return GF_NOT_SUPPORTED;
        switch (com->command_type) {
        case GF_NET_CHAN_SET_PULL:
                return GF_OK;
        case GF_NET_CHAN_INTERACTIVE:
                return GF_OK;
        case GF_NET_CHAN_BUFFER:
                return GF_OK;
        case GF_NET_CHAN_SET_PADDING:
                read->pad_bytes = com->pad.padding_bytes;
                return GF_OK;
        case GF_NET_CHAN_DURATION:
                com->duration.duration = read->duration;
                com->duration.duration /= read->sample_rate;
                return GF_OK;
        case GF_NET_CHAN_PLAY:
                read->start_range = com->play.start_range;
                read->end_range = com->play.end_range;
                read->current_time = 0;
                if (read->stream) fseek(read->stream, read->start_offset, SEEK_SET);

                if (read->ch == com->base.on_channel) {
                        read->done = GF_FALSE;
                        /*PLAY after complete download, estimate duration*/
                        if (!read->is_remote && !read->duration) {
                                AMR_ConfigureFromFile(read);
                                if (read->duration) {
                                        GF_NetworkCommand rcfg;
                                        rcfg.base.on_channel = read->ch;
                                        rcfg.base.command_type = GF_NET_CHAN_DURATION;
                                        rcfg.duration.duration = read->duration;
                                        rcfg.duration.duration /= read->sample_rate;
                                        gf_service_command(read->service, &rcfg, GF_OK);
                                }
                        }
                }
                return GF_OK;
        case GF_NET_CHAN_STOP:
                return GF_OK;
        default:
                return GF_OK;
        }
}


static GF_Err AMR_ChannelGetSLP(GF_InputService *plug, LPNETCHANNEL channel, char **out_data_ptr, u32 *out_data_size, GF_SLHeader *out_sl_hdr, Bool *sl_compressed, GF_Err *out_reception_status, Bool *is_new_data)
{
        u32 start_from, i;
        u8 toc, ft;
        AMR_Reader *read = (AMR_Reader*)plug->priv;

        *out_reception_status = GF_OK;
        *sl_compressed = GF_FALSE;
        *is_new_data = GF_FALSE;

        memset(&read->sl_hdr, 0, sizeof(GF_SLHeader));
        read->sl_hdr.randomAccessPointFlag = 1;
        read->sl_hdr.compositionTimeStampFlag = 1;
        if (read->ch != channel) return GF_STREAM_NOT_FOUND;

        /*fetching es data*/
        if (read->done) {
                *out_reception_status = GF_EOS;
                return GF_OK;
        }

        if (!read->data) {
                if (!read->stream) {
                        *out_data_ptr = NULL;
                        *out_data_size = 0;
                        return GF_OK;
                }
                *is_new_data = GF_TRUE;

fetch_next:
                //pos = (u32) ftell(read->stream);
                if (feof(read->stream)) {
                        read->done = GF_TRUE;
                        *out_reception_status = GF_EOS;
                        return GF_OK;
                }

                toc = fgetc(read->stream);
                if (feof(read->stream)) {
                        read->done = GF_TRUE;
                        *out_reception_status = GF_EOS;
                        return GF_OK;
                }
                
                switch (read->mtype) {
                case TYPE_AMR:
                        ft = (toc >> 3) & 0x0F;
                        read->data_size = (u32)GF_AMR_FRAME_SIZE[ft];
                        break;
                case TYPE_AMR_WB:
                        ft = (toc >> 3) & 0x0F;
                        read->data_size = (u32)GF_AMR_WB_FRAME_SIZE[ft];
                        break;
                default:
                        for (i=0; i<GF_SMV_EVRC_RATE_TO_SIZE_NB; i++) {
                                if (GF_SMV_EVRC_RATE_TO_SIZE[2*i]==toc) {
                                        /*remove rate_type byte*/
                                        read->data_size = (u32)GF_SMV_EVRC_RATE_TO_SIZE[2*i+1] - 1;
                                        break;
                                }
                        }
                        break;
                }
                /*we're seeking*/
                if (read->start_range && read->duration) {
                        start_from = (u32) (read->start_range * read->sample_rate);
                        if (read->current_time + read->block_size < start_from) {
                                read->current_time += read->block_size;
                                fseek(read->stream, read->data_size, SEEK_CUR);
                                goto fetch_next;
                        } else {
                                read->start_range = 0;
                        }
                }
                assert(read->data_size);
                read->data_size++;
                read->sl_hdr.compositionTimeStamp = read->current_time;
                read->data = (unsigned char*)gf_malloc(sizeof(char) * (read->data_size+read->pad_bytes));
                read->data[0] = toc;
                if (read->data_size>1) {
                        u32 bytes_read = (u32) fread(read->data + 1, read->data_size-1, 1, read->stream);
                        if (bytes_read != read->data_size - 1) read->data_size = bytes_read+1; 
                }
                if (read->pad_bytes) memset(read->data + read->data_size, 0, sizeof(char) * read->pad_bytes);
        }
        *out_sl_hdr = read->sl_hdr;
        *out_data_ptr = (char*)read->data;
        *out_data_size = read->data_size;
        return GF_OK;
}

static GF_Err AMR_ChannelReleaseSLP(GF_InputService *plug, LPNETCHANNEL channel)
{
        AMR_Reader *read = (AMR_Reader*)plug->priv;

        if (read->ch == channel) {
                if (!read->data) return GF_BAD_PARAM;
                gf_free(read->data);
                read->data = NULL;
                read->current_time += read->block_size;
                return GF_OK;
        }
        return GF_OK;
}

GF_InputService *NewAESReader()
{
        AMR_Reader *reader;
        GF_InputService *plug = (GF_InputService*)gf_malloc(sizeof(GF_InputService));
        memset(plug, 0, sizeof(GF_InputService));
        GF_REGISTER_MODULE_INTERFACE(plug, GF_NET_CLIENT_INTERFACE, "GPAC AMR/EVRC/SMV Reader", "gpac distribution")

        plug->RegisterMimeTypes = AMR_RegisterMimeTypes;
        plug->CanHandleURL = AMR_CanHandleURL;
        plug->ConnectService = AMR_ConnectService;
        plug->CloseService = AMR_CloseService;
        plug->GetServiceDescriptor = AMR_GetServiceDesc;
        plug->ConnectChannel = AMR_ConnectChannel;
        plug->DisconnectChannel = AMR_DisconnectChannel;
        plug->ServiceCommand = AMR_ServiceCommand;
        /*we do support pull mode*/
        plug->ChannelGetSLP = AMR_ChannelGetSLP;
        plug->ChannelReleaseSLP = AMR_ChannelReleaseSLP;

        reader = (AMR_Reader*)gf_malloc(sizeof(AMR_Reader));
        memset(reader, 0, sizeof(AMR_Reader));
        plug->priv = reader;
        return plug;
}

void DeleteAESReader(void *ifce)
{
        GF_InputService *plug = (GF_InputService *) ifce;
        AMR_Reader *read = (AMR_Reader*)plug->priv;
        gf_free(read);
        gf_free(plug);
}


#ifdef GPAC_AMR_IN_STANDALONE

GPAC_MODULE_EXPORT
const u32 *QueryInterfaces()
{
        static u32 si [] = {
                GF_NET_CLIENT_INTERFACE,
                0
        };
        return si;
}

GPAC_MODULE_EXPORT
GF_BaseInterface *LoadInterface(u32 InterfaceType)
{
        switch (InterfaceType) {
        case GF_NET_CLIENT_INTERFACE:
                return (GF_BaseInterface *) NewAESReader();
        default:
                return NULL;
        }
}

GPAC_MODULE_EXPORT
void ShutdownInterface(GF_BaseInterface *ifce)
{
        switch (ifce->InterfaceType) {
        case GF_NET_CLIENT_INTERFACE:
                DeleteAESReader(ifce);
                break;
        }
}

GPAC_MODULE_STATIC_DECLARATION(amr_in)

#endif