dialog.hpp

#pragma once
#include <cstdint>
#include "util.hpp"
 
// pre-declare the class, since a pointer to it is placed in the vtable
class GUIDialog;
 
struct GUIDialog_Wrapped_VTable {
    GUIDialog *me;
    uint32_t fakeentrypadding[2];
 
    VTABLE_FAKE_ENTRY(1, 0);
    
    // Args: event
    // Return: unknown?
    VTableFunction<int, struct GUIDialog_OnEvent_Data *> OnEvent;
 
    VTABLE_FAKE_ENTRY(1, 1);
 
    // Args: element, unknown0
    // unknown0 - always pass 0
    VTableFunction<void, void *, int> AddElement;
 
    VTABLE_FAKE_ENTRY(4, 2);
 
    VTableFunction<void> Refresh;
 
    VTABLE_FAKE_ENTRY(23, 3);
 
    // Return: dialog result
    VTableFunction<int> ShowDialog;
 
    VTABLE_FAKE_ENTRY(20, 4);
};
 
struct GUIDialog_Wrapped {
    uint8_t unknown0[0x10];
 
    // refer to accessors in GUIDialog class for documentation
    uint16_t leftX;
    uint16_t topY;
    uint16_t rightX;
    uint16_t bottomY;
 
    uint8_t unknown1[0x34];
 
    struct GUIDialog_Wrapped_VTable *vtable;
 
    uint8_t unknown2[0x58];
};
#ifndef __clang__
static_assert(sizeof(struct GUIDialog_Wrapped) == 0xA8);
#endif
 
struct GUIDialog_OnEvent_Data {
    uint16_t type;
 
    uint16_t data;
 
    void *element;
 
    constexpr uint16_t GetEventID() {
        return (type >> 4) - 8;
    }
};
 
class GUIDialog : public Wrapped {
public:
    enum Height : int {
        Height25 = 0,
        Height55 = 1,
        Height75 = 2,
        Height95 = 3,
        Height35 = 4,
        Height60 = 5
    };
 
    enum Alignment : int {
        AlignTop = 0,
        AlignCenter = 1,
        AlignBottom = 2
    };
 
    enum KeyboardState : int {
        KeyboardStateNone = 0, // 2 gives same effect
        KeyboardStateMath1 = 1, // 3 gives same effect
        KeyboardStateMath2 = 4,
        KeyboardStateMath3 = 5,
        KeyboardStateTrig = 6,
        KeyboardStateVar = 7,
        KeyboardStateABC = 8,
        KeyboardStateCatalog = 9,
        KeyboardStateAdvance = 10,
        KeyboardStateNumber = 11
    };
 
    enum DialogResult : int {
        DialogResultOK = 0x3EA,
        DialogResultCancel = 0x3EB
    };
 
    GUIDialog(
        enum Height height, enum Alignment alignment,
        const char* title,
        enum KeyboardState keyboard
    );
 
    virtual int OnEvent(struct GUIDialog_Wrapped *dialog, struct GUIDialog_OnEvent_Data *event);
    
    uint16_t GetLeftX();
    uint16_t GetTopY();
    uint16_t GetRightX();
    uint16_t GetBottomY();
 
    void AddElement(GUIElement &element);
    void Refresh();
    DialogResult ShowDialog();
 
private:
    struct GUIDialog_Wrapped_VTable *m_oldVTable;
    struct GUIDialog_Wrapped_VTable m_vtable;
 
    static int OnEvent_Wrap(struct GUIDialog_Wrapped *dialog, struct GUIDialog_OnEvent_Data *event);
};