CheatMenuSA/src/ui.h

#pragma once
#include "pch.h"

// TODO: Fix this messy code
class Ui
{
public:
    struct NamedMemory
    {
        std::string name;
        int addr;
    };

    struct NamedValue
    {
        std::string name;
        int value;
    };

    Ui() = delete;
    Ui(Ui&) = delete;

    static bool ColorButton(int color_id, std::vector<float>& color, ImVec2 size);
    static bool CheckboxAddress(const char* label, int addr = NULL, const char* hint = nullptr);
    static bool CheckboxAddressEx(const char* label, int addr = NULL, int enabled_val = 1, int disabled_val = 0,
                                  const char* hint = nullptr);
    static bool CheckboxBitFlag(const char* label, uint flag, const char* hint = nullptr);
    static bool CheckboxWithHint(const char* label, bool* state, const char* hint = nullptr, bool is_disabled = false);

    static bool DrawTitleBar();
    template <typename T>
    static void EditAddress(const char* label, int address, int min = 0, int def = 0, int max = 100);
    static void EditBits(const char* label, int address, const std::vector<std::string>& names);
    static void EditFloat(const char* label, int address, float min, float def, float max, float mul = 1, float change = 1.0f);
    template <typename T>
    static void EditReference(const char* label, T& address, int min = 0, int def = 0, int max = 100);
    static void EditRadioButtonAddress(const char* label, std::vector<NamedMemory>& named_mem);
    static void EditRadioButtonAddressEx(const char* label, int addr, std::vector<NamedValue>& named_val);

#ifdef GTASA
    static void EditStat(const char* label, int stat_id, int min = 0, int def = 0, int max = 1000);
#endif


    static ImVec2 GetSize(short count = 1, bool spacing = true);

    static bool ListBox(const char* label, const std::vector<std::string>& all_items, int& selected);
    static bool ListBoxStr(const char* label, const std::vector<std::string>& all_items, std::string& selected);

    static void RadioButtonAddress(const char* label, std::vector<NamedMemory>& named_mem);
    static void RadioButtonAddressEx(const char* label, int addr, std::vector<NamedValue>& named_val);
    static void ColorPickerAddress(const char* label, int base_addr, ImVec4&& default_color);
};

template <typename T>
void Ui::EditAddress(const char* label, const int address, const int min, const int def, const int max)
{
    if (ImGui::CollapsingHeader(label))
    {
        int val = patch::Get<T>(address, false);

        int items = 3;

        if (min == def)
        {
            items = 2;
        }

        ImGui::Columns(items, nullptr, false);
        ImGui::Text(("Min: " + std::to_string(min)).c_str());

        if (items == 3)
        {
            ImGui::NextColumn();
            ImGui::Text(("Def: " + std::to_string(def)).c_str());
        }

        ImGui::NextColumn();
        ImGui::Text(("Max: " + std::to_string(max)).c_str());
        ImGui::Columns(1);

        ImGui::Spacing();

        if (ImGui::InputInt(("Set value##" + std::string(label)).c_str(), &val))
        {
            patch::Set<T>(address, val, false);
        }

        ImGui::Spacing();

        if (val < min)
        {
            val = min;
        }

        if (val > max)
        {
            val = max;
        }

        if (ImGui::Button(("Minimum##" + std::string(label)).c_str(), GetSize(items)))
        {
            patch::Set<T>(address, min, false);
        }

        if (items == 3)
        {
            ImGui::SameLine();

            if (ImGui::Button(("Default##" + std::string(label)).c_str(), GetSize(3)))
            {
                patch::Set<T>(address, def, false);
            }
        }

        ImGui::SameLine();

        if (ImGui::Button(("Maximum##" + std::string(label)).c_str(), GetSize(items)))
        {
            patch::Set<T>(address, max, false);
        }

        ImGui::Spacing();
        ImGui::Separator();
    }
}

template <typename T>
void Ui::EditReference(const char* label, T& address, const int min, const int def, const int max)
{
    if (ImGui::CollapsingHeader(label))
    {
        int val = static_cast<int>(address);

        ImGui::Columns(3, nullptr, false);
        ImGui::Text(("Min: " + std::to_string(min)).c_str());
        ImGui::NextColumn();
        ImGui::Text(("Def: " + std::to_string(def)).c_str());
        ImGui::NextColumn();
        ImGui::Text(("Max: " + std::to_string(max)).c_str());
        ImGui::Columns(1);

        ImGui::Spacing();

        if (ImGui::InputInt(("Set value##" + std::string(label)).c_str(), &val))
        {
            address = static_cast<float>(val);
        }

        ImGui::Spacing();

        if (ImGui::Button(("Minimum##" + std::string(label)).c_str(), GetSize(3)))
        {
            address = static_cast<float>(min);
        }

        ImGui::SameLine();

        if (ImGui::Button(("Default##" + std::string(label)).c_str(), GetSize(3)))
        {
            address = static_cast<float>(def);
        }

        ImGui::SameLine();

        if (ImGui::Button(("Maximum##" + std::string(label)).c_str(), GetSize(3)))
        {
            address = static_cast<float>(max);
        }

        ImGui::Spacing();
        ImGui::Separator();
    }
}