mirror of
https://github.com/RGBCube/serenity
synced 2025-10-31 21:42:43 +00:00
b33a71a35e
We can just create a `BigEndianInputBitStream` where it is needed instead of storing one in the class. After making `read_header()` static, the only other user of the field was `read_size_information()`, so let's do that there and then remove the field.
78 lines
3.4 KiB
C++
78 lines
3.4 KiB
C++
/*
|
|
* Copyright (c) 2021, Arne Elster <arne@elster.li>
|
|
*
|
|
* SPDX-License-Identifier: BSD-2-Clause
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include "Loader.h"
|
|
#include "MP3Types.h"
|
|
#include <AK/BitStream.h>
|
|
#include <AK/MemoryStream.h>
|
|
#include <AK/Tuple.h>
|
|
#include <LibDSP/MDCT.h>
|
|
|
|
namespace Audio {
|
|
|
|
namespace MP3::Tables {
|
|
struct ScaleFactorBand;
|
|
}
|
|
|
|
class MP3LoaderPlugin : public LoaderPlugin {
|
|
public:
|
|
explicit MP3LoaderPlugin(NonnullOwnPtr<SeekableStream> stream);
|
|
virtual ~MP3LoaderPlugin() = default;
|
|
|
|
static bool sniff(SeekableStream& stream);
|
|
static ErrorOr<NonnullOwnPtr<LoaderPlugin>, LoaderError> create(NonnullOwnPtr<SeekableStream>);
|
|
|
|
virtual ErrorOr<Vector<FixedArray<Sample>>, LoaderError> load_chunks(size_t samples_to_read_from_input) override;
|
|
|
|
virtual MaybeLoaderError reset() override;
|
|
virtual MaybeLoaderError seek(int const position) override;
|
|
|
|
virtual int loaded_samples() override { return m_loaded_samples; }
|
|
virtual int total_samples() override { return m_total_samples; }
|
|
virtual u32 sample_rate() override { return m_sample_rate; }
|
|
virtual u16 num_channels() override { return m_num_channels; }
|
|
virtual PcmSampleFormat pcm_format() override { return m_sample_format; }
|
|
virtual DeprecatedString format_name() override { return "MP3 (.mp3)"; }
|
|
|
|
private:
|
|
MaybeLoaderError initialize();
|
|
static MaybeLoaderError skip_id3(SeekableStream& stream);
|
|
static MaybeLoaderError synchronize(SeekableStream& stream, size_t sample_index);
|
|
static ErrorOr<MP3::Header, LoaderError> read_header(SeekableStream& stream, size_t sample_index);
|
|
static ErrorOr<MP3::Header, LoaderError> synchronize_and_read_header(SeekableStream& stream, size_t sample_index);
|
|
ErrorOr<MP3::Header, LoaderError> synchronize_and_read_header();
|
|
MaybeLoaderError build_seek_table();
|
|
ErrorOr<MP3::MP3Frame, LoaderError> read_next_frame();
|
|
ErrorOr<MP3::MP3Frame, LoaderError> read_frame_data(MP3::Header const&);
|
|
MaybeLoaderError read_side_information(MP3::MP3Frame&);
|
|
ErrorOr<size_t, LoaderError> read_scale_factors(MP3::MP3Frame&, BigEndianInputBitStream& reservoir, size_t granule_index, size_t channel_index);
|
|
MaybeLoaderError read_huffman_data(MP3::MP3Frame&, BigEndianInputBitStream& reservoir, size_t granule_index, size_t channel_index, size_t granule_bits_read);
|
|
static AK::Array<float, 576> calculate_frame_exponents(MP3::MP3Frame const&, size_t granule_index, size_t channel_index);
|
|
static void reorder_samples(MP3::Granule&, u32 sample_rate);
|
|
static void reduce_alias(MP3::Granule&, size_t max_subband_index = 576);
|
|
static void process_stereo(MP3::MP3Frame&, size_t granule_index);
|
|
static void transform_samples_to_time(Array<float, 576> const& input, size_t input_offset, Array<float, 36>& output, MP3::BlockType block_type);
|
|
static void synthesis(Array<float, 1024>& V, Array<float, 32>& samples, Array<float, 32>& result);
|
|
static ReadonlySpan<MP3::Tables::ScaleFactorBand> get_scalefactor_bands(MP3::Granule const&, int samplerate);
|
|
|
|
SeekTable m_seek_table;
|
|
AK::Array<AK::Array<AK::Array<float, 18>, 32>, 2> m_last_values {};
|
|
AK::Array<AK::Array<float, 1024>, 2> m_synthesis_buffer {};
|
|
static DSP::MDCT<36> s_mdct_36;
|
|
static DSP::MDCT<12> s_mdct_12;
|
|
|
|
u32 m_sample_rate { 0 };
|
|
u8 m_num_channels { 0 };
|
|
PcmSampleFormat m_sample_format { PcmSampleFormat::Int16 };
|
|
int m_total_samples { 0 };
|
|
size_t m_loaded_samples { 0 };
|
|
|
|
AllocatingMemoryStream m_bit_reservoir;
|
|
};
|
|
|
|
}
|