CSC111 A class for a simple car
--D. Thiebaut 12:51, 5 April 2010 (UTC)
wheel1.py
# wheel1.py
# first program using a class to define
# a new graphic object.
from graphics import *
W = 400
H = 400
#----------------------------------------------------------------
class Wheel:
"""A class with two concentric circles"""
def __init__( self, center, r1, r2 ):
self.circ1 = Circle( center, r1 )
self.circ2 = Circle( center, r2 )
r1, r2 = min( r1, r2 ), max( r1, r2 )
self.radius1 = r1
self.radius2 = r2
def draw( self, win ):
self.circ2.draw( win )
self.circ1.draw( win )
def getRadius1( self ):
return self.radius1
def getRadius2( self ):
return self.radius2
#----------------------------------------------------------------
def waitForClick( win, message ):
""" waitForClick: stops the GUI and displays a message.
Returns when the user clicks the window. The message is erased."""
# wait for user to click mouse to start
startMsg = Text( Point( win.getWidth()/2, win.getHeight()/2 ), message )
startMsg.draw( win ) # display message
win.getMouse() # wait
startMsg.undraw() # erase
def main():
global W, H
win = GraphWin( "wheel demo", W, H )
waitForClick( win, "click to start" )
w = Wheel( Point( W/2, H/2 ), 20, 40 )
w.draw( win )
waitForClick( win, "click to end" )
main()
wheel2.py
# wheel2.py
# Same program as wheel1.py, but this time we add
# a method to define the color of the wheel.
from graphics import *
W = 400
H = 400
#----------------------------------------------------------------
class Wheel:
"""A class with two concentric circles"""
def __init__( self, center, r1, r2 ):
self.circ1 = Circle( center, r1 )
self.circ2 = Circle( center, r2 )
r1, r2 = min( r1, r2 ), max( r1, r2 )
self.radius1 = r1
self.radius2 = r2
def draw( self, win ):
self.circ2.draw( win )
self.circ1.draw( win )
def setFill( self, color1, color2 ):
self.circ1.setFill( color1 )
self.circ2.setFill( color2 )
def getRadius1( self ):
return self.radius1
def getRadius2( self ):
return self.radius2
#----------------------------------------------------------------
def waitForClick( win, message ):
""" waitForClick: stops the GUI and displays a message.
Returns when the user clicks the window. The message is erased."""
# wait for user to click mouse to start
startMsg = Text( Point( win.getWidth()/2, win.getHeight()/2 ), message )
startMsg.draw( win ) # display message
win.getMouse() # wait
startMsg.undraw() # erase
def main():
global W, H
win = GraphWin( "wheel demo", W, H )
waitForClick( win, "click to start" )
w = Wheel( Point( W/2, H/2 ), 20, 40 )
w.draw( win )
w.setFill( "yellow", "black" )
waitForClick( win, "click to end" )
main()
wheel3.py
# wheel3.py
# A new method is added to wheel2.py so that the
# wheel can now move with a dx, dy speed and
# direction.
from graphics import *
W = 400
H = 400
#----------------------------------------------------------------
class Wheel:
"""A class with two concentric circles"""
def __init__( self, center, r1, r2 ):
self.circ1 = Circle( center, r1 )
self.circ2 = Circle( center, r2 )
r1, r2 = min( r1, r2 ), max( r1, r2 )
self.radius1 = r1
self.radius2 = r2
def draw( self, win ):
self.circ2.draw( win )
self.circ1.draw( win )
def setFill( self, color1, color2 ):
self.circ1.setFill( color1 )
self.circ2.setFill( color2 )
def move( self, dx, dy ):
self.circ1.move( dx, dy )
self.circ2.move( dx, dy )
def getRadius1( self ):
return self.radius1
def getRadius2( self ):
return self.radius2
#----------------------------------------------------------------
def waitForClick( win, message ):
""" waitForClick: stops the GUI and displays a message.
Returns when the user clicks the window. The message is erased."""
# wait for user to click mouse to start
startMsg = Text( Point( win.getWidth()/2, win.getHeight()/2 ), message )
startMsg.draw( win ) # display message
win.getMouse() # wait
startMsg.undraw() # erase
def main():
global W, H
win = GraphWin( "wheel demo", W, H )
waitForClick( win, "click to start" )
w = Wheel( Point( W/2, H/2 ), 20, 40 )
w.draw( win )
w.setFill( "yellow", "black" )
waitForClick( win, "click to move" )
for i in range( 300 ):
w.move( 5, 3 )
waitForClick( win, "click to end" )
win.close()
main()
car1.py
See last version of car.py for full documentation.
# car1.py
# We add a rectangle and two wheels to create a new class for a
# car!
from graphics import *
W = 400
H = 400
#----------------------------------------------------------------
class Wheel:
"""A class with two concentric circles"""
def __init__( self, center, r1, r2 ):
self.circ1 = Circle( center, r1 )
self.circ2 = Circle( center, r2 )
r1, r2 = min( r1, r2 ), max( r1, r2 )
self.radius1 = r1
self.radius2 = r2
def draw( self, win ):
self.circ2.draw( win )
self.circ1.draw( win )
def setFill( self, color1, color2 ):
self.circ1.setFill( color1 )
self.circ2.setFill( color2 )
def getRadius1( self ):
return self.radius1
def getRadius2( self ):
return self.radius2
#----------------------------------------------------------------
class Car:
def __init__( self, P1, P2 ):
self.P1 = P1
self.P2 = P2
self.width = abs( P1.getX()-P2.getX() )
self.height= abs( P1.getY()-P2.getY() )
self.rect = Rectangle( P1, P2 )
center1 = Point( P1.getX()+self.width/8, P2.getY() )
center2 = Point( P1.getX()+self.width*7/8, P2.getY() )
radius2 = self.height/3
radius1 = radius2/2
self.wheel1 = Wheel( center1, radius1, radius2 )
self.wheel2 = Wheel( center2, radius1, radius2 )
def draw( self, win ):
self.rect.draw( win )
self.wheel1.draw( win )
self.wheel2.draw( win )
#----------------------------------------------------------------
def waitForClick( win, message ):
""" waitForClick: stops the GUI and displays a message.
Returns when the user clicks the window. The message is erased."""
# wait for user to click mouse to start
startMsg = Text( Point( win.getWidth()/2, win.getHeight()/2 ), message )
startMsg.draw( win ) # display message
win.getMouse() # wait
startMsg.undraw() # erase
def main():
global W, H
win = GraphWin( "wheel demo", W, H )
waitForClick( win, "click to start" )
car = Car( Point( 100,100 ), Point( 250, 170 ) )
car.draw( win )
waitForClick( win, "click to end" )
main()
car2.py
# car2.py
# Same as car1, but we add a method to define the color of
# the car, including the wheels
#
from graphics import *
W = 400
H = 400
#----------------------------------------------------------------
class Wheel:
"""A class with two concentric circles"""
def __init__( self, center, r1, r2 ):
self.circ1 = Circle( center, r1 )
self.circ2 = Circle( center, r2 )
r1, r2 = min( r1, r2 ), max( r1, r2 )
self.radius1 = r1
self.radius2 = r2
def draw( self, win ):
self.circ2.draw( win )
self.circ1.draw( win )
def setFill( self, color1, color2 ):
self.circ1.setFill( color1 )
self.circ2.setFill( color2 )
def getRadius1( self ):
return self.radius1
def getRadius2( self ):
return self.radius2
#----------------------------------------------------------------
class Car:
def __init__( self, P1, P2 ):
self.P1 = P1
self.P2 = P2
self.width = abs( P1.getX()-P2.getX() )
self.height= abs( P1.getY()-P2.getY() )
self.rect = Rectangle( P1, P2 )
center1 = Point( P1.getX()+self.width/8, P2.getY() )
center2 = Point( P1.getX()+self.width*7/8, P2.getY() )
radius2 = self.height/3
radius1 = radius2/2
self.wheel1 = Wheel( center1, radius1, radius2 )
self.wheel2 = Wheel( center2, radius1, radius2 )
def draw( self, win ):
self.rect.draw( win )
self.wheel1.draw( win )
self.wheel2.draw( win )
def setFill( self, color1, color2, color3 ):
self.rect.setFill( color1 )
self.wheel1.setFill( color2, color3 )
self.wheel2.setFill( color2, color3 )
#----------------------------------------------------------------
def waitForClick( win, message ):
""" waitForClick: stops the GUI and displays a message.
Returns when the user clicks the window. The message is erased."""
# wait for user to click mouse to start
startMsg = Text( Point( win.getWidth()/2, win.getHeight()/2 ), message )
startMsg.draw( win ) # display message
win.getMouse() # wait
startMsg.undraw() # erase
def main():
global W, H
win = GraphWin( "wheel demo", W, H )
waitForClick( win, "click to start" )
car = Car( Point( 100,100 ), Point( 250, 170 ) )
car.draw( win )
car.setFill( "red", "yellow", "black" )
waitForClick( win, "click to end" )
main()
car3.py
# car3.py
# Now the car can move!
from graphics import *
W = 400
H = 400
#----------------------------------------------------------------
class Wheel:
"""A class with two concentric circles"""
def __init__( self, center, r1, r2 ):
self.circ1 = Circle( center, r1 )
self.circ2 = Circle( center, r2 )
r1, r2 = min( r1, r2 ), max( r1, r2 )
self.radius1 = r1
self.radius2 = r2
def draw( self, win ):
self.circ2.draw( win )
self.circ1.draw( win )
def setFill( self, color1, color2 ):
self.circ1.setFill( color1 )
self.circ2.setFill( color2 )
def getRadius1( self ):
return self.radius1
def getRadius2( self ):
return self.radius2
def move( self, dx, dy ):
self.circ1.move( dx, dy )
self.circ2.move( dx, dy )
#----------------------------------------------------------------
class Car:
"""a class containing a rectangle and 2 wheels"""
def __init__( self, P1, P2 ):
"""constructs the car. The top-left and bottom-right points
defining the body of the car are given."""
self.P1 = P1
self.P2 = P2
self.width = abs( P1.getX()-P2.getX() )
self.height= abs( P1.getY()-P2.getY() )
#--- define rectangle---
self.rect = Rectangle( P1, P2 )
#--- and the two wheels ---
center1 = Point( P1.getX()+self.width/8, P2.getY() )
center2 = Point( P1.getX()+self.width*7/8, P2.getY() )
radius2 = self.height/3
radius1 = radius2/2
self.wheel1 = Wheel( center1, radius1, radius2 )
self.wheel2 = Wheel( center2, radius1, radius2 )
def draw( self, win ):
"""draw rectangle and 2 wheels on window"""
self.rect.draw( win )
self.wheel1.draw( win )
self.wheel2.draw( win )
def setFill( self, color1, color2, color3 ):
"""defines the color of the car. First is body, then inside wheel, then tire color"""
self.rect.setFill( color1 )
self.wheel1.setFill( color2, color3 )
self.wheel2.setFill( color2, color3 )
def move( self, dx, dy ):
"""defines direction of movement for all 3 elements of car"""
self.rect.move( dx, dy )
self.wheel1.move( dx, dy )
self.wheel2.move( dx, dy )
#----------------------------------------------------------------
def waitForClick( win, message ):
""" waitForClick: stops the GUI and displays a message.
Returns when the user clicks the window. The message is erased."""
# wait for user to click mouse to start
startMsg = Text( Point( win.getWidth()/2, win.getHeight()/2 ), message )
startMsg.draw( win ) # display message
win.getMouse() # wait
startMsg.undraw() # erase
def main():
global W, H
win = GraphWin( "wheel demo", W, H )
waitForClick( win, "click to start" )
car = Car( Point( 100,100 ), Point( 250, 170 ) )
car.draw( win )
car.setFill( "red", "yellow", "black" )
for i in range( 100 ):
car.move( 2, 0 )
waitForClick( win, "click to end" )
main()
