src/third_party/libjingle/source/talk/media/base/yuvframegenerator.h

   1 // Generates YUV420 frames with a "landscape with striped crosshair" in the
   2 // Y-plane, plus a horizontal gradient in the U-plane and a vertical one in the
   3 // V-plane. This makes for a nice mix of colours that is suited for both
   4 // catching visual errors and making sure e.g. YUV->RGB/BGR conversion looks
   5 // the same on different platforms.
   6 // There is also a solid box bouncing around in the Y-plane, and two differently
   7 // coloured lines bouncing horizontally and vertically in the U and V plane.
   8 // This helps illustrating how the frame boundary goes, and can aid as a quite
   9 // handy visual help for noticing e.g. packet loss if the frames are encoded
  10 // and sent over the network.
  11
  12 #ifndef TALK_MEDIA_BASE_YUVFRAMEGENERATOR_H_
  13 #define TALK_MEDIA_BASE_YUVFRAMEGENERATOR_H_
  14
  15 #include "webrtc/base/basictypes.h"
  16
  17 namespace cricket {
  18
  19 class YuvFrameGenerator {
  20  public:
  21   // Constructs a frame-generator that produces frames of size |width|x|height|.
  22   // If |enable_barcode| is specified, barcodes can be included in the frames
  23   // when calling |GenerateNextFrame(uint8*, uint32)|. If |enable_barcode| is
  24   // |true| then |width|x|height| should be at least 160x100; otherwise this
  25   // constructor will abort.
  26   YuvFrameGenerator(int width, int height, bool enable_barcode);
  27   ~YuvFrameGenerator();
  28
  29   int GetFrameSize() { return frame_data_size_; }
  30
  31   // Generate the next frame and return it in the provided |frame_buffer|. If
  32   // barcode_value is not |nullptr| the value referred by it will be encoded
  33   // into a barcode in the frame.  The value should in the range:
  34   // [0..9,999,999]. If the value exceeds this range or barcodes were not
  35   // requested in the constructor, this function will abort.
  36   void GenerateNextFrame(uint8* frame_buffer, int32 barcode_value);
  37
  38   int GetHeight() { return height_; }
  39   int GetWidth() { return width_; }
  40
  41   // Fetch the bounds of the barcode from the generator. The barcode will
  42   // always be at this location. This function will abort if barcodes were not
  43   // requested in the constructor.
  44   void GetBarcodeBounds(int* top, int* left, int* width, int* height);
  45
  46  private:
  47   void DrawLandscape(uint8 *p, int w, int h);
  48   void DrawGradientX(uint8 *p, int w, int h);
  49   void DrawGradientY(uint8 *p, int w, int h);
  50   void DrawMovingLineX(uint8 *p, int w, int h, int n);
  51   void DrawMovingLineY(uint8 *p, int w, int h, int n);
  52   void DrawBouncingCube(uint8 *p, int w, int h, int n);
  53
  54   void DrawBarcode(uint32 value);
  55   int DrawSideGuardBars(int x, int y, int height);
  56   int DrawMiddleGuardBars(int x, int y, int height);
  57   int DrawEanEncodedDigit(int digit, int x, int y, int height, bool r_code);
  58   void DrawBlockRectangle(uint8* p, int x_start, int y_start,
  59                           int width, int height, int pitch, uint8 value);
  60
  61  private:
  62   int width_;
  63   int height_;
  64   int frame_index_;
  65   int frame_data_size_;
  66   uint8* y_data_;
  67   uint8* u_data_;
  68   uint8* v_data_;
  69
  70   int barcode_start_x_;
  71   int barcode_start_y_;
  72
  73   DISALLOW_COPY_AND_ASSIGN(YuvFrameGenerator);
  74 };
  75
  76 }  // namespace cricket
  77
  78 #endif  // TALK_MEDIA_BASE_YUVFRAMEGENERATOR_H_