103 lines
3.0 KiB
Python
103 lines
3.0 KiB
Python
|
#!/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()
|