/*  RetroArch - A frontend for libretro.
 *  Copyright (C) 2013-2014 - Jason Fetters
 *
 *  RetroArch is free software: you can redistribute it and/or modify it under the terms
 *  of the GNU General Public License as published by the Free Software Found-
 *  ation, either version 3 of the License, or (at your option) any later version.
 *
 *  RetroArch is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
 *  without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
 *  PURPOSE.  See the GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License along with RetroArch.
 *  If not, see <http://www.gnu.org/licenses/>.
 */

#include <stdio.h>
#include <string.h>
#include <stdlib.h>

#include "../boolean.h"
#include "wiimote.h"

static void* hidpad_wii_connect(struct apple_pad_connection* connection, uint32_t slot)
{
   struct wiimote_t* device = (struct wiimote_t*)calloc(1, sizeof(struct wiimote_t));

   if (!device)
      return NULL;

   device->connection = connection;
   device->unid = slot;
   device->state = WIIMOTE_STATE_CONNECTED;
   device->exp.type = EXP_NONE;

   wiimote_handshake(device, -1, NULL, -1);

   return device;
}

static void hidpad_wii_disconnect(struct wiimote_t* device)
{
   if (device)
      free(device);
}

static int16_t hidpad_wii_get_axis(struct wiimote_t* device, unsigned axis)
{
   if (device->exp.type == EXP_CLASSIC)
   {
      switch (axis)
      {
         case 0:
            return device->exp.cc.classic.ljs.x.value * 0x7FFF;
         case 1:
            return device->exp.cc.classic.ljs.y.value * 0x7FFF;
         case 2:
            return device->exp.cc.classic.rjs.x.value * 0x7FFF;
         case 3:
            return device->exp.cc.classic.rjs.y.value * 0x7FFF;
         default:
            return 0;
      }
   }

   return 0;
}

static void hidpad_wii_packet_handler(struct wiimote_t* device,
      uint8_t *packet, uint16_t size)
{
   int i;
   byte* msg = packet + 2;

   switch (packet[1])
   {
      case WM_RPT_BTN:
         wiimote_pressed_buttons(device, msg);
         break;
      case WM_RPT_READ:
         wiimote_pressed_buttons(device, msg);
         wiimote_handshake(device, WM_RPT_READ, msg + 5,
               ((msg[2] & 0xF0) >> 4) + 1);
         break;
      case WM_RPT_CTRL_STATUS:
         wiimote_pressed_buttons(device, msg);
         wiimote_handshake(device,WM_RPT_CTRL_STATUS,msg,-1);
         break;
      case WM_RPT_BTN_EXP:
         wiimote_pressed_buttons(device, msg);
         wiimote_handle_expansion(device, msg+2);
         break;
   }

   g_current_input_data.pad_buttons[device->unid] = device->btns | 
      (device->exp.cc.classic.btns << 16);
   for (i = 0; i < 4; i++)
      g_current_input_data.pad_axis[device->unid][i] = hidpad_wii_get_axis(device, i);
}

static void hidpad_wii_set_rumble(struct wiimote_t* device, enum retro_rumble_effect effect, uint16_t strength)
{
   // TODO
}

struct apple_pad_interface apple_pad_wii =
{
   (void*)&hidpad_wii_connect,
   (void*)&hidpad_wii_disconnect,
   (void*)&hidpad_wii_packet_handler,
   (void*)&hidpad_wii_set_rumble
};