skia2/tools/gpu/YUVUtils.h
Brian Salomon 0c0b5a6bb2 Remove SkYUVASizeInfo, make SkYUVAIndex a private part of SkYUVAInfo
Bug: skia:10632
Change-Id: If4dd7779b0856f6d0b441381bf7f2f51527cdb9d
Reviewed-on: https://skia-review.googlesource.com/c/skia/+/352497
Reviewed-by: Jim Van Verth <jvanverth@google.com>
Commit-Queue: Brian Salomon <bsalomon@google.com>
2021-01-12 15:06:26 +00:00

62 lines
2.0 KiB
C++

/*
* Copyright 2019 Google Inc.
*
* Use of this source code is governed by a BSD-style license that can be
* found in the LICENSE file.
*/
#ifndef YUVUtils_DEFINED
#define YUVUtils_DEFINED
#include "include/core/SkImage.h"
#include "include/core/SkYUVAPixmaps.h"
#include "include/gpu/GrBackendSurface.h"
#include "src/core/SkAutoMalloc.h"
class SkData;
namespace sk_gpu_test {
// Utility that decodes a JPEG but preserves the YUVA8 planes in the image, and uses
// MakeFromYUVAPixmaps to create a GPU multiplane YUVA image for a context. It extracts the planar
// data once, and lazily creates the actual SkImage when the GrContext is provided (and refreshes
// the image if the context has changed, as in Viewer)
class LazyYUVImage {
public:
// Returns null if the data could not be extracted into YUVA planes
static std::unique_ptr<LazyYUVImage> Make(sk_sp<SkData> data,
GrMipmapped = GrMipmapped::kNo,
sk_sp<SkColorSpace> = nullptr);
static std::unique_ptr<LazyYUVImage> Make(SkYUVAPixmaps,
GrMipmapped = GrMipmapped::kNo,
sk_sp<SkColorSpace> = nullptr);
enum class Type { kFromPixmaps, kFromGenerator, kFromTextures };
SkISize dimensions() const { return fPixmaps.yuvaInfo().dimensions(); }
sk_sp<SkImage> refImage(GrRecordingContext* rContext, Type);
private:
// Decoded YUV data
SkYUVAPixmaps fPixmaps;
GrMipmapped fMipmapped;
sk_sp<SkColorSpace> fColorSpace;
// Memoized SkImages formed with planes, one for each Type.
sk_sp<SkImage> fYUVImage[4];
LazyYUVImage() = default;
bool reset(sk_sp<SkData> data, GrMipmapped, sk_sp<SkColorSpace>);
bool reset(SkYUVAPixmaps pixmaps, GrMipmapped, sk_sp<SkColorSpace>);
bool ensureYUVImage(GrRecordingContext* rContext, Type type);
};
} // namespace sk_gpu_test
#endif // YUVUtils_DEFINED