import pyiforce as ifr
import serial
import pyvjoy
import pyvjoy_ffb as ffb

PORT = 'COM7'
BAUD = 38400


def send_pkt(port, command, data):
    pkt = bytes(ifr.tx_packet(command, data))
    port.write(pkt)

def hex_print(d):
    print(' '.join('%02x'%x for x in d))

def query_print(d):
    print('QUERY[%02x]:%s'%(d['query'],' '.join('%02x'%x for x in d['data'])))

def setup_wheel(port):
    ifr.set_handler(ifr.Packet.QUERY, query_print)
    send_pkt(port, ifr.Packet.QUERY, [ifr.Query.Manufacturer])
    ifr.rx_pump()
    send_pkt(port, ifr.Packet.QUERY, [ifr.Query.Product])
    ifr.rx_pump()
    send_pkt(port, ifr.Packet.QUERY, [ifr.Query.Version])
    ifr.rx_pump()
    send_pkt(port, ifr.Packet.QUERY, [ifr.Query.NumberOfEffects])
    ifr.rx_pump()
    send_pkt(port, ifr.Packet.QUERY, [ifr.Query.BufferSize])
    ifr.rx_pump()

    # send_pkt(port, ifr.Packet.SET_OVERALL, [ifr.OverallControlType.GAIN, 0x80])
    send_pkt(port, ifr.Packet.QUERY, [ifr.Query.Open])


def serial_reader(sp):
    while sp.is_open:
        in_byte = sp.read(1)[0]
        # print(f'rx:%02X'%in_byte)
        yield in_byte


with serial.Serial(port=PORT, baudrate=BAUD) as port:

    port.rts = True
    ifr.create_rx_stream(serial_reader(port))

    setup_wheel(port)

    j = pyvjoy.VJoyDevice(1)
    ffb.add_ffb(j)


    # def handle_wheel(w):
    #     j.data.wAxisX = int(0x4000 * (w['axis']['wheel'] + 1))
    #     j.data.wAxisY = int(0x8000 * w['axis']['gas'])
    #     j.data.wAxisZ = int(0x8000 * w['axis']['brake'])
    #     j.data.lButtons = (
    #         0x0000000000000001 if w['buttons']['A'] else 0) + (
    #         0x0000000000000002 if w['buttons']['B'] else 0) + (
    #         0x0000000000000004 if w['buttons']['1_PaddleUp_Start'] else 0) + (
    #         0x0000000000000008 if w['buttons']['2_PaddleDown'] else 0) + (
    #         0x0000000000000010 if w['buttons']['3'] else 0) + (
    #         0x0000000000000020 if w['buttons']['4'] else 0)
    #     j.data.bHats = (
    #         -1 if w['hat']['center'] else
    #         0 if w['hat']['up'] else
    #         1 if w['hat']['right'] else
    #         2 if w['hat']['down'] else
    #         3 if w['hat']['left'] else -1)
    #     j.update()

    # ifr.set_handler(ifr.Packet.INPUT_WHEEL, handle_wheel)
    # ifr.set_handler(ifr.Packet.INPUT_WHEEL, print)

    while True:
        ifr.rx_pump()