X-Git-Url: http://git.mutantstargoat.com/user/nuclear/?a=blobdiff_plain;ds=sidebyside;f=src%2Fmachine.h;h=e0f2155b389906b157205ac62730594af468789f;hb=1548fea8c64ebdfde728f33fd2d1537954ebf38b;hp=5513e027264eb5573581799e0a4f0efd9d07a41b;hpb=2c8e92970f198061a0cefdb59c2d0ec8c58409c9;p=antikythera

diff --git a/src/machine.h b/src/machine.h
index 5513e02..e0f2155 100644
--- a/src/machine.h
+++ b/src/machine.h
@@ -2,6 +2,7 @@
 #define MACHINE_H_
 
 #include <vector>
+#include <map>
 #include "gear.h"
 
 struct Motor {
@@ -12,6 +13,7 @@ struct Motor {
 class Machine {
 private:
 	std::vector<Gear*> gears;
+	std::map<Gear*, int> gearidx;
 	bool **meshing;
 	bool meshing_valid;
 	bool *visited;	/* used for update_gear */
@@ -32,6 +34,8 @@ public:
 
 	void update(float dt);
 	void draw() const;
+
+	Gear *intersect_gear(const Ray &ray, HitPoint *hitp = 0) const;
 };
 
 #endif	// MACHINE_H_