BlockStructure.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
#pragma once
 
#include "CoreMinimal.h"
#include "Core/BlockRange.h"
 
namespace BuildPatchServices
{
    struct FBlockEntry
    {
        friend class FBlockStructure;
    public:
        FBlockEntry(uint64 InOffset, uint64 InSize);
        ~FBlockEntry();
 
        uint64 GetOffset() const;
        uint64 GetSize() const;
        const FBlockEntry* GetNext() const;
        const FBlockEntry* GetPrevious() const;
        FBlockRange AsRange() const { return FBlockRange::FromFirstAndSize(Offset, Size); }
 
    private:
        void InsertBefore(FBlockEntry* NewEntry, FBlockEntry** Head);
        void InsertAfter(FBlockEntry* NewEntry, FBlockEntry** Tail);
        void Unlink(FBlockEntry** Head, FBlockEntry** Tail);
        void Merge(uint64 InOffset, uint64 InSize);
        void Chop(uint64 InOffset, uint64 InSize, FBlockEntry** Head, FBlockEntry** Tail);
 
    private:
        uint64 Offset;
        uint64 Size;
        FBlockEntry* Prev;
        FBlockEntry* Next;
    };
 
    class FBlockStructure
    {
    public:
        FBlockStructure();
        FBlockStructure(uint64 Offset, uint64 Size);
        FBlockStructure(const FBlockStructure&);
        FBlockStructure(FBlockStructure&&);
        ~FBlockStructure();
        FBlockStructure& operator=(const FBlockStructure&);
        FBlockStructure& operator=(FBlockStructure&&);
 
        const FBlockEntry* GetHead() const;
        const FBlockEntry* GetTail() const;
 
        void Empty();
 
        void Add(uint64 Offset, uint64 Size, ESearchDir::Type SearchDir = ESearchDir::FromStart);
 
        void Add(const FBlockRange& BlockRange, ESearchDir::Type SearchDir = ESearchDir::FromStart);
 
        void Add(const FBlockStructure& OtherStructure, ESearchDir::Type SearchDir = ESearchDir::FromStart);
 
        void Remove(uint64 Offset, uint64 Size, ESearchDir::Type SearchDir = ESearchDir::FromStart);
 
        void Remove(const FBlockRange& BlockRange, ESearchDir::Type SearchDir = ESearchDir::FromStart);
 
        void Remove(const FBlockStructure& OtherStructure, ESearchDir::Type SearchDir = ESearchDir::FromStart);
 
        uint64 SelectSerialBytes(uint64 FirstByte, uint64 Count, FBlockStructure& OutputStructure) const;
 
        FBlockStructure Intersect(const FBlockStructure& OtherStructure) const;
 
        FString ToString(uint64 BlockCountLimit = 20) const;
 
    private:
        FBlockEntry* Head;
        FBlockEntry* Tail;
        void CollectOverlaps(FBlockEntry* First, ESearchDir::Type SearchDir);
    };
 
    namespace BlockStructureHelpers
    {
        uint64 CountSize(const FBlockStructure& Structure);
        bool HasIntersection(const FBlockStructure& ByteStructure, const FBlockStructure& Intersection);
        FBlockStructure SerializeIntersection(const FBlockStructure& ByteStructure, const FBlockStructure& Intersection);
    }
}