#include "WavWriter.h"

WavWriter::WavWriter(std::ostream& stream)
    : m_stream(stream)
{
}

void WavWriter::WritePcmHeader(const WavMetaData& metaData, const size_t dataLen) const
{
    constexpr auto riffMasterChunkSize = sizeof(WAV_CHUNK_ID_RIFF) + sizeof(uint32_t) + sizeof(WAV_WAVE_ID) + sizeof(WavChunkHeader) + sizeof(WavFormatChunkPcm)
                                         + sizeof(WavChunkHeader) + sizeof(dataLen);

    m_stream.write(reinterpret_cast<const char*>(&WAV_CHUNK_ID_RIFF), sizeof(WAV_CHUNK_ID_RIFF));
    m_stream.write(reinterpret_cast<const char*>(&riffMasterChunkSize), sizeof(riffMasterChunkSize));
    m_stream.write(reinterpret_cast<const char*>(&WAV_WAVE_ID), sizeof(WAV_WAVE_ID));

    constexpr WavChunkHeader formatChunkHeader{WAV_CHUNK_ID_FMT, sizeof(WavFormatChunkPcm)};
    m_stream.write(reinterpret_cast<const char*>(&formatChunkHeader), sizeof(formatChunkHeader));

    const WavFormatChunkPcm formatChunk{
        WavFormat::PCM,
        static_cast<uint16_t>(metaData.channelCount),
        metaData.samplesPerSec,
        metaData.samplesPerSec * metaData.channelCount * metaData.bitsPerSample / 8,
        static_cast<uint16_t>(metaData.channelCount * (metaData.bitsPerSample / 8)),
        static_cast<uint16_t>(metaData.bitsPerSample),
    };
    m_stream.write(reinterpret_cast<const char*>(&formatChunk), sizeof(formatChunk));

    const WavChunkHeader dataChunkHeader{WAV_CHUNK_ID_DATA, dataLen};
    m_stream.write(reinterpret_cast<const char*>(&dataChunkHeader), sizeof(dataChunkHeader));
}

void WavWriter::WritePcmData(const void* data, const size_t dataLen) const
{
    m_stream.write(static_cast<const char*>(data), dataLen);
}