X-Git-Url: http://git.mutantstargoat.com/user/nuclear/?p=laserbrain_demo;a=blobdiff_plain;f=src%2Fmachine.cc;h=441826e4ca08b21bc5824450e25fc2545f54e3c8;hp=3d4dd73fdc070f9ff03ad7ed1f947e8ec473f733;hb=21d0490039c490c6abcc80ebf2142a3376f2be43;hpb=ccc1a688b59e25bb934a0d3e2bbf477960068d4f

diff --git a/src/machine.cc b/src/machine.cc
index 3d4dd73..441826e 100644
--- a/src/machine.cc
+++ b/src/machine.cc
@@ -47,6 +47,15 @@ void Machine::add_motor(int gearidx, float speed_hz)
 	motors.push_back(m);
 }
 
+int Machine::get_gear_index(Gear *g) const
+{
+	std::map<Gear*, int>::const_iterator it = gearidx.find(g);
+	if(it == gearidx.end()) {
+		return -1;
+	}
+	return it->second;
+}
+
 void Machine::invalidate_meshing()
 {
 	meshing_valid = false;