OpenGL_Animation/rolling_ball.py at main · devaprasadnm/OpenGL_Animation · 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
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160

from OpenGL.GL import *
from OpenGL.GLUT import *
from OpenGL.GLU import *
import math


WINDOW_SIZE = 1000
GLOBAL_X_POSTION = -500
GLOBAL_Y_POSTION = 0
TARGET_FPS = 60
STATE = 1
SPEED = 4
ANGLE = 0
THETA = 0
choice = 0
X1_GROUND = -1000
Y1_GROUND = 0
X2_GROUND = 1000
Y2_GROUND = 0


BALL_RADIUS = float(input("Ball Radius: "))

TOP_X = GLOBAL_X_POSTION
TOP_Y = GLOBAL_Y_POSTION + BALL_RADIUS
BOTTOM_X = GLOBAL_X_POSTION
BOTTOM_Y = GLOBAL_Y_POSTION - BALL_RADIUS


def init():
    glClearColor(0.0, 0.0, 0.0, 1.0)
    gluOrtho2D(-WINDOW_SIZE, WINDOW_SIZE, -WINDOW_SIZE, WINDOW_SIZE)


def update(value):
    global GLOBAL_X_POSTION, GLOBAL_Y_POSTION, STATE, WINDOW_SIZE, SPEED, ANGLE, THETA, TOP_X, TOP_Y, BOTTOM_X, BOTTOM_Y
    glutPostRedisplay()
    glutTimerFunc(int(1000/TARGET_FPS), update, int(0))

    if(STATE == 1):
        if(GLOBAL_X_POSTION < WINDOW_SIZE and GLOBAL_Y_POSTION > -WINDOW_SIZE):
            GLOBAL_X_POSTION = SPEED+GLOBAL_X_POSTION
            GLOBAL_Y_POSTION = GLOBAL_Y_POSTION-(SPEED*math.tan(ANGLE))
            TOP_X = GLOBAL_X_POSTION
            TOP_Y = GLOBAL_Y_POSTION + BALL_RADIUS
            BOTTOM_X = GLOBAL_X_POSTION
            BOTTOM_Y = GLOBAL_Y_POSTION - BALL_RADIUS
            THETA = THETA-1
            TOP_X, TOP_Y, BOTTOM_X, BOTTOM_Y = rotation(
                TOP_X, TOP_Y, BOTTOM_X, BOTTOM_Y, THETA, GLOBAL_X_POSTION, GLOBAL_Y_POSTION)
        else:
            STATE = -1
    elif(STATE == -1):
        if(GLOBAL_X_POSTION > -WINDOW_SIZE and GLOBAL_Y_POSTION < WINDOW_SIZE):
            GLOBAL_X_POSTION = -SPEED+GLOBAL_X_POSTION
            GLOBAL_Y_POSTION = GLOBAL_Y_POSTION-(-SPEED*math.tan(ANGLE))
            TOP_X = GLOBAL_X_POSTION
            TOP_Y = GLOBAL_Y_POSTION + BALL_RADIUS
            BOTTOM_X = GLOBAL_X_POSTION
            BOTTOM_Y = GLOBAL_Y_POSTION - BALL_RADIUS
            THETA = THETA+1
            TOP_X, TOP_Y, BOTTOM_X, BOTTOM_Y = rotation(
                TOP_X, TOP_Y, BOTTOM_X, BOTTOM_Y, THETA, GLOBAL_X_POSTION, GLOBAL_Y_POSTION)
        else:
            STATE = 1

# Here is my keyboard input code


def buttons(key, x, y):
    global SPEED, STATE
    if key == b'f':
        SPEED = SPEED+1
        print(SPEED)
    if key == b's':
        SPEED = SPEED-1
        print(SPEED)
    if key == b'r':
        STATE = 1
        print(SPEED)
    if key == b'l':
        STATE = -1
        print(SPEED)
    glutPostRedisplay()


def rotation(x1, y1, x2, y2, angle, xr, yr):
    cosT = math.cos(angle)
    sinT = math.sin(angle)
    X1 = (x1-xr)*cosT + (y1-yr)*sinT + xr
    Y1 = -(x1-xr)*sinT + (y1-yr)*cosT + yr
    X2 = (x2-xr)*cosT + (y2-yr)*sinT + xr
    Y2 = -(x2-xr)*sinT + (y2-yr)*cosT + yr
    rgb = (0, 1, 0)
    return(X1, Y1, X2, Y2)


def drawCircle(x, y):
    i = 0.0
    glBegin(GL_TRIANGLE_FAN)
    glVertex2f(x, y)

    while(i <= 360):

        glVertex2f(BALL_RADIUS*math.sin(math.pi * i / 180.0) + GLOBAL_X_POSTION,
                   BALL_RADIUS*math.cos(math.pi * i / 180.0) + GLOBAL_Y_POSTION)
        i = i+1.0

    glEnd()

    glBegin(GL_LINES)
    glColor3f(1.0, 0.0, 0.0)
    glVertex2f(TOP_X, TOP_Y)
    glVertex2f(BOTTOM_X, BOTTOM_Y)
    glEnd()


def drawBall():
    glClear(GL_COLOR_BUFFER_BIT)
    glColor3f(1.0, 1.0, 0.0)
    drawCircle(GLOBAL_X_POSTION, GLOBAL_Y_POSTION)
    draw_ground()
    glutSwapBuffers()


def draw_ground():
    glColor3f(1.0, 0.0, 0.0)
    glPointSize(10.0)
    glBegin(GL_LINES)
    glVertex2f(X1_GROUND, Y1_GROUND)
    glVertex2f(X2_GROUND, Y2_GROUND)
    glEnd()
    glFlush()


def main():
    global choice, ANGLE, GLOBAL_X_POSTION, GLOBAL_Y_POSTION, X1_GROUND, Y1_GROUND, X2_GROUND, Y2_GROUND

    glutInit(sys.argv)
    glutInitDisplayMode(GLUT_RGB | GLUT_DOUBLE)
    glutInitWindowSize(500, 500)
    glutInitWindowPosition(0, 0)

    glutCreateWindow("Rolling Ball About Inclined Plane")
    ANGLE = int(input('angle='))
    ANGLE = (ANGLE*(22/7))/180
    X1_GROUND, Y1_GROUND, X2_GROUND, Y2_GROUND = rotation(
        -2828.5, 0, 2828.5, 0, ANGLE, 0, 0)
    GLOBAL_Y_POSTION = 0+(BALL_RADIUS/math.cos(ANGLE))
    GLOBAL_X_POSTION = 0
    glutDisplayFunc(drawBall)
    glutTimerFunc(0, update, 0)
    glutIdleFunc(drawBall)
    glutKeyboardFunc(buttons)
    init()
    glutMainLoop()


main()