source/src/render/passes/material-pass.hpp

/*
 * Copyright (C) 2021  Christopher J. Howard
 *
 * This file is part of Antkeeper source code.
 *
 * Antkeeper source code 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.
 *
 * Antkeeper source code 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 Antkeeper source code.  If not, see <http://www.gnu.org/licenses/>.
 */

#ifndef ANTKEEPER_RENDER_MATERIAL_PASS_HPP
#define ANTKEEPER_RENDER_MATERIAL_PASS_HPP

#include "render/pass.hpp"
#include "render/material.hpp"
#include "utility/fundamental-types.hpp"
#include "event/event-handler.hpp"
#include "event/input-events.hpp"
#include "gl/shader-program.hpp"
#include "gl/shader-input.hpp"
#include "gl/texture-2d.hpp"
#include <unordered_map>

class resource_manager;

namespace render {

class shadow_map_pass;

/**
 * Renders scene objects using their material-specified shaders and properties.
 */
class material_pass: public pass,
	public event_handler<mouse_moved_event>
{
public:
	material_pass(gl::rasterizer* rasterizer, const gl::framebuffer* framebuffer, resource_manager* resource_manager);
	virtual ~material_pass();
	virtual void render(const render::context& ctx, render::queue& queue) const final;
	
	/// Sets the material to be used when a render operation is missing a material. If no fallback material is specified, render operations without materials will not be processed.
	void set_fallback_material(const material* fallback);
	
	const render::shadow_map_pass* shadow_map_pass;
	const gl::texture_2d* shadow_map;
	
private:
	virtual void handle_event(const mouse_moved_event& event);
	
	/**
	 * Sets of known shader input parameters. Each time a new shader is encountered, a parameter set will be created and its inputs connected to the shader program. A null input indiciates that the shader doesn't have that parameter.
	 */
	struct parameter_set
	{
		const gl::shader_input* time;
		const gl::shader_input* mouse;
		const gl::shader_input* resolution;
		const gl::shader_input* camera_position;
		const gl::shader_input* camera_exposure;
		const gl::shader_input* model;
		const gl::shader_input* view;
		const gl::shader_input* projection;
		const gl::shader_input* model_view;
		const gl::shader_input* view_projection;
		const gl::shader_input* model_view_projection;
		const gl::shader_input* normal_model;
		const gl::shader_input* normal_model_view;
		const gl::shader_input* clip_depth;
		const gl::shader_input* log_depth_coef;
		
		const gl::shader_input* ambient_light_count;
		const gl::shader_input* ambient_light_colors;
		const gl::shader_input* point_light_count;
		const gl::shader_input* point_light_colors;
		const gl::shader_input* point_light_positions;
		const gl::shader_input* point_light_attenuations;
		const gl::shader_input* directional_light_count;
		const gl::shader_input* directional_light_colors;
		const gl::shader_input* directional_light_directions;
		const gl::shader_input* directional_light_textures;
		const gl::shader_input* directional_light_texture_matrices;
		const gl::shader_input* directional_light_texture_opacities;
		const gl::shader_input* spot_light_count;
		const gl::shader_input* spot_light_colors;
		const gl::shader_input* spot_light_positions;
		const gl::shader_input* spot_light_directions;
		const gl::shader_input* spot_light_attenuations;
		const gl::shader_input* spot_light_cutoffs;
		
		const gl::shader_input* shadow_map_directional;
		const gl::shader_input* shadow_splits_directional;
		const gl::shader_input* shadow_matrices_directional;
		
		const gl::shader_input* skinning_palette;
	};

	const parameter_set* load_parameter_set(const gl::shader_program* program) const;

	mutable std::unordered_map<const gl::shader_program*, parameter_set*> parameter_sets;
	const material* fallback_material;
	float2 mouse_position;
	
	int max_ambient_light_count;
	int max_point_light_count;
	int max_directional_light_count;
	int max_spot_light_count;
	int max_bone_count;
	
	mutable int ambient_light_count;
	mutable int point_light_count;
	mutable int directional_light_count;
	mutable int spot_light_count;

	float3* ambient_light_colors;
	float3* point_light_colors;
	float3* point_light_positions;
	float3* point_light_attenuations;
	float3* directional_light_colors;
	float3* directional_light_directions;
	const gl::texture_2d** directional_light_textures;
	float4x4* directional_light_texture_matrices;
	float* directional_light_texture_opacities;
	float3* spot_light_colors;
	float3* spot_light_positions;
	float3* spot_light_directions;
	float3* spot_light_attenuations;
	float2* spot_light_cutoffs;
};

} // namespace render

#endif // ANTKEEPER_RENDER_MATERIAL_PASS_HPP