OpenRCT2/test/testpaint/Printer.cpp

#pragma region Copyright (c) 2014-2016 OpenRCT2 Developers
/*****************************************************************************
 * OpenRCT2, an open source clone of Roller Coaster Tycoon 2.
 *
 * OpenRCT2 is the work of many authors, a full list can be found in contributors.md
 * For more information, visit https://github.com/OpenRCT2/OpenRCT2
 *
 * OpenRCT2 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.
 *
 * A full copy of the GNU General Public License can be found in licence.txt
 *****************************************************************************/
#pragma endregion

#include "Printer.hpp"
#include "String.hpp"
#include "intercept.h"

namespace Printer {

    static const char *functionNames[] = {
        "sub_98196C",
        "sub_98197C",
        "sub_98198C",
        "sub_98199C",
        "metal_a_supports_paint_setup",
        "metal_b_supports_paint_setup",
        "wooden_a_supports_paint_setup",
        "wooden_b_supports_paint_setup",
    };

    static std::string GetImageIdString(uint32 imageId);

    static std::string GetHeightOffset(uint16 height, uint16 baseHeight);

    static std::string GetOffsetExpressionString(int offset);

    std::string PrintFunctionCalls(std::vector<function_call> calls, uint16 baseHeight) {
        std::string out;

        for (auto &&call : calls) {
            out += PrintFunctionCall(call, baseHeight).c_str();
            out += "\n";
        }

        return out;
    }

    std::string PrintFunctionCall(function_call call, uint16 baseHeight) {
        std::string imageId = GetImageIdString(call.supports.colour_flags);
        const char *functionName = functionNames[call.function];

        switch (call.function) {
            case SUPPORTS_WOOD_A:
            case SUPPORTS_WOOD_B:
                return String::Format(
                    "%s(%d, %d, %s, %s)", functionName, call.supports.type, call.supports.special,
                    GetHeightOffset(call.supports.height, baseHeight).c_str(), imageId.c_str()
                );

            case SUPPORTS_METAL_A:
            case SUPPORTS_METAL_B:
                return String::Format(
                    "%s(%d, %d, %d, %s, %s)", functionName, call.supports.type, call.supports.segment, call.supports.special,
                    GetHeightOffset(call.supports.height, baseHeight).c_str(), imageId.c_str()
                );

            case SET_SEGMENT_HEIGHT:
                return "paint_util_set_segment_support_height";
        }

        std::string s = String::Format("%s(", functionName);

        imageId = GetImageIdString(call.paint.image_id);
        s += String::Format("%s, ", imageId.c_str());
        s += String::Format("%d, %d, ", call.paint.offset.x, call.paint.offset.y);
        s += String::Format(
            "%d, %d, %d, ",
            call.paint.bound_box_length.x, call.paint.bound_box_length.y, call.paint.bound_box_length.z
        );
        s += String::Format("%s, ", GetHeightOffset(call.paint.z_offset, baseHeight).c_str());

        if (call.function != PAINT_98196C) {
            s += String::Format(
                "%d, %d, %s, ",
                call.paint.bound_box_offset.x, call.paint.bound_box_offset.y, GetHeightOffset(call.paint.bound_box_offset.z, baseHeight).c_str()
            );
        }

        s += String::Format("%d)", call.paint.rotation);


        if (call.function != PAINT_98196C) {
            s += String::Format(
                "    = { %d, %d, %s }, { %d, %d, %s }, { %d, %d, %d }",
                call.paint.offset.x, call.paint.offset.y, GetHeightOffset(call.paint.z_offset, baseHeight).c_str(),
                call.paint.bound_box_offset.x, call.paint.bound_box_offset.y,
                GetHeightOffset(call.paint.bound_box_offset.z, baseHeight).c_str(),
                call.paint.bound_box_length.x, call.paint.bound_box_length.y, call.paint.bound_box_length.z);
        }

        return s;
    }

    static std::string GetImageIdString(uint32 imageId)
    {
        std::string result;

        uint32 image = imageId & 0x7FFFF;
        uint32 palette = imageId & ~0x7FFFF;

        std::string paletteName;
        if (palette == Intercept2::DEFAULT_SCHEME_TRACK) paletteName = "SCHEME_TRACK";
        else if (palette == Intercept2::DEFAULT_SCHEME_SUPPORTS) paletteName = "SCHEME_SUPPORTS";
        else if (palette == Intercept2::DEFAULT_SCHEME_MISC) paletteName = "SCHEME_MISC";
        else if (palette == Intercept2::DEFAULT_SCHEME_3) paletteName = "SCHEME_3";
        else {
            paletteName = String::Format("0x%08X", palette);
        }

        if (image == 0) {
            result = paletteName;
        } else if (image & 0x70000) {
            result = String::Format("%s | vehicle.base_image_id + %d", paletteName.c_str(), image & ~0x70000);
        } else {
            result = String::Format("%s | %d", paletteName.c_str(), image);
        }

        return result;
    }

    static std::string GetHeightOffset(uint16 height, uint16 baseHeight) {
        int offset = height - baseHeight;

        return String::Format("height%s", GetOffsetExpressionString(offset).c_str());
    }

    static std::string GetOffsetExpressionString(int offset)
    {
        if (offset < 0) return std::string(" - ") + std::to_string(-offset);
        if (offset > 0) return std::string(" + ") + std::to_string(offset);
        return std::string();
    }
};