tron@2186: /* $Id$ */
tron@2186: 
maedhros@7271: /** @file articulated_vehicles.cpp */
belugas@6918: 
truelight@0: #include "stdafx.h"
Darkvater@1891: #include "openttd.h"
maedhros@7268: #include "articulated_vehicles.h"
truelight@0: #include "engine.h"
bjarni@2676: #include "train.h"
maedhros@7353: #include "roadveh.h"
peter1138@2982: #include "newgrf_callbacks.h"
peter1138@2962: #include "newgrf_engine.h"
rubidium@8640: #include "vehicle_func.h"
truelight@0: 
maedhros@7353: 
rubidium@8091: uint CountArticulatedParts(EngineID engine_type, bool purchase_window)
peter1138@2602: {
skidd13@8424: 	if (!HasBit(EngInfo(engine_type)->callbackmask, CBM_VEHICLE_ARTIC_ENGINE)) return 0;
tron@6476: 
rubidium@8091: 	Vehicle *v = NULL;;
rubidium@8091: 	if (!purchase_window) {
rubidium@8091: 		v = new InvalidVehicle();
rubidium@8091: 		v->engine_type = engine_type;
rubidium@8091: 	}
rubidium@8091: 
peter1138@2602: 	uint i;
rubidium@8089: 	for (i = 1; i < MAX_UVALUE(EngineID); i++) {
rubidium@8091: 		uint16 callback = GetVehicleCallback(CBID_VEHICLE_ARTIC_ENGINE, i, 0, engine_type, v);
rubidium@10330: 		if (callback == CALLBACK_FAILED || GB(callback, 0, 8) == 0xFF) break;
peter1138@2602: 	}
peter1138@2602: 
rubidium@8091: 	delete v;
rubidium@8091: 
peter1138@2608: 	return i - 1;
peter1138@2602: }
peter1138@2602: 
peter1138@9055: 
peter1138@9055: uint16 *GetCapacityOfArticulatedParts(EngineID engine, VehicleType type)
peter1138@9055: {
peter1138@9055: 	static uint16 capacity[NUM_CARGO];
peter1138@9055: 	memset(capacity, 0, sizeof(capacity));
peter1138@9055: 
peter1138@9055: 	if (type == VEH_TRAIN) {
peter1138@9055: 		const RailVehicleInfo *rvi = RailVehInfo(engine);
peter1138@9055: 		capacity[rvi->cargo_type] = rvi->capacity;
peter1138@9055: 		if (rvi->railveh_type == RAILVEH_MULTIHEAD) capacity[rvi->cargo_type] += rvi->capacity;
peter1138@9055: 	} else if (type == VEH_ROAD) {
peter1138@9055: 		const RoadVehicleInfo *rvi = RoadVehInfo(engine);
peter1138@9055: 		capacity[rvi->cargo_type] = rvi->capacity;
peter1138@9055: 	}
peter1138@9055: 
peter1138@9055: 	if (!HasBit(EngInfo(engine)->callbackmask, CBM_VEHICLE_ARTIC_ENGINE)) return capacity;
peter1138@9055: 
peter1138@9055: 	for (uint i = 1; i < MAX_UVALUE(EngineID); i++) {
peter1138@9055: 		uint16 callback = GetVehicleCallback(CBID_VEHICLE_ARTIC_ENGINE, i, 0, engine, NULL);
rubidium@10330: 		if (callback == CALLBACK_FAILED || GB(callback, 0, 8) == 0xFF) break;
peter1138@9055: 
peter1138@9055: 		EngineID artic_engine = GetFirstEngineOfType(type) + GB(callback, 0, 7);
peter1138@9055: 
peter1138@9055: 		if (type == VEH_TRAIN) {
peter1138@9055: 			const RailVehicleInfo *rvi = RailVehInfo(artic_engine);
peter1138@9055: 			capacity[rvi->cargo_type] += GetEngineProperty(artic_engine, 0x14, rvi->capacity);
peter1138@9055: 		} else if (type == VEH_ROAD) {
peter1138@9055: 			const RoadVehicleInfo *rvi = RoadVehInfo(artic_engine);
peter1138@9055: 			capacity[rvi->cargo_type] += GetEngineProperty(artic_engine, 0x0F, rvi->capacity);
peter1138@9055: 		}
peter1138@9055: 	}
peter1138@9055: 
peter1138@9055: 	return capacity;
peter1138@9055: }
peter1138@9055: 
peter1138@9055: 
maedhros@7353: void AddArticulatedParts(Vehicle **vl, VehicleType type)
peter1138@2602: {
tron@6476: 	const Vehicle *v = vl[0];
tron@6476: 	Vehicle *u = vl[0];
peter1138@2602: 
skidd13@8424: 	if (!HasBit(EngInfo(v->engine_type)->callbackmask, CBM_VEHICLE_ARTIC_ENGINE)) return;
peter1138@2602: 
rubidium@8089: 	for (uint i = 1; i < MAX_UVALUE(EngineID); i++) {
rubidium@7711: 		uint16 callback = GetVehicleCallback(CBID_VEHICLE_ARTIC_ENGINE, i, 0, v->engine_type, v);
rubidium@10330: 		if (callback == CALLBACK_FAILED || GB(callback, 0, 8) == 0xFF) return;
peter1138@2602: 
peter1138@4831: 		/* Attempt to use pre-allocated vehicles until they run out. This can happen
peter1138@4831: 		 * if the callback returns different values depending on the cargo type. */
rubidium@7988: 		u->SetNext(vl[i]);
rubidium@7988: 		if (u->Next() == NULL) return;
peter1138@4831: 
rubidium@8014: 		Vehicle *previous = u;
rubidium@7988: 		u = u->Next();
peter1138@2602: 
maedhros@7353: 		EngineID engine_type = GetFirstEngineOfType(type) + GB(callback, 0, 7);
skidd13@8424: 		bool flip_image = HasBit(callback, 7);
peter1138@2602: 
belugas@6918: 		/* get common values from first engine */
peter1138@2602: 		u->direction = v->direction;
peter1138@2602: 		u->owner = v->owner;
peter1138@2602: 		u->tile = v->tile;
peter1138@2602: 		u->x_pos = v->x_pos;
peter1138@2602: 		u->y_pos = v->y_pos;
peter1138@2602: 		u->z_pos = v->z_pos;
peter1138@2602: 		u->build_year = v->build_year;
peter1138@2602: 		u->vehstatus = v->vehstatus & ~VS_STOPPED;
peter1138@2602: 
peter1138@3870: 		u->cargo_subtype = 0;
peter1138@2602: 		u->max_speed = 0;
peter1138@2602: 		u->max_age = 0;
peter1138@2602: 		u->engine_type = engine_type;
peter1138@2602: 		u->value = 0;
bjarni@2676: 		u->subtype = 0;
peter1138@2602: 		u->cur_image = 0xAC2;
peter1138@2804: 		u->random_bits = VehicleRandomBits();
peter1138@2602: 
maedhros@7353: 		switch (type) {
maedhros@7353: 			default: NOT_REACHED();
maedhros@7353: 
maedhros@7353: 			case VEH_TRAIN: {
maedhros@7353: 				const RailVehicleInfo *rvi_artic = RailVehInfo(engine_type);
maedhros@7353: 
maedhros@7353: 				u = new (u) Train();
rubidium@8024: 				previous->SetNext(u);
maedhros@7353: 				u->u.rail.track = v->u.rail.track;
maedhros@7353: 				u->u.rail.railtype = v->u.rail.railtype;
maedhros@7353: 				u->u.rail.first_engine = v->engine_type;
maedhros@7353: 
maedhros@7353: 				u->spritenum = rvi_artic->image_index;
maedhros@7353: 				u->cargo_type = rvi_artic->cargo_type;
maedhros@7353: 				u->cargo_cap = rvi_artic->capacity;
maedhros@7353: 
maedhros@7353: 				SetArticulatedPart(u);
maedhros@7353: 			} break;
maedhros@7353: 
maedhros@7353: 			case VEH_ROAD: {
maedhros@7353: 				const RoadVehicleInfo *rvi_artic = RoadVehInfo(engine_type);
maedhros@7353: 
maedhros@7353: 				u = new (u) RoadVehicle();
rubidium@8024: 				previous->SetNext(u);
maedhros@7353: 				u->u.road.first_engine = v->engine_type;
maedhros@7353: 				u->u.road.cached_veh_length = GetRoadVehLength(u);
maedhros@7353: 				u->u.road.state = RVSB_IN_DEPOT;
maedhros@7353: 
maedhros@7353: 				u->u.road.roadtype = v->u.road.roadtype;
maedhros@7353: 				u->u.road.compatible_roadtypes = v->u.road.compatible_roadtypes;
maedhros@7353: 
maedhros@7353: 				u->spritenum = rvi_artic->image_index;
maedhros@7353: 				u->cargo_type = rvi_artic->cargo_type;
maedhros@7353: 				u->cargo_cap = rvi_artic->capacity;
maedhros@7353: 
maedhros@7353: 				SetRoadVehArticPart(u);
maedhros@7353: 			} break;
maedhros@7353: 		}
maedhros@7353: 
maedhros@7353: 		if (flip_image) u->spritenum++;
maedhros@7353: 
peter1138@2602: 		VehiclePositionChanged(u);
peter1138@2602: 	}
peter1138@2602: }