python-projects/5_bouncing_dvd.py at main · sunil-dhaka/python-projects · GitHub

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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
import sys, time, random

try:
    import bext
except ImportError:
    print('get bext module first')
    exit()

WIDTH,HEIGHT=bext.size()

##
WIDTH-=1 #??
##
NO_OF_DVDS=10
PAUSE_TIME=0.1

COLORS = ['red', 'green', 'yellow', 'blue', 'magenta', 'cyan', 'white']

UP_RIGHT='ur'
UP_LEFT='ul'
DOWN_RIGHT='dr'
DOWN_LEFT='dl'

DIRECTIONS=(UP_RIGHT,UP_LEFT,DOWN_RIGHT,DOWN_LEFT)

COLOR='color'
X='x'
Y='y'
DIR='direction'

def main():

    # clear screen before starting bounce
    bext.clear()
    # create dvd logos
    dvd_logos=[]
    for i in range(NO_OF_DVDS):
        dvd_item={
            COLOR:random.choice(COLORS),
            X:random.randint(1,WIDTH-4), # why -4??
            Y:random.randint(1,HEIGHT-4),
            DIR:random.choice(DIRECTIONS)
        }
        # Make sure X is even so it can hit the corner.
        if dvd_item[X]%2==1:
            dvd_item[X]-=1
            dvd_logos.append(dvd_item)
        else:
            dvd_logos.append(dvd_item)

    # create bounce
    corner_bounce_count=0
    while True:
        '''
        WIDTH - 3 because 'DVD' has 3 letters
        '''
        for logo in dvd_logos:
            # go to logos current location
            bext.goto(logo[X],logo[Y])
            print('    ',end='')#??

            original_dir=logo[DIR]

            # corner cases
            if logo[X]==0 and logo[Y]==0:
                logo[DIR]=DOWN_RIGHT
                corner_bounce_count+=1
            elif logo[X]==0 and logo[Y]==HEIGHT-1:
                logo[DIR]=UP_RIGHT
                corner_bounce_count+=1
            elif logo[X]==WIDTH-3 and logo[Y]==0:
                logo[DIR]=DOWN_LEFT
                corner_bounce_count+=1
            elif logo[X]==WIDTH-3 and logo[Y]==HEIGHT-1:
                logo[DIR]=UP_LEFT
                corner_bounce_count+=1

            # right edge
            elif logo[X]==WIDTH-3 and logo[DIR]==DOWN_RIGHT:
                logo[DIR]=DOWN_LEFT
            elif logo[X]==WIDTH-3 and logo[DIR]==UP_RIGHT:
                logo[DIR]=UP_LEFT

            # left edge
            elif logo[X]==0 and logo[DIR]==DOWN_LEFT:
                logo[DIR]=DOWN_RIGHT
            elif logo[X]==0 and logo[DIR]==UP_LEFT:
                logo[DIR]=UP_RIGHT

            # upper edge
            elif logo[Y]==0 and logo[DIR]==UP_RIGHT:
                logo[DIR]=DOWN_RIGHT
            elif logo[Y]==0 and logo[DIR]==UP_LEFT:
                logo[DIR]=DOWN_LEFT

            # bottom edge
            elif logo[Y]==HEIGHT-1 and logo[DIR]==DOWN_RIGHT:
                logo[DIR]=UP_RIGHT
            elif logo[Y]==HEIGHT-1 and logo[DIR]==DOWN_LEFT:
                logo[DIR]=UP_LEFT

            # Change color when the logo bounces:
            if logo[DIR]!=original_dir:
                logo[COLOR]=random.choice(COLORS)

            # move logos
            if logo[DIR]==UP_RIGHT:
                logo[X]+=2
                logo[Y]-=1
            elif logo[DIR]==UP_LEFT:
                logo[X]-=2
                logo[Y]-=1
            elif logo[DIR]==DOWN_LEFT:
                logo[X]-=2
                logo[Y]+=1
            elif logo[DIR]==DOWN_RIGHT:
                logo[X]+=2
                logo[Y]+=1

        # show how many times corner bounces
        bext.goto(5,0)
        bext.fg('blue')
        print('Corner Bounces:',corner_bounce_count,end='')

        # draw logo at new location
        for logo in dvd_logos:
            bext.goto(logo[X],logo[Y])
            bext.fg(logo[COLOR])
            print('DVD',end='')
        bext.goto(0,0)
        # (Required for bext-using programs.)
        sys.stdout.flush()
        time.sleep(PAUSE_TIME)
if __name__=='__main__':
    try:
        main()
    except:
        # when pressed Ctrl-C exit
        print('Stopped DVD bouncing.')
        exit()