VideoOutput.h

Go to the documentation of this file.

#pragma once
#include "AudioTools/CoreAudio/AudioOutput.h"
#include "stdint.h"
 
namespace audio_tools {

enum class VideoFormat {
  H264,
  MJPEG,
  MPEG4,
  RAW,
  YUV422,
  RGB565,
  RGB666,
  RGB888,
  I420,
  MPEG1,
  UNKNOWN
};

struct VideoInfo {
  uint16_t width = 0;
  uint16_t height = 0;
  float fps = 0;
  VideoFormat format = VideoFormat::UNKNOWN;
  uint32_t frame_size = 0;
  uint32_t total_file_size = 0;
 
  bool operator==(const VideoInfo& alt) const {
    return width == alt.width && height == alt.height && fps == alt.fps &&
           format == alt.format && frame_size == alt.frame_size &&
           total_file_size == alt.total_file_size;
  }
  bool operator!=(const VideoInfo& alt) const { return !(*this == alt); }
  operator bool() const { return width > 0 && height > 0; }
  void logInfo(const char* source = "") {
    LOGI(
        "%s width: %d / height: %d / fps: %f / format: %d / frame_size: %d / "
        "total_file_size: %d",
        source, (int)width, (int)height, fps, (int)format, (int)frame_size,
        (int)total_file_size);
  }
};

class VideoInfoSource {
 public:
  virtual VideoInfo videoInfo() = 0;
};

class VideoOutput {
 public:
  virtual size_t write(const uint8_t* data, size_t len) = 0;
  virtual void flush() {}
  virtual void setSkipRender(bool skip) {}

  virtual bool isKeyFrame(const uint8_t* data, size_t len) { return false; }

  virtual uint32_t getWriteTimeMs() const { return 0; }

  virtual bool hadOutput() const { return true; }

  virtual void setVideoInfoSource(VideoInfoSource& source) {}

  virtual uint64_t totalDecodeMs() const { return 0; }
};
 
}  // namespace audio_tools