/*
 * Copyright (C) 2020 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 <mutex>
#include <cutils/properties.h>
#include <log/log.h>
#include "talsa.h"
#include "debug.h"

namespace android {
namespace hardware {
namespace audio {
namespace CPP_VERSION {
namespace implementation {
namespace talsa {

namespace {

struct mixer *gMixer0 = nullptr;
int gMixerRefcounter0 = 0;
std::mutex gMixerMutex;
PcmPeriodSettings gPcmPeriodSettings;
unsigned gPcmHostLatencyMs;

void mixerSetValueAll(struct mixer_ctl *ctl, int value) {
    const unsigned int n = mixer_ctl_get_num_values(ctl);
    for (unsigned int i = 0; i < n; i++) {
        ::mixer_ctl_set_value(ctl, i, value);
    }
}

void mixerSetPercentAll(struct mixer_ctl *ctl, int percent) {
    const unsigned int n = mixer_ctl_get_num_values(ctl);
    for (unsigned int i = 0; i < n; i++) {
        ::mixer_ctl_set_percent(ctl, i, percent);
    }
}

struct mixer *mixerGetOrOpenImpl(const unsigned card,
                                 struct mixer *&gMixer,
                                 int &refcounter) {
    if (!gMixer) {
        struct mixer *mixer = ::mixer_open(card);
        if (!mixer) {
            return FAILURE(nullptr);
        }

        mixerSetPercentAll(::mixer_get_ctl_by_name(mixer, "Master Playback Volume"), 100);
        mixerSetPercentAll(::mixer_get_ctl_by_name(mixer, "Capture Volume"), 100);

        mixerSetValueAll(::mixer_get_ctl_by_name(mixer, "Master Playback Switch"), 1);
        mixerSetValueAll(::mixer_get_ctl_by_name(mixer, "Capture Switch"), 1);

        gMixer = mixer;
    }

    ++refcounter;
    return gMixer;
}

struct mixer *mixerGetOrOpen(const unsigned card) {
    std::lock_guard<std::mutex> guard(gMixerMutex);

    switch (card) {
    case 0:  return mixerGetOrOpenImpl(card, gMixer0, gMixerRefcounter0);
    default: return FAILURE(nullptr);
    }
}

bool mixerUnrefImpl(struct mixer *mixer, struct mixer *&gMixer, int &refcounter) {
    if (mixer == gMixer) {
        if (0 == --refcounter) {
            ::mixer_close(mixer);
            gMixer = nullptr;
        }
        return true;
    } else {
        return false;
    }
}

bool mixerUnref(struct mixer *mixer) {
    std::lock_guard<std::mutex> guard(gMixerMutex);

    return mixerUnrefImpl(mixer, gMixer0, gMixerRefcounter0);
}

unsigned readUnsignedProperty(const char *propName, const unsigned defaultValue) {
    char propValue[PROPERTY_VALUE_MAX];

    if (property_get(propName, propValue, nullptr) < 0) {
        return defaultValue;
    }

    unsigned value;
    return (sscanf(propValue, "%u", &value) == 1) ? value : defaultValue;
}
}  // namespace

void init() {
    gPcmPeriodSettings.periodCount =
        readUnsignedProperty("ro.hardware.audio.tinyalsa.period_count", 4);

    gPcmPeriodSettings.periodSizeMultiplier =
        readUnsignedProperty("ro.hardware.audio.tinyalsa.period_size_multiplier", 1);

    gPcmHostLatencyMs =
        readUnsignedProperty("ro.hardware.audio.tinyalsa.host_latency_ms", 0);
}

PcmPeriodSettings pcmGetPcmPeriodSettings() {
    return gPcmPeriodSettings;
}

unsigned pcmGetHostLatencyMs() {
    return gPcmHostLatencyMs;
}

void PcmDeleter::operator()(pcm_t *x) const {
    LOG_ALWAYS_FATAL_IF(::pcm_close(x) != 0);
};

std::unique_ptr<pcm_t, PcmDeleter> pcmOpen(const unsigned int dev,
                                           const unsigned int card,
                                           const unsigned int nChannels,
                                           const size_t sampleRateHz,
                                           const size_t frameCount,
                                           const bool isOut) {
    const PcmPeriodSettings periodSettings = pcmGetPcmPeriodSettings();

    struct pcm_config pcm_config;
    memset(&pcm_config, 0, sizeof(pcm_config));

    pcm_config.channels = nChannels;
    pcm_config.rate = sampleRateHz;
    // Approx interrupts per buffer
    pcm_config.period_count = periodSettings.periodCount;
    // Approx frames between interrupts
    pcm_config.period_size =
        periodSettings.periodSizeMultiplier * frameCount / periodSettings.periodCount;
    pcm_config.format = PCM_FORMAT_S16_LE;

    PcmPtr pcm =
        PcmPtr(::pcm_open(dev, card,
                          (isOut ? PCM_OUT : PCM_IN) | PCM_MONOTONIC,
                           &pcm_config));
    if (::pcm_is_ready(pcm.get())) {
        return pcm;
    } else {
        ALOGE("%s:%d pcm_open failed for nChannels=%u sampleRateHz=%zu "
              "period_count=%d period_size=%d isOut=%d with %s", __func__, __LINE__,
              nChannels, sampleRateHz, pcm_config.period_count, pcm_config.period_size, isOut,
              pcm_get_error(pcm.get()));
        return FAILURE(nullptr);
    }
}

bool pcmPrepare(pcm_t *pcm) {
    if (!pcm) {
        return FAILURE(false);
    }

    const int r = ::pcm_prepare(pcm);
    if (r) {
        ALOGE("%s:%d pcm_prepare failed with %s",
              __func__, __LINE__, ::pcm_get_error(pcm));
        return FAILURE(false);
    } else {
        return true;
    }
}

bool pcmStart(pcm_t *pcm) {
    if (!pcm) {
        return FAILURE(false);
    }

    const int r = ::pcm_start(pcm);
    if (r) {
        ALOGE("%s:%d pcm_start failed with %s",
              __func__, __LINE__, ::pcm_get_error(pcm));
        return FAILURE(false);
    } else {
        return true;
    }
}

bool pcmStop(pcm_t *pcm) {
    if (!pcm) {
        return FAILURE(false);
    }

    const int r = ::pcm_stop(pcm);
    if (r) {
        ALOGE("%s:%d pcm_stop failed with %s",
              __func__, __LINE__, ::pcm_get_error(pcm));
        return FAILURE(false);
    } else {
        return true;
    }
}

bool pcmRead(pcm_t *pcm, void *data, unsigned int count) {
    if (!pcm) {
        return FAILURE(false);
    }

    const int r = ::pcm_read(pcm, data, count);
    if (r) {
        ALOGE("%s:%d pcm_read failed with %s (%d)",
              __func__, __LINE__, ::pcm_get_error(pcm), r);
        return FAILURE(false);
    } else {
        return true;
    }
}

bool pcmWrite(pcm_t *pcm, const void *data, unsigned int count) {
    if (!pcm) {
        return FAILURE(false);
    }

    const int r = ::pcm_write(pcm, data, count);
    if (r) {
        ALOGE("%s:%d pcm_write failed with %s (%d)",
              __func__, __LINE__, ::pcm_get_error(pcm), r);
        return FAILURE(false);
    } else {
        return true;
    }
}

Mixer::Mixer(unsigned card): mMixer(mixerGetOrOpen(card)) {}

Mixer::~Mixer() {
    if (mMixer) {
        LOG_ALWAYS_FATAL_IF(!mixerUnref(mMixer));
    }
}

}  // namespace talsa
}  // namespace implementation
}  // namespace CPP_VERSION
}  // namespace audio
}  // namespace hardware
}  // namespace android