diff --git a/Libraries/LibAudio/Makefile b/Libraries/LibAudio/Makefile index 6a043b51b7f..51453a9665e 100644 --- a/Libraries/LibAudio/Makefile +++ b/Libraries/LibAudio/Makefile @@ -1,5 +1,6 @@ OBJS = \ ClientConnection.o \ + WavWriter.o \ WavLoader.o LIBRARY = libaudio.a diff --git a/Libraries/LibAudio/WavWriter.cpp b/Libraries/LibAudio/WavWriter.cpp new file mode 100644 index 00000000000..4741c963d67 --- /dev/null +++ b/Libraries/LibAudio/WavWriter.cpp @@ -0,0 +1,124 @@ +/* + * Copyright (c) 2020-2020, William McPherson + * All rights reserved. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions are met: + * + * 1. Redistributions of source code must retain the above copyright notice, this + * list of conditions and the following disclaimer. + * + * 2. 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. + * + * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "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 COPYRIGHT HOLDER OR CONTRIBUTORS 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. + */ + +#include + +namespace Audio { + +WavWriter::WavWriter(const StringView& path, int sample_rate, int num_channels, int bits_per_sample) + : m_sample_rate(sample_rate) + , m_num_channels(num_channels) + , m_bits_per_sample(bits_per_sample) +{ + set_file(path); +} + +WavWriter::WavWriter(int sample_rate, int num_channels, int bits_per_sample) + : m_sample_rate(sample_rate) + , m_num_channels(num_channels) + , m_bits_per_sample(bits_per_sample) +{ +} + +WavWriter::~WavWriter() +{ + if (!m_finalized) + finalize(); +} + +void WavWriter::set_file(const StringView& path) +{ + m_file = Core::File::construct(path); + if (!m_file->open(Core::IODevice::ReadWrite)) { + m_error_string = String::format("Can't open file: %s", m_file->error_string()); + return; + } + m_file->seek(44); + m_finalized = false; +} + +void WavWriter::write_samples(const u8* samples, size_t size) +{ + m_data_sz += size; + m_file->write(samples, size); +} + +void WavWriter::finalize() +{ + ASSERT(!m_finalized); + m_finalized = true; + m_file->seek(0); + write_header(); + m_file->close(); + m_data_sz = 0; +} + +void WavWriter::write_header() +{ + // "RIFF" + static u32 riff = 0x46464952; + m_file->write(reinterpret_cast(&riff), sizeof(riff)); + + // Size of data + (size of header - previous field - this field) + u32 sz = m_data_sz + (44 - 4 - 4); + m_file->write(reinterpret_cast(&sz), sizeof(sz)); + + // "WAVE" + static u32 wave = 0x45564157; + m_file->write(reinterpret_cast(&wave), sizeof(wave)); + + // "fmt " + static u32 fmt_id = 0x20746D66; + m_file->write(reinterpret_cast(&fmt_id), sizeof(fmt_id)); + + // Size of the next 6 fields + static u32 fmt_size = 16; + m_file->write(reinterpret_cast(&fmt_size), sizeof(fmt_size)); + + // 1 for PCM + static u16 audio_format = 1; + m_file->write(reinterpret_cast(&audio_format), sizeof(audio_format)); + + m_file->write(reinterpret_cast(&m_num_channels), sizeof(m_num_channels)); + + m_file->write(reinterpret_cast(&m_sample_rate), sizeof(m_sample_rate)); + + u32 byte_rate = m_sample_rate * m_num_channels * (m_bits_per_sample / 8); + m_file->write(reinterpret_cast(&byte_rate), sizeof(byte_rate)); + + u16 block_align = m_num_channels * (m_bits_per_sample / 8); + m_file->write(reinterpret_cast(&block_align), sizeof(block_align)); + + m_file->write(reinterpret_cast(&m_bits_per_sample), sizeof(m_bits_per_sample)); + + // "data" + static u32 chunk_id = 0x61746164; + m_file->write(reinterpret_cast(&chunk_id), sizeof(chunk_id)); + + m_file->write(reinterpret_cast(&m_data_sz), sizeof(m_data_sz)); +} + +} diff --git a/Libraries/LibAudio/WavWriter.h b/Libraries/LibAudio/WavWriter.h new file mode 100644 index 00000000000..bab7e297bd4 --- /dev/null +++ b/Libraries/LibAudio/WavWriter.h @@ -0,0 +1,70 @@ +/* + * Copyright (c) 2020-2020, William McPherson + * All rights reserved. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions are met: + * + * 1. Redistributions of source code must retain the above copyright notice, this + * list of conditions and the following disclaimer. + * + * 2. 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. + * + * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "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 COPYRIGHT HOLDER OR CONTRIBUTORS 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. + */ + +#pragma once + +#include +#include + +namespace Audio { + +class WavWriter { +public: + WavWriter(const StringView& path, int sample_rate = 44100, int num_channels = 2, int bits_per_sample = 16); + WavWriter(int sample_rate = 44100, int num_channels = 2, int bits_per_sample = 16); + ~WavWriter(); + + bool has_error() const { return !m_error_string.is_null(); } + const char* error_string() const { return m_error_string.characters(); } + + void write_samples(const u8* samples, size_t size); + void finalize(); // You can finalize manually or let the destructor do it. + + u32 sample_rate() const { return m_sample_rate; } + u16 num_channels() const { return m_num_channels; } + u16 bits_per_sample() const { return m_bits_per_sample; } + RefPtr file() const { return m_file; } + + void set_file(const StringView& path); + void set_num_channels(int num_channels) { m_num_channels = num_channels; } + void set_sample_rate(int sample_rate) { m_sample_rate = sample_rate; } + void set_bits_per_sample(int bits_per_sample) { m_bits_per_sample = bits_per_sample; } + + void clear_error() { m_error_string = String(); } + +private: + void write_header(); + RefPtr m_file; + String m_error_string; + bool m_finalized { false }; + + u32 m_sample_rate; + u16 m_num_channels; + u16 m_bits_per_sample; + u32 m_data_sz { 0 }; +}; + +}