X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Flibr2c2%2Ftrain.h;h=461a27e583146bde1080e64acbb6384b0c33a3e6;hb=bde934f7d5ba07a9442a305192e10024820d13ed;hp=fa638f8d5034390b25388a1fa06b7e125538ef80;hpb=5b1b2196908c159d4a1339bce2682afa22700c35;p=r2c2.git

diff --git a/source/libr2c2/train.h b/source/libr2c2/train.h
index fa638f8..461a27e 100644
--- a/source/libr2c2/train.h
+++ b/source/libr2c2/train.h
@@ -109,9 +109,17 @@ public:
 
 	void add_ai(TrainAI &);
 	void remove_ai(TrainAI &);
-	TrainAI *get_tagged_ai(const std::string &) const;
 	void ai_message(const TrainAI::Message &);
 
+	template<typename T>
+	T *get_ai_of_type() const
+	{
+		for(std::list<TrainAI *>::const_iterator i=ais.begin(); i!=ais.end(); ++i)
+			if(T *ai = dynamic_cast<T *>(*i))
+				return ai;
+		return 0;
+	}
+
 	void place(Block &, unsigned);
 	void unplace();
 	bool is_placed() const { return !blocks.empty(); }