face_renderer.h

#ifndef FACE_SWAP_FACE_RENDERER_H
#define FACE_SWAP_FACE_RENDERER_H

#include "face_swap/basel_3dmm.h"
#include <opencv2/core.hpp>

namespace face_swap
{
    class FaceRenderer
    {
    public:

        FaceRenderer();

        ~FaceRenderer();

        void init(int width, int height, bool force = false);

        void clear();

        void clearFrameBuffer();

        void clearMesh();

        void setViewport(int width, int height);

        void setProjection(float f, float z_near = 10, float z_far= 10000);

        void setMesh(const Mesh& mesh);

        void render(const cv::Mat& vecR, const cv::Mat& vecT);

        void getFrameBuffer(cv::Mat& img);

        void getDepthBuffer(cv::Mat& depth);

        void setLight(const cv::Mat& pos_dir, const cv::Mat& ambient, const cv::Mat& diffuse);

        void clearLight();

    private:
        void drawMesh();

    private:
        int m_width = 0, m_height = 0;
        int m_tex_width = 0, m_tex_height = 0;
        unsigned int m_dynamic_tex_id = 0;
        unsigned int m_fbo = 0;
        unsigned int m_depth_rb = 0;

        // Mesh
        unsigned int m_mesh_vert_id = 0;
        unsigned int m_mesh_uv_id = 0;
        unsigned int m_mesh_faces_id = 0;
        unsigned int m_mesh_total_faces = 0;
        unsigned int m_mesh_total_vertices = 0;
        unsigned int m_mesh_tex_id = 0;
        unsigned int m_mesh_tex_width = 0;
        unsigned int m_mesh_tex_height = 0;
        unsigned int m_mesh_normals_id = 0;
    };

}   // namespace face_swap

#endif //FACE_SWAP_FACE_RENDERER_H