nick/WickedEngine
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
Files
39d3ab1e19cd143d09e8d6f6aea0800e04b92ee7
WickedEngine/WickedEngine/offlineshadercompiler.cpp
T
Turánszki János 39d3ab1e19 Impostor persistent data (#465) 
* Impostor persistent gpu data

* impostor buffer frustum culling

* fixes

* impostor ao

* impostor wireframe fix
2022-06-17 15:19:42 +02:00

569 lines
26 KiB
C++
Raw Blame History

#include "WickedEngine.h"
#include <iostream>
#include <iomanip>
#include <mutex>
#include <string>
std::mutex locker;
struct ShaderEntry
{
std::string name;
wi::graphics::ShaderStage stage = wi::graphics::ShaderStage::Count;
wi::graphics::ShaderModel minshadermodel = wi::graphics::ShaderModel::SM_5_0;
struct Permutation
{
wi::vector<std::string> defines;
};
wi::vector<Permutation> permutations;
};
wi::vector<ShaderEntry> shaders = {
{"hairparticle_simulateCS", wi::graphics::ShaderStage::CS},
{"hairparticle_finishUpdateCS", wi::graphics::ShaderStage::CS},
{"emittedparticle_simulateCS", wi::graphics::ShaderStage::CS},
{"generateMIPChainCubeCS_float4", wi::graphics::ShaderStage::CS},
{"generateMIPChainCubeCS_unorm4", wi::graphics::ShaderStage::CS},
{"generateMIPChainCubeArrayCS_float4", wi::graphics::ShaderStage::CS},
{"generateMIPChainCubeArrayCS_unorm4", wi::graphics::ShaderStage::CS},
{"generateMIPChain3DCS_float4", wi::graphics::ShaderStage::CS},
{"generateMIPChain3DCS_unorm4", wi::graphics::ShaderStage::CS},
{"generateMIPChain2DCS_float4", wi::graphics::ShaderStage::CS},
{"generateMIPChain2DCS_unorm4", wi::graphics::ShaderStage::CS},
{"blur_gaussian_float4CS", wi::graphics::ShaderStage::CS},
{"bloomseparateCS", wi::graphics::ShaderStage::CS},
{"depthoffield_mainCS", wi::graphics::ShaderStage::CS},
{"depthoffield_neighborhoodMaxCOCCS", wi::graphics::ShaderStage::CS},
{"depthoffield_prepassCS", wi::graphics::ShaderStage::CS},
{"depthoffield_upsampleCS", wi::graphics::ShaderStage::CS},
{"depthoffield_tileMaxCOC_verticalCS", wi::graphics::ShaderStage::CS},
{"depthoffield_tileMaxCOC_horizontalCS", wi::graphics::ShaderStage::CS},
{"voxelRadianceSecondaryBounceCS", wi::graphics::ShaderStage::CS},
{"voxelSceneCopyClearCS", wi::graphics::ShaderStage::CS},
{"voxelClearOnlyNormalCS", wi::graphics::ShaderStage::CS},
{"upsample_bilateral_float1CS", wi::graphics::ShaderStage::CS},
{"upsample_bilateral_float4CS", wi::graphics::ShaderStage::CS},
{"upsample_bilateral_unorm1CS", wi::graphics::ShaderStage::CS},
{"upsample_bilateral_unorm4CS", wi::graphics::ShaderStage::CS},
{"temporalaaCS", wi::graphics::ShaderStage::CS},
{"tileFrustumsCS", wi::graphics::ShaderStage::CS},
{"tonemapCS", wi::graphics::ShaderStage::CS},
{"underwaterCS", wi::graphics::ShaderStage::CS},
{"fsr_upscalingCS", wi::graphics::ShaderStage::CS},
{"fsr_sharpenCS", wi::graphics::ShaderStage::CS},
{"ssaoCS", wi::graphics::ShaderStage::CS},
{"rtreflectionCS", wi::graphics::ShaderStage::CS, wi::graphics::ShaderModel::SM_6_5},
{"ssr_tileMaxRoughness_horizontalCS", wi::graphics::ShaderStage::CS},
{"ssr_tileMaxRoughness_verticalCS", wi::graphics::ShaderStage::CS},
{"ssr_kickjobsCS", wi::graphics::ShaderStage::CS},
{"ssr_depthHierarchyCS", wi::graphics::ShaderStage::CS},
{"ssr_resolveCS", wi::graphics::ShaderStage::CS},
{"ssr_temporalCS", wi::graphics::ShaderStage::CS},
{"ssr_bilateralCS", wi::graphics::ShaderStage::CS},
{"ssr_raytraceCS", wi::graphics::ShaderStage::CS},
{"ssr_raytraceCS_cheap", wi::graphics::ShaderStage::CS},
{"ssr_raytraceCS_earlyexit", wi::graphics::ShaderStage::CS},
{"sharpenCS", wi::graphics::ShaderStage::CS},
{"skinningCS", wi::graphics::ShaderStage::CS},
{"resolveMSAADepthStencilCS", wi::graphics::ShaderStage::CS},
{"raytraceCS", wi::graphics::ShaderStage::CS},
{"raytraceCS_rtapi", wi::graphics::ShaderStage::CS, wi::graphics::ShaderModel::SM_6_5},
{"paint_textureCS", wi::graphics::ShaderStage::CS},
{"oceanUpdateDisplacementMapCS", wi::graphics::ShaderStage::CS},
{"oceanUpdateGradientFoldingCS", wi::graphics::ShaderStage::CS},
{"oceanSimulatorCS", wi::graphics::ShaderStage::CS},
{"msao_interleaveCS", wi::graphics::ShaderStage::CS},
{"msao_preparedepthbuffers1CS", wi::graphics::ShaderStage::CS},
{"msao_preparedepthbuffers2CS", wi::graphics::ShaderStage::CS},
{"msao_blurupsampleCS", wi::graphics::ShaderStage::CS},
{"msao_blurupsampleCS_blendout", wi::graphics::ShaderStage::CS},
{"msao_blurupsampleCS_premin", wi::graphics::ShaderStage::CS},
{"msao_blurupsampleCS_premin_blendout", wi::graphics::ShaderStage::CS},
{"msaoCS", wi::graphics::ShaderStage::CS},
{"motionblur_neighborhoodMaxVelocityCS", wi::graphics::ShaderStage::CS},
{"motionblur_tileMaxVelocity_horizontalCS", wi::graphics::ShaderStage::CS},
{"motionblur_tileMaxVelocity_verticalCS", wi::graphics::ShaderStage::CS},
{"luminancePass2CS", wi::graphics::ShaderStage::CS},
{"motionblur_kickjobsCS", wi::graphics::ShaderStage::CS},
{"motionblurCS", wi::graphics::ShaderStage::CS},
{"motionblurCS_cheap", wi::graphics::ShaderStage::CS},
{"motionblurCS_earlyexit", wi::graphics::ShaderStage::CS},
{"luminancePass1CS", wi::graphics::ShaderStage::CS},
{"lightShaftsCS", wi::graphics::ShaderStage::CS},
{"lightCullingCS_ADVANCED_DEBUG", wi::graphics::ShaderStage::CS},
{"lightCullingCS_DEBUG", wi::graphics::ShaderStage::CS},
{"lightCullingCS", wi::graphics::ShaderStage::CS},
{"lightCullingCS_ADVANCED", wi::graphics::ShaderStage::CS},
{"hbaoCS", wi::graphics::ShaderStage::CS},
{"gpusortlib_sortInnerCS", wi::graphics::ShaderStage::CS},
{"gpusortlib_sortStepCS", wi::graphics::ShaderStage::CS},
{"gpusortlib_kickoffSortCS", wi::graphics::ShaderStage::CS},
{"gpusortlib_sortCS", wi::graphics::ShaderStage::CS},
{"fxaaCS", wi::graphics::ShaderStage::CS},
{"filterEnvMapCS", wi::graphics::ShaderStage::CS},
{"fft_512x512_c2c_CS", wi::graphics::ShaderStage::CS},
{"fft_512x512_c2c_v2_CS", wi::graphics::ShaderStage::CS},
{"emittedparticle_sphpartitionCS", wi::graphics::ShaderStage::CS},
{"emittedparticle_sphpartitionoffsetsCS", wi::graphics::ShaderStage::CS},
{"emittedparticle_sphpartitionoffsetsresetCS", wi::graphics::ShaderStage::CS},
{"emittedparticle_simulateCS_SORTING", wi::graphics::ShaderStage::CS},
{"emittedparticle_simulateCS_SORTING_DEPTHCOLLISIONS", wi::graphics::ShaderStage::CS},
{"emittedparticle_sphdensityCS", wi::graphics::ShaderStage::CS},
{"emittedparticle_sphforceCS", wi::graphics::ShaderStage::CS},
{"emittedparticle_kickoffUpdateCS", wi::graphics::ShaderStage::CS},
{"emittedparticle_simulateCS_DEPTHCOLLISIONS", wi::graphics::ShaderStage::CS},
{"emittedparticle_emitCS", wi::graphics::ShaderStage::CS},
{"emittedparticle_emitCS_FROMMESH", wi::graphics::ShaderStage::CS},
{"emittedparticle_emitCS_volume", wi::graphics::ShaderStage::CS},
{"emittedparticle_finishUpdateCS", wi::graphics::ShaderStage::CS},
{"downsample4xCS", wi::graphics::ShaderStage::CS},
{"depthoffield_prepassCS_earlyexit", wi::graphics::ShaderStage::CS},
{"depthoffield_mainCS_cheap", wi::graphics::ShaderStage::CS},
{"depthoffield_mainCS_earlyexit", wi::graphics::ShaderStage::CS },
{"depthoffield_postfilterCS", wi::graphics::ShaderStage::CS },
{"depthoffield_kickjobsCS", wi::graphics::ShaderStage::CS },
{"copytexture2D_float4_borderexpandCS", wi::graphics::ShaderStage::CS },
{"copytexture2D_unorm4_borderexpandCS", wi::graphics::ShaderStage::CS },
{"copytexture2D_unorm4CS", wi::graphics::ShaderStage::CS },
{"copytexture2D_float4CS", wi::graphics::ShaderStage::CS },
{"chromatic_aberrationCS", wi::graphics::ShaderStage::CS },
{"bvh_hierarchyCS", wi::graphics::ShaderStage::CS },
{"bvh_primitivesCS", wi::graphics::ShaderStage::CS },
{"bvh_propagateaabbCS", wi::graphics::ShaderStage::CS },
{"blur_gaussian_wide_unorm1CS", wi::graphics::ShaderStage::CS },
{"blur_gaussian_wide_unorm4CS", wi::graphics::ShaderStage::CS },
{"blur_gaussian_unorm1CS", wi::graphics::ShaderStage::CS },
{"blur_gaussian_unorm4CS", wi::graphics::ShaderStage::CS },
{"blur_gaussian_wide_float1CS", wi::graphics::ShaderStage::CS },
{"blur_gaussian_wide_float3CS", wi::graphics::ShaderStage::CS },
{"blur_gaussian_wide_float4CS", wi::graphics::ShaderStage::CS },
{"blur_bilateral_wide_unorm4CS", wi::graphics::ShaderStage::CS },
{"blur_gaussian_float1CS", wi::graphics::ShaderStage::CS },
{"blur_gaussian_float3CS", wi::graphics::ShaderStage::CS },
{"blur_bilateral_unorm4CS", wi::graphics::ShaderStage::CS },
{"blur_bilateral_wide_float1CS", wi::graphics::ShaderStage::CS },
{"blur_bilateral_wide_float3CS", wi::graphics::ShaderStage::CS },
{"blur_bilateral_wide_float4CS", wi::graphics::ShaderStage::CS },
{"blur_bilateral_wide_unorm1CS", wi::graphics::ShaderStage::CS },
{"blur_bilateral_float1CS", wi::graphics::ShaderStage::CS },
{"blur_bilateral_float3CS", wi::graphics::ShaderStage::CS },
{"blur_bilateral_float4CS", wi::graphics::ShaderStage::CS },
{"blur_bilateral_unorm1CS", wi::graphics::ShaderStage::CS },
{"voxelSceneCopyClearCS_TemporalSmoothing", wi::graphics::ShaderStage::CS },
{"normalsfromdepthCS", wi::graphics::ShaderStage::CS },
{"volumetricCloud_shapenoiseCS", wi::graphics::ShaderStage::CS },
{"volumetricCloud_detailnoiseCS", wi::graphics::ShaderStage::CS },
{"volumetricCloud_curlnoiseCS", wi::graphics::ShaderStage::CS },
{"volumetricCloud_weathermapCS", wi::graphics::ShaderStage::CS },
{"volumetricCloud_renderCS", wi::graphics::ShaderStage::CS },
{"volumetricCloud_reprojectCS", wi::graphics::ShaderStage::CS },
{"volumetricCloud_temporalCS", wi::graphics::ShaderStage::CS },
{"shadingRateClassificationCS", wi::graphics::ShaderStage::CS },
{"shadingRateClassificationCS_DEBUG", wi::graphics::ShaderStage::CS },
{"skyAtmosphere_transmittanceLutCS", wi::graphics::ShaderStage::CS },
{"skyAtmosphere_skyViewLutCS", wi::graphics::ShaderStage::CS },
{"skyAtmosphere_multiScatteredLuminanceLutCS", wi::graphics::ShaderStage::CS },
{"skyAtmosphere_skyLuminanceLutCS", wi::graphics::ShaderStage::CS },
{"upsample_bilateral_uint4CS", wi::graphics::ShaderStage::CS },
{"screenspaceshadowCS", wi::graphics::ShaderStage::CS },
{"rtshadowCS", wi::graphics::ShaderStage::CS, wi::graphics::ShaderModel::SM_6_5 },
{"rtshadow_denoise_tileclassificationCS", wi::graphics::ShaderStage::CS },
{"rtshadow_denoise_filterCS", wi::graphics::ShaderStage::CS },
{"rtshadow_denoise_temporalCS", wi::graphics::ShaderStage::CS },
{"rtaoCS", wi::graphics::ShaderStage::CS, wi::graphics::ShaderModel::SM_6_5 },
{"rtao_denoise_tileclassificationCS", wi::graphics::ShaderStage::CS },
{"rtao_denoise_filterCS", wi::graphics::ShaderStage::CS },
{"visibility_resolveCS", wi::graphics::ShaderStage::CS },
{"visibility_resolveCS_MSAA", wi::graphics::ShaderStage::CS },
{"visibility_indirect_prepareCS", wi::graphics::ShaderStage::CS },
{"visibility_velocityCS", wi::graphics::ShaderStage::CS },
{"visibility_skyCS", wi::graphics::ShaderStage::CS },
{"surfel_coverageCS", wi::graphics::ShaderStage::CS },
{"surfel_indirectprepareCS", wi::graphics::ShaderStage::CS },
{"surfel_updateCS", wi::graphics::ShaderStage::CS },
{"surfel_gridresetCS", wi::graphics::ShaderStage::CS },
{"surfel_gridoffsetsCS", wi::graphics::ShaderStage::CS },
{"surfel_binningCS", wi::graphics::ShaderStage::CS },
{"surfel_raytraceCS_rtapi", wi::graphics::ShaderStage::CS, wi::graphics::ShaderModel::SM_6_5 },
{"surfel_raytraceCS", wi::graphics::ShaderStage::CS },
{"surfel_integrateCS", wi::graphics::ShaderStage::CS },
{"ddgi_raytraceCS", wi::graphics::ShaderStage::CS },
{"ddgi_raytraceCS_rtapi", wi::graphics::ShaderStage::CS, wi::graphics::ShaderModel::SM_6_5 },
{"ddgi_updateCS", wi::graphics::ShaderStage::CS },
{"ddgi_updateCS_depth", wi::graphics::ShaderStage::CS },
{"terrainVirtualTextureUpdateCS", wi::graphics::ShaderStage::CS },
{"meshlet_prepareCS", wi::graphics::ShaderStage::CS },
{"impostor_prepareCS", wi::graphics::ShaderStage::CS },
{"emittedparticlePS_soft", wi::graphics::ShaderStage::PS },
{"imagePS", wi::graphics::ShaderStage::PS },
{"emittedparticlePS_soft_lighting", wi::graphics::ShaderStage::PS },
{"oceanSurfacePS", wi::graphics::ShaderStage::PS },
{"hairparticlePS", wi::graphics::ShaderStage::PS },
{"volumetricLight_SpotPS", wi::graphics::ShaderStage::PS },
{"volumetricLight_PointPS", wi::graphics::ShaderStage::PS },
{"volumetricLight_DirectionalPS", wi::graphics::ShaderStage::PS },
{"voxelPS", wi::graphics::ShaderStage::PS },
{"vertexcolorPS", wi::graphics::ShaderStage::PS },
{"upsample_bilateralPS", wi::graphics::ShaderStage::PS },
{"sunPS", wi::graphics::ShaderStage::PS },
{"skyPS_dynamic", wi::graphics::ShaderStage::PS },
{"skyPS_static", wi::graphics::ShaderStage::PS },
{"shadowPS_transparent", wi::graphics::ShaderStage::PS },
{"shadowPS_water", wi::graphics::ShaderStage::PS },
{"shadowPS_alphatest", wi::graphics::ShaderStage::PS },
{"renderlightmapPS", wi::graphics::ShaderStage::PS },
{"renderlightmapPS_rtapi", wi::graphics::ShaderStage::PS, wi::graphics::ShaderModel::SM_6_5 },
{"raytrace_debugbvhPS", wi::graphics::ShaderStage::PS },
{"outlinePS", wi::graphics::ShaderStage::PS },
{"oceanSurfaceSimplePS", wi::graphics::ShaderStage::PS },
{"objectPS_transparent_pom", wi::graphics::ShaderStage::PS },
{"objectPS_water", wi::graphics::ShaderStage::PS },
{"objectPS_voxelizer", wi::graphics::ShaderStage::PS },
{"objectPS_transparent", wi::graphics::ShaderStage::PS },
{"objectPS_transparent_planarreflection", wi::graphics::ShaderStage::PS },
{"objectPS_planarreflection", wi::graphics::ShaderStage::PS },
{"objectPS_pom", wi::graphics::ShaderStage::PS },
{"objectPS", wi::graphics::ShaderStage::PS },
{"objectPS_hologram", wi::graphics::ShaderStage::PS },
{"objectPS_paintradius", wi::graphics::ShaderStage::PS },
{"objectPS_simple", wi::graphics::ShaderStage::PS },
{"objectPS_debug", wi::graphics::ShaderStage::PS },
{"objectPS_prepass", wi::graphics::ShaderStage::PS },
{"objectPS_prepass_alphatest", wi::graphics::ShaderStage::PS },
{"objectPS_anisotropic", wi::graphics::ShaderStage::PS },
{"objectPS_transparent_anisotropic", wi::graphics::ShaderStage::PS },
{"objectPS_cloth", wi::graphics::ShaderStage::PS },
{"objectPS_transparent_cloth", wi::graphics::ShaderStage::PS },
{"objectPS_clearcoat", wi::graphics::ShaderStage::PS },
{"objectPS_transparent_clearcoat", wi::graphics::ShaderStage::PS },
{"objectPS_cloth_clearcoat", wi::graphics::ShaderStage::PS },
{"objectPS_transparent_cloth_clearcoat", wi::graphics::ShaderStage::PS },
{"objectPS_cartoon", wi::graphics::ShaderStage::PS },
{"objectPS_transparent_cartoon", wi::graphics::ShaderStage::PS },
{"objectPS_unlit", wi::graphics::ShaderStage::PS },
{"objectPS_transparent_unlit", wi::graphics::ShaderStage::PS },
{"lightVisualizerPS", wi::graphics::ShaderStage::PS },
{"lensFlarePS", wi::graphics::ShaderStage::PS },
{"impostorPS", wi::graphics::ShaderStage::PS },
{"impostorPS_simple", wi::graphics::ShaderStage::PS },
{"impostorPS_prepass", wi::graphics::ShaderStage::PS },
{"hairparticlePS_simple", wi::graphics::ShaderStage::PS },
{"hairparticlePS_prepass", wi::graphics::ShaderStage::PS },
{"forceFieldVisualizerPS", wi::graphics::ShaderStage::PS },
{"fontPS", wi::graphics::ShaderStage::PS },
{"envMap_skyPS_static", wi::graphics::ShaderStage::PS },
{"envMap_skyPS_dynamic", wi::graphics::ShaderStage::PS },
{"envMapPS", wi::graphics::ShaderStage::PS },
{"emittedparticlePS_soft_distortion", wi::graphics::ShaderStage::PS },
{"downsampleDepthBuffer4xPS", wi::graphics::ShaderStage::PS },
{"emittedparticlePS_simple", wi::graphics::ShaderStage::PS },
{"cubeMapPS", wi::graphics::ShaderStage::PS },
{"circlePS", wi::graphics::ShaderStage::PS },
{"captureImpostorPS", wi::graphics::ShaderStage::PS },
{"ddgi_debugPS", wi::graphics::ShaderStage::PS },
{"hairparticleVS", wi::graphics::ShaderStage::VS },
{"emittedparticleVS", wi::graphics::ShaderStage::VS },
{"imageVS", wi::graphics::ShaderStage::VS },
{"fontVS", wi::graphics::ShaderStage::VS },
{"voxelVS", wi::graphics::ShaderStage::VS },
{"vertexcolorVS", wi::graphics::ShaderStage::VS },
{"volumetriclight_directionalVS", wi::graphics::ShaderStage::VS },
{"volumetriclight_pointVS", wi::graphics::ShaderStage::VS },
{"volumetriclight_spotVS", wi::graphics::ShaderStage::VS },
{"vSpotLightVS", wi::graphics::ShaderStage::VS },
{"vPointLightVS", wi::graphics::ShaderStage::VS },
{"sphereVS", wi::graphics::ShaderStage::VS },
{"skyVS", wi::graphics::ShaderStage::VS },
{"shadowVS_transparent", wi::graphics::ShaderStage::VS },
{"shadowVS", wi::graphics::ShaderStage::VS },
{"shadowVS_alphatest", wi::graphics::ShaderStage::VS },
{"postprocessVS", wi::graphics::ShaderStage::VS },
{"renderlightmapVS", wi::graphics::ShaderStage::VS },
{"raytrace_screenVS", wi::graphics::ShaderStage::VS },
{"oceanSurfaceVS", wi::graphics::ShaderStage::VS },
{"objectVS_simple", wi::graphics::ShaderStage::VS },
{"objectVS_voxelizer", wi::graphics::ShaderStage::VS },
{"objectVS_common", wi::graphics::ShaderStage::VS },
{"objectVS_common_tessellation", wi::graphics::ShaderStage::VS },
{"objectVS_prepass", wi::graphics::ShaderStage::VS },
{"objectVS_prepass_alphatest", wi::graphics::ShaderStage::VS },
{"objectVS_prepass_tessellation", wi::graphics::ShaderStage::VS },
{"objectVS_prepass_alphatest_tessellation", wi::graphics::ShaderStage::VS },
{"objectVS_simple_tessellation", wi::graphics::ShaderStage::VS },
{"objectVS_debug", wi::graphics::ShaderStage::VS },
{"lensFlareVS", wi::graphics::ShaderStage::VS },
{"impostorVS", wi::graphics::ShaderStage::VS },
{"forceFieldPointVisualizerVS", wi::graphics::ShaderStage::VS },
{"forceFieldPlaneVisualizerVS", wi::graphics::ShaderStage::VS },
{"envMap_skyVS", wi::graphics::ShaderStage::VS },
{"envMapVS", wi::graphics::ShaderStage::VS },
{"envMap_skyVS_emulation", wi::graphics::ShaderStage::VS },
{"envMapVS_emulation", wi::graphics::ShaderStage::VS },
{"cubeShadowVS", wi::graphics::ShaderStage::VS },
{"cubeShadowVS_alphatest", wi::graphics::ShaderStage::VS },
{"cubeShadowVS_emulation", wi::graphics::ShaderStage::VS },
{"cubeShadowVS_alphatest_emulation", wi::graphics::ShaderStage::VS },
{"cubeShadowVS_transparent", wi::graphics::ShaderStage::VS },
{"cubeShadowVS_transparent_emulation", wi::graphics::ShaderStage::VS },
{"occludeeVS", wi::graphics::ShaderStage::VS },
{"ddgi_debugVS", wi::graphics::ShaderStage::VS },
{"envMap_skyGS_emulation", wi::graphics::ShaderStage::GS },
{"envMapGS_emulation", wi::graphics::ShaderStage::GS },
{"cubeShadowGS_emulation", wi::graphics::ShaderStage::GS },
{"cubeShadowGS_alphatest_emulation", wi::graphics::ShaderStage::GS },
{"cubeShadowGS_transparent_emulation", wi::graphics::ShaderStage::GS },
{"voxelGS", wi::graphics::ShaderStage::GS },
{"objectGS_voxelizer", wi::graphics::ShaderStage::GS },
{"objectDS", wi::graphics::ShaderStage::DS },
{"objectDS_prepass", wi::graphics::ShaderStage::DS },
{"objectDS_prepass_alphatest", wi::graphics::ShaderStage::DS },
{"objectDS_simple", wi::graphics::ShaderStage::DS },
{"objectHS", wi::graphics::ShaderStage::HS },
{"objectHS_prepass", wi::graphics::ShaderStage::HS },
{"objectHS_prepass_alphatest", wi::graphics::ShaderStage::HS },
{"objectHS_simple", wi::graphics::ShaderStage::HS },
{"emittedparticleMS", wi::graphics::ShaderStage::MS },
{"rtreflectionLIB", wi::graphics::ShaderStage::LIB },
};
struct Target
{
wi::graphics::ShaderFormat format;
std::string dir;
};
wi::vector<Target> targets;
wi::unordered_map<std::string, wi::shadercompiler::CompilerOutput> results;
bool rebuild = false;
bool shaderdump_enabled = false;
using namespace wi::graphics;
int main(int argc, char* argv[])
{
std::cout << "[Wicked Engine Offline Shader Compiler]" << std::endl;
std::cout << "Available command arguments:" << std::endl;
std::cout << "\thlsl5 : \tCompile shaders to hlsl5 (dx11) format (using d3dcompiler)" << std::endl;
std::cout << "\thlsl6 : \tCompile shaders to hlsl6 (dx12) format (using dxcompiler)" << std::endl;
std::cout << "\tspirv : \tCompile shaders to spirv (vulkan) format (using dxcompiler)" << std::endl;
std::cout << "\trebuild : \tAll shaders will be rebuilt, regardless if they are outdated or not" << std::endl;
std::cout << "\tshaderdump : \tShaders will be saved to wiShaderDump.h C++ header file (rebuild is assumed)" << std::endl;
std::cout << "Command arguments used: ";
wi::arguments::Parse(argc, argv);
if (wi::arguments::HasArgument("hlsl5"))
{
targets.push_back({ ShaderFormat::HLSL5, "shaders/hlsl5/" });
std::cout << "hlsl5 ";
}
if (wi::arguments::HasArgument("hlsl6"))
{
targets.push_back({ ShaderFormat::HLSL6, "shaders/hlsl6/" });
std::cout << "hlsl6 ";
}
if (wi::arguments::HasArgument("spirv"))
{
targets.push_back({ ShaderFormat::SPIRV, "shaders/spirv/" });
std::cout << "spirv ";
}
if (wi::arguments::HasArgument("shaderdump"))
{
shaderdump_enabled = true;
rebuild = true;
std::cout << "shaderdump ";
}
if (wi::arguments::HasArgument("rebuild"))
{
rebuild = true;
std::cout << "rebuild ";
}
std::cout << std::endl;
if (targets.empty())
{
targets = {
//{ ShaderFormat::HLSL5, "shaders/hlsl5/" },
{ ShaderFormat::HLSL6, "shaders/hlsl6/" },
{ ShaderFormat::SPIRV, "shaders/spirv/" },
};
std::cout << "No shader formats were specified, assuming command arguments: hlsl5 spirv hlsl6" << std::endl;
}
// permutations for visibility_surfaceCS:
shaders.push_back({ "visibility_surfaceCS", wi::graphics::ShaderStage::CS });
for (auto& x : wi::scene::MaterialComponent::shaderTypeDefines)
{
shaders.back().permutations.emplace_back().defines = x;
}
// permutations for visibility_surfaceCS REDUCED:
shaders.push_back({ "visibility_surfaceCS", wi::graphics::ShaderStage::CS });
for (auto& x : wi::scene::MaterialComponent::shaderTypeDefines)
{
auto defines = x;
defines.push_back("REDUCED");
shaders.back().permutations.emplace_back().defines = defines;
}
// permutations for visibility_shadeCS:
shaders.push_back({ "visibility_shadeCS", wi::graphics::ShaderStage::CS });
for (auto& x : wi::scene::MaterialComponent::shaderTypeDefines)
{
shaders.back().permutations.emplace_back().defines = x;
}
wi::jobsystem::Initialize();
wi::jobsystem::context ctx;
std::string SHADERSOURCEPATH = wi::renderer::GetShaderSourcePath();
wi::helper::MakePathAbsolute(SHADERSOURCEPATH);
std::cout << "[Wicked Engine Offline Shader Compiler] Searching for outdated shaders..." << std::endl;
wi::Timer timer;
for (auto& target : targets)
{
std::string SHADERPATH = target.dir;
wi::helper::DirectoryCreate(SHADERPATH);
for (auto& shader : shaders)
{
if (target.format == ShaderFormat::HLSL5)
{
if (
shader.stage == ShaderStage::MS ||
shader.stage == ShaderStage::AS ||
shader.stage == ShaderStage::LIB
)
{
// shader stage not applicable to HLSL5
continue;
}
}
wi::vector<ShaderEntry::Permutation> permutations = shader.permutations;
if (permutations.empty())
{
permutations.emplace_back();
}
for (auto permutation : permutations)
{
wi::jobsystem::Execute(ctx, [=](wi::jobsystem::JobArgs args) {
std::string shaderbinaryfilename = SHADERPATH + shader.name;
for (auto& def : permutation.defines)
{
shaderbinaryfilename += "_" + def;
}
shaderbinaryfilename += ".cso";
if (!rebuild && !wi::shadercompiler::IsShaderOutdated(shaderbinaryfilename))
{
return;
}
wi::shadercompiler::CompilerInput input;
input.format = target.format;
input.stage = shader.stage;
input.shadersourcefilename = SHADERSOURCEPATH + shader.name + ".hlsl";
input.include_directories.push_back(SHADERSOURCEPATH);
input.minshadermodel = shader.minshadermodel;
input.defines = permutation.defines;
if (input.minshadermodel > ShaderModel::SM_5_0 && target.format == ShaderFormat::HLSL5)
{
// if shader format cannot support shader model, then we cancel the task without returning error
return;
}
wi::shadercompiler::CompilerOutput output;
wi::shadercompiler::Compile(input, output);
if (output.IsValid())
{
wi::shadercompiler::SaveShaderAndMetadata(shaderbinaryfilename, output);
locker.lock();
if (!output.error_message.empty())
{
std::cerr << output.error_message << std::endl;
}
std::cout << "shader compiled: " << shaderbinaryfilename << std::endl;
if (shaderdump_enabled)
{
results[shaderbinaryfilename] = output;
}
locker.unlock();
}
else
{
locker.lock();
std::cerr << "shader compile FAILED: " << shaderbinaryfilename << std::endl << output.error_message;
locker.unlock();
std::exit(1);
}
});
}
}
}
wi::jobsystem::Wait(ctx);
std::cout << "[Wicked Engine Offline Shader Compiler] Finished in " << std::setprecision(4) << timer.elapsed_seconds() << " seconds" << std::endl;
if (shaderdump_enabled)
{
std::cout << "[Wicked Engine Offline Shader Compiler] Creating ShaderDump..." << std::endl;
timer.record();
std::string ss;
ss += "namespace wiShaderDump {\n";
for (auto& x : results)
{
auto& name = x.first;
auto& output = x.second;
std::string name_repl = name;
std::replace(name_repl.begin(), name_repl.end(), '/', '_');
std::replace(name_repl.begin(), name_repl.end(), '.', '_');
ss += "const uint8_t " + name_repl + "[] = {";
for (size_t i = 0; i < output.shadersize; ++i)
{
ss += std::to_string((uint32_t)output.shaderdata[i]) + ",";
}
ss += "};\n";
}
ss += "struct ShaderDumpEntry{const uint8_t* data; size_t size;};\n";
ss += "const wi::unordered_map<std::string, ShaderDumpEntry> shaderdump = {\n";
for (auto& x : results)
{
auto& name = x.first;
auto& output = x.second;
std::string name_repl = name;
std::replace(name_repl.begin(), name_repl.end(), '/', '_');
std::replace(name_repl.begin(), name_repl.end(), '.', '_');
ss += "std::pair<std::string, ShaderDumpEntry>(\"" + name + "\", {" + name_repl + ",sizeof(" + name_repl + ")}),\n";
}
ss += "};\n"; // map end
ss += "}\n"; // namespace end
wi::helper::FileWrite("wiShaderDump.h", (uint8_t*)ss.c_str(), ss.length());
std::cout << "[Wicked Engine Offline Shader Compiler] ShaderDump written to wiShaderDump.h in " << std::setprecision(4) << timer.elapsed_seconds() << " seconds" << std::endl;
}
return 0;
}
Reference in New Issue View Git Blame Copy Permalink