libs/qglengine/renderer.cpp

/*
    QGLView
    Copyright (C) 2019  Ivan Pelipenko peri4ko@yandex.ru

    This program is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program.  If not, see <http://www.gnu.org/licenses/>.
*/

#define GL_GLEXT_PROTOTYPES
#include <QOpenGLExtraFunctions>
#include "renderer.h"
#include "qglview.h"
#include "glmesh.h"
#include "gltexture_manager.h"
#include <qad_types.h>

using namespace QGLEngineShaders;


Renderer::Renderer(QGLView * view_): RendererBase(view_),
		fbo_ds       (view_, QVector<GLenum>() << GL_RGBA16F << GL_RGBA32F << GL_RGBA16F << GL_RGBA16F << GL_RGBA16F),
		fbo_out      (view_, 3, false, GL_RGBA16F),
		fbo_hsmall   (view_, 1, false, GL_RGB16F ),
		rend_mat(this), rend_service(this), rend_selection(this) {
	quad = Primitive::plane(2., 2.);
	cam_light = new Light();
	cam_light->intensity = 0.75;
	cam_light->setName("Camera_Light");

	shader_files[srSelectionFill ] = "selection.glsl";
	shader_files[srSelectionHalo ] = "selection_halo.glsl";
	shader_files[srSelectionApply] = "selection_apply.glsl";
	shader_files[srSelectionFrame] = "selection_frame.glsl";

	shader_files[srServiceFill ] = "service_fill.glsl";
	shader_files[srServiceFrame] = "service_frame.glsl";

	shader_files[srGeometryPass ] = "ds_geom.glsl";
	shader_files[srLightOmniPass] = "ds_light.glsl"; shader_defines[srLightOmniPass] << "FIRST_PASS";
	shader_files[srLightSpotPass] = "ds_light.glsl"; shader_defines[srLightSpotPass] << "SPOT";
	shader_files[srFinalPass    ] = "ds_final.glsl";

	/*shaders << ShaderPair("FXAA", &shader_fxaa)
			<< ShaderPair("dsl_pass_0", &shader_ds_0)
			<< ShaderPair("dsl_pass_1", &shader_ds_1)
			<< ShaderPair("hdr", &shader_hdr)
			<< ShaderPair("downscale", &shader_small)
			<< ShaderPair("bloom_pass_0", &shader_bloom_0)
			<< ShaderPair("bloom_pass_1", &shader_bloom_1)
			<< ShaderPair("fbo_add", &shader_fbo_add)
			<< ShaderPair("motion_blur", &shader_motion_blur)
			<< ShaderPair("shadow", &shader_shadow)
			<< ShaderPair("ssr", &shader_ssr)
			<< ShaderPair("ssr_blur", &shader_ssr_blur)
			<< ShaderPair("ssr_merge", &shader_ssr_merge)
			<< ShaderPair("ssao_blur", &shader_ssao_blur)
			<< ShaderPair("ssao_merge", &shader_ssao_merge)
			<< ShaderPair("dof", &shader_dof);*/
	exposure_ = 1.;
	edit_mode = need_init_shaders = is_camera_light = true;
}


Renderer::~Renderer() {
	delete quad;
	delete cam_light;
	qDeleteAll(shaders.values());
}


void Renderer::init(int width, int height) {
	resize(width, height);
	rend_mat.init(width, height);
	rend_service.init(width, height);
	rend_selection.init(width, height);
	initQuad(quad);
	initTextureArrays();
	need_init_shaders = true;
}


void Renderer::resize(int width, int height) {
	rend_mat.resize(width, height);
	rend_service.resize(width, height);
	rend_selection.resize(width, height);
	fbo_ds       .resize(width, height);
	fbo_out      .resize(width, height);
	fbo_hsmall   .resize(width / 16, height / 16);
}


void Renderer::reloadShaders() {
	QMapIterator<ShaderRole, QString> it(shader_files);
	while (it.hasNext()) {
		it.next();
		loadShadersMulti(shaders[it.key()], "shaders/" + it.value(), shader_defines.value(it.key()));
	}
	need_init_shaders = true;
	view->scene()->setLightsChanged();
	view->scene()->setTreeStructChanged();
	view->scene()->setMaterialsChanged();
}


bool Renderer::bindShader(Renderer::ShaderRole role, QOpenGLShaderProgram ** ret) {
	QOpenGLShaderProgram * prog = shaders.value(role);
	if (ret) *ret = prog;
	if (!prog) return false;
	if (!prog->isLinked()) return false;
	prog->bind();
	return true;
}


void Renderer::initShaders() {
	if (!need_init_shaders) return;
	need_init_shaders = false;
	initUniformBuffer(shaders.value(srGeometryPass ), &buffer_materials , bpMaterials      , "QGLMaterialData"      );
	initUniformBuffer(shaders.value(srLightOmniPass), &buffer_materials , bpMaterials      , "QGLMaterialData"      );
	initUniformBuffer(shaders.value(srLightOmniPass), &buffer_lights    , bpLightParameters, "QGLLightParameterData");
	initUniformBuffer(shaders.value(srLightOmniPass), &buffer_lights_pos, bpLightPositions , "QGLLightPositionData" );
	initUniformBuffer(shaders.value(srLightSpotPass), &buffer_materials , bpMaterials      , "QGLMaterialData"      );
	initUniformBuffer(shaders.value(srLightSpotPass), &buffer_lights    , bpLightParameters, "QGLLightParameterData");
	initUniformBuffer(shaders.value(srLightSpotPass), &buffer_lights_pos, bpLightPositions , "QGLLightPositionData" );
	ShaderRole roles[] = {srLightOmniPass, srLightSpotPass};
	for (int p = 0; p < 2; ++p) {
		QOpenGLShaderProgram * prog = 0;
		if (!bindShader(roles[p], &prog)) continue;
		for (int i = 0; i < 5; ++i)
			prog->setUniformValue(QString("tex_%1").arg(i).toLatin1().constData(), i);
		prog->setUniformValue("tex_d", 5);
		prog->setUniformValue("tex_sum", 7);
	}
}


