include "target.tmh"

//Variables
char operatingmode; 

//Initialization
int main()
{
	//Exclude all but WH Throttle
	Configure(&HCougar, MODE_EXCLUDED); 
	Configure(&Joystick, MODE_EXCLUDED);
	Configure(&T16000, MODE_EXCLUDED);
	Configure(&LMFD, MODE_EXCLUDED);
	Configure(&RMFD, MODE_EXCLUDED);
    if(Init(&EventHandle)) return 1; 
  
    //Mode Lights Off
    ActKey(PULSE+KEYON+LED(&Throttle, LED_ONOFF, LED_CURRENT-LED1));
	ActKey(PULSE+KEYON+LED(&Throttle, LED_ONOFF, LED_CURRENT-LED2));
	ActKey(PULSE+KEYON+LED(&Throttle, LED_ONOFF, LED_CURRENT-LED3));
	ActKey(PULSE+KEYON+LED(&Throttle, LED_ONOFF, LED_CURRENT-LED4));
	ActKey(PULSE+KEYON+LED(&Throttle, LED_ONOFF, LED_CURRENT-LED5));
    //Backlight Control
	MapKey(&Throttle, PSF, PULSE+LED(&Throttle, LED_INTENSITY, 86)); 
	MapKey(&Throttle, PSM, PULSE+LED(&Throttle, LED_INTENSITY, 0));
	MapKey(&Throttle, PSB, PULSE+LED(&Throttle, LED_INTENSITY, 43));    

	//Mode Switching 
	SetShiftButton(&Throttle, PSF, &Throttle, 0, 0);
	MapKeyIO(&Throttle, CHF, EXEC("mode_change(operatingmode,1);") , 0);
	MapKeyIO(&Throttle, CHB, EXEC("mode_change(operatingmode,0);"), 0);
}

//Mode Change Primary Function
int mode_change(int operatingmode, int increase)
{
	if(increase != 0)	{ operatingmode=operatingmode+1; }
	else operatingmode=operatingmode-1;
	
	mode_exec(operatingmode);
	mode_lights(operatingmode);
	
	return operatingmode;
}

//Remapping Sub Function
int mode_exec (int operatingmode)
{

}

//Light Indicator Sub Function
int mode_lights (int operatingmode)
{
	printf("Test");
}

//Event Handler
int EventHandle(int type, alias o, int x)
{
	//Event Handler code
	DefaultMapping(&o, x);
}

