From a8cf0c8c524ecfbf69cde72f1f194bba41484dda Mon Sep 17 00:00:00 2001
From: nrp <olpc@spongezone.net>
Date: Sun, 31 Aug 2008 15:49:07 +0000
Subject: fixing train addition bug, colors

---
(limited to 'bridge.py')

diff --git a/bridge.py b/bridge.py
index 45b718a..f389206 100644
--- a/bridge.py
+++ b/bridge.py
@@ -73,12 +73,15 @@ class Bridge:
             
             rearaxle = self.world.get_bodies_at_pos(rearwheel)
             frontaxle = self.world.get_bodies_at_pos(frontwheel)
-            self.world.add.jointMotor(rearaxle[0],rearaxle[1],rearwheel)
-            self.world.add.jointMotor(frontaxle[0],frontaxle[1],frontwheel)
+            if len(rearaxle) == 2:
+                self.world.add.jointMotor(rearaxle[0],rearaxle[1],rearwheel)
+            if len(frontaxle) == 2:
+                self.world.add.jointMotor(frontaxle[0],frontaxle[1],frontwheel)
             
         for i in range(1,len(points)):
             backlink = (points[i][0]+train[0]-1,points[i][1]+train[1]-1)
             frontlink = (points[i-1][0]+1,points[i-1][1]+train[1]-1)
             btrain = self.world.get_bodies_at_pos(backlink)
-            ftrain = self.world.get_bodies_at_pos(frontlink)
-            self.world.add.distanceJoint(btrain[0], ftrain[0], backlink, frontlink)
+            ftrain = self.world.get_bodies_at_pos(frontlink)
+            if len(ftrain) and len(btrain):
+                self.world.add.distanceJoint(btrain[0], ftrain[0], backlink, frontlink)
--
cgit v0.9.1