You've already forked UnrealEngineUWP
mirror of
https://github.com/izzy2lost/UnrealEngineUWP.git
synced 2026-03-26 18:15:20 -07:00
910404ad86
1. Fix header parsing to check for actual header termination (blank line) 2. Header lines are only ever terminated by LF, never CR. Treat NUL as error. 3. The only supported pixel layout is -Y <Height> +X (the default). Require that explicitly. 4. Require that widths and heights be positive and enforce a maximum image size 5. Fix several potential integer overflows 6. Fix several potential out-of-bounds reads/writes 7. Don't cast possibly unaligned uint8 pointers to FColor* 8. Fix potential UB when shift amount in old RLE encoding gets large 9. Don't allow runs as first thing in an old-RLE scan line, since they want to repeat the prev pixel 10. Factor out a lot of the error handling code, make sure error messages get logged 11. Fix several typos while I'm here #jira UE-166514 #jira UE-166515 #jira UE-166516 #jira UE-166551 #preflight 6345d5bef93be0f634181cb9 [CL 22479168 by fabian giesen in ue5-main branch]
81 lines
3.0 KiB
C++
81 lines
3.0 KiB
C++
// Copyright Epic Games, Inc. All Rights Reserved.
|
|
|
|
#pragma once
|
|
|
|
#include "Containers/ContainersFwd.h"
|
|
#include "CoreMinimal.h"
|
|
#include "CoreTypes.h"
|
|
#include "IImageWrapper.h"
|
|
#include "ImageCore.h"
|
|
#include "Internationalization/Text.h"
|
|
|
|
// http://radsite.lbl.gov/radiance/refer/Notes/picture_format.html
|
|
// http://paulbourke.net/dataformats/pic/
|
|
|
|
/** To load the HDR file image format. Does not support all possible types HDR formats (e.g. xyze is not supported) */
|
|
// does not use ImageWrapperBase , unlike all other imagewrappers
|
|
class IMAGEWRAPPER_API FHdrImageWrapper : public IImageWrapper
|
|
{
|
|
public:
|
|
|
|
// Todo we should have this for all image wrapper.
|
|
bool SetCompressedFromView(TArrayView64<const uint8> Data);
|
|
|
|
// IIMageWrapper Interface begin
|
|
virtual bool SetCompressed(const void* InCompressedData, int64 InCompressedSize) override;
|
|
virtual bool SetRaw(const void* InRawData, int64 InRawSize, const int32 InWidth, const int32 InHeight, const ERGBFormat InFormat, const int32 InBitDepth, const int32 InBytesPerRow) override;
|
|
virtual TArray64<uint8> GetCompressed(int32 Quality = 0) override;
|
|
virtual bool GetRaw(const ERGBFormat InFormat, int32 InBitDepth, TArray64<uint8>& OutRawData) override;
|
|
|
|
virtual bool CanSetRawFormat(const ERGBFormat InFormat, const int32 InBitDepth) const override;
|
|
virtual ERawImageFormat::Type GetSupportedRawFormat(const ERawImageFormat::Type InFormat) const override;
|
|
|
|
virtual int32 GetWidth() const override;
|
|
virtual int32 GetHeight() const override;
|
|
virtual int32 GetBitDepth() const override;
|
|
virtual ERGBFormat GetFormat() const override;
|
|
|
|
// IImageWrapper Interface end
|
|
|
|
// GetErrorMessage : nice idea, but not virtual, never called by the standard import path
|
|
const FText& GetErrorMessage() const;
|
|
|
|
void FreeCompressedData();
|
|
|
|
using IImageWrapper::GetRaw;
|
|
private:
|
|
// Helpers for error exits. Set error messages and return false.
|
|
void SetAndLogError(const FText& InText);
|
|
bool FailHeaderParsing(); // also calls FreeCompressData.
|
|
bool FailUnexpectedEOB();
|
|
bool FailMalformedScanline();
|
|
|
|
bool GetHeaderLine(const uint8*& BufferPos, char Line[256]);
|
|
|
|
static bool ParseMatchString(const char*& InOutCursor, const char* InExpected);
|
|
static bool ParsePositiveInt(const char*& InOutCursor, int* OutValue);
|
|
static bool ParseImageSize(const char* InLine, int* OutWidth, int* OutHeight);
|
|
|
|
static bool HaveBytes(const uint8* InCursor, const uint8* InEnd, int InAmount);
|
|
|
|
/** @param Out order in bytes: RGBE */
|
|
bool DecompressScanline(uint8* Out, const uint8*& In, const uint8* InEnd);
|
|
|
|
bool OldDecompressScanline(uint8* Out, const uint8*& In, const uint8* InEnd, int32 Length, bool bInitialRunAllowed);
|
|
|
|
bool IsCompressedImageValid() const;
|
|
|
|
TArrayView64<const uint8> CompressedData;
|
|
const uint8* RGBDataStart = nullptr;
|
|
|
|
TArray64<uint8> CompressedDataHolder;
|
|
TArray64<uint8> RawDataHolder;
|
|
|
|
/** INDEX_NONE if not valid */
|
|
int32 Width = INDEX_NONE;
|
|
/** INDEX_NONE if not valid */
|
|
int32 Height = INDEX_NONE;
|
|
|
|
// Reported error
|
|
FText ErrorMessage;
|
|
}; |