MassBuilderSaveTool/src/UESaveFile/Serialisers/EnumPropertySerialiser.cpp

// MassBuilderSaveTool
// Copyright (C) 2021-2022 Guillaume Jacquemin
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program.  If not, see <https://www.gnu.org/licenses/>.

#include "../BinaryReader.h"
#include "../BinaryWriter.h"

#include "EnumPropertySerialiser.h"

auto EnumPropertySerialiser::types() -> Containers::ArrayView<const std::string> {
    static const Containers::Array<std::string> types{InPlaceInit, {"EnumProperty"}};
    return types;
}

auto EnumPropertySerialiser::deserialise(const std::string& name, const std::string& type, UnsignedLong value_length,
                                         BinaryReader& reader, PropertySerialiser& serialiser) -> UnrealPropertyBase::ptr
{
    auto prop = Containers::pointer<EnumProperty>();

    if(!reader.readUEString(prop->enumType)) {
        return nullptr;
    }

    char terminator;
    if(!reader.readChar(terminator) || terminator != '\0') {
        return nullptr;
    }

    if(!reader.readUEString(prop->value)) {
        return nullptr;
    }

    return prop;
}

auto EnumPropertySerialiser::serialise(UnrealPropertyBase::ptr& prop, UnsignedLong& bytes_written,
                                       BinaryWriter& writer, PropertySerialiser& serialiser) -> bool
{
    auto enum_prop = dynamic_cast<EnumProperty*>(prop.get());

    if(!enum_prop) {
        return false;
    }

    writer.writeUEStringToArray(enum_prop->enumType);
    writer.writeValueToArray<char>('\0');
    bytes_written += writer.writeUEStringToArray(enum_prop->value);

    return true;
}