dom/media/gtest/YUVBufferGenerator.h

   1 /* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
   2 /* vim: set ts=8 sts=2 et sw=2 tw=80: */
   3 /* This Source Code Form is subject to the terms of the Mozilla Public
   4  * License, v. 2.0. If a copy of the MPL was not distributed with this
   5  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
   6
   7 #ifndef YUVBufferGenerator_h
   8 #define YUVBufferGenerator_h
   9
  10 #include "ImageContainer.h"
  11 #include "mozilla/AlreadyAddRefed.h"
  12 #include "nsTArray.h"
  13 #include "Point.h"  // mozilla::gfx::IntSize
  14
  15 // A helper object to generate of different YUV planes.
  16 class YUVBufferGenerator {
  17  public:
  18   void Init(const mozilla::gfx::IntSize& aSize);
  19   mozilla::gfx::IntSize GetSize() const;
  20   already_AddRefed<mozilla::layers::Image> GenerateI420Image();
  21   already_AddRefed<mozilla::layers::Image> GenerateNV12Image();
  22   already_AddRefed<mozilla::layers::Image> GenerateNV21Image();
  23
  24  private:
  25   mozilla::layers::Image* CreateI420Image();
  26   mozilla::layers::Image* CreateNV12Image();
  27   mozilla::layers::Image* CreateNV21Image();
  28   mozilla::gfx::IntSize mImageSize;
  29   nsTArray<uint8_t> mSourceBuffer;
  30 };
  31
  32 #endif  // YUVBufferGenerator_h