From 480624ff5275225123d373526d8d30f844a4d99e Mon Sep 17 00:00:00 2001
From: Michael Steenbeek <m.o.steenbeek@gmail.com>
Date: Thu, 29 Dec 2016 20:32:33 +0100
Subject: [PATCH] Implement code for detecting supported track pieces (#4946)

---
 src/ride/ride.c | 23 +++++++++++++++++++++++
 src/ride/ride.h |  1 +
 2 files changed, 24 insertions(+)

diff --git a/src/ride/ride.c b/src/ride/ride.c
index 71adf591f3..48d5a3c2f6 100644
--- a/src/ride/ride.c
+++ b/src/ride/ride.c
@@ -7455,6 +7455,29 @@ uint8 ride_entry_get_vehicle_at_position(int rideEntryIndex,int numCarsPerTrain,
 	}
 }
 
+//Finds track pieces that a given ride entry doesn't have the sprites for
+uint64 ride_entry_get_supported_track_pieces(rct_ride_entry* rideEntry)
+{
+	uint64 supportedPieces = 0xFFFFFFFFFFFFFFFFULL;
+	uint16 trackPieceRequiredSprites[55] = {
+		0x0001u, 0x0001u, 0x0001u, 0x0000u, 0x0006u, 0x0002u, 0x0020u, 0x000E,
+		0x0003u, 0x0006u, 0x0007u, 0x0002u, 0x0004u, 0x0001u, 0x0001u, 0x0001u,
+		0x0001u, 0x0061u, 0x000E, 0x1081u, 0x0001u, 0x0020u, 0x0020u, 0x0001u,
+		0x0001u, 0x0000u, 0x0001u, 0x0001u, 0x000C, 0x0061u, 0x0002u, 0x000E,
+		0x0480u, 0x0001u, 0x0061u, 0x0001u, 0x0001u, 0x000Fu, 0x0001u, 0x0200u,
+		0x0007u, 0x0008u, 0x0000u, 0x0000u, 0x4000u, 0x0008u, 0x0001u, 0x0001u,
+		0x0061u, 0x0061u, 0x0008u, 0x0008u, 0x0001u, 0x000Eu, 0x000Eu
+	};
+
+	//Only check default vehicle; it's assumed the others will have correct sprites if this one does (I've yet to find an exception, at least)
+	for (int j = 0; j < 55; j++) {
+		if ((rideEntry->vehicles[rideEntry->default_vehicle].sprite_flags & trackPieceRequiredSprites[j]) != trackPieceRequiredSprites[j])
+			supportedPieces &= ~(1ULL << j);
+	}
+
+	return supportedPieces;
+}
+
 static int ride_get_smallest_station_length(rct_ride *ride)
 {
 	uint32 result = -1;
diff --git a/src/ride/ride.h b/src/ride/ride.h
index 63863f0dc2..0c5dd75b8d 100644
--- a/src/ride/ride.h
+++ b/src/ride/ride.h
@@ -1123,6 +1123,7 @@ void ride_fix_breakdown(int rideIndex, int reliabilityIncreaseFactor);
 void ride_entry_get_train_layout(int rideEntryIndex, int numCarsPerTrain, uint8 *trainLayout);
 uint8 ride_entry_get_vehicle_at_position(int rideEntryIndex, int numCarsPerTrain, int position);
 void ride_update_max_vehicles(int rideIndex);
+uint64 ride_entry_get_supported_track_pieces(rct_ride_entry* rideEntry);
 
 void ride_set_ride_entry(int rideIndex, int rideEntry);
 void ride_set_num_vehicles(int rideIndex, int numVehicles);