git.sesse.net Git - nageru/blob - jpeg_frame_view.h

   1 #ifndef _JPEG_FRAME_VIEW_H
   2 #define _JPEG_FRAME_VIEW_H 1
   3
   4 #include <epoxy/gl.h>
   5 #include <QGLWidget>
   6
   7 #include <stdint.h>
   8
   9 #include <movit/effect_chain.h>
  10 #include <movit/ycbcr_input.h>
  11
  12 #include <memory>
  13
  14 struct Frame {
  15         std::unique_ptr<uint8_t[]> y, cb, cr;
  16         unsigned width, height;
  17         unsigned chroma_subsampling_x, chroma_subsampling_y;
  18         unsigned pitch_y, pitch_chroma;
  19 };
  20
  21 class JPEGFrameView : public QGLWidget {
  22         Q_OBJECT
  23
  24 public:
  25         JPEGFrameView(QWidget *parent);
  26
  27         void setFrame(unsigned stream_idx, int64_t pts)
  28         {
  29                 this->stream_idx = stream_idx;
  30                 this->pts = pts;
  31                 update_frame();
  32         }
  33
  34         void setDecodedFrame(std::shared_ptr<Frame> frame);
  35
  36 protected:
  37         void initializeGL() override;
  38         void resizeGL(int width, int height) override;
  39         void paintGL() override;
  40
  41 private:
  42         void update_frame();
  43
  44         unsigned stream_idx;
  45         int64_t pts;
  46
  47         std::unique_ptr<movit::EffectChain> chain;
  48         std::shared_ptr<Frame> current_frame;  // So that we hold on to the pixels.
  49         movit::YCbCrInput *ycbcr_input;
  50         movit::YCbCrFormat ycbcr_format;
  51 };
  52
  53 #endif  // !defined(_JPEG_FRAME_VIEW_H)