Web   ·   Wiki   ·   Activities   ·   Blog   ·   Lists   ·   Chat   ·   Meeting   ·   Bugs   ·   Git   ·   Translate   ·   Archive   ·   People   ·   Donate
summaryrefslogtreecommitdiffstats
path: root/bridge.py
blob: bebd9e1b1f9836fb67861a5f21994ecea5d8afe1 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
import pygame

def create_world(game):
    rect = pygame.Rect((0,800), (350, -250))
    rect.normalize()
    pygame.draw.rect(game.screen, (100,180,255), rect, 3)
    game.world.add.rect(rect.center, rect.width / 2, rect.height / 2,
        dynamic=False)
    rect = pygame.Rect((1200,800), (-350, -250))
    rect.normalize()
    pygame.draw.rect(game.screen, (100,180,255), rect, 3)
    game.world.add.rect(rect.center, rect.width / 2, rect.height / 2,
        dynamic=False)

def create_train(game, worldpoint = (300,500), train = (100, 50), wheelrad = 20, cars = 3):
    points = []
    for i in range(0,cars):
        startpoint = (worldpoint[0]-(train[0]+7)*i, worldpoint[1])
        points.append(startpoint)
        rect = pygame.Rect(startpoint, train)
        rect.normalize()
        pygame.draw.rect(game.screen, (200, 50, 100), rect, 3)

        game.world.add.rect(rect.center, rect.width / 2, rect.height / 2,
            dynamic = True, density=1.0, restitution=0.16, friction=0.5)

        rearwheel = (startpoint[0]+wheelrad,startpoint[1]+train[1]-wheelrad/2)
        pygame.draw.circle(game.screen, (0,0,0), rearwheel, wheelrad, 3)
        game.world.add.ball(rearwheel,wheelrad, dynamic=True, density=1.0, 
            restitution=0.16, friction=0.5)

        frontwheel = (startpoint[0]+train[0]-wheelrad,startpoint[1]+train[1]-wheelrad/2)
        pygame.draw.circle(game.screen, (0,0,0), frontwheel, wheelrad, 3)
        game.world.add.ball(frontwheel,wheelrad, dynamic=True, density=1.0, 
            restitution=0.16, friction=0.5)

        rearaxle = game.world.get_bodies_at_pos(rearwheel)
        frontaxle = game.world.get_bodies_at_pos(frontwheel)
        game.world.add.jointMotor(rearaxle[0],rearaxle[1],rearwheel)
        game.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 = game.world.get_bodies_at_pos(backlink)
        ftrain = game.world.get_bodies_at_pos(frontlink)
        game.world.add.distanceJoint(btrain[0], ftrain[0], backlink, frontlink)