EXEC("mode_change(operatingmode,1);"

include "target.tmh"
include "Drakoz_TrainSimWorld.ttm"

// Global variables

// Default Engine Profile 
//	This script is designed to have different control setups for different engines.  
//	This is mainly to control the difference of separate throttle and brake levers (e.g. SD40-2, GP38)
//  vs. combined throttle/brake levers (e.g. AC4400CW). 
//	Choices:
//	  pro_ProfileSelection = 
//			pro_SD40GP38		- EMD SD40-2, GP38-2, GP40-2	(LED 00001 = 1)
//			pro_AC4400CW		- GE AC4400CW					(LED 00010 = 2)
//
// Uncomment the line below for the engine you want to be the default engine.  
int pro_ProfileSelection = pro_SD40GP38;

int main()
{

	if(Init(&EventHandle)) return 1;

	//******************************************************************
	// !!!!! Don't add any code above the Configure and EventHandle commands above !!!!!
	//******************************************************************
	
	printf("----------------------------------\xa");

	//******************************************************************
	//	Setup some basic settings

	SetKBLayout(KB_ENG);											// this file designed for English keyboard.
	SetKBRate(100, 100); 											// SetKBRate(PULSE delay, D() delay) in ms - long delays needed for TSW
	SetShiftButton(&Throttle, MSP, &Throttle, FLAPU, FLAPD, 0);		// IO Shift and UMD Setup (%DEV1, I button, $DEV, U, D, Toggle settings)

	//******************************************************************
	//  Warthog Throttle LED Setup 

	ActKey(PULSE+KEYON+LED(&Throttle, LED_INTENSITY, LEDLevel1));		// set Throttle LED back light intensity
	ActKey(PULSE+KEYON+LED(&Throttle, LED_ONOFF, LED_CURRENT-LED1)); 	// LED 1 OFF
	ActKey(PULSE+KEYON+LED(&Throttle, LED_ONOFF, LED_CURRENT-LED2)); 	// LED 2 OFF
	ActKey(PULSE+KEYON+LED(&Throttle, LED_ONOFF, LED_CURRENT-LED3)); 	// LED 3 OFF
	ActKey(PULSE+KEYON+LED(&Throttle, LED_ONOFF, LED_CURRENT-LED4)); 	// LED 4 OFF
	ActKey(PULSE+KEYON+LED(&Throttle, LED_ONOFF, LED_CURRENT-LED5)); 	// LED 5 OFF
	
	//******************************************************************
	//  Configure buttons and axis

	// Configuration that is common among all engines can be set in the function CommonConfig().  
	// pro_CallCustomConfigs() calls CommonConfig() first to make sure all common setups are done first
		
	// Call functions to make custom configuration for each profile
	//	Allows creating MapKey() and other configurations specific to each engine type.
	//	Configurations in pro_CallCustomConfigs() will override any previous MapKey(), KeyAxis(), etc. commands.  
	pro_CallCustomConfigs();

	// Set LEDs to match current profile
	setLEDbyBinary(&Throttle, pro_ProfileSelection);

	printf("----------------------------------\xa");
	
}	// end main()

//******************************************************************
// CommonConfig()
//	Configurations (MapKey, KeyAxis, Axis setup, etc.) that is common to all engines should be set up here. 
//  This function will be called every time a profile change occurs before setting any custom configuration for specific engines.  
int CommonConfig()
{

	//******************************************************************
	//	Warthog Throttle Button Controls

    //  Add all common configuration statements here....  (some examples shown)

	// Slew Control Button
	MapKeyIO(&Throttle, SC, 0, L_SHIFT);
	
	// Coolie Switch (HAT)
	MapKeyIO(&Throttle, CSU, L_CTL+DARROW, DARROW);
	MapKeyIO(&Throttle, CSD, L_CTL+UARROW, UARROW);
	MapKeyIO(&Throttle, CSL, L_CTL+LARROW, LARROW);
	MapKeyIO(&Throttle, CSR, L_CTL+RARROW, RARROW);
	
	//  Landing Gear Horn Silence Button    
	MapKeyIO(&Throttle, LDGH, 
		EXEC("pro_RotateProfile();"),   // (Map MSP + LDGH to call pro_RotateProfile() and change to next profile)
		AWS_Reset               // Otherwise, LDGH does AWS_Reset button press
	);
	
}

//******************************************************************
// pro_CallCustomConfigs()
//	Call functions to make custom configuration for each profile
int pro_CallCustomConfigs()
{
	// Do common config shared by all engines first
	CommonConfig();		
	
	// Then do custom config depending on profile.  
	// MapKey(), etc. setup in custom config will overwrite the button/axis configs from the common config.
	if      (pro_ProfileSelection == pro_SD40GP38)	pro_ConfigureSD40GP38();
	else if (pro_ProfileSelection == pro_AC4400CW)	pro_ConfigureAC4400CW();

}

//******************************************************************
// pro_ConfigureSD40GP38()
//	Custom configuration for SD40-2, GP38-2, GP40-2 and other similar diesel engines
int pro_ConfigureSD40GP38()
{
	printf("Profile %d: pro_SD40GP38\xa", pro_SD40GP38);
	
	// Reverser - using Throttle Friction Control
	KeyAxis(&Throttle, THR_FC, 'ioumd', AXMAP1(LIST(0,40,60,100), 
		CHAIN(LOCK+DOWN+Reverser_Backwards, D(400), UP+Reverser_Backwards, D(50), LOCK), 
		CHAIN(LOCK+DOWN+Reverser_Forward, D(400), UP+Reverser_Forward, D(50), LOCK))
	);
}
		
//******************************************************************
// pro_ConfigureAC4400CW()
//	Custom configuration for AC4400CW. 
int pro_ConfigureAC4400CW()
{
	printf("Profile %d: pro_AC4400CW\xa", pro_AC4400CW);
	
	// Reverser - using Throttle Friction Control
	KeyAxis(&Throttle, THR_FC, 'ioumd', AXMAP1(LIST(0,40,60,100), 
		CHAIN(LOCK+DOWN+Reverser_Backwards, D(400), UP+Reverser_Backwards, D(50), LOCK), 
		CHAIN(LOCK+DOWN+Reverser_Forward, D(400), UP+Reverser_Forward, D(50), LOCK))
	);
	
	// Headlight Switch - different delays needed vs. other engines. 
	// Following complicated CHAIN commands needed to overcome issues with SHIFT & CTL not being received properly by TSW (bug with TSW?)
	int headlightFrontFwd = CHAIN(LOCK+DOWN+'h', D(250), UP+'h', D(50), LOCK);
	int headlightFrontBkd = CHAIN(LOCK+DOWN+L_SHIFT, D(50), DOWN+'h', D(250), UP+'h', D(50), UP+L_SHIFT, D(50), LOCK);
	int headlightRearFwd = CHAIN(LOCK+DOWN+L_CTL, D(50), DOWN+'h', D(300), UP+'h', D(50), UP+L_CTL, D(50), LOCK);
	int headlightRearBkd = CHAIN(LOCK+DOWN+L_SHIFT, D(50), DOWN+L_CTL, D(50), DOWN+'h', D(300), UP+'h', D(50), UP+L_CTL, D(50), UP+L_SHIFT, D(50), LOCK);
	// Engine Operate Switch - Right
	MapKeyIO (&Throttle, EORIGN, 0, headlightFrontFwd);
	MapKeyIO (&Throttle, EORNORM, 0, 0);
	MapKeyRIO(&Throttle, EORMOTOR, 0, 0);
	MapKeyIO (&Throttle, EORMOTOR, 0, headlightFrontBkd);	
	// Engine Operate Switch - Left
	MapKeyIO (&Throttle, EOLIGN, 0, headlightRearFwd);
	MapKeyIO (&Throttle, EOLNORM, 0, 0);
	MapKeyRIO(&Throttle, EOLMOTOR, 0, 0);
	MapKeyIO (&Throttle, EOLMOTOR, 0, headlightRearBkd);	
}



// ******************* End of User Configuration ******************************************************************************************
//	Users should not make changes to the functions below.



//******************************************************************
// pro_RotateProfile()
//	Rotate through engine profiles and implement changes for specific engines.
//	Rotates LEDs to represent current profile
//	Note sections below for custom programming for each engine.  
//	
int pro_RotateProfile()
{
	pro_ProfileSelection = (pro_ProfileSelection + 1) % (pro_NumOfProfiles + 1);
	if (pro_ProfileSelection == 0) pro_ProfileSelection = 1;
	setLEDbyBinary(&Throttle, pro_ProfileSelection);
	//printf("Profile changed to: %d\xa", pro_ProfileSelection);
	pro_CallCustomConfigs();
}

//******************************************************************
// setLEDbyBinary()
//	Set LEDs to match the binary equivalent of p 
//	dev - device (e.g. &Throttle)
int LEDListArray[5] = {LED1, LED2, LED3, LED4, LED5};
int setLEDbyBinary(alias dev, int p)
{
	int x = 0;
	while (x < 5) {
		if (p & 0x1) ActKey(PULSE+KEYON+LED(&dev, LED_ONOFF, LED_CURRENT+LEDListArray[x]));
		else ActKey(PULSE+KEYON+LED(&dev, LED_ONOFF, LED_CURRENT-LEDListArray[x]));
		x = x + 1;
		p = p >> 0x1;
	}
}



//******************************************************************
int EventHandle(int type, alias o, int x)
{	
	DefaultMapping(&o, x);
}
//******************************************************************

//******************************************************************
//	Warthog Throttle LED Settings
define		LEDLevelOff		0				// LED intensity off
define		LEDLevel1		50				// LED intensity level 1
define		LEDLevel2		100				// LED intensity level 2
define		LEDLevel3		150				// LED intensity level 3
define		LEDLevel4		200				// LED intensity level 4
define		LEDLevel5		255				// LED intensity level 5

//******************************************************************
//	Profile Selection Labels (used for the variable pro_ProfileSelection)
//	Do not define any profile at 0, only 1 through 31 allowed.  
define		pro_SD40GP38		1		// EMD SD40-2, GP38-2, GP40-2	(LED 00001 = 1)
define		pro_AC4400CW		2		// GE AC4400CW					(LED 00010 = 2)

define		pro_NumOfProfiles	2		// Set to the number of profiles listed  (31 max allowed)

	pro_CallCustomConfigs();
	setLEDbyBinary(&Throttle, pro_ProfileSelection);

	if      (pro_ProfileSelection == pro_SD40GP38)	pro_ConfigureSD40GP38();
	else if (pro_ProfileSelection == pro_AC4400CW)	pro_ConfigureAC4400CW();

	
//  Landing Gear Horn Silence Button    
	MapKeyIO(&Throttle, LDGH, 
		EXEC("pro_RotateProfile();"),   // (Map MSP + LDGH to call pro_RotateProfile() and change to next profile)
		AWS_Reset               // Otherwise, LDGH does AWS_Reset button press
	);
	

define Interconnect_RPM EXEC("sync_rpm = !sync_rpm;")

...

// INTERCONNECT THROTTLE WITH PROP PITCH 

	int SyncRPM()
	{
		if (sync_rpm) DXAxis(DX_Z_AXIS, -T16000[THR]); // throttle and rpm connected
	}

int main()
{

...

//	B14
	MapKeyIO(&T16000, B14, 0, Interconnect_RPM);

...

}
int EventHandle(int type, alias o, int x)
{
	DefaultMapping(&o, x);
	
	if ((&o == &T16000) & (x == THR)) SyncRPM();
}