#include "lolevel.h"
#include "platform.h"
#include "keyboard.h"
#include "kbd_common.h"

long kbd_new_state[3] = { 0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF };
long kbd_prev_state[3] = { 0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF };
long kbd_mod_state[3] = { 0xFFFFFFFF, 0xFFFFFFFF, 0xFFFFFFFF };

extern void _GetKbdState(long*);


int get_usb_bit() {
    long usb_physw[3];
    usb_physw[USB_IDX] = 0;
    _kbd_read_keys_r2(usb_physw);
    return(( usb_physw[USB_IDX] & USB_MASK)==USB_MASK) ;
}

int get_hdmi_hpd_bit() {
    long hpd_physw[3];
    hpd_physw[HDMI_HPD_IDX] = 0;
    _GetKbdState(hpd_physw);
    return( ((hpd_physw[HDMI_HPD_IDX] & HDMI_HPD_FLAG)==HDMI_HPD_FLAG)?0:1) ;
}

int get_analog_av_bit() {
    long av_physw[3];
    av_physw[ANALOG_AV_IDX] = 0;
    _GetKbdState(av_physw);
    return( ((av_physw[ANALOG_AV_IDX] & ANALOG_AV_FLAG)==ANALOG_AV_FLAG)?0:1) ;
}

KeyMap keymap[] = {
    { 0, KEY_SHOOT_FULL      ,0x00040001 }, // Found @0xfc613714, levent 0x01
    { 0, KEY_SHOOT_HALF      ,0x00040000 }, // Found @0xfc61378c, levent 0x00
    { 0, KEY_SHOOT_FULL_ONLY ,0x00000001 }, // Found @0xfc613714, levent 0x01
    { 0, KEY_ZOOM_OUT        ,0x00000002 }, // Found @0xfc61371c, levent 0x04
    { 0, KEY_ZOOM_IN         ,0x00000004 }, // Found @0xfc613724, levent 0x03
    { 0, KEY_VIDEO           ,0x00000008 }, // Found @0xfc61372c, levent 0x02
    { 0, KEY_MENU            ,0x00000010 }, // Found @0xfc613734, levent 0x14
    { 0, KEY_UP              ,0x00000020 }, // Found @0xfc61373c, levent 0x06
    { 0, KEY_DOWN            ,0x00000040 }, // Found @0xfc613744, levent 0x07
    { 0, KEY_RIGHT           ,0x00000080 }, // Found @0xfc61374c, levent 0x09
    { 0, KEY_LEFT            ,0x00000100 }, // Found @0xfc613754, levent 0x08
    { 0, KEY_SET             ,0x00000200 }, // Found @0xfc61375c, levent 0x0a
    { 0, KEY_ERASE           ,0x00000400 }, // ring func / erase, from watching  physw
    { 0, KEY_PLAYBACK        ,0x00000800 }, // Found @0xfc61376c, levent 0x101
    { 0, KEY_WIFI            ,0x00010000 }, // wifi button, from watching physw
//    { 0, KEY_POWER           ,0x00020000 }, // Found @0xfc613784, levent 0x100
    { 0, 0, 0 }
};
// ev comp and mode dial is in word 1, lower bits, encoding unclear
// flash open word 0, 0x00004000 = open
// no battery door switch, cam runs with door open, no physw change

long __attribute__((naked,noinline)) wrap_kbd_p1_f() {

    asm volatile(
        "push    {r1-r7, lr}\n"
        "movs    r4, #0\n"
        "bl      my_kbd_read_keys\n"
        "b       _kbd_p1_f_cont\n"
    );

    return 0;
}

// no stack manipulation needed here, since we create the task directly
void __attribute__((noinline)) mykbd_task() {
    extern void kbd_p2_f_my();

    while (physw_run) {
        _SleepTask(physw_sleep_delay);

        if (wrap_kbd_p1_f() == 1) {
            kbd_p2_f_my();                      // replacement of _kbd_p2_f (in sub/<fwver>/boot.c)
        }
    }

    _ExitTask();
}

// g7x jogdial hw counters (19 bits) are at 0xd9854004 and 0xd9855004, use fw func to read + sign extend them
// 0x7fff8 .. 0x7fffc .. 0 (start pos) .. 4
// intermediate positions are also available, but they are ignored by the fw for a good reason
extern int _get_dial_hw_position(int dial);
#define DIAL_HW_REAR  4
#define DIAL_HW_FRONT 5
int get_dial_hw_position(int dial)
{
    // mask low bits
    return _get_dial_hw_position(dial)&~3;
}
int jogdial_stopped=0;

extern long dial_positions[4];

long get_jogdial_direction(void) {
    static int new_jogdial=0, old_jogdial=0, new_frontdial=0, old_frontdial=0;
    
    old_jogdial=new_jogdial;
    new_jogdial=get_dial_hw_position(DIAL_HW_REAR);

    old_frontdial=new_frontdial;
    new_frontdial=get_dial_hw_position(DIAL_HW_FRONT);

    if (old_jogdial>new_jogdial) return JOGDIAL_LEFT; 
    else if (old_jogdial<new_jogdial) return JOGDIAL_RIGHT;
    if (old_frontdial<new_frontdial) return FRONTDIAL_LEFT;  // counter clockwise with camera aimed away from you
    else if (old_frontdial>new_frontdial) return FRONTDIAL_RIGHT;
    else return 0;
}

int handle_jogdial() {
    // return 0 to prevent fw dial handler
    if (jogdial_stopped) {
        // update positions in RAM
        dial_positions[0] = dial_positions[2] = get_dial_hw_position(DIAL_HW_REAR);
        dial_positions[1] = dial_positions[3] = get_dial_hw_position(DIAL_HW_FRONT);
        return 0;
    }
    return 1;
}

void jogdial_control(int c) {
    jogdial_stopped = c;
}

void my_kbd_read_keys() {
    kbd_update_key_state();
    kbd_update_physw_bits();
}

void kbd_fetch_data(long *dst)
{
    _GetKbdState(dst);
    _kbd_read_keys_r2(dst);
}