format_registry.hpp

Go to the documentation of this file.

#pragma once
 
// Audio format codec registry — register readers/writers by extension/MIME type.
// Provides unified audio file I/O across all supported formats.
 
#include <pulp/audio/audio_file.hpp>
#include <cstdint>
#include <string>
#include <string_view>
#include <vector>
#include <functional>
#include <memory>
 
namespace pulp::audio {

struct DecodeLimits {
    std::uint64_t max_frames = 1ull << 30;
    std::uint32_t max_channels = 64u;
    std::uint64_t max_output_bytes = 4ull * 1024u * 1024u * 1024u;

    static DecodeLimits unbounded() {
        DecodeLimits limits;
        limits.max_frames = 0u;
        limits.max_channels = 0u;
        limits.max_output_bytes = 0u;
        return limits;
    }
};

class FormatReader {
public:
    virtual ~FormatReader() = default;

    virtual std::optional<AudioFileInfo> read_info(const std::string& path) = 0;

    virtual std::optional<AudioFileData> read(const std::string& path) = 0;

    virtual std::optional<AudioFileData> read(const std::string& path, const DecodeLimits& limits) {
        (void)limits;
        return read(path);
    }

    virtual bool supports_extension(std::string_view ext) const = 0;

    virtual std::string format_name() const = 0;
};

class FormatWriter {
public:
    virtual ~FormatWriter() = default;

    virtual bool write(const std::string& path, const AudioFileData& data) = 0;

    virtual bool supports_extension(std::string_view ext) const = 0;

    virtual std::string format_name() const = 0;
};

class FormatRegistry {
public:
    static FormatRegistry& instance();

    void register_reader(std::unique_ptr<FormatReader> reader);

    void register_writer(std::unique_ptr<FormatWriter> writer);

    FormatReader* find_reader(std::string_view extension) const;

    FormatWriter* find_writer(std::string_view extension) const;

    std::optional<AudioFileInfo> read_info(const std::string& path) const;
    std::optional<AudioFileData> read(const std::string& path) const;

    std::optional<AudioFileData> read(const std::string& path, const DecodeLimits& limits) const;

    bool write(const std::string& path, const AudioFileData& data) const;

    std::vector<std::string> supported_read_extensions() const;

    std::vector<std::string> supported_write_extensions() const;
 
private:
    FormatRegistry();
    std::vector<std::unique_ptr<FormatReader>> readers_;
    std::vector<std::unique_ptr<FormatWriter>> writers_;
};
 
}  // namespace pulp::audio