Add decoder class

torchaudio/csrc/ffmpeg/decoder.cpp

@@ -0,0 +1,20 @@
+#include <torchaudio/csrc/ffmpeg/decoder.h>
+
+namespace torchaudio {
+namespace ffmpeg {
+
+////////////////////////////////////////////////////////////////////////////////
+// Decoder
+////////////////////////////////////////////////////////////////////////////////
+Decoder::Decoder(AVCodecParameters* pParam) : pCodecContext(pParam) {}
+
+int Decoder::process_packet(AVPacket* pPacket) {
+  return avcodec_send_packet(pCodecContext, pPacket);
+}
+
+int Decoder::get_frame(AVFrame* pFrame) {
+  return avcodec_receive_frame(pCodecContext, pFrame);
+}
+
+} // namespace ffmpeg
+} // namespace torchaudio

torchaudio/csrc/ffmpeg/decoder.h

@@ -0,0 +1,30 @@
+#pragma once
+
+#include <torchaudio/csrc/ffmpeg/ffmpeg.h>
+
+namespace torchaudio {
+namespace ffmpeg {
+
+class Decoder {
+  AVCodecContextPtr pCodecContext;
+
+ public:
+  // Default constructable
+  Decoder(AVCodecParameters* pParam);
+  // Custom destructor to clean up the resources
+  ~Decoder() = default;
+  // Non-copyable
+  Decoder(const Decoder&) = delete;
+  Decoder& operator=(const Decoder&) = delete;
+  // Movable
+  Decoder(Decoder&&) = default;
+  Decoder& operator=(Decoder&&) = default;
+
+  // Process incoming packet
+  int process_packet(AVPacket* pPacket);
+  // Fetch a decoded frame
+  int get_frame(AVFrame* pFrame);
+};
+
+} // namespace ffmpeg
+} // namespace torchaudio