#pragma once

#include <vector>

namespace input
{
	enum class product_type
	{
		playstation_3_controller,
		red_octane_gh_guitar,
		red_octane_gh_drum_kit,
		dance_dance_revolution_mat,
		dj_hero_turntable,
		harmonix_rockband_guitar,
		harmonix_rockband_drum_kit,
		harmonix_rockband_drum_kit_2,
		rock_revolution_drum_kit
	};

	enum vendor_id
	{
		sony_corp = 0x054C, // Sony Corp.
		sony_cea  = 0x12BA, // Sony Computer Entertainment America
		konami_de = 0x1CCF, // Konami Digital Entertainment
	};

	enum product_id
	{
		red_octane_gh_guitar         = 0x0100, // RedOctane Guitar (Guitar Hero 4 Guitar Controller)
		red_octane_gh_drum_kit       = 0x0120, // RedOctane Drum Kit (Guitar Hero 4 Drum Controller)
		dance_dance_revolution_mat   = 0x0140, // Dance Dance Revolution Dance Mat Controller
		dj_hero_turntable            = 0x0140, // DJ Hero Turntable Controller
		harmonix_rockband_guitar     = 0x0200, // Harmonix Guitar (Rock Band II Guitar Controller)
		harmonix_rockband_drum_kit   = 0x0210, // Harmonix Drum Kit (Rock Band II Drum Controller)
		harmonix_rockband_drum_kit_2 = 0x0218, // Harmonix Pro-Drum Kit (Rock Band III Pro-Drum Controller)
		playstation_3_controller     = 0x0268, // PlayStation 3 Controller
		rock_revolution_drum_kit     = 0x0300, // Rock Revolution Drum Controller
	};

	struct product_info
	{
		product_type type;
		unsigned short vendor_id;
		unsigned short product_id;
		unsigned int pclass_profile; // See CELL_PAD_PCLASS_PROFILE flags
	};

	inline product_info get_product_info(product_type type)
	{
		switch (type)
		{
		case product_type::dance_dance_revolution_mat:
		{
			return product_info{ type, vendor_id::konami_de, product_id::dance_dance_revolution_mat, 0x000000FF };
		}
		case product_type::dj_hero_turntable:
		{
			return product_info{ type, vendor_id::sony_cea, product_id::dj_hero_turntable, 0x000007FF };
		}
		case product_type::harmonix_rockband_drum_kit:
		{
			return product_info{ type, vendor_id::sony_cea, product_id::harmonix_rockband_drum_kit, 0x000000FF };
		}
		case product_type::harmonix_rockband_drum_kit_2:
		{
			return product_info{ type, vendor_id::sony_cea, product_id::harmonix_rockband_drum_kit_2, 0x000000BF };
		}
		case product_type::harmonix_rockband_guitar:
		{
			return product_info{ type, vendor_id::sony_cea, product_id::harmonix_rockband_guitar, 0x00007FFF };
		}
		case product_type::red_octane_gh_drum_kit:
		{
			return product_info{ type, vendor_id::sony_cea, product_id::red_octane_gh_drum_kit, 0x000000BB };
		}
		case product_type::red_octane_gh_guitar:
		{
			return product_info{ type, vendor_id::sony_cea, product_id::red_octane_gh_guitar, 0x000000FF };
		}
		case product_type::rock_revolution_drum_kit:
		{
			return product_info{ type, vendor_id::sony_cea, product_id::rock_revolution_drum_kit, 0x000000FB };
		}
		case product_type::playstation_3_controller:
		default: // GCC doesn't like it when there is no return value if if all enum values are covered
		{
			return product_info{ type, vendor_id::sony_corp, product_id::playstation_3_controller, 0x0 };
		}
		}
	}

	inline std::vector<product_info> get_products_by_class(int class_id)
	{
		switch (class_id)
		{
		default:
		case 0: // CELL_PAD_PCLASS_TYPE_STANDARD
		{
			return
			{
				get_product_info(product_type::playstation_3_controller)
			};
		}
		case 1: // CELL_PAD_PCLASS_TYPE_GUITAR
		{
			return
			{
				get_product_info(product_type::red_octane_gh_guitar),
				get_product_info(product_type::harmonix_rockband_guitar)
			};
		}
		case 2: // CELL_PAD_PCLASS_TYPE_DRUM
		{
			return
			{
				get_product_info(product_type::red_octane_gh_drum_kit),
				get_product_info(product_type::harmonix_rockband_drum_kit),
				get_product_info(product_type::harmonix_rockband_drum_kit_2),
				get_product_info(product_type::rock_revolution_drum_kit)
			};
		}
		case 3: // CELL_PAD_PCLASS_TYPE_DJ
		{
			return
			{
				get_product_info(product_type::dj_hero_turntable)
			};
		}
		case 4: // CELL_PAD_PCLASS_TYPE_DANCEMAT
		{
			return
			{
				get_product_info(product_type::dance_dance_revolution_mat)
			};
		}
		case 5: // CELL_PAD_PCLASS_TYPE_NAVIGATION
		{
			return
			{
				get_product_info(product_type::playstation_3_controller)
			};
		}
		}
	}
};