You've already forked UnrealEngineUWP
mirror of
https://github.com/izzy2lost/UnrealEngineUWP.git
synced 2026-03-26 18:15:20 -07:00
d5b66a61f2
* In this case, instead of baking each mesh material property to it's own buffer, we can write all the meshes to the same buffer as their UVs are already the final ones that do not overlap * Memory usage is greatly reduced (in test case with ~300 materials to bake, peak is reduced by 5.5GB) * A lot faster to process (~2x in most cases) as we are doing a single RT readback per baked material property * We can also skip the final merge of the baked outputs, as we have a single one Refactored the material baking module to have both code paths (single output and multi outputs) share as much code as possible #rb jeanfrancois.dube #preflight 63eba6e0284e76cdadbecb48 [CL 24226039 by sebastien lussier in ue5-main branch]
105 lines
4.5 KiB
C++
105 lines
4.5 KiB
C++
// Copyright Epic Games, Inc. All Rights Reserved.
|
|
|
|
#pragma once
|
|
|
|
#include "CoreMinimal.h"
|
|
#include "Modules/ModuleInterface.h"
|
|
#include "PixelFormat.h"
|
|
#include "SceneTypes.h"
|
|
#include "IMaterialBakingModule.h"
|
|
#include "MaterialPropertyEx.h"
|
|
|
|
class UTextureRenderTarget2D;
|
|
class UMaterialOptions;
|
|
class UMaterialInterface;
|
|
|
|
class FExportMaterialProxy;
|
|
class FTextureRenderTargetResource;
|
|
|
|
struct FMaterialData;
|
|
struct FMeshData;
|
|
struct FBakeOutput;
|
|
|
|
class MATERIALBAKING_API FMaterialBakingModule : public IMaterialBakingModule
|
|
{
|
|
public:
|
|
/** IModuleInterface overrides begin */
|
|
virtual void StartupModule();
|
|
virtual void ShutdownModule();
|
|
/** IModuleInterface overrides end */
|
|
|
|
/** Bakes out material properties according to MaterialSettings using MeshSettings and stores the output in Output */
|
|
virtual void BakeMaterials(const TArray<FMaterialData*>& MaterialSettings, const TArray<FMeshData*>& MeshSettings, TArray<FBakeOutput>& Output) override;
|
|
|
|
/** Bakes out material properties according to extended MaterialSettings using MeshSettings and stores the output in Output */
|
|
virtual void BakeMaterials(const TArray<FMaterialDataEx*>& MaterialSettings, const TArray<FMeshData*>& MeshSettings, TArray<FBakeOutputEx>& Output) override;
|
|
|
|
/** Bakes out material properties according to MaterialSettings using MeshSettings and stores the output in Output */
|
|
virtual void BakeMaterials(const TArray<FMaterialData*>& MaterialSettings, const TArray<FMeshData*>& MeshSettings, FBakeOutput& Output) override;
|
|
|
|
/** Bakes out material properties according to extended MaterialSettings using MeshSettings and stores the output in Output */
|
|
virtual void BakeMaterials(const TArray<FMaterialDataEx*>& MaterialSettings, const TArray<FMeshData*>& MeshSettings, FBakeOutputEx& Output) override;
|
|
|
|
/** Promps a slate window to allow the user to populate specific material baking settings used while baking out materials */
|
|
virtual bool SetupMaterialBakeSettings(TArray<TWeakObjectPtr<UObject>>& OptionObjects, int32 NumLODs) override;
|
|
|
|
/** Outputs true HDR version of emissive color */
|
|
virtual void SetEmissiveHDR(bool bHDR) override;
|
|
|
|
/** Bakes all material properties to linear textures, except for colors */
|
|
virtual void SetLinearBake(bool bCorrectLinear) override;
|
|
|
|
/** Returns whether a specific material property is baked to a linear texture or not */
|
|
virtual bool IsLinearBake(FMaterialPropertyEx Property) override;
|
|
|
|
/** Obtain a CRC than can help trigger a rebake if code/global settings impacting the bake result change */
|
|
virtual uint32 GetCRC() const override;
|
|
|
|
protected:
|
|
friend class FMaterialBakingProcessor;
|
|
|
|
/* Creates and adds or reuses a RenderTarget from the pool */
|
|
UTextureRenderTarget2D* CreateRenderTarget(FMaterialPropertyEx InProperty, const FIntPoint& InTargetSize, bool bInUsePooledRenderTargets, const FColor& BackgroundColor);
|
|
|
|
/* Creates and adds (or reuses a ExportMaterialProxy from the pool if MaterialBaking.UseMaterialProxyCaching is set to 1) */
|
|
FExportMaterialProxy* CreateMaterialProxy(const FMaterialDataEx* MaterialSettings, const FMaterialPropertyEx& Property);
|
|
|
|
/** Cleans up all cached material proxies in MaterialProxyPool */
|
|
void CleanupMaterialProxies();
|
|
|
|
/** Free up all pooled render targets */
|
|
void CleanupRenderTargets();
|
|
|
|
/** Callback for modified objects which should be removed from MaterialProxyPool in that case */
|
|
void OnObjectModified(UObject* Object);
|
|
|
|
/** Callback used to clear material proxy cache on garbage collection */
|
|
void OnPreGarbageCollect();
|
|
|
|
private:
|
|
enum EPropertyColorSpace
|
|
{
|
|
Linear,
|
|
sRGB,
|
|
};
|
|
|
|
/** Pool of available render targets, cached for re-using on consecutive property rendering */
|
|
TArray<UTextureRenderTarget2D*> RenderTargetPool;
|
|
|
|
/** Pool of cached material proxies to optimize material baking workflow, stays resident when MaterialBaking.UseMaterialProxyCaching is set to 1 */
|
|
typedef TWeakObjectPtr<UMaterialInterface> FMaterialPoolKey;
|
|
typedef TPair<FMaterialPropertyEx, FExportMaterialProxy*> FMaterialPoolValue;
|
|
typedef TMultiMap<FMaterialPoolKey, FMaterialPoolValue, FDefaultSetAllocator, TWeakObjectPtrMapKeyFuncs<FMaterialPoolKey, FMaterialPoolValue, true /*bInAllowDuplicateKeys*/>> FMaterialPoolMap;
|
|
FMaterialPoolMap MaterialProxyPool;
|
|
|
|
/** Pixel formats to use for baking out specific material properties */
|
|
TMap<FMaterialPropertyEx, EPixelFormat> PerPropertyFormat;
|
|
|
|
/** Whether or not to enforce gamma correction while baking out specific material properties */
|
|
TMap<FMaterialPropertyEx, EPropertyColorSpace> PerPropertyColorSpace;
|
|
|
|
EPropertyColorSpace DefaultColorSpace;
|
|
|
|
bool bEmissiveHDR;
|
|
};
|