/*
Copyright (C) 1997-2001 Id Software, Inc.

This program 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 Foundation; either version 2
of the License, or (at your option) any later version.

This program 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 this program; if not, write to the Free Software
Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.

*/
// in_win.c -- windows 95 mouse and joystick code
// 02/21/97 JCB Added extended DirectInput code to support external controllers.

#include "client.h"
#include <GLFW/glfw3.h>

extern unsigned sys_msg_time;
extern GLFWwindow *glfw_window;

// joystick defines and variables
// where should defines be moved?
#define JOY_ABSOLUTE_AXIS 0x00000000 // control like a joystick
#define JOY_RELATIVE_AXIS 0x00000010 // control like a mouse, spinner, trackball
#define JOY_MAX_AXES 6 // X, Y, Z, R, U, V
#define JOY_AXIS_X 0
#define JOY_AXIS_Y 1
#define JOY_AXIS_Z 2
#define JOY_AXIS_R 3
#define JOY_AXIS_U 4
#define JOY_AXIS_V 5

enum _ControlList { AxisNada = 0, AxisForward, AxisLook, AxisSide, AxisTurn, AxisUp };

int dwAxisFlags[JOY_MAX_AXES];

int dwAxisMap[JOY_MAX_AXES];
int dwControlMap[JOY_MAX_AXES];
int pdwRawValue[JOY_MAX_AXES];

cvar_t *in_mouse;
cvar_t *in_joystick;

// none of these cvars are saved over a session
// this means that advanced controller configuration needs to be executed
// each time.  this avoids any problems with getting back to a default usage
// or when changing from one controller to another.  this way at least something
// works.
cvar_t *joy_name;
cvar_t *joy_advanced;
cvar_t *joy_advaxisx;
cvar_t *joy_advaxisy;
cvar_t *joy_advaxisz;
cvar_t *joy_advaxisr;
cvar_t *joy_advaxisu;
cvar_t *joy_advaxisv;
cvar_t *joy_forwardthreshold;
cvar_t *joy_sidethreshold;
cvar_t *joy_pitchthreshold;
cvar_t *joy_yawthreshold;
cvar_t *joy_forwardsensitivity;
cvar_t *joy_sidesensitivity;
cvar_t *joy_pitchsensitivity;
cvar_t *joy_yawsensitivity;
cvar_t *joy_upthreshold;
cvar_t *joy_upsensitivity;

bool joy_avail, joy_advancedinit, joy_haspov;
unsigned int joy_oldbuttonstate, joy_oldpovstate;

int joy_id;
unsigned int joy_flags;
int joy_numbuttons;

bool in_appactive;
extern bool ActiveApp;

// forward-referenced functions
void IN_StartupJoystick(void);
void IN_JoyMove(usercmd_t *cmd);

/*
============================================================

  MOUSE CONTROL

============================================================
*/

// mouse variables
cvar_t *m_filter;

bool mlooking;

void IN_MLookDown(void) { mlooking = true; }
void IN_MLookUp(void) {
  mlooking = false;
  if(!freelook->value && lookspring->value)
    IN_CenterView();
}

int mouse_buttons;
int mouse_oldbuttonstate;
int mouse_x, mouse_y, old_mouse_x, old_mouse_y, mx_accum, my_accum;

// int old_x, old_y;

bool mouseactive; // false when not focus app

bool restore_spi;
bool mouseinitialized;

// int window_center_x, window_center_y;
// RECT window_rect;

/*
===========
IN_ActivateMouse

Called when the window gains focus or changes in some way
===========
*/
void IN_ActivateMouse(void) {
  // int width, height;

  if(!mouseinitialized)
    return;
  if(!in_mouse->value) {
    mouseactive = false;
    return;
  }
  if(mouseactive)
    return;

  mouseactive = true;

  if(glfwRawMouseMotionSupported())
    glfwSetInputMode(glfw_window, GLFW_RAW_MOUSE_MOTION, GLFW_TRUE);

  glfwSetInputMode(glfw_window, GLFW_CURSOR, GLFW_CURSOR_DISABLED);
}

/*
===========
IN_DeactivateMouse

Called when the window loses focus
===========
*/
void IN_DeactivateMouse(void) {
  if(!mouseinitialized)
    return;
  if(!mouseactive)
    return;

  mouseactive = false;

  glfwSetInputMode(glfw_window, GLFW_CURSOR, GLFW_CURSOR_NORMAL);
}

/*
===========
IN_StartupMouse
===========
*/
void IN_StartupMouse(void) {
  cvar_t *cv;

  cv = Cvar_Get("in_initmouse", "1", CVAR_NOSET);
  if(!cv->value)
    return;

  mouseinitialized = true;
  mouse_buttons = 3;
}

/*
===========
IN_MouseMove
===========
*/
void IN_MouseMove(usercmd_t *cmd) {
  static double omx, omy;
  double mx, my, temp;

  if(!mouseactive)
    return;

  // find mouse movement
  glfwGetCursorPos(glfw_window, &mx, &my);

  temp = mx - omx;
  omx = mx;
  mx = temp;

  temp = my - omy;
  omy = my;
  my = temp;

  if(m_filter->value) {
    mouse_x = (mx + old_mouse_x) * 0.5;
    mouse_y = (my + old_mouse_y) * 0.5;
  } else {
    mouse_x = mx;
    mouse_y = my;
  }

  old_mouse_x = mx;
  old_mouse_y = my;

  mouse_x *= sensitivity->value;
  mouse_y *= sensitivity->value;

  // add mouse X/Y movement to cmd
  if((in_strafe.state & 1) || (lookstrafe->value && mlooking))
    cmd->sidemove += m_side->value * mouse_x;
  else
    cl.viewangles[YAW] -= m_yaw->value * mouse_x;

  if((mlooking || freelook->value) && !(in_strafe.state & 1)) {
    cl.viewangles[PITCH] += m_pitch->value * mouse_y;
  } else {
    cmd->forwardmove -= m_forward->value * mouse_y;
  }
}

/*
=========================================================================

VIEW CENTERING

=========================================================================
*/

cvar_t *v_centermove;
cvar_t *v_centerspeed;

/*
===========
IN_Init
===========
*/
void IN_Init(void) {
  // mouse variables
  m_filter = Cvar_Get("m_filter", "0", 0);
  in_mouse = Cvar_Get("in_mouse", "1", CVAR_ARCHIVE);

  // joystick variables
  in_joystick = Cvar_Get("in_joystick", "0", CVAR_ARCHIVE);
  joy_name = Cvar_Get("joy_name", "joystick", 0);
  joy_advanced = Cvar_Get("joy_advanced", "0", 0);
  joy_advaxisx = Cvar_Get("joy_advaxisx", "0", 0);
  joy_advaxisy = Cvar_Get("joy_advaxisy", "0", 0);
  joy_advaxisz = Cvar_Get("joy_advaxisz", "0", 0);
  joy_advaxisr = Cvar_Get("joy_advaxisr", "0", 0);
  joy_advaxisu = Cvar_Get("joy_advaxisu", "0", 0);
  joy_advaxisv = Cvar_Get("joy_advaxisv", "0", 0);
  joy_forwardthreshold = Cvar_Get("joy_forwardthreshold", "0.15", 0);
  joy_sidethreshold = Cvar_Get("joy_sidethreshold", "0.15", 0);
  joy_upthreshold = Cvar_Get("joy_upthreshold", "0.15", 0);
  joy_pitchthreshold = Cvar_Get("joy_pitchthreshold", "0.15", 0);
  joy_yawthreshold = Cvar_Get("joy_yawthreshold", "0.15", 0);
  joy_forwardsensitivity = Cvar_Get("joy_forwardsensitivity", "-1", 0);
  joy_sidesensitivity = Cvar_Get("joy_sidesensitivity", "-1", 0);
  joy_upsensitivity = Cvar_Get("joy_upsensitivity", "-1", 0);
  joy_pitchsensitivity = Cvar_Get("joy_pitchsensitivity", "1", 0);
  joy_yawsensitivity = Cvar_Get("joy_yawsensitivity", "-1", 0);

  // centering
  v_centermove = Cvar_Get("v_centermove", "0.15", 0);
  v_centerspeed = Cvar_Get("v_centerspeed", "500", 0);

  Cmd_AddCommand("+mlook", IN_MLookDown);
  Cmd_AddCommand("-mlook", IN_MLookUp);

  //   Cmd_AddCommand("joy_advancedupdate", Joy_AdvancedUpdate_f);

  IN_StartupMouse();
  IN_StartupJoystick();
}

/*
===========
IN_Shutdown
===========
*/
void IN_Shutdown(void) { IN_DeactivateMouse(); }

/*
===========
IN_Activate

Called when the main window gains or loses focus.
The window may have been destroyed and recreated
between a deactivate and an activate.
===========
*/
void IN_Activate(bool active) {
  in_appactive = active;
  mouseactive = !active; // force a new window check or turn off
}

/*
==================
IN_Frame

Called every frame, even if not generating commands
==================
*/
void IN_Frame(void) {
  if(!mouseinitialized)
    return;

  if(!in_mouse || !in_appactive) {
    IN_DeactivateMouse();
    return;
  }

  if(!cl.refresh_prepped || cls.key_dest == key_console || cls.key_dest == key_menu) {
    // temporarily deactivate if in fullscreen
    if(Cvar_VariableValue("vid_fullscreen") == 0) {
      IN_DeactivateMouse();
      return;
    }
  }

  IN_ActivateMouse();
}

/*
===========
IN_Move
===========
*/
void IN_Move(usercmd_t *cmd) {
  IN_MouseMove(cmd);

  if(ActiveApp)
    IN_JoyMove(cmd);
}

/*
===================
IN_ClearStates
===================
*/
void IN_ClearStates(void) {
  mx_accum = 0;
  my_accum = 0;
  mouse_oldbuttonstate = 0;
}

/*
=========================================================================

JOYSTICK

=========================================================================
*/

/*
===============
IN_StartupJoystick
===============
*/
void IN_StartupJoystick(void) {
  //int numdevs;
  cvar_t *cv;

  // assume no joystick
  joy_avail = false;

  // abort startup if user requests no joystick
  cv = Cvar_Get("in_initjoy", "1", CVAR_NOSET);
  if(!cv->value)
    return;

  // cycle through the joystick ids for the first valid one
  for(joy_id = 0; joy_id < GLFW_JOYSTICK_LAST; joy_id++) {
    if(glfwJoystickPresent(GLFW_JOYSTICK_1 + joy_id))
      break;
  }

  // abort startup if we didn't find a valid joystick
  if(joy_id >= GLFW_JOYSTICK_LAST) {
    Com_Printf("\njoystick not found -- no valid joysticks\n\n");
    return;
  }

  // save the joystick's number of buttons and POV status
  glfwGetJoystickButtons(joy_id, &joy_numbuttons);
  int hats;
  glfwGetJoystickHats(joy_id, &hats);
  joy_haspov = hats > 0;

  // old button and POV states default to no buttons pressed
  joy_oldbuttonstate = joy_oldpovstate = 0;

  // mark the joystick as available and advanced initialization not completed
  // this is needed as cvars are not available during initialization

  joy_avail = true;

  Com_Printf("\njoystick detected\n\n");
}

/*
===========
IN_Commands
===========
*/
void IN_Commands(void) {
  int i, key_index;
  int buttonstate, povstate;

  if(!joy_avail) {
    return;
  }

  // loop through the joystick buttons
  // key a joystick event or auxillary event for higher number buttons for each state change
  int num_buttons;
  const unsigned char *buttons = glfwGetJoystickButtons(joy_id, &num_buttons);

  buttonstate = 0;
  for(i = 0; i < joy_numbuttons; i++) {
    if(buttons[i] == GLFW_PRESS)
      buttonstate |= (1 << i);

    if((buttonstate & (1 << i)) && !(joy_oldbuttonstate & (1 << i))) {
      key_index = (i < 4) ? K_JOY1 : K_AUX1;
      Key_Event(key_index + i, true, 0);
    }

    if(!(buttonstate & (1 << i)) && (joy_oldbuttonstate & (1 << i))) {
      key_index = (i < 4) ? K_JOY1 : K_AUX1;
      Key_Event(key_index + i, false, 0);
    }
  }
  joy_oldbuttonstate = buttonstate;

  if(joy_haspov) {
    int num_hats;
    const unsigned char *hats = glfwGetJoystickHats(joy_id, &num_hats);

    // convert POV information into 4 bits of state information
    // this avoids any potential problems related to moving from one
    // direction to another without going through the center position
    povstate = 0;
    if(hats[0] != 0) {
      if(hats[0] & GLFW_HAT_UP)
        povstate |= 0x01;
      if(hats[0] & GLFW_HAT_RIGHT)
        povstate |= 0x02;
      if(hats[0] & GLFW_HAT_DOWN)
        povstate |= 0x04;
      if(hats[0] & GLFW_HAT_LEFT)
        povstate |= 0x08;
    }

    // determine which bits have changed and key an auxillary event for each change
    for(i = 0; i < 4; i++) {
      if((povstate & (1 << i)) && !(joy_oldpovstate & (1 << i))) {
        Key_Event(K_AUX29 + i, true, 0);
      }

      if(!(povstate & (1 << i)) && (joy_oldpovstate & (1 << i))) {
        Key_Event(K_AUX29 + i, false, 0);
      }
    }
    joy_oldpovstate = povstate;
  }
}

/*
===========
IN_JoyMove
===========
*/
void IN_JoyMove(usercmd_t *cmd) {
  float speed, aspeed;
  float fAxisValue;
  int i;

  // verify joystick is available and that the user wants to use it
  if(!joy_avail || !in_joystick->value) {
    return;
  }

  int num_axis;
  const float *axes = glfwGetJoystickAxes(joy_id, &num_axis);

  if((in_speed.state & 1) ^ (int)cl_run->value)
    speed = 2;
  else
    speed = 1;
  aspeed = speed * cls.frametime;

  // loop through the axes
  for(i = 0; i < num_axis; i++) {
    // get the floating point zero-centered, potentially-inverted data for the current axis
    fAxisValue = axes[i];

    switch(dwAxisMap[i]) {
    case AxisForward:
      if((joy_advanced->value == 0.0) && mlooking) {
        // user wants forward control to become look control
        if(fabs(fAxisValue) > joy_pitchthreshold->value) {
          // if mouse invert is on, invert the joystick pitch value
          // only absolute control support here (joy_advanced is false)
          if(m_pitch->value < 0.0) {
            cl.viewangles[PITCH] -= (fAxisValue * joy_pitchsensitivity->value) * aspeed * cl_pitchspeed->value;
          } else {
            cl.viewangles[PITCH] += (fAxisValue * joy_pitchsensitivity->value) * aspeed * cl_pitchspeed->value;
          }
        }
      } else {
        // user wants forward control to be forward control
        if(fabs(fAxisValue) > joy_forwardthreshold->value) {
          cmd->forwardmove += (fAxisValue * joy_forwardsensitivity->value) * speed * cl_forwardspeed->value;
        }
      }
      break;

    case AxisSide:
      if(fabs(fAxisValue) > joy_sidethreshold->value) {
        cmd->sidemove += (fAxisValue * joy_sidesensitivity->value) * speed * cl_sidespeed->value;
      }
      break;

    case AxisUp:
      if(fabs(fAxisValue) > joy_upthreshold->value) {
        cmd->upmove += (fAxisValue * joy_upsensitivity->value) * speed * cl_upspeed->value;
      }
      break;

    case AxisTurn:
      if((in_strafe.state & 1) || (lookstrafe->value && mlooking)) {
        // user wants turn control to become side control
        if(fabs(fAxisValue) > joy_sidethreshold->value) {
          cmd->sidemove -= (fAxisValue * joy_sidesensitivity->value) * speed * cl_sidespeed->value;
        }
      } else {
        // user wants turn control to be turn control
        if(fabs(fAxisValue) > joy_yawthreshold->value) {
          if(dwControlMap[i] == JOY_ABSOLUTE_AXIS) {
            cl.viewangles[YAW] += (fAxisValue * joy_yawsensitivity->value) * aspeed * cl_yawspeed->value;
          } else {
            cl.viewangles[YAW] += (fAxisValue * joy_yawsensitivity->value) * speed * 180.0;
          }
        }
      }
      break;

    case AxisLook:
      if(mlooking) {
        if(fabs(fAxisValue) > joy_pitchthreshold->value) {
          // pitch movement detected and pitch movement desired by user
          if(dwControlMap[i] == JOY_ABSOLUTE_AXIS) {
            cl.viewangles[PITCH] += (fAxisValue * joy_pitchsensitivity->value) * aspeed * cl_pitchspeed->value;
          } else {
            cl.viewangles[PITCH] += (fAxisValue * joy_pitchsensitivity->value) * speed * 180.0;
          }
        }
      }
      break;

    default:
      break;
    }
  }
}