oggcontainer.h

Go to the documentation of this file.

#ifndef TAG_PARSER_OGGCONTAINER_H
#define TAG_PARSER_OGGCONTAINER_H
 
#include "./oggiterator.h"
#include "./oggpage.h"
#include "./oggstream.h"
 
#include "../vorbis/vorbiscomment.h"
 
#include "../genericcontainer.h"
 
#include <tuple>
#include <unordered_map>
 
namespace CppUtilities {
template <std::size_t bufferSize> class CopyHelper;
}
 
namespace TagParser {
 
class MediaFileInfo;
class OggContainer;
 
struct TAG_PARSER_EXPORT OggParameter {
    constexpr OggParameter();
    void set(std::size_t pageIndex, std::size_t segmentIndex, bool lastMetaDataBlock, GeneralMediaFormat streamFormat = GeneralMediaFormat::Vorbis);
 
    std::size_t firstPageIndex;
    std::size_t firstSegmentIndex;
    std::size_t lastPageIndex;
    std::size_t lastSegmentIndex;
    GeneralMediaFormat streamFormat;
    bool lastMetaDataBlock;
    bool removed;
};
 
constexpr OggParameter::OggParameter()
    : firstPageIndex(0)
    , firstSegmentIndex(0)
    , lastPageIndex(0)
    , lastSegmentIndex(0)
    , streamFormat(GeneralMediaFormat::Vorbis)
    , lastMetaDataBlock(false)
    , removed(false)
{
}
 
inline void OggParameter::set(std::size_t pageIndex, std::size_t segmentIndex, bool lastMetaDataBlock, GeneralMediaFormat streamFormat)
{
    firstPageIndex = lastPageIndex = pageIndex;
    firstSegmentIndex = lastSegmentIndex = segmentIndex;
    this->lastMetaDataBlock = lastMetaDataBlock;
    this->streamFormat = streamFormat;
}
 
class TAG_PARSER_EXPORT OggVorbisComment final : public VorbisComment {
    friend class OggContainer;
 
public:
    OggVorbisComment();
 
    static constexpr TagType tagType = TagType::OggVorbisComment;
    static constexpr std::string_view tagName = "OGG Vorbis comment";
    TagType type() const override;
    std::string_view typeName() const override;
    bool supportsTarget() const override;
 
    OggParameter &oggParams();
    const OggParameter &oggParams() const;
 
private:
    OggParameter m_oggParams;
};
 
inline OggVorbisComment::OggVorbisComment()
{
}
 
inline TagType OggVorbisComment::type() const
{
    return TagType::OggVorbisComment;
}
 
inline bool OggVorbisComment::supportsTarget() const
{
    return true;
}
 
inline OggParameter &OggVorbisComment::oggParams()
{
    return m_oggParams;
}
 
inline const OggParameter &OggVorbisComment::oggParams() const
{
    return m_oggParams;
}
 
class TAG_PARSER_EXPORT OggContainer final : public GenericContainer<MediaFileInfo, OggVorbisComment, OggStream, OggPage> {
    friend class OggStream;
 
public:
    OggContainer(MediaFileInfo &fileInfo, std::uint64_t startOffset);
    ~OggContainer() override;
 
    bool isChecksumValidationEnabled() const;
    void setChecksumValidationEnabled(bool enabled);
    void reset() override;
 
    OggVorbisComment *createTag(const TagTarget &target) override;
    OggVorbisComment *tag(std::size_t index) override;
    std::size_t tagCount() const override;
    bool removeTag(Tag *tag) override;
    void removeAllTags() override;
 
protected:
    void internalParseHeader(Diagnostics &diag, AbortableProgressFeedback &progress) override;
    void internalParseTags(Diagnostics &diag, AbortableProgressFeedback &progress) override;
    void internalParseTracks(Diagnostics &diag, AbortableProgressFeedback &progress) override;
    void internalMakeFile(Diagnostics &diag, AbortableProgressFeedback &progress) override;
 
private:
    void announceComment(
        std::size_t pageIndex, std::size_t segmentIndex, bool lastMetaDataBlock, GeneralMediaFormat mediaFormat = GeneralMediaFormat::Vorbis);
    void makeVorbisCommentSegment(std::stringstream &buffer, CppUtilities::CopyHelper<65307> &copyHelper, std::vector<std::uint32_t> &newSegmentSizes,
        VorbisComment *comment, OggParameter *params, Diagnostics &diag);
 
    std::unordered_map<std::uint32_t, std::vector<std::unique_ptr<OggStream>>::size_type> m_streamsBySerialNo;
 
    OggIterator m_iterator;
    bool m_validateChecksums;
};
 
inline bool OggContainer::isChecksumValidationEnabled() const
{
    return m_validateChecksums;
}
 
inline void OggContainer::setChecksumValidationEnabled(bool enabled)
{
    m_validateChecksums = enabled;
}
 
} // namespace TagParser
 
#endif // TAG_PARSER_OGGCONTAINER_H