#!/usr/bin/env python
import 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")
main_dir = os.path.split(os.path.abspath(__file__))[0]
data_dir = os.path.join(main_dir, 'data')
def main():
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_dir, "liquid.bmp")).convert()
blendimage = pygame.image.load (os.path.join (data_dir, "liquid.bmp")).convert()
screen.blit (image, (10, 10))
screen.blit (blendimage, (200, 10))
pygame.display.flip ()
pygame.key.set_repeat (500, 30)
usage()
going = True
while going:
for event in pygame.event.get ():
if event.type == QUIT:
going = False
if event.type == KEYDOWN:
usage ()
if event.key == K_ESCAPE:
going = False
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 ()
pygame.quit()
if __name__ == '__main__':
main()
|