MultiVideoDemuxer.h

Go to the documentation of this file.

#pragma once
 
#include "AudioTools/AudioCodecs/ContainerCommon.h"
#include "AudioTools/CoreAudio/AudioBasic/Collections/Vector.h"
#include "AudioTools/CoreAudio/AudioBasic/StrView.h"
 
namespace audio_tools {
 
class MultiVideoDemuxer : public Demuxer {
 public:
  void addDemuxer(Demuxer &demuxer) {
    const char *mime_video = demuxer.mimeVideo();
    for (int i = 0; i < demuxers.size(); i++) {
      if (StrView(demuxers[i]->mimeVideo()).equals(mime_video)) {
        demuxers.erase(i);
        break;
      }
    }
    demuxer.addNotifyAudioChange(*this);
    demuxers.push_back(&demuxer);
  }
 
  void setOutputAudio(Print &out) override {
    for (int i = 0; i < demuxers.size(); i++) demuxers[i]->setOutputAudio(out);
  }
 
  void setOutput(Print &out) override { setOutputAudio(out); }
 
  void setOutputVideo(Print &out) override {
    for (int i = 0; i < demuxers.size(); i++) demuxers[i]->setOutputVideo(out);
  }
  void setOutputVideo(VideoOutput &out) override {
    for (int i = 0; i < demuxers.size(); i++) demuxers[i]->setOutputVideo(out);
  }
 
  void setSendWavHeader(bool flag) override {
    for (int i = 0; i < demuxers.size(); i++)
      demuxers[i]->setSendWavHeader(flag);
  }
 
  const char *mimeVideo() override {
    return p_selected != nullptr ? p_selected->mimeVideo() : nullptr;
  }
 
  const char *mime() override {
    return p_selected != nullptr ? p_selected->mime() : nullptr;
  }
 
  VideoInfo getVideoInfo() override {
    return p_selected != nullptr ? p_selected->getVideoInfo() : VideoInfo{};
  }
 
  AudioInfoFormat getAudioInfo() override {
    return p_selected != nullptr ? p_selected->getAudioInfo()
                                  : AudioInfoFormat{};
  }
 
  bool begin() override {
    p_selected = nullptr;
    is_first = true;
    return true;
  }
 
  void end() override {
    if (p_selected != nullptr) p_selected->end();
    p_selected = nullptr;
    is_first = true;
  }
 
  size_t write(const uint8_t *data, size_t len) override {
    if (len == 0) return 0;
    if (is_first) {
      is_first = false;
      for (int i = 0; i < demuxers.size(); i++) {
        if (demuxers[i]->isValid(data, len)) {
          select(demuxers[i]);
          break;
        }
      }
      if (p_selected == nullptr) {
        LOGE(
            "MultiVideoDemuxer: could not determine container format from "
            "first %u bytes",
            (unsigned)len);
      }
    }
    return p_selected != nullptr ? p_selected->write(data, len) : len;
  }
 
  operator bool() override {
    return p_selected != nullptr ? (bool)(*p_selected) : is_first;
  }
 
  Demuxer *selectedDemuxer() const { return p_selected; }
 
 protected:
  Vector<Demuxer *> demuxers{0};
  Demuxer *p_selected = nullptr;
  bool is_first = true;
 
  void select(Demuxer *demuxer) {
    p_selected = demuxer;
    p_selected->begin();
    LOGI("MultiVideoDemuxer: selected demuxer for %s", demuxer->mimeVideo());
  }
};
 
}  // namespace audio_tools