X-Git-Url: http://git.tdb.fi/?p=r2c2.git;a=blobdiff_plain;f=source%2F3d%2Fallocation.h;h=13a6112882295e9362e0ad630bfbf2b2fbb9dc40;hp=87ea3096be28c6fe19faea8aa02647d30fce5941;hb=9b8a32cf7371bc5fae8ce239a212469511c459e6;hpb=aa2c48c635ad0be4eb465a72861a8d5403647b0a

diff --git a/source/3d/allocation.h b/source/3d/allocation.h
index 87ea309..13a6112 100644
--- a/source/3d/allocation.h
+++ b/source/3d/allocation.h
@@ -24,8 +24,9 @@ public:
 	Allocation3D(Layout3D &, Train &);
 
 	void set_color(const Msp::GL::Color &);
-
 private:
+	Msp::GL::Color get_color_for_block(const Block &) const;
+
 	void block_reserved(Block &, Train *);
 	void train_advanced(Block &);
 };