UE4测试CopyTexture和CopyToResolveTarget的代码

// Fill out your copyright notice in the Description page of Project Settings. #include "TestCopyRenderFunctionLibrary.h" #include "Engine/TextureRenderTarget2D.h" #include "Engine/World.h" #include "Engine/Texture2DArray.h" #include "GlobalShader.h" #include "CommonRenderResources.h" #include "ScreenRendering.h" void UTestCopyRenderFunctionLibrary::CopyTextureToRenderTextureAllMips( const UObject* WorldContextObject, const UTexture2D* InTexture, class UTextureRenderTarget2D* OutputRenderTargetA) { UWorld* World = WorldContextObject->GetWorld(); ERHIFeatureLevel::Type FeatureLevel = World->Scene->GetFeatureLevel(); if (FeatureLevel < ERHIFeatureLevel::SM5) { UE_LOG(LogTemp, Warning, TEXT("FeatureLevel < ERHIFeatureLevel::SM5")); return; } if (nullptr == InTexture) { UE_LOG(LogTemp, Warning, TEXT("InTexture Is NULL")); return; } FTextureRenderTargetResource* TextureRenderTargetSource = OutputRenderTargetA->GameThread_GetRenderTargetResource(); FTexture* TextureSource = InTexture->Resource; int32 CommonMip = FMath::Min(TextureRenderTargetSource->TextureRHI->GetNumMips(), TextureSource->TextureRHI->GetNumMips()); ENQUEUE_RENDER_COMMAND(TestShaderCommand)( [TextureSource, TextureRenderTargetSource, FeatureLevel, CommonMip](FRHICommandListImmediate& RHICmdList) { if (CommonMip >= 2) { //Test Copy Subregion for (int32 MipIndex = 0; MipIndex < CommonMip - 1; ++MipIndex) { FRHICopyTextureInfo CopyInfo; CopyInfo.SourcePosition = FIntVector(0, 0, 0); CopyInfo.DestPosition = FIntVector(0, 0, 0); CopyInfo.SourceMipIndex = MipIndex; CopyInfo.DestMipIndex = MipIndex; CopyInfo.Size = FIntVector(TextureSource->GetSizeX() / FMath::Pow(2, MipIndex + 1), TextureSource->GetSizeY() / FMath::Pow(2, MipIndex + 1), TextureSource->GetSizeZ() / FMath::Pow(2, MipIndex + 1)); RHICmdList.Transition(FRHITransitionInfo(TextureRenderTargetSource->TextureRHI, ERHIAccess::Unknown, ERHIAccess::CopyDest)); RHICmdList.Transition(FRHITransitionInfo(TextureSource->TextureRHI, ERHIAccess::SRVMask, ERHIAccess::CopySrc)); FRHITexture* Test = TextureSource->TextureRHI.GetReference(); RHICmdList.CopyTexture(TextureSource->TextureRHI, TextureRenderTargetSource->TextureRHI, CopyInfo); RHICmdList.Transition(FRHITransitionInfo(TextureRenderTargetSource->TextureRHI, ERHIAccess::CopyDest, ERHIAccess::SRVMask)); RHICmdList.Transition(FRHITransitionInfo(TextureRenderTargetSource->TextureRHI, ERHIAccess::CopySrc, ERHIAccess::SRVMask)); } } else { FRHICopyTextureInfo CopyInfo; CopyInfo.SourcePosition = FIntVector(0, 0, 0); CopyInfo.DestPosition = FIntVector(0, 0, 0); CopyInfo.Size = FIntVector(TextureSource->GetSizeX() / 2, TextureSource->GetSizeY() /2, TextureSource->GetSizeZ() / 2); RHICmdList.Transition(FRHITransitionInfo(TextureRenderTargetSource->TextureRHI, ERHIAccess::Unknown, ERHIAccess::CopyDest)); RHICmdList.Transition(FRHITransitionInfo(TextureSource->TextureRHI, ERHIAccess::SRVMask, ERHIAccess::CopySrc)); RHICmdList.CopyTexture(TextureSource->TextureRHI, TextureRenderTargetSource->TextureRHI, CopyInfo); RHICmdList.Transition(FRHITransitionInfo(TextureRenderTargetSource->TextureRHI, ERHIAccess::CopyDest, ERHIAccess::SRVMask)); RHICmdList.Transition(FRHITransitionInfo(TextureRenderTargetSource->TextureRHI, ERHIAccess::CopySrc, ERHIAccess::SRVMask)); } } ); FlushRenderingCommands(); } void UTestCopyRenderFunctionLibrary::CopyTextureToTextureAllMips( const UObject* WorldContextObject, const UTexture2D* InTexture, class UTexture2D* OutputRenderTarget) { UWorld* World = WorldContextObject->GetWorld(); ERHIFeatureLevel::Type FeatureLevel = World->Scene->GetFeatureLevel(); if (FeatureLevel < ERHIFeatureLevel::SM5) { UE_LOG(LogTemp, Warning, TEXT("FeatureLevel < ERHIFeatureLevel::SM5")); return; } if (nullptr == InTexture) { UE_LOG(LogTemp, Warning, TEXT("InTexture Is NULL")); return; } FTexture* TextureRenderTargetSource = OutputRenderTarget->Resource; FTexture* TextureSource = InTexture->Resource; int32 CommonMip = FMath::Min(TextureRenderTargetSource->TextureRHI->GetNumMips(), TextureSource->TextureRHI->GetNumMips()); ENQUEUE_RENDER_COMMAND(TestShaderCommand)( [TextureSource, TextureRenderTargetSource, FeatureLevel, CommonMip](FRHICommandListImmediate& RHICmdList) { if (CommonMip >= 2) { //Test Copy Subregion for (int32 MipIndex = 0; MipIndex < CommonMip - 1; ++MipIndex) { FRHICopyTextureInfo CopyInfo; CopyInfo.SourcePosition = FIntVector(0, 0, 0); CopyInfo.DestPosition = FIntVector(0, 0, 0); CopyInfo.SourceMipIndex = MipIndex; CopyInfo.DestMipIndex = MipIndex; CopyInfo.Size = FIntVector(TextureSource->GetSizeX() / FMath::Pow(2, MipIndex + 1), TextureSource->GetSizeY() / FMath::Pow(2, MipIndex + 1), TextureSource->GetSizeZ() / FMath::Pow(2, MipIndex + 1)); RHICmdList.Transition(FRHITransitionInfo(TextureRenderTargetSource->TextureRHI, ERHIAccess::Unknown, ERHIAccess::CopyDest)); RHICmdList.Transition(FRHITransitionInfo(TextureSource->TextureRHI, ERHIAccess::SRVMask, ERHIAccess::CopySrc)); FRHITexture* Test = TextureSource->TextureRHI.GetReference(); RHICmdList.CopyTexture(TextureSource->TextureRHI, TextureRenderTargetSource->TextureRHI, CopyInfo); RHICmdList.Transition(FRHITransitionInfo(TextureRenderTargetSource->TextureRHI, ERHIAccess::CopyDest, ERHIAccess::SRVMask)); RHICmdList.Transition(FRHITransitionInfo(TextureRenderTargetSource->TextureRHI, ERHIAccess::CopySrc, ERHIAccess::SRVMask)); } } else { FRHICopyTextureInfo CopyInfo; CopyInfo.SourcePosition = FIntVector(0, 0, 0); CopyInfo.DestPosition = FIntVector(0, 0, 0); CopyInfo.Size = FIntVector(TextureSource->GetSizeX() / 2, TextureSource->GetSizeY() / 2, TextureSource->GetSizeZ() / 2); RHICmdList.Transition(FRHITransitionInfo(TextureRenderTargetSource->TextureRHI, ERHIAccess::Unknown, ERHIAccess::CopyDest)); RHICmdList.Transition(FRHITransitionInfo(TextureSource->TextureRHI, ERHIAccess::SRVMask, ERHIAccess::CopySrc)); RHICmdList.CopyTexture(TextureSource->TextureRHI, TextureRenderTargetSource->TextureRHI, CopyInfo); RHICmdList.Transition(FRHITransitionInfo(TextureRenderTargetSource->TextureRHI, ERHIAccess::CopyDest, ERHIAccess::SRVMask)); RHICmdList.Transition(FRHITransitionInfo(TextureRenderTargetSource->TextureRHI, ERHIAccess::CopySrc, ERHIAccess::SRVMask)); } } ); FlushRenderingCommands(); } void UTestCopyRenderFunctionLibrary::CopyTextureOfTheSlice( const UObject* WorldContextObject, const UTexture2D* InTexture, class UTexture2DArray* OutputTextureArray, int32 TargetSliceIndex) { UWorld* World = WorldContextObject->GetWorld(); ERHIFeatureLevel::Type FeatureLevel = World->Scene->GetFeatureLevel(); if (FeatureLevel < ERHIFeatureLevel::SM5) { UE_LOG(LogTemp, Warning, TEXT("FeatureLevel < ERHIFeatureLevel::SM5")); return; } if (nullptr == InTexture) { UE_LOG(LogTemp, Warning, TEXT("InTexture Is NULL")); return; } FTexture* SourceTexture = InTexture->Resource; FTexture* TargetTexture = OutputTextureArray->Resource; ENQUEUE_RENDER_COMMAND(TestShaderCommand)( [SourceTexture, TargetTexture, FeatureLevel, TargetSliceIndex](FRHICommandListImmediate& RHICmdList) { FRHICopyTextureInfo CopyInfo; CopyInfo.SourcePosition = FIntVector(0, 0, 0); CopyInfo.DestPosition = FIntVector(0, 0, 0); CopyInfo.Size = FIntVector(SourceTexture->GetSizeX(), SourceTexture->GetSizeY(), SourceTexture->GetSizeZ()); CopyInfo.SourceSliceIndex = 0; CopyInfo.DestSliceIndex = TargetSliceIndex; RHICmdList.Transition(FRHITransitionInfo(TargetTexture->TextureRHI, ERHIAccess::Unknown, ERHIAccess::CopyDest)); RHIC