void Renderer::releaseShader() {
	view->glUseProgram(0);
}


void Renderer::fillObjectsBuffer(const QList<ObjectBase *> & ol, RenderPass pass) {
	cur_objects_.resize(ol.size());
	for (int i = 0; i < ol.size(); ++i) {
		Object & so(cur_objects_[i]);
		ObjectBase * o = ol[i];
		if (o->material()) {
			so.material = o->material()->_index;
			so.color = QVector4D(1,1,1,1);
		} else {
			so.material = 0;
			so.color = QColor2QVector(o->color_);
		}
		so.object_id = o->id_;
		o->worldTransform().transposed().copyDataTo(so.modelmatrix);
		//qDebug() << "load obj" << o->name() << o->worldTransform();
	}
	//qDebug() << "fillObjectsBuffer" << ol.size();

}


void Renderer::renderObjects(Scene & scene, RenderPass pass) {
	QOpenGLExtraFunctions * f = view;
	QMapIterator<Mesh*, QList<ObjectBase*> > it(scene.geometries_used);
	while (it.hasNext()) {
		it.next();
		Mesh * mesh = it.key();
		if (mesh->objects_changed) {
			mesh->objects_changed = false;
			fillObjectsBuffer(it.value(), pass);
			mesh->loadObjects(f, cur_objects_);
		}
		if (mesh->selected_changed && edit_mode) {
			mesh->selected_changed = false;
			fillSelectionsBuffer(rend_selection.cur_selections_, it.value());
			mesh->loadSelections(f, rend_selection.cur_selections_);
		}
		mesh->draw(f, it.value().size());
	}
}


void Renderer::renderScene() {
	initShaders();
	QOpenGLExtraFunctions * f = view;
	Scene & scene(*(view->scene_));
	Camera * cam = view->camera();
	QOpenGLShaderProgram * prog = 0;
	bool scene_changed = scene.prepare();

	/// reload materials on change
	if (scene_changed || scene.need_reload_materials) {
		//if (scene.need_reload_materials)
		//	maps_hash = 0;
		rend_selection.generateObjectsID(scene);
		reloadMaterials(scene);
		if (edit_mode)
			recreateMaterialThumbnails();
		emit view->materialsChanged();
	}

	/// material thumbnails
	if (edit_mode && !scene_changed) {
		rend_mat.procQueue();
	}

	/// lights
	QMap<int, QList<Light*>> ll = scene.lights_used;
	if (is_camera_light) {
		ll[Light::Omni] << cam_light;
		cam_light->setPos(cam->pos());
	}
	if (scene.lights_changed) {
		scene.lights_changed = false;
		reloadLightsParameters(ll);
	}
	reloadLightsPositions(cam);

	/// selection
	if (edit_mode) {
		rend_selection.renderSelection(scene);
	}

	/// geometry pass
	fbo_ds.bind();
	glEnableDepth();
	glClearFramebuffer();
	if (bindShader(srGeometryPass, &prog)) {
		setUniformMaps(prog);
		setUniformCamera(prog, cam);
		textures_empty.bind(f, tarEmpty);
		textures_maps .bind(f, tarMaps );
		renderObjects(scene, rpSolid);
	}
	fbo_ds.release();

	/// lighting passes
	fbo_ds.bindColorTextures();
	fbo_ds.bindDepthTexture(5);
	fbo_out.bind();
	int ri = 1, wi = 0;
	typedef QPair<Renderer::ShaderRole, Light::Type> PassPair;
	QVector<PassPair> passes;
	passes << PassPair(srLightOmniPass, Light::Omni) << PassPair(srLightSpotPass, Light::Cone);
	foreach (PassPair pass, passes) {
		if (bindShader(pass.first, &prog)) {
			setUniformCamera(prog, cam);
			setUniformViewCorners(prog, cam);
			prog->setUniformValue("lights_start", lights_start[pass.second]);
			prog->setUniformValue("lights_count", ll[pass.second].size());
			fbo_out.setWriteBuffer(wi);
			fbo_out.bindColorTexture(ri, 7);
			glClearFramebuffer(Qt::black, false);
			renderQuad(prog, quad, cam);
			piSwap(ri, wi);
		}
	}
	fbo_out.release();

	/// apply hovers and selection frame
	if (edit_mode) {
		rend_selection.drawSelection(fbo_out, ri);
		rend_service.renderService();
	} else {
		fbo_out.blit(ri, 0, 0, fbo_out.rect(), QRect(QPoint(), view->size()));
	}

}


void Renderer::setCameraLightOn(bool on) {
	is_camera_light = on;
	view->scene()->setLightsChanged();
}