egl.h

/* SPDX-License-Identifier: LGPL-2.1-or-later */
/*
 * Copyright (C) 2024, Linaro Ltd.
 *
 * Authors:
 * Bryan O'Donoghue <bryan.odonoghue@linaro.org>
 *
 */
 
#pragma once
 
#include <sys/types.h>
#include <unistd.h>
 
#include <libcamera/base/log.h>
#include <libcamera/base/span.h>
#include <libcamera/base/utils.h>
 
/*
 * Workaround for build issues on Mesa <= 22.2, see
 * https://github.com/KhronosGroup/EGL-Registry/pull/130
 */
#define EGL_NO_X11
 
#define EGL_EGLEXT_PROTOTYPES
#include <EGL/egl.h>
#include <EGL/eglext.h>
#define GL_GLEXT_PROTOTYPES
#include <GLES2/gl2.h>
#include <GLES2/gl2ext.h>
 
namespace libcamera {
 
LOG_DECLARE_CATEGORY(eGL)
 

class eGLImage
{
public:
    eGLImage(GLint format, uint32_t width, uint32_t height, uint32_t stride, GLenum texture_unit, uint32_t texture_unit_uniform_id)
        : format_(format), width_(width), height_(height), stride_(stride),
          framesize_(stride * height),
          texture_unit_uniform_id_(texture_unit_uniform_id),
          texture_unit_(texture_unit)
    {
        glGenTextures(1, &texture_);
        glGenFramebuffers(1, &fbo_);
    }

    ~eGLImage()
    {
        glDeleteFramebuffers(1, &fbo_);
        glDeleteTextures(1, &texture_);
    }
 
    GLint format_; 
    uint32_t width_; 
    uint32_t height_; 
    uint32_t stride_; 
    uint32_t offset_; 
    uint32_t framesize_; 
    uint32_t texture_unit_uniform_id_; 
    GLenum texture_unit_; 
    GLuint texture_; 
    GLuint fbo_; 
    bool dmabuf_import_failed_ = false; 
 
private:
    LIBCAMERA_DISABLE_COPY_AND_MOVE(eGLImage)
};
 
class eGL
{
public:
    eGL();
    ~eGL();
 
    int initEGLContext();
    static bool isAvailable();
 
    int createInputDMABufTexture2D(eGLImage &eglImage, int fd);
    int createOutputDMABufTexture2D(eGLImage &eglImage, int fd);
    void createTexture2D(eGLImage &eglImage, void *data);
    void updateTexture2D(eGLImage &eglImage, void *data);
    void createOutputTexture2D(eGLImage &eglImage);
 
    int attachTextureToFBO(eGLImage &eglImage);
    void activateBindTexture(eGLImage &eglImage);
 
    void pushEnv(std::vector<std::string> &shaderEnv, const char *str);
    void makeCurrent();
 
    int compileVertexShader(GLuint &shaderId, Span<const unsigned char> shaderData,
                Span<const std::string> shaderEnv);
    int compileFragmentShader(GLuint &shaderId, Span<const unsigned char> shaderData,
                  Span<const std::string> shaderEnv);
    int linkProgram(GLuint &programId, GLuint fragmentshaderId, GLuint vertexshaderId);
    void dumpShaderSource(GLuint shaderId);
    void useProgram(GLuint programId);
    void deleteProgram(GLuint programId);
    void syncOutput();
    void flushOutput();
 
private:
    LIBCAMERA_DISABLE_COPY_AND_MOVE(eGL)
 
    pid_t tid_;
 
    EGLDisplay display_ = EGL_NO_DISPLAY;
    EGLContext context_ = EGL_NO_CONTEXT;
    EGLSurface surface_ = EGL_NO_SURFACE;
 
    static EGLDisplay probeDisplay();
    int compileShader(int shaderType, GLuint &shaderId, Span<const unsigned char> shaderData,
              Span<const std::string> shaderEnv);
 
    int createDMABufTexture2D(eGLImage &eglImage, int fd, bool output);
 
    PFNGLEGLIMAGETARGETTEXTURE2DOESPROC glEGLImageTargetTexture2DOES;
    PFNEGLCREATEIMAGEKHRPROC eglCreateImageKHR;
    PFNEGLDESTROYIMAGEKHRPROC eglDestroyImageKHR;
    PFNGLGETSTRINGPROC glGetString;
};
} //namespace libcamera