package org.usfirst.frc.team4272.robot2016;

import org.usfirst.frc.team4272.robotlib.PIDOutputSplitter;
import edu.wpi.first.wpilibj.smartdashboard.SmartDashboard;

import edu.wpi.first.wpilibj.DigitalInput;
import edu.wpi.first.wpilibj.Encoder;
import edu.wpi.first.wpilibj.PIDOutput;
import edu.wpi.first.wpilibj.PIDSourceType;
import edu.wpi.first.wpilibj.Talon;

public class HwRobot {
	/* Relay == a Spike */
	/* PCM = Pneumatics Control Module */

	/* All of the numbered inputs are in the classes:
	 * - DIO: 0-9
	 * - Relay: 0-3
	 * - Analog In: 0-3
	 * - PWM: 0-9
	 * - PCM: 0-7 (the PCM is connected via CAN).
	 * - CAN
	 *
	 * For completeness, the roboRIO also has: i2c, RS-232, SPI,
	 * RSL, 2xUSB-A, an accelerometer, and an expansion port.
	 *
	 * And, for communication: USB-B and Ethernet.
	 */

	// naming convention: {side_letter}{Thing}{E|M (encoder/motor)}

	private final PIDOutput lDriveM1 = new Talon(/*PWM*/1), lDriveM2 = new Talon(/*PWM*/0);
	private final PIDOutput rDriveM1 = new Talon(/*PWM*/2), rDriveM2 = new Talon(/*PWM*/4);
	private final PIDOutput armM    = new Talon(/*PWM*/8);
	private final PIDOutput fBallM  = new PIDOutputSplitter(new Talon(/*PWM*/9), new Talon(/*PWM*/3));
	private final PIDOutput bBallM  = new Talon(/*PWM*/6);

	public final Encoder   armE = new Encoder(/*DIO*/0, /*DIO*/1, /*DIO*/2, true);
	public final DigitalInput armL = new DigitalInput(/*DIO*/3);
	public final DigitalInput ballL = new DigitalInput(/*DIO*/4);
	public final Encoder   rDriveE = new Encoder(/*DIO*/5,/*DIO*/6);
	public final Encoder   lDriveE = new Encoder(/*DIO*/7,/*DIO*/8, true);

	public HwRobot() {
		armE.setPIDSourceType(PIDSourceType.kDisplacement);
		lDriveE.setPIDSourceType(PIDSourceType.kRate);
		rDriveE.setPIDSourceType(PIDSourceType.kRate);
	}

	public void run(Control c) {
		lDriveM1.pidWrite( c.lDrive1);
		lDriveM2.pidWrite( c.lDrive2);
		rDriveM1.pidWrite(-c.rDrive1);
		rDriveM2.pidWrite(-c.rDrive2);
		armM.pidWrite(c.arm * 0.5);
		fBallM.pidWrite(-c.fBall);
		bBallM.pidWrite(-c.bBall);

		SmartDashboard.putNumber("fBall", c.fBall);
		SmartDashboard.putNumber("bBall", c.bBall);
		SmartDashboard.putNumber("armM", c.arm);

		SmartDashboard.putNumber("armE", armE.pidGet());
		SmartDashboard.putNumber("lDriveE", lDriveE.pidGet());
		SmartDashboard.putNumber("rDriveE", rDriveE.pidGet());
		SmartDashboard.putBoolean("ballL", ballL.get());
		SmartDashboard.putBoolean("armL", armL.get());
	}
}