fs_encode.c

Go to the documentation of this file.

/*
 * FreeSWITCH Modular Media Switching Software Library / Soft-Switch Application
 * Copyright (C) 2005-2015, Anthony Minessale II <anthm@freeswitch.org>
 *
 * Version: MPL 1.1
 *
 * The contents of this file are subject to the Mozilla Public License Version
 * 1.1 (the "License"); you may not use this file except in compliance with
 * the License. You may obtain a copy of the License at
 * http://www.mozilla.org/MPL/
 *
 * Software distributed under the License is distributed on an "AS IS" basis,
 * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
 * for the specific language governing rights and limitations under the
 * License.
 *
 * The Original Code is FreeSWITCH Modular Media Switching Software Library / Soft-Switch Application
 *
 * The Initial Developer of the Original Code is
 * Anthony Minessale II <anthm@freeswitch.org>
 * Portions created by the Initial Developer are Copyright (C)
 * the Initial Developer. All Rights Reserved.
 *
 * Contributor(s):
 *
 * Mathieu Rene <mrene@avgs.ca>
 *
 * fs_encode.c -- Encode a native file
 *
 */

#ifndef _XOPEN_SOURCE
#define _XOPEN_SOURCE 600
#endif

#ifndef WIN32
#ifdef HAVE_SETRLIMIT
#include <sys/resource.h>
#endif
#endif

#include <switch.h>

/* Picky compiler */
#ifdef __ICC
#pragma warning (disable:167)
#endif

static void fs_encode_cleanup(void)
{
        switch_safe_free(SWITCH_GLOBAL_dirs.conf_dir);
        switch_safe_free(SWITCH_GLOBAL_dirs.mod_dir);
        switch_safe_free(SWITCH_GLOBAL_dirs.log_dir);
}

int main(int argc, char *argv[])
{
        int r = 1;
        switch_bool_t verbose = SWITCH_FALSE;
        const char *err = NULL;
        int i;
        char *extra_modules[100] = { 0 };
        int extra_modules_count = 0;
        int cmd_fail = 0;
        const char *fmtp = "";
        int ptime = 20;
        const char *input, *output, *format = NULL;
        int channels = 1;
        int rate = 8000;
        switch_file_handle_t fh_input = { 0 }, fh_output = { 0 };
        switch_codec_t codec = { 0 };
        switch_codec_t raw_codec = { 0 };
        char buf[2048];
        switch_size_t len = sizeof(buf)/2;
        switch_memory_pool_t *pool = NULL;
        int bitrate = 0;
        int blocksize;
        int in_asis = 0;
        int out_asis = 0;
        int out_flags = SWITCH_FILE_FLAG_WRITE;

        for (i = 1; i < argc; i++) {
                if (argv[i][0] == '-') {
                        switch(argv[i][1]) {
                                case 'c':
                                        i++;
                                        if((SWITCH_GLOBAL_dirs.conf_dir = (char *) malloc(strlen(argv[i]) + 1)) == NULL) {
                                                return 255;
                                        }
                                        strcpy(SWITCH_GLOBAL_dirs.conf_dir, argv[i]);
                                        break;
                                case 'k':
                                        i++;
                                        if((SWITCH_GLOBAL_dirs.log_dir = (char *) malloc(strlen(argv[i]) + 1)) == NULL) {
                                                return 255;
                                        }
                                        strcpy(SWITCH_GLOBAL_dirs.log_dir, argv[i]);
                                        break;
                                case 'm':
                                        i++;
                                        if((SWITCH_GLOBAL_dirs.mod_dir = (char *) malloc(strlen(argv[i]) + 1)) == NULL) {
                                                return 255;
                                        }
                                        strcpy(SWITCH_GLOBAL_dirs.mod_dir, argv[i]);
                                        break;
                                case 'l':
                                        i++;
                                        /* Load extra modules */
                                        if (strchr(argv[i], ','))  {
                                                extra_modules_count = switch_split(argv[i], ',', extra_modules);
                                        } else {
                                                extra_modules_count = 1;
                                                extra_modules[0] = argv[i];
                                        }
                                        break;
                                case 'f':
                                        fmtp = argv[++i];
                                        break;
                                case 'p':
                                        ptime = atoi(argv[++i]);
                                        break;
                                case 'r':
                                        rate = atoi(argv[++i]);
                                        break;
                                case 'b':
                                        bitrate = atoi(argv[++i]);
                                        break;
                                case 'v':
                                        verbose = SWITCH_TRUE;
                                        break;
                                default:
                                        printf("Command line option not recognized: %s\n", argv[i]);
                                        cmd_fail = 1;
                        }
                } else {
                        break;
                }
        }

        if (argc - i < 2 || cmd_fail) {
                goto usage;
        }

        input = argv[i++];
        output = argv[i++];
        if (zstr(input) || zstr(output) || !(format = strchr(output, '.'))) {
                goto usage;
        }

        format++;

        if (switch_core_init(SCF_MINIMAL, verbose, &err) != SWITCH_STATUS_SUCCESS) {
                fprintf(stderr, "Cannot init core [%s]\n", err);
                goto end;
        }

        switch_loadable_module_init(SWITCH_FALSE);
        switch_loadable_module_load_module("", "CORE_PCM_MODULE", SWITCH_TRUE, &err);
        switch_loadable_module_load_module("", "CORE_SPEEX_MODULE", SWITCH_TRUE, &err);
        switch_loadable_module_load_module("", "CORE_SOFTTIMER_MODULE", SWITCH_TRUE, &err);

        for (i = 0; i < extra_modules_count; i++) {
                if (switch_loadable_module_load_module((char *) SWITCH_GLOBAL_dirs.mod_dir, (char *) extra_modules[i], SWITCH_TRUE, &err) != SWITCH_STATUS_SUCCESS) {
                        fprintf(stderr, "Cannot init %s [%s]\n", extra_modules[i], err);
                        goto end;
                }
        }

        if (switch_loadable_module_load_module((char *) SWITCH_GLOBAL_dirs.mod_dir, (char *) "mod_spandsp", SWITCH_TRUE, &err) != SWITCH_STATUS_SUCCESS) {
                fprintf(stderr, "Cannot init mod_spandsp [%s]\n", err);
                goto end;
        }

        if (switch_loadable_module_load_module((char *) SWITCH_GLOBAL_dirs.mod_dir, (char *) "mod_sndfile", SWITCH_TRUE, &err) != SWITCH_STATUS_SUCCESS) {
                fprintf(stderr, "Cannot init mod_sndfile [%s]\n", err);
                goto end;
        }

        if (switch_loadable_module_load_module((char *) SWITCH_GLOBAL_dirs.mod_dir, (char *) "mod_native_file", SWITCH_TRUE, &err) != SWITCH_STATUS_SUCCESS) {
                fprintf(stderr, "Cannot init mod_native_file [%s]\n", err);
                goto end;
        }

        switch_core_new_memory_pool(&pool);
        if (verbose) {
                fprintf(stderr, "Opening file %s\n", input);
        }
        if (switch_core_file_open(&fh_input, input, channels, rate, SWITCH_FILE_FLAG_READ | SWITCH_FILE_DATA_SHORT, NULL) != SWITCH_STATUS_SUCCESS) {
                fprintf(stderr, "Couldn't open %s\n", input);
                goto end;
        }


        if (verbose) {
                fprintf(stderr, "Opening file %s\n", output);
        }

        if (switch_stristr(".wav", output)) {
                out_asis = 0;
                out_flags |= SWITCH_FILE_DATA_SHORT;
        } else {
                out_asis = 1;
                out_flags |= SWITCH_FILE_NATIVE;
        }


        if (out_asis) {
                if (switch_core_codec_init_with_bitrate(&codec, format, NULL, fmtp, rate, ptime, channels, bitrate, SWITCH_CODEC_FLAG_ENCODE, NULL, pool) != SWITCH_STATUS_SUCCESS) {
                        fprintf(stderr, "Couldn't initialize codec for %s@%dh@%di\n", format, rate, ptime);
                        goto end;
                }
        } else {
                char *p;

                if ((p = strchr(input, '.'))) {
                        p++;
                }
                if (!p || switch_core_codec_init_with_bitrate(&codec, p, NULL, fmtp, rate, ptime, channels, bitrate, SWITCH_CODEC_FLAG_ENCODE|SWITCH_CODEC_FLAG_DECODE, NULL, pool) != SWITCH_STATUS_SUCCESS) {
                        fprintf(stderr, "Couldn't initialize codec for %s@%dh@%di\n", p, rate, ptime);
                        goto end;
                }

                if (switch_core_codec_init_with_bitrate(&raw_codec, "L16", NULL, fmtp, rate, ptime, channels, bitrate, SWITCH_CODEC_FLAG_ENCODE|SWITCH_CODEC_FLAG_DECODE, NULL, pool) != SWITCH_STATUS_SUCCESS) {
                fprintf(stderr, "Couldn't initialize codec for %s@%dh@%di\n", "L16", rate, ptime);
                goto end;
                }
        }




        if (switch_core_file_open(&fh_output, output, channels, codec.implementation->actual_samples_per_second, out_flags, NULL) != SWITCH_STATUS_SUCCESS) {
                fprintf(stderr, "Couldn't open %s\n", output);
                goto end;
        }

        if (switch_test_flag(&fh_input, SWITCH_FILE_NATIVE)) {
                in_asis = 1;
        }



        if (in_asis) {
                blocksize = len = codec.implementation->encoded_bytes_per_packet;
        } else {
                blocksize = len = (rate*ptime)/1000;
        }

        switch_assert(sizeof(buf) >= len * 2);

        if (verbose) {
                fprintf(stderr, "Frame size is %d\n", blocksize);
        }

        while (switch_core_file_read(&fh_input, buf, &len) == SWITCH_STATUS_SUCCESS) {
                char encode_buf[2048];
                uint32_t encoded_len = sizeof(buf);
                uint32_t encoded_rate = rate;
                unsigned int flags = 0;

                if (out_asis) {
                        if (switch_core_codec_encode(&codec, NULL, buf, len*2, rate, encode_buf, &encoded_len, &encoded_rate, &flags) != SWITCH_STATUS_SUCCESS) {
                                fprintf(stderr, "Codec encoder error\n");
                                goto end;
                        }

                        len = encoded_len;
                } else {
                        if (!in_asis) {
                                encoded_len = len;
                        } else if (in_asis) {

                                switch_core_codec_decode(&codec,
                                                                                 &raw_codec,
                                                                                 buf,
                                                                                 len,
                                                                                 rate,
                                                                                 encode_buf,
                                                                                 &encoded_len,
                                                                                 &encoded_rate,
                                                                                 &flags);
                                encoded_len /= 2;
                                len = encoded_len;
                        }
                }


                if (switch_core_file_write(&fh_output, encode_buf, &len) != SWITCH_STATUS_SUCCESS) {
                        fprintf(stderr, "Write error\n");
                        goto end;
                }

                if (len != encoded_len) {
                        printf("Short write: wrote %"SWITCH_SIZE_T_FMT"/%d bytes\n", len, encoded_len);
                }

                len = blocksize;
        }

        r = 0;

end:


        switch_core_codec_destroy(&codec);
        switch_core_codec_destroy(&raw_codec);


        if (fh_input.file_interface) {
                switch_core_file_close(&fh_input);
        }

        if (fh_output.file_interface) {
                switch_core_file_close(&fh_output);
        }

        if (pool) {
                switch_core_destroy_memory_pool(&pool);
        }

        fs_encode_cleanup();

        //switch_core_destroy();

        return r;
usage:
        printf("Usage: %s [options] input output\n\n", argv[0]);
        printf("The output must end in the format, e.g., myfile.SPEEX\n");
        printf("\t\t -c path\t\t Path to the FS configurations.\n");
        printf("\t\t -k path\t\t Path to the FS log directory\n");
        printf("\t\t -l module[,module]\t Load additional modules (comma-separated)\n");
        printf("\t\t -m path\t\t Path to the modules.\n");
        printf("\t\t -f format\t\t fmtp to pass to the codec\n");
        printf("\t\t -p ptime\t\t ptime to use while encoding\n");
        printf("\t\t -r rate\t\t sampling rate\n");
        printf("\t\t -b bitrate\t\t codec bitrate (if supported)\n");
        printf("\t\t -v\t\t\t verbose\n");
        fs_encode_cleanup();
        return 1;
}

/* For Emacs:
 * Local Variables:
 * mode:c
 * indent-tabs-mode:t
 * tab-width:4
 * c-basic-offset:4
 * End:
 * For VIM:
 * vim:set softtabstop=4 shiftwidth=4 tabstop=4 noet:
 */