/*
   ___         _           _   ___             _
  / __|___ _ _| |_ _ _ ___| | | __| _ ___ __ _| |__
 | (__/ _ \ ' \  _| '_/ _ \ | | _| '_/ -_) _` | / /
  \___\___/_||_\__|_| \___/_| |_||_| \___\__,_|_\_\

  https://hackaday.io/project/170908-control-freak

  Danjovic 2020

  PC Joystick - 12/04/2020
*/



//    _ _ _                 _
//   | (_) |__ _ _ __ _ _ _(_)___ ___
//   | | | '_ \ '_/ _` | '_| / -_|_-<
//   |_|_|_.__/_| \__,_|_| |_\___/__/
//
#include "HID-Project.h"


//       _      __ _      _ _   _
//    __| |___ / _(_)_ _ (_) |_(_)___ _ _  ___
//   / _` / -_)  _| | ' \| |  _| / _ \ ' \(_-<
//   \__,_\___|_| |_|_||_|_|\__|_\___/_||_/__/
//

//                             AVR
#define joy1Button1 2    //  PD1/INT1
#define joy1Button2 3    //  PD0/INT0
#define joy2Button1 4    //  PD4
#define joy2Button2 6    //  PD7

#define joy1Xaxis   A3   //  PF4
#define joy1Yaxis   A2   //  PF5
#define joy2Xaxis   A1   //  PF6
#define joy2Yaxis   A0   //  PF7


#define holdCaps()    DDRF  = 0xf0
#define releaseCaps() DDRF  = 0x00
#define pullupsOff()  PORTF = 0x00

#define maxCounts     1024
#define midRange       200

//#define DEBUG
//                 _      _    _
//   __ ____ _ _ _(_)__ _| |__| |___ ___
//   \ V / _` | '_| / _` | '_ \ / -_|_-<
//    \_/\__,_|_| |_\__,_|_.__/_\___/__/
//
uint16_t a1X, a1Y, a2X, a2Y;




//    ___      _
//   / __| ___| |_ _  _ _ __
//   \__ \/ -_)  _| || | '_ \
//   |___/\___|\__|\_,_| .__/
//                     |_|
void setup() {

  pullupsOff(); // turn off pullups
  holdCaps();   // start with caps discharged

  pinMode(joy1Button1 , INPUT_PULLUP);
  pinMode(joy1Button2 , INPUT_PULLUP);

  pinMode(joy2Button1 , INPUT_PULLUP);
  pinMode(joy2Button2 , INPUT_PULLUP);

  // Sends a clean report to the host. This is important on any Arduino type.
  Gamepad.begin();
  
#if defined (DEBUG)  
  Serial.begin(9600);
#endif
}


//    __  __      _        _
//   |  \/  |__ _(_)_ _   | |   ___  ___ _ __
//   | |\/| / _` | | ' \  | |__/ _ \/ _ \ '_ \
//   |_|  |_\__,_|_|_||_| |____\___/\___/ .__/
//                                      |_|
void loop() {

  // Start with a clean slate
  Gamepad.releaseAll(); 

  // Update Analog Axes   
  sampleAnalogAxes();
  Gamepad.xAxis((int16_t)a1X);
  Gamepad.yAxis((int16_t)a1Y);
  Gamepad.rxAxis((int16_t)a2X);
  Gamepad.ryAxis((int16_t)a2Y);

  // Update Buttons
  if (digitalRead(joy1Button1) == 0) Gamepad.press(1);
  if (digitalRead(joy1Button2) == 0) Gamepad.press(2);

  if (digitalRead(joy2Button1) == 0) Gamepad.press(3);
  if (digitalRead(joy2Button2) == 0) Gamepad.press(4);

  // Functions above only set the values.
  // This writes the report to the host.
  Gamepad.write();

#if defined (DEBUG) 
    Serial.print("Joy1->");

    // Print Axes status
    Serial.print(" aX:"); Serial.print(a1X);
    Serial.print(" aY:"); Serial.print(a1Y);

    Serial.print("   Joy2->");

    // Print Axes status
    Serial.print(" aX:"); Serial.print(a2X);
    Serial.print(" aY:"); Serial.print(a2Y);

    Serial.println();
#endif

  // Simple debounce
  delay(10);

}



//     __              _   _
//    / _|_  _ _ _  __| |_(_)___ _ _  ___
//   |  _| || | ' \/ _|  _| / _ \ ' \(_-<
//   |_|  \_,_|_||_\__|\__|_\___/_||_/__/
//
void sampleAnalogAxes() {
  static volatile uint16_t counter = 0;
  uint8_t  sample;

  a1X = 0;
  a1Y = 0;
  a2X = 0;
  a2Y = 0;

  releaseCaps();
  delayMicroseconds(5);

  // sample inputs
  counter = maxCounts;
  do  {
    sample = ~PINF;
    delayMicroseconds(3);
    sample >>= 4;
    a1X += sample & 1; // F4
    sample >>= 1;
    a1Y += sample & 1; // F5
    sample >>= 1;
    a2X += sample & 1; // F6
    sample >>= 1;
    a2Y += sample & 1; // F7

  } while (--counter);

  // Take care of disconnected axes
  if (a1X == maxCounts ) a1X=midRange;
  if (a1Y == maxCounts ) a1Y=midRange;
  if (a2X == maxCounts ) a2X=midRange;
  if (a2Y == maxCounts ) a2Y=midRange;
  

  // reset caps
  holdCaps();
}