#!/usr/bin/env python
import sys, os
import pygame
from pygame.locals import *

def usage ():
    print "Press R, G, B to increase the color channel values,"
    print "1-9 to set the step range for the increment,"
    print "A - ADD, S- SUB, M- MULT, - MIN, + MAX"
    print "  to change the blend modes"

color = [0, 0, 0]
changed = False
blendtype = 0
step = 5

pygame.init ()
screen = pygame.display.set_mode ((640, 480), 0, 32)
screen.fill ((100, 100, 100))

image = pygame.image.load (os.path.join ("data", "liquid.bmp")).convert()
blendimage = pygame.image.load (os.path.join ("data", "liquid.bmp")).convert()
screen.blit (image, (10, 10))
screen.blit (blendimage, (200, 10))

pygame.display.flip ()
pygame.key.set_repeat (500, 30)
usage()

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

        if event.type == KEYDOWN:
            usage ()

            if event.key == K_ESCAPE:
                sys.exit ()
            
            if event.key == K_r:
                color[0] += step
                if color[0] > 255:
                    color[0] = 0
                changed = True

            elif event.key == K_g:
                color[1] += step
                if color[1] > 255:
                    color[1] = 0
                changed = True

            elif event.key == K_b:
                color[2] += step
                if color[2] > 255:
                    color[2] = 0
                changed = True

            elif event.key == K_a:
                blendtype = BLEND_ADD
                changed = True
            elif event.key == K_s:
                blendtype = BLEND_SUB
                changed = True
            elif event.key == K_m:
                blendtype = BLEND_MULT
                changed = True
            elif event.key == K_PLUS:
                blendtype = BLEND_MAX
                changed = True
            elif event.key == K_MINUS:
                blendtype = BLEND_MIN
                changed = True

            elif event.key in (K_1, K_2, K_3, K_4, K_5, K_6, K_7, K_8, K_9):
                step = int (event.unicode)

        if changed:
            screen.fill ((100, 100, 100))
            screen.blit (image, (10, 10))
            blendimage.blit (image, (0, 0))
            #blendimage.fill (color, (0, 0, 20, 20), blendtype)
            blendimage.fill (color, None, blendtype)
            screen.blit (blendimage, (200, 10))
            print "Color: %s, Pixel (0,0): %s" % (tuple(color),
                                                  [blendimage.get_at ((0, 0))])
            changed = False
            pygame.display.flip ()