M4AAudioFileDemuxer.h

Go to the documentation of this file.

#pragma once
 
#include <Arduino.h>
#include <SD.h>
 
#include "M4AAudioDemuxer.h"
 
namespace audio_tools {
 
class M4AAudioFileDemuxer : public M4ACommonDemuxer {
 public:
  using M4ACommonDemuxer::Frame;
  using M4ACommonDemuxer::FrameCallback;
 
  M4AAudioFileDemuxer() { setupParser(); };
 
  M4AAudioFileDemuxer(MultiDecoder& decoder) : M4AAudioFileDemuxer() {
    setDecoder(decoder);
  }
 
  bool setDecoder(MultiDecoder& decoder) {
    this->p_decoder = &decoder;
    if (decoder.getOutput() == nullptr) {
      LOGE("No output defined for MultiDecoder");
      return false;
    }
    setCallback([&decoder](const Frame& frame, void* /*ref*/) {
      LOGI("Decoding frame: %s with %d bytes", frame.mime, (int)frame.size);
      if (!decoder.selectDecoder(frame.mime)) {
        LOGE("Failed to select decoder for %s", frame.mime);
        return;
      }
      decoder.write(frame.data, frame.size);
    });
    return true;
  }
 
  void setCallback(FrameCallback cb) override { frame_callback = cb; }
 
  void setSamplesBufferSize(int size) {
    stsz_bufsize = size / 4;
    stsz_buf.resize(stsz_bufsize);
  }
 
  bool begin(File& file) {
    M4ACommonDemuxer::begin();
    this->p_file = &file;
    if (!file) return false;
    parser.begin();
    end();
    if (p_decoder) p_decoder->begin();
    if (!parseFile()) return false;
    if (!readStszHeader()) return false;
    if (!checkMdat()) return false;
    mdat_sample_pos = mdat_offset + mdat_pos;
 
    return true;
  }
 
  void end() {
    audio_config.codec = M4ACommonDemuxer::Codec::Unknown;
    audio_config.alacMagicCookie.clear();
    // resize(default_size);
    sample_index = 0;
    sample_count = 0;
    mdat_pos = 0;
    stsd_processed = false;
    mdat_offset = 0;
    mdat_size = 0;
    stsz_offset = 0;
    stsz_size = 0;
    mdat_pos = 0;
    fixed_sample_size = 0;
  }
 
  bool copy() {
    if (!p_file || sample_index >= sample_count) return false;
    size_t currentSize = getNextSampleSize();
    if (currentSize == 0) return false;
    if (!p_file->seek(mdat_sample_pos)) return false;
    if (buffer.size() < currentSize) buffer.resize(currentSize);
    size_t bytesRead = p_file->read(buffer.data(), currentSize);
    if (bytesRead != currentSize) return false;
    buffer.setWritePos(bytesRead);
    executeCallback(currentSize, buffer);
    mdat_sample_pos += currentSize;
    return true;
  }
 
  operator bool() { return sample_count > 0 && sample_index < sample_count; }
 
  uint32_t sampleIndex() const { return sample_index; }
 
  uint32_t size() const { return sample_count; }
 
  uint32_t getMdatOffset() const { return mdat_offset; }
 
  uint32_t getNextSampleSize() {
    assert(p_file != nullptr);
    if (sample_index >= sample_count) return 0;
    uint32_t currentSize = 0;
    if (fixed_sample_size) {
      currentSize = fixed_sample_size;
    } else {
      // if buffer is empty, fill it again
      if (stsz_buf.isEmpty()) {
        uint64_t pos = stsz_offset + 20 + sample_index * 4;
        if (!p_file->seek(pos)) return false;
        stsz_buf.clear();
        size_t read_bytes = p_file->read(
            reinterpret_cast<uint8_t*>(stsz_buf.data()), stsz_bufsize * 4);
        stsz_buf.setWritePos(read_bytes / 4);
        if (stsz_buf.isEmpty()) return 0;
      }
      // provide next size
      uint32_t val = 0;
      if (!stsz_buf.read(val)) return 0;
      currentSize = readU32(val);
    }
    sample_index++;
    return currentSize;
  }
 
