X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Flibr2c2%2Ftrain.cpp;h=02e9d31797e2faf6737288ba68c38829b98a0bdb;hb=0c1214ad0680c3bf65c1ee5f366dbafabc9d188d;hp=6cb3205eb175fad4f6e6d8b18997a725df2da7b5;hpb=57e0944ee8ed442e2d0aa7c9964cac00450af5e7;p=r2c2.git

diff --git a/source/libr2c2/train.cpp b/source/libr2c2/train.cpp
index 6cb3205..02e9d31 100644
--- a/source/libr2c2/train.cpp
+++ b/source/libr2c2/train.cpp
@@ -6,6 +6,7 @@
 #include <msp/time/units.h>
 #include <msp/time/utils.h>
 #include "aicontrol.h"
+#include "beamgate.h"
 #include "block.h"
 #include "catalogue.h"
 #include "driver.h"
@@ -53,6 +54,7 @@ Train::Train(Layout &l, const VehicleType &t, unsigned a, const string &p):
 		speed_quantizer = new SpeedQuantizer(speed_steps);
 
 	vehicles.push_back(new Vehicle(layout, loco_type));
+	vehicles.back()->set_train(this);
 
 	layout.add_train(*this);
 
@@ -87,6 +89,8 @@ void Train::add_vehicle(const VehicleType &vt)
 	Vehicle *veh = new Vehicle(layout, vt);
 	vehicles.back()->attach_back(*veh);
 	vehicles.push_back(veh);
+	veh->set_train(this);
+	signal_vehicle_added.emit(vehicles.size()-1, *veh);
 }
 
 void Train::remove_vehicle(unsigned i)
@@ -95,10 +99,17 @@ void Train::remove_vehicle(unsigned i)
 		throw out_of_range("Train::remove_vehicle");
 	if(i==0)
 		throw logic_error("can't remove locomotive");
-	delete vehicles[i];
+
+	Vehicle *veh = vehicles[i];
 	vehicles.erase(vehicles.begin()+i);
+	veh->detach_front();
 	if(i<vehicles.size())
+	{
+		veh->detach_back();
 		vehicles[i-1]->attach_back(*vehicles[i]);
+	}
+	signal_vehicle_removed.emit(i, *veh);
+	delete veh;
 }
 
 unsigned Train::get_n_vehicles() const
@@ -183,6 +194,7 @@ void Train::place(const BlockIter &block)
 
 	allocator.start_from(block);
 	accurate_position = false;
+	last_entry_block = BlockIter();
 
 	if(reverse)
 		vehicles.front()->place(block.reverse().track_iter(), 0, Vehicle::FRONT_BUFFER);
@@ -197,6 +209,7 @@ void Train::unplace()
 
 	allocator.clear();
 	accurate_position = false;
+	last_entry_block = BlockIter();
 
 	for(vector<Vehicle *>::iterator i=vehicles.begin(); i!=vehicles.end(); ++i)
 		(*i)->unplace();
@@ -299,6 +312,7 @@ void Train::tick(const Time::TimeDelta &dt)
 		driver.set_loco_reverse(address, r);
 
 		allocator.reverse();
+		last_entry_block = BlockIter();
 	}
 
 	if(speed_quantizer)
@@ -338,6 +352,8 @@ void Train::tick(const Time::TimeDelta &dt)
 			}
 		}
 	}
+	else if(intent_to_move && !allocator.is_active())
+		allocator.set_active(true);
 	else if(allocator.is_active() && !intent_to_move && !stop_timeout)
 		stop_timeout = 2*Time::sec;
 }
@@ -410,27 +426,28 @@ void Train::loco_func_event(unsigned addr, unsigned func, bool state)
 
 void Train::sensor_state_changed(Sensor &sensor, Sensor::State state)
 {
-	Block *block = 0;
-	if(TrackCircuit *tc = dynamic_cast<TrackCircuit *>(&sensor))
-		block = &tc->get_block();
-	else
+	if(!current_speed_step || state!=Sensor::MAYBE_ACTIVE)
+		return;
+
+	Block *block = sensor.get_block();
+	if(!block || block->get_train()!=this)
 		return;
 
-	if(block->get_train()==this && state==Sensor::MAYBE_ACTIVE)
+	if(last_entry_block && &*last_entry_block!=block)
 	{
-		if(last_entry_block)
+		for(BlockIter i=last_entry_block.next(); (i && &*i!=block); i=i.next())
+			if(i->get_train()!=this || i->get_sensor_id())
+				return;
+	}
+
+	if(dynamic_cast<TrackCircuit *>(&sensor))
+	{
+		if(last_entry_block && pure_speed && speed_quantizer)
 		{
-			float travel_distance = -1;
-			if(pure_speed && speed_quantizer && current_speed_step>0)
-				travel_distance = 0;
+			float travel_distance = 0;
 
 			for(BlockIter i=last_entry_block; &*i!=block; i=i.next())
-			{
-				if(i->get_sensor_id())
-					return;
-				if(travel_distance>=0)
-					travel_distance += i->get_path_length(i.entry());
-			}
+				travel_distance += i->get_path_length(i.entry());
 
 			if(travel_distance>0)
 			{
@@ -459,6 +476,25 @@ void Train::sensor_state_changed(Sensor &sensor, Sensor::State state)
 				vehicles.front()->place(track, 0, Vehicle::FRONT_AXLE);
 		}
 	}
+	else if(BeamGate *gate = dynamic_cast<BeamGate *>(&sensor))
+	{
+		if(!advancing && vehicles.front()->get_track())
+		{
+			TrackIter track = allocator.iter_for(*block).track_iter();
+			for(; (track && &track->get_block()==block); track=track.next())
+				if(track.track()==gate->get_track())
+				{
+					if(reverse)
+						track = track.reverse();
+					float offset = gate->get_offset_from_endpoint(track.entry());
+					if(reverse)
+						vehicles.back()->place(track, offset, Vehicle::BACK_BUFFER);
+					else
+						vehicles.front()->place(track, offset, Vehicle::FRONT_BUFFER);
+					break;
+				}
+		}
+	}
 }
 
 void Train::halt_event(bool h)
@@ -478,8 +514,7 @@ float Train::get_reserved_distance_until(const Block *until_block) const
 	if(!track)  // XXX Probably unnecessary
 		return 0;
 
-	BlockIter block = track.block_iter();
-	if(&*block==until_block)
+	if(&track->get_block()==until_block)
 		return 0;
 
 	// Account for the vehicle's offset on its current track
@@ -490,6 +525,8 @@ float Train::get_reserved_distance_until(const Block *until_block) const
 		result = track->get_type().get_path_length(track->get_active_path())-result;
 	result -= veh.get_type().get_length()/2;
 
+	BlockIter block = track.block_iter();
+
 	// Count remaining distance in the vehicle's current block
 	for(track=track.next(); &track->get_block()==&*block; track=track.next())
 		result += track->get_type().get_path_length(track->get_active_path());
@@ -561,6 +598,7 @@ void Train::Loader::vehicle(ArticleNumber art_nr)
 	Vehicle *veh = new Vehicle(obj.layout, vtype);
 	obj.vehicles.back()->attach_back(*veh);
 	obj.vehicles.push_back(veh);
+	veh->set_train(&obj);
 }
 
 } // namespace R2C2