pygame__3

# -*- coding: utf-8 -*-
"""
Created on Wed Apr 20 08:16:32 2016

@author: User
"""
import random
import pygame
from pygame.locals import *
import sys

SIZE = (640, 760)
WHITE = (255, 255, 255)
RED = (255, 0, 0)
BLUE = (40, 40, 200)
BLACK = (0, 0, 0)
TRANSPARENT = (0, 0, 0, 0)
FPS = 600
pygame.init()
surface = pygame.display.set_mode(SIZE)
screen = pygame.Surface(SIZE)
clock = pygame.time.Clock()


class Circle():

    def __init__(self, color, position, radius):
        self.surface = pygame.Surface((2*radius, 2*radius)).convert_alpha()
        self.surface.fill(TRANSPARENT)
        self.circle = pygame.draw.circle(self.surface, color,
                                         (radius, radius), radius)
        self.x, self.y = position
        self.radius = radius
        self.attached = False
        self.vx = 1.2
        self.vy = 0.8
        self.friction = 0.99

    def distanceTo(self,obj):
        return ((obj.x-self.x)**2+(obj.y-self.y)**2)**0.5

    def move(self):
        if not self.attached:
            self.vx *= self.friction
            self.vy *= self.friction
            self.x += self.vx
            self.y += self.vy

            if self.x > 500:
                self.x = 500
                self.vx = -self.vx

            if self.x < 40:
                self.x = 40
                self.vx = -self.vx

            if self.y < 40:
                self.y = 40
                self.vy = -self.vy

            if self.y > 620:
                self.y = 620
                self.vy = -self.vy

        for c in circles:
            if c != self:
                if self.distanceTo(c) < self.radius + c.radius:
                    self.vx += (self.x-c.x)/100
                    self.vy += (self.y-c.y)/100
                    #~ c.vx += -self.vx
                    #~ c.vy += -self.vy

    def draw(self, surface):
        self.rect = surface.blit(self.surface, (self.x, self.y))

    def point_in(self, point):
        return self.rect.collidepoint(point)

    def change_color(self):
        color = (int(random.random()*256),
                 int(random.random()*256),
                 int(random.random()*256))
        self.circle = pygame.draw.circle(self.surface, color,
                                        (self.radius, self.radius),
                                         self.radius)



class Platform:
    def __init__(self):
        self.img = pygame.image.load('e.jpg')


def make_level(level, platform):
    a = 0
    b = 0
    for row in level:
        for col in row:
            if col == '-':
                screen.blit(platform.img, (a, b))
            a += 20
        b += 20
        a = 0

level = [
    '-----------          -----------',
    '-                              -',
    '-                              -',
    '-                              -',
    '-                              -',
    '-                              -',
    '-                              -',
    '-                              -',
    '-                              -',
    '-                              -',
    '-                              -',
    '-                              -',
    '-                              -',
    '-                              -',
    '-                              -',
    '-                              -',
    '-                              -',
    '-                              -',
    '-                              -',
    '-                              -',
    '-                              -',
    '-                              -',
    '-                              -',
    '-                              -',
    '-                              -',
    '-                              -',
    '-                              -',
    '-                              -',
    '-                              -',
    '-                              -',
    '-                              -',
    '-                              -',
    '-                              -',
    '-                              -',
    '-                              -',
    '-                              -',
    '-                              -',
    '-----------          -----------']


circles = [Circle(WHITE, (300, 600), 50),Circle(BLUE, (320, 380), 40),Circle(RED, (320, 100), 50)]
circles.append(Circle(WHITE, (200, 600), 20))
circles.append(Circle(WHITE, (300, 500), 20))
circles[0].friction = 1

pl = Platform()

while True:
    for event in pygame.event.get():
        if event.type == QUIT:
            pygame.quit()
            sys.exit()

        elif event.type == KEYDOWN and event.key == K_SPACE:
            for c in circles:
                c.vx = 0
                c.vy = 0

        elif event.type == MOUSEBUTTONDOWN:
            for c in circles:
                if c.point_in(event.pos):
                    c.attached = True

        elif event.type == MOUSEBUTTONUP:
            for c in circles:
                c.attached = False

        elif event.type == MOUSEMOTION:
            for c in circles:
                if c.attached:
                    c.vx,c.vy = event.rel
                    c.x = event.pos[0]-c.radius
                    c.y = event.pos[1]-c.radius


    surface.fill(BLACK)
    make_level(level, pl)
    surface.blit(screen,(0, 0))

    for c in circles:
        c.move()
        c.draw(surface)

    pygame.display.update()
    clock.tick(FPS)