#ifndef _JIT_H
#define _JIT_H

#include "types.h"
#include "dmg.h"

#define HALT_SENTINEL 0xffffffff

// runtime context for JIT - TODO remove all the padding from fields i removed
typedef struct {
    /*  0 */ void *dmg;
    /*  4 */ void *read_func;
    /*  8 */ void *write_func;
    /*  c */ void *ei_di_func;
    /* 10 */ u8 trace_enabled;           // if set, dispatcher always returns to C
    /* 11 */ u8 current_rom_bank;
    /* 12 */ u8 _pad[2];
    /* 14 */ void **bank0_cache;
    /* 18 */ void ***banked_cache;
    /* 1c */ void **upper_cache;
    /* 20 */ void *dispatcher_return;
    /* 24 */ void *read16_func;
    /* 28 */ void *write16_func;
    /* 2c */ u32 cycles_accumulated;     // GB cycles accumulated by compiled code
    /* 30 */ void *patch_helper;         // patch_helper routine for lazy block patching
    /* 34 */ u32 read_cycles;            // GB cycles at time of dmg_read call
    /* 38 */ u32 _pad2;
    /* 3c */ u32 *frame_cycles_ptr;      // pointer to dmg->frame_cycles for HALT
    /* 40 */ void *stop_func;            // STOP handler (for CGB speed switch)
    /* 44 */ u8 effective_double_speed;  // 1 when cgb double_speed && !ignore_double_speed
    /* 45 */ u8 _pad4[3];
    /* 48 */ u16 gb_sp;
    /* 4a */ u16 _pad3;
    /* 4c */ long stack_in_ram;          // non-zero if A3 points to native WRAM/HRAM
    /* 50 */ u32 wake_limit;             // CPU cycles from the last sync to the
                                         // earliest deadline
    /* 54 */ u32 skipped_cycles;         // GB6_PROFILING: cycles fast-forwarded
    /* 58 */ u32 ly_clamp_skips;         // GB6_PROFILING: LY-wait clamp skip count
    /* 5c */ void *mbc_write_func;       // per-type mbcN_write, for constant
                                         // ROM-range write sites
} jit_context;

extern jit_context jit_ctx;
extern int jit_halted;

void jit_init(struct dmg *dmg);

int jit_run(struct dmg *dmg);

int jit_clear_all_blocks(void);

// batch compilation of a saved block list (blocklist.c)
int jit_precompile(u8 bank, u16 pc);
void jit_precompile_finish(void);

void jit_cleanup(void);

#endif