#pragma once

#include "OpenGLRenderPass.h"
#include "OpenGLRenderPipeline.h"
#include "OpenGLRenderer.h"
#include "OpenGLShaderPrograms.h"
#include "RenderCommandQueue.h"
#include "RenderFrameState.h"
#include "RenderFrameStateResolver.h"
#include "HealthTelemetry.h"
#include "RuntimeCoordinator.h"
#include "RuntimeSnapshotProvider.h"

#include <windows.h>

#include <cstdint>
#include <chrono>
#include <functional>
#include <string>
#include <vector>

class RenderEngine
{
public:
	using RenderEffectCallback = std::function<void()>;
	using ScreenshotCallback = std::function<void()>;
	using PreviewPaintCallback = std::function<void()>;

	struct OscOverlayUpdate
	{
		std::string routeKey;
		std::string layerKey;
		std::string parameterKey;
		JsonValue targetValue;
	};

	struct OscOverlayCommitCompletion
	{
		std::string routeKey;
		uint64_t generation = 0;
	};

	struct OscOverlayCommitRequest
	{
		std::string routeKey;
		std::string layerKey;
		std::string parameterKey;
		JsonValue value;
		uint64_t generation = 0;
	};

	RenderEngine(
		RuntimeSnapshotProvider& runtimeSnapshotProvider,
		HealthTelemetry& healthTelemetry,
		CRITICAL_SECTION& mutex,
		HDC hdc,
		HGLRC hglrc,
		RenderEffectCallback renderEffect,
		ScreenshotCallback screenshotReady,
		PreviewPaintCallback previewPaint);
	~RenderEngine();

	bool CompileDecodeShader(int errorMessageSize, char* errorMessage);
	bool CompileOutputPackShader(int errorMessageSize, char* errorMessage);
	bool InitializeResources(
		unsigned inputFrameWidth,
		unsigned inputFrameHeight,
		unsigned captureTextureWidth,
		unsigned outputFrameWidth,
		unsigned outputFrameHeight,
		unsigned outputPackTextureWidth,
		std::string& error);
	bool CompileLayerPrograms(unsigned inputFrameWidth, unsigned inputFrameHeight, int errorMessageSize, char* errorMessage);
	bool ApplyPreparedShaderBuild(
		const PreparedShaderBuild& preparedBuild,
		unsigned inputFrameWidth,
		unsigned inputFrameHeight,
		bool preserveFeedbackState,
		int errorMessageSize,
		char* errorMessage);

	void ResetTemporalHistoryState();
	void ResetShaderFeedbackState();
	void ApplyRuntimeCoordinatorRenderReset(RuntimeCoordinatorRenderResetScope resetScope);
	void ClearOscOverlayState();
	void UpdateOscOverlayState(
		const std::vector<OscOverlayUpdate>& updates,
		const std::vector<OscOverlayCommitCompletion>& completedCommits);
	void ResizeView(int width, int height);
	bool TryPresentPreview(bool force, unsigned previewFps, unsigned outputFrameWidth, unsigned outputFrameHeight);
	bool TryUploadInputFrame(const VideoIOFrame& inputFrame, const VideoIOState& videoState);
	bool RenderOutputFrame(const RenderPipelineFrameContext& context, VideoIOOutputFrame& outputFrame);
	bool ResolveRenderFrameState(
		const RenderFrameInput& input,
		std::vector<OscOverlayCommitRequest>* commitRequests,
		RenderFrameState& frameState);
	void RenderPreparedFrame(const RenderFrameState& frameState);
	bool CaptureOutputFrameRgbaTopDown(unsigned width, unsigned height, std::vector<unsigned char>& topDownPixels);

private:
	bool CommitPreparedLayerPrograms(const PreparedShaderBuild& preparedBuild, unsigned inputFrameWidth, unsigned inputFrameHeight, int errorMessageSize, char* errorMessage);
	void RenderLayerStack(
		bool hasInputSource,
		const std::vector<RuntimeRenderState>& layerStates,
		unsigned inputFrameWidth,
		unsigned inputFrameHeight,
		unsigned captureTextureWidth,
		VideoIOPixelFormat inputPixelFormat,
		unsigned historyCap);
	void ResetTemporalHistoryStateOnRenderThread();
	void ResetShaderFeedbackStateOnRenderThread();
	void ApplyRenderResetOnRenderThread(RenderCommandResetScope resetScope);
	void ProcessRenderResetCommandsOnRenderThread();
	bool PresentPreviewOnRenderThread(unsigned outputFrameWidth, unsigned outputFrameHeight);
	bool UploadInputFrameOnRenderThread(const VideoIOFrame& inputFrame, const VideoIOState& videoState);
	bool RenderOutputFrameOnRenderThread(const RenderPipelineFrameContext& context, VideoIOOutputFrame& outputFrame);
	bool ReadOutputFrameRgbaOnRenderThread(unsigned width, unsigned height, std::vector<unsigned char>& bottomUpPixels);
	bool CaptureOutputFrameRgbaTopDownOnRenderThread(unsigned width, unsigned height, std::vector<unsigned char>& topDownPixels);

	OpenGLRenderer mRenderer;
	OpenGLRenderPass mRenderPass;
	OpenGLRenderPipeline mRenderPipeline;
	OpenGLShaderPrograms mShaderPrograms;
	CRITICAL_SECTION& mMutex;
	HDC mHdc;
	HGLRC mHglrc;

	std::chrono::steady_clock::time_point mLastPreviewPresentTime;
	RenderCommandQueue mRenderCommandQueue;
	RenderFrameStateResolver mFrameStateResolver;
	RuntimeLiveState mRuntimeLiveState;
};