Fix audioloop test program

#include "SineSource.h"

/*

 * Copyright (C) 2014 The Android Open Source Project

 *

 * Licensed under the Apache License, Version 2.0 (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.apache.org/licenses/LICENSE-2.0

 *

 * Unless required by applicable law or agreed to in writing, software

 * distributed under the License is distributed on an "AS IS" BASIS,

 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

 * See the License for the specific language governing permissions and

 * limitations under the License.

 */

#include <binder/ProcessState.h>

#include <media/mediarecorder.h>

#include <media/stagefright/MetaData.h>

#include <media/stagefright/OMXClient.h>

#include <media/stagefright/OMXCodec.h>

#include <system/audio.h>

#include "SineSource.h"

using namespace android;

int main() {

    // We only have an AMR-WB encoder on sholes...

    static bool outputWBAMR = false;

    static const int32_t kSampleRate = outputWBAMR ? 16000 : 8000;

    static const int32_t kNumChannels = 1;

static void usage(const char* name)

{

    fprintf(stderr, "Usage: %s [-d duration] [-m] [-w] [<output-file>]\n", name);

    fprintf(stderr, "Encodes either a sine wave or microphone input to AMR format\n");

    fprintf(stderr, "    -d    duration in seconds, default 5 seconds\n");

    fprintf(stderr, "    -m    use microphone for input, default sine source\n");

    fprintf(stderr, "    -w    use AMR wideband (default narrowband)\n");

    fprintf(stderr, "    <output-file> output file for AMR encoding,"

            " if unspecified, decode to speaker.\n");

}

    android::ProcessState::self()->startThreadPool();

int main(int argc, char* argv[])

{

    static const int channels = 1; // not permitted to be stereo now

    unsigned duration = 5;

    bool useMic = false;

    bool outputWBAMR = false;

    bool playToSpeaker = true;

    const char* fileOut = NULL;

    int ch;

    while ((ch = getopt(argc, argv, "d:mw")) != -1) {

        switch (ch) {

        case 'd':

            duration = atoi(optarg);

            break;

        case 'm':

            useMic = true;

            break;

        case 'w':

            outputWBAMR = true;

            break;

        default:

            usage(argv[0]);

            return -1;

        }

    }

    argc -= optind;

    argv += optind;

    if (argc == 1) {

        fileOut = argv[0];

    }

    const int32_t kSampleRate = outputWBAMR ? 16000 : 8000;

    const int32_t kBitRate = outputWBAMR ? 16000 : 8000;

    android::ProcessState::self()->startThreadPool();

    OMXClient client;

    CHECK_EQ(client.connect(), (status_t)OK);

    sp<MediaSource> source;

#if 0

    sp<MediaSource> source = new SineSource(kSampleRate, kNumChannels);

#else

    sp<MediaSource> source = new AudioSource(

            AUDIO_SOURCE_DEFAULT,

    if (useMic) {

        // talk into the appropriate microphone for the duration

        source = new AudioSource(

                AUDIO_SOURCE_MIC,

                kSampleRate,

            audio_channel_in_mask_from_count(kNumChannels));

#endif

                channels);

    } else {

        // use a sine source at 500 hz.

        source = new SineSource(kSampleRate, channels);

    }

    sp<MetaData> meta = new MetaData;

    meta->setCString(

            kKeyMIMEType,

            outputWBAMR ? MEDIA_MIMETYPE_AUDIO_AMR_WB

                    : MEDIA_MIMETYPE_AUDIO_AMR_NB);

    meta->setInt32(kKeyChannelCount, kNumChannels);

    meta->setInt32(kKeyChannelCount, channels);

    meta->setInt32(kKeySampleRate, kSampleRate);

    meta->setInt32(kKeyBitRate, kBitRate);

    int32_t maxInputSize;

    if (source->getFormat()->findInt32(kKeyMaxInputSize, &maxInputSize)) {

        meta->setInt32(kKeyMaxInputSize, maxInputSize);

            meta, true /* createEncoder */,

            source);

#if 1

    sp<AMRWriter> writer = new AMRWriter("/sdcard/out.amr");

    if (fileOut != NULL) {

        // target file specified, write encoded AMR output

        sp<AMRWriter> writer = new AMRWriter(fileOut);

        writer->addSource(encoder);

        writer->start();

    sleep(10);

        sleep(duration);

        writer->stop();

#else

    } else {

        // otherwise decode to speaker

        sp<MediaSource> decoder = OMXCodec::Create(

                client.interface(),

                meta, false /* createEncoder */,

                encoder);

#if 0

        if (playToSpeaker) {

            AudioPlayer *player = new AudioPlayer(NULL);

            player->setSource(decoder);

            player->start();

    sleep(10);

    player->stop();

    delete player;

    player = NULL;

#elif 0

            sleep(duration);

            source->stop(); // must stop source otherwise delete player will hang

            delete player; // there is no player->stop()...

        } else {

            CHECK_EQ(decoder->start(), (status_t)OK);

            MediaBuffer* buffer;

            while (decoder->read(&buffer) == OK) {

        // do something with buffer

                // do something with buffer (save it eventually?)

                // need to stop after some count though...

                putchar('.');

                fflush(stdout);

                buffer->release();

                buffer = NULL;

            }

            CHECK_EQ(decoder->stop(), (status_t)OK);

#endif

#endif

        }

    }

    return 0;

}