summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--train_cmd.c7
1 files changed, 4 insertions, 3 deletions
diff --git a/train_cmd.c b/train_cmd.c
index bf8b29629..7becdb023 100644
--- a/train_cmd.c
+++ b/train_cmd.c
@@ -59,7 +59,7 @@ static void TrainCargoChanged(Vehicle* v)
// powered wagons have extra weight added
if (HASBIT(u->u.rail.flags, VRF_POWEREDWAGON))
- vweight += RailVehInfo(v->engine_type)->pow_wag_weight;
+ vweight += RailVehInfo(u->u.rail.first_engine)->pow_wag_weight;
}
// consist weight is the sum of the weight of all vehicles in the consist
@@ -79,7 +79,6 @@ static void TrainCargoChanged(Vehicle* v)
*/
void TrainPowerChanged(Vehicle* v)
{
- const RailVehicleInfo *rvi_v = RailVehInfo(v->engine_type);
Vehicle* u;
uint32 power = 0;
@@ -106,7 +105,7 @@ void TrainPowerChanged(Vehicle* v)
if (engine_has_power) power += rvi_u->power;
if (HASBIT(u->u.rail.flags, VRF_POWEREDWAGON) && (wagon_has_power)) {
- power += rvi_v->pow_wag_power;
+ power += RailVehInfo(u->u.rail.first_engine)->pow_wag_power;
}
}
@@ -149,6 +148,8 @@ void TrainConsistChanged(Vehicle* v)
u->u.rail.first_engine = (v == u) ? INVALID_ENGINE : first_engine;
u->u.rail.railtype = GetEngine(u->engine_type)->railtype;
+ if (IsTrainEngine(u)) first_engine = u->engine_type;
+
if (rvi_u->visual_effect != 0) {
u->u.rail.cached_vis_effect = rvi_u->visual_effect;
} else {