首页 > 代码库 > ue4 C++ 导入图片

ue4 C++ 导入图片

2024-11-21 06:49:02   203人阅读

void SDrawHouseTypeTools::OnButtonClickLoadImage()
{
       // 当前如果是在硬装模块，则可进行导入户型操作
       FHardEdMode* HardEdMode = (FHardEdMode*)GLevelEditorModeTools().GetActiveMode(FBuiltinEditorModes::EM_HardEdModeId);
       if (HardEdMode)
       {
              //打开文件夹
              TArray<FString> OpenFilenames;
              TSharedPtr<SWindow> ParentWindow = FSlateApplication::Get().GetActiveTopLevelWindow();
              void* ParentWindowHandle = (ParentWindow.IsValid() && ParentWindow->GetNativeWindow().IsValid()) ? ParentWindow->GetNativeWindow()->GetOSWindowHandle() : nullptr;
              if (ExtendFunc::OpenFileDialog(
                     ParentWindowHandle,
                     NSLOCTEXT("UIHome", "OpenReferencePictureTitle", "打开户型图片文件...").ToString(),
                     FPaths::GameDir(),
                     TEXT(""),
                     NSLOCTEXT("UIHome", "OpenReferencePictureFilter", "All Files (*.*)|*.*|户型图片文件 (*.png)|*.png|户型图片文件 (*.jpg)|*.jpg").ToString(),
                     ExtendFunc::EFileDialogFlags::None,
                     OpenFilenames))
              {
                     if (OpenFilenames.Num() > 0)
                     {
                           FString OpenFilename = OpenFilenames[0];
                           FName BrushName(*OpenFilename);
                           //传了图片路径，返回可以用来显示在runtime中的图片纹理
                           if (SpriteTexture.IsValid() && SpriteTexture->IsRooted())
                           {
                                  SpriteTexture->RemoveFromRoot();
                           }
                           // 判断图片得格式是否正确
                           SpriteTexture = TWeakObjectPtr<UTexture2D>(GetTexture2DFromDiskFile(OpenFilename));
                           if (SpriteTexture.IsValid())
                           {
                                  SpriteTexture->AddToRoot();
                                  //将数据存储并导入到画图方法中
                                  FIntPoint ImageSize = FIntPoint(SpriteTexture->GetSizeX(), SpriteTexture->GetSizeY());
                                  IHardModeModule::Get().GetImageSize() = ImageSize;
                                  TextureAlpha = 0.5;
                                  HardEdMode->OnHouseImgAlphaChanged(TextureAlpha);
                                  HardEdMode->SetHouseImg(SpriteTexture.Get(), ImageSize);
                           }
                           else
                           {
                                  FMessageDialog::Open(EAppMsgType::Ok, FText::FromString(TEXT("请导入正确格式的图片！")));
                                  OnButtonClickLoadImage();
                           }
                     }
              }
       }
       else
       {
              FMessageDialog::Open(EAppMsgType::Ok, FText::FromString(TEXT(" ")));
       }
}
//获取图片纹理
UTexture2D* SDrawHouseTypeTools::GetTexture2DFromDiskFile(const FString& FilePath)
{
       // 如果找到该文件，则返回true，否则为false
       if (!FPaths::FileExists(FilePath))
       {
              UE_LOG(LogTemp, Error, TEXT("File not found: %s"), *FilePath);
              return nullptr;
       }
       // 从文件中加载压缩的字节数据
       TArray<uint8> FileData;
       if (!FFileHelper::LoadFileToArray(FileData, *FilePath))
       {
              UE_LOG(LogTemp, Error, TEXT("Failed to load file: %s"), *FilePath);
              return nullptr;
       }
       // 使用ImageWrapper模块检测图像类型
       IImageWrapperModule& ImageWrapperModule = FModuleManager::LoadModuleChecked<IImageWrapperModule>(TEXT("ImageWrapper"));
       EImageFormat::Type ImageFormat = ImageWrapperModule.DetectImageFormat(FileData.GetData(), FileData.Num());
       if (ImageFormat == EImageFormat::Invalid)
       {
              UE_LOG(LogTemp, Error, TEXT("Unrecognized image file format: %s"), *FilePath);
              return nullptr;
       }
       // 为检测到的图像格式创建一个图像包装
       IImageWrapperPtr ImageWrapper = ImageWrapperModule.CreateImageWrapper(ImageFormat);
       if (!ImageWrapper.IsValid())
       {
              UE_LOG(LogTemp, Error, TEXT("Failed to create image wrapper for file: %s"), *FilePath);
              return nullptr;
       }
       // 解压图像数据
       const TArray<uint8>* RawData = nullptr;
       ImageWrapper->SetCompressed(FileData.GetData(), FileData.Num());
       ImageWrapper->GetRaw(ERGBFormat::BGRA, 8, RawData);
       if (RawData == nullptr)
       {
              UE_LOG(LogTemp, Error, TEXT("Failed to decompress image file: %s"), *FilePath);
              return nullptr;
       }
       // 创建纹理并上传未压缩的图像数据
       FString TextureBaseName = TEXT("Texture_") + FPaths::GetBaseFilename(FilePath);
       // UObject* Outer = nullptr;
       UTexture2D* mytexture = MyCreateTexture(*RawData, ImageWrapper->GetWidth(), ImageWrapper->GetHeight(), EPixelFormat::PF_B8G8R8A8, FName(*TextureBaseName));
       
       return mytexture;
}
UTexture2D* SDrawHouseTypeTools::MyCreateTexture(const TArray<uint8>& PixelData, int32 InSizeX, int32 InSizeY, EPixelFormat InFormat, FName BaseName)
{
       // 无耻地抄袭UTexture2D:CreateTransient一些修改
       if (InSizeX <= 0 || InSizeY <= 0 || (InSizeX % GPixelFormats[InFormat].BlockSizeX) != 0 || (InSizeY % GPixelFormats[InFormat].BlockSizeY) != 0)
       {
              return nullptr;
       }
       // 最重要的不同之处在于UTexture2D::CreateTransient:我们为新结构提供一个名称和一个所有者
       UTexture2D* NewTexture = NewObject<UTexture2D>(GetTransientPackage(), NAME_None, RF_Transient);
       NewTexture->PlatformData = new FTexturePlatformData();
       NewTexture->PlatformData->SizeX = InSizeX;
       NewTexture->PlatformData->SizeY = InSizeY;
       NewTexture->PlatformData->PixelFormat = InFormat;
       // 分配第一个纹理映射并上传像素数据
       int32 NumBlocksX = InSizeX / GPixelFormats[InFormat].BlockSizeX;
       int32 NumBlocksY = InSizeY / GPixelFormats[InFormat].BlockSizeY;
       FTexture2DMipMap* Mip = new(NewTexture->PlatformData->Mips) FTexture2DMipMap();
       Mip->SizeX = InSizeX;
       Mip->SizeY = InSizeY;
       Mip->BulkData.Lock(LOCK_READ_WRITE);
       void* TextureData = Mip->BulkData.Realloc(NumBlocksX * NumBlocksY * GPixelFormats[InFormat].BlockBytes);
       FMemory::Memcpy(TextureData, PixelData.GetData(), PixelData.Num());
       Mip->BulkData.Unlock();
       NewTexture->UpdateResource();
       return NewTexture;
}

ue4 C++ 导入图片


联系
我们