a1q25_h_cos3x.py

from __future__ import division
from __future__ import print_function


import glfw
from OpenGL.GL import *
from OpenGL.GLU import *
import numpy as np
import math 


def key_callback(window, key, scancode, action, mode):
    if key == glfw.KEY_ESCAPE and action == glfw.PRESS:
        print("closed window for pressing Escape key")
        glfw.set_window_should_close(window, True)
def reshape_callback(window, width, height):
    glViewport(0,0,width, height)


def drawAxes():
    glColor3f(0,0,1)
    glBegin(GL_LINES)
    glVertex2f(-500,0)
    glVertex2f(1900,0)
    glVertex2f(0,-500)
    glVertex2f(0,500)
    glEnd()



def setpixel(x,y,color):
    glColor3fv(color)
    glPointSize(2.3)
    glBegin(GL_POINTS)
    glVertex2f(x,y)
    glEnd()




def main():
    if not glfw.init():
        raise Exception("glfw nt initialized")

    window=glfw.create_window(900,600, "B117049- Cosine 3x Wave",None,None)

    if not window:
        glfw.terminate()
        raise Exception("glfw window not created")

    w,h = glfw.get_framebuffer_size(window)
    print("width: {}, height:{}".format(w,h))

    glfw.set_window_pos(window, 350,85)

    glfw.make_context_current(window)
    glfw.set_key_callback(window, key_callback)

    glfw.set_window_size_callback(window, reshape_callback)

    gluOrtho2D(-400,3600/2,-450,450)

    while not glfw.window_should_close(window):

        glClear(GL_COLOR_BUFFER_BIT)
        glClearColor(1,1,1,1.0)
        drawAxes()


        for i in range(-400 ,1800,1):
            x = float(i); 
            y = 100.0 * math.cos(i *(6.284/360.0))*3;
            setpixel(x,y,(1,0,0))
   
         

        glfw.swap_buffers(window)
        glfw.poll_events()

    glfw.terminate()

if __name__ =="__main__":
    main()