No public description
PiperOrigin-RevId: 578496866
This commit is contained in:
parent
c6aa9cbaef
commit
3a55f1156a
|
@ -14,6 +14,8 @@
|
||||||
|
|
||||||
#include "mediapipe/gpu/gl_texture_buffer.h"
|
#include "mediapipe/gpu/gl_texture_buffer.h"
|
||||||
|
|
||||||
|
#include <cstdint>
|
||||||
|
|
||||||
#include "absl/log/absl_check.h"
|
#include "absl/log/absl_check.h"
|
||||||
#include "absl/log/absl_log.h"
|
#include "absl/log/absl_log.h"
|
||||||
#include "mediapipe/framework/formats/image_frame.h"
|
#include "mediapipe/framework/formats/image_frame.h"
|
||||||
|
@ -131,6 +133,13 @@ bool GlTextureBuffer::CreateInternal(const void* data, int alignment) {
|
||||||
SymbolAvailable(&glTexStorage2D)) {
|
SymbolAvailable(&glTexStorage2D)) {
|
||||||
ABSL_CHECK(data == nullptr) << "unimplemented";
|
ABSL_CHECK(data == nullptr) << "unimplemented";
|
||||||
glTexStorage2D(target_, 1, info.gl_internal_format, width_, height_);
|
glTexStorage2D(target_, 1, info.gl_internal_format, width_, height_);
|
||||||
|
} else if (info.immutable) {
|
||||||
|
ABSL_CHECK(SymbolAvailable(&glTexStorage2D) &&
|
||||||
|
context->GetGlVersion() != GlVersion::kGLES2)
|
||||||
|
<< "Immutable GpuBuffer format requested is not supported in this "
|
||||||
|
<< "GlContext. Format was " << static_cast<uint32_t>(format_);
|
||||||
|
ABSL_CHECK(data == nullptr) << "unimplemented";
|
||||||
|
glTexStorage2D(target_, 1, info.gl_internal_format, width_, height_);
|
||||||
} else {
|
} else {
|
||||||
glTexImage2D(target_, 0 /* level */, info.gl_internal_format, width_,
|
glTexImage2D(target_, 0 /* level */, info.gl_internal_format, width_,
|
||||||
height_, 0 /* border */, info.gl_format, info.gl_type, data);
|
height_, 0 /* border */, info.gl_format, info.gl_type, data);
|
||||||
|
|
|
@ -35,6 +35,10 @@ namespace mediapipe {
|
||||||
#endif // GL_HALF_FLOAT_OES
|
#endif // GL_HALF_FLOAT_OES
|
||||||
#endif // __EMSCRIPTEN__
|
#endif // __EMSCRIPTEN__
|
||||||
|
|
||||||
|
#ifndef GL_RGBA8
|
||||||
|
#define GL_RGBA8 0x8058
|
||||||
|
#endif // GL_RGBA8
|
||||||
|
|
||||||
#if !MEDIAPIPE_DISABLE_GPU
|
#if !MEDIAPIPE_DISABLE_GPU
|
||||||
#ifdef GL_ES_VERSION_2_0
|
#ifdef GL_ES_VERSION_2_0
|
||||||
static void AdaptGlTextureInfoForGLES2(GlTextureInfo* info) {
|
static void AdaptGlTextureInfoForGLES2(GlTextureInfo* info) {
|
||||||
|
@ -163,6 +167,14 @@ const GlTextureInfo& GlTextureInfoForGpuBufferFormat(GpuBufferFormat format,
|
||||||
{
|
{
|
||||||
{GL_RGBA32F, GL_RGBA, GL_FLOAT, 1},
|
{GL_RGBA32F, GL_RGBA, GL_FLOAT, 1},
|
||||||
}},
|
}},
|
||||||
|
{GpuBufferFormat::kImmutableRGBAFloat128,
|
||||||
|
{
|
||||||
|
{GL_RGBA32F, GL_RGBA, GL_FLOAT, 1, true /* immutable */},
|
||||||
|
}},
|
||||||
|
{GpuBufferFormat::kImmutableRGBA32,
|
||||||
|
{
|
||||||
|
{GL_RGBA8, GL_RGBA, GL_UNSIGNED_BYTE, 1, true /* immutable */},
|
||||||
|
}},
|
||||||
}};
|
}};
|
||||||
|
|
||||||
static const auto* gles2_format_info = ([] {
|
static const auto* gles2_format_info = ([] {
|
||||||
|
@ -206,6 +218,7 @@ const GlTextureInfo& GlTextureInfoForGpuBufferFormat(GpuBufferFormat format,
|
||||||
|
|
||||||
ImageFormat::Format ImageFormatForGpuBufferFormat(GpuBufferFormat format) {
|
ImageFormat::Format ImageFormatForGpuBufferFormat(GpuBufferFormat format) {
|
||||||
switch (format) {
|
switch (format) {
|
||||||
|
case GpuBufferFormat::kImmutableRGBA32:
|
||||||
case GpuBufferFormat::kBGRA32:
|
case GpuBufferFormat::kBGRA32:
|
||||||
// TODO: verify we are handling order of channels correctly.
|
// TODO: verify we are handling order of channels correctly.
|
||||||
return ImageFormat::SRGBA;
|
return ImageFormat::SRGBA;
|
||||||
|
@ -221,6 +234,7 @@ ImageFormat::Format ImageFormatForGpuBufferFormat(GpuBufferFormat format) {
|
||||||
return ImageFormat::SRGB;
|
return ImageFormat::SRGB;
|
||||||
case GpuBufferFormat::kTwoComponentFloat32:
|
case GpuBufferFormat::kTwoComponentFloat32:
|
||||||
return ImageFormat::VEC32F2;
|
return ImageFormat::VEC32F2;
|
||||||
|
case GpuBufferFormat::kImmutableRGBAFloat128:
|
||||||
case GpuBufferFormat::kRGBAFloat128:
|
case GpuBufferFormat::kRGBAFloat128:
|
||||||
return ImageFormat::VEC32F4;
|
return ImageFormat::VEC32F4;
|
||||||
case GpuBufferFormat::kRGBA32:
|
case GpuBufferFormat::kRGBA32:
|
||||||
|
|
|
@ -53,6 +53,10 @@ enum class GpuBufferFormat : uint32_t {
|
||||||
kRGB24 = 0x00000018, // Note: prefer BGRA32 whenever possible.
|
kRGB24 = 0x00000018, // Note: prefer BGRA32 whenever possible.
|
||||||
kRGBAHalf64 = MEDIAPIPE_FOURCC('R', 'G', 'h', 'A'),
|
kRGBAHalf64 = MEDIAPIPE_FOURCC('R', 'G', 'h', 'A'),
|
||||||
kRGBAFloat128 = MEDIAPIPE_FOURCC('R', 'G', 'f', 'A'),
|
kRGBAFloat128 = MEDIAPIPE_FOURCC('R', 'G', 'f', 'A'),
|
||||||
|
// Immutable version of kRGBA32
|
||||||
|
kImmutableRGBA32 = MEDIAPIPE_FOURCC('4', 'C', 'I', '8'),
|
||||||
|
// Immutable version of kRGBAFloat128
|
||||||
|
kImmutableRGBAFloat128 = MEDIAPIPE_FOURCC('4', 'C', 'I', 'f'),
|
||||||
// 8-bit Y plane + interleaved 8-bit U/V plane with 2x2 subsampling.
|
// 8-bit Y plane + interleaved 8-bit U/V plane with 2x2 subsampling.
|
||||||
kNV12 = MEDIAPIPE_FOURCC('N', 'V', '1', '2'),
|
kNV12 = MEDIAPIPE_FOURCC('N', 'V', '1', '2'),
|
||||||
// 8-bit Y plane + interleaved 8-bit V/U plane with 2x2 subsampling.
|
// 8-bit Y plane + interleaved 8-bit V/U plane with 2x2 subsampling.
|
||||||
|
@ -78,6 +82,9 @@ struct GlTextureInfo {
|
||||||
// For multiplane buffers, this represents how many times smaller than
|
// For multiplane buffers, this represents how many times smaller than
|
||||||
// the nominal image size a plane is.
|
// the nominal image size a plane is.
|
||||||
int downscale;
|
int downscale;
|
||||||
|
// For GLES3.1+ compute shaders, users may explicitly request immutable
|
||||||
|
// textures.
|
||||||
|
bool immutable = false;
|
||||||
};
|
};
|
||||||
|
|
||||||
const GlTextureInfo& GlTextureInfoForGpuBufferFormat(GpuBufferFormat format,
|
const GlTextureInfo& GlTextureInfoForGpuBufferFormat(GpuBufferFormat format,
|
||||||
|
@ -121,6 +128,8 @@ inline OSType CVPixelFormatForGpuBufferFormat(GpuBufferFormat format) {
|
||||||
return kCVPixelFormatType_64RGBAHalf;
|
return kCVPixelFormatType_64RGBAHalf;
|
||||||
case GpuBufferFormat::kRGBAFloat128:
|
case GpuBufferFormat::kRGBAFloat128:
|
||||||
return kCVPixelFormatType_128RGBAFloat;
|
return kCVPixelFormatType_128RGBAFloat;
|
||||||
|
case GpuBufferFormat::kImmutableRGBA32:
|
||||||
|
case GpuBufferFormat::kImmutableRGBAFloat128:
|
||||||
case GpuBufferFormat::kNV12:
|
case GpuBufferFormat::kNV12:
|
||||||
case GpuBufferFormat::kNV21:
|
case GpuBufferFormat::kNV21:
|
||||||
case GpuBufferFormat::kI420:
|
case GpuBufferFormat::kI420:
|
||||||
|
|
Loading…
Reference in New Issue
Block a user