//
//  PositionalAudioStream.cpp
//  libraries/audio/src
//
//  Created by Stephen Birarda on 6/5/13.
//  Copyright 2013 High Fidelity, Inc.
//
//  Distributed under the Apache License, Version 2.0.
//  See the accompanying file LICENSE or http://www.apache.org/licenses/LICENSE-2.0.html
//

#include "PositionalAudioStream.h"
#include "SharedUtil.h"

#include <cstring>

#include <glm/detail/func_common.hpp>
#include <QtCore/QDataStream>

#include <Node.h>
#include <udt/PacketHeaders.h>
#include <UUID.h>

PositionalAudioStream::PositionalAudioStream(PositionalAudioStream::Type type, bool isStereo, const InboundAudioStream::Settings& settings) :
    InboundAudioStream(isStereo
                       ? AudioConstants::NETWORK_FRAME_SAMPLES_STEREO
                       : AudioConstants::NETWORK_FRAME_SAMPLES_PER_CHANNEL,
    AUDIOMIXER_INBOUND_RING_BUFFER_FRAME_CAPACITY, settings),
    _type(type),
    _position(0.0f, 0.0f, 0.0f),
    _orientation(0.0f, 0.0f, 0.0f, 0.0f),
    _shouldLoopbackForNode(false),
    _isStereo(isStereo),
    _ignorePenumbra(false),
    _lastPopOutputTrailingLoudness(0.0f),
    _lastPopOutputLoudness(0.0f),
    _quietestTrailingFrameLoudness(std::numeric_limits<float>::max()),
    _quietestFrameLoudness(0.0f),
    _frameCounter(0)
{
}

void PositionalAudioStream::resetStats() {
    _lastPopOutputTrailingLoudness = 0.0f;
    _lastPopOutputLoudness = 0.0f;
}

void PositionalAudioStream::updateLastPopOutputLoudnessAndTrailingLoudness() {
    _lastPopOutputLoudness = _ringBuffer.getFrameLoudness(_lastPopOutput);

    const int TRAILING_MUTE_THRESHOLD_FRAMES = 400;
    const int TRAILING_LOUDNESS_FRAMES = 200;
    const float CURRENT_FRAME_RATIO = 1.0f / TRAILING_LOUDNESS_FRAMES;
    const float PREVIOUS_FRAMES_RATIO = 1.0f - CURRENT_FRAME_RATIO;
    const float LOUDNESS_EPSILON = 0.000001f;

    if (_lastPopOutputLoudness >= _lastPopOutputTrailingLoudness) {
        _lastPopOutputTrailingLoudness = _lastPopOutputLoudness;
    } else {
        _lastPopOutputTrailingLoudness = (_lastPopOutputTrailingLoudness * PREVIOUS_FRAMES_RATIO) + (CURRENT_FRAME_RATIO * _lastPopOutputLoudness);

        if (_lastPopOutputTrailingLoudness < LOUDNESS_EPSILON) {
            _lastPopOutputTrailingLoudness = 0;
        }
    }
    if (_frameCounter++ == TRAILING_MUTE_THRESHOLD_FRAMES) {
        _quietestFrameLoudness = _quietestTrailingFrameLoudness;
        _frameCounter = 0;
        _quietestTrailingFrameLoudness = std::numeric_limits<float>::max();

    }
    if (_lastPopOutputLoudness < _quietestTrailingFrameLoudness) {
        _quietestTrailingFrameLoudness = _lastPopOutputLoudness;
    }
}

int PositionalAudioStream::parsePositionalData(const QByteArray& positionalByteArray) {
    QDataStream packetStream(positionalByteArray);

    packetStream.readRawData(reinterpret_cast<char*>(&_position), sizeof(_position));
    packetStream.readRawData(reinterpret_cast<char*>(&_orientation), sizeof(_orientation));

    // if this node sent us a NaN for first float in orientation then don't consider this good audio and bail
    if (glm::isnan(_orientation.x)) {
        // NOTE: why would we reset the ring buffer here?
        _ringBuffer.reset();
        return 0;
    }

    return packetStream.device()->pos();
}

AudioStreamStats PositionalAudioStream::getAudioStreamStats() const {
    AudioStreamStats streamStats = InboundAudioStream::getAudioStreamStats();
    streamStats._streamType = _type;
    return streamStats;
}