source: mainline/uspace/app/drec/drec.c@ e1b7e36

/*
 * Copyright (c) 2011 Jan Vesely
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 * - Redistributions of source code must retain the above copyright
 *   notice, this list of conditions and the following disclaimer.
 * - Redistributions in binary form must reproduce the above copyright
 *   notice, this list of conditions and the following disclaimer in the
 *   documentation and/or other materials provided with the distribution.
 * - The name of the author may not be used to endorse or promote products
 *   derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

/** @addtogroup dplay
 * @{
 */
/**
 * @file PCM playback audio devices
 */

#include <assert.h>
#include <errno.h>
#include <str_error.h>
#include <str.h>
#include <devman.h>
#include <audio_pcm_buffer_iface.h>
#include <fibril_synch.h>
#include <stdio.h>
#include <sys/mman.h>
#include <sys/time.h>

#include <stdio.h>
#include <macros.h>

#include "wave.h"

#define DEFAULT_DEVICE "/hw/pci0/00:01.0/sb16/dsp"
#define SUBBUFFERS 2

const unsigned sampling_rate = 44100, sample_size = 16, channels = 2;
bool sign = true;

typedef struct {
        struct {
                void *base;
                size_t size;
                unsigned id;
                void* position;
        } buffer;
        FILE* file;
        async_exch_t *device;
} record_t;

static void record_initialize(record_t *rec, async_exch_t *exch)
{
        assert(exch);
        assert(rec);
        rec->buffer.id = 0;
        rec->buffer.base = NULL;
        rec->buffer.size = 0;
        rec->buffer.position = NULL;
        rec->file = NULL;
        rec->device = exch;
}


static void device_event_callback(ipc_callid_t iid, ipc_call_t *icall, void* arg)
{
        async_answer_0(iid, EOK);
        record_t *rec = arg;
        const size_t buffer_part = rec->buffer.size / SUBBUFFERS;
        while (1) {
                ipc_call_t call;
                ipc_callid_t callid = async_get_call(&call);
                if (IPC_GET_IMETHOD(call) != IPC_FIRST_USER_METHOD) {
                        printf("Unknown event.\n");
                        break;
                }
                const size_t bytes = fwrite(rec->buffer.position,
                   sizeof(uint8_t), buffer_part, rec->file);
                printf("%zu ", bytes);
                rec->buffer.position += buffer_part;

                if (rec->buffer.position >= (rec->buffer.base + rec->buffer.size))
                        rec->buffer.position = rec->buffer.base;
                async_answer_0(callid, EOK);
        }
}


static void record(record_t *rec, unsigned sampling_rate, unsigned sample_size,
    unsigned channels, bool sign)
{
        assert(rec);
        assert(rec->device);
        rec->buffer.position = rec->buffer.base;
        printf("Recording: %dHz, %d-bit %ssigned samples, %d channel(s).\n",
            sampling_rate, sample_size, sign ? "": "un", channels);
        int ret = audio_pcm_buffer_start_record(rec->device, rec->buffer.id,
            SUBBUFFERS, sampling_rate, sample_size, channels, sign);
        if (ret != EOK) {
                printf("Failed to start recording: %s.\n", str_error(ret));
                return;
        }

        getchar();
        printf("\n");
        audio_pcm_buffer_stop_record(rec->device, rec->buffer.id);
}

int main(int argc, char *argv[])
{
        const char *device = DEFAULT_DEVICE;
        const char *file;
        switch (argc) {
        case 2:
                file = argv[1];
                break;
        case 3:
                device = argv[1];
                file = argv[2];
                break;
        default:
                printf("Usage: %s [device] file.\n", argv[0]);
                return 1;
        }

        devman_handle_t pcm_handle;
        int ret = devman_fun_get_handle(device, &pcm_handle, 0);
        if (ret != EOK) {
                printf("Failed to get device(%s) handle: %s.\n",
                    device, str_error(ret));
                return 1;
        }

        async_sess_t *session = devman_device_connect(
            EXCHANGE_SERIALIZE, pcm_handle, IPC_FLAG_BLOCKING);
        if (!session) {
                printf("Failed to connect to device.\n");
                return 1;
        }

        async_exch_t *exch = async_exchange_begin(session);
        if (!exch) {
                ret = EPARTY;
                printf("Failed to start session exchange.\n");
                goto close_session;
        }
        const char* info = NULL;
        ret = audio_pcm_buffer_get_info_str(exch, &info);
        if (ret != EOK) {
                printf("Failed to get PCM info.\n");
                goto close_session;
        }
        printf("Playing on %s.\n", info);
        free(info);

        record_t rec;
        record_initialize(&rec, exch);

        ret = audio_pcm_buffer_get_buffer(rec.device, &rec.buffer.base,
            &rec.buffer.size, &rec.buffer.id, device_event_callback, &rec);
        if (ret != EOK) {
                printf("Failed to get PCM buffer: %s.\n", str_error(ret));
                goto close_session;
        }
        printf("Buffer (%u): %p %zu.\n", rec.buffer.id, rec.buffer.base,
            rec.buffer.size);
        uintptr_t ptr = 0;
        as_get_physical_mapping(rec.buffer.base, &ptr);
        printf("buffer mapped at %x.\n", ptr);

        rec.file = fopen(file, "wb");
        if (rec.file == NULL) {
                ret = ENOENT;
                printf("Failed to open %s.\n", file);
                goto cleanup;
        }
        wave_header_t header = {
                .chunk_id = CHUNK_ID,
                .format = FORMAT_STR,
                .subchunk1_id = SUBCHUNK1_ID,
                .subchunk1_size = PCM_SUBCHUNK1_SIZE,
                .audio_format = FORMAT_LINEAR_PCM,
                .channels = channels,
                .sampling_rate = sampling_rate,
                .sample_size = sample_size,
                .byte_rate = (sampling_rate / 8) * channels,
                .block_align = (sampling_rate / 8) * channels,
                .subchunk2_id = SUBCHUNK2_ID,
        };
        fwrite(&header, sizeof(header), 1, rec.file);
        record(&rec, sampling_rate, sample_size, channels, sign);
        fclose(rec.file);

cleanup:
        munmap(rec.buffer.base, rec.buffer.size);
        audio_pcm_buffer_release_buffer(exch, rec.buffer.id);
close_session:
        async_exchange_end(exch);
        async_hangup(session);
        return ret == EOK ? 0 : 1;
}
/**
 * @}
 */