#!/usr/bin/python2.5

# In this Python program, I test my "camera" by

# drawing a coordinate axis and several rotating

# cubes.

# As much as I wanted to use PyAllegro, I had to move

# to PyGame instead; alpy just needs too much work, is

# too unstable, and is too slow in making progress.

import sys

import pygame

# This module imports all the little modules I will

# need for this program.

import math3d

from vertex import egacolor

from math import sqrt

from worm import *

# First, we need to initialize the graphics system.

pygame.init()

# size = width, height = 320, 400

size = width, height = 625, 400

xoffset = 50

yoffset = 50

offset = xoffset/2, yoffset/2

screen = pygame.display.set_mode(size)

black = 0, 0, 0

white = 255, 255, 255

bgcolor = white

## For visual reference, we will also draw a

## coordinate axis.

coords = math3d.wireframe('coords.dat')

## This is where I will be experimenting with movement...

nextscreen = 0

pygame.mouse.set_visible(False)

center_x = width/2; center_y = height/2

inner_r = 50*50; outer_r = 100*100

m_0 = (float(height)/width); m_1 =-(float(height)/width)

clockwise_image = \

pygame.image.load('../images/arrow_rotate_clockwise.png')

counter_clockwise_image = \

pygame.image.load('../images/arrow_rotate_anticlockwise.png')

fwd_image = pygame.image.load('../images/add.png')

back_image = pygame.image.load('../images/money_yen.png')

up_image = pygame.image.load('../images/arrow_up.png')

left_image = pygame.image.load('../images/arrow_left.png')

right_image = pygame.image.load('../images/arrow_right.png')

down_image = pygame.image.load('../images/arrow_down.png')

cursor_image = up_image

dest_offset = up_image.get_size()

dest_offset = (dest_offset[0]/2, dest_offset[1]/2)

#m_offset = (0, 0)

show_cursor = False

hyp = sqrt(height*height + width*width)

line_y = 100 * height / hyp

line_x = 100 * width / hyp

def direction(pos):

return RIGHT, m_offset

def cursor(direction):

return right_image

def drawScene():

pygame.display.flip()

font = pygame.font.Font(None, 24)

my_worm = worm(sx=width-xoffset, sy=height-yoffset)

useFlyCamera = False

while 1:

events = pygame.event.get()

for event in events:

if event.type == pygame.QUIT:

sys.exit()

elif event.type == pygame.KEYDOWN:

# print event

print event

if event.unicode == 'f' or event.unicode == 'F':

print 'Toggle useFlyCamera!!!'

if useFlyCamera:

useFlyCamera = False

else:

useFlyCamera = True

my_worm.cycleCamera()

elif event.unicode == 'n' or event.unicode == 'N':

nextscreen = 1

elif event.unicode == 'p' or event.unicode == 'P':

my_worm.pop()

elif event.type == pygame.MOUSEMOTION:

# print event.dict

# print event.pos, event.buttons

mouse_dir, m_offset = direction(event.pos)

if mouse_dir == BORDER:

show_cursor = False

else:

show_cursor = True

cursor_image = cursor(mouse_dir)

# print event.pos, event.button

#elif event.type == pygame.MOUSEBUTTONDOWN:

# print event.dict

elif event.type == pygame.MOUSEBUTTONDOWN:

if useFlyCamera:

print "hello!"

# i = 0

button_down = True

while button_down:

# print i,

# i = (i+1)%256

next_direction, m_offset = direction(event.pos)

my_worm.move_flyCamera(next_direction, event.button)

for up_event in pygame.event.get():

if up_event.type == pygame.MOUSEBUTTONUP:

button_down = False

drawScene()

else:

next_direction, m_offset = direction(event.pos)

my_worm.add_segment(next_direction)

# print event.pos, event.button

drawScene()

if nextscreen == 1:

break

pygame.mouse.set_visible(True)