#ifndef  WII_EC_H_
#define  WII_EC_H_

#include  <stdbool.h>

#include  <furi.h>

#include  "wii_ec_nunchuck.h"
#include  "wii_ec_classic.h"
#include  "wii_ec_udraw.h"

//----------------------------------------------------------------------------- ----------------------------------------
// Crypto key (PSK),      base register : {0x40..0x4F}[2][8]
#define  ENC_LEN  (2*8)

// Controller State data, base register : {0x00..0x05}[6]
#define  JOY_LEN  (6)

// Calibration data,      base register : {0x20..0x2F}[16]
#define  CAL_LEN  (16)

// Controller ID,         base register : {0xFA..0xFF}[6]
#define  PID_LEN  (6)

//----------------------------------------------------------------------------- ----------------------------------------
// Perhipheral specific parameters union
//
typedef
	union ecDec {
		ecDecNunchuck_t  nunchuck;
		ecDecClassic_t   classic;
	}
ecDec_t;

//-----------------------------------------------------------------------------
typedef
	union ecCal {
		// 0=lowest seen ; 1=min ; 2=mid ; 3=max ; 4=highest seen
		ecCalNunchuck_t  nunchuck[5];
		ecCalClassic_t   classic[5];
	}
ecCal_t;

//----------------------------------------------------------------------------- ----------------------------------------
// Wii Extension Controller events
//
typedef
	enum wiiEcEventType {
		WIIEC_NONE,
		WIIEC_CONN,     // Connect
		WIIEC_DISCONN,  // Disconnect
		WIIEC_PRESS,    // Press button
		WIIEC_RELEASE,  // Release button
		WIIEC_ANALOG,   // Analogue change (Joystick/Trigger)
		WIIEC_ACCEL,    // Accelerometer change
	}
wiiEcEventType_t;

//-----------------------------------------------------------------------------
typedef
	struct wiiEcEvent {
		wiiEcEventType_t  type;  // event type
		char              in;    // input (see device specific options)
		uint32_t          val;   // new value - meaningless for digital button presses
	}
wiiEcEvent_t;

//----------------------------------------------------------------------------- ----------------------------------------
// Known perhipheral types
//
typedef
	enum ecPid {
		PID_UNKNOWN  = 0,
		PID_FIRST    = 1,
		PID_NUNCHUCK = PID_FIRST,

		// If you're wise, ONLY edit this section
			PID_NUNCHUCK_R2,
			PID_CLASSIC,
			PID_CLASSIC_PRO,
			PID_BALANCE,
			PID_GH_GUITAR,
			PID_GH_DRUMS,
			PID_TURNTABLE,
			PID_TAIKO_DRUMS,
			PID_UDRAW,  //! same as drawsome?
		// -----

		PID_ERROR,
		PID_NULL,
		PID_CNT,
	}
ecPid_t;

//-----------------------------------------------------------------------------
// Calibration strategies
//
typedef
	enum ecCalib {
		CAL_FACTORY = 0x01,  // (re)set to factory defaults
		CAL_TRACK   = 0x02,  // track maximum and minimum values seen
		CAL_RESET   = 0x04,  // initialise ready for software calibration
		CAL_RANGE   = 0x08,  // perform software calibration step
		CAL_CENTRE  = 0x10,  // reset centre point of joystick
		CAL_NOTJOY  = 0x20,  // do NOT calibrate the joystick
	}
ecCalib_t;

//-----------------------------------------------------------------------------
// ecId table entry
//
typedef
	struct ecId {
		uint8_t   id[6];                                             // 6 byte ID string returned by Extension Controller
		char*     name;                                              // Friendly name
		scene_t   scene;                                             // Default scene
		bool      (*init)(wiiEC_t*);                                 // Additional initialisation code
		void      (*decode)(wiiEC_t*);                               // Decode function
		void      (*check)(wiiEC_t*, FuriMessageQueue*);             // check (for action) function
		void      (*calib)(wiiEC_t*, ecCalib_t);                     // calibrate analogue controllers [SOFTWARE]
		void      (*show)(Canvas* const, state_t* const);            // Draw scene
		bool      (*keys)(const eventMsg_t* const, state_t* const);  // Interpret keys
	}
ecId_t;

//-----------------------------------------------------------------------------
// List of known perhipherals
//
// More perhipheral ID codes here:  https://wiibrew.org/wiki/Wiimote/Extension_Controllers#The_New_Way
//
extern  const ecId_t  ecId[PID_CNT] ;

//----------------------------------------------------------------------------- ----------------------------------------
// Data pertaining to a single Perhipheral instance
//
typedef
	struct wiiEC {
		// Perhipheral state
		bool         init;             // Initialised?

		uint8_t      pid[PID_LEN];     // PID string - eg. {0x00, 0x00, 0xA4, 0x20, 0x00, 0x00}
		ecPid_t      pidx;             // Index in to ecId table
		const char*  sid;              // just for convenience

		bool         encrypt;          // encryption enabled?
		uint8_t      encKey[ENC_LEN];  // encryption key

		uint8_t      calF[CAL_LEN];    // factory calibration data (not software)
		uint8_t      joy[JOY_LEN];     // Perhipheral raw data

		ecDec_t      dec[2];           // device specific decode (two, so we can spot changes)
		int          decN;             // which decode set is most recent {0, 1}
		ecCal_t      calS;             // software calibration data
	}
wiiEC_t;

//----------------------------------------------------------------------------- ----------------------------------------
// Function prototypes
//
// top level calls will work out which sub-function to call
// top level check() function will handle connect/disconnect messages
//

#include  <gui/gui.h>   // Canvas
typedef   struct wiiEC     wiiEC_t ;
typedef   enum   ecCalib   ecCalib_t ;
typedef   struct state     state_t ;
typedef   struct eventMsg  eventMsg_t ;

void  ecDecode    (wiiEC_t* const pec) ;
void  ecPoll      (wiiEC_t* const pec,  FuriMessageQueue* const queue) ;
void  ecCalibrate (wiiEC_t* const pec,  ecCalib_t c) ;

void  ec_show     (       Canvas* const canvas,  state_t* const state) ;
bool  ec_key      (const eventMsg_t* const msg,  state_t* const state) ;

#endif //WII_EC_H_