  void beginSampleSizeAccess(File* filePtr, uint32_t sampleCount,
                             uint32_t stszOffset) {
    p_file = filePtr;
    sample_index = 0;
    sample_count = sampleCount;
    stsz_offset = stszOffset;
  }
 
  bool parseFile() {
    uint8_t buffer[1024];
    p_file->seek(0);
    while (p_file->available()) {
      int to_read = min(sizeof(buffer), parser.availableForWrite());
      size_t len = p_file->read(buffer, to_read);
      parser.write(buffer, len);
      // stop if we have all the data
      if (stsd_processed && mdat_offset && stsz_offset) return true;
    }
    return false;
  }
 
 protected:
  File* p_file = nullptr;        
  uint64_t mdat_offset = 0;      
  uint64_t mdat_size = 0;        
  uint64_t stsz_offset = 0;      
  uint64_t stsz_size = 0;        
  uint32_t sample_index = 0;     
  uint64_t mdat_pos = 0;         
  SingleBuffer<uint8_t> buffer;  
  int stsz_bufsize = 256;        
  SingleBuffer<uint32_t> stsz_buf{
      stsz_bufsize};                  
  uint32_t fixed_sample_size = 0;     
  MultiDecoder* p_decoder = nullptr;  
  uint64_t mdat_sample_pos = 0;
  void setupParser() override {
    parser.setReference(this);
 
    // Callback for ESDS box (AAC config)
    parser.setCallback(
        "esds",
        [](MP4Parser::Box& box, void* ref) {
          static_cast<M4AAudioFileDemuxer*>(ref)->onEsds(box);
        },
        false);
 
    // Callback for MP4A box (AAC sample entry)
    parser.setCallback(
        "mp4a",
        [](MP4Parser::Box& box, void* ref) {
          static_cast<M4AAudioFileDemuxer*>(ref)->onMp4a(box);
        },
        false);
 
    // Callback for ALAC box (ALAC sample entry)
    parser.setCallback(
        "alac",
        [](MP4Parser::Box& box, void* ref) {
          static_cast<M4AAudioFileDemuxer*>(ref)->onAlac(box);
        },
        false);
 
    // Callback for STSZ box (sample sizes)
    parser.setCallback(
        "stsz",
        [](MP4Parser::Box& box, void* ref) {
          auto* self = static_cast<M4AAudioFileDemuxer*>(ref);
          if (box.seq == 0) {
            self->stsz_offset = box.file_offset;
            self->stsz_size = box.size;
          }
        },
        false);
 
    // Callback for MDAT box (media data)
    parser.setCallback(
        "mdat",
        [](MP4Parser::Box& box, void* ref) {
          auto* self = static_cast<M4AAudioFileDemuxer*>(ref);
          if (box.seq == 0) {
            self->mdat_offset = box.file_offset + 8;  // skip box header
            self->mdat_size = box.size;
          }
        },
        false);
 
    // Callback for STSD box (sample description)
    parser.setCallback(
        "stsd",
        [](MP4Parser::Box& box, void* ref) {
          auto* self = static_cast<M4AAudioFileDemuxer*>(ref);
          self->onStsd(box);  // for aac and alac
          self->stsd_processed = true;
        },
        false);
  }
 
  void executeCallback(size_t size, SingleBuffer<uint8_t>& buffer) {
    Frame frame = sampleExtractor.getFrame(size, buffer);
    if (frame_callback)
      frame_callback(frame, nullptr);
    else
      LOGW("No frame callback defined");
  }
 
  bool readStszHeader() {
    if (!p_file || stsz_offset == 0) return false;
    uint8_t buffer[20];
    if (!p_file->seek(stsz_offset)) return false;
    if (p_file->read(buffer, 20) != 20) return false;
    if (!checkType(buffer, "stsz", 4)) return false;
    uint8_t* cont = buffer + 8;
    fixed_sample_size = readU32(cont + 4);
    sample_count = readU32(cont + 8);
    stsz_processed = true;
    return true;
  }
 
  bool checkMdat() {
    p_file->seek(mdat_offset - 8);
    uint8_t buffer[8];
    if (p_file->read(buffer, 8) != 8) return false;
    return checkType(buffer, "mdat", 4);
  }
};
 
}  // namespace audio